Add another missing aput

run-llama · Dec 27, 2024 · b29230e · b29230e
1 parent 3ab3e26
commit b29230e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama-index-integrations/agent/llama-index-agent-openai/llama_index/agent/openai/step.py b/llama-index-integrations/agent/llama-index-agent-openai/llama_index/agent/openai/step.py
@@ -516,7 +516,7 @@ async def _acall_function(
 
             event.on_end(payload={EventPayload.FUNCTION_OUTPUT: str(tool_output)})
         sources.append(tool_output)
-        memory.put(function_message)
+        await memory.aput(function_message)
 
         return (
             tool.metadata.return_direct and not tool_output.is_error