diff --git a/agent_api/main.py b/agent_api/main.py index 7612a16..e3374b1 100644 --- a/agent_api/main.py +++ b/agent_api/main.py @@ -41,4 +41,25 @@ def chat(req: ChatRequest): return { "response": response, "session_id": req.session_id + } + +@app.post("/v1/chat/completions") +def chat_completions(req: dict): + messages = req["messages"] + user_message = messages[-1]["content"] + + response = run_agent(user_message) + + return { + "id": "chatcmpl-local", + "object": "chat.completion", + "choices": [ + { + "index": 0, + "message": { + "role": "assistant", + "content": response + } + } + ] } \ No newline at end of file