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

SelaseKay · web-flow · commit e9ffbad814f5 · 2026-01-19T13:33:14.000Z
* feat(firebase_ai): add LiveServerGoAway message for session termination

* feat(firebase_ai): add test for parsing goAway message in LiveServer response

* chore: rename LiveServerGoAway to GoingAwayNotice and update related tests
diff --git a/packages/firebase_ai/firebase_ai/lib/firebase_ai.dart b/packages/firebase_ai/firebase_ai/lib/firebase_ai.dart
@@ -107,6 +107,7 @@ export 'src/live_api.dart'
         LiveServerToolCall,
         LiveServerToolCallCancellation,
         LiveServerResponse,
+        GoingAwayNotice,
         Transcription;
 export 'src/live_session.dart' show LiveSession;
 export 'src/schema.dart' show Schema, SchemaType;
diff --git a/packages/firebase_ai/firebase_ai/lib/src/live_api.dart b/packages/firebase_ai/firebase_ai/lib/src/live_api.dart
@@ -209,6 +209,19 @@ class LiveServerToolCallCancellation implements LiveServerMessage {
   final List<String>? functionIds;
 }
 
+/// 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.
+  const GoingAwayNotice({this.timeLeft});
+
+  /// The remaining time before the connection will be terminated as ABORTED.
+  final String? timeLeft;
+}
+
 /// A single response chunk received during a live content generation.
 ///
 /// It can contain generated content, function calls to be executed, or
@@ -435,6 +448,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 GoingAwayNotice(timeLeft: goAwayJson['timeLeft'] as String?);
   } else {
     throw unhandledFormat('LiveServerMessage', json);
   }
diff --git a/packages/firebase_ai/firebase_ai/test/live_test.dart b/packages/firebase_ai/firebase_ai/test/live_test.dart
@@ -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<GoingAwayNotice>());
+      final goAwayMessage = response.message as GoingAwayNotice;
+      expect(goAwayMessage.timeLeft, '50s');
+    });
+
     test('parseServerMessage throws VertexAIException for error message', () {
       final jsonObject = {'error': {}};
       expect(() => parseServerResponse(jsonObject),
