From 98528113e9a2537b5be17a387d48a207e0b013c7 Mon Sep 17 00:00:00 2001
From: a-maurice <amaurice@google.com>
Date: Tue, 25 Nov 2025 16:36:04 -0800
Subject: [PATCH 1/2] [Firebase AI] Disable some Imagen tests from CI

---
 .../Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs b/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
index edcbdee6..07403a2a 100644
--- a/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
+++ b/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
@@ -75,7 +75,10 @@ protected override void Start()
         TestCountTokens,
         TestYoutubeLink,
         TestGenerateImage,
+#if !(FIREBASE_RUNNING_FROM_CI)
+        // Disabled from CI, because of rate limit issues
         TestImagenGenerateImage,
+#endif
         TestImagenGenerateImageOptions,
         TestThinkingBudget,
         TestIncludeThoughts,
@@ -83,7 +86,10 @@ protected override void Start()
         TestUrlContext,
         TestTemplateGenerateContent,
         TestTemplateGenerateContentStream,
+#if !(FIREBASE_RUNNING_FROM_CI)
+        // Disabled from CI, because of rate limit issues
         TestTemplateImagenGenerateImage,
+#endif
       };
       // Set of tests that only run the single time.
       Func<Task>[] singleTests = {

From 9dc01e2c3258bc20deca3076906bc1a41a1c21bd Mon Sep 17 00:00:00 2001
From: a-maurice <amaurice@google.com>
Date: Tue, 25 Nov 2025 16:54:14 -0800
Subject: [PATCH 2/2] Update UIHandlerAutomated.cs

---
 .../Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs b/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
index 07403a2a..d4b4a162 100644
--- a/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
+++ b/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
@@ -75,7 +75,7 @@ protected override void Start()
         TestCountTokens,
         TestYoutubeLink,
         TestGenerateImage,
-#if !(FIREBASE_RUNNING_FROM_CI)
+#if !(FIREBASE_RUNNING_FROM_CI && !UNITY_EDITOR)
         // Disabled from CI, because of rate limit issues
         TestImagenGenerateImage,
 #endif
@@ -86,7 +86,7 @@ protected override void Start()
         TestUrlContext,
         TestTemplateGenerateContent,
         TestTemplateGenerateContentStream,
-#if !(FIREBASE_RUNNING_FROM_CI)
+#if !(FIREBASE_RUNNING_FROM_CI && !UNITY_EDITOR)
         // Disabled from CI, because of rate limit issues
         TestTemplateImagenGenerateImage,
 #endif