From 5575079e6be893d2c9cdbe2227c62409d147ebf7 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 12 May 2023 18:23:13 +0800 Subject: [PATCH] update api.py --- api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api.py b/api.py index 822288f..c1a590b 100644 --- a/api.py +++ b/api.py @@ -253,15 +253,15 @@ async def no_knowledge_chat( ): for resp, history in local_doc_qa.llm._call( - query=question, chat_history=history, streaming=True + prompt=question, history=history, streaming=True ): pass return ChatMessage( question=question, - response=resp["result"], + response=resp, history=history, - source_documents=None, + source_documents=[], ) async def stream_chat(websocket: WebSocket, knowledge_base_id: str):