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"])