github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-29 11:09:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix get_tool_return (#11346)

Browse Source
This commit is contained in:
Harrison Chase 2023-10-03 09:01:05 -07:00 committed by GitHub
parent 49b34e2293
commit 88bad37ec2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/langchain/langchain/agents/agent.py
View File

@ -1229,11 +1229,14 @@ class AgentExecutor(Chain):
"""Check if the tool is a returning tool."""
agent_action, observation = next_step_output
name_to_tool_map = {tool.name: tool for tool in self.tools}
return_value_key = "output"
if len(self.agent.return_values) > 0:
return_value_key = self.agent.return_values[0]
# Invalid tools won't be in the map, so we return False.
if agent_action.tool in name_to_tool_map:
if name_to_tool_map[agent_action.tool].return_direct:
return AgentFinish(
{self.agent.return_values[0]: observation},
{return_value_key: observation},
"",
)
return None