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 =