Move helper methods to align with Vertex api.dart

Author: andrewheard

packages/firebase_ai/firebase_ai/lib/src/developer/api.dart

@@ -40,31 +40,6 @@ import '../content.dart'
 import '../error.dart';
 import '../tool.dart' show Tool, ToolConfig;
 
-ContentModality _parseContentModality(Object jsonObject) {
-  return switch (jsonObject) {
-    'MODALITY_UNSPECIFIED' => ContentModality.unspecified,
-    'TEXT' => ContentModality.text,
-    'IMAGE' => ContentModality.image,
-    'VIDEO' => ContentModality.video,
-    'AUDIO' => ContentModality.audio,
-    'DOCUMENT' => ContentModality.document,
-    _ => throw unhandledFormat('ContentModality', jsonObject),
-  };
-}
-
-ModalityTokenCount _parseModalityTokenCount(Object? jsonObject) {
-  if (jsonObject is! Map) {
-    throw unhandledFormat('ModalityTokenCount', jsonObject);
-  }
-  final modality = _parseContentModality(jsonObject['modality']);
-
-  if (jsonObject.containsKey('tokenCount')) {
-    return ModalityTokenCount(modality, jsonObject['tokenCount'] as int);
-  } else {
-    return ModalityTokenCount(modality, 0);
-  }
-}
-
 HarmProbability _parseHarmProbability(Object jsonObject) =>
     switch (jsonObject) {
       'UNSPECIFIED' => HarmProbability.unknown,
@@ -298,6 +273,31 @@ UsageMetadata _parseUsageMetadata(Object jsonObject) {
   );
 }
 
+ModalityTokenCount _parseModalityTokenCount(Object? jsonObject) {
+  if (jsonObject is! Map) {
+    throw unhandledFormat('ModalityTokenCount', jsonObject);
+  }
+  final modality = _parseContentModality(jsonObject['modality']);
+
+  if (jsonObject.containsKey('tokenCount')) {
+    return ModalityTokenCount(modality, jsonObject['tokenCount'] as int);
+  } else {
+    return ModalityTokenCount(modality, 0);
+  }
+}
+
+ContentModality _parseContentModality(Object jsonObject) {
+  return switch (jsonObject) {
+    'MODALITY_UNSPECIFIED' => ContentModality.unspecified,
+    'TEXT' => ContentModality.text,
+    'IMAGE' => ContentModality.image,
+    'VIDEO' => ContentModality.video,
+    'AUDIO' => ContentModality.audio,
+    'DOCUMENT' => ContentModality.document,
+    _ => throw unhandledFormat('ContentModality', jsonObject),
+  };
+}
+
 SafetyRating _parseSafetyRating(Object? jsonObject) {
   return switch (jsonObject) {
     {