You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
2.2 KiB
65 lines
2.2 KiB
from swarms.structs.agent import Agent
|
|
from swarms.structs.multi_agent_router import MultiAgentRouter
|
|
|
|
# Example usage:
|
|
agents = [
|
|
Agent(
|
|
agent_name="ResearchAgent",
|
|
agent_description="Specializes in researching topics and providing detailed, factual information",
|
|
system_prompt="You are a research specialist. Provide detailed, well-researched information about any topic, citing sources when possible.",
|
|
max_loops=1,
|
|
),
|
|
Agent(
|
|
agent_name="CodeExpertAgent",
|
|
agent_description="Expert in writing, reviewing, and explaining code across multiple programming languages",
|
|
system_prompt="You are a coding expert. Write, review, and explain code with a focus on best practices and clean code principles.",
|
|
max_loops=1,
|
|
),
|
|
Agent(
|
|
agent_name="WritingAgent",
|
|
agent_description="Skilled in creative and technical writing, content creation, and editing",
|
|
system_prompt="You are a writing specialist. Create, edit, and improve written content while maintaining appropriate tone and style.",
|
|
max_loops=1,
|
|
),
|
|
]
|
|
|
|
models_to_test = [
|
|
"gpt-4.1",
|
|
"gpt-4o",
|
|
"gpt-5-mini",
|
|
"o4-mini",
|
|
"o3",
|
|
"claude-opus-4-20250514",
|
|
"claude-sonnet-4-20250514",
|
|
"claude-3-7-sonnet-20250219",
|
|
"gemini/gemini-2.5-flash",
|
|
"gemini/gemini-2.5-pro",
|
|
]
|
|
|
|
task = "Use all the agents available to you to remake the Fibonacci function in Python, providing both an explanation and code."
|
|
|
|
model_logs = []
|
|
|
|
for model_name in models_to_test:
|
|
print(f"\n--- Testing model: {model_name} ---")
|
|
router_execute = MultiAgentRouter(
|
|
agents=agents,
|
|
temperature=0.5,
|
|
model=model_name,
|
|
)
|
|
try:
|
|
result = router_execute.run(task)
|
|
print(f"Run completed successfully for {model_name}")
|
|
model_logs.append(
|
|
{"model": model_name, "status": "✅ Success"}
|
|
)
|
|
except Exception as e:
|
|
print(f"An error occurred for {model_name}")
|
|
model_logs.append(
|
|
{"model": model_name, "status": f"❌ Error: {e}"}
|
|
)
|
|
|
|
print("\n===== Model Run Summary =====")
|
|
for log in model_logs:
|
|
print(f"{log['model']}: {log['status']}")
|