From 8635f63c31c481d6b909d4cc478e5ef9792abd60 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 12 May 2023 18:13:35 +0800 Subject: [PATCH] update api.py --- api.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/api.py b/api.py index be6ea52..822288f 100644 --- 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)