diff --git a/langchain/agents/agent.py b/langchain/agents/agent.py index fa52fe1655..0587228f01 100644 --- a/langchain/agents/agent.py +++ b/langchain/agents/agent.py @@ -363,6 +363,10 @@ class AgentExecutor(Chain, BaseModel): else: return self.agent.return_values + def lookup_tool(self, name: str) -> BaseTool: + """Lookup tool by name.""" + return {tool.name: tool for tool in self.tools}[name] + def _should_continue(self, iterations: int) -> bool: if self.max_iterations is None: return True diff --git a/tests/unit_tests/agents/test_agent.py b/tests/unit_tests/agents/test_agent.py index e9d0831779..7db3894d95 100644 --- a/tests/unit_tests/agents/test_agent.py +++ b/tests/unit_tests/agents/test_agent.py @@ -289,3 +289,25 @@ def test_agent_with_new_prefix_suffix() -> None: prompt_str = agent.agent.llm_chain.prompt.template assert prompt_str.startswith(prefix), "Prompt does not start with prefix" assert prompt_str.endswith(suffix), "Prompt does not end with suffix" + + +def test_agent_lookup_tool() -> None: + """Test agent lookup tool.""" + fake_llm = FakeListLLM( + responses=["FooBarBaz\nAction: Search\nAction Input: misalignment"] + ) + tools = [ + Tool( + name="Search", + func=lambda x: x, + description="Useful for searching", + return_direct=True, + ), + ] + agent = initialize_agent( + tools=tools, + llm=fake_llm, + agent="zero-shot-react-description", + ) + + assert agent.lookup_tool("Search") == tools[0]