phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

temp debug

Browse source
This commit is contained in:
Xi Yan 2025-02-20 16:30:27 -08:00
parent fb0d992f99
commit 96c521ada6
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

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