github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 18:53:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
The callback argument was missing, preventing me to get callbacks to
work properly when using it async
This commit is contained in:
Rogério Chaves 2023-06-22 17:18:31 +02:00 committed by GitHub
parent b909bc8b58
commit 3436da65a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/agents/openai_functions_agent/base.py
View File

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