From b3c677da4b0182753584cba8fce8695de660a754 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 1 Oct 2025 16:57:06 -0400
Subject: [PATCH] [Firebase AI] Use 2.5-flash-lite for URLContext test

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

diff --git a/FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift b/FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift
index 747b1dc5bea..aeeaeb76c87 100644
--- a/FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift
+++ b/FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift
@@ -430,7 +430,7 @@ struct GenerateContentIntegrationTests {
   )
   func generateContent_withURLContext_succeeds(_ config: InstanceConfig) async throws {
     let model = FirebaseAI.componentInstance(config).generativeModel(
-      modelName: ModelNames.gemini2_5_Flash,
+      modelName: ModelNames.gemini2_5_FlashLite,
       tools: [.urlContext()]
     )
     let prompt = """