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>
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
from typing import Any, Dict
|
|
|
|
from langchain_core.documents import Document
|
|
from pytest_mock import MockerFixture
|
|
|
|
from langchain_community.retrievers import RemoteLangChainRetriever
|
|
|
|
|
|
class MockResponse:
|
|
def __init__(self, json_data: Dict, status_code: int):
|
|
self.json_data = json_data
|
|
self.status_code = status_code
|
|
|
|
def json(self) -> Dict:
|
|
return self.json_data
|
|
|
|
|
|
def mocked_requests_post(*args: Any, **kwargs: Any) -> MockResponse:
|
|
return MockResponse(
|
|
json_data={
|
|
"response": [
|
|
{
|
|
"page_content": "I like apples",
|
|
"metadata": {
|
|
"test": 0,
|
|
},
|
|
},
|
|
{
|
|
"page_content": "I like pineapples",
|
|
"metadata": {
|
|
"test": 1,
|
|
},
|
|
},
|
|
]
|
|
},
|
|
status_code=200,
|
|
)
|
|
|
|
|
|
def test_RemoteLangChainRetriever_invoke(
|
|
mocker: MockerFixture,
|
|
) -> None:
|
|
mocker.patch("requests.post", side_effect=mocked_requests_post)
|
|
|
|
remote_langchain_retriever = RemoteLangChainRetriever(
|
|
url="http://localhost:8000",
|
|
)
|
|
response = remote_langchain_retriever.invoke("I like apples")
|
|
want = [
|
|
Document(page_content="I like apples", metadata={"test": 0}),
|
|
Document(page_content="I like pineapples", metadata={"test": 1}),
|
|
]
|
|
|
|
assert len(response) == len(want)
|
|
for r, w in zip(response, want):
|
|
assert r.page_content == w.page_content
|
|
assert r.metadata == w.metadata
|
|
|
|
|
|
# TODO: _ainvoke test
|