Fix callback forwarding in async plan method for OpenAI function agent (#6584)

rogeriochaves · web-flow · commit 3436da65a4b1 · 2023-06-22T08:18:31.000-07:00
The callback argument was missing, preventing me to get callbacks to
work properly when using it async
diff --git a/langchain/agents/openai_functions_agent/base.py b/langchain/agents/openai_functions_agent/base.py
@@ -236,7 +236,7 @@ async def aplan(
         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

Original file line number	Diff line number	Diff line change
`@@ -236,7 +236,7 @@ async def aplan(`
`236`	`236`	`prompt = self.prompt.format_prompt(**full_inputs)`
`237`	`237`	`messages = prompt.to_messages()`
`238`	`238`	`predicted_message = await self.llm.apredict_messages(`
`239`		`- messages, functions=self.functions`
	`239`	`+ messages, functions=self.functions, callbacks=callbacks`
`240`	`240`	`)`
`241`	`241`	`agent_decision = _parse_ai_message(predicted_message)`
`242`	`242`	`return agent_decision`