lfoppiano commited on
Commit
398556b
1 Parent(s): cdfc7ae

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
@@ -143,7 +143,7 @@ class DocumentQAEngine:
143
 
144
  return parsed_output
145
 
146
- def _run_query(self, doc_id, query, memory=None, context_size=4):
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,