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.
52 lines
1.2 KiB
52 lines
1.2 KiB
#!/usr/bin/env python3
|
|
"""
|
|
Basic Graph Workflow Example
|
|
|
|
A minimal example showing how to use GraphWorkflow with backend selection.
|
|
"""
|
|
|
|
from swarms.structs.graph_workflow import GraphWorkflow
|
|
from swarms.structs.agent import Agent
|
|
|
|
agent_one = Agent(agent_name="research_agent", model="gpt-4o-mini")
|
|
agent_two = Agent(
|
|
agent_name="research_agent_two", model="gpt-4o-mini"
|
|
)
|
|
agent_three = Agent(
|
|
agent_name="research_agent_three", model="gpt-4o-mini"
|
|
)
|
|
|
|
|
|
def main():
|
|
"""
|
|
Run a basic graph workflow example without print statements.
|
|
"""
|
|
# Create agents
|
|
|
|
# Create workflow with backend selection
|
|
workflow = GraphWorkflow(
|
|
name="Basic Example",
|
|
verbose=True,
|
|
)
|
|
|
|
# Add agents to workflow
|
|
workflow.add_node(agent_one)
|
|
workflow.add_node(agent_two)
|
|
workflow.add_node(agent_three)
|
|
|
|
# Create simple chain using the actual agent names
|
|
workflow.add_edge("research_agent", "research_agent_two")
|
|
workflow.add_edge("research_agent_two", "research_agent_three")
|
|
|
|
# Compile the workflow
|
|
workflow.compile()
|
|
|
|
# Run the workflow
|
|
task = "Complete a simple task"
|
|
results = workflow.run(task)
|
|
return results
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|