Method _run_query

document_qa/document_qa_engine.py:460–463

(self, doc_id, query, context_size=4)

        return parsed_output

    def _run_query(self, doc_id, query, context_size=4) -> tuple[Any, list]:
        relevant_documents, relevant_document_coordinates = self._get_context(doc_id, query, context_size)
        response = self.chain.invoke({"context": relevant_documents, "question": query})
        return response, relevant_document_coordinates

    def _get_context(self, doc_id, query, context_size=4) -> tuple[List[Document], list]:
        db = self.data_storage.embeddings_dict[doc_id]