Merge pull request #137 from leehom0123/master

issue fix: Call api api/local_doc_qa/local_doc_chat return data:None
netease-youdao · Feb 26, 2024 · feba5fb · feba5fb
2 parents 082fc67 + b1bad93
commit feba5fb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/qanything_kernel/connector/llm/llm_for_fastchat.py b/qanything_kernel/connector/llm/llm_for_fastchat.py
@@ -164,7 +164,10 @@ def generatorAnswer(self, prompt: str,
             history[-1] = [prompt, complete_answer]
             answer_result = AnswerResult()
             answer_result.history = history
-            answer_result.llm_output = {"answer": response_text}
+            if streaming:
+                answer_result.llm_output = {"answer": response_text}
+            else:
+                answer_result.llm_output = {"answer": complete_answer}
             answer_result.prompt = prompt
             yield answer_result