diff --git a/agent_api/main.py b/agent_api/main.py index e3374b1..55d62c3 100644 --- a/agent_api/main.py +++ b/agent_api/main.py @@ -43,6 +43,25 @@ def chat(req: ChatRequest): "session_id": req.session_id } + +@app.get("/v1/models") +def list_models(): + return { + "object": "list", + "data": [ + { + "id": "agent-model", + "object": "model", + "created": 0, + "owned_by": "local-agent" + } + ] + } + +@app.get("/models") +def legacy_models(): + return list_models() + @app.post("/v1/chat/completions") def chat_completions(req: dict): messages = req["messages"]