github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-20 13:54:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

infra: fix breebs test lint (#17075)

Browse Source
This commit is contained in:
Bagatur 2024-02-05 16:09:48 -08:00 committed by GitHub
parent cf01fc3790
commit 6e2ed9671f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/community/tests/integration_tests/retrievers/test_breebs.py
View File

@ -1,6 +1,5 @@
from typing import List from typing import List
from langchain_core.callbacks.manager import CallbackManagerForRetrieverRun
from langchain_core.documents import Document from langchain_core.documents import Document
from langchain_community.retrievers.breebs import BreebsRetriever from langchain_community.retrievers.breebs import BreebsRetriever
@ -11,9 +10,7 @@ class TestBreebsRetriever:
breeb_key = "Parivoyage" breeb_key = "Parivoyage"
query = "What are the best churches to visit in Paris?" query = "What are the best churches to visit in Paris?"
breeb_retriever = BreebsRetriever(breeb_key) breeb_retriever = BreebsRetriever(breeb_key)
documents: List[Document] = breeb_retriever._get_relevant_documents( documents: List[Document] = breeb_retriever.get_relevant_documents(query)
query, run_manager=CallbackManagerForRetrieverRun
)
assert isinstance(documents, list), "Documents should be a list" assert isinstance(documents, list), "Documents should be a list"
for doc in documents: for doc in documents:
assert doc.page_content, "Document page_content should not be None" assert doc.page_content, "Document page_content should not be None"