Merge branch 'ankush/callbackhandler' into harrison/callback-updates

This commit is contained in:
Harrison Chase 2023-01-03 12:31:45 -08:00
commit 3086b752a3
2 changed files with 2 additions and 2 deletions

View File

@ -73,7 +73,7 @@ class Agent(BaseModel):
parsed_output = self._extract_tool_and_input(full_output)
tool, tool_input = parsed_output
if tool == self.finish_tool_name:
return AgentFinish(full_output, {"output": tool_input})
return AgentFinish({"output": tool_input}, full_output)
return AgentAction(tool, tool_input, full_output)
def prepare_for_new_call(self) -> None:

View File

@ -14,8 +14,8 @@ class AgentAction(NamedTuple):
class AgentFinish(NamedTuple):
"""Agent's return value."""
log: str
return_values: dict
log: str
class Generation(NamedTuple):