app/api/v2/chat.py
@@ -42,7 +42,7 @@ return StreamingResponse(f"data: {error_msg}\n\n", media_type="text/event-stream") session_id = session.get("data", {}).get("id") return StreamingResponse(service_chat_dialog(db, chatId, dialog.query, session_id, current_user.id, chat_info.mode), return StreamingResponse(service_chat_dialog(db, chatId, dialog.query, session_id, current_user.id, chat_info.mode, chat_info.get_kb_ids()), media_type="text/event-stream") @chat_router_v2.post("/agent/{chatId}/completions")