WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-11 04:07:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #60 from langchain-ai/bagatur/tool_exec_config

Browse Source 
patch: add config to ToolExecutor._execute
This commit is contained in:
Nuno Campos
2024-01-23 11:48:09 -08:00
committed by GitHub
parent 1e3cca4015 18b2519356
commit 10ee6f27ab
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/prebuilt/tool_executor.py
+9 -5
View File
@@ -1,7 +1,7 @@
from typing import Any, Sequence, Union
from langchain_core.load.serializable import Serializable
from langchain_core.runnables import RunnableBinding, RunnableLambda
from langchain_core.runnables import RunnableBinding, RunnableConfig, RunnableLambda
from langchain_core.tools import BaseTool
INVALID_TOOL_MSG_TEMPLATE = (
@@ -47,7 +47,9 @@ class ToolExecutor(RunnableBinding):
**kwargs,
)
def _execute(self, tool_invocation: ToolInvocationInterface) -> Any:
def _execute(
self, tool_invocation: ToolInvocationInterface, *, config: RunnableConfig
) -> Any:
if tool_invocation.tool not in self.tool_map:
return self.invalid_tool_msg_template.format(
requested_tool_name=tool_invocation.tool,
@@ -55,10 +57,12 @@ class ToolExecutor(RunnableBinding):
)
else:
tool = self.tool_map[tool_invocation.tool]
output = tool.invoke(tool_invocation.tool_input)
output = tool.invoke(tool_invocation.tool_input, config=config)
return output
async def _aexecute(self, tool_invocation: ToolInvocationInterface) -> Any:
async def _aexecute(
self, tool_invocation: ToolInvocationInterface, *, config: RunnableConfig
) -> Any:
if tool_invocation.tool not in self.tool_map:
return self.invalid_tool_msg_template.format(
requested_tool_name=tool_invocation.tool,
@@ -66,5 +70,5 @@ class ToolExecutor(RunnableBinding):
)
else:
tool = self.tool_map[tool_invocation.tool]
output = await tool.ainvoke(tool_invocation.tool_input)
output = await tool.ainvoke(tool_invocation.tool_input, config=config)
return output