update api.py

2023-05-12 18:13:35 +08:00 · 2023-05-12 18:13:35 +08:00 · 8635f63c31
parent f173d8b6b1
commit 8635f63c31
1 changed files with 8 additions and 1 deletions
--- a/api.py
+++ b/api.py
@ -252,11 +252,18 @@ async def no_knowledge_chat(
        ),
 ):

-    for resp, history in local_doc_qa._call(
+    for resp, history in local_doc_qa.llm._call(
            query=question, chat_history=history, streaming=True
    ):
        pass

+    return ChatMessage(
+        question=question,
+        response=resp["result"],
+        history=history,
+        source_documents=None,
+    )
+
 async def stream_chat(websocket: WebSocket, knowledge_base_id: str):
    await websocket.accept()
    vs_path = os.path.join(VS_ROOT_PATH, knowledge_base_id)