From 3436da65a4b1549d47bebefb265da41b60e10dde Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rog=C3=A9rio=20Chaves?= Date: Thu, 22 Jun 2023 17:18:31 +0200 Subject: [PATCH] Fix callback forwarding in async plan method for OpenAI function agent (#6584) The callback argument was missing, preventing me to get callbacks to work properly when using it async --- langchain/agents/openai_functions_agent/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/agents/openai_functions_agent/base.py b/langchain/agents/openai_functions_agent/base.py index 704cbf04..bcfb234e 100644 --- a/langchain/agents/openai_functions_agent/base.py +++ b/langchain/agents/openai_functions_agent/base.py @@ -236,7 +236,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent): prompt = self.prompt.format_prompt(**full_inputs) messages = prompt.to_messages() predicted_message = await self.llm.apredict_messages( - messages, functions=self.functions + messages, functions=self.functions, callbacks=callbacks ) agent_decision = _parse_ai_message(predicted_message) return agent_decision