update the test internal constructor and add more annotations

Author: cynthiajoan

packages/firebase_ai/firebase_ai/lib/src/imagen/imagen_model.dart

@@ -39,7 +39,9 @@ final class ImagenModel extends BaseApiClientModel {
         _safetySettings = safetySettings,
         _useVertexBackend = useVertexBackend,
         super(
-            serializationStrategy: VertexSerialization(),
+            serializationStrategy: useVertexBackend
+                ? VertexSerialization()
+                : DeveloperSerialization(),
             modelUri: useVertexBackend
                 ? _VertexUri(app: app, model: model, location: location)
                 : _GoogleAIUri(app: app, model: model),

packages/firebase_ai/firebase_ai/lib/src/server_template/template_generative_model.dart

@@ -18,15 +18,10 @@ part of '../base_model.dart';
 /// A generative model that connects to a remote server template.
 @experimental
 final class TemplateGenerativeModel extends BaseTemplateApiClientModel {
-  /// Constructor only for test usage
-  @internal
-  TemplateGenerativeModel.internal({
+  TemplateGenerativeModel._test({
     required String location,
     required FirebaseApp app,
     required bool useVertexBackend,
-    bool? useLimitedUseAppCheckTokens,
-    FirebaseAppCheck? appCheck,
-    FirebaseAuth? auth,
     http.Client? httpClient,
   }) : super(
           serializationStrategy: useVertexBackend
@@ -38,8 +33,7 @@ final class TemplateGenerativeModel extends BaseTemplateApiClientModel {
           client: HttpApiClient(
               apiKey: app.options.apiKey,
               httpClient: httpClient,
-              requestHeaders: BaseModel.firebaseTokens(
-                  appCheck, auth, app, useLimitedUseAppCheckTokens)),
+              requestHeaders: BaseModel.firebaseTokens(null, null, app, false)),
           templateUri: useVertexBackend
               ? _TemplateVertexUri(app: app, location: location)
               : _TemplateGoogleAIUri(app: app),
@@ -97,6 +91,7 @@ final class TemplateGenerativeModel extends BaseTemplateApiClientModel {
 
 /// Returns a [TemplateGenerativeModel] using its private constructor.
 @experimental
+@internal
 TemplateGenerativeModel createTemplateGenerativeModel({
   required FirebaseApp app,
   required String location,
@@ -113,3 +108,19 @@ TemplateGenerativeModel createTemplateGenerativeModel({
       auth: auth,
       location: location,
     );
+
+/// Returns a [TemplateGenerativeModel] for test case.
+@experimental
+@internal
+TemplateGenerativeModel createTestTemplateGenerativeModel({
+  required FirebaseApp app,
+  required String location,
+  required bool useVertexBackend,
+  required http.Client client,
+}) =>
+    TemplateGenerativeModel._test(
+      app: app,
+      useVertexBackend: useVertexBackend,
+      location: location,
+      httpClient: client,
+    );

packages/firebase_ai/firebase_ai/lib/src/server_template/template_imagen_model.dart

@@ -17,15 +17,10 @@ part of '../base_model.dart';
 /// An image model that connects to a remote server template.
 @experimental
 final class TemplateImagenModel extends BaseTemplateApiClientModel {
-  /// Constructor only for test usage
-  @internal
-  TemplateImagenModel.internal(
+  TemplateImagenModel._testModel(
       {required FirebaseApp app,
       required String location,
       required bool useVertexBackend,
-      bool? useLimitedUseAppCheckTokens,
-      FirebaseAppCheck? appCheck,
-      FirebaseAuth? auth,
       http.Client? httpClient})
       : super(
           serializationStrategy: VertexSerialization(),
@@ -35,8 +30,7 @@ final class TemplateImagenModel extends BaseTemplateApiClientModel {
           client: HttpApiClient(
               apiKey: app.options.apiKey,
               httpClient: httpClient,
-              requestHeaders: BaseModel.firebaseTokens(
-                  appCheck, auth, app, useLimitedUseAppCheckTokens)),
+              requestHeaders: BaseModel.firebaseTokens(null, null, app, false)),
           templateUri: useVertexBackend
               ? _TemplateVertexUri(app: app, location: location)
               : _TemplateGoogleAIUri(app: app),
@@ -50,7 +44,9 @@ final class TemplateImagenModel extends BaseTemplateApiClientModel {
       FirebaseAppCheck? appCheck,
       FirebaseAuth? auth})
       : super(
-          serializationStrategy: VertexSerialization(),
+          serializationStrategy: useVertexBackend
+              ? VertexSerialization()
+              : DeveloperSerialization(),
           modelUri: useVertexBackend
               ? _VertexUri(app: app, model: '', location: location)
               : _GoogleAIUri(app: app, model: ''),
@@ -78,8 +74,9 @@ final class TemplateImagenModel extends BaseTemplateApiClientModel {
       );
 }
 
-/// Returns a [TemplateImagenModel] using it's private constructor.
+/// Returns a [TemplateImagenModel] using its private constructor.
 @experimental
+@internal
 TemplateImagenModel createTemplateImagenModel({
   required FirebaseApp app,
   required String location,
@@ -96,3 +93,19 @@ TemplateImagenModel createTemplateImagenModel({
       useVertexBackend: useVertexBackend,
       useLimitedUseAppCheckTokens: useLimitedUseAppCheckTokens,
     );
+
+/// Returns a [TemplateImagenModel] using its private constructor.
+@experimental
+@internal
+TemplateImagenModel createTestTemplateImagenModel({
+  required FirebaseApp app,
+  required String location,
+  required bool useVertexBackend,
+  required http.Client client,
+}) =>
+    TemplateImagenModel._testModel(
+      app: app,
+      location: location,
+      useVertexBackend: useVertexBackend,
+      httpClient: client,
+    );

packages/firebase_ai/firebase_ai/test/server_template_test.dart

@@ -15,6 +15,7 @@
 import 'dart:convert';
 
 import 'package:firebase_ai/firebase_ai.dart';
+import 'package:firebase_ai/src/base_model.dart';
 import 'package:firebase_core/firebase_core.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:http/http.dart' as http;
@@ -61,11 +62,11 @@ void main() {
     TemplateGenerativeModel createModel(http.Client client,
         {bool useVertexBackend = true}) {
       // ignore: invalid_use_of_internal_member
-      return TemplateGenerativeModel.internal(
+      return createTestTemplateGenerativeModel(
         app: app,
         location: location,
         useVertexBackend: useVertexBackend,
-        httpClient: client,
+        client: client,
       );
     }
 
@@ -113,12 +114,11 @@ void main() {
     TemplateImagenModel createModel(http.Client client,
         {bool useVertexBackend = true}) {
       // ignore: invalid_use_of_internal_member
-      return TemplateImagenModel.internal(
-        app: app,
-        location: location,
-        useVertexBackend: useVertexBackend,
-        httpClient: client,
-      );
+      return createTestTemplateImagenModel(
+          app: app,
+          location: location,
+          useVertexBackend: useVertexBackend,
+          client: client);
     }
 
     test('generateImages can make successful request', () async {