[Firebase AI] Handle known URL Context issue in integration test (#15386)

Author: andrewheard

FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift

@@ -433,22 +433,28 @@ struct GenerateContentIntegrationTests {
       modelName: ModelNames.gemini2_5_Flash,
       tools: [.urlContext()]
     )
-    let prompt = """
-    Write a one paragraph summary of this blog post: \
-    https://developers.googleblog.com/en/introducing-gemma-3-270m/
-    """
+    let url = "https://developers.googleblog.com/en/introducing-gemma-3-270m/"
+    let prompt = "Write a one paragraph summary of this blog post: \(url)"
 
-    let response = try await model.generateContent(prompt)
+    // TODO(#15385): Remove `withKnownIssue` when the URL Context tool works consistently using the
+    // Gemini Developer API.
+    try await withKnownIssue(isIntermittent: true) {
+      let response = try await model.generateContent(prompt)
 
-    let candidate = try #require(response.candidates.first)
-    let urlContextMetadata = try #require(candidate.urlContextMetadata)
-    #expect(urlContextMetadata.urlMetadata.count == 1)
-    let urlMetadata = try #require(urlContextMetadata.urlMetadata.first)
-    let retrievedURL = try #require(urlMetadata.retrievedURL)
-    #expect(
-      retrievedURL == URL(string: "https://developers.googleblog.com/en/introducing-gemma-3-270m/")
-    )
-    #expect(urlMetadata.retrievalStatus == .success)
+      let candidate = try #require(response.candidates.first)
+      let urlContextMetadata = try #require(candidate.urlContextMetadata)
+      #expect(urlContextMetadata.urlMetadata.count == 1)
+      let urlMetadata = try #require(urlContextMetadata.urlMetadata.first)
+      let retrievedURL = try #require(urlMetadata.retrievedURL)
+      #expect(retrievedURL == URL(string: url))
+      #expect(urlMetadata.retrievalStatus == .success)
+    } when: {
+      // This issue only impacts the Gemini Developer API (Google AI), Vertex AI is unaffected.
+      if case .googleAI = config.apiConfig.service {
+        return true
+      }
+      return false
+    }
   }
 
   @Test(arguments: InstanceConfig.allConfigs)