fix(firebaseai): fix the json parse for toolCallCancellation (#17690)

Author: cynthiajoan

packages/firebase_ai/firebase_ai/lib/src/live_api.dart

@@ -322,8 +322,13 @@ LiveServerMessage _parseServerMessage(Object jsonObject) {
 
     return LiveServerToolCall(functionCalls: functionCalls);
   } else if (json.containsKey('toolCallCancellation')) {
-    final toolCancelJson =
-        json['toolCallCancellation'] as Map<String, List<String>>;
+    final toolCancelData = json['toolCallCancellation'] as Map;
+    final Map<String, List<String>> toolCancelJson = toolCancelData.map(
+      (key, value) => MapEntry(
+        key as String,
+        (value as List).cast<String>(),
+      ),
+    );
     return LiveServerToolCallCancellation(functionIds: toolCancelJson['ids']);
   } else if (json.containsKey('setupComplete')) {
     return LiveServerSetupComplete();

packages/firebase_ai/firebase_ai/test/live_test.dart

@@ -11,6 +11,7 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
+import 'dart:convert';
 import 'dart:typed_data';
 
 import 'package:firebase_ai/src/api.dart';
@@ -206,11 +207,13 @@ void main() {
 
     test('parseServerMessage parses toolCallCancellation message correctly',
         () {
-      final jsonObject = {
-        'toolCallCancellation': {
-          'ids': ['1', '2']
+      final jsonObject = jsonDecode('''
+        {
+          "toolCallCancellation": {
+            "ids": ["1", "2"]
+          }
         }
-      };
+        ''') as Map<String, dynamic>;
       final response = parseServerResponse(jsonObject);
       expect(response.message, isA<LiveServerToolCallCancellation>());
       final cancellationMessage =