update function name

langchain-ai · Apr 10, 2024 · 8a2e364 · 8a2e364
1 parent 37e72fc
commit 8a2e364
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/langchain_benchmarks/tool_usage/agents/tool_using_agent.py b/langchain_benchmarks/tool_usage/agents/tool_using_agent.py
@@ -2,7 +2,7 @@
 
 This is useful for agents that follow the standard LangChain tool format.
 """
-from langchain.agents import AgentExecutor, create_tools_agent
+from langchain.agents import AgentExecutor, create_tool_calling_agent
 from langchain_core.language_models import BaseChatModel
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.runnables import Runnable
@@ -56,7 +56,7 @@ def __call__(self) -> Runnable:
         else:
             finalized_prompt = self.prompt
 
-        agent = create_tools_agent(self.model, env.tools, finalized_prompt)
+        agent = create_tool_calling_agent(self.model, env.tools, finalized_prompt)
 
         executor = AgentExecutor(
             agent=agent,