ids.append(_id)
        return ids
    def _get_relevant_documents(
        self, query: str, *, run_manager: CallbackManagerForRetrieverRun
    ) -> List[Document]:
        res = self.client.search(query)
        docs = []
        for r in res:
            docs.append(
                Document(
                    page_content=r["context"],
                    metadata={"source": r["source"], "document_id": r["document_id"]},
                )
            )
        return docs