|
|
@ -3,12 +3,19 @@ from tempfile import TemporaryDirectory
|
|
|
|
|
|
|
|
|
|
|
|
from langchain.agents.agent_types import AgentType
|
|
|
|
from langchain.agents.agent_types import AgentType
|
|
|
|
from langchain.agents.initialize import initialize_agent, load_agent
|
|
|
|
from langchain.agents.initialize import initialize_agent, load_agent
|
|
|
|
|
|
|
|
from langchain.agents.tools import Tool
|
|
|
|
from langchain.llms.fake import FakeListLLM
|
|
|
|
from langchain.llms.fake import FakeListLLM
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_mrkl_serialization() -> None:
|
|
|
|
def test_mrkl_serialization() -> None:
|
|
|
|
agent = initialize_agent(
|
|
|
|
agent = initialize_agent(
|
|
|
|
[],
|
|
|
|
[
|
|
|
|
|
|
|
|
Tool(
|
|
|
|
|
|
|
|
name="Test tool",
|
|
|
|
|
|
|
|
func=lambda x: x,
|
|
|
|
|
|
|
|
description="Test description",
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
],
|
|
|
|
FakeListLLM(responses=[]),
|
|
|
|
FakeListLLM(responses=[]),
|
|
|
|
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
|
|
|
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
|
|
|
verbose=True,
|
|
|
|
verbose=True,
|
|
|
|