Skip to content

Commit

Permalink
Update auto_swarm_builder.py
Browse files Browse the repository at this point in the history
  • Loading branch information
kyegomez authored Dec 12, 2024
1 parent e6e989d commit 770b4a1
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions swarms/structs/auto_swarm_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,13 +50,11 @@ class SwarmConfig(BaseModel):
name="Research-Agent",
description="Gathers information",
system_prompt="You are a research agent...",
max_loops=2,
),
AgentConfig(
name="Writing-Agent",
description="Writes content",
system_prompt="You are a writing agent...",
max_loops=1,
),
],
)
Expand Down Expand Up @@ -195,7 +193,7 @@ def _create_agents(self, task: str, *args, **kwargs):
self.name = agents_dictionary.name
self.description = agents_dictionary.description
self.max_loops = getattr(
agents_dictionary, "max_loops", 1
agents_dictionary
) # Default to 1 if not set

logger.info(
Expand All @@ -213,7 +211,6 @@ def _create_agents(self, task: str, *args, **kwargs):
agent_name=agent_config.name,
agent_description=agent_config.description,
agent_system_prompt=agent_config.system_prompt,
# max_loops=agent_config.max_loops,
)
agents.append(agent)

Expand Down

0 comments on commit 770b4a1

Please sign in to comment.