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

datetime nit

Browse source
This commit is contained in:
Xi Yan 2025-02-21 11:42:18 -08:00
parent b1b45ed320
commit ea050f7fa8
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -280,6 +280,7 @@ class ChatAgent(ShieldRunnerMixin):
in_progress_tool_call_step = await self.storage.get_in_progress_tool_call_step( in_progress_tool_call_step = await self.storage.get_in_progress_tool_call_step(
request.session_id, request.turn_id request.session_id, request.turn_id
) )
now = datetime.now()
tool_execution_step = ToolExecutionStep( tool_execution_step = ToolExecutionStep(
step_id=(in_progress_tool_call_step.step_id if in_progress_tool_call_step else str(uuid.uuid4())), step_id=(in_progress_tool_call_step.step_id if in_progress_tool_call_step else str(uuid.uuid4())),
turn_id=request.turn_id, turn_id=request.turn_id,
@ -292,8 +293,8 @@ class ChatAgent(ShieldRunnerMixin):
) )
for x in request.tool_responses for x in request.tool_responses
], ],
completed_at=datetime.now(), completed_at=now,
started_at=(in_progress_tool_call_step.started_at if in_progress_tool_call_step else datetime.now()), started_at=(in_progress_tool_call_step.started_at if in_progress_tool_call_step else now),
) )
steps.append(tool_execution_step) steps.append(tool_execution_step)
yield AgentTurnResponseStreamChunk( yield AgentTurnResponseStreamChunk(