From 8de1b4c4c20ea81f44628a1c42fbc1bbfff37520 Mon Sep 17 00:00:00 2001 From: Ankush Gola <9536492+agola11@users.noreply.github.com> Date: Fri, 5 May 2023 00:52:16 -0700 Subject: [PATCH] Revert "fix: #4128 missing run_manager parameter" (#4159) Reverts hwchase17/langchain#4130 --- langchain/agents/agent.py | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/langchain/agents/agent.py b/langchain/agents/agent.py index 0784d54d..fbb1ff6c 100644 --- a/langchain/agents/agent.py +++ b/langchain/agents/agent.py @@ -957,11 +957,7 @@ class AgentExecutor(Chain): run_manager=run_manager, ) if isinstance(next_step_output, AgentFinish): - return await self._areturn( - next_step_output, - intermediate_steps, - run_manager=run_manager, - ) + return await self._areturn(next_step_output, intermediate_steps) intermediate_steps.extend(next_step_output) if len(next_step_output) == 1: @@ -969,9 +965,7 @@ class AgentExecutor(Chain): # See if tool should return directly tool_return = self._get_tool_return(next_step_action) if tool_return is not None: - return await self._areturn( - tool_return, intermediate_steps, run_manager=run_manager - ) + return await self._areturn(tool_return, intermediate_steps) iterations += 1 time_elapsed = time.time() - start_time @@ -986,9 +980,7 @@ class AgentExecutor(Chain): output = self.agent.return_stopped_response( self.early_stopping_method, intermediate_steps, **inputs ) - return await self._areturn( - output, intermediate_steps, run_manager=run_manager - ) + return await self._areturn(output, intermediate_steps) def _get_tool_return( self, next_step_output: Tuple[AgentAction, str]