Skip to content

fix(google): replay Gemini realtime tool result across session restart - #6481

Open
ByteMaster-1 wants to merge 3 commits into
livekit:mainfrom
ByteMaster-1:fix/gemini-realtime-tool-update
Open

fix(google): replay Gemini realtime tool result across session restart#6481
ByteMaster-1 wants to merge 3 commits into
livekit:mainfrom
ByteMaster-1:fix/gemini-realtime-tool-update

Conversation

@ByteMaster-1

Copy link
Copy Markdown
Contributor

When update_tools() restarts the Gemini Live websocket mid-turn, a function-tool result produced in that window was sent on the closing session and never replayed, so the model never received it and the turn hung. Buffer the tool result while the socket is restarting and replay it via send_tool_response once the new session is established, then nudge the model to continue the turn.
Also decouple generation-completion handling from the model name so newer Live preview models work: track generation state explicitly via _generation_completed, drop trailing model_turn frames that arrive with no active generation, and allow generate_reply() for models without mutable_chat_context (gated on the capability, not a model-name literal). Fixes #6479

@ByteMaster-1
ByteMaster-1 requested a review from a team as a code owner July 18, 2026 14:10
devin-ai-integration[bot]

This comment was marked as resolved.

@ByteMaster-1
ByteMaster-1 force-pushed the fix/gemini-realtime-tool-update branch from b16c719 to 30dea9c Compare July 18, 2026 14:29
ronanmcnulty pushed a commit to GreyhoundRecyclingIreland/agents that referenced this pull request Jul 23, 2026
Routine + handled on our deployment: post-completion re-emissions are dropped
by design (livekit#6431/livekit#6481) and the worker's capture wrap folds the text into the
stored transcript before this handler runs. The WARNING paged Rollbar on
essentially every Gemini call (cai items livekit#1543/livekit#1544, livekit#1606/livekit#1607).
@ByteMaster-1
ByteMaster-1 force-pushed the fix/gemini-realtime-tool-update branch from 30dea9c to 87e3efe Compare July 31, 2026 09:22
devin-ai-integration[bot]

This comment was marked as resolved.

@ByteMaster-1
ByteMaster-1 force-pushed the fix/gemini-realtime-tool-update branch from 1730cca to 0523c25 Compare August 11, 2026 14:47
When update_tools() restarts the Gemini Live websocket mid-turn, a
function-tool result produced in that window was sent on the closing
session and never replayed, so the model never received it and the turn
hung. Buffer the tool result whenever there is no live socket to send on
(no active session, or a restart pending) and replay it via
send_tool_response once the new session is established. Gemini Live
auto-generates a reply after a function response, so no extra user-turn
nudge is sent. Multiple results landing during one reconnect window are
accumulated so none are dropped.

Also decouple generation-completion handling from the model name so newer
Live preview models work: track generation state explicitly via
_generation_completed, drop trailing model_turn frames that arrive with no
active generation, and allow generate_reply() for models without
mutable_chat_context (gated on the capability, not a model-name literal).

Fixes livekit#6479

Co-Authored-By: heer <heer434@users.noreply.github.com>
devin-ai-integration[bot]

This comment was marked as resolved.

Restrict tool-result buffering to a genuine restart window (track whether a
session was ever established) so the initial context sync / agent handoff no
longer resends historical function outputs. Strip only the stray model_turn
in the recv guard instead of dropping the whole message, so bundled
turn_complete/usage/go_away/resumption events are still handled.
devin-ai-integration[bot]

This comment was marked as resolved.

The guard's condition (generation done AND not _generation_completed) was
unreachable — every path that marks a generation done sets the flag True
first — so it never fired and processing was unchanged. Remove it and the
dead _generation_completed state rather than ship code that implies a
protection it never provided.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

1 participant