class MyPathWayRetriever(BaseRetriever):
def _get_relevant_documents(
self, query: str, *, run_manager: CallbackManagerForRetrieverRun
) -> List[Document]:
# docs = client.query(query)
documents = []
# for doc in docs:
# documents.append(Document(page_content=doc.get(
# "text"), metadata=doc.get("metadata")))
return documents
retriever = MyPathWayRetriever()
documents = retriever.invoke("what is bitcoin")
print(documents)this gets logged but this doesnt
class MyPathWayRetriever(BaseRetriever):
def _get_relevant_documents(
self, query: str, *, run_manager: CallbackManagerForRetrieverRun
) -> List[Document]:
docs = client.query(query)
documents = []
for doc in docs:
documents.append(Document(page_content=doc.get(
"text"), metadata=doc.get("metadata")))
return documents
retriever = MyPathWayRetriever()
documents = retriever.invoke("what is bitcoin")
print(documents)