apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-26 15:39:40 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix tests

Browse source
This commit is contained in:
Adil Hafeez 2025-03-19 15:04:22 -07:00
parent 786bb40253
commit 00cf848a8f
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
4 changed files with 35 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
demos/use_cases/orchestrating_agents/main.py
View file

@ -26,13 +26,14 @@ class ChatCompletionsRequest(BaseModel):
openai_client = openai.OpenAI(
api_key=" -- ", # Replace with your OpenAI API key
api_key="None", # archgw picks the API key from the config file
base_url="http://host.docker.internal:12000/v1",
)
def call_openai(messages: List[Dict[str, str]], stream: bool):
completion = openai_client.chat.completions.create(
model="gpt-4o", # archgw picks the default LLM configured in the config file
model="None", # archgw picks the default LLM configured in the config file
messages=messages,
stream=stream,
)
@ -93,7 +94,12 @@ AGENTS = {
@app.post("/v1/chat/completions")
def completion_api(req: ChatCompletionsRequest, request: Request):
agent_name = req.metadata.get("agent-name", "unknown_agent")
agent = AGENTS.get(agent_name, AGENTS["unknown_agent"])
agent = AGENTS.get(agent_name)
logger.info(f"Routing to agent: {agent_name}")
return agent.handle(req)
@app.get("/healthz")
async def healthz():
return {"status": "ok"}