Update swarm_router.py

pull/1237/head
CI-DEV 2 days ago committed by GitHub
parent 15e897a4e0
commit 521f883a67
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -26,7 +26,7 @@ from swarms.structs.council_as_judge import CouncilAsAJudge
from swarms.structs.debate_with_judge import DebateWithJudge from swarms.structs.debate_with_judge import DebateWithJudge
from swarms.structs.groupchat import GroupChat from swarms.structs.groupchat import GroupChat
from swarms.structs.heavy_swarm import HeavySwarm from swarms.structs.heavy_swarm import HeavySwarm
from swarms.structs.hiearchical_swarm import HiearchicalSwarm from swarms.structs.hierarchical_swarm import HierarchicalSwarm
from swarms.structs.interactive_groupchat import InteractiveGroupChat from swarms.structs.interactive_groupchat import InteractiveGroupChat
from swarms.structs.ma_utils import list_all_agents from swarms.structs.ma_utils import list_all_agents
from swarms.structs.majority_voting import MajorityVoting from swarms.structs.majority_voting import MajorityVoting
@ -50,7 +50,7 @@ SwarmType = Literal[
"GroupChat", "GroupChat",
"MultiAgentRouter", "MultiAgentRouter",
"AutoSwarmBuilder", "AutoSwarmBuilder",
"HiearchicalSwarm", "HierarchicalSwarm",
"auto", "auto",
"MajorityVoting", "MajorityVoting",
"MALT", "MALT",
@ -434,7 +434,7 @@ class SwarmRouter:
"MALT": self._create_malt, "MALT": self._create_malt,
"CouncilAsAJudge": self._create_council_as_judge, "CouncilAsAJudge": self._create_council_as_judge,
"InteractiveGroupChat": self._create_interactive_group_chat, "InteractiveGroupChat": self._create_interactive_group_chat,
"HiearchicalSwarm": self._create_hierarchical_swarm, "HierarchicalSwarm": self._create_hierarchical_swarm,
"MixtureOfAgents": self._create_mixture_of_agents, "MixtureOfAgents": self._create_mixture_of_agents,
"MajorityVoting": self._create_majority_voting, "MajorityVoting": self._create_majority_voting,
"GroupChat": self._create_group_chat, "GroupChat": self._create_group_chat,
@ -471,6 +471,16 @@ class SwarmRouter:
chairman_model=self.chairman_model, chairman_model=self.chairman_model,
) )
def _create_debate_with_judge(self, *args, **kwargs):
"""Factory function for DebateWithJudge."""
return DebateWithJudge(
pro_agent=self.agents[0],
con_agent=self.agents[1],
judge_agent=self.agents[2],
max_rounds=self.max_loops,
output_type=self.output_type,
verbose=self.verbose,
)
def _create_agent_rearrange(self, *args, **kwargs): def _create_agent_rearrange(self, *args, **kwargs):
"""Factory function for AgentRearrange.""" """Factory function for AgentRearrange."""

Loading…
Cancel
Save