From 96c521ada631ebacbecf238acac552b2af46a46c Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 20 Feb 2025 16:30:27 -0800 Subject: [PATCH] temp debug --- .../providers/inline/agents/meta_reference/agent_instance.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index 743b77adf..8da3f3a14 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -225,6 +225,7 @@ class ChatAgent(ShieldRunnerMixin): completed_at=datetime.now(), steps=steps, ) + await self.storage.add_turn_to_session(request.session_id, turn) if output_message.tool_calls: chunk = AgentTurnResponseStreamChunk( @@ -242,8 +243,6 @@ class ChatAgent(ShieldRunnerMixin): ) ) ) - # only add to storage if turn is complete - await self.storage.add_turn_to_session(request.session_id, turn) yield chunk