Instantiate agents if they have not been initialized in agency chart

VRSEN · Sep 2, 2024 · 4a674d6 · 4a674d6
1 parent 2107467
commit 4a674d6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/agency_swarm/agency/agency.py b/agency_swarm/agency/agency.py
@@ -905,7 +905,17 @@ def _parse_agency_chart(self, agency_chart):
 
         if len(agency_chart) == 0:
             raise Exception("Agency chart cannot be empty.")
+
+        # instantiate agents if they are classes
+        for i, node in enumerate(agency_chart):
+            if inspect.isclass(node) and issubclass(node, Agent):
+                agency_chart[i] = node()
+            elif isinstance(node, list):
+                for j, agent in enumerate(node):
+                    if inspect.isclass(agent) and issubclass(agent, Agent):
+                        node[j] = agent()
 
+        # add agents to the agency
         for node in agency_chart:
             if isinstance(node, Agent):
                 if not self.ceo:

diff --git a/tests/test_agency.py b/tests/test_agency.py
@@ -345,12 +345,10 @@ def test_6_load_from_db(self):
             "type": "json_object",
         }
 
-        ceo = CEO()
-
         # check that agents are loaded
         agency = Agency([
-            ceo,
-            [ceo, agent1],
+            CEO,
+            [CEO, agent1],
             [agent1, agent2]],
             shared_instructions="This is a shared instruction",
             settings_path="./settings2.json",