This commit is contained in:
Harrison Chase 2023-01-03 12:31:29 -08:00
parent 23b8cfc123
commit 03e3cd468b
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):