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
multi_strategy_parser
Rogério Chaves 11 months ago committed by GitHub
parent b909bc8b58
commit 3436da65a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save