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

Merge pull request #193 from langchain-ai/wfh/aexec_tool_support

Browse Source 
Add tools upport to agentexecutor
This commit is contained in:
Nuno Campos
2024-03-07 09:01:39 -08:00
committed by GitHub
parent 4569be0e02 16970aca08
commit 069ff067a5
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/prebuilt/agent_executor.py
+14
View File
@@ -72,12 +72,26 @@ def create_agent_executor(agent_runnable, tools, input_schema=None):
def execute_tools(data):
# Get the most recent agent_outcome - this is the key added in the `agent` above
agent_action = data["agent_outcome"]
if isinstance(agent_action, list):
output = tool_executor.batch(agent_action, return_exceptions=True)
return {
"intermediate_steps": [
(action, str(out)) for action, out in zip(agent_action, output)
]
}
output = tool_executor.invoke(agent_action)
return {"intermediate_steps": [(agent_action, str(output))]}
async def aexecute_tools(data):
# Get the most recent agent_outcome - this is the key added in the `agent` above
agent_action = data["agent_outcome"]
if isinstance(agent_action, list):
output = await tool_executor.abatch(agent_action, return_exceptions=True)
return {
"intermediate_steps": [
(action, str(out)) for action, out in zip(agent_action, output)
]
}
output = await tool_executor.ainvoke(agent_action)
return {"intermediate_steps": [(agent_action, str(output))]}