lfoppiano commited on
Commit
37619a9
1 Parent(s): cbdc1a4

fix signature

Browse files
Files changed (1) hide show
  1. document_qa/document_qa_engine.py +1 -1
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, memory=None, context_size=4):
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,