diff --git a/chains/local_doc_qa.py b/chains/local_doc_qa.py index 1b968cc0..548b5046 100644 --- a/chains/local_doc_qa.py +++ b/chains/local_doc_qa.py @@ -133,7 +133,7 @@ class LocalDocQA: ) knowledge_chain.return_source_documents = True + result = knowledge_chain({"query": query}) - self.llm.history[-1][0] = query return result, self.llm.history