Spaces:
Running
Running
fix signature
Browse files
document_qa/document_qa_engine.py
CHANGED
@@ -143,7 +143,7 @@ class DocumentQAEngine:
|
|
143 |
|
144 |
return parsed_output
|
145 |
|
146 |
-
def _run_query(self, doc_id, query,
|
147 |
relevant_documents = self._get_context(doc_id, query, context_size)
|
148 |
if memory:
|
149 |
return self.chain.run(input_documents=relevant_documents,
|
|
|
143 |
|
144 |
return parsed_output
|
145 |
|
146 |
+
def _run_query(self, doc_id, query, context_size=4, memory=None):
|
147 |
relevant_documents = self._get_context(doc_id, query, context_size)
|
148 |
if memory:
|
149 |
return self.chain.run(input_documents=relevant_documents,
|