Skip to content

Commit 2e7de49

Browse files
committed
fix agent event logger
1 parent e783720 commit 2e7de49

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/llama_stack_client/lib/agents/event_logger.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -94,18 +94,18 @@ def _get_log_event(self, chunk, previous_event_type=None, previous_step_type=Non
9494
if event_type == "step_start":
9595
yield LogEvent(role=step_type, content="", end="", color="yellow")
9696
elif event_type == "step_progress":
97-
if event.payload.tool_call_delta:
98-
if isinstance(event.payload.tool_call_delta.content, str):
97+
if event.payload.delta.type == "tool_call":
98+
if isinstance(event.payload.delta.content, str):
9999
yield LogEvent(
100100
role=None,
101-
content=event.payload.tool_call_delta.content,
101+
content=event.payload.delta.content,
102102
end="",
103103
color="cyan",
104104
)
105-
else:
105+
elif event.payload.delta.type == "text":
106106
yield LogEvent(
107107
role=None,
108-
content=event.payload.text_delta,
108+
content=event.payload.delta.text,
109109
end="",
110110
color="yellow",
111111
)

0 commit comments

Comments
 (0)