mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
f92006de3c
0.2rc migrations - [x] Move memory - [x] Move remaining retrievers - [x] graph_qa chains - [x] some dependency from evaluation code potentially on math utils - [x] Move openapi chain from `langchain.chains.api.openapi` to `langchain_community.chains.openapi` - [x] Migrate `langchain.chains.ernie_functions` to `langchain_community.chains.ernie_functions` - [x] migrate `langchain/chains/llm_requests.py` to `langchain_community.chains.llm_requests` - [x] Moving `langchain_community.cross_enoders.base:BaseCrossEncoder` -> `langchain_community.retrievers.document_compressors.cross_encoder:BaseCrossEncoder` (namespace not ideal, but it needs to be moved to `langchain` to avoid circular deps) - [x] unit tests langchain -- add pytest.mark.community to some unit tests that will stay in langchain - [x] unit tests community -- move unit tests that depend on community to community - [x] mv integration tests that depend on community to community - [x] mypy checks Other todo - [x] Make deprecation warnings not noisy (need to use warn deprecated and check that things are implemented properly) - [x] Update deprecation messages with timeline for code removal (likely we actually won't be removing things until 0.4 release) -- will give people more time to transition their code. - [ ] Add information to deprecation warning to show users how to migrate their code base using langchain-cli - [ ] Remove any unnecessary requirements in langchain (e.g., is SQLALchemy required?) --------- Co-authored-by: Erick Friis <erick@langchain.dev>
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
"""Unit tests for ReAct."""
|
|
|
|
from typing import Union
|
|
|
|
from langchain.agents.react.base import ReActChain, ReActDocstoreAgent
|
|
from langchain_core.agents import AgentAction
|
|
from langchain_core.documents import Document
|
|
from langchain_core.language_models import FakeListLLM
|
|
from langchain_core.prompts.prompt import PromptTemplate
|
|
from langchain_core.tools import Tool
|
|
|
|
from langchain_community.docstore.base import Docstore
|
|
|
|
_PAGE_CONTENT = """This is a page about LangChain.
|
|
|
|
It is a really cool framework.
|
|
|
|
What isn't there to love about langchain?
|
|
|
|
Made in 2022."""
|
|
|
|
_FAKE_PROMPT = PromptTemplate(input_variables=["input"], template="{input}")
|
|
|
|
|
|
class FakeDocstore(Docstore):
|
|
"""Fake docstore for testing purposes."""
|
|
|
|
def search(self, search: str) -> Union[str, Document]:
|
|
"""Return the fake document."""
|
|
document = Document(page_content=_PAGE_CONTENT)
|
|
return document
|
|
|
|
|
|
def test_predict_until_observation_normal() -> None:
|
|
"""Test predict_until_observation when observation is made normally."""
|
|
outputs = ["foo\nAction: Search[foo]"]
|
|
fake_llm = FakeListLLM(responses=outputs)
|
|
tools = [
|
|
Tool(name="Search", func=lambda x: x, description="foo"),
|
|
Tool(name="Lookup", func=lambda x: x, description="bar"),
|
|
]
|
|
agent = ReActDocstoreAgent.from_llm_and_tools(fake_llm, tools)
|
|
output = agent.plan([], input="")
|
|
expected_output = AgentAction("Search", "foo", outputs[0])
|
|
assert output == expected_output
|
|
|
|
|
|
def test_react_chain() -> None:
|
|
"""Test react chain."""
|
|
responses = [
|
|
"I should probably search\nAction: Search[langchain]",
|
|
"I should probably lookup\nAction: Lookup[made]",
|
|
"Ah okay now I know the answer\nAction: Finish[2022]",
|
|
]
|
|
fake_llm = FakeListLLM(responses=responses)
|
|
react_chain = ReActChain(llm=fake_llm, docstore=FakeDocstore())
|
|
output = react_chain.run("when was langchain made")
|
|
assert output == "2022"
|
|
|
|
|
|
def test_react_chain_bad_action() -> None:
|
|
"""Test react chain when bad action given."""
|
|
bad_action_name = "BadAction"
|
|
responses = [
|
|
f"I'm turning evil\nAction: {bad_action_name}[langchain]",
|
|
"Oh well\nAction: Finish[curses foiled again]",
|
|
]
|
|
fake_llm = FakeListLLM(responses=responses)
|
|
react_chain = ReActChain(llm=fake_llm, docstore=FakeDocstore())
|
|
output = react_chain.run("when was langchain made")
|
|
assert output == "curses foiled again"
|