chef_agent

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

api/scheduler_api.py

@@ -1,7 +1,5 @@
 from fastapi import APIRouter, HTTPException
 from typing import Any, Dict
-from core.db import SessionDep
-from core.dto.api_response import ApiResponse
 from core.task.dlt_scheduler import scheduler
 
 # 彩票相关的路由