feat(chat): add configurable streaming output

2026-07-17 13:34:27 +08:00
parent f7076569a7
commit 13fed0467a
11 changed files with 172 additions and 24 deletions
--- a/ai_knowledge_base_v2/apps/backend/app/api/chat.py
+++ b/ai_knowledge_base_v2/apps/backend/app/api/chat.py
@@ -83,7 +83,15 @@ def completions(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
 ) -> StreamingResponse:
-    return StreamingResponse(_chat_stream(payload, db, current_user), media_type="text/event-stream")
+    return StreamingResponse(
+        _chat_stream(payload, db, current_user),
+        media_type="text/event-stream",
+        headers={
+            "Cache-Control": "no-cache, no-transform",
+            "Connection": "keep-alive",
+            "X-Accel-Buffering": "no",
+        },
+    )


@router.post("/stop")