mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +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>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
import pytest
|
|
from langchain_core.documents import Document
|
|
|
|
from langchain_community.retrievers import DriaRetriever
|
|
|
|
|
|
# Set a fixture for DriaRetriever
|
|
@pytest.fixture
|
|
def dria_retriever() -> DriaRetriever:
|
|
api_key = "<YOUR_API_KEY>"
|
|
contract_id = "B16z9i3rRi0KEeibrzzMU33YTB4WDtos1vdiMBTmKgs"
|
|
retriever = DriaRetriever(api_key=api_key, contract_id=contract_id)
|
|
return retriever
|
|
|
|
|
|
def test_dria_retriever(dria_retriever: DriaRetriever) -> None:
|
|
texts = [
|
|
{
|
|
"text": "Langchain",
|
|
"metadata": {
|
|
"source": "source#1",
|
|
"document_id": "doc123",
|
|
"content": "Langchain",
|
|
},
|
|
}
|
|
]
|
|
dria_retriever.add_texts(texts)
|
|
|
|
# Assuming invoke returns a list of Document instances
|
|
docs = dria_retriever.invoke("Langchain")
|
|
|
|
# Perform assertions
|
|
assert len(docs) > 0, "Expected at least one document"
|
|
doc = docs[0]
|
|
assert isinstance(doc, Document), "Expected a Document instance"
|
|
assert isinstance(doc.page_content, str), (
|
|
"Expected document content type " "to be string"
|
|
)
|
|
assert isinstance(
|
|
doc.metadata, dict
|
|
), "Expected document metadata content to be a dictionary"
|