From 6e2ed9671f5eb6a27c09aaf164ed0720f2c88f39 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Mon, 5 Feb 2024 16:09:48 -0800 Subject: [PATCH] infra: fix breebs test lint (#17075) --- .../tests/integration_tests/retrievers/test_breebs.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/libs/community/tests/integration_tests/retrievers/test_breebs.py b/libs/community/tests/integration_tests/retrievers/test_breebs.py index c5c75fcdd33..1dd6b470a57 100644 --- a/libs/community/tests/integration_tests/retrievers/test_breebs.py +++ b/libs/community/tests/integration_tests/retrievers/test_breebs.py @@ -1,6 +1,5 @@ from typing import List -from langchain_core.callbacks.manager import CallbackManagerForRetrieverRun from langchain_core.documents import Document from langchain_community.retrievers.breebs import BreebsRetriever @@ -11,9 +10,7 @@ class TestBreebsRetriever: breeb_key = "Parivoyage" query = "What are the best churches to visit in Paris?" breeb_retriever = BreebsRetriever(breeb_key) - documents: List[Document] = breeb_retriever._get_relevant_documents( - query, run_manager=CallbackManagerForRetrieverRun - ) + documents: List[Document] = breeb_retriever.get_relevant_documents(query) assert isinstance(documents, list), "Documents should be a list" for doc in documents: assert doc.page_content, "Document page_content should not be None"