search_engine_chat bug

2023-08-24 17:25:54 +08:00 · 2023-08-24 17:25:54 +08:00 · 2a2b95d214
parent 63ca393731
commit 2a2b95d214
2 changed files with 1 additions and 1 deletions
--- a/common/init.py
+++ b/common/init.py
--- a/server/chat/search_engine_chat.py
+++ b/server/chat/search_engine_chat.py
@ -121,7 +121,7 @@ def search_engine_chat(query: str = Body(..., description="用户输入", exampl
            answer = ""
            async for token in callback.aiter():
                answer += token
-            yield json.dumps({"answer": token,
+            yield json.dumps({"answer": answer,
                              "docs": source_documents},
                             ensure_ascii=False)
        await task