langgraph[fix]: pass config to tools (#2724)

Fixes #2723
langchain-ai · Dec 11, 2024 · f642fb6 · f642fb6
1 parent ff3bc2f
commit f642fb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/langgraph/langgraph/prebuilt/tool_node.py b/libs/langgraph/langgraph/prebuilt/tool_node.py
@@ -297,7 +297,7 @@ def _run_one(
 
         try:
             input = {**call, **{"type": "tool_call"}}
-            response = self.tools_by_name[call["name"]].invoke(input)
+            response = self.tools_by_name[call["name"]].invoke(input, config)
 
         # GraphInterrupt is a special exception that will always be raised.
         # It can be triggered in the following scenarios:
@@ -352,7 +352,7 @@ async def _arun_one(
 
         try:
             input = {**call, **{"type": "tool_call"}}
-            response = await self.tools_by_name[call["name"]].ainvoke(input)
+            response = await self.tools_by_name[call["name"]].ainvoke(input, config)
 
         # GraphInterrupt is a special exception that will always be raised.
         # It can be triggered in the following scenarios: