mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-04 06:03:31 +00:00
c010ec8b71
- `.get_relevant_documents(query)` -> `.invoke(query)`
- `.get_relevant_documents(query=query)` -> `.invoke(query)`
- `.get_relevant_documents(query, callbacks=callbacks)` ->
`.invoke(query, config={"callbacks": callbacks})`
- `.get_relevant_documents(query, **kwargs)` -> `.invoke(query,
**kwargs)`
---------
Co-authored-by: Erick Friis <erick@langchain.dev>
17 lines
410 B
Python
17 lines
410 B
Python
import os
|
|
|
|
from langchain_community.retrievers.you import YouRetriever
|
|
|
|
|
|
class TestYouRetriever:
|
|
@classmethod
|
|
def setup_class(cls) -> None:
|
|
if not os.getenv("YDC_API_KEY"):
|
|
raise ValueError("YDC_API_KEY environment variable is not set")
|
|
|
|
def test_invoke(self) -> None:
|
|
retriever = YouRetriever()
|
|
actual = retriever.invoke("test")
|
|
|
|
assert len(actual) > 0
|