From 03e3cd468be163f01a8cfc774922f6981bc6dd78 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Tue, 3 Jan 2023 12:31:29 -0800 Subject: [PATCH] cr --- langchain/agents/agent.py | 2 +- langchain/schema.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/langchain/agents/agent.py b/langchain/agents/agent.py index 2b67ae0d567..811a48e3697 100644 --- a/langchain/agents/agent.py +++ b/langchain/agents/agent.py @@ -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: diff --git a/langchain/schema.py b/langchain/schema.py index 8c6d8bbea60..a4b4e6267ce 100644 --- a/langchain/schema.py +++ b/langchain/schema.py @@ -14,8 +14,8 @@ class AgentAction(NamedTuple): class AgentFinish(NamedTuple): """Agent's return value.""" - log: str return_values: dict + log: str class Generation(NamedTuple):