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):