Merge 22afe4b24b into 458b4921d2

swarms/structs/auto_swarm_builder.py

@@ -408,6 +408,10 @@ class AutoSwarmBuilder:
 
             agents_dictionary = model.run(task)
             
+            # Parse JSON string if needed
+            if isinstance(agents_dictionary, str):
+                agents_dictionary = json.loads(agents_dictionary)
+
             return agents_dictionary
 
         except Exception as e:
@@ -438,6 +442,10 @@ class AutoSwarmBuilder:
                 f"Create the swarm spec for the following task: {task}"
             )
             
+            # Parse JSON string if needed
+            if isinstance(swarm_spec, str):
+                swarm_spec = json.loads(swarm_spec)
+
             print(swarm_spec)
 
             print(type(swarm_spec))
@@ -504,7 +512,15 @@ class AutoSwarmBuilder:
             if isinstance(agent_config, dict):
                 agent_config = AgentSpec(**agent_config)
             
-            agent = Agent(**agent_config)
+            # Convert AgentSpec to dict for Agent constructor
+            if hasattr(agent_config, 'dict'):
+                agent_dict = agent_config.dict()
+            elif hasattr(agent_config, 'model_dump'):
+                agent_dict = agent_config.model_dump()
+            else:
+                agent_dict = agent_config
+
+            agent = Agent(**agent_dict)
             agents.append(agent)
 
         return agents
@@ -535,6 +551,8 @@ class AutoSwarmBuilder:
 
             if self.execution_type == "return-agents":
                 return self.create_agents(task)
+            elif self.execution_type == "execute-swarm-router":
+                return self._execute_task(task)
             elif self.execution_type == "return-swarm-router-config":
                 return self.create_router_config(task)
             elif self.execution_type == "return-agents-objects":