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