Skip to content

fix(runner): add timestamps to Gemini CLI messages in UI #892

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Gkrumbach07 merged 1 commit into from
Mar 13, 2026
Merged

fix(runner): add timestamps to Gemini CLI messages in UI #892

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 21 additions & 0 deletions components/runners/ambient-runner/ag_ui_gemini_cli/adapter.py
100644 → 100755
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
import json
import logging
import uuid
from datetime import datetime
from typing import AsyncIterator, Optional

from ag_ui.core import (
Expand Down Expand Up @@ -40,6 +41,15 @@
logger = logging.getLogger(__name__)


def _iso_to_ms(iso_timestamp: str) -> Optional[int]:
"""Convert an ISO 8601 timestamp string to epoch milliseconds."""
try:
dt = datetime.fromisoformat(iso_timestamp.replace("Z", "+00:00"))
return int(dt.timestamp() * 1000)
except (ValueError, AttributeError):
return None


def _summarize_event(event: object) -> str:
"""One-line summary of a Gemini CLI event for logging."""
if isinstance(event, InitEvent):
Expand Down Expand Up @@ -88,6 +98,7 @@ async def run(
text_message_open = False
current_message_id: Optional[str] = None
accumulated_text = ""
message_timestamp_ms: Optional[int] = None

# Tool tracking
current_tool_call_id: Optional[str] = None
Expand Down Expand Up @@ -141,10 +152,12 @@ async def run(
# First text chunk: open a text message
if not text_message_open:
current_message_id = str(uuid.uuid4())
message_timestamp_ms = _iso_to_ms(event.timestamp)
yield TextMessageStartEvent(
type=EventType.TEXT_MESSAGE_START,
message_id=current_message_id,
role="assistant",
timestamp=message_timestamp_ms,
)
text_message_open = True
accumulated_text = ""
Expand All @@ -163,10 +176,12 @@ async def run(
if event.role == "assistant" and not event.delta:
if not text_message_open:
current_message_id = str(uuid.uuid4())
message_timestamp_ms = _iso_to_ms(event.timestamp)
yield TextMessageStartEvent(
type=EventType.TEXT_MESSAGE_START,
message_id=current_message_id,
role="assistant",
timestamp=message_timestamp_ms,
)
text_message_open = True
accumulated_text = ""
Expand All @@ -193,10 +208,12 @@ async def run(
id=current_message_id,
role="assistant",
content=accumulated_text,
timestamp=message_timestamp_ms,
)
)
current_message_id = None
accumulated_text = ""
message_timestamp_ms = None
continue

# ── tool_use ──
Expand All @@ -213,11 +230,13 @@ async def run(
id=current_message_id,
role="assistant",
content=accumulated_text,
timestamp=message_timestamp_ms,
)
)
text_message_open = False
current_message_id = None
accumulated_text = ""
message_timestamp_ms = None

current_tool_call_id = event.tool_id or str(uuid.uuid4())
yield ToolCallStartEvent(
Expand Down Expand Up @@ -297,11 +316,13 @@ async def run(
id=current_message_id,
role="assistant",
content=accumulated_text,
timestamp=message_timestamp_ms,
)
)
text_message_open = False
current_message_id = None
accumulated_text = ""
message_timestamp_ms = None

if event.status == "error":
error_msg = "Gemini CLI run failed"
Expand Down
Loading