From ea2a498d5ce7cdcd61cb2cafc3d6cca37feee274 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Wed, 11 Jun 2025 19:22:48 +0000
Subject: [PATCH 1/2] Jules was unable to complete the task in time. Please
 review the work done so far and provide feedback for Jules to continue.

---
 .../Tests/TestApp/Tests/Integration/IntegrationTests.swift      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/FirebaseAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift b/FirebaseAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
index 0cfe64d3086..da2bdfa555e 100644
--- a/FirebaseAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
+++ b/FirebaseAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
@@ -67,7 +67,7 @@ final class IntegrationTests: XCTestCase {
   func testCountTokens_text() async throws {
     let prompt = "Why is the sky blue?"
     model = vertex.generativeModel(
-      modelName: "gemini-1.5-pro",
+      modelName: ModelNames.gemini2Flash,
       generationConfig: generationConfig,
       safetySettings: [
         SafetySetting(harmCategory: .harassment, threshold: .blockLowAndAbove, method: .severity),

From ecee8519a9aa7ee5707cbcd477d13c90033c4581 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Wed, 11 Jun 2025 19:23:45 +0000
Subject: [PATCH 2/2] I've updated the `testCountTokens_text` function to use
 `ModelNames.gemini2Flash`.

- I found the `testCountTokens_text` function in `FirebaseAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift`.
- I then updated the function to use `ModelNames.gemini2Flash` instead of the hardcoded string `"gemini-1.5-pro"`.