[Vertex AI] Test SDK with v1 API instead of v1beta

FirebaseVertexAI/Sources/Constants.swift

@@ -17,7 +17,7 @@ import Foundation
 /// Constants associated with the Vertex AI for Firebase SDK.
 enum Constants {
   /// The Vertex AI backend endpoint URL.
-  static let baseURL = "https://firebasevertexai.googleapis.com"
+  static let baseURL = "https://staging-firebasevertexai.sandbox.googleapis.com"
 
   /// The base reverse-DNS name for `NSError` or `CustomNSError` error domains.
   ///

FirebaseVertexAI/Sources/GenerativeAIRequest.swift

@@ -31,7 +31,7 @@ public struct RequestOptions {
   let timeout: TimeInterval
 
   /// The API version to use in requests to the backend.
-  let apiVersion = "v1beta"
+  let apiVersion = "v1"
 
   /// Initializes a request options object.
   ///

FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift

@@ -66,7 +66,7 @@ final class IntegrationTests: XCTestCase {
 
   // MARK: - Generate Content
 
-  func testGenerateContent() async throws {
+  func testGenerateContent_text() async throws {
     let prompt = "Where is Google headquarters located? Answer with the city name only."
 
     let response = try await model.generateContent(prompt)
@@ -75,6 +75,31 @@ final class IntegrationTests: XCTestCase {
     XCTAssertEqual(text, "Mountain View")
   }
 
+  func testGenerateContent_image_fileData_public() async throws {
+    let storageRef = storage.reference(withPath: "vertexai/public/green.png")
+    let fileData = FileDataPart(uri: storageRef.gsURI, mimeType: "image/png")
+
+    let response = try await model.generateContent(fileData, "What color is this?")
+
+    XCTAssertNotNil(response.text)
+  }
+
+  func testGenerateContent_imageData_requiresAuth_wrongUser_permissionDenied() async throws {
+    let userID = "3MjEzU6JIobWvHdCYHicnDMcPpQ2"
+    let storageRef = storage.reference(withPath: "vertexai/authenticated/user/\(userID)/pink.webp")
+
+    let fileData = FileDataPart(uri: storageRef.gsURI, mimeType: "image/webp")
+
+    do {
+      let response = try await model.generateContent(fileData, "What color is this?")
+      XCTFail("Expected to throw an error, got response: \(response)")
+    } catch {
+      let errorDescription = String(describing: error)
+      XCTAssertTrue(errorDescription.contains("403"))
+      XCTAssertTrue(errorDescription.contains("The caller does not have permission"))
+    }
+  }
+
   func testGenerateContent_appCheckNotConfigured_shouldFail() async throws {
     let app = try FirebaseApp.defaultNamedCopy(name: TestAppCheckProviderFactory.notConfiguredName)
     addTeardownBlock { await app.delete() }
@@ -83,13 +108,63 @@ final class IntegrationTests: XCTestCase {
     let prompt = "Where is Google headquarters located? Answer with the city name only."
 
     do {
-      _ = try await model.generateContent(prompt)
-      XCTFail("Expected a Firebase App Check error; none thrown.")
+      let response = try await model.generateContent(prompt)
+      XCTFail("Expected a Firebase App Check error, got response: \(response)")
     } catch let GenerateContentError.internalError(error) {
       XCTAssertTrue(String(describing: error).contains("Firebase App Check token is invalid"))
     }
   }
 
+  // MARK: - Generate Content Streaming
+
+  func testGenerateContentStream_text() async throws {
+    let prompt = "Where is Google headquarters located? Answer with the city name only."
+
+    var text = ""
+    let contentStream = try model.generateContentStream(prompt)
+    for try await chunk in contentStream {
+      text += try XCTUnwrap(chunk.text)
+    }
+
+    text = text.trimmingCharacters(in: .whitespacesAndNewlines)
+    XCTAssertEqual(text, "Mountain View")
+  }
+
+  func testGenerateContentStream_image_fileData_public() async throws {
+    let storageRef = storage.reference(withPath: "vertexai/public/green.png")
+    let fileData = FileDataPart(uri: storageRef.gsURI, mimeType: "image/png")
+
+    var text = ""
+    let contentStream = try model.generateContentStream(fileData, "What color is this?")
+    for try await chunk in contentStream {
+      text += try XCTUnwrap(chunk.text)
+    }
+
+    text = text.trimmingCharacters(in: .whitespacesAndNewlines)
+    XCTAssertFalse(text.isEmpty)
+  }
+
+  func testGenContentStream_image_fileData_requiresAuth_wrongUser_permissionDenied() async throws {
+    let userID = "3MjEzU6JIobWvHdCYHicnDMcPpQ2"
+    let storageRef = storage.reference(withPath: "vertexai/authenticated/user/\(userID)/pink.webp")
+
+    let fileData = FileDataPart(uri: storageRef.gsURI, mimeType: "image/webp")
+
+    let contentStream = try model.generateContentStream(fileData, "What color is this?")
+
+    do {
+      var responses = [GenerateContentResponse]()
+      for try await response in contentStream {
+        responses.append(response)
+      }
+      XCTFail("Expected to throw an error, got response(s): \(responses)")
+    } catch {
+      let errorDescription = String(describing: error)
+      XCTAssertTrue(errorDescription.contains("403"))
+      XCTAssertTrue(errorDescription.contains("The caller does not have permission"))
+    }
+  }
+
   // MARK: - Count Tokens
 
   func testCountTokens_text() async throws {
@@ -166,8 +241,8 @@ final class IntegrationTests: XCTestCase {
     let fileData = FileDataPart(uri: storageRef.gsURI, mimeType: "image/webp")
 
     do {
-      _ = try await model.countTokens(fileData)
-      XCTFail("Expected to throw an error.")
+      let response = try await model.countTokens(fileData)
+      XCTFail("Expected to throw an error, got response: \(response)")
     } catch {
       let errorDescription = String(describing: error)
       XCTAssertTrue(errorDescription.contains("403"))
@@ -229,8 +304,8 @@ final class IntegrationTests: XCTestCase {
     let prompt = "Why is the sky blue?"
 
     do {
-      _ = try await model.countTokens(prompt)
-      XCTFail("Expected a Firebase App Check error; none thrown.")
+      let response = try await model.countTokens(prompt)
+      XCTFail("Expected a Firebase App Check error, got response: \(response)")
     } catch {
       XCTAssertTrue(String(describing: error).contains("Firebase App Check token is invalid"))
     }