update api.py

2023-05-12 18:23:13 +08:00 · 2023-05-12 18:23:13 +08:00 · 5575079e6b
parent 8635f63c31
commit 5575079e6b
1 changed files with 3 additions and 3 deletions
--- 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):