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

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(
request.session_id, request.turn_id
)
now = datetime.now()
tool_execution_step = ToolExecutionStep(
step_id=(in_progress_tool_call_step.step_id if in_progress_tool_call_step else str(uuid.uuid4())),
turn_id=request.turn_id,
@ -292,8 +293,8 @@ class ChatAgent(ShieldRunnerMixin):
)
for x in request.tool_responses
],
completed_at=datetime.now(),
started_at=(in_progress_tool_call_step.started_at if in_progress_tool_call_step else datetime.now()),
completed_at=now,
started_at=(in_progress_tool_call_step.started_at if in_progress_tool_call_step else now),
)
steps.append(tool_execution_step)
yield AgentTurnResponseStreamChunk(