forked from Archives/langchain
Harrison/error zero tools (#6340)
Co-authored-by: Juhee Kim <46583939+juppytt@users.noreply.github.com>
This commit is contained in:
parent
eef62bf4e9
commit
42a28ac1ba
@ -124,6 +124,10 @@ class ZeroShotAgent(Agent):
|
||||
@classmethod
|
||||
def _validate_tools(cls, tools: Sequence[BaseTool]) -> None:
|
||||
validate_tools_single_input(cls.__name__, tools)
|
||||
if len(tools) == 0:
|
||||
raise ValueError(
|
||||
f"Got no tools for {cls.__name__}. At least one tool must be provided."
|
||||
)
|
||||
for tool in tools:
|
||||
if tool.description is None:
|
||||
raise ValueError(
|
||||
|
@ -3,12 +3,19 @@ from tempfile import TemporaryDirectory
|
||||
|
||||
from langchain.agents.agent_types import AgentType
|
||||
from langchain.agents.initialize import initialize_agent, load_agent
|
||||
from langchain.agents.tools import Tool
|
||||
from langchain.llms.fake import FakeListLLM
|
||||
|
||||
|
||||
def test_mrkl_serialization() -> None:
|
||||
agent = initialize_agent(
|
||||
[],
|
||||
[
|
||||
Tool(
|
||||
name="Test tool",
|
||||
func=lambda x: x,
|
||||
description="Test description",
|
||||
)
|
||||
],
|
||||
FakeListLLM(responses=[]),
|
||||
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
||||
verbose=True,
|
||||
|
Loading…
Reference in New Issue
Block a user