chef_agent

2026-09-07 13:24:09 +08:00
parent 86ece1926d
commit bc44de93bc
14 changed files with 133 additions and 63 deletions
--- a/api/agent_api.py
+++ b/api/agent_api.py
@@ -38,15 +38,10 @@ def chat(request: ChatRequest):
@agent_router.post("/chat_stream")
 def chat(request: ChatRequest):

-    def generate_stream():
-        result = ChefService.chat_stream(
+    return StreamingResponse(
+        ChefService.chat_stream(
            thread_id=request.thread_id,
            message=request.message,
-        )
-        for chunk in result:
-            yield chunk
-
-    return StreamingResponse(
-        generate_stream(),
+        ),
        media_type="text/plain; charset=utf-8"
    )