Skip to content

feat(firebase_ai): add LiveServerGoAway message for session termination #17843

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
SelaseKay merged 5 commits into from
Jan 19, 2026
Merged

feat(firebase_ai): add LiveServerGoAway message for session termination #17843

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
c498d76
feat(firebase_ai): add LiveServerGoAway message for session termination
SelaseKay Nov 4, 2025
9cce6cb
feat(firebase_ai): add test for parsing goAway message in LiveServer …
SelaseKay Dec 2, 2025
53c91a6
Merge branch 'main' into feat-live-server-go-away
SelaseKay Dec 2, 2025
042d9b1
chore: rename LiveServerGoAway to GoingAwayNotice and update related …
SelaseKay Jan 16, 2026
f2d0a50
Merge branch 'main' into feat-live-server-go-away
SelaseKay Jan 16, 2026
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
12 changes: 12 additions & 0 deletions packages/firebase_ai/firebase_ai/lib/src/live_api.dart
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -209,6 +209,15 @@ class LiveServerToolCallCancellation implements LiveServerMessage {
final List<String>? functionIds;
}

/// A message indicating that the live server is going away.
class LiveServerGoAway implements LiveServerMessage {
Copy link
Copy Markdown
Collaborator

@cynthiajoan cynthiajoan Jan 15, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

with the API review, let's update the name to the follow

/// A server message indicating that the server will not be able to service the
/// client soon.
class GoingAwayNotice implements LiveServerMessage {
/// Creates a [GoingAwayNotice] instance.
///
/// [timeLeft] (optional): The remaining time before the connection will be
/// terminated.
GoingAwayNotice({this.timeLeft});

/// The remaining time before the connection will be terminated as ABORTED.
final Duration timeLeft;
}

Besides that, PR lgtm

/// Creates a [LiveServerGoAway] instance.
const LiveServerGoAway({this.timeLeft});

/// The time left in seconds for the live session to be terminated.
final String? timeLeft;
}

/// A single response chunk received during a live content generation.
///
/// It can contain generated content, function calls to be executed, or
Expand Down Expand Up @@ -435,6 +444,9 @@ LiveServerMessage _parseServerMessage(Object jsonObject) {
return LiveServerToolCallCancellation(functionIds: toolCancelJson['ids']);
} else if (json.containsKey('setupComplete')) {
return LiveServerSetupComplete();
} else if (json.containsKey('goAway')) {
final goAwayJson = json['goAway'] as Map;
return LiveServerGoAway(timeLeft: goAwayJson['timeLeft'] as String?);
} else {
throw unhandledFormat('LiveServerMessage', json);
}
Expand Down
10 changes: 10 additions & 0 deletions packages/firebase_ai/firebase_ai/test/live_test.dart
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -228,6 +228,16 @@ void main() {
expect(response.message, isA<LiveServerSetupComplete>());
});

test('parseServerMessage parses goAway message correctly', () {
final jsonObject = {
'goAway': {'timeLeft': '50s'}
};
final response = parseServerResponse(jsonObject);
expect(response.message, isA<LiveServerGoAway>());
final goAwayMessage = response.message as LiveServerGoAway;
expect(goAwayMessage.timeLeft, '50s');
});

test('parseServerMessage throws VertexAIException for error message', () {
final jsonObject = {'error': {}};
expect(() => parseServerResponse(jsonObject),
Expand Down
Loading