/ examples / routing / routellm_workflow.py
routellm_workflow.py
 1  """
 2  RouteLLM Workflow Example
 3  
 4  Prerequisites:
 5  1. pip install routellm
 6  2. export OPENAI_API_KEY=your-api-key
 7  3. Start RouteLLM server:
 8     python -m routellm.openai_server \
 9       --routers mf \
10       --strong-model gpt-4o \
11       --weak-model gpt-4o-mini \
12       --port 6060
13  """
14  
15  from praisonaiagents import Agent, Workflow
16  
17  ROUTELLM_URL = "http://localhost:6060/v1"
18  
19  workflow = AgentFlow(
20      name="Analysis Pipeline",
21      steps=[
22          Agent(
23              name="Analyzer",
24              role="Data analyst",
25              goal="Analyze data and extract insights",
26              llm="router-mf-0.5",
27              base_url=ROUTELLM_URL
28          ),
29          Agent(
30              name="Reporter",
31              role="Report writer",
32              goal="Create clear reports from analysis",
33              llm="router-mf-0.5",
34              base_url=ROUTELLM_URL
35          )
36      ]
37  )
38  
39  result = workflow.run("Analyze the impact of AI on software development")
40  print(result["output"])