Remove Imagen tests

Author: a-maurice

firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs

@@ -70,8 +70,6 @@ protected override void Start() {
         TestCountTokens,
         TestYoutubeLink,
         TestGenerateImage,
-        TestImagenGenerateImage,
-        TestImagenGenerateImageOptions
       };
       // Set of tests that only run the single time.
       Func<Task>[] singleTests = {
@@ -88,9 +86,6 @@ protected override void Start() {
         InternalTestBasicResponseLongUsageMetadata,
         InternalTestGoogleAIBasicReplyShort,
         InternalTestGoogleAICitations,
-        InternalTestGenerateImagesBase64,
-        InternalTestGenerateImagesAllFiltered,
-        InternalTestGenerateImagesBase64SomeFiltered,
       };
 
       // Create the set of tests, combining the above lists.
@@ -688,58 +683,6 @@ async Task TestGenerateImage(Backend backend) {
       Assert($"Missing expected modalities. Text: {foundText}, Image: {foundImage}", foundText && foundImage);
     }
 
-    async Task TestImagenGenerateImage(Backend backend) {
-      var model = GetFirebaseAI(backend).GetImagenModel("imagen-3.0-generate-002");
-
-      var response = await model.GenerateImagesAsync(
-          "Generate an image of a cartoon dog.");
-
-      // We can't easily test if the image is correct, but can check other random data.
-      AssertEq("FilteredReason", response.FilteredReason, null);
-      AssertEq("Image Count", response.Images.Count, 1);
-
-      AssertEq($"Image MimeType", response.Images[0].MimeType, "image/png");
-
-      var texture = response.Images[0].AsTexture2D();
-      Assert($"Image as Texture2D", texture != null);
-      // By default the image should be Square 1x1, so check for that.
-      Assert($"Image Height > 0", texture.height > 0);
-      AssertEq($"Image Height = Width", texture.height, texture.width);
-    }
-
-    async Task TestImagenGenerateImageOptions(Backend backend) {
-      var model = GetFirebaseAI(backend).GetImagenModel(
-          modelName: "imagen-3.0-generate-002",
-          generationConfig: new ImagenGenerationConfig(
-            // negativePrompt and addWatermark are not supported on this version of the model.
-            numberOfImages: 2,
-            aspectRatio: ImagenAspectRatio.Landscape4x3,
-            imageFormat: ImagenImageFormat.Jpeg(50)
-          ),
-          safetySettings: new ImagenSafetySettings(
-            safetyFilterLevel: ImagenSafetySettings.SafetyFilterLevel.BlockLowAndAbove,
-            personFilterLevel: ImagenSafetySettings.PersonFilterLevel.BlockAll),
-          requestOptions: new RequestOptions(timeout: TimeSpan.FromMinutes(1)));
-
-      var response = await model.GenerateImagesAsync(
-          "Generate an image of a cartoon dog.");
-
-      // We can't easily test if the image is correct, but can check other random data.
-      AssertEq("FilteredReason", response.FilteredReason, null);
-      AssertEq("Image Count", response.Images.Count, 2);
-
-      for (int i = 0; i < 2; i++) {
-        AssertEq($"Image {i} MimeType", response.Images[i].MimeType, "image/jpeg");
-
-        var texture = response.Images[i].AsTexture2D();
-        Assert($"Image {i} as Texture2D", texture != null);
-        // By default the image should be Landscape 4x3, so check for that.
-        Assert($"Image {i} Height > 0", texture.height > 0);
-        Assert($"Image {i} Height < Width {texture.height} < {texture.width}",
-            texture.height < texture.width);
-      }
-    }
-
     // Test providing a file from a GCS bucket (Firebase Storage) to the model.
     async Task TestReadFile() {
       // GCS is currently only supported with VertexAI.
@@ -1236,54 +1179,5 @@ async Task InternalTestGoogleAICitations() {
       AssertEq("CandidatesTokensDetails[0].Modality", candidatesDetails[0].Modality, ContentModality.Text);
       AssertEq("CandidatesTokensDetails[0].TokenCount", candidatesDetails[0].TokenCount, 1667);
     }
-
-    async Task InternalTestGenerateImagesBase64() {
-      Dictionary<string, object> json = await GetVertexJsonTestData("unary-success-generate-images-base64.json");
-      var response = ImagenGenerationResponse<ImagenInlineImage>.FromJson(json);
-
-      AssertEq("FilteredReason", response.FilteredReason, null);
-      AssertEq("Image Count", response.Images.Count, 4);
-
-      for (int i = 0; i < response.Images.Count; i++) {
-        var image = response.Images[i];
-        AssertEq($"Image {i} MimeType", image.MimeType, "image/png");
-        Assert($"Image {i} Length: {image.Data.Length}", image.Data.Length > 0);
-
-        var texture = image.AsTexture2D();
-        Assert($"Failed to convert Image {i}", texture != null);
-      }
-    }
-
-    async Task InternalTestGenerateImagesAllFiltered() {
-      Dictionary<string, object> json = await GetVertexJsonTestData("unary-failure-generate-images-all-filtered.json");
-      var response = ImagenGenerationResponse<ImagenInlineImage>.FromJson(json);
-
-      AssertEq("FilteredReason", response.FilteredReason,
-        "Unable to show generated images. All images were filtered out because " +
-        "they violated Vertex AI's usage guidelines. You will not be charged for " +
-        "blocked images. Try rephrasing the prompt. If you think this was an error, " +
-        "send feedback. Support codes: 39322892, 29310472");
-      AssertEq("Image Count", response.Images.Count, 0);
-    }
-
-    async Task InternalTestGenerateImagesBase64SomeFiltered() {
-      Dictionary<string, object> json = await GetVertexJsonTestData("unary-failure-generate-images-base64-some-filtered.json");
-      var response = ImagenGenerationResponse<ImagenInlineImage>.FromJson(json);
-
-      AssertEq("FilteredReason", response.FilteredReason,
-        "Your current safety filter threshold filtered out 2 generated images. " +
-        "You will not be charged for blocked images. Try rephrasing the prompt. " +
-        "If you think this was an error, send feedback.");
-      AssertEq("Image Count", response.Images.Count, 2);
-
-      for (int i = 0; i < response.Images.Count; i++) {
-        var image = response.Images[i];
-        AssertEq($"Image {i} MimeType", image.MimeType, "image/png");
-        Assert($"Image {i} Length: {image.Data.Length}", image.Data.Length > 0);
-
-        var texture = image.AsTexture2D();
-        Assert($"Failed to convert Image {i}", texture != null);
-      }
-    }
   }
 }