From e290ab4407174885e0722d68a3b31e2b9638271c Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 5 Feb 2024 17:29:57 -0500
Subject: [PATCH 001/124] Add prototype for Firebase Vertex AI

---
 FirebaseCore/Sources/FIRApp.m                 |   1 +
 FirebaseVertexAI/Sources/VertexAI.swift       | 154 ++++++++++++++++++
 .../Sources/VertexAIComponent.swift           |  80 +++++++++
 Package.swift                                 |  21 +++
 4 files changed, 256 insertions(+)
 create mode 100644 FirebaseVertexAI/Sources/VertexAI.swift
 create mode 100644 FirebaseVertexAI/Sources/VertexAIComponent.swift

diff --git a/FirebaseCore/Sources/FIRApp.m b/FirebaseCore/Sources/FIRApp.m
index 8f4aefe4bc7..a8cc38c12de 100644
--- a/FirebaseCore/Sources/FIRApp.m
+++ b/FirebaseCore/Sources/FIRApp.m
@@ -829,6 +829,7 @@ + (void)registerSwiftComponents {
     @"FIRSessions" : @"fire-ses",
     @"FIRFunctionsComponent" : @"fire-fun",
     @"FIRStorageComponent" : @"fire-str",
+    @"FIRVertexAIComponent" : @"fire-vtx",
   };
   for (NSString *className in swiftComponents.allKeys) {
     Class klass = NSClassFromString(className);
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
new file mode 100644
index 00000000000..41aafaeee3d
--- /dev/null
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -0,0 +1,154 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+import FirebaseAppCheckInterop
+import FirebaseCore
+import GoogleGenerativeAI
+
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+@objc(FIRVertexAI)
+open class VertexAI: NSObject {
+  // MARK: - Public APIs
+
+  /// The default `VertexAI` instance.
+  ///
+  /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
+  public static func vertexAI(modelName: String, location: String) -> VertexAI {
+    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location)
+  }
+
+  public static func vertexAI(app: FirebaseApp, modelName: String, location: String) -> VertexAI {
+    let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                            in: app.container)
+    let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
+    return provider.vertexAI(location: location, modelResourceName: modelResourceName)
+  }
+
+  public func generateContentStream(prompt: String) async
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    return model.generateContentStream(prompt)
+  }
+
+  // MARK: - Private
+
+  /// The `FirebaseApp` associated with this `VertexAI` instance.
+  private let app: FirebaseApp
+
+  private let appCheck: AppCheckInterop?
+
+  private let location: String
+
+  private let modelResouceName: String
+
+  lazy var model: GenerativeModel = {
+    let options = RequestOptions(hooks: [
+      setVertexAIEndpoint,
+      addAccessTokenHeader,
+      addAppCheckHeader,
+    ])
+    return GenerativeModel(
+      name: modelResouceName,
+      apiKey: app.options.apiKey!,
+      requestOptions: options
+    )
+  }()
+
+  private static let accessTokenEnvKey = "FIRVertexAIAccessToken"
+
+  init(app: FirebaseApp, location: String, modelResourceName: String) {
+    self.app = app
+    appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
+    self.location = location
+    modelResouceName = modelResourceName
+  }
+
+  private static func modelResourceName(app: FirebaseApp, modelName: String,
+                                        location: String) -> String {
+    if modelName.contains("/") {
+      return modelName
+    }
+    guard let projectID = app.options.projectID else {
+      print("The FirebaseApp is missing a project ID.")
+      return modelName
+    }
+
+    return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
+  }
+
+  // MARK: Request Hooks
+
+  /// Replace the Labs endpoint with a Vertex AI endpoint in the provided request.
+  ///
+  /// This is temporary workaround until the Google Generative AI SDK supports setting an endpoint.
+  ///
+  /// - Parameter request: The `URLRequest` to modify with a Vertex AI hostname.
+  func setVertexAIEndpoint(request: inout URLRequest) {
+    guard let requestURL = request.url else {
+      return
+    }
+    guard var urlComponents = URLComponents(url: requestURL, resolvingAgainstBaseURL: false) else {
+      return
+    }
+    urlComponents.host = "\(location)-aiplatform.googleapis.com"
+
+    guard let componentsURL = urlComponents.url else {
+      return
+    }
+
+    request.url = componentsURL
+  }
+
+  /// Add a Google Cloud access token in an Authorization header in the provided request.
+  ///
+  /// This is a temporary workaround until Vertex AI can be called with an API key.
+  ///
+  /// - Parameter request: The `URLRequest` to modify by adding an access token.
+  func addAccessTokenHeader(request: inout URLRequest) {
+    // Remove the API key header, it is not supported by Vertex AI.
+    if var headers = request.allHTTPHeaderFields {
+      headers.removeValue(forKey: "x-goog-api-key")
+    }
+
+    guard let accessToken = ProcessInfo.processInfo.environment[VertexAI.accessTokenEnvKey] else {
+      print("""
+      Vertex AI requires an Access Token for authorization:
+      1. Get an access token by running `gcloud auth print-access-token`
+      2. Set it in the \(VertexAI.accessTokenEnvKey) environment variable.
+      """)
+      return
+    }
+
+    request.addValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
+  }
+
+  /// Adds an App Check token to the provided request, if App Check is included in the app.
+  ///
+  /// This demonstrates how an App Check token can be added to requests; it is currently ignored by
+  /// the backend.
+  ///
+  /// - Parameter request: The `URLRequest` to modify by adding an App Check token header.
+  func addAppCheckHeader(request: inout URLRequest) async {
+    guard let appCheck = appCheck else {
+      return
+    }
+
+    let tokenResult = await appCheck.getToken(forcingRefresh: false)
+    request.addValue(tokenResult.token, forHTTPHeaderField: "X-Firebase-AppCheck")
+  }
+}
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
new file mode 100644
index 00000000000..a8d6c177c74
--- /dev/null
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -0,0 +1,80 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseAppCheckInterop
+import FirebaseCore
+import Foundation
+
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+@objc(FIRVertexAIProvider)
+protocol VertexAIProvider {
+  @objc func vertexAI(location: String, modelResourceName: String) -> VertexAI
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+@objc(FIRVertexAIComponent)
+class VertexAIComponent: NSObject, Library, VertexAIProvider {
+  // MARK: - Private Variables
+
+  /// The app associated with all `VertexAI` instances in this container.
+  /// This is `unowned` instead of `weak` so it can be used without unwrapping in `vertexAI(...)`
+  private unowned let app: FirebaseApp
+
+  /// A map of active  `VertexAI` instances for `app`, keyed by model resource names
+  /// (e.g., "projects/my-project-id/locations/us-central1/publishers/google/models/gemini-pro").
+  private var instances: [String: VertexAI] = [:]
+
+  /// Lock to manage access to the `instances` array to avoid race conditions.
+  private var instancesLock: os_unfair_lock = .init()
+
+  // MARK: - Initializers
+
+  required init(app: FirebaseApp) {
+    self.app = app
+  }
+
+  // MARK: - Library conformance
+
+  static func componentsToRegister() -> [Component] {
+    let appCheckInterop = Dependency(with: AppCheckInterop.self, isRequired: false)
+    return [Component(VertexAIProvider.self,
+                      instantiationTiming: .lazy,
+                      dependencies: [
+                        appCheckInterop,
+                      ]) { container, isCacheable in
+        guard let app = container.app else { return nil }
+        isCacheable.pointee = true
+        return self.init(app: app)
+      }]
+  }
+
+  // MARK: - VertexAIProvider conformance
+
+  func vertexAI(location: String, modelResourceName: String) -> VertexAI {
+    os_unfair_lock_lock(&instancesLock)
+
+    // Unlock before the function returns.
+    defer { os_unfair_lock_unlock(&instancesLock) }
+
+    if let instance = instances[modelResourceName] {
+      return instance
+    }
+    let newInstance = VertexAI(app: app, location: location, modelResourceName: modelResourceName)
+    instances[modelResourceName] = newInstance
+    return newInstance
+  }
+}
diff --git a/Package.swift b/Package.swift
index f78597e8408..97f52f6fc79 100644
--- a/Package.swift
+++ b/Package.swift
@@ -137,6 +137,10 @@ let package = Package(
       name: "FirebaseStorage",
       targets: ["FirebaseStorage"]
     ),
+    .library(
+      name: "FirebaseVertexAI",
+      targets: ["FirebaseVertexAI"]
+    ),
   ],
   dependencies: [
     .package(
@@ -183,6 +187,10 @@ let package = Package(
       "100.0.0" ..< "101.0.0"
     ),
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
+    .package(
+      url: "https://github.com/google/generative-ai-swift.git",
+      revision: "f59b91c7df6ae4d48ff85ab58e640b5084712fa6"
+    ),
   ],
   targets: [
     .target(
@@ -1347,6 +1355,19 @@ let package = Package(
         .headerSearchPath("../../.."),
       ]
     ),
+
+    // MARK: - Firebase Vertex AI
+
+    .target(
+      name: "FirebaseVertexAI",
+      dependencies: [
+        "FirebaseAppCheckInterop",
+        "FirebaseCore",
+        "FirebaseCoreExtension",
+        .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
+      ],
+      path: "FirebaseVertexAI/Sources"
+    ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,
   cxxLanguageStandard: CXXLanguageStandard.gnucxx14

From da7e247b5a9e4f976970542ae18330b8ed8ad6ad Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 5 Feb 2024 17:53:44 -0500
Subject: [PATCH 002/124] Update generative-ai-swift dependency

---
 Package.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Package.swift b/Package.swift
index 97f52f6fc79..125aa4a46bd 100644
--- a/Package.swift
+++ b/Package.swift
@@ -189,7 +189,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "f59b91c7df6ae4d48ff85ab58e640b5084712fa6"
+      revision: "efcee26db89d1948ebd581b3097c94186d615282"
     ),
   ],
   targets: [

From 0d8715192448058f28b38eb73a5e134e720dd552 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 14 Feb 2024 16:01:53 -0500
Subject: [PATCH 003/124] Use Swift 5.9 `AccessLevelOnImport`

---
 FirebaseVertexAI/Sources/VertexAI.swift       | 29 ++++++++++++++-----
 .../Sources/VertexAIComponent.swift           |  8 ++---
 Package.swift                                 |  9 ++++--
 3 files changed, 32 insertions(+), 14 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 41aafaeee3d..429ad02491a 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -14,12 +14,16 @@
 
 import Foundation
 
+public import FirebaseCore
+public import class GoogleGenerativeAI.Chat
+public import protocol GoogleGenerativeAI.PartsRepresentable
+public import struct GoogleGenerativeAI.GenerateContentResponse
+public import struct GoogleGenerativeAI.ModelContent
+
+import FirebaseCoreExtension
 import FirebaseAppCheckInterop
-import FirebaseCore
-import GoogleGenerativeAI
 
-// Avoids exposing internal FirebaseCore APIs to Swift users.
-@_implementationOnly import FirebaseCoreExtension
+private import GoogleGenerativeAI
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAI)
@@ -40,9 +44,20 @@ open class VertexAI: NSObject {
     return provider.vertexAI(location: location, modelResourceName: modelResourceName)
   }
 
-  public func generateContentStream(prompt: String) async
-    -> AsyncThrowingStream<GenerateContentResponse, Error> {
-    return model.generateContentStream(prompt)
+  public func generateContentStream(_ parts: GoogleGenerativeAI
+    .PartsRepresentable...)
+    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
+    return model.generateContentStream([GoogleGenerativeAI.ModelContent(parts: parts)])
+  }
+
+  public func generateContentStream(_ content: [GoogleGenerativeAI.ModelContent])
+    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
+    return model.generateContentStream(content)
+  }
+
+  public func startChat(history: [GoogleGenerativeAI.ModelContent] = []) -> GoogleGenerativeAI
+    .Chat {
+    return model.startChat(history: history)
   }
 
   // MARK: - Private
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a8d6c177c74..68cc0220849 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -12,12 +12,12 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import FirebaseAppCheckInterop
-import FirebaseCore
 import Foundation
 
-// Avoids exposing internal FirebaseCore APIs to Swift users.
-@_implementationOnly import FirebaseCoreExtension
+public import FirebaseCore
+
+import FirebaseAppCheckInterop
+import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
diff --git a/Package.swift b/Package.swift
index 125aa4a46bd..28e98373c3e 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1,4 +1,4 @@
-// swift-tools-version:5.7.1
+// swift-tools-version:5.9
 // The swift-tools-version declares the minimum version of Swift required to
 // build this package.
 
@@ -189,7 +189,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "efcee26db89d1948ebd581b3097c94186d615282"
+      revision: "4fe215743fd02c8182efdcfff5e7571873d2c158"
     ),
   ],
   targets: [
@@ -1366,7 +1366,10 @@ let package = Package(
         "FirebaseCoreExtension",
         .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources"
+      path: "FirebaseVertexAI/Sources",
+      swiftSettings: [
+        .enableExperimentalFeature("AccessLevelOnImport"),
+      ]
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From 71ac32686a73a6f03161d885bbd988875a08800e Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 20 Feb 2024 10:33:11 -0500
Subject: [PATCH 004/124] Set Vertex AI endpoint in `RequestOptions`

---
 FirebaseVertexAI/Sources/VertexAI.swift | 35 ++++++-------------------
 Package.swift                           |  2 +-
 2 files changed, 9 insertions(+), 28 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 429ad02491a..e0478ea3a33 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -20,8 +20,8 @@ public import protocol GoogleGenerativeAI.PartsRepresentable
 public import struct GoogleGenerativeAI.GenerateContentResponse
 public import struct GoogleGenerativeAI.ModelContent
 
-import FirebaseCoreExtension
 import FirebaseAppCheckInterop
+import FirebaseCoreExtension
 
 private import GoogleGenerativeAI
 
@@ -72,11 +72,13 @@ open class VertexAI: NSObject {
   private let modelResouceName: String
 
   lazy var model: GenerativeModel = {
-    let options = RequestOptions(hooks: [
-      setVertexAIEndpoint,
-      addAccessTokenHeader,
-      addAppCheckHeader,
-    ])
+    let options = RequestOptions(
+      endpoint: "\(location)-aiplatform.googleapis.com",
+      hooks: [
+        addAccessTokenHeader,
+        addAppCheckHeader,
+      ]
+    )
     return GenerativeModel(
       name: modelResouceName,
       apiKey: app.options.apiKey!,
@@ -108,27 +110,6 @@ open class VertexAI: NSObject {
 
   // MARK: Request Hooks
 
-  /// Replace the Labs endpoint with a Vertex AI endpoint in the provided request.
-  ///
-  /// This is temporary workaround until the Google Generative AI SDK supports setting an endpoint.
-  ///
-  /// - Parameter request: The `URLRequest` to modify with a Vertex AI hostname.
-  func setVertexAIEndpoint(request: inout URLRequest) {
-    guard let requestURL = request.url else {
-      return
-    }
-    guard var urlComponents = URLComponents(url: requestURL, resolvingAgainstBaseURL: false) else {
-      return
-    }
-    urlComponents.host = "\(location)-aiplatform.googleapis.com"
-
-    guard let componentsURL = urlComponents.url else {
-      return
-    }
-
-    request.url = componentsURL
-  }
-
   /// Add a Google Cloud access token in an Authorization header in the provided request.
   ///
   /// This is a temporary workaround until Vertex AI can be called with an API key.
diff --git a/Package.swift b/Package.swift
index 28e98373c3e..951dd278cea 100644
--- a/Package.swift
+++ b/Package.swift
@@ -189,7 +189,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "4fe215743fd02c8182efdcfff5e7571873d2c158"
+      revision: "46e56ae138f152528b7b50520a22777f3d8adb18"
     ),
   ],
   targets: [

From 0f0d35ad6d98e0688ece21f3e312d236f7ff9e60 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 20 Feb 2024 10:49:58 -0500
Subject: [PATCH 005/124] Revert "Use Swift 5.9 `AccessLevelOnImport`"

This reverts commit af7d7836bc5f4fc5576a06bc7af646f3b179a66a.

# Conflicts:
#	FirebaseVertexAI/Sources/VertexAI.swift
#	Package.swift
---
 FirebaseVertexAI/Sources/VertexAI.swift          | 12 ++++--------
 FirebaseVertexAI/Sources/VertexAIComponent.swift |  8 ++++----
 Package.swift                                    |  7 ++-----
 3 files changed, 10 insertions(+), 17 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index e0478ea3a33..9e3809c9828 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -14,16 +14,12 @@
 
 import Foundation
 
-public import FirebaseCore
-public import class GoogleGenerativeAI.Chat
-public import protocol GoogleGenerativeAI.PartsRepresentable
-public import struct GoogleGenerativeAI.GenerateContentResponse
-public import struct GoogleGenerativeAI.ModelContent
-
 import FirebaseAppCheckInterop
-import FirebaseCoreExtension
+import FirebaseCore
+import GoogleGenerativeAI
 
-private import GoogleGenerativeAI
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAI)
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index 68cc0220849..a8d6c177c74 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -12,12 +12,12 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
+import FirebaseCore
 import Foundation
 
-public import FirebaseCore
-
-import FirebaseAppCheckInterop
-import FirebaseCoreExtension
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
diff --git a/Package.swift b/Package.swift
index 951dd278cea..a491ecfacd2 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1,4 +1,4 @@
-// swift-tools-version:5.9
+// swift-tools-version:5.7.1
 // The swift-tools-version declares the minimum version of Swift required to
 // build this package.
 
@@ -1366,10 +1366,7 @@ let package = Package(
         "FirebaseCoreExtension",
         .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources",
-      swiftSettings: [
-        .enableExperimentalFeature("AccessLevelOnImport"),
-      ]
+      path: "FirebaseVertexAI/Sources"
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From 7c5e5c38779eb12bd6a2e0cb8b84397455b64944 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 20 Feb 2024 12:23:31 -0500
Subject: [PATCH 006/124] Add alternative API surface that returns
 `GenerativeModel` instances

---
 FirebaseVertexAI/Sources/VertexAI.swift | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 9e3809c9828..02fdb82807f 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -26,6 +26,27 @@ import GoogleGenerativeAI
 open class VertexAI: NSObject {
   // MARK: - Public APIs
 
+  // Option 1: Return `GoogleGenerativeAI.GenerativeModel` instances.
+
+  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  ///
+  /// This instance is configured with the default `FirebaseApp`.
+  public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
+    .GenerativeModel {
+    return vertexAI(modelName: modelName, location: location).model
+  }
+
+  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  public static func generativeModel(app: FirebaseApp, modelName: String,
+                                     location: String) -> GoogleGenerativeAI.GenerativeModel {
+    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location).model
+  }
+
+  // Option 2: Return `VertexAI` instances with a similar API surface to
+  // `GoogleGenerativeAI.GenerativeModel`.
+  // Some types are re-used from the Google AI SDK for requests and responses, e.g.,
+  // `GenerateContentResponse`.
+
   /// The default `VertexAI` instance.
   ///
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.

From ee6651c766d00c001fcd77fb2b37645e6235a59f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 22 Feb 2024 11:01:44 -0500
Subject: [PATCH 007/124] Add sample apps using Vertex AI

---
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../ChatSample/Assets.xcassets/Contents.json  |    6 +
 .../Sample/ChatSample/ChatSampleApp.swift     |   36 +
 .../ChatSample/Models/ChatMessage.swift       |   64 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../Screens/ConversationScreen.swift          |  127 ++
 .../ViewModels/ConversationViewModel.swift    |  130 ++
 .../ChatSample/Views/BouncingDots.swift       |   77 ++
 .../ChatSample/Views/ErrorDetailsView.swift   |  247 ++++
 .../Sample/ChatSample/Views/ErrorView.swift   |   65 +
 .../Sample/ChatSample/Views/MessageView.swift |  108 ++
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../Assets.xcassets/Contents.json             |    6 +
 .../GenerativeAIMultimodalSampleApp.swift     |   31 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../Screens/PhotoReasoningScreen.swift        |   65 +
 .../ViewModels/PhotoReasoningViewModel.swift  |  119 ++
 .../project.pbxproj                           | 1053 +++++++++++++++++
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../Assets.xcassets/Contents.json             |    6 +
 .../GenerativeAISample/ContentView.swift      |   48 +
 .../GenerativeAISampleApp.swift               |   29 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../Assets.xcassets/Contents.json             |    6 +
 .../GenerativeAITextSampleApp.swift           |   31 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../Screens/SummarizeScreen.swift             |   74 ++
 .../ViewModels/SummarizeViewModel.swift       |   68 ++
 .../GenerativeAIUIComponents/Package.swift    |   35 +
 .../GenerativeAIUIComponents/InputField.swift |   82 ++
 .../MultimodalInputField.swift                |  183 +++
 FirebaseVertexAI/Sources/VertexAI.swift       |   37 +-
 37 files changed, 2822 insertions(+), 31 deletions(-)
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift

diff --git a/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift b/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
new file mode 100644
index 00000000000..7d7e02d7e3c
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
@@ -0,0 +1,36 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import FirebaseVertexAI
+import SwiftUI
+
+@main
+struct ChatSampleApp: App {
+  @StateObject
+  var viewModel = ConversationViewModel()
+
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      NavigationStack {
+        ConversationScreen()
+          .environmentObject(viewModel)
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift b/FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift
new file mode 100644
index 00000000000..6f7ab321b12
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift
@@ -0,0 +1,64 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+enum Participant {
+  case system
+  case user
+}
+
+struct ChatMessage: Identifiable, Equatable {
+  let id = UUID().uuidString
+  var message: String
+  let participant: Participant
+  var pending = false
+
+  static func pending(participant: Participant) -> ChatMessage {
+    Self(message: "", participant: participant, pending: true)
+  }
+}
+
+extension ChatMessage {
+  static var samples: [ChatMessage] = [
+    .init(message: "Hello. What can I do for you today?", participant: .system),
+    .init(message: "Show me a simple loop in Swift.", participant: .user),
+    .init(message: """
+    Sure, here is a simple loop in Swift:
+
+    # Example 1
+    ```
+    for i in 1...5 {
+      print("Hello, world!")
+    }
+    ```
+
+    This loop will print the string "Hello, world!" five times. The for loop iterates over a range of numbers,
+    in this case the numbers from 1 to 5. The variable i is assigned each number in the range, and the code inside the loop is executed.
+
+    **Here is another example of a simple loop in Swift:**
+    ```swift
+    var sum = 0
+    for i in 1...100 {
+      sum += i
+    }
+    print("The sum of the numbers from 1 to 100 is \\(sum).")
+    ```
+
+    This loop calculates the sum of the numbers from 1 to 100. The variable sum is initialized to 0, and then the for loop iterates over the range of numbers from 1 to 100. The variable i is assigned each number in the range, and the value of i is added to the sum variable. After the loop has finished executing, the value of sum is printed to the console.
+    """, participant: .system),
+  ]
+
+  static var sample = samples[0]
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
new file mode 100644
index 00000000000..a3044a1f430
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
@@ -0,0 +1,127 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import GenerativeAIUIComponents
+import SwiftUI
+
+struct ConversationScreen: View {
+  @EnvironmentObject
+  var viewModel: ConversationViewModel
+
+  @State
+  private var userPrompt = ""
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      ScrollViewReader { scrollViewProxy in
+        List {
+          ForEach(viewModel.messages) { message in
+            MessageView(message: message)
+          }
+          if let error = viewModel.error {
+            ErrorView(error: error)
+              .tag("errorView")
+          }
+        }
+        .listStyle(.plain)
+        .onChange(of: viewModel.messages, perform: { newValue in
+          if viewModel.hasError {
+            // wait for a short moment to make sure we can actually scroll to the bottom
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo("errorView", anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          } else {
+            guard let lastMessage = viewModel.messages.last else { return }
+
+            // wait for a short moment to make sure we can actually scroll to the bottom
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo(lastMessage.id, anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          }
+        })
+      }
+      InputField("Message...", text: $userPrompt) {
+        Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
+          .font(.title)
+      }
+      .focused($focusedField, equals: .message)
+      .onSubmit { sendOrStop() }
+    }
+    .toolbar {
+      ToolbarItem(placement: .primaryAction) {
+        Button(action: newChat) {
+          Image(systemName: "square.and.pencil")
+        }
+      }
+    }
+    .navigationTitle("Chat sample")
+    .onAppear {
+      focusedField = .message
+    }
+  }
+
+  private func sendMessage() {
+    Task {
+      let prompt = userPrompt
+      userPrompt = ""
+      await viewModel.sendMessage(prompt, streaming: true)
+    }
+  }
+
+  private func sendOrStop() {
+    if viewModel.busy {
+      viewModel.stop()
+    } else {
+      sendMessage()
+    }
+  }
+
+  private func newChat() {
+    viewModel.startNewChat()
+  }
+}
+
+struct ConversationScreen_Previews: PreviewProvider {
+  struct ContainerView: View {
+    @StateObject var viewModel = ConversationViewModel()
+
+    var body: some View {
+      ConversationScreen()
+        .environmentObject(viewModel)
+        .onAppear {
+          viewModel.messages = ChatMessage.samples
+        }
+    }
+  }
+
+  static var previews: some View {
+    NavigationStack {
+      ConversationScreen()
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
new file mode 100644
index 00000000000..465789be387
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -0,0 +1,130 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import UIKit
+
+@MainActor
+class ConversationViewModel: ObservableObject {
+  /// This array holds both the user's and the system's chat messages
+  @Published var messages = [ChatMessage]()
+
+  /// Indicates we're waiting for the model to finish
+  @Published var busy = false
+
+  @Published var error: Error?
+  var hasError: Bool {
+    return error != nil
+  }
+
+  private var model: GenerativeModel
+  private var chat: Chat
+  private var stopGenerating = false
+
+  private var chatTask: Task<Void, Never>?
+
+  init() {
+    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    chat = model.startChat()
+  }
+
+  func sendMessage(_ text: String, streaming: Bool = true) async {
+    error = nil
+    if streaming {
+      await internalSendMessageStreaming(text)
+    } else {
+      await internalSendMessage(text)
+    }
+  }
+
+  func startNewChat() {
+    stop()
+    error = nil
+    chat = model.startChat()
+    messages.removeAll()
+  }
+
+  func stop() {
+    chatTask?.cancel()
+    error = nil
+  }
+
+  private func internalSendMessageStreaming(_ text: String) async {
+    chatTask?.cancel()
+
+    chatTask = Task {
+      busy = true
+      defer {
+        busy = false
+      }
+
+      // first, add the user's message to the chat
+      let userMessage = ChatMessage(message: text, participant: .user)
+      messages.append(userMessage)
+
+      // add a pending message while we're waiting for a response from the backend
+      let systemMessage = ChatMessage.pending(participant: .system)
+      messages.append(systemMessage)
+
+      do {
+        let responseStream = chat.sendMessageStream(text)
+        for try await chunk in responseStream {
+          messages[messages.count - 1].pending = false
+          if let text = chunk.text {
+            messages[messages.count - 1].message += text
+          }
+        }
+      } catch {
+        self.error = error
+        print(error.localizedDescription)
+        messages.removeLast()
+      }
+    }
+  }
+
+  private func internalSendMessage(_ text: String) async {
+    chatTask?.cancel()
+
+    chatTask = Task {
+      busy = true
+      defer {
+        busy = false
+      }
+
+      // first, add the user's message to the chat
+      let userMessage = ChatMessage(message: text, participant: .user)
+      messages.append(userMessage)
+
+      // add a pending message while we're waiting for a response from the backend
+      let systemMessage = ChatMessage.pending(participant: .system)
+      messages.append(systemMessage)
+
+      do {
+        var response: GenerateContentResponse?
+        response = try await chat.sendMessage(text)
+
+        if let responseText = response?.text {
+          // replace pending message with backend response
+          messages[messages.count - 1].message = responseText
+          messages[messages.count - 1].pending = false
+        }
+      } catch {
+        self.error = error
+        print(error.localizedDescription)
+        messages.removeLast()
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift b/FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift
new file mode 100644
index 00000000000..6895e6723da
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift
@@ -0,0 +1,77 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import SwiftUI
+
+struct BouncingDots: View {
+  @State
+  private var dot1YOffset: CGFloat = 0.0
+
+  @State
+  private var dot2YOffset: CGFloat = 0.0
+
+  @State
+  private var dot3YOffset: CGFloat = 0.0
+
+  let animation = Animation.easeInOut(duration: 0.8)
+    .repeatForever(autoreverses: true)
+
+  var body: some View {
+    HStack(spacing: 8) {
+      Circle()
+        .fill(Color.white)
+        .frame(width: 10, height: 10)
+        .offset(y: dot1YOffset)
+        .onAppear {
+          withAnimation(self.animation.delay(0.0)) {
+            self.dot1YOffset = -5
+          }
+        }
+      Circle()
+        .fill(Color.white)
+        .frame(width: 10, height: 10)
+        .offset(y: dot2YOffset)
+        .onAppear {
+          withAnimation(self.animation.delay(0.2)) {
+            self.dot2YOffset = -5
+          }
+        }
+      Circle()
+        .fill(Color.white)
+        .frame(width: 10, height: 10)
+        .offset(y: dot3YOffset)
+        .onAppear {
+          withAnimation(self.animation.delay(0.4)) {
+            self.dot3YOffset = -5
+          }
+        }
+    }
+    .onAppear {
+      let baseOffset: CGFloat = -2
+
+      self.dot1YOffset = baseOffset
+      self.dot2YOffset = baseOffset
+      self.dot3YOffset = baseOffset
+    }
+  }
+}
+
+struct BouncingDots_Previews: PreviewProvider {
+  static var previews: some View {
+    BouncingDots()
+      .frame(width: 200, height: 50)
+      .background(.blue)
+      .roundedCorner(10, corners: [.allCorners])
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
new file mode 100644
index 00000000000..e601876b2d9
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -0,0 +1,247 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import MarkdownUI
+import SwiftUI
+
+extension SafetySetting.HarmCategory: CustomStringConvertible {
+  public var description: String {
+    switch self {
+    case .dangerousContent: "Dangerous content"
+    case .harassment: "Harassment"
+    case .hateSpeech: "Hate speech"
+    case .sexuallyExplicit: "Sexually explicit"
+    case .unknown: "Unknown"
+    case .unspecified: "Unspecified"
+    }
+  }
+}
+
+extension SafetyRating.HarmProbability: CustomStringConvertible {
+  public var description: String {
+    switch self {
+    case .high: "High"
+    case .low: "Low"
+    case .medium: "Medium"
+    case .negligible: "Negligible"
+    case .unknown: "Unknown"
+    case .unspecified: "Unspecified"
+    }
+  }
+}
+
+private struct SubtitleFormRow: View {
+  var title: String
+  var value: String
+
+  var body: some View {
+    VStack(alignment: .leading) {
+      Text(title)
+        .font(.subheadline)
+      Text(value)
+    }
+  }
+}
+
+private struct SubtitleMarkdownFormRow: View {
+  var title: String
+  var value: String
+
+  var body: some View {
+    VStack(alignment: .leading) {
+      Text(title)
+        .font(.subheadline)
+      Markdown(value)
+    }
+  }
+}
+
+private struct SafetyRatingsSection: View {
+  var ratings: [SafetyRating]
+
+  var body: some View {
+    Section("Safety ratings") {
+      List(ratings, id: \.self) { rating in
+        HStack {
+          Text("\(String(describing: rating.category))")
+            .font(.subheadline)
+          Spacer()
+          Text("\(String(describing: rating.probability))")
+        }
+      }
+    }
+  }
+}
+
+struct ErrorDetailsView: View {
+  var error: Error
+
+  var body: some View {
+    NavigationView {
+      Form {
+        switch error {
+        case let GenerateContentError.internalError(underlying: underlyingError):
+          Section("Error Type") {
+            Text("Internal error")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description",
+                            value: underlyingError.localizedDescription)
+          }
+
+        case let GenerateContentError.promptBlocked(response: generateContentResponse):
+          Section("Error Type") {
+            Text("Your prompt was blocked")
+          }
+
+          Section("Details") {
+            if let reason = generateContentResponse.promptFeedback?.blockReason {
+              SubtitleFormRow(title: "Reason for blocking", value: reason.rawValue)
+            }
+
+            if let text = generateContentResponse.text {
+              SubtitleMarkdownFormRow(title: "Last chunk for the response", value: text)
+            }
+          }
+
+          if let ratings = generateContentResponse.candidates.first?.safetyRatings {
+            SafetyRatingsSection(ratings: ratings)
+          }
+
+        case let GenerateContentError.responseStoppedEarly(
+          reason: finishReason,
+          response: generateContentResponse
+        ):
+
+          Section("Error Type") {
+            Text("Response stopped early")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Reason for finishing early", value: finishReason.rawValue)
+
+            if let text = generateContentResponse.text {
+              SubtitleMarkdownFormRow(title: "Last chunk for the response", value: text)
+            }
+          }
+
+          if let ratings = generateContentResponse.candidates.first?.safetyRatings {
+            SafetyRatingsSection(ratings: ratings)
+          }
+
+        case GenerateContentError.invalidAPIKey:
+          Section("Error Type") {
+            Text("Invalid API Key")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
+            SubtitleMarkdownFormRow(
+              title: "Help",
+              value: """
+              Please provide a valid value for `API_KEY` in the `GenerativeAI-Info.plist` file.
+              """
+            )
+          }
+
+        case GenerateContentError.unsupportedUserLocation:
+          Section("Error Type") {
+            Text("Unsupported User Location")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
+            SubtitleMarkdownFormRow(
+              title: "Help",
+              value: """
+              The API is unsupported in your location (country / territory); please see the list of
+              [available regions](https://ai.google.dev/available_regions#available_regions).
+              """
+            )
+          }
+
+        default:
+          Section("Error Type") {
+            Text("Some other error")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
+          }
+        }
+      }
+      .navigationTitle("Error details")
+      .navigationBarTitleDisplayMode(.inline)
+    }
+  }
+}
+
+#Preview("Response Stopped Early") {
+  let error = GenerateContentError.responseStoppedEarly(
+    reason: .maxTokens,
+    response: GenerateContentResponse(candidates: [
+      CandidateResponse(content: ModelContent(role: "model", [
+        """
+        A _hypothetical_ model response.
+        Cillum ex aliqua amet aliquip labore amet eiusmod consectetur reprehenderit sit commodo.
+        """,
+      ]),
+      safetyRatings: [
+        SafetyRating(category: .dangerousContent, probability: .high),
+        SafetyRating(category: .harassment, probability: .low),
+        SafetyRating(category: .hateSpeech, probability: .low),
+        SafetyRating(category: .sexuallyExplicit, probability: .low),
+      ],
+      finishReason: FinishReason.maxTokens,
+      citationMetadata: nil),
+    ],
+    promptFeedback: nil)
+  )
+
+  return ErrorDetailsView(error: error)
+}
+
+#Preview("Prompt Blocked") {
+  let error = GenerateContentError.promptBlocked(
+    response: GenerateContentResponse(candidates: [
+      CandidateResponse(content: ModelContent(role: "model", [
+        """
+        A _hypothetical_ model response.
+        Cillum ex aliqua amet aliquip labore amet eiusmod consectetur reprehenderit sit commodo.
+        """,
+      ]),
+      safetyRatings: [
+        SafetyRating(category: .dangerousContent, probability: .high),
+        SafetyRating(category: .harassment, probability: .low),
+        SafetyRating(category: .hateSpeech, probability: .low),
+        SafetyRating(category: .sexuallyExplicit, probability: .low),
+      ],
+      finishReason: FinishReason.other,
+      citationMetadata: nil),
+    ],
+    promptFeedback: nil)
+  )
+
+  return ErrorDetailsView(error: error)
+}
+
+#Preview("Invalid API Key") {
+  ErrorDetailsView(error: GenerateContentError.invalidAPIKey)
+}
+
+#Preview("Unsupported User Location") {
+  ErrorDetailsView(error: GenerateContentError.unsupportedUserLocation)
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
new file mode 100644
index 00000000000..1307eee62d4
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
@@ -0,0 +1,65 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import SwiftUI
+
+struct ErrorView: View {
+  var error: Error
+  @State private var isDetailsSheetPresented = false
+  var body: some View {
+    HStack {
+      Text("An error occurred.")
+      Button(action: { isDetailsSheetPresented.toggle() }) {
+        Image(systemName: "info.circle")
+      }
+    }
+    .frame(maxWidth: .infinity, alignment: .center)
+    .listRowSeparator(.hidden)
+    .sheet(isPresented: $isDetailsSheetPresented) {
+      ErrorDetailsView(error: error)
+    }
+  }
+}
+
+#Preview {
+  NavigationView {
+    let errorPromptBlocked = GenerateContentError.promptBlocked(
+      response: GenerateContentResponse(candidates: [
+        CandidateResponse(content: ModelContent(role: "model", [
+          """
+            A _hypothetical_ model response.
+            Cillum ex aliqua amet aliquip labore amet eiusmod consectetur reprehenderit sit commodo.
+          """,
+        ]),
+        safetyRatings: [
+          SafetyRating(category: .dangerousContent, probability: .high),
+          SafetyRating(category: .harassment, probability: .low),
+          SafetyRating(category: .hateSpeech, probability: .low),
+          SafetyRating(category: .sexuallyExplicit, probability: .low),
+        ],
+        finishReason: FinishReason.other,
+        citationMetadata: nil),
+      ],
+      promptFeedback: nil)
+    )
+    List {
+      MessageView(message: ChatMessage.samples[0])
+      MessageView(message: ChatMessage.samples[1])
+      ErrorView(error: errorPromptBlocked)
+    }
+    .listStyle(.plain)
+    .navigationTitle("Chat sample")
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift
new file mode 100644
index 00000000000..79894503ffd
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift
@@ -0,0 +1,108 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import MarkdownUI
+import SwiftUI
+
+struct RoundedCorner: Shape {
+  var radius: CGFloat = .infinity
+  var corners: UIRectCorner = .allCorners
+
+  func path(in rect: CGRect) -> Path {
+    let path = UIBezierPath(
+      roundedRect: rect,
+      byRoundingCorners: corners,
+      cornerRadii: CGSize(width: radius, height: radius)
+    )
+    return Path(path.cgPath)
+  }
+}
+
+extension View {
+  func roundedCorner(_ radius: CGFloat, corners: UIRectCorner) -> some View {
+    clipShape(RoundedCorner(radius: radius, corners: corners))
+  }
+}
+
+struct MessageContentView: View {
+  var message: ChatMessage
+
+  var body: some View {
+    if message.pending {
+      BouncingDots()
+    } else {
+      Markdown(message.message)
+        .markdownTextStyle {
+          FontFamilyVariant(.normal)
+          FontSize(.em(0.85))
+          ForegroundColor(message.participant == .system ? Color(UIColor.label) : .white)
+        }
+        .markdownBlockStyle(\.codeBlock) { configuration in
+          configuration.label
+            .relativeLineSpacing(.em(0.25))
+            .markdownTextStyle {
+              FontFamilyVariant(.monospaced)
+              FontSize(.em(0.85))
+              ForegroundColor(Color(.label))
+            }
+            .padding()
+            .background(Color(.secondarySystemBackground))
+            .clipShape(RoundedRectangle(cornerRadius: 8))
+            .markdownMargin(top: .zero, bottom: .em(0.8))
+        }
+    }
+  }
+}
+
+struct MessageView: View {
+  var message: ChatMessage
+
+  var body: some View {
+    HStack {
+      if message.participant == .user {
+        Spacer()
+      }
+      MessageContentView(message: message)
+        .padding(10)
+        .background(message.participant == .system
+          ? Color(UIColor.systemFill)
+          : Color(UIColor.systemBlue))
+        .roundedCorner(10,
+                       corners: [
+                         .topLeft,
+                         .topRight,
+                         message.participant == .system ? .bottomRight : .bottomLeft,
+                       ])
+      if message.participant == .system {
+        Spacer()
+      }
+    }
+    .listRowSeparator(.hidden)
+  }
+}
+
+struct MessageView_Previews: PreviewProvider {
+  static var previews: some View {
+    NavigationView {
+      List {
+        MessageView(message: ChatMessage.samples[0])
+        MessageView(message: ChatMessage.samples[1])
+        MessageView(message: ChatMessage.samples[2])
+        MessageView(message: ChatMessage(message: "Hello!", participant: .system, pending: true))
+      }
+      .listStyle(.plain)
+      .navigationTitle("Chat sample")
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
new file mode 100644
index 00000000000..49565de52d1
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
@@ -0,0 +1,31 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import SwiftUI
+
+@main
+struct GenerativeAIMultimodalSampleApp: App {
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      NavigationStack {
+        PhotoReasoningScreen()
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
new file mode 100644
index 00000000000..98f327585db
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
@@ -0,0 +1,65 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import GenerativeAIUIComponents
+import MarkdownUI
+import PhotosUI
+import SwiftUI
+
+struct PhotoReasoningScreen: View {
+  @StateObject var viewModel = PhotoReasoningViewModel()
+
+  var body: some View {
+    VStack {
+      MultimodalInputField(text: $viewModel.userInput, selection: $viewModel.selectedItems)
+        .onSubmit {
+          onSendTapped()
+        }
+
+      ScrollViewReader { scrollViewProxy in
+        List {
+          if let outputText = viewModel.outputText {
+            HStack(alignment: .top) {
+              if viewModel.inProgress {
+                ProgressView()
+              } else {
+                Image(systemName: "cloud.circle.fill")
+                  .font(.title2)
+              }
+
+              Markdown("\(outputText)")
+            }
+            .listRowSeparator(.hidden)
+          }
+        }
+        .listStyle(.plain)
+      }
+    }
+    .navigationTitle("Multimodal sample")
+  }
+
+  // MARK: - Actions
+
+  private func onSendTapped() {
+    Task {
+      await viewModel.reason()
+    }
+  }
+}
+
+#Preview {
+  NavigationStack {
+    PhotoReasoningScreen()
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
new file mode 100644
index 00000000000..aa646c127df
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -0,0 +1,119 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import OSLog
+import PhotosUI
+import SwiftUI
+
+@MainActor
+class PhotoReasoningViewModel: ObservableObject {
+  // Maximum value for the larger of the two image dimensions (height and width) in pixels. This is
+  // being used to reduce the image size in bytes.
+  private static let largestImageDimension = 768.0
+
+  private var logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "generative-ai")
+
+  @Published
+  var userInput: String = ""
+
+  @Published
+  var selectedItems = [PhotosPickerItem]()
+
+  @Published
+  var outputText: String? = nil
+
+  @Published
+  var errorMessage: String?
+
+  @Published
+  var inProgress = false
+
+  private var model: GenerativeModel?
+
+  init() {
+    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro-vision", location: "us-central1")
+  }
+
+  func reason() async {
+    defer {
+      inProgress = false
+    }
+    guard let model else {
+      return
+    }
+
+    do {
+      inProgress = true
+      errorMessage = nil
+      outputText = ""
+
+      let prompt = "Look at the image(s), and then answer the following question: \(userInput)"
+
+      var images = [PartsRepresentable]()
+      for item in selectedItems {
+        if let data = try? await item.loadTransferable(type: Data.self) {
+          guard let image = UIImage(data: data) else {
+            logger.error("Failed to parse data as an image, skipping.")
+            continue
+          }
+          if image.size.fits(largestDimension: PhotoReasoningViewModel.largestImageDimension) {
+            images.append(image)
+          } else {
+            guard let resizedImage = image
+              .preparingThumbnail(of: image.size
+                .aspectFit(largestDimension: PhotoReasoningViewModel.largestImageDimension)) else {
+              logger.error("Failed to resize image: \(image)")
+              continue
+            }
+
+            images.append(resizedImage)
+          }
+        }
+      }
+
+      let outputContentStream = model.generateContentStream(prompt, images)
+
+      // stream response
+      for try await outputContent in outputContentStream {
+        guard let line = outputContent.text else {
+          return
+        }
+
+        outputText = (outputText ?? "") + line
+      }
+    } catch {
+      logger.error("\(error.localizedDescription)")
+      errorMessage = error.localizedDescription
+    }
+  }
+}
+
+private extension CGSize {
+  func fits(largestDimension length: CGFloat) -> Bool {
+    return width <= length && height <= length
+  }
+
+  func aspectFit(largestDimension length: CGFloat) -> CGSize {
+    let aspectRatio = width / height
+    if width > height {
+      let width = min(self.width, length)
+      return CGSize(width: width, height: round(width / aspectRatio))
+    } else {
+      let height = min(self.height, length)
+      return CGSize(width: round(height * aspectRatio), height: height)
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
new file mode 100644
index 00000000000..20f07a5acca
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -0,0 +1,1053 @@
+// !$*UTF8*$!
+{
+	archiveVersion = 1;
+	classes = {
+	};
+	objectVersion = 60;
+	objects = {
+
+/* Begin PBXBuildFile section */
+		869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AA2B86BE3100482873 /* FirebaseVertexAI */; };
+		869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AC2B86BEA100482873 /* FirebaseVertexAI */; };
+		869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AE2B86BFBC00482873 /* FirebaseVertexAI */; };
+		869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200B02B86BFF500482873 /* FirebaseVertexAI */; };
+		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */; };
+		880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */; };
+		88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
+		88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
+		88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88209C232B0FBE1700F64795 /* MarkdownUI */; };
+		88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
+		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
+		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
+		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
+		8848C8352B0D04BC007B434F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8342B0D04BC007B434F /* ContentView.swift */; };
+		8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8362B0D04BD007B434F /* Assets.xcassets */; };
+		8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8392B0D04BD007B434F /* Preview Assets.xcassets */; };
+		8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */; };
+		8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84A2B0D051F007B434F /* Assets.xcassets */; };
+		8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84D2B0D051F007B434F /* Preview Assets.xcassets */; };
+		8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */; };
+		8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85C2B0D056D007B434F /* Assets.xcassets */; };
+		8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85F2B0D056D007B434F /* Preview Assets.xcassets */; };
+		886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
+		886F95D62B17BA010036F07A /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
+		886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95D72B17BA420036F07A /* MarkdownUI */; };
+		886F95DB2B17BAEF0036F07A /* PhotoReasoningViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */; };
+		886F95DC2B17BAEF0036F07A /* PhotoReasoningScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */; };
+		886F95DD2B17D5010036F07A /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
+		886F95DE2B17D5010036F07A /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
+		886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
+		886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
+		886F95E12B17D5010036F07A /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
+		886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95E22B17D6630036F07A /* GenerativeAIUIComponents */; };
+		889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
+		88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A91D2B0FC55100424728 /* MarkdownUI */; };
+		88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */; };
+		88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88D9474C2B14F27E008B5580 /* MarkdownUI */; };
+		88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F442B110D5300C08E95 /* ChatSampleApp.swift */; };
+		88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F482B110D5400C08E95 /* Assets.xcassets */; };
+		88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */; };
+		88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
+		88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
+		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
+		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
+		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
+		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
+/* End PBXBuildFile section */
+
+/* Begin PBXFileReference section */
+		869200B22B879C4F00482873 /* GoogleService-Info.plist */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text.plist.xml; path = "GoogleService-Info.plist"; sourceTree = "<group>"; };
+		8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningViewModel.swift; sourceTree = "<group>"; };
+		880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningScreen.swift; sourceTree = "<group>"; };
+		88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeScreen.swift; sourceTree = "<group>"; };
+		88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeViewModel.swift; sourceTree = "<group>"; };
+		88263BEE2B239BFE008AB09B /* ErrorView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = ErrorView.swift; sourceTree = "<group>"; };
+		8848C82F2B0D04BC007B434F /* GenerativeAISample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAISample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAISampleApp.swift; sourceTree = "<group>"; };
+		8848C8342B0D04BC007B434F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = "<group>"; };
+		8848C8362B0D04BD007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		8848C8392B0D04BD007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		8848C8442B0D051E007B434F /* GenerativeAITextSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAITextSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAITextSampleApp.swift; sourceTree = "<group>"; };
+		8848C84A2B0D051F007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		8848C84D2B0D051F007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAIMultimodalSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAIMultimodalSampleApp.swift; sourceTree = "<group>"; };
+		8848C85C2B0D056D007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		8848C85F2B0D056D007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		889873842B208563005B4896 /* ErrorDetailsView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ErrorDetailsView.swift; sourceTree = "<group>"; };
+		88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = GenerativeAIUIComponents; sourceTree = "<group>"; };
+		88E10F422B110D5300C08E95 /* ChatSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = ChatSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		88E10F442B110D5300C08E95 /* ChatSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatSampleApp.swift; sourceTree = "<group>"; };
+		88E10F482B110D5400C08E95 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		88E10F542B1112CA00C08E95 /* ConversationScreen.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ConversationScreen.swift; sourceTree = "<group>"; };
+		88E10F562B1112F600C08E95 /* ConversationViewModel.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ConversationViewModel.swift; sourceTree = "<group>"; };
+		88E10F582B11131900C08E95 /* ChatMessage.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatMessage.swift; sourceTree = "<group>"; };
+		88E10F5A2B11133E00C08E95 /* MessageView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = MessageView.swift; sourceTree = "<group>"; };
+		88E10F5C2B11135000C08E95 /* BouncingDots.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = BouncingDots.swift; sourceTree = "<group>"; };
+/* End PBXFileReference section */
+
+/* Begin PBXFrameworksBuildPhase section */
+		8848C82C2B0D04BC007B434F /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
+				869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */,
+				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8412B0D051E007B434F /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
+				869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8532B0D056C007B434F /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
+				869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */,
+				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		88E10F3F2B110D5300C08E95 /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
+				869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */,
+				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXFrameworksBuildPhase section */
+
+/* Begin PBXGroup section */
+		8802666E2B0FC39000CF7CB6 /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		880266742B0FC39000CF7CB6 /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+		88209C1A2B0FBDC300F64795 /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+		88209C1C2B0FBDC300F64795 /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		88209C222B0FBE1700F64795 /* Frameworks */ = {
+			isa = PBXGroup;
+			children = (
+			);
+			name = Frameworks;
+			sourceTree = "<group>";
+		};
+		8848C8262B0D04BC007B434F = {
+			isa = PBXGroup;
+			children = (
+				88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */,
+				869200B22B879C4F00482873 /* GoogleService-Info.plist */,
+				8848C8312B0D04BC007B434F /* GenerativeAISample */,
+				8848C8452B0D051E007B434F /* GenerativeAITextSample */,
+				8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */,
+				88E10F432B110D5300C08E95 /* ChatSample */,
+				8848C8302B0D04BC007B434F /* Products */,
+				88209C222B0FBE1700F64795 /* Frameworks */,
+			);
+			sourceTree = "<group>";
+		};
+		8848C8302B0D04BC007B434F /* Products */ = {
+			isa = PBXGroup;
+			children = (
+				8848C82F2B0D04BC007B434F /* GenerativeAISample.app */,
+				8848C8442B0D051E007B434F /* GenerativeAITextSample.app */,
+				8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */,
+				88E10F422B110D5300C08E95 /* ChatSample.app */,
+			);
+			name = Products;
+			sourceTree = "<group>";
+		};
+		8848C8312B0D04BC007B434F /* GenerativeAISample */ = {
+			isa = PBXGroup;
+			children = (
+				8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */,
+				8848C8342B0D04BC007B434F /* ContentView.swift */,
+				8848C8362B0D04BD007B434F /* Assets.xcassets */,
+				8848C8382B0D04BD007B434F /* Preview Content */,
+			);
+			path = GenerativeAISample;
+			sourceTree = "<group>";
+		};
+		8848C8382B0D04BD007B434F /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				8848C8392B0D04BD007B434F /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		8848C8452B0D051E007B434F /* GenerativeAITextSample */ = {
+			isa = PBXGroup;
+			children = (
+				88209C1C2B0FBDC300F64795 /* ViewModels */,
+				88209C1A2B0FBDC300F64795 /* Screens */,
+				8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */,
+				8848C84A2B0D051F007B434F /* Assets.xcassets */,
+				8848C84C2B0D051F007B434F /* Preview Content */,
+			);
+			path = GenerativeAITextSample;
+			sourceTree = "<group>";
+		};
+		8848C84C2B0D051F007B434F /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				8848C84D2B0D051F007B434F /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */ = {
+			isa = PBXGroup;
+			children = (
+				8802666E2B0FC39000CF7CB6 /* ViewModels */,
+				880266742B0FC39000CF7CB6 /* Screens */,
+				8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */,
+				8848C85C2B0D056D007B434F /* Assets.xcassets */,
+				8848C85E2B0D056D007B434F /* Preview Content */,
+			);
+			path = GenerativeAIMultimodalSample;
+			sourceTree = "<group>";
+		};
+		8848C85E2B0D056D007B434F /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				8848C85F2B0D056D007B434F /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		88E10F432B110D5300C08E95 /* ChatSample */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F522B11124A00C08E95 /* Models */,
+				88E10F502B11123600C08E95 /* ViewModels */,
+				88E10F512B11124100C08E95 /* Views */,
+				88E10F532B1112B900C08E95 /* Screens */,
+				88E10F442B110D5300C08E95 /* ChatSampleApp.swift */,
+				88E10F482B110D5400C08E95 /* Assets.xcassets */,
+				88E10F4A2B110D5400C08E95 /* Preview Content */,
+			);
+			path = ChatSample;
+			sourceTree = "<group>";
+		};
+		88E10F4A2B110D5400C08E95 /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		88E10F502B11123600C08E95 /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F562B1112F600C08E95 /* ConversationViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		88E10F512B11124100C08E95 /* Views */ = {
+			isa = PBXGroup;
+			children = (
+				88263BEE2B239BFE008AB09B /* ErrorView.swift */,
+				88E10F5A2B11133E00C08E95 /* MessageView.swift */,
+				88E10F5C2B11135000C08E95 /* BouncingDots.swift */,
+				889873842B208563005B4896 /* ErrorDetailsView.swift */,
+			);
+			path = Views;
+			sourceTree = "<group>";
+		};
+		88E10F522B11124A00C08E95 /* Models */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F582B11131900C08E95 /* ChatMessage.swift */,
+			);
+			path = Models;
+			sourceTree = "<group>";
+		};
+		88E10F532B1112B900C08E95 /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F542B1112CA00C08E95 /* ConversationScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+/* End PBXGroup section */
+
+/* Begin PBXNativeTarget section */
+		8848C82E2B0D04BC007B434F /* GenerativeAISample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */;
+			buildPhases = (
+				8848C82B2B0D04BC007B434F /* Sources */,
+				8848C82C2B0D04BC007B434F /* Frameworks */,
+				8848C82D2B0D04BC007B434F /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = GenerativeAISample;
+			packageProductDependencies = (
+				886F95D72B17BA420036F07A /* MarkdownUI */,
+				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
+				869200AA2B86BE3100482873 /* FirebaseVertexAI */,
+			);
+			productName = GenerativeAISample;
+			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
+			productType = "com.apple.product-type.application";
+		};
+		8848C8432B0D051E007B434F /* GenerativeAITextSample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */;
+			buildPhases = (
+				8848C8402B0D051E007B434F /* Sources */,
+				8848C8412B0D051E007B434F /* Frameworks */,
+				8848C8422B0D051E007B434F /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = GenerativeAITextSample;
+			packageProductDependencies = (
+				88209C232B0FBE1700F64795 /* MarkdownUI */,
+				869200B02B86BFF500482873 /* FirebaseVertexAI */,
+			);
+			productName = GenerativeAITextSample;
+			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
+			productType = "com.apple.product-type.application";
+		};
+		8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */;
+			buildPhases = (
+				8848C8522B0D056C007B434F /* Sources */,
+				8848C8532B0D056C007B434F /* Frameworks */,
+				8848C8542B0D056C007B434F /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = GenerativeAIMultimodalSample;
+			packageProductDependencies = (
+				88B8A91D2B0FC55100424728 /* MarkdownUI */,
+				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
+				869200AE2B86BFBC00482873 /* FirebaseVertexAI */,
+			);
+			productName = GenerativeAIMultimodalSample;
+			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
+			productType = "com.apple.product-type.application";
+		};
+		88E10F412B110D5300C08E95 /* ChatSample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */;
+			buildPhases = (
+				88E10F3E2B110D5300C08E95 /* Sources */,
+				88E10F3F2B110D5300C08E95 /* Frameworks */,
+				88E10F402B110D5300C08E95 /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = ChatSample;
+			packageProductDependencies = (
+				88D9474C2B14F27E008B5580 /* MarkdownUI */,
+				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
+				869200AC2B86BEA100482873 /* FirebaseVertexAI */,
+			);
+			productName = ChatSample;
+			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
+			productType = "com.apple.product-type.application";
+		};
+/* End PBXNativeTarget section */
+
+/* Begin PBXProject section */
+		8848C8272B0D04BC007B434F /* Project object */ = {
+			isa = PBXProject;
+			attributes = {
+				BuildIndependentTargetsInParallel = 1;
+				LastSwiftUpdateCheck = 1510;
+				LastUpgradeCheck = 1510;
+				TargetAttributes = {
+					8848C82E2B0D04BC007B434F = {
+						CreatedOnToolsVersion = 15.1;
+					};
+					8848C8432B0D051E007B434F = {
+						CreatedOnToolsVersion = 15.1;
+					};
+					8848C8552B0D056C007B434F = {
+						CreatedOnToolsVersion = 15.1;
+					};
+					88E10F412B110D5300C08E95 = {
+						CreatedOnToolsVersion = 15.1;
+					};
+				};
+			};
+			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */;
+			compatibilityVersion = "Xcode 14.0";
+			developmentRegion = en;
+			hasScannedForEncodings = 0;
+			knownRegions = (
+				en,
+				Base,
+			);
+			mainGroup = 8848C8262B0D04BC007B434F;
+			packageReferences = (
+				88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */,
+				DEA09AC32B1FCE22001962D9 /* XCRemoteSwiftPackageReference "NetworkImage" */,
+				869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */,
+			);
+			productRefGroup = 8848C8302B0D04BC007B434F /* Products */;
+			projectDirPath = "";
+			projectRoot = "";
+			targets = (
+				8848C82E2B0D04BC007B434F /* GenerativeAISample */,
+				8848C8432B0D051E007B434F /* GenerativeAITextSample */,
+				8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */,
+				88E10F412B110D5300C08E95 /* ChatSample */,
+			);
+		};
+/* End PBXProject section */
+
+/* Begin PBXResourcesBuildPhase section */
+		8848C82D2B0D04BC007B434F /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */,
+				8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */,
+				869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8422B0D051E007B434F /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */,
+				8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */,
+				869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8542B0D056C007B434F /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */,
+				8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */,
+				869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		88E10F402B110D5300C08E95 /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */,
+				88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */,
+				869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXResourcesBuildPhase section */
+
+/* Begin PBXSourcesBuildPhase section */
+		8848C82B2B0D04BC007B434F /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */,
+				886F95DE2B17D5010036F07A /* ChatMessage.swift in Sources */,
+				88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */,
+				8848C8352B0D04BC007B434F /* ContentView.swift in Sources */,
+				886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */,
+				8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */,
+				886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */,
+				886F95DD2B17D5010036F07A /* MessageView.swift in Sources */,
+				886F95DC2B17BAEF0036F07A /* PhotoReasoningScreen.swift in Sources */,
+				886F95DB2B17BAEF0036F07A /* PhotoReasoningViewModel.swift in Sources */,
+				886F95E12B17D5010036F07A /* ConversationScreen.swift in Sources */,
+				88263BF02B239C09008AB09B /* ErrorView.swift in Sources */,
+				886F95D62B17BA010036F07A /* SummarizeViewModel.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8402B0D051E007B434F /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */,
+				8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */,
+				88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8522B0D056C007B434F /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */,
+				880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */,
+				8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		88E10F3E2B110D5300C08E95 /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */,
+				88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */,
+				88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */,
+				88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */,
+				88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */,
+				88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */,
+				88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */,
+				889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXSourcesBuildPhase section */
+
+/* Begin XCBuildConfiguration section */
+		8848C83B2B0D04BD007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ALWAYS_SEARCH_USER_PATHS = NO;
+				ASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;
+				CLANG_ANALYZER_NONNULL = YES;
+				CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
+				CLANG_CXX_LANGUAGE_STANDARD = "gnu++20";
+				CLANG_ENABLE_MODULES = YES;
+				CLANG_ENABLE_OBJC_ARC = YES;
+				CLANG_ENABLE_OBJC_WEAK = YES;
+				CLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;
+				CLANG_WARN_BOOL_CONVERSION = YES;
+				CLANG_WARN_COMMA = YES;
+				CLANG_WARN_CONSTANT_CONVERSION = YES;
+				CLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;
+				CLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;
+				CLANG_WARN_DOCUMENTATION_COMMENTS = YES;
+				CLANG_WARN_EMPTY_BODY = YES;
+				CLANG_WARN_ENUM_CONVERSION = YES;
+				CLANG_WARN_INFINITE_RECURSION = YES;
+				CLANG_WARN_INT_CONVERSION = YES;
+				CLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;
+				CLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;
+				CLANG_WARN_OBJC_LITERAL_CONVERSION = YES;
+				CLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;
+				CLANG_WARN_QUOTED_INCLUDE_IN_FRAMEWORK_HEADER = YES;
+				CLANG_WARN_RANGE_LOOP_ANALYSIS = YES;
+				CLANG_WARN_STRICT_PROTOTYPES = YES;
+				CLANG_WARN_SUSPICIOUS_MOVE = YES;
+				CLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;
+				CLANG_WARN_UNREACHABLE_CODE = YES;
+				CLANG_WARN__DUPLICATE_METHOD_MATCH = YES;
+				COPY_PHASE_STRIP = NO;
+				DEBUG_INFORMATION_FORMAT = dwarf;
+				ENABLE_STRICT_OBJC_MSGSEND = YES;
+				ENABLE_TESTABILITY = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = YES;
+				GCC_C_LANGUAGE_STANDARD = gnu17;
+				GCC_DYNAMIC_NO_PIC = NO;
+				GCC_NO_COMMON_BLOCKS = YES;
+				GCC_OPTIMIZATION_LEVEL = 0;
+				GCC_PREPROCESSOR_DEFINITIONS = (
+					"DEBUG=1",
+					"$(inherited)",
+				);
+				GCC_WARN_64_TO_32_BIT_CONVERSION = YES;
+				GCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;
+				GCC_WARN_UNDECLARED_SELECTOR = YES;
+				GCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;
+				GCC_WARN_UNUSED_FUNCTION = YES;
+				GCC_WARN_UNUSED_VARIABLE = YES;
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LOCALIZATION_PREFERS_STRING_CATALOGS = YES;
+				MTL_ENABLE_DEBUG_INFO = INCLUDE_SOURCE;
+				MTL_FAST_MATH = YES;
+				ONLY_ACTIVE_ARCH = YES;
+				SDKROOT = iphoneos;
+				SWIFT_ACTIVE_COMPILATION_CONDITIONS = "DEBUG $(inherited)";
+				SWIFT_OPTIMIZATION_LEVEL = "-Onone";
+			};
+			name = Debug;
+		};
+		8848C83C2B0D04BD007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ALWAYS_SEARCH_USER_PATHS = NO;
+				ASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;
+				CLANG_ANALYZER_NONNULL = YES;
+				CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
+				CLANG_CXX_LANGUAGE_STANDARD = "gnu++20";
+				CLANG_ENABLE_MODULES = YES;
+				CLANG_ENABLE_OBJC_ARC = YES;
+				CLANG_ENABLE_OBJC_WEAK = YES;
+				CLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;
+				CLANG_WARN_BOOL_CONVERSION = YES;
+				CLANG_WARN_COMMA = YES;
+				CLANG_WARN_CONSTANT_CONVERSION = YES;
+				CLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;
+				CLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;
+				CLANG_WARN_DOCUMENTATION_COMMENTS = YES;
+				CLANG_WARN_EMPTY_BODY = YES;
+				CLANG_WARN_ENUM_CONVERSION = YES;
+				CLANG_WARN_INFINITE_RECURSION = YES;
+				CLANG_WARN_INT_CONVERSION = YES;
+				CLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;
+				CLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;
+				CLANG_WARN_OBJC_LITERAL_CONVERSION = YES;
+				CLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;
+				CLANG_WARN_QUOTED_INCLUDE_IN_FRAMEWORK_HEADER = YES;
+				CLANG_WARN_RANGE_LOOP_ANALYSIS = YES;
+				CLANG_WARN_STRICT_PROTOTYPES = YES;
+				CLANG_WARN_SUSPICIOUS_MOVE = YES;
+				CLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;
+				CLANG_WARN_UNREACHABLE_CODE = YES;
+				CLANG_WARN__DUPLICATE_METHOD_MATCH = YES;
+				COPY_PHASE_STRIP = NO;
+				DEBUG_INFORMATION_FORMAT = "dwarf-with-dsym";
+				ENABLE_NS_ASSERTIONS = NO;
+				ENABLE_STRICT_OBJC_MSGSEND = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = YES;
+				GCC_C_LANGUAGE_STANDARD = gnu17;
+				GCC_NO_COMMON_BLOCKS = YES;
+				GCC_WARN_64_TO_32_BIT_CONVERSION = YES;
+				GCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;
+				GCC_WARN_UNDECLARED_SELECTOR = YES;
+				GCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;
+				GCC_WARN_UNUSED_FUNCTION = YES;
+				GCC_WARN_UNUSED_VARIABLE = YES;
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LOCALIZATION_PREFERS_STRING_CATALOGS = YES;
+				MTL_ENABLE_DEBUG_INFO = NO;
+				MTL_FAST_MATH = YES;
+				SDKROOT = iphoneos;
+				SWIFT_COMPILATION_MODE = wholemodule;
+				VALIDATE_PRODUCT = YES;
+			};
+			name = Release;
+		};
+		8848C83E2B0D04BD007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		8848C83F2B0D04BD007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+		8848C8502B0D051F007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		8848C8512B0D051F007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+		8848C8622B0D056D007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		8848C8632B0D056D007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+		88E10F4D2B110D5400C08E95 /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		88E10F4E2B110D5400C08E95 /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+/* End XCBuildConfiguration section */
+
+/* Begin XCConfigurationList section */
+		8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C83B2B0D04BD007B434F /* Debug */,
+				8848C83C2B0D04BD007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C83E2B0D04BD007B434F /* Debug */,
+				8848C83F2B0D04BD007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C8502B0D051F007B434F /* Debug */,
+				8848C8512B0D051F007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C8622B0D056D007B434F /* Debug */,
+				8848C8632B0D056D007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				88E10F4D2B110D5400C08E95 /* Debug */,
+				88E10F4E2B110D5400C08E95 /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+/* End XCConfigurationList section */
+
+/* Begin XCLocalSwiftPackageReference section */
+		869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */ = {
+			isa = XCLocalSwiftPackageReference;
+			relativePath = ../..;
+		};
+/* End XCLocalSwiftPackageReference section */
+
+/* Begin XCRemoteSwiftPackageReference section */
+		88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */ = {
+			isa = XCRemoteSwiftPackageReference;
+			repositoryURL = "https://github.com/gonzalezreal/swift-markdown-ui";
+			requirement = {
+				kind = revision;
+				revision = 5df8a4adedd6ae4eb2455ef60ff75183984daeb8;
+			};
+		};
+		DEA09AC32B1FCE22001962D9 /* XCRemoteSwiftPackageReference "NetworkImage" */ = {
+			isa = XCRemoteSwiftPackageReference;
+			repositoryURL = "https://github.com/gonzalezreal/NetworkImage";
+			requirement = {
+				kind = revision;
+				revision = 7aff8d1b31148d32c5933d75557d42f6323ee3d1;
+			};
+		};
+/* End XCRemoteSwiftPackageReference section */
+
+/* Begin XCSwiftPackageProductDependency section */
+		869200AA2B86BE3100482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		869200AC2B86BEA100482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		869200AE2B86BFBC00482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		869200B02B86BFF500482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		88209C232B0FBE1700F64795 /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		886F95D72B17BA420036F07A /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		886F95E22B17D6630036F07A /* GenerativeAIUIComponents */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = GenerativeAIUIComponents;
+		};
+		88B8A91D2B0FC55100424728 /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = GenerativeAIUIComponents;
+		};
+		88D9474C2B14F27E008B5580 /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = GenerativeAIUIComponents;
+		};
+/* End XCSwiftPackageProductDependency section */
+	};
+	rootObject = 8848C8272B0D04BC007B434F /* Project object */;
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
new file mode 100644
index 00000000000..34331bf6eb7
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
@@ -0,0 +1,48 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import SwiftUI
+
+struct ContentView: View {
+  @StateObject
+  var viewModel = ConversationViewModel()
+
+  var body: some View {
+    NavigationStack {
+      List {
+        NavigationLink {
+          SummarizeScreen()
+        } label: {
+          Label("Text", systemImage: "doc.text")
+        }
+        NavigationLink {
+          PhotoReasoningScreen()
+        } label: {
+          Label("Multi-modal", systemImage: "doc.richtext")
+        }
+        NavigationLink {
+          ConversationScreen()
+            .environmentObject(viewModel)
+        } label: {
+          Label("Chat", systemImage: "ellipsis.message.fill")
+        }
+      }
+      .navigationTitle("Generative AI Samples")
+    }
+  }
+}
+
+#Preview {
+  ContentView()
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
new file mode 100644
index 00000000000..7c709111b7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
@@ -0,0 +1,29 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import SwiftUI
+
+@main
+struct GenerativeAISampleApp: App {
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      ContentView()
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
new file mode 100644
index 00000000000..c800e1f7be8
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
@@ -0,0 +1,31 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import SwiftUI
+
+@main
+struct GenerativeAITextSampleApp: App {
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      NavigationStack {
+        SummarizeScreen()
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
new file mode 100644
index 00000000000..8fbb89f4482
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -0,0 +1,74 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import MarkdownUI
+import SwiftUI
+
+struct SummarizeScreen: View {
+  @StateObject var viewModel = SummarizeViewModel()
+  @State var userInput = ""
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      Text("Enter some text, then tap on _Go_ to summarize it.")
+      HStack(alignment: .top) {
+        TextField("Enter text summarize", text: $userInput, axis: .vertical)
+          .textFieldStyle(.roundedBorder)
+          .onSubmit {
+            onSummarizeTapped()
+          }
+        Button("Go") {
+          onSummarizeTapped()
+        }
+        .padding(.top, 4)
+      }
+      .padding([.horizontal, .bottom])
+
+      List {
+        HStack(alignment: .top) {
+          if viewModel.inProgress {
+            ProgressView()
+          } else {
+            Image(systemName: "cloud.circle.fill")
+              .font(.title2)
+          }
+
+          Markdown("\(viewModel.outputText)")
+        }
+        .listRowSeparator(.hidden)
+      }
+      .listStyle(.plain)
+    }
+    .navigationTitle("Text sample")
+  }
+
+  private func onSummarizeTapped() {
+    Task {
+      await viewModel.summarize(inputText: userInput)
+    }
+  }
+}
+
+#Preview {
+  NavigationStack {
+    SummarizeScreen()
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
new file mode 100644
index 00000000000..fb5b349ac82
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -0,0 +1,68 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import OSLog
+
+@MainActor
+class SummarizeViewModel: ObservableObject {
+  private var logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "generative-ai")
+
+  @Published
+  var outputText = ""
+
+  @Published
+  var errorMessage: String?
+
+  @Published
+  var inProgress = false
+
+  private var model: GenerativeModel?
+
+  init() {
+    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+  }
+
+  func summarize(inputText: String) async {
+    defer {
+      inProgress = false
+    }
+    guard let model else {
+      return
+    }
+
+    do {
+      inProgress = true
+      errorMessage = nil
+      outputText = ""
+
+      let prompt = "Summarize the following text for me: \(inputText)"
+
+      let outputContentStream = model.generateContentStream(prompt)
+
+      // stream response
+      for try await outputContent in outputContentStream {
+        guard let line = outputContent.text else {
+          return
+        }
+
+        outputText = outputText + line
+      }
+    } catch {
+      logger.error("\(error.localizedDescription)")
+      errorMessage = error.localizedDescription
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift
new file mode 100644
index 00000000000..808f5f42a97
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift
@@ -0,0 +1,35 @@
+// swift-tools-version: 5.9
+// The swift-tools-version declares the minimum version of Swift required to build this package.
+
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+import PackageDescription
+
+let package = Package(
+  name: "GenerativeAIUIComponents",
+  platforms: [
+    .iOS(.v16),
+  ],
+  products: [
+    .library(
+      name: "GenerativeAIUIComponents",
+      targets: ["GenerativeAIUIComponents"]
+    ),
+  ],
+  targets: [
+    .target(
+      name: "GenerativeAIUIComponents"
+    ),
+  ]
+)
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
new file mode 100644
index 00000000000..317baaaf3c1
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
@@ -0,0 +1,82 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import SwiftUI
+public struct InputField<Label>: View where Label: View {
+  @Binding
+  private var text: String
+
+  private var title: String?
+  private var label: () -> Label
+
+  @Environment(\.submitHandler)
+  var submitHandler
+
+  private func submit() {
+    if let submitHandler {
+      submitHandler()
+    }
+  }
+
+  public init(_ title: String? = nil, text: Binding<String>,
+              @ViewBuilder label: @escaping () -> Label) {
+    self.title = title
+    _text = text
+    self.label = label
+  }
+
+  public var body: some View {
+    VStack(alignment: .leading) {
+      HStack(alignment: .bottom) {
+        VStack(alignment: .leading) {
+          TextField(
+            title ?? "",
+            text: $text,
+            axis: .vertical
+          )
+          .padding(.vertical, 4)
+          .onSubmit(submit)
+        }
+        .padding(.horizontal, 8)
+        .padding(.vertical, 4)
+        .overlay {
+          RoundedRectangle(
+            cornerRadius: 8,
+            style: .continuous
+          )
+          .stroke(Color(UIColor.systemFill), lineWidth: 1)
+        }
+
+        Button(action: submit, label: label)
+          .padding(.top, 4)
+      }
+    }
+    .padding(.horizontal)
+  }
+}
+
+#Preview {
+  struct Wrapper: View {
+    @State var userInput: String = ""
+
+    var body: some View {
+      InputField("Message", text: $userInput) {
+        Image(systemName: "arrow.up.circle.fill")
+          .font(.title)
+      }
+    }
+  }
+
+  return Wrapper()
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
new file mode 100644
index 00000000000..9519fb38a4b
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
@@ -0,0 +1,183 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import PhotosUI
+import SwiftUI
+
+struct MultimodalInputFieldSubmitHandler: EnvironmentKey {
+  static var defaultValue: (() -> Void)?
+}
+
+extension EnvironmentValues {
+  var submitHandler: (() -> Void)? {
+    get { self[MultimodalInputFieldSubmitHandler.self] }
+    set { self[MultimodalInputFieldSubmitHandler.self] = newValue }
+  }
+}
+
+public extension View {
+  func onSubmit(submitHandler: @escaping () -> Void) -> some View {
+    environment(\.submitHandler, submitHandler)
+  }
+}
+
+public struct MultimodalInputField: View {
+  @Binding public var text: String
+  @Binding public var selection: [PhotosPickerItem]
+
+  @Environment(\.submitHandler) var submitHandler
+
+  @State private var selectedImages = [Image]()
+
+  @State private var isChooseAttachmentTypePickerShowing = false
+  @State private var isAttachmentPickerShowing = false
+
+  private func showChooseAttachmentTypePicker() {
+    isChooseAttachmentTypePickerShowing.toggle()
+  }
+
+  private func showAttachmentPicker() {
+    isAttachmentPickerShowing.toggle()
+  }
+
+  private func submit() {
+    if let submitHandler {
+      submitHandler()
+    }
+  }
+
+  public init(text: Binding<String>,
+              selection: Binding<[PhotosPickerItem]>) {
+    _text = text
+    _selection = selection
+  }
+
+  public var body: some View {
+    VStack(alignment: .leading) {
+      HStack(alignment: .top) {
+        Button(action: showChooseAttachmentTypePicker) {
+          Image(systemName: "plus")
+        }
+        .padding(.top, 4)
+
+        VStack(alignment: .leading) {
+          TextField(
+            "Upload an image, and then ask a question about it",
+            text: $text,
+            axis: .vertical
+          )
+          .padding(.vertical, 4)
+          .onSubmit(submit)
+
+          if selectedImages.count > 0 {
+            ScrollView(.horizontal) {
+              LazyHStack {
+                ForEach(0 ..< selectedImages.count, id: \.self) { i in
+                  HStack {
+                    selectedImages[i]
+                      .resizable()
+                      .scaledToFill()
+                      .frame(width: 50, height: 50)
+                      .cornerRadius(8)
+                  }
+                }
+              }
+            }
+            .frame(height: 50)
+          }
+        }
+        .padding(.horizontal, 8)
+        .padding(.vertical, 4)
+        .overlay {
+          RoundedRectangle(
+            cornerRadius: 8,
+            style: .continuous
+          )
+          .stroke(Color(UIColor.systemFill), lineWidth: 1)
+        }
+
+        Button(action: submit) {
+          Text("Go")
+        }
+        .padding(.top, 4)
+      }
+    }
+    .padding(.horizontal)
+    .confirmationDialog(
+      "Select an image",
+      isPresented: $isChooseAttachmentTypePickerShowing,
+      titleVisibility: .hidden
+    ) {
+      Button(action: showAttachmentPicker) {
+        Text("Photo & Video Library")
+      }
+    }
+    .photosPicker(isPresented: $isAttachmentPickerShowing, selection: $selection)
+    .onChange(of: selection) { _ in
+      Task {
+        selectedImages.removeAll()
+
+        for item in selection {
+          if let data = try? await item.loadTransferable(type: Data.self) {
+            if let uiImage = UIImage(data: data) {
+              let image = Image(uiImage: uiImage)
+              selectedImages.append(image)
+            }
+          }
+        }
+      }
+    }
+  }
+}
+
+#Preview {
+  struct Wrapper: View {
+    @State var userInput: String = ""
+    @State var selectedItems = [PhotosPickerItem]()
+
+    @State private var selectedImages = [Image]()
+
+    var body: some View {
+      MultimodalInputField(text: $userInput, selection: $selectedItems)
+        .onChange(of: selectedItems) { _ in
+          Task {
+            selectedImages.removeAll()
+
+            for item in selectedItems {
+              if let data = try? await item.loadTransferable(type: Data.self) {
+                if let uiImage = UIImage(data: data) {
+                  let image = Image(uiImage: uiImage)
+                  selectedImages.append(image)
+                }
+              }
+            }
+          }
+        }
+
+      List {
+        ForEach(0 ..< $selectedImages.count, id: \.self) { i in
+          HStack {
+            selectedImages[i]
+              .resizable()
+              .scaledToFill()
+              .frame(width: .infinity)
+              .cornerRadius(8)
+          }
+        }
+      }
+    }
+  }
+
+  return Wrapper()
+}
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 02fdb82807f..8a5b7b05415 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -16,7 +16,9 @@ import Foundation
 
 import FirebaseAppCheckInterop
 import FirebaseCore
-import GoogleGenerativeAI
+
+// Exports the GoogleGenerativeAI module to users of the SDK.
+@_exported import GoogleGenerativeAI
 
 // Avoids exposing internal FirebaseCore APIs to Swift users.
 @_implementationOnly import FirebaseCoreExtension
@@ -90,11 +92,9 @@ open class VertexAI: NSObject {
 
   lazy var model: GenerativeModel = {
     let options = RequestOptions(
-      endpoint: "\(location)-aiplatform.googleapis.com",
-      hooks: [
-        addAccessTokenHeader,
-        addAppCheckHeader,
-      ]
+      apiVersion: "v2beta",
+      endpoint: "staging-firebaseml.sandbox.googleapis.com",
+      hooks: [addAppCheckHeader]
     )
     return GenerativeModel(
       name: modelResouceName,
@@ -103,8 +103,6 @@ open class VertexAI: NSObject {
     )
   }()
 
-  private static let accessTokenEnvKey = "FIRVertexAIAccessToken"
-
   init(app: FirebaseApp, location: String, modelResourceName: String) {
     self.app = app
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
@@ -127,29 +125,6 @@ open class VertexAI: NSObject {
 
   // MARK: Request Hooks
 
-  /// Add a Google Cloud access token in an Authorization header in the provided request.
-  ///
-  /// This is a temporary workaround until Vertex AI can be called with an API key.
-  ///
-  /// - Parameter request: The `URLRequest` to modify by adding an access token.
-  func addAccessTokenHeader(request: inout URLRequest) {
-    // Remove the API key header, it is not supported by Vertex AI.
-    if var headers = request.allHTTPHeaderFields {
-      headers.removeValue(forKey: "x-goog-api-key")
-    }
-
-    guard let accessToken = ProcessInfo.processInfo.environment[VertexAI.accessTokenEnvKey] else {
-      print("""
-      Vertex AI requires an Access Token for authorization:
-      1. Get an access token by running `gcloud auth print-access-token`
-      2. Set it in the \(VertexAI.accessTokenEnvKey) environment variable.
-      """)
-      return
-    }
-
-    request.addValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
-  }
-
   /// Adds an App Check token to the provided request, if App Check is included in the app.
   ///
   /// This demonstrates how an App Check token can be added to requests; it is currently ignored by

From 202192f4bbf563866525fe336dd848a9e1057d5a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 22 Feb 2024 11:26:08 -0500
Subject: [PATCH 008/124] Update generative-ai-swift dependency

---
 FirebaseVertexAI/Sources/VertexAI.swift | 38 ++-----------------------
 Package.swift                           |  2 +-
 2 files changed, 3 insertions(+), 37 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 8a5b7b05415..850ed74a226 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -28,55 +28,21 @@ import FirebaseCore
 open class VertexAI: NSObject {
   // MARK: - Public APIs
 
-  // Option 1: Return `GoogleGenerativeAI.GenerativeModel` instances.
-
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   ///
   /// This instance is configured with the default `FirebaseApp`.
   public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
     .GenerativeModel {
-    return vertexAI(modelName: modelName, location: location).model
+    return generativeModel(app: FirebaseApp.app()!, modelName: modelName, location: location)
   }
 
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   public static func generativeModel(app: FirebaseApp, modelName: String,
                                      location: String) -> GoogleGenerativeAI.GenerativeModel {
-    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location).model
-  }
-
-  // Option 2: Return `VertexAI` instances with a similar API surface to
-  // `GoogleGenerativeAI.GenerativeModel`.
-  // Some types are re-used from the Google AI SDK for requests and responses, e.g.,
-  // `GenerateContentResponse`.
-
-  /// The default `VertexAI` instance.
-  ///
-  /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI(modelName: String, location: String) -> VertexAI {
-    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location)
-  }
-
-  public static func vertexAI(app: FirebaseApp, modelName: String, location: String) -> VertexAI {
     let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                             in: app.container)
     let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
-    return provider.vertexAI(location: location, modelResourceName: modelResourceName)
-  }
-
-  public func generateContentStream(_ parts: GoogleGenerativeAI
-    .PartsRepresentable...)
-    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
-    return model.generateContentStream([GoogleGenerativeAI.ModelContent(parts: parts)])
-  }
-
-  public func generateContentStream(_ content: [GoogleGenerativeAI.ModelContent])
-    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
-    return model.generateContentStream(content)
-  }
-
-  public func startChat(history: [GoogleGenerativeAI.ModelContent] = []) -> GoogleGenerativeAI
-    .Chat {
-    return model.startChat(history: history)
+    return provider.vertexAI(location: location, modelResourceName: modelResourceName).model
   }
 
   // MARK: - Private
diff --git a/Package.swift b/Package.swift
index a491ecfacd2..78d7cfa9967 100644
--- a/Package.swift
+++ b/Package.swift
@@ -189,7 +189,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "46e56ae138f152528b7b50520a22777f3d8adb18"
+      revision: "ae0c9b580d2ea5d6003739b1e14cc5c62f2ffa62"
     ),
   ],
   targets: [

From 9698be2ad1722b681ab150cae8850414ecb78d20 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 22 Feb 2024 11:47:55 -0500
Subject: [PATCH 009/124] Fix optional VertexAIProvider unwrapping

---
 FirebaseVertexAI/Sources/VertexAI.swift | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 850ed74a226..9d1b8b4016c 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -39,10 +39,14 @@ open class VertexAI: NSObject {
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   public static func generativeModel(app: FirebaseApp, modelName: String,
                                      location: String) -> GoogleGenerativeAI.GenerativeModel {
-    let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
-                                                            in: app.container)
+    guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                                  in: app.container) else {
+      fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
+    }
     let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
-    return provider.vertexAI(location: location, modelResourceName: modelResourceName).model
+    let vertexAI = provider.vertexAI(location: location, modelResourceName: modelResourceName)
+
+    return vertexAI.model
   }
 
   // MARK: - Private

From d0b4d1b9a59c54481d21a7e1fec6022dfbb916a1 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 23 Feb 2024 14:36:12 -0500
Subject: [PATCH 010/124] Update `generative-ai-swift` dependency

---
 Package.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Package.swift b/Package.swift
index 78d7cfa9967..9abcecd7edd 100644
--- a/Package.swift
+++ b/Package.swift
@@ -189,7 +189,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "ae0c9b580d2ea5d6003739b1e14cc5c62f2ffa62"
+      revision: "c9f2c4913bc65aa267815962c7e91358c2d8463f"
     ),
   ],
   targets: [

From 25bda5a7c2994a1d021c7db632fc6fc8f6bcd43c Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 4 Mar 2024 16:50:37 -0500
Subject: [PATCH 011/124] Prototype and sample fixes

---
 .../ChatSample/Views/ErrorDetailsView.swift   | 19 ++-------------
 .../ViewModels/PhotoReasoningViewModel.swift  |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift       | 24 ++++++++++++++-----
 3 files changed, 21 insertions(+), 24 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
index e601876b2d9..399777ad1e8 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -152,23 +152,8 @@ struct ErrorDetailsView: View {
             SubtitleMarkdownFormRow(
               title: "Help",
               value: """
-              Please provide a valid value for `API_KEY` in the `GenerativeAI-Info.plist` file.
-              """
-            )
-          }
-
-        case GenerateContentError.unsupportedUserLocation:
-          Section("Error Type") {
-            Text("Unsupported User Location")
-          }
-
-          Section("Details") {
-            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
-            SubtitleMarkdownFormRow(
-              title: "Help",
-              value: """
-              The API is unsupported in your location (country / territory); please see the list of
-              [available regions](https://ai.google.dev/available_regions#available_regions).
+              The `API_KEY` provided in the `GoogleService-Info.plist` file is invalid. Download a
+              new copy of the file from the [Firebase Console](https://console.firebase.google.com).
               """
             )
           }
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index aa646c127df..dc41e00444a 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -62,7 +62,7 @@ class PhotoReasoningViewModel: ObservableObject {
 
       let prompt = "Look at the image(s), and then answer the following question: \(userInput)"
 
-      var images = [PartsRepresentable]()
+      var images = [any ThrowingPartsRepresentable]()
       for item in selectedItems {
         if let data = try? await item.loadTransferable(type: Data.self) {
           guard let image = UIImage(data: data) else {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 9d1b8b4016c..8fb0f9e8c68 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -33,7 +33,10 @@ open class VertexAI: NSObject {
   /// This instance is configured with the default `FirebaseApp`.
   public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
     .GenerativeModel {
-    return generativeModel(app: FirebaseApp.app()!, modelName: modelName, location: location)
+    guard let app = FirebaseApp.app() else {
+      fatalError("No instance of the default Firebase app was found.")
+    }
+    return generativeModel(app: app, modelName: modelName, location: location)
   }
 
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
@@ -66,9 +69,12 @@ open class VertexAI: NSObject {
       endpoint: "staging-firebaseml.sandbox.googleapis.com",
       hooks: [addAppCheckHeader]
     )
+    guard let apiKey = app.options.apiKey else {
+      fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
+    }
     return GenerativeModel(
       name: modelResouceName,
-      apiKey: app.options.apiKey!,
+      apiKey: apiKey,
       requestOptions: options
     )
   }()
@@ -86,8 +92,14 @@ open class VertexAI: NSObject {
       return modelName
     }
     guard let projectID = app.options.projectID else {
-      print("The FirebaseApp is missing a project ID.")
-      return modelName
+      fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
+    }
+    guard !location.isEmpty else {
+      fatalError("""
+      No location specified; see
+      https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
+      list of available regions.
+      """)
     }
 
     return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
@@ -95,14 +107,14 @@ open class VertexAI: NSObject {
 
   // MARK: Request Hooks
 
-  /// Adds an App Check token to the provided request, if App Check is included in the app.
+  /// Adds an App Check token to the provided request if App Check is included in the app.
   ///
   /// This demonstrates how an App Check token can be added to requests; it is currently ignored by
   /// the backend.
   ///
   /// - Parameter request: The `URLRequest` to modify by adding an App Check token header.
   func addAppCheckHeader(request: inout URLRequest) async {
-    guard let appCheck = appCheck else {
+    guard let appCheck else {
       return
     }
 

From 6174462ea74d63d4dbb3c5632f2ebdf358f2c514 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 18 Mar 2024 19:27:53 +0000
Subject: [PATCH 012/124] Fork `google-generative-ai` for Firebase (#12564)

---
 FirebaseVertexAI/Sources/Chat.swift           | 184 +++++++++++
 .../Sources/CountTokensRequest.swift          |  45 +++
 FirebaseVertexAI/Sources/Errors.swift         | 189 +++++++++++
 .../Sources/GenerateContentError.swift        |  44 +++
 .../Sources/GenerateContentRequest.swift      |  49 +++
 .../Sources/GenerateContentResponse.swift     | 286 +++++++++++++++++
 .../Sources/GenerationConfig.swift            |  86 +++++
 .../Sources/GenerativeAIRequest.swift         |  46 +++
 .../Sources/GenerativeAIService.swift         | 257 +++++++++++++++
 .../Sources/GenerativeAISwift.swift           |  26 ++
 .../Sources/GenerativeModel.swift             | 298 ++++++++++++++++++
 FirebaseVertexAI/Sources/Logging.swift        |  56 ++++
 FirebaseVertexAI/Sources/ModelContent.swift   | 141 +++++++++
 .../Sources/PartsRepresentable+Image.swift    | 107 +++++++
 .../Sources/PartsRepresentable.swift          |  66 ++++
 FirebaseVertexAI/Sources/Safety.swift         | 182 +++++++++++
 Package.swift                                 |   6 +-
 17 files changed, 2067 insertions(+), 1 deletion(-)
 create mode 100644 FirebaseVertexAI/Sources/Chat.swift
 create mode 100644 FirebaseVertexAI/Sources/CountTokensRequest.swift
 create mode 100644 FirebaseVertexAI/Sources/Errors.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerateContentError.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerateContentRequest.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerateContentResponse.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerationConfig.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeAIRequest.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeAIService.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeAISwift.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeModel.swift
 create mode 100644 FirebaseVertexAI/Sources/Logging.swift
 create mode 100644 FirebaseVertexAI/Sources/ModelContent.swift
 create mode 100644 FirebaseVertexAI/Sources/PartsRepresentable+Image.swift
 create mode 100644 FirebaseVertexAI/Sources/PartsRepresentable.swift
 create mode 100644 FirebaseVertexAI/Sources/Safety.swift

diff --git a/FirebaseVertexAI/Sources/Chat.swift b/FirebaseVertexAI/Sources/Chat.swift
new file mode 100644
index 00000000000..c7cfb859fa8
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Chat.swift
@@ -0,0 +1,184 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// An object that represents a back-and-forth chat with a model, capturing the history and saving
+/// the context in memory between each message sent.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public class Chat {
+  private let model: GenerativeModel
+
+  /// Initializes a new chat representing a 1:1 conversation between model and user.
+  init(model: GenerativeModel, history: [ModelContent]) {
+    self.model = model
+    self.history = history
+  }
+
+  /// The previous content from the chat that has been successfully sent and received from the
+  /// model. This will be provided to the model for each message sent as context for the discussion.
+  public var history: [ModelContent]
+
+  /// See ``sendMessage(_:)-3ify5``.
+  public func sendMessage(_ parts: any ThrowingPartsRepresentable...) async throws
+    -> GenerateContentResponse {
+    return try await sendMessage([ModelContent(parts: parts)])
+  }
+
+  /// Sends a message using the existing history of this chat as context. If successful, the message
+  /// and response will be added to the history. If unsuccessful, history will remain unchanged.
+  /// - Parameter content: The new content to send as a single chat message.
+  /// - Returns: The model's response if no error occurred.
+  /// - Throws: A ``GenerateContentError`` if an error occurred.
+  public func sendMessage(_ content: @autoclosure () throws -> [ModelContent]) async throws
+    -> GenerateContentResponse {
+    // Ensure that the new content has the role set.
+    let newContent: [ModelContent]
+    do {
+      newContent = try content().map(populateContentRole(_:))
+    } catch let underlying {
+      if let contentError = underlying as? ImageConversionError {
+        throw GenerateContentError.promptImageContentError(underlying: contentError)
+      } else {
+        throw GenerateContentError.internalError(underlying: underlying)
+      }
+    }
+
+    // Send the history alongside the new message as context.
+    let request = history + newContent
+    let result = try await model.generateContent(request)
+    guard let reply = result.candidates.first?.content else {
+      let error = NSError(domain: "com.google.generative-ai",
+                          code: -1,
+                          userInfo: [
+                            NSLocalizedDescriptionKey: "No candidates with content available.",
+                          ])
+      throw GenerateContentError.internalError(underlying: error)
+    }
+
+    // Make sure we inject the role into the content received.
+    let toAdd = ModelContent(role: "model", parts: reply.parts)
+
+    // Append the request and successful result to history, then return the value.
+    history.append(contentsOf: newContent)
+    history.append(toAdd)
+    return result
+  }
+
+  /// See ``sendMessageStream(_:)-4abs3``.
+  @available(macOS 12.0, *)
+  public func sendMessageStream(_ parts: any ThrowingPartsRepresentable...)
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    return try sendMessageStream([ModelContent(parts: parts)])
+  }
+
+  /// Sends a message using the existing history of this chat as context. If successful, the message
+  /// and response will be added to the history. If unsuccessful, history will remain unchanged.
+  /// - Parameter content: The new content to send as a single chat message.
+  /// - Returns: A stream containing the model's response or an error if an error occurred.
+  @available(macOS 12.0, *)
+  public func sendMessageStream(_ content: @autoclosure () throws -> [ModelContent])
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    let resolvedContent: [ModelContent]
+    do {
+      resolvedContent = try content()
+    } catch let underlying {
+      return AsyncThrowingStream { continuation in
+        let error: Error
+        if let contentError = underlying as? ImageConversionError {
+          error = GenerateContentError.promptImageContentError(underlying: contentError)
+        } else {
+          error = GenerateContentError.internalError(underlying: underlying)
+        }
+        continuation.finish(throwing: error)
+      }
+    }
+
+    return AsyncThrowingStream { continuation in
+      Task {
+        var aggregatedContent: [ModelContent] = []
+
+        // Ensure that the new content has the role set.
+        let newContent: [ModelContent] = resolvedContent.map(populateContentRole(_:))
+
+        // Send the history alongside the new message as context.
+        let request = history + newContent
+        let stream = model.generateContentStream(request)
+        do {
+          for try await chunk in stream {
+            // Capture any content that's streaming. This should be populated if there's no error.
+            if let chunkContent = chunk.candidates.first?.content {
+              aggregatedContent.append(chunkContent)
+            }
+
+            // Pass along the chunk.
+            continuation.yield(chunk)
+          }
+        } catch {
+          // Rethrow the error that the underlying stream threw. Don't add anything to history.
+          continuation.finish(throwing: error)
+          return
+        }
+
+        // Save the request.
+        history.append(contentsOf: newContent)
+
+        // Aggregate the content to add it to the history before we finish.
+        let aggregated = aggregatedChunks(aggregatedContent)
+        history.append(aggregated)
+
+        continuation.finish()
+      }
+    }
+  }
+
+  private func aggregatedChunks(_ chunks: [ModelContent]) -> ModelContent {
+    var parts: [ModelContent.Part] = []
+    var combinedText = ""
+    for aggregate in chunks {
+      // Loop through all the parts, aggregating the text and adding the images.
+      for part in aggregate.parts {
+        switch part {
+        case let .text(str):
+          combinedText += str
+
+        case .data(mimetype: _, _):
+          // Don't combine it, just add to the content. If there's any text pending, add that as
+          // a part.
+          if !combinedText.isEmpty {
+            parts.append(.text(combinedText))
+            combinedText = ""
+          }
+
+          parts.append(part)
+        }
+      }
+    }
+
+    if !combinedText.isEmpty {
+      parts.append(.text(combinedText))
+    }
+
+    return ModelContent(role: "model", parts: parts)
+  }
+
+  /// Populates the `role` field with `user` if it doesn't exist. Required in chat sessions.
+  private func populateContentRole(_ content: ModelContent) -> ModelContent {
+    if content.role != nil {
+      return content
+    } else {
+      return ModelContent(role: "user", parts: content.parts)
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
new file mode 100644
index 00000000000..0a58d40acc0
--- /dev/null
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -0,0 +1,45 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct CountTokensRequest {
+  let model: String
+  let contents: [ModelContent]
+  let options: RequestOptions
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensRequest: Encodable {
+  enum CodingKeys: CodingKey {
+    case contents
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensRequest: GenerativeAIRequest {
+  typealias Response = CountTokensResponse
+
+  var url: URL {
+    URL(string: "\(GenerativeAISwift.baseURL)/\(options.apiVersion)/\(model):countTokens")!
+  }
+}
+
+/// The model's response to a count tokens request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct CountTokensResponse: Decodable {
+  /// The total number of tokens in the input given to the model as a prompt.
+  public let totalTokens: Int
+}
diff --git a/FirebaseVertexAI/Sources/Errors.swift b/FirebaseVertexAI/Sources/Errors.swift
new file mode 100644
index 00000000000..35e7e6b813b
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Errors.swift
@@ -0,0 +1,189 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+struct RPCError: Error {
+  let httpResponseCode: Int
+  let message: String
+  let status: RPCStatus
+  let details: [ErrorDetails]
+
+  private var errorInfo: ErrorDetails? {
+    return details.first { $0.isErrorInfo() }
+  }
+
+  init(httpResponseCode: Int, message: String, status: RPCStatus, details: [ErrorDetails]) {
+    self.httpResponseCode = httpResponseCode
+    self.message = message
+    self.status = status
+    self.details = details
+  }
+
+  func isInvalidAPIKeyError() -> Bool {
+    return errorInfo?.reason == "API_KEY_INVALID"
+  }
+
+  func isUnsupportedUserLocationError() -> Bool {
+    return message == RPCErrorMessage.unsupportedUserLocation.rawValue
+  }
+}
+
+extension RPCError: Decodable {
+  enum CodingKeys: CodingKey {
+    case error
+  }
+
+  init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    let status = try container.decode(ErrorStatus.self, forKey: .error)
+
+    if let code = status.code {
+      httpResponseCode = code
+    } else {
+      httpResponseCode = -1
+    }
+
+    if let message = status.message {
+      self.message = message
+    } else {
+      message = "Unknown error."
+    }
+
+    if let rpcStatus = status.status {
+      self.status = rpcStatus
+    } else {
+      self.status = .unknown
+    }
+
+    details = status.details
+  }
+}
+
+struct ErrorStatus {
+  let code: Int?
+  let message: String?
+  let status: RPCStatus?
+  let details: [ErrorDetails]
+}
+
+struct ErrorDetails {
+  static let errorInfoType = "type.googleapis.com/google.rpc.ErrorInfo"
+
+  let type: String
+  let reason: String?
+  let domain: String?
+
+  func isErrorInfo() -> Bool {
+    return type == ErrorDetails.errorInfoType
+  }
+}
+
+extension ErrorDetails: Decodable, Equatable {
+  enum CodingKeys: String, CodingKey {
+    case type = "@type"
+    case reason
+    case domain
+  }
+}
+
+extension ErrorStatus: Decodable {
+  enum CodingKeys: CodingKey {
+    case code
+    case message
+    case status
+    case details
+  }
+
+  init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    code = try container.decodeIfPresent(Int.self, forKey: .code)
+    message = try container.decodeIfPresent(String.self, forKey: .message)
+    do {
+      status = try container.decodeIfPresent(RPCStatus.self, forKey: .status)
+    } catch {
+      status = .unknown
+    }
+    if container.contains(.details) {
+      details = try container.decode([ErrorDetails].self, forKey: .details)
+    } else {
+      details = []
+    }
+  }
+}
+
+enum RPCStatus: String, Decodable {
+  // Not an error; returned on success.
+  case ok = "OK"
+
+  // The operation was cancelled, typically by the caller.
+  case cancelled = "CANCELLED"
+
+  // Unknown error.
+  case unknown = "UNKNOWN"
+
+  // The client specified an invalid argument.
+  case invalidArgument = "INVALID_ARGUMENT"
+
+  // The deadline expired before the operation could complete.
+  case deadlineExceeded = "DEADLINE_EXCEEDED"
+
+  // Some requested entity (e.g., file or directory) was not found.
+  case notFound = "NOT_FOUND"
+
+  // The entity that a client attempted to create (e.g., file or directory) already exists.
+  case alreadyExists = "ALREADY_EXISTS"
+
+  // The caller does not have permission to execute the specified operation.
+  case permissionDenied = "PERMISSION_DENIED"
+
+  // The request does not have valid authentication credentials for the operation.
+  case unauthenticated = "UNAUTHENTICATED"
+
+  // Some resource has been exhausted, perhaps a per-user quota, or perhaps the entire file system
+  // is out of space.
+  case resourceExhausted = "RESOURCE_EXHAUSTED"
+
+  // The operation was rejected because the system is not in a state required for the operation's
+  // execution.
+  case failedPrecondition = "FAILED_PRECONDITION"
+
+  // The operation was aborted, typically due to a concurrency issue such as a sequencer check
+  // failure or transaction abort.
+  case aborted = "ABORTED"
+
+  // The operation was attempted past the valid range.
+  case outOfRange = "OUT_OF_RANGE"
+
+  // The operation is not implemented or is not supported/enabled in this service.
+  case unimplemented = "UNIMPLEMENTED"
+
+  // Internal errors.
+  case internalError = "INTERNAL"
+
+  // The service is currently unavailable.
+  case unavailable = "UNAVAILABLE"
+
+  // Unrecoverable data loss or corruption.
+  case dataLoss = "DATA_LOSS"
+}
+
+enum RPCErrorMessage: String {
+  case unsupportedUserLocation = "User location is not supported for the API use."
+}
+
+enum InvalidCandidateError: Error {
+  case emptyContent(underlyingError: Error)
+  case malformedContent(underlyingError: Error)
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentError.swift b/FirebaseVertexAI/Sources/GenerateContentError.swift
new file mode 100644
index 00000000000..055928e5f1b
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerateContentError.swift
@@ -0,0 +1,44 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// Errors that occur when generating content from a model.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum GenerateContentError: Error {
+  /// An error occurred when constructing the prompt. Examine the related error for details.
+  case promptImageContentError(underlying: ImageConversionError)
+
+  /// An internal error occurred. See the underlying error for more context.
+  case internalError(underlying: Error)
+
+  /// A prompt was blocked. See the response's `promptFeedback.blockReason` for more information.
+  case promptBlocked(response: GenerateContentResponse)
+
+  /// A response didn't fully complete. See the `FinishReason` for more information.
+  case responseStoppedEarly(reason: FinishReason, response: GenerateContentResponse)
+
+  /// The provided API key is invalid.
+  case invalidAPIKey(message: String)
+
+  /// The user's location (region) is not supported by the API.
+  ///
+  /// See the Google documentation for a
+  /// [list of regions](https://ai.google.dev/available_regions#available_regions)
+  /// (countries and territories) where the API is available.
+  ///
+  /// - Important: The API is only available in
+  /// [specific regions](https://ai.google.dev/available_regions#available_regions).
+  case unsupportedUserLocation
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
new file mode 100644
index 00000000000..417260bb700
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -0,0 +1,49 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct GenerateContentRequest {
+  /// Model name.
+  let model: String
+  let contents: [ModelContent]
+  let generationConfig: GenerationConfig?
+  let safetySettings: [SafetySetting]?
+  let isStreaming: Bool
+  let options: RequestOptions
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentRequest: Encodable {
+  enum CodingKeys: String, CodingKey {
+    case contents
+    case generationConfig
+    case safetySettings
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentRequest: GenerativeAIRequest {
+  typealias Response = GenerateContentResponse
+
+  var url: URL {
+    let modelURL = "\(GenerativeAISwift.baseURL)/\(options.apiVersion)/\(model)"
+    if isStreaming {
+      return URL(string: "\(modelURL):streamGenerateContent?alt=sse")!
+    } else {
+      return URL(string: "\(modelURL):generateContent")!
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
new file mode 100644
index 00000000000..03153da9ae7
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -0,0 +1,286 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// The model's response to a generate content request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct GenerateContentResponse {
+  /// A list of candidate response content, ordered from best to worst.
+  public let candidates: [CandidateResponse]
+
+  /// A value containing the safety ratings for the response, or, if the request was blocked, a
+  /// reason for blocking the request.
+  public let promptFeedback: PromptFeedback?
+
+  /// The response's content as text, if it exists.
+  public var text: String? {
+    guard let candidate = candidates.first else {
+      Logging.default.error("Could not get text from a response that had no candidates.")
+      return nil
+    }
+    guard let text = candidate.content.parts.first?.text else {
+      Logging.default.error("Could not get a text part from the first candidate.")
+      return nil
+    }
+    return text
+  }
+
+  /// Initializer for SwiftUI previews or tests.
+  public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback?) {
+    self.candidates = candidates
+    self.promptFeedback = promptFeedback
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case candidates
+    case promptFeedback
+  }
+
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+
+    guard container.contains(CodingKeys.candidates) || container
+      .contains(CodingKeys.promptFeedback) else {
+      let context = DecodingError.Context(
+        codingPath: [],
+        debugDescription: "Failed to decode GenerateContentResponse;" +
+          " missing keys 'candidates' and 'promptFeedback'."
+      )
+      throw DecodingError.dataCorrupted(context)
+    }
+
+    if let candidates = try container.decodeIfPresent(
+      [CandidateResponse].self,
+      forKey: .candidates
+    ) {
+      self.candidates = candidates
+    } else {
+      candidates = []
+    }
+    promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
+  }
+}
+
+/// A struct representing a possible reply to a content generation prompt. Each content generation
+/// prompt may produce multiple candidate responses.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct CandidateResponse {
+  /// The response's content.
+  public let content: ModelContent
+
+  /// The safety rating of the response content.
+  public let safetyRatings: [SafetyRating]
+
+  /// The reason the model stopped generating content, if it exists; for example, if the model
+  /// generated a predefined stop sequence.
+  public let finishReason: FinishReason?
+
+  /// Cited works in the model's response content, if it exists.
+  public let citationMetadata: CitationMetadata?
+
+  /// Initializer for SwiftUI previews or tests.
+  public init(content: ModelContent, safetyRatings: [SafetyRating], finishReason: FinishReason?,
+              citationMetadata: CitationMetadata?) {
+    self.content = content
+    self.safetyRatings = safetyRatings
+    self.finishReason = finishReason
+    self.citationMetadata = citationMetadata
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CandidateResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case content
+    case safetyRatings
+    case finishReason
+    case finishMessage
+    case citationMetadata
+  }
+
+  /// Initializes a response from a decoder. Used for decoding server responses; not for public
+  /// use.
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+
+    do {
+      if let content = try container.decodeIfPresent(ModelContent.self, forKey: .content) {
+        self.content = content
+      } else {
+        content = ModelContent(parts: [])
+      }
+    } catch {
+      // Check if `content` can be decoded as an empty dictionary to detect the `"content": {}` bug.
+      if let content = try? container.decode([String: String].self, forKey: .content),
+         content.isEmpty {
+        throw InvalidCandidateError.emptyContent(underlyingError: error)
+      } else {
+        throw InvalidCandidateError.malformedContent(underlyingError: error)
+      }
+    }
+
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+
+    finishReason = try container.decodeIfPresent(FinishReason.self, forKey: .finishReason)
+
+    citationMetadata = try container.decodeIfPresent(
+      CitationMetadata.self,
+      forKey: .citationMetadata
+    )
+  }
+}
+
+/// A collection of source attributions for a piece of content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct CitationMetadata: Decodable {
+  /// A list of individual cited sources and the parts of the content to which they apply.
+  public let citationSources: [Citation]
+}
+
+/// A struct describing a source attribution.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct Citation: Decodable {
+  /// The inclusive beginning of a sequence in a model response that derives from a cited source.
+  public let startIndex: Int
+
+  /// The exclusive end of a sequence in a model response that derives from a cited source.
+  public let endIndex: Int
+
+  /// A link to the cited source.
+  public let uri: String
+
+  /// The license the cited source work is distributed under.
+  public let license: String
+}
+
+/// A value enumerating possible reasons for a model to terminate a content generation request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum FinishReason: String {
+  case unknown = "FINISH_REASON_UNKNOWN"
+
+  case unspecified = "FINISH_REASON_UNSPECIFIED"
+
+  /// Natural stop point of the model or provided stop sequence.
+  case stop = "STOP"
+
+  /// The maximum number of tokens as specified in the request was reached.
+  case maxTokens = "MAX_TOKENS"
+
+  /// The token generation was stopped because the response was flagged for safety reasons.
+  /// NOTE: When streaming, the Candidate.content will be empty if content filters blocked the
+  /// output.
+  case safety = "SAFETY"
+
+  /// The token generation was stopped because the response was flagged for unauthorized citations.
+  case recitation = "RECITATION"
+
+  /// All other reasons that stopped token generation.
+  case other = "OTHER"
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension FinishReason: Decodable {
+  /// Do not explicitly use. Initializer required for Decodable conformance.
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedFinishReason = FinishReason(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized FinishReason with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedFinishReason
+  }
+}
+
+/// A metadata struct containing any feedback the model had on the prompt it was provided.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct PromptFeedback {
+  /// A type describing possible reasons to block a prompt.
+  public enum BlockReason: String, Decodable {
+    /// The block reason is unknown.
+    case unknown = "UNKNOWN"
+
+    /// The block reason was not specified in the server response.
+    case unspecified = "BLOCK_REASON_UNSPECIFIED"
+
+    /// The prompt was blocked because it was deemed unsafe.
+    case safety = "SAFETY"
+
+    /// All other block reasons.
+    case other = "OTHER"
+
+    /// Do not explicitly use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedBlockReason = BlockReason(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized BlockReason with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedBlockReason
+    }
+  }
+
+  /// The reason a prompt was blocked, if it was blocked.
+  public let blockReason: BlockReason?
+
+  /// The safety ratings of the prompt.
+  public let safetyRatings: [SafetyRating]
+
+  /// Initializer for SwiftUI previews or tests.
+  public init(blockReason: BlockReason?, safetyRatings: [SafetyRating]) {
+    self.blockReason = blockReason
+    self.safetyRatings = safetyRatings
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension PromptFeedback: Decodable {
+  enum CodingKeys: CodingKey {
+    case blockReason
+    case safetyRatings
+  }
+
+  /// Do not explicitly use. Initializer required for Decodable conformance.
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    blockReason = try container.decodeIfPresent(
+      PromptFeedback.BlockReason.self,
+      forKey: .blockReason
+    )
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerationConfig.swift b/FirebaseVertexAI/Sources/GenerationConfig.swift
new file mode 100644
index 00000000000..2d1016c965c
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerationConfig.swift
@@ -0,0 +1,86 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A struct defining model parameters to be used when sending generative AI
+/// requests to the backend model.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct GenerationConfig: Encodable {
+  /// A parameter controlling the degree of randomness in token selection. A
+  /// temperature of zero is deterministic, always choosing the
+  /// highest-probability response. Typical values are between 0 and 1
+  /// inclusive. Defaults to 0 if unspecified.
+  public let temperature: Float?
+
+  /// The `topP` parameter changes how the model selects tokens for output.
+  /// Tokens are selected from the most to least probable until the sum of
+  /// their probabilities equals the `topP` value. For example, if tokens A, B,
+  /// and C have probabilities of 0.3, 0.2, and 0.1 respectively and the topP
+  /// value is 0.5, then the model will select either A or B as the next token
+  /// by using the `temperature` and exclude C as a candidate.
+  /// Defaults to 0.95 if unset.
+  public let topP: Float?
+
+  /// The `topK` parameter changes how the model selects tokens for output. A
+  /// `topK` of 1 means the selected token is the most probable among all the
+  /// tokens in the model's vocabulary, while a `topK` of 3 means that the next
+  /// token is selected from among the 3 most probable using the `temperature`.
+  /// For each token selection step, the `topK` tokens with the highest
+  /// probabilities are sampled. Tokens are then further filtered based on
+  /// `topP` with the final token selected using `temperature` sampling.
+  /// Defaults to 40 if unspecified.
+  public let topK: Int?
+
+  /// The maximum number of generated response messages to return. This value
+  /// must be between [1, 8], inclusive. If unset, this will default to 1.
+  ///
+  /// - Note: Only unique candidates are returned. Higher temperatures are more
+  ///     likely to produce unique candidates. Setting `temperature` to 0 will
+  ///     always produce exactly one candidate regardless of the
+  ///     `candidateCount`.
+  public let candidateCount: Int?
+
+  /// Specifies the maximum number of tokens that can be generated in the
+  /// response. The number of tokens per word varies depending on the
+  /// language outputted. The maximum value is capped at 1024. Defaults to 0
+  /// (unbounded).
+  public let maxOutputTokens: Int?
+
+  /// A set of up to 5 `String`s that will stop output generation. If
+  /// specified, the API will stop at the first appearance of a stop sequence.
+  /// The stop sequence will not be included as part of the response.
+  public let stopSequences: [String]?
+
+  /// Creates a new `GenerationConfig` value.
+  ///
+  /// - Parameter temperature: See ``temperature``
+  /// - Parameter topP: See ``topP``
+  /// - Parameter topK: See ``topK``
+  /// - Parameter candidateCount: See ``candidateCount``
+  /// - Parameter maxOutputTokens: See ``maxOutputTokens``
+  /// - Parameter stopSequences: See ``stopSequences``
+  public init(temperature: Float? = nil, topP: Float? = nil, topK: Int? = nil,
+              candidateCount: Int? = nil, maxOutputTokens: Int? = nil,
+              stopSequences: [String]? = nil) {
+    // Explicit init because otherwise if we re-arrange the above variables it changes the API
+    // surface.
+    self.temperature = temperature
+    self.topP = topP
+    self.topK = topK
+    self.candidateCount = candidateCount
+    self.maxOutputTokens = maxOutputTokens
+    self.stopSequences = stopSequences
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
new file mode 100644
index 00000000000..2bf94503b1a
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
@@ -0,0 +1,46 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+protocol GenerativeAIRequest: Encodable {
+  associatedtype Response: Decodable
+
+  var url: URL { get }
+
+  var options: RequestOptions { get }
+}
+
+/// Configuration parameters for sending requests to the backend.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct RequestOptions {
+  /// The request’s timeout interval in seconds; if not specified uses the default value for a
+  /// `URLRequest`.
+  let timeout: TimeInterval?
+
+  /// The API version to use in requests to the backend.
+  let apiVersion: String
+
+  /// Initializes a request options object.
+  ///
+  /// - Parameters:
+  ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
+  ///   for a `URLRequest`.
+  ///   - apiVersion The API version to use in requests to the backend; defaults to "v1".
+  public init(timeout: TimeInterval? = nil, apiVersion: String = "v1") {
+    self.timeout = timeout
+    self.apiVersion = apiVersion
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
new file mode 100644
index 00000000000..8d904738927
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -0,0 +1,257 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct GenerativeAIService {
+  /// Gives permission to talk to the backend.
+  private let apiKey: String
+
+  private let urlSession: URLSession
+
+  init(apiKey: String, urlSession: URLSession) {
+    self.apiKey = apiKey
+    self.urlSession = urlSession
+  }
+
+  func loadRequest<T: GenerativeAIRequest>(request: T) async throws -> T.Response {
+    let urlRequest = try urlRequest(request: request)
+
+    #if DEBUG
+      printCURLCommand(from: urlRequest)
+    #endif
+
+    let data: Data
+    let rawResponse: URLResponse
+    (data, rawResponse) = try await urlSession.data(for: urlRequest)
+
+    let response = try httpResponse(urlResponse: rawResponse)
+
+    // Verify the status code is 200
+    guard response.statusCode == 200 else {
+      Logging.default.error("[GoogleGenerativeAI] The server responded with an error: \(response)")
+      if let responseString = String(data: data, encoding: .utf8) {
+        Logging.network.error("[GoogleGenerativeAI] Response payload: \(responseString)")
+      }
+
+      throw parseError(responseData: data)
+    }
+
+    return try parseResponse(T.Response.self, from: data)
+  }
+
+  @available(macOS 12.0, *)
+  func loadRequestStream<T: GenerativeAIRequest>(request: T)
+    -> AsyncThrowingStream<T.Response, Error> {
+    return AsyncThrowingStream { continuation in
+      Task {
+        let urlRequest: URLRequest
+        do {
+          urlRequest = try self.urlRequest(request: request)
+        } catch {
+          continuation.finish(throwing: error)
+          return
+        }
+
+        #if DEBUG
+          printCURLCommand(from: urlRequest)
+        #endif
+
+        let stream: URLSession.AsyncBytes
+        let rawResponse: URLResponse
+        do {
+          (stream, rawResponse) = try await urlSession.bytes(for: urlRequest)
+        } catch {
+          continuation.finish(throwing: error)
+          return
+        }
+
+        // Verify the status code is 200
+        let response: HTTPURLResponse
+        do {
+          response = try httpResponse(urlResponse: rawResponse)
+        } catch {
+          continuation.finish(throwing: error)
+          return
+        }
+
+        // Verify the status code is 200
+        guard response.statusCode == 200 else {
+          Logging.default
+            .error("[GoogleGenerativeAI] The server responded with an error: \(response)")
+          var responseBody = ""
+          for try await line in stream.lines {
+            responseBody += line + "\n"
+          }
+
+          Logging.network.error("[GoogleGenerativeAI] Response payload: \(responseBody)")
+          continuation.finish(throwing: parseError(responseBody: responseBody))
+
+          return
+        }
+
+        // Received lines that are not server-sent events (SSE); these are not prefixed with "data:"
+        var extraLines: String = ""
+
+        let decoder = JSONDecoder()
+        decoder.keyDecodingStrategy = .convertFromSnakeCase
+        for try await line in stream.lines {
+          Logging.network.debug("[GoogleGenerativeAI] Stream response: \(line)")
+
+          if line.hasPrefix("data:") {
+            // We can assume 5 characters since it's utf-8 encoded, removing `data:`.
+            let jsonText = String(line.dropFirst(5))
+            let data: Data
+            do {
+              data = try jsonData(jsonText: jsonText)
+            } catch {
+              continuation.finish(throwing: error)
+              return
+            }
+
+            // Handle the content.
+            do {
+              let content = try parseResponse(T.Response.self, from: data)
+              continuation.yield(content)
+            } catch {
+              continuation.finish(throwing: error)
+              return
+            }
+          } else {
+            extraLines += line
+          }
+        }
+
+        if extraLines.count > 0 {
+          continuation.finish(throwing: parseError(responseBody: extraLines))
+          return
+        }
+
+        continuation.finish(throwing: nil)
+      }
+    }
+  }
+
+  // MARK: - Private Helpers
+
+  private func urlRequest<T: GenerativeAIRequest>(request: T) throws -> URLRequest {
+    var urlRequest = URLRequest(url: request.url)
+    urlRequest.httpMethod = "POST"
+    urlRequest.setValue(apiKey, forHTTPHeaderField: "x-goog-api-key")
+    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version)",
+                        forHTTPHeaderField: "x-goog-api-client")
+    urlRequest.setValue("application/json", forHTTPHeaderField: "Content-Type")
+    let encoder = JSONEncoder()
+    encoder.keyEncodingStrategy = .convertToSnakeCase
+    urlRequest.httpBody = try encoder.encode(request)
+
+    if let timeoutInterval = request.options.timeout {
+      urlRequest.timeoutInterval = timeoutInterval
+    }
+
+    return urlRequest
+  }
+
+  private func httpResponse(urlResponse: URLResponse) throws -> HTTPURLResponse {
+    // Verify the status code is 200
+    guard let response = urlResponse as? HTTPURLResponse else {
+      Logging.default
+        .error(
+          "[GoogleGenerativeAI] Response wasn't an HTTP response, internal error \(urlResponse)"
+        )
+      throw NSError(
+        domain: "com.google.generative-ai",
+        code: -1,
+        userInfo: [NSLocalizedDescriptionKey: "Response was not an HTTP response."]
+      )
+    }
+
+    return response
+  }
+
+  private func jsonData(jsonText: String) throws -> Data {
+    guard let data = jsonText.data(using: .utf8) else {
+      let error = NSError(
+        domain: "com.google.generative-ai",
+        code: -1,
+        userInfo: [NSLocalizedDescriptionKey: "Could not parse response as UTF8."]
+      )
+      throw error
+    }
+
+    return data
+  }
+
+  private func parseError(responseBody: String) -> Error {
+    do {
+      let data = try jsonData(jsonText: responseBody)
+      return parseError(responseData: data)
+    } catch {
+      return error
+    }
+  }
+
+  private func parseError(responseData: Data) -> Error {
+    do {
+      return try JSONDecoder().decode(RPCError.self, from: responseData)
+    } catch {
+      // TODO: Return an error about an unrecognized error payload with the response body
+      return error
+    }
+  }
+
+  private func parseResponse<T: Decodable>(_ type: T.Type, from data: Data) throws -> T {
+    do {
+      return try JSONDecoder().decode(type, from: data)
+    } catch {
+      if let json = String(data: data, encoding: .utf8) {
+        Logging.network.error("[GoogleGenerativeAI] JSON response: \(json)")
+      }
+      Logging.default.error("[GoogleGenerativeAI] Error decoding server JSON: \(error)")
+      throw error
+    }
+  }
+
+  #if DEBUG
+    private func cURLCommand(from request: URLRequest) -> String {
+      var returnValue = "curl "
+      if let allHeaders = request.allHTTPHeaderFields {
+        for (key, value) in allHeaders {
+          returnValue += "-H '\(key): \(value)' "
+        }
+      }
+
+      guard let url = request.url else { return "" }
+      returnValue += "'\(url.absoluteString)' "
+
+      guard let body = request.httpBody,
+            let jsonStr = String(bytes: body, encoding: .utf8) else { return "" }
+      let escapedJSON = jsonStr.replacingOccurrences(of: "'", with: "'\\''")
+      returnValue += "-d '\(escapedJSON)'"
+
+      return returnValue
+    }
+
+    private func printCURLCommand(from request: URLRequest) {
+      let command = cURLCommand(from: request)
+      Logging.verbose.debug("""
+      [GoogleGenerativeAI] Creating request with the equivalent cURL command:
+      ----- cURL command -----
+      \(command, privacy: .private)
+      ------------------------
+      """)
+    }
+  #endif // DEBUG
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
new file mode 100644
index 00000000000..4146b60d989
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -0,0 +1,26 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+import Foundation
+
+#if !os(macOS) && !os(iOS)
+  #warning("Only iOS, macOS, and Catalyst targets are currently fully supported.")
+#endif
+
+/// Constants associated with the GenerativeAISwift SDK
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum GenerativeAISwift {
+  /// String value of the SDK version
+  public static let version = "0.4.8"
+  static let baseURL = "https://generativelanguage.googleapis.com"
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
new file mode 100644
index 00000000000..3617cd3a251
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -0,0 +1,298 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A type that represents a remote multimodal model (like Gemini), with the ability to generate
+/// content based on various input types.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public final class GenerativeModel {
+  // The prefix for a model resource in the Gemini API.
+  private static let modelResourcePrefix = "models/"
+
+  /// The resource name of the model in the backend; has the format "models/model-name".
+  let modelResourceName: String
+
+  /// The backing service responsible for sending and receiving model requests to the backend.
+  let generativeAIService: GenerativeAIService
+
+  /// Configuration parameters used for the MultiModalModel.
+  let generationConfig: GenerationConfig?
+
+  /// The safety settings to be used for prompts.
+  let safetySettings: [SafetySetting]?
+
+  /// Configuration parameters for sending requests to the backend.
+  let requestOptions: RequestOptions
+
+  /// Initializes a new remote model with the given parameters.
+  ///
+  /// - Parameters:
+  ///   - name: The name of the model to use, e.g., `"gemini-1.0-pro"`; see
+  ///     [Gemini models](https://ai.google.dev/models/gemini) for a list of supported model names.
+  ///   - apiKey: The API key for your project.
+  ///   - generationConfig: The content generation parameters your model should use.
+  ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - requestOptions Configuration parameters for sending requests to the backend.
+  public convenience init(name: String,
+                          apiKey: String,
+                          generationConfig: GenerationConfig? = nil,
+                          safetySettings: [SafetySetting]? = nil,
+                          requestOptions: RequestOptions = RequestOptions()) {
+    self.init(
+      name: name,
+      apiKey: apiKey,
+      generationConfig: generationConfig,
+      safetySettings: safetySettings,
+      requestOptions: requestOptions,
+      urlSession: .shared
+    )
+  }
+
+  /// The designated initializer for this class.
+  init(name: String,
+       apiKey: String,
+       generationConfig: GenerationConfig? = nil,
+       safetySettings: [SafetySetting]? = nil,
+       requestOptions: RequestOptions = RequestOptions(),
+       urlSession: URLSession) {
+    modelResourceName = GenerativeModel.modelResourceName(name: name)
+    generativeAIService = GenerativeAIService(apiKey: apiKey, urlSession: urlSession)
+    self.generationConfig = generationConfig
+    self.safetySettings = safetySettings
+    self.requestOptions = requestOptions
+
+    Logging.default.info("""
+    [GoogleGenerativeAI] Model \(
+      name,
+      privacy: .public
+    ) initialized. To enable additional logging, add \
+    `\(Logging.enableArgumentKey, privacy: .public)` as a launch argument in Xcode.
+    """)
+    Logging.verbose.debug("[GoogleGenerativeAI] Verbose logging enabled.")
+  }
+
+  /// Generates content from String and/or image inputs, given to the model as a prompt, that are
+  /// representable as one or more ``ModelContent/Part``s.
+  ///
+  /// Since ``ModelContent/Part``s do not specify a role, this method is intended for generating
+  /// content from
+  /// [zero-shot](https://developers.google.com/machine-learning/glossary/generative#zero-shot-prompting)
+  /// or "direct" prompts. For
+  /// [few-shot](https://developers.google.com/machine-learning/glossary/generative#few-shot-prompting)
+  /// prompts, see ``generateContent(_:)-58rm0``.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt (see
+  /// ``ThrowingPartsRepresentable``
+  /// for conforming types).
+  /// - Returns: The content generated by the model.
+  /// - Throws: A ``GenerateContentError`` if the request failed.
+  public func generateContent(_ parts: any ThrowingPartsRepresentable...)
+    async throws -> GenerateContentResponse {
+    return try await generateContent([ModelContent(parts: parts)])
+  }
+
+  /// Generates new content from input content given to the model as a prompt.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt.
+  /// - Returns: The generated content response from the model.
+  /// - Throws: A ``GenerateContentError`` if the request failed.
+  public func generateContent(_ content: @autoclosure () throws -> [ModelContent]) async throws
+    -> GenerateContentResponse {
+    let response: GenerateContentResponse
+    do {
+      let generateContentRequest = try GenerateContentRequest(model: modelResourceName,
+                                                              contents: content(),
+                                                              generationConfig: generationConfig,
+                                                              safetySettings: safetySettings,
+                                                              isStreaming: false,
+                                                              options: requestOptions)
+      response = try await generativeAIService.loadRequest(request: generateContentRequest)
+    } catch {
+      if let imageError = error as? ImageConversionError {
+        throw GenerateContentError.promptImageContentError(underlying: imageError)
+      }
+      throw GenerativeModel.generateContentError(from: error)
+    }
+
+    // Check the prompt feedback to see if the prompt was blocked.
+    if response.promptFeedback?.blockReason != nil {
+      throw GenerateContentError.promptBlocked(response: response)
+    }
+
+    // Check to see if an error should be thrown for stop reason.
+    if let reason = response.candidates.first?.finishReason, reason != .stop {
+      throw GenerateContentError.responseStoppedEarly(reason: reason, response: response)
+    }
+
+    return response
+  }
+
+  /// Generates content from String and/or image inputs, given to the model as a prompt, that are
+  /// representable as one or more ``ModelContent/Part``s.
+  ///
+  /// Since ``ModelContent/Part``s do not specify a role, this method is intended for generating
+  /// content from
+  /// [zero-shot](https://developers.google.com/machine-learning/glossary/generative#zero-shot-prompting)
+  /// or "direct" prompts. For
+  /// [few-shot](https://developers.google.com/machine-learning/glossary/generative#few-shot-prompting)
+  /// prompts, see ``generateContent(_:)-58rm0``.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt (see
+  /// ``ThrowingPartsRepresentable``
+  /// for conforming types).
+  /// - Returns: A stream wrapping content generated by the model or a ``GenerateContentError``
+  ///     error if an error occurred.
+  @available(macOS 12.0, *)
+  public func generateContentStream(_ parts: any ThrowingPartsRepresentable...)
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    return try generateContentStream([ModelContent(parts: parts)])
+  }
+
+  /// Generates new content from input content given to the model as a prompt.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt.
+  /// - Returns: A stream wrapping content generated by the model or a ``GenerateContentError``
+  ///     error if an error occurred.
+  @available(macOS 12.0, *)
+  public func generateContentStream(_ content: @autoclosure () throws -> [ModelContent])
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    let evaluatedContent: [ModelContent]
+    do {
+      evaluatedContent = try content()
+    } catch let underlying {
+      return AsyncThrowingStream { continuation in
+        let error: Error
+        if let contentError = underlying as? ImageConversionError {
+          error = GenerateContentError.promptImageContentError(underlying: contentError)
+        } else {
+          error = GenerateContentError.internalError(underlying: underlying)
+        }
+        continuation.finish(throwing: error)
+      }
+    }
+
+    let generateContentRequest = GenerateContentRequest(model: modelResourceName,
+                                                        contents: evaluatedContent,
+                                                        generationConfig: generationConfig,
+                                                        safetySettings: safetySettings,
+                                                        isStreaming: true,
+                                                        options: requestOptions)
+
+    var responseIterator = generativeAIService.loadRequestStream(request: generateContentRequest)
+      .makeAsyncIterator()
+    return AsyncThrowingStream {
+      let response: GenerateContentResponse?
+      do {
+        response = try await responseIterator.next()
+      } catch {
+        throw GenerativeModel.generateContentError(from: error)
+      }
+
+      // The responseIterator will return `nil` when it's done.
+      guard let response = response else {
+        // This is the end of the stream! Signal it by sending `nil`.
+        return nil
+      }
+
+      // Check the prompt feedback to see if the prompt was blocked.
+      if response.promptFeedback?.blockReason != nil {
+        throw GenerateContentError.promptBlocked(response: response)
+      }
+
+      // If the stream ended early unexpectedly, throw an error.
+      if let finishReason = response.candidates.first?.finishReason, finishReason != .stop {
+        throw GenerateContentError.responseStoppedEarly(reason: finishReason, response: response)
+      } else {
+        // Response was valid content, pass it along and continue.
+        return response
+      }
+    }
+  }
+
+  /// Creates a new chat conversation using this model with the provided history.
+  public func startChat(history: [ModelContent] = []) -> Chat {
+    return Chat(model: self, history: history)
+  }
+
+  /// Runs the model's tokenizer on String and/or image inputs that are representable as one or more
+  /// ``ModelContent/Part``s.
+  ///
+  /// Since ``ModelContent/Part``s do not specify a role, this method is intended for tokenizing
+  /// [zero-shot](https://developers.google.com/machine-learning/glossary/generative#zero-shot-prompting)
+  /// or "direct" prompts. For
+  /// [few-shot](https://developers.google.com/machine-learning/glossary/generative#few-shot-prompting)
+  /// input, see ``countTokens(_:)-9spwl``.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt (see
+  /// ``ThrowingPartsRepresentable``
+  /// for conforming types).
+  /// - Returns: The results of running the model's tokenizer on the input; contains
+  /// ``CountTokensResponse/totalTokens``.
+  /// - Throws: A ``CountTokensError`` if the tokenization request failed.
+  public func countTokens(_ parts: any ThrowingPartsRepresentable...) async throws
+    -> CountTokensResponse {
+    return try await countTokens([ModelContent(parts: parts)])
+  }
+
+  /// Runs the model's tokenizer on the input content and returns the token count.
+  ///
+  /// - Parameter content: The input given to the model as a prompt.
+  /// - Returns: The results of running the model's tokenizer on the input; contains
+  /// ``CountTokensResponse/totalTokens``.
+  /// - Throws: A ``CountTokensError`` if the tokenization request failed or the input content was
+  /// invalid.
+  public func countTokens(_ content: @autoclosure () throws -> [ModelContent]) async throws
+    -> CountTokensResponse {
+    do {
+      let countTokensRequest = try CountTokensRequest(
+        model: modelResourceName,
+        contents: content(),
+        options: requestOptions
+      )
+      return try await generativeAIService.loadRequest(request: countTokensRequest)
+    } catch {
+      throw CountTokensError.internalError(underlying: error)
+    }
+  }
+
+  /// Returns a model resource name of the form "models/model-name" based on `name`.
+  private static func modelResourceName(name: String) -> String {
+    if name.contains("/") {
+      return name
+    } else {
+      return modelResourcePrefix + name
+    }
+  }
+
+  /// Returns a `GenerateContentError` (for public consumption) from an internal error.
+  ///
+  /// If `error` is already a `GenerateContentError` the error is returned unchanged.
+  private static func generateContentError(from error: Error) -> GenerateContentError {
+    if let error = error as? GenerateContentError {
+      return error
+    } else if let error = error as? RPCError, error.isInvalidAPIKeyError() {
+      return GenerateContentError.invalidAPIKey(message: error.message)
+    } else if let error = error as? RPCError, error.isUnsupportedUserLocationError() {
+      return GenerateContentError.unsupportedUserLocation
+    }
+    return GenerateContentError.internalError(underlying: error)
+  }
+}
+
+/// See ``GenerativeModel/countTokens(_:)-9spwl``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum CountTokensError: Error {
+  case internalError(underlying: Error)
+}
diff --git a/FirebaseVertexAI/Sources/Logging.swift b/FirebaseVertexAI/Sources/Logging.swift
new file mode 100644
index 00000000000..458c34ed18d
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Logging.swift
@@ -0,0 +1,56 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+import OSLog
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct Logging {
+  /// Subsystem that should be used for all Loggers.
+  static let subsystem = "com.google.generative-ai"
+
+  /// Default category used for most loggers, unless specialized.
+  static let defaultCategory = ""
+
+  /// The argument required to enable additional logging.
+  static let enableArgumentKey = "-GoogleGenerativeAIDebugLogEnabled"
+
+  // No initializer available.
+  @available(*, unavailable)
+  private init() {}
+
+  /// The default logger that is visible for all users. Note: we shouldn't be using anything lower
+  /// than `.notice`.
+  static var `default` = Logger(subsystem: subsystem, category: defaultCategory)
+
+  /// A non default
+  static var network: Logger = {
+    if ProcessInfo.processInfo.arguments.contains(enableArgumentKey) {
+      return Logger(subsystem: subsystem, category: "NetworkResponse")
+    } else {
+      // Return a valid logger that's using `OSLog.disabled` as the logger, hiding everything.
+      return Logger(.disabled)
+    }
+  }()
+
+  ///
+  static var verbose: Logger = {
+    if ProcessInfo.processInfo.arguments.contains(enableArgumentKey) {
+      return Logger(subsystem: subsystem, category: defaultCategory)
+    } else {
+      // Return a valid logger that's using `OSLog.disabled` as the logger, hiding everything.
+      return Logger(.disabled)
+    }
+  }()
+}
diff --git a/FirebaseVertexAI/Sources/ModelContent.swift b/FirebaseVertexAI/Sources/ModelContent.swift
new file mode 100644
index 00000000000..44648c57852
--- /dev/null
+++ b/FirebaseVertexAI/Sources/ModelContent.swift
@@ -0,0 +1,141 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A type describing data in media formats interpretable by an AI model. Each generative AI
+/// request or response contains an `Array` of ``ModelContent``s, and each ``ModelContent`` value
+/// may comprise multiple heterogeneous ``ModelContent/Part``s.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct ModelContent: Codable, Equatable {
+  /// A discrete piece of data in a media format intepretable by an AI model. Within a single value
+  /// of ``Part``, different data types may not mix.
+  public enum Part: Codable, Equatable {
+    enum CodingKeys: String, CodingKey {
+      case text
+      case inlineData
+    }
+
+    enum InlineDataKeys: String, CodingKey {
+      case mimeType = "mime_type"
+      case bytes = "data"
+    }
+
+    /// Text value.
+    case text(String)
+
+    /// Data with a specified media type. Not all media types may be supported by the AI model.
+    case data(mimetype: String, Data)
+
+    // MARK: Convenience Initializers
+
+    /// Convenience function for populating a Part with JPEG data.
+    public static func jpeg(_ data: Data) -> Self {
+      return .data(mimetype: "image/jpeg", data)
+    }
+
+    /// Convenience function for populating a Part with PNG data.
+    public static func png(_ data: Data) -> Self {
+      return .data(mimetype: "image/png", data)
+    }
+
+    // MARK: Codable Conformance
+
+    public func encode(to encoder: Encoder) throws {
+      var container = encoder.container(keyedBy: ModelContent.Part.CodingKeys.self)
+      switch self {
+      case let .text(a0):
+        try container.encode(a0, forKey: .text)
+      case let .data(mimetype, bytes):
+        var inlineDataContainer = container.nestedContainer(
+          keyedBy: InlineDataKeys.self,
+          forKey: .inlineData
+        )
+        try inlineDataContainer.encode(mimetype, forKey: .mimeType)
+        try inlineDataContainer.encode(bytes, forKey: .bytes)
+      }
+    }
+
+    public init(from decoder: Decoder) throws {
+      let values = try decoder.container(keyedBy: CodingKeys.self)
+      if values.contains(.text) {
+        self = try .text(values.decode(String.self, forKey: .text))
+      } else if values.contains(.inlineData) {
+        let dataContainer = try values.nestedContainer(
+          keyedBy: InlineDataKeys.self,
+          forKey: .inlineData
+        )
+        let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
+        let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
+        self = .data(mimetype: mimetype, bytes)
+      } else {
+        throw DecodingError.dataCorrupted(.init(
+          codingPath: [CodingKeys.text, CodingKeys.inlineData],
+          debugDescription: "Neither text or inline data was found."
+        ))
+      }
+    }
+
+    /// Returns the text contents of this ``Part``, if it contains text.
+    public var text: String? {
+      switch self {
+      case let .text(contents): return contents
+      default: return nil
+      }
+    }
+  }
+
+  /// The role of the entity creating the ``ModelContent``. For user-generated client requests,
+  /// for example, the role is `user`.
+  public let role: String?
+
+  /// The data parts comprising this ``ModelContent`` value.
+  public let parts: [Part]
+
+  /// Creates a new value from any data or `Array` of data interpretable as a
+  /// ``Part``. See ``ThrowingPartsRepresentable`` for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", parts: some ThrowingPartsRepresentable) throws {
+    self.role = role
+    try self.parts = parts.tryPartsValue()
+  }
+
+  /// Creates a new value from any data or `Array` of data interpretable as a
+  /// ``Part``. See ``ThrowingPartsRepresentable`` for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", parts: some PartsRepresentable) {
+    self.role = role
+    self.parts = parts.partsValue
+  }
+
+  /// Creates a new value from a list of ``Part``s.
+  public init(role: String? = "user", parts: [Part]) {
+    self.role = role
+    self.parts = parts
+  }
+
+  /// Creates a new value from any data interpretable as a ``Part``. See
+  /// ``ThrowingPartsRepresentable``
+  /// for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", _ parts: any ThrowingPartsRepresentable...) throws {
+    let content = try parts.flatMap { try $0.tryPartsValue() }
+    self.init(role: role, parts: content)
+  }
+
+  /// Creates a new value from any data interpretable as a ``Part``. See
+  /// ``ThrowingPartsRepresentable``
+  /// for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", _ parts: [PartsRepresentable]) {
+    let content = parts.flatMap { $0.partsValue }
+    self.init(role: role, parts: content)
+  }
+}
diff --git a/FirebaseVertexAI/Sources/PartsRepresentable+Image.swift b/FirebaseVertexAI/Sources/PartsRepresentable+Image.swift
new file mode 100644
index 00000000000..052a004e52f
--- /dev/null
+++ b/FirebaseVertexAI/Sources/PartsRepresentable+Image.swift
@@ -0,0 +1,107 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import UniformTypeIdentifiers
+#if canImport(UIKit)
+  import UIKit // For UIImage extensions.
+#elseif canImport(AppKit)
+  import AppKit // For NSImage extensions.
+#endif
+
+private let imageCompressionQuality: CGFloat = 0.8
+
+/// An enum describing failures that can occur when converting image types to model content data.
+/// For some image types like `CIImage`, creating valid model content requires creating a JPEG
+/// representation of the image that may not yet exist, which may be computationally expensive.
+public enum ImageConversionError: Error {
+  /// The image (the receiver of the call `toModelContentParts()`) was invalid.
+  case invalidUnderlyingImage
+
+  /// A valid image destination could not be allocated.
+  case couldNotAllocateDestination
+
+  /// JPEG image data conversion failed, accompanied by the original image, which may be an
+  /// instance of `NSImageRep`, `UIImage`, `CGImage`, or `CIImage`.
+  case couldNotConvertToJPEG(Any)
+}
+
+#if canImport(UIKit)
+  /// Enables images to be representable as ``ThrowingPartsRepresentable``.
+  @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+  extension UIImage: ThrowingPartsRepresentable {
+    public func tryPartsValue() throws -> [ModelContent.Part] {
+      guard let data = jpegData(compressionQuality: imageCompressionQuality) else {
+        throw ImageConversionError.couldNotConvertToJPEG(self)
+      }
+      return [ModelContent.Part.data(mimetype: "image/jpeg", data)]
+    }
+  }
+
+#elseif canImport(AppKit)
+  /// Enables images to be representable as ``ThrowingPartsRepresentable``.
+  @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+  extension NSImage: ThrowingPartsRepresentable {
+    public func tryPartsValue() throws -> [ModelContent.Part] {
+      guard let cgImage = cgImage(forProposedRect: nil, context: nil, hints: nil) else {
+        throw ImageConversionError.invalidUnderlyingImage
+      }
+      let bmp = NSBitmapImageRep(cgImage: cgImage)
+      guard let data = bmp.representation(using: .jpeg, properties: [.compressionFactor: 0.8])
+      else {
+        throw ImageConversionError.couldNotConvertToJPEG(bmp)
+      }
+      return [ModelContent.Part.data(mimetype: "image/jpeg", data)]
+    }
+  }
+#endif
+
+/// Enables `CGImages` to be representable as model content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CGImage: ThrowingPartsRepresentable {
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    let output = NSMutableData()
+    guard let imageDestination = CGImageDestinationCreateWithData(
+      output, UTType.jpeg.identifier as CFString, 1, nil
+    ) else {
+      throw ImageConversionError.couldNotAllocateDestination
+    }
+    CGImageDestinationAddImage(imageDestination, self, nil)
+    CGImageDestinationSetProperties(imageDestination, [
+      kCGImageDestinationLossyCompressionQuality: imageCompressionQuality,
+    ] as CFDictionary)
+    if CGImageDestinationFinalize(imageDestination) {
+      return [.data(mimetype: "image/jpeg", output as Data)]
+    }
+    throw ImageConversionError.couldNotConvertToJPEG(self)
+  }
+}
+
+/// Enables `CIImages` to be representable as model content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CIImage: ThrowingPartsRepresentable {
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    let context = CIContext()
+    let jpegData = (colorSpace ?? CGColorSpace(name: CGColorSpace.sRGB))
+      .flatMap {
+        // The docs specify kCGImageDestinationLossyCompressionQuality as a supported option, but
+        // Swift's type system does not allow this.
+        // [kCGImageDestinationLossyCompressionQuality: imageCompressionQuality]
+        context.jpegRepresentation(of: self, colorSpace: $0, options: [:])
+      }
+    if let jpegData = jpegData {
+      return [.data(mimetype: "image/jpeg", jpegData)]
+    }
+    throw ImageConversionError.couldNotConvertToJPEG(self)
+  }
+}
diff --git a/FirebaseVertexAI/Sources/PartsRepresentable.swift b/FirebaseVertexAI/Sources/PartsRepresentable.swift
new file mode 100644
index 00000000000..05ba0d9dabc
--- /dev/null
+++ b/FirebaseVertexAI/Sources/PartsRepresentable.swift
@@ -0,0 +1,66 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A protocol describing any data that could be serialized to model-interpretable input data,
+/// where the serialization process might fail with an error.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public protocol ThrowingPartsRepresentable {
+  func tryPartsValue() throws -> [ModelContent.Part]
+}
+
+/// A protocol describing any data that could be serialized to model-interpretable input data,
+/// where the serialization process cannot fail with an error. For a failable conversion, see
+/// ``ThrowingPartsRepresentable``
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public protocol PartsRepresentable: ThrowingPartsRepresentable {
+  var partsValue: [ModelContent.Part] { get }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public extension PartsRepresentable {
+  func tryPartsValue() throws -> [ModelContent.Part] {
+    return partsValue
+  }
+}
+
+/// Enables a ``ModelContent.Part`` to be passed in as ``ThrowingPartsRepresentable``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension ModelContent.Part: ThrowingPartsRepresentable {
+  public typealias ErrorType = Never
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    return [self]
+  }
+}
+
+/// Enable an `Array` of ``ThrowingPartsRepresentable`` values to be passed in as a single
+/// ``ThrowingPartsRepresentable``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension [ThrowingPartsRepresentable]: ThrowingPartsRepresentable {
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    return try compactMap { element in
+      try element.tryPartsValue()
+    }
+    .flatMap { $0 }
+  }
+}
+
+/// Enables a `String` to be passed in as ``ThrowingPartsRepresentable``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension String: PartsRepresentable {
+  public var partsValue: [ModelContent.Part] {
+    return [.text(self)]
+  }
+}
diff --git a/FirebaseVertexAI/Sources/Safety.swift b/FirebaseVertexAI/Sources/Safety.swift
new file mode 100644
index 00000000000..d59511f2476
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Safety.swift
@@ -0,0 +1,182 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A type defining potentially harmful media categories and their model-assigned ratings. A value
+/// of this type may be assigned to a category for every model-generated response, not just
+/// responses that exceed a certain threshold.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct SafetyRating: Decodable, Equatable, Hashable {
+  /// The category describing the potential harm a piece of content may pose. See
+  /// ``SafetySetting/HarmCategory`` for a list of possible values.
+  public let category: SafetySetting.HarmCategory
+
+  /// The model-generated probability that a given piece of content falls under the harm category
+  /// described in ``category``. This does not
+  /// indiciate the severity of harm for a piece of content. See ``HarmProbability`` for a list of
+  /// possible values.
+  public let probability: HarmProbability
+
+  /// Initializes a new `SafetyRating` instance with the given category and probability.
+  /// Use this initializer for SwiftUI previews or tests.
+  public init(category: SafetySetting.HarmCategory, probability: HarmProbability) {
+    self.category = category
+    self.probability = probability
+  }
+
+  /// The probability that a given model output falls under a harmful content category. This does
+  /// not indicate the severity of harm for a piece of content.
+  public enum HarmProbability: String, Codable {
+    /// Unknown. A new server value that isn't recognized by the SDK.
+    case unknown = "UNKNOWN"
+
+    /// The probability was not specified in the server response.
+    case unspecified = "HARM_PROBABILITY_UNSPECIFIED"
+
+    /// The probability is zero or close to zero. For benign content, the probability across all
+    /// categories will be this value.
+    case negligible = "NEGLIGIBLE"
+
+    /// The probability is small but non-zero.
+    case low = "LOW"
+
+    /// The probability is moderate.
+    case medium = "MEDIUM"
+
+    /// The probability is high. The content described is very likely harmful.
+    case high = "HIGH"
+
+    /// Initializes a new `SafetyRating` from a decoder.
+    /// Not for external use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedProbability = HarmProbability(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized HarmProbability with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedProbability
+    }
+  }
+}
+
+/// Safety feedback for an entire request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct SafetyFeedback: Decodable {
+  /// Safety rating evaluated from content.
+  public let rating: SafetyRating
+
+  /// Safety settings applied to the request.
+  public let setting: SafetySetting
+
+  /// Internal initializer.
+  init(rating: SafetyRating, setting: SafetySetting) {
+    self.rating = rating
+    self.setting = setting
+  }
+}
+
+/// A type used to specify a threshold for harmful content, beyond which the model will return a
+/// fallback response instead of generated content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct SafetySetting: Codable {
+  /// A type describing safety attributes, which include harmful categories and topics that can
+  /// be considered sensitive.
+  public enum HarmCategory: String, Codable {
+    /// Unknown. A new server value that isn't recognized by the SDK.
+    case unknown = "HARM_CATEGORY_UNKNOWN"
+
+    /// Unspecified by the server.
+    case unspecified = "HARM_CATEGORY_UNSPECIFIED"
+
+    /// Harassment content.
+    case harassment = "HARM_CATEGORY_HARASSMENT"
+
+    /// Negative or harmful comments targeting identity and/or protected attributes.
+    case hateSpeech = "HARM_CATEGORY_HATE_SPEECH"
+
+    /// Contains references to sexual acts or other lewd content.
+    case sexuallyExplicit = "HARM_CATEGORY_SEXUALLY_EXPLICIT"
+
+    /// Promotes or enables access to harmful goods, services, or activities.
+    case dangerousContent = "HARM_CATEGORY_DANGEROUS_CONTENT"
+
+    /// Do not explicitly use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedCategory = HarmCategory(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized HarmCategory with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedCategory
+    }
+  }
+
+  /// Block at and beyond a specified ``SafetyRating/HarmProbability``.
+  public enum BlockThreshold: String, Codable {
+    /// Unknown. A new server value that isn't recognized by the SDK.
+    case unknown = "UNKNOWN"
+
+    /// Threshold is unspecified.
+    case unspecified = "HARM_BLOCK_THRESHOLD_UNSPECIFIED"
+
+    // Content with `.negligible` will be allowed.
+    case blockLowAndAbove = "BLOCK_LOW_AND_ABOVE"
+
+    /// Content with `.negligible` and `.low` will be allowed.
+    case blockMediumAndAbove = "BLOCK_MEDIUM_AND_ABOVE"
+
+    /// Content with `.negligible`, `.low`, and `.medium` will be allowed.
+    case blockOnlyHigh = "BLOCK_ONLY_HIGH"
+
+    /// All content will be allowed.
+    case blockNone = "BLOCK_NONE"
+
+    /// Do not explicitly use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedThreshold = BlockThreshold(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized BlockThreshold with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedThreshold
+    }
+  }
+
+  enum CodingKeys: String, CodingKey {
+    case harmCategory = "category"
+    case threshold
+  }
+
+  /// The category this safety setting should be applied to.
+  public let harmCategory: HarmCategory
+
+  /// The threshold describing what content should be blocked.
+  public let threshold: BlockThreshold
+
+  /// Initializes a new safety setting with the given category and threshold.
+  public init(harmCategory: HarmCategory, threshold: BlockThreshold) {
+    self.harmCategory = harmCategory
+    self.threshold = threshold
+  }
+}
diff --git a/Package.swift b/Package.swift
index 9abcecd7edd..55ca4f8b833 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1366,7 +1366,11 @@ let package = Package(
         "FirebaseCoreExtension",
         .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources"
+      path: "FirebaseVertexAI/Sources",
+      sources: [
+        "VertexAI.swift",
+        "VertexAIComponent.swift",
+      ]
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From 575a69b5d5c9986c96315b4752420706355e9dab Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 18 Mar 2024 21:16:22 +0000
Subject: [PATCH 013/124] Use forked `google-generative-ai` in Vertex AI
 samples (#12574)

---
 .../ChatSample/Views/ErrorDetailsView.swift   | 24 -----------
 FirebaseVertexAI/Sources/Errors.swift         | 12 ------
 .../Sources/GenerateContentError.swift        | 13 ------
 .../Sources/GenerativeAIRequest.swift         |  4 +-
 .../Sources/GenerativeAIService.swift         | 28 ++++++++++---
 .../Sources/GenerativeAISwift.swift           |  5 +--
 .../Sources/GenerativeModel.swift             | 35 +++++-----------
 FirebaseVertexAI/Sources/VertexAI.swift       | 41 +++++--------------
 Package.swift                                 | 11 +----
 9 files changed, 48 insertions(+), 125 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
index 399777ad1e8..7b3b78753db 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -142,22 +142,6 @@ struct ErrorDetailsView: View {
             SafetyRatingsSection(ratings: ratings)
           }
 
-        case GenerateContentError.invalidAPIKey:
-          Section("Error Type") {
-            Text("Invalid API Key")
-          }
-
-          Section("Details") {
-            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
-            SubtitleMarkdownFormRow(
-              title: "Help",
-              value: """
-              The `API_KEY` provided in the `GoogleService-Info.plist` file is invalid. Download a
-              new copy of the file from the [Firebase Console](https://console.firebase.google.com).
-              """
-            )
-          }
-
         default:
           Section("Error Type") {
             Text("Some other error")
@@ -222,11 +206,3 @@ struct ErrorDetailsView: View {
 
   return ErrorDetailsView(error: error)
 }
-
-#Preview("Invalid API Key") {
-  ErrorDetailsView(error: GenerateContentError.invalidAPIKey)
-}
-
-#Preview("Unsupported User Location") {
-  ErrorDetailsView(error: GenerateContentError.unsupportedUserLocation)
-}
diff --git a/FirebaseVertexAI/Sources/Errors.swift b/FirebaseVertexAI/Sources/Errors.swift
index 35e7e6b813b..0b1b8848411 100644
--- a/FirebaseVertexAI/Sources/Errors.swift
+++ b/FirebaseVertexAI/Sources/Errors.swift
@@ -30,14 +30,6 @@ struct RPCError: Error {
     self.status = status
     self.details = details
   }
-
-  func isInvalidAPIKeyError() -> Bool {
-    return errorInfo?.reason == "API_KEY_INVALID"
-  }
-
-  func isUnsupportedUserLocationError() -> Bool {
-    return message == RPCErrorMessage.unsupportedUserLocation.rawValue
-  }
 }
 
 extension RPCError: Decodable {
@@ -179,10 +171,6 @@ enum RPCStatus: String, Decodable {
   case dataLoss = "DATA_LOSS"
 }
 
-enum RPCErrorMessage: String {
-  case unsupportedUserLocation = "User location is not supported for the API use."
-}
-
 enum InvalidCandidateError: Error {
   case emptyContent(underlyingError: Error)
   case malformedContent(underlyingError: Error)
diff --git a/FirebaseVertexAI/Sources/GenerateContentError.swift b/FirebaseVertexAI/Sources/GenerateContentError.swift
index 055928e5f1b..31501a7744c 100644
--- a/FirebaseVertexAI/Sources/GenerateContentError.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentError.swift
@@ -28,17 +28,4 @@ public enum GenerateContentError: Error {
 
   /// A response didn't fully complete. See the `FinishReason` for more information.
   case responseStoppedEarly(reason: FinishReason, response: GenerateContentResponse)
-
-  /// The provided API key is invalid.
-  case invalidAPIKey(message: String)
-
-  /// The user's location (region) is not supported by the API.
-  ///
-  /// See the Google documentation for a
-  /// [list of regions](https://ai.google.dev/available_regions#available_regions)
-  /// (countries and territories) where the API is available.
-  ///
-  /// - Important: The API is only available in
-  /// [specific regions](https://ai.google.dev/available_regions#available_regions).
-  case unsupportedUserLocation
 }
diff --git a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
index 2bf94503b1a..21f35b1b728 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
@@ -38,8 +38,8 @@ public struct RequestOptions {
   /// - Parameters:
   ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
   ///   for a `URLRequest`.
-  ///   - apiVersion The API version to use in requests to the backend; defaults to "v1".
-  public init(timeout: TimeInterval? = nil, apiVersion: String = "v1") {
+  ///   - apiVersion The API version to use in requests to the backend; defaults to "v2beta".
+  public init(timeout: TimeInterval? = nil, apiVersion: String = "v2beta") {
     self.timeout = timeout
     self.apiVersion = apiVersion
   }
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index 8d904738927..e1df1010c06 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -12,6 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
+import FirebaseCore
 import Foundation
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -19,15 +21,18 @@ struct GenerativeAIService {
   /// Gives permission to talk to the backend.
   private let apiKey: String
 
+  private let appCheck: AppCheckInterop?
+
   private let urlSession: URLSession
 
-  init(apiKey: String, urlSession: URLSession) {
+  init(apiKey: String, appCheck: AppCheckInterop?, urlSession: URLSession) {
     self.apiKey = apiKey
+    self.appCheck = appCheck
     self.urlSession = urlSession
   }
 
   func loadRequest<T: GenerativeAIRequest>(request: T) async throws -> T.Response {
-    let urlRequest = try urlRequest(request: request)
+    let urlRequest = try await urlRequest(request: request)
 
     #if DEBUG
       printCURLCommand(from: urlRequest)
@@ -59,7 +64,7 @@ struct GenerativeAIService {
       Task {
         let urlRequest: URLRequest
         do {
-          urlRequest = try self.urlRequest(request: request)
+          urlRequest = try await self.urlRequest(request: request)
         } catch {
           continuation.finish(throwing: error)
           return
@@ -146,13 +151,24 @@ struct GenerativeAIService {
 
   // MARK: - Private Helpers
 
-  private func urlRequest<T: GenerativeAIRequest>(request: T) throws -> URLRequest {
+  private func urlRequest<T: GenerativeAIRequest>(request: T) async throws -> URLRequest {
     var urlRequest = URLRequest(url: request.url)
     urlRequest.httpMethod = "POST"
     urlRequest.setValue(apiKey, forHTTPHeaderField: "x-goog-api-key")
-    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version)",
-                        forHTTPHeaderField: "x-goog-api-client")
+    // TODO: Determine the right client header to use.
+    //    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version))",
+    //                        forHTTPHeaderField: "x-goog-api-client")
     urlRequest.setValue("application/json", forHTTPHeaderField: "Content-Type")
+
+    if let appCheck {
+      let tokenResult = await appCheck.getToken(forcingRefresh: false)
+      urlRequest.setValue(tokenResult.token, forHTTPHeaderField: "X-Firebase-AppCheck")
+      if let error = tokenResult.error {
+        Logging.default
+          .debug("[GoogleGenerativeAI] Failed to fetch AppCheck token. Error: \(error)")
+      }
+    }
+
     let encoder = JSONEncoder()
     encoder.keyEncodingStrategy = .convertToSnakeCase
     urlRequest.httpBody = try encoder.encode(request)
diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
index 4146b60d989..0f0af2bc3e2 100644
--- a/FirebaseVertexAI/Sources/GenerativeAISwift.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -11,6 +11,7 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
+
 import Foundation
 
 #if !os(macOS) && !os(iOS)
@@ -20,7 +21,5 @@ import Foundation
 /// Constants associated with the GenerativeAISwift SDK
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public enum GenerativeAISwift {
-  /// String value of the SDK version
-  public static let version = "0.4.8"
-  static let baseURL = "https://generativelanguage.googleapis.com"
+  static let baseURL = "https://staging-firebaseml.sandbox.googleapis.com"
 }
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index 3617cd3a251..aad2b3e99c0 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -12,6 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
 import Foundation
 
 /// A type that represents a remote multimodal model (like Gemini), with the ability to generate
@@ -44,31 +45,21 @@ public final class GenerativeModel {
   ///   - apiKey: The API key for your project.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
-  ///   - requestOptions Configuration parameters for sending requests to the backend.
-  public convenience init(name: String,
-                          apiKey: String,
-                          generationConfig: GenerationConfig? = nil,
-                          safetySettings: [SafetySetting]? = nil,
-                          requestOptions: RequestOptions = RequestOptions()) {
-    self.init(
-      name: name,
-      apiKey: apiKey,
-      generationConfig: generationConfig,
-      safetySettings: safetySettings,
-      requestOptions: requestOptions,
-      urlSession: .shared
-    )
-  }
-
-  /// The designated initializer for this class.
+  ///   - requestOptions: Configuration parameters for sending requests to the backend.
+  ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
        apiKey: String,
        generationConfig: GenerationConfig? = nil,
        safetySettings: [SafetySetting]? = nil,
-       requestOptions: RequestOptions = RequestOptions(),
-       urlSession: URLSession) {
+       requestOptions: RequestOptions,
+       appCheck: AppCheckInterop?,
+       urlSession: URLSession = .shared) {
     modelResourceName = GenerativeModel.modelResourceName(name: name)
-    generativeAIService = GenerativeAIService(apiKey: apiKey, urlSession: urlSession)
+    generativeAIService = GenerativeAIService(
+      apiKey: apiKey,
+      appCheck: appCheck,
+      urlSession: urlSession
+    )
     self.generationConfig = generationConfig
     self.safetySettings = safetySettings
     self.requestOptions = requestOptions
@@ -282,10 +273,6 @@ public final class GenerativeModel {
   private static func generateContentError(from error: Error) -> GenerateContentError {
     if let error = error as? GenerateContentError {
       return error
-    } else if let error = error as? RPCError, error.isInvalidAPIKeyError() {
-      return GenerateContentError.invalidAPIKey(message: error.message)
-    } else if let error = error as? RPCError, error.isUnsupportedUserLocationError() {
-      return GenerateContentError.unsupportedUserLocation
     }
     return GenerateContentError.internalError(underlying: error)
   }
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 8fb0f9e8c68..0f79e1be99c 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -12,13 +12,9 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import Foundation
-
 import FirebaseAppCheckInterop
 import FirebaseCore
-
-// Exports the GoogleGenerativeAI module to users of the SDK.
-@_exported import GoogleGenerativeAI
+import Foundation
 
 // Avoids exposing internal FirebaseCore APIs to Swift users.
 @_implementationOnly import FirebaseCoreExtension
@@ -31,8 +27,9 @@ open class VertexAI: NSObject {
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   ///
   /// This instance is configured with the default `FirebaseApp`.
-  public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
-    .GenerativeModel {
+  ///
+  /// TODO: Add RequestOptions to public API.
+  public static func generativeModel(modelName: String, location: String) -> GenerativeModel {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
@@ -40,8 +37,10 @@ open class VertexAI: NSObject {
   }
 
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  ///
+  /// TODO: Add RequestOptions to public API.
   public static func generativeModel(app: FirebaseApp, modelName: String,
-                                     location: String) -> GoogleGenerativeAI.GenerativeModel {
+                                     location: String) -> GenerativeModel {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
@@ -64,18 +63,15 @@ open class VertexAI: NSObject {
   private let modelResouceName: String
 
   lazy var model: GenerativeModel = {
-    let options = RequestOptions(
-      apiVersion: "v2beta",
-      endpoint: "staging-firebaseml.sandbox.googleapis.com",
-      hooks: [addAppCheckHeader]
-    )
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
     }
     return GenerativeModel(
       name: modelResouceName,
       apiKey: apiKey,
-      requestOptions: options
+      // TODO: Add RequestOptions to public API.
+      requestOptions: RequestOptions(),
+      appCheck: appCheck
     )
   }()
 
@@ -104,21 +100,4 @@ open class VertexAI: NSObject {
 
     return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
   }
-
-  // MARK: Request Hooks
-
-  /// Adds an App Check token to the provided request if App Check is included in the app.
-  ///
-  /// This demonstrates how an App Check token can be added to requests; it is currently ignored by
-  /// the backend.
-  ///
-  /// - Parameter request: The `URLRequest` to modify by adding an App Check token header.
-  func addAppCheckHeader(request: inout URLRequest) async {
-    guard let appCheck else {
-      return
-    }
-
-    let tokenResult = await appCheck.getToken(forcingRefresh: false)
-    request.addValue(tokenResult.token, forHTTPHeaderField: "X-Firebase-AppCheck")
-  }
 }
diff --git a/Package.swift b/Package.swift
index 55ca4f8b833..71a2436591a 100644
--- a/Package.swift
+++ b/Package.swift
@@ -187,10 +187,6 @@ let package = Package(
       "100.0.0" ..< "101.0.0"
     ),
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
-    .package(
-      url: "https://github.com/google/generative-ai-swift.git",
-      revision: "c9f2c4913bc65aa267815962c7e91358c2d8463f"
-    ),
   ],
   targets: [
     .target(
@@ -1364,13 +1360,8 @@ let package = Package(
         "FirebaseAppCheckInterop",
         "FirebaseCore",
         "FirebaseCoreExtension",
-        .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources",
-      sources: [
-        "VertexAI.swift",
-        "VertexAIComponent.swift",
-      ]
+      path: "FirebaseVertexAI/Sources"
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From 6f9a34480c8f388970897ed5e69feb8dc5a783bd Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 18 Mar 2024 22:30:24 +0000
Subject: [PATCH 014/124] Add CI workflow for Vertex AI (#12577)

---
 .github/workflows/vertexai.yml | 37 ++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 .github/workflows/vertexai.yml

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
new file mode 100644
index 00000000000..63cfd774e43
--- /dev/null
+++ b/.github/workflows/vertexai.yml
@@ -0,0 +1,37 @@
+name: vertexai
+
+on:
+  pull_request:
+    paths:
+        - 'FirebaseVertexAI**'
+        - '.github/workflows/vertexai.yml'
+        - 'Gemfile*'
+  schedule:
+    # Run every day at 11pm (PST) - cron uses UTC times
+    - cron:  '0 7 * * *'
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  spm:
+    strategy:
+      matrix:
+        target: [iOS, macOS, catalyst]
+        os: [macos-13]
+        include:
+          - os: macos-13
+            xcode: Xcode_15.2
+    runs-on: ${{ matrix.os }}
+    env:
+      FIREBASECI_USE_LATEST_GOOGLEAPPMEASUREMENT: 1
+    steps:
+    - uses: actions/checkout@v4
+    - name: Xcode
+      run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
+    - name: Initialize xcodebuild
+      run: xcodebuild -list
+    # TODO: Add unit tests and switch from `spmbuildonly` to `spm`.
+    - name: Build
+      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAI ${{ matrix.target }} spmbuildonly

From cb2f1434dad1a7a216de016d6fa3e8f7a2960388 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 19 Mar 2024 15:44:55 +0000
Subject: [PATCH 015/124] Fork `generative-ai-swift` tests for Vertex AI
 (#12584)

---
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |  67 ++
 .../failure-model-not-found.json              |  13 +
 .../success-total-tokens.json                 |   3 +
 .../streaming-failure-empty-content.txt       |   1 +
 .../streaming-failure-error-mid-stream.txt    |  17 +
 ...streaming-failure-finish-reason-safety.txt |   2 +
 .../streaming-failure-invalid-json.txt        |   1 +
 .../streaming-failure-malformed-content.txt   |   1 +
 ...treaming-failure-prompt-blocked-safety.txt |   2 +
 ...treaming-failure-recitation-no-content.txt |   6 +
 .../streaming-failure-unknown-finish-enum.txt |  11 +
 .../streaming-success-basic-reply-long.txt    |  19 +
 .../streaming-success-basic-reply-short.txt   |   2 +
 .../streaming-success-citations.txt           |  13 +
 .../streaming-success-citations2.txt          |   5 +
 .../streaming-success-unknown-safety-enum.txt |   2 +
 .../unary-failure-api-key.json                |  21 +
 .../unary-failure-empty-content.json          |  28 +
 ...e-finish-reason-recitation-no-content.json |  46 +
 ...ilure-finish-reason-safety-no-content.json |  46 +
 .../unary-failure-finish-reason-safety.json   |  53 +
 .../unary-failure-image-rejected.json         |  13 +
 .../unary-failure-invalid-response.json       |  14 +
 .../unary-failure-malformed-content.json      |  30 +
 .../unary-failure-prompt-blocked-safety.json  |  23 +
 ...ry-failure-unknown-enum-finish-reason.json |  53 +
 ...y-failure-unknown-enum-prompt-blocked.json |  23 +
 .../unary-failure-unknown-model.json          |  13 +
 ...ary-failure-unsupported-user-location.json |  13 +
 .../unary-success-basic-reply-long.json       |  54 +
 .../unary-success-basic-reply-short.json      |  54 +
 .../unary-success-citations.json              |  64 ++
 .../unary-success-missing-safety-ratings.json |  16 +
 .../unary-success-quote-reply.json            |  54 +
 ...y-success-unknown-enum-safety-ratings.json |  46 +
 .../Tests/Unit/GenerativeModelTests.swift     | 930 ++++++++++++++++++
 .../Tests/Unit/GoogleAITests.swift            | 175 ++++
 .../Tests/Unit/MockURLProtocol.swift          |  62 ++
 .../Tests/Unit/PartsRepresentableTests.swift  | 138 +++
 39 files changed, 2134 insertions(+)
 create mode 100644 FirebaseVertexAI/Tests/Unit/ChatTests.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift

diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
new file mode 100644
index 00000000000..668e3704122
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -0,0 +1,67 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+@testable import GoogleGenerativeAI
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+final class ChatTests: XCTestCase {
+  var urlSession: URLSession!
+
+  override func setUp() {
+    let configuration = URLSessionConfiguration.default
+    configuration.protocolClasses = [MockURLProtocol.self]
+    urlSession = URLSession(configuration: configuration)
+  }
+
+  override func tearDown() {
+    MockURLProtocol.requestHandler = nil
+  }
+
+  func testMergingText() async throws {
+    let fileURL = try XCTUnwrap(Bundle.module.url(
+      forResource: "streaming-success-basic-reply-long",
+      withExtension: "txt"
+    ))
+
+    MockURLProtocol.requestHandler = { request in
+      let response = HTTPURLResponse(
+        url: request.url!,
+        statusCode: 200,
+        httpVersion: nil,
+        headerFields: nil
+      )!
+      return (response, fileURL.lines)
+    }
+
+    let model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+    let chat = Chat(model: model, history: [])
+    let input = "Test input"
+    let stream = chat.sendMessageStream(input)
+
+    // Ensure the values are parsed correctly
+    for try await value in stream {
+      XCTAssertNotNil(value.text)
+    }
+
+    XCTAssertEqual(chat.history.count, 2)
+    XCTAssertEqual(chat.history[0].parts[0].text, input)
+
+    let finalText = "1 2 3 4 5 6 7 8 9 10"
+    let assembledExpectation = ModelContent(role: "model", parts: finalText)
+    XCTAssertEqual(chat.history[0].parts[0].text, input)
+    XCTAssertEqual(chat.history[1], assembledExpectation)
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json
new file mode 100644
index 00000000000..50fcb725667
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 404,
+    "message": "models/test-model-name is not found for API version v1beta, or is not supported for countTokens. Call ListModels to see the list of available models and their supported methods.",
+    "status": "NOT_FOUND",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::not_found: models/test-model-name is not found for API version v1beta, or is not supported for countTokens. Call ListModels to see the list of available models and their supported methods. [google.rpc.error_details_ext] { message: \"models/test-model-name is not found for API version v1beta, or is not supported for countTokens. Call ListModels to see the list of available models and their supported methods.\" }"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
new file mode 100644
index 00000000000..0bccd9e7c3d
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
@@ -0,0 +1,3 @@
+{
+  "totalTokens": 6
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt
new file mode 100644
index 00000000000..5762b515325
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt
@@ -0,0 +1 @@
+data: {"candidates": [{"content": {},"index": 0}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt
new file mode 100644
index 00000000000..aeb4eb09389
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt
@@ -0,0 +1,17 @@
+data: {"candidates": [{"content": {"parts": [{"text": "First "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Second "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+{
+  "error": {
+    "code": 499,
+    "message": "The operation was cancelled.",
+    "status": "CANCELLED",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::cancelled: "
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt
new file mode 100644
index 00000000000..b73c75cf505
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt
@@ -0,0 +1,2 @@
+data: {"candidates": [{"content": {"parts": [{"text": "No"}]},"finishReason": "SAFETY","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "HIGH"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt
new file mode 100644
index 00000000000..3aea78474cb
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt
@@ -0,0 +1 @@
+data: {"this": [{"is": {"not": [{"a": "valid"}]}, "response": {}}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt
new file mode 100644
index 00000000000..273b6cdc5ca
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt
@@ -0,0 +1 @@
+data: {"candidates": [{"content": {"missing-parts": true},"index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_TOXICITY","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_VIOLENCE","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_TOXICITY","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_VIOLENCE","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt
new file mode 100644
index 00000000000..58c914af08e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt
@@ -0,0 +1,2 @@
+data: {"promptFeedback": {"blockReason": "SAFETY","safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "HIGH"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt
new file mode 100644
index 00000000000..60ec91d6dfc
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt
@@ -0,0 +1,6 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Some information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Some information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "LOW"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 30,"endIndex": 179,"uri": "https://www.example.com/some-citation","license": ""}]}}]}
+
+data: {"candidates": [{"finishReason": "RECITATION","index": 0}]}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt
new file mode 100644
index 00000000000..6194abd8957
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt
@@ -0,0 +1,11 @@
+data: {"candidates": [{"content": {"parts": [{"text": "**Cats:**\n\n- **Physical Characteristics:**\n  - Size: Cats come"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": " in a wide range of sizes, from small breeds like the Singapura to large breeds like the Maine Coon.\n  - Fur: Cats have soft, furry coats"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": " that can vary in length and texture depending on the breed.\n  - Eyes: Cats have large, expressive eyes that can be various colors, including green, blue, yellow, and hazel.\n  - Ears: Cats have pointed, erect ears that are sensitive to sound.\n  - Tail: Cats have long"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": ", flexible tails that they use for balance and communication.\n\n- **Behavior and Personality:**\n  - Independent: Cats are often described as independent animals that enjoy spending time alone.\n  - Affectionate: Despite their independent nature, cats can be very affectionate and form strong bonds with their owners.\n  - Playful: Cats are naturally playful and enjoy engaging in activities such as chasing toys, climbing, and pouncing.\n  - Curious: Cats are curious creatures that love to explore their surroundings.\n  - Vocal: Cats communicate through a variety of vocalizations, including meows, purrs, hisses, and grow"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "ls.\n\n- **Health and Care:**\n  - Diet: Cats are obligate carnivores, meaning they require animal-based protein for optimal health.\n  - Grooming: Cats spend a significant amount of time grooming themselves to keep their fur clean and free of mats.\n  - Exercise: Cats need regular exercise to stay healthy and active. This can be achieved through play sessions or access to outdoor space.\n  - Veterinary Care: Regular veterinary checkups are essential for maintaining a cat's health and detecting any potential health issues early on.\n\n**Dogs:**\n\n- **Physical Characteristics:**\n  - Size: Dogs come in a wide range of sizes, from small breeds like the Chihuahua to giant breeds like the Great Dane.\n  - Fur: Dogs have fur coats that can vary in length, texture, and color depending on the breed.\n  - Eyes: Dogs have expressive eyes that can be various colors, including brown, blue, green, and hazel.\n  - Ears: Dogs have floppy or erect ears that are sensitive to sound.\n  - Tail: Dogs have long, wagging tails that they use for communication and expressing emotions.\n\n- **Behavior and Personality:**\n  - Loyal: Dogs are known for their loyalty and"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": " devotion to their owners.\n  - Friendly: Dogs are generally friendly and outgoing animals that enjoy interacting with people and other animals.\n  - Playful: Dogs are playful and energetic creatures that love to engage in activities such as fetching, running, and playing with toys.\n  - Trainable: Dogs are highly trainable and can learn a variety of commands and tricks.\n  - Vocal: Dogs communicate through a variety of vocalizations, including barking, howling, whining, and growling.\n\n- **Health and Care:**\n  - Diet: Dogs are omnivores and can eat a variety of foods, including meat, vegetables, and grains.\n  - Grooming: Dogs require regular grooming to keep their fur clean and free of mats. The frequency of grooming depends on the breed and coat type.\n  - Exercise: Dogs need regular exercise to stay healthy and active. The amount of exercise required varies depending on the breed and age of the dog.\n  - Veterinary Care: Regular veterinary checkups are essential for maintaining a dog's health and detecting any potential health issues early on."}]},"finishReason": "FAKE_ENUM","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT_NEW_ENUM","probability": "NEGLIGIBLE_UNKNOWN_ENUM"}]}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
new file mode 100644
index 00000000000..bca95140490
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
@@ -0,0 +1,19 @@
+data: {"candidates": [{"content": {"parts": [{"text": "1 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": "2 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "3 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "4 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "5 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "6 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "7 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "8 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "9 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "10"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
new file mode 100644
index 00000000000..a7f5476954e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
@@ -0,0 +1,2 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
new file mode 100644
index 00000000000..4f50be7d32f
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
@@ -0,0 +1,13 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Some information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": " More information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": ", Even more information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": " Some information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-1","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-2","license": ""}]}}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "More information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-3","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-4","license": ""}]}}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Even more information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-5","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-6","license": ""}]}}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Physics (YouTube Channel)"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.google.com","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.google.com","license": ""}]}}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
new file mode 100644
index 00000000000..665e993ad75
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
@@ -0,0 +1,5 @@
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "In the context of the science fiction comedy novel \"The Hitchhiker's Guide to the Galaxy\" by Douglas Adams, the answer to the \"Ultimate Question of"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.055720285,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.062674366},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03904829,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03339982},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12220858,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0540987},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05781161,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03149938}]}]}
+
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Life, the Universe, and Everything\" is given as 42. However, in real life, there is no universally accepted meaning of life. The meaning"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.05910154,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.038321976},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.049589027,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.019271139},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03986249},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.098946586,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04177388}],"citationMetadata": {"citations": [{"startIndex": 52,"endIndex": 181,"uri": "https://imcsteakhouse.com.au/faqs"}]}}]}
+
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " of life is a philosophical question that has been pondered by humans for centuries, and there is no single answer that is widely agreed upon."}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.044764314,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.028870905},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04240383,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.012576347},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.10650458,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034880884},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.11085559,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03191915}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 91,"totalTokenCount": 100}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt
new file mode 100644
index 00000000000..8f7a8ff3c1c
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt
@@ -0,0 +1,2 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_SOMETHING_NEW","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
new file mode 100644
index 00000000000..ecf6f6b53fa
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -0,0 +1,21 @@
+{
+  "error": {
+    "code": 400,
+    "message": "API key not valid. Please pass a valid API key.",
+    "status": "INVALID_ARGUMENT",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.ErrorInfo",
+        "reason": "API_KEY_INVALID",
+        "domain": "googleapis.com",
+        "metadata": {
+          "service": "generativelanguage.googleapis.com"
+        }
+      },
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "Invalid API key: AIzv00G7VmUCUeC-5OglO3hcXM"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json
new file mode 100644
index 00000000000..4e1889660f2
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json
@@ -0,0 +1,28 @@
+{
+  "candidates": [
+    {
+      "content": {},
+      "index": 0
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
new file mode 100644
index 00000000000..0d99cd71d51
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
@@ -0,0 +1,46 @@
+{
+  "candidates": [
+    {
+      "finishReason": "RECITATION",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "HIGH"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json
new file mode 100644
index 00000000000..03958d4afc1
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json
@@ -0,0 +1,46 @@
+{
+    "candidates": [
+        {
+            "finishReason": "SAFETY",
+            "index": 0,
+            "safetyRatings": [
+                {
+                    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_HATE_SPEECH",
+                    "probability": "HIGH"
+                },
+                {
+                    "category": "HARM_CATEGORY_HARASSMENT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                    "probability": "NEGLIGIBLE"
+                }
+            ]
+        }
+    ],
+    "promptFeedback": {
+        "safetyRatings": [
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "probability": "NEGLIGIBLE"
+            }
+        ]
+    }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json
new file mode 100644
index 00000000000..3249c739cf8
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json
@@ -0,0 +1,53 @@
+{
+    "candidates": [
+        {
+            "content": {
+                "parts": [
+                    {
+                        "text": "No"
+                    }
+                ]
+            },
+            "finishReason": "SAFETY",
+            "index": 0,
+            "safetyRatings": [
+                {
+                    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_HATE_SPEECH",
+                    "probability": "HIGH"
+                },
+                {
+                    "category": "HARM_CATEGORY_HARASSMENT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                    "probability": "NEGLIGIBLE"
+                }
+            ]
+        }
+    ],
+    "promptFeedback": {
+        "safetyRatings": [
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "probability": "NEGLIGIBLE"
+            }
+        ]
+    }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json
new file mode 100644
index 00000000000..9dacdc71e7a
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 400,
+    "message": "Request contains an invalid argument.",
+    "status": "INVALID_ARGUMENT",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::invalid_argument: invalid status photos.thumbnailer.Status.Code::5: Source image 0 too short"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json
new file mode 100644
index 00000000000..49d05e1840b
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json
@@ -0,0 +1,14 @@
+{
+  "this": [
+    {
+      "is": {
+        "not": [
+          {
+            "a": "valid"
+          }
+        ]
+      },
+      "response": {}
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json
new file mode 100644
index 00000000000..737f2e08548
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json
@@ -0,0 +1,30 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "invalid-field": true
+      },
+      "index": 0
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json
new file mode 100644
index 00000000000..9d2abbb23d6
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json
@@ -0,0 +1,23 @@
+{
+  "promptFeedback": {
+    "blockReason": "SAFETY",
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "HIGH"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json
new file mode 100644
index 00000000000..1adbc70cae4
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json
@@ -0,0 +1,53 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Some text"
+          }
+        ]
+      },
+      "finishReason": "FAKE_NEW_FINISH_REASON",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json
new file mode 100644
index 00000000000..a00cbb72cfc
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json
@@ -0,0 +1,23 @@
+{
+  "promptFeedback": {
+    "blockReason": "FAKE_NEW_BLOCK_REASON",
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json
new file mode 100644
index 00000000000..60b3f55c978
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 404,
+    "message": "models/unknown is not found for API version v1, or is not supported for GenerateContent. Call ListModels to see the list of available models and their supported methods.",
+    "status": "NOT_FOUND",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::not_found: models/unknown is not found for API version v1, or is not supported for GenerateContent. Call ListModels to see the list of available models and their supported methods. [google.rpc.error_details_ext] { message: \"models/unknown is not found for API version v1, or is not supported for GenerateContent. Call ListModels to see the list of available models and their supported methods.\" }"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
new file mode 100644
index 00000000000..c4c2ace4e20
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 400,
+    "message": "User location is not supported for the API use.",
+    "status": "FAILED_PRECONDITION",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::failed_precondition: User location is not supported for the API use. [google.rpc.error_details_ext] { message: \"User location is not supported for the API use.\" }"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
new file mode 100644
index 00000000000..59b84de92fe
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
@@ -0,0 +1,54 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "You can ask me a wide range of questions on various topics. Here are some examples:\n\n1. **General Knowledge:**\n   - What is the capital of France?\n   - Who painted the Mona Lisa?\n   - What is the largest ocean in the world?\n\n2. **Science and Technology:**\n   - How does a computer work?\n   - What is the difference between a virus and a bacteria?\n   - What are the latest advancements in artificial intelligence?\n\n3. **History and Culture:**\n   - Who was the first president of the United States?\n   - What is the significance of the Great Wall of China?\n   - What are some of the most famous works of Shakespeare?\n\n4. **Current Events:**\n   - What is the latest news about the COVID-19 pandemic?\n   - Who is the current president of Ukraine?\n   - What are the major issues being discussed in the upcoming election?\n\n5. **Personal Questions:**\n   - What are your hobbies?\n   - What is your favorite book or movie?\n   - What are your thoughts on the future of technology?\n\n6. **Fun and Games:**\n   - Can you tell me a joke?\n   - What is the answer to this riddle: \"I have keys but no locks. I have space but no room. You can enter, but can't go outside.\" (Answer: a keyboard)\n   - Let's play a game of 20 questions.\n\n7. **Hypothetical Questions:**\n   - What would you do if you won the lottery?\n   - What would happen if time travel were possible?\n   - What is the meaning of life?\n\n8. **Philosophical Questions:**\n   - What is the nature of reality?\n   - Does free will exist?\n   - What is the difference between right and wrong?\n\n9. **Creative Questions:**\n   - Write a poem about a sunset.\n   - Design a logo for a new company.\n   - Compose a song about your favorite season.\n\n10. **Technical Questions:**\n    - How can I improve the performance of my computer?\n    - What is the best way to troubleshoot a network issue?\n    - How do I create a website using HTML and CSS?\n\nRemember, I am still under development and may not be able to answer all questions perfectly. However, I will do my best to provide you with accurate and informative responses."
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
new file mode 100644
index 00000000000..40a9a6da58e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
@@ -0,0 +1,54 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Mountain View, California, United States"
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
new file mode 100644
index 00000000000..ac99fcf0944
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
@@ -0,0 +1,64 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Some information cited from an external source"
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ],
+      "citationMetadata": {
+        "citationSources": [
+          {
+            "startIndex": 574,
+            "endIndex": 705,
+            "uri": "https://www.example.com/some-citation",
+            "license": ""
+          }
+        ]
+      }
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json
new file mode 100644
index 00000000000..645d3e6425e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json
@@ -0,0 +1,16 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "This is the generated content."
+          }
+        ]
+      },
+      "index": 0
+    }
+  ],
+  "promptFeedback": {
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json
new file mode 100644
index 00000000000..e5bf4f0b497
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json
@@ -0,0 +1,54 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Google's mission is to \"organize the world's information and make it universally accessible and useful.\""
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json
new file mode 100644
index 00000000000..3c47a21322c
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json
@@ -0,0 +1,46 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Some text"
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "MEDIUM"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "FAKE_NEW_HARM_PROBABILITY"
+        },
+        {
+          "category": "FAKE_NEW_HARM_CATEGORY",
+          "probability": "HIGH"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "MEDIUM"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "FAKE_NEW_HARM_PROBABILITY"
+      },
+      {
+        "category": "FAKE_NEW_HARM_CATEGORY",
+        "probability": "HIGH"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
new file mode 100644
index 00000000000..b835a9ea05f
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -0,0 +1,930 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+@testable import GoogleGenerativeAI
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+final class GenerativeModelTests: XCTestCase {
+  let testPrompt = "What sorts of questions can I ask you?"
+  let safetyRatingsNegligible: [SafetyRating] = [
+    .init(category: .sexuallyExplicit, probability: .negligible),
+    .init(category: .hateSpeech, probability: .negligible),
+    .init(category: .harassment, probability: .negligible),
+    .init(category: .dangerousContent, probability: .negligible),
+  ]
+
+  var urlSession: URLSession!
+  var model: GenerativeModel!
+
+  override func setUp() async throws {
+    let configuration = URLSessionConfiguration.default
+    configuration.protocolClasses = [MockURLProtocol.self]
+    urlSession = try XCTUnwrap(URLSession(configuration: configuration))
+    model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+  }
+
+  override func tearDown() {
+    MockURLProtocol.requestHandler = nil
+  }
+
+  // MARK: - Generate Content
+
+  func testGenerateContent_success_basicReplyLong() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-long",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    let finishReason = try XCTUnwrap(candidate.finishReason)
+    XCTAssertEqual(finishReason, .stop)
+    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    let partText = try XCTUnwrap(part.text)
+    XCTAssertTrue(partText.hasPrefix("You can ask me a wide range of questions"))
+    XCTAssertEqual(response.text, partText)
+    let promptFeedback = try XCTUnwrap(response.promptFeedback)
+    XCTAssertNil(promptFeedback.blockReason)
+    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+  }
+
+  func testGenerateContent_success_basicReplyShort() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    let finishReason = try XCTUnwrap(candidate.finishReason)
+    XCTAssertEqual(finishReason, .stop)
+    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    XCTAssertEqual(part.text, "Mountain View, California, United States")
+    XCTAssertEqual(response.text, part.text)
+    let promptFeedback = try XCTUnwrap(response.promptFeedback)
+    XCTAssertNil(promptFeedback.blockReason)
+    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+  }
+
+  func testGenerateContent_success_citations() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-citations",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    XCTAssertEqual(response.text, "Some information cited from an external source")
+    let citationMetadata = try XCTUnwrap(candidate.citationMetadata)
+    XCTAssertEqual(citationMetadata.citationSources.count, 1)
+    let citationSource = try XCTUnwrap(citationMetadata.citationSources.first)
+    XCTAssertEqual(citationSource.uri, "https://www.example.com/some-citation")
+    XCTAssertEqual(citationSource.startIndex, 574)
+    XCTAssertEqual(citationSource.endIndex, 705)
+    XCTAssertEqual(citationSource.license, "")
+  }
+
+  func testGenerateContent_success_quoteReply() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-quote-reply",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    let finishReason = try XCTUnwrap(candidate.finishReason)
+    XCTAssertEqual(finishReason, .stop)
+    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    let partText = try XCTUnwrap(part.text)
+    XCTAssertTrue(partText.hasPrefix("Google"))
+    XCTAssertEqual(response.text, part.text)
+    let promptFeedback = try XCTUnwrap(response.promptFeedback)
+    XCTAssertNil(promptFeedback.blockReason)
+    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+  }
+
+  func testGenerateContent_success_unknownEnum_safetyRatings() async throws {
+    let expectedSafetyRatings = [
+      SafetyRating(category: .harassment, probability: .medium),
+      SafetyRating(category: .dangerousContent, probability: .unknown),
+      SafetyRating(category: .unknown, probability: .high),
+    ]
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-unknown-enum-safety-ratings",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.text, "Some text")
+    XCTAssertEqual(response.candidates.first?.safetyRatings, expectedSafetyRatings)
+    XCTAssertEqual(response.promptFeedback?.safetyRatings, expectedSafetyRatings)
+  }
+
+  func testGenerateContent_success_prefixedModelName() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json"
+      )
+    let model = GenerativeModel(
+      // Model name is prefixed with "models/".
+      name: "models/test-model",
+      apiKey: "API_KEY",
+      urlSession: urlSession
+    )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_failure_invalidAPIKey() async throws {
+    let expectedStatusCode = 400
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-api-key",
+        withExtension: "json",
+        statusCode: expectedStatusCode
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError.invalidAPIKey(message) {
+      XCTAssertEqual(message, "API key not valid. Please pass a valid API key.")
+    } catch {
+      XCTFail("Should throw GenerateContentError.invalidAPIKey; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_emptyContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-empty-content",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError
+      .internalError(underlying: invalidCandidateError as InvalidCandidateError) {
+      guard case let .emptyContent(decodingError) = invalidCandidateError else {
+        XCTFail("Not an InvalidCandidateError.emptyContent error: \(invalidCandidateError)")
+        return
+      }
+      _ = try XCTUnwrap(decodingError as? DecodingError,
+                        "Not a DecodingError: \(decodingError)")
+    } catch {
+      XCTFail("Should throw GenerateContentError.internalError; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_finishReasonSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-finish-reason-safety",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.responseStoppedEarly(reason, response) {
+      XCTAssertEqual(reason, .safety)
+      XCTAssertEqual(response.text, "No")
+    } catch {
+      XCTFail("Should throw a responseStoppedEarly")
+    }
+  }
+
+  func testGenerateContent_failure_finishReasonSafety_noContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-finish-reason-safety-no-content",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.responseStoppedEarly(reason, response) {
+      XCTAssertEqual(reason, .safety)
+      XCTAssertNil(response.text)
+    } catch {
+      XCTFail("Should throw a responseStoppedEarly")
+    }
+  }
+
+  func testGenerateContent_failure_imageRejected() async throws {
+    let expectedStatusCode = 400
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-image-rejected",
+        withExtension: "json",
+        statusCode: 400
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError.internalError(underlying: rpcError as RPCError) {
+      XCTAssertEqual(rpcError.status, .invalidArgument)
+      XCTAssertEqual(rpcError.httpResponseCode, expectedStatusCode)
+      XCTAssertEqual(rpcError.message, "Request contains an invalid argument.")
+    } catch {
+      XCTFail("Should throw GenerateContentError.internalError; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_promptBlockedSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-prompt-blocked-safety",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.promptBlocked(response) {
+      XCTAssertNil(response.text)
+    } catch {
+      XCTFail("Should throw a promptBlocked")
+    }
+  }
+
+  func testGenerateContent_failure_unknownEnum_finishReason() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unknown-enum-finish-reason",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.responseStoppedEarly(reason, response) {
+      XCTAssertEqual(reason, .unknown)
+      XCTAssertEqual(response.text, "Some text")
+    } catch {
+      XCTFail("Should throw a responseStoppedEarly")
+    }
+  }
+
+  func testGenerateContent_failure_unknownEnum_promptBlocked() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unknown-enum-prompt-blocked",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.promptBlocked(response) {
+      let promptFeedback = try XCTUnwrap(response.promptFeedback)
+      XCTAssertEqual(promptFeedback.blockReason, .unknown)
+    } catch {
+      XCTFail("Should throw a promptBlocked")
+    }
+  }
+
+  func testGenerateContent_failure_unknownModel() async throws {
+    let expectedStatusCode = 404
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unknown-model",
+        withExtension: "json",
+        statusCode: 404
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError.internalError(underlying: rpcError as RPCError) {
+      XCTAssertEqual(rpcError.status, .notFound)
+      XCTAssertEqual(rpcError.httpResponseCode, expectedStatusCode)
+      XCTAssertTrue(rpcError.message.hasPrefix("models/unknown is not found"))
+    } catch {
+      XCTFail("Should throw GenerateContentError.internalError; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_unsupportedUserLocation() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unsupported-user-location",
+        withExtension: "json",
+        statusCode: 400
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.unsupportedUserLocation; no error thrown.")
+    } catch GenerateContentError.unsupportedUserLocation {
+      return
+    }
+
+    XCTFail("Expected an unsupported user location error.")
+  }
+
+  func testGenerateContent_failure_nonHTTPResponse() async throws {
+    MockURLProtocol.requestHandler = try nonHTTPRequestHandler()
+
+    var responseError: Error?
+    var content: GenerateContentResponse?
+    do {
+      content = try await model.generateContent(testPrompt)
+    } catch {
+      responseError = error
+    }
+
+    XCTAssertNil(content)
+    XCTAssertNotNil(responseError)
+    let generateContentError = try XCTUnwrap(responseError as? GenerateContentError)
+    guard case let .internalError(underlyingError) = generateContentError else {
+      XCTFail("Not an internal error: \(generateContentError)")
+      return
+    }
+    XCTAssertEqual(underlyingError.localizedDescription, "Response was not an HTTP response.")
+  }
+
+  func testGenerateContent_failure_invalidResponse() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "unary-failure-invalid-response",
+      withExtension: "json"
+    )
+
+    var responseError: Error?
+    var content: GenerateContentResponse?
+    do {
+      content = try await model.generateContent(testPrompt)
+    } catch {
+      responseError = error
+    }
+
+    XCTAssertNil(content)
+    XCTAssertNotNil(responseError)
+    let generateContentError = try XCTUnwrap(responseError as? GenerateContentError)
+    guard case let .internalError(underlyingError) = generateContentError else {
+      XCTFail("Not an internal error: \(generateContentError)")
+      return
+    }
+    let decodingError = try XCTUnwrap(underlyingError as? DecodingError)
+    guard case let .dataCorrupted(context) = decodingError else {
+      XCTFail("Not a data corrupted error: \(decodingError)")
+      return
+    }
+    XCTAssert(context.debugDescription.hasPrefix("Failed to decode GenerateContentResponse"))
+  }
+
+  func testGenerateContent_failure_malformedContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-malformed-content",
+        withExtension: "json"
+      )
+
+    var responseError: Error?
+    var content: GenerateContentResponse?
+    do {
+      content = try await model.generateContent(testPrompt)
+    } catch {
+      responseError = error
+    }
+
+    XCTAssertNil(content)
+    XCTAssertNotNil(responseError)
+    let generateContentError = try XCTUnwrap(responseError as? GenerateContentError)
+    guard case let .internalError(underlyingError) = generateContentError else {
+      XCTFail("Not an internal error: \(generateContentError)")
+      return
+    }
+    let invalidCandidateError = try XCTUnwrap(underlyingError as? InvalidCandidateError)
+    guard case let .malformedContent(malformedContentUnderlyingError) = invalidCandidateError else {
+      XCTFail("Not a malformed content error: \(invalidCandidateError)")
+      return
+    }
+    _ = try XCTUnwrap(
+      malformedContentUnderlyingError as? DecodingError,
+      "Not a decoding error: \(malformedContentUnderlyingError)"
+    )
+  }
+
+  func testGenerateContentMissingSafetyRatings() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "unary-success-missing-safety-ratings",
+      withExtension: "json"
+    )
+
+    let content = try await model.generateContent(testPrompt)
+    let promptFeedback = try XCTUnwrap(content.promptFeedback)
+    XCTAssertEqual(promptFeedback.safetyRatings.count, 0)
+    XCTAssertEqual(content.text, "This is the generated content.")
+  }
+
+  func testGenerateContent_requestOptions_customTimeout() async throws {
+    let expectedTimeout = 150.0
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        timeout: expectedTimeout
+      )
+    let requestOptions = RequestOptions(timeout: expectedTimeout)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: requestOptions,
+      urlSession: urlSession
+    )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+  }
+
+  // MARK: - Generate Content (Streaming)
+
+  func testGenerateContentStream_failureInvalidAPIKey() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-api-key",
+        withExtension: "json"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("No content is there, this shouldn't happen.")
+      }
+    } catch GenerateContentError.invalidAPIKey {
+      // invalidAPIKey error is as expected, nothing else to check.
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failureEmptyContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-empty-content",
+        withExtension: "txt"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("No content is there, this shouldn't happen.")
+      }
+    } catch GenerateContentError.internalError(_ as InvalidCandidateError) {
+      // Underlying error is as expected, nothing else to check.
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failureFinishReasonSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-finish-reason-safety",
+        withExtension: "txt"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("Content shouldn't be shown, this shouldn't happen.")
+      }
+    } catch let GenerateContentError.responseStoppedEarly(reason, _) {
+      XCTAssertEqual(reason, .safety)
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failurePromptBlockedSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-prompt-blocked-safety",
+        withExtension: "txt"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("Content shouldn't be shown, this shouldn't happen.")
+      }
+    } catch let GenerateContentError.promptBlocked(response) {
+      XCTAssertEqual(response.promptFeedback?.blockReason, .safety)
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failureUnknownFinishEnum() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-unknown-finish-enum",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream("Hi")
+    do {
+      for try await content in stream {
+        XCTAssertNotNil(content.text)
+      }
+    } catch let GenerateContentError.responseStoppedEarly(reason, _) {
+      XCTAssertEqual(reason, .unknown)
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_successBasicReplyLong() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-long",
+        withExtension: "txt"
+      )
+
+    var responses = 0
+    let stream = model.generateContentStream("Hi")
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      responses += 1
+    }
+
+    XCTAssertEqual(responses, 10)
+  }
+
+  func testGenerateContentStream_successBasicReplyShort() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt"
+      )
+
+    var responses = 0
+    let stream = model.generateContentStream("Hi")
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      responses += 1
+    }
+
+    XCTAssertEqual(responses, 1)
+  }
+
+  func testGenerateContentStream_successUnknownSafetyEnum() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-unknown-safety-enum",
+        withExtension: "txt"
+      )
+
+    var hadUnknown = false
+    let stream = model.generateContentStream("Hi")
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      if let ratings = content.candidates.first?.safetyRatings,
+         ratings.contains(where: { $0.category == .unknown }) {
+        hadUnknown = true
+      }
+    }
+
+    XCTAssertTrue(hadUnknown)
+  }
+
+  func testGenerateContentStream_successWithCitations() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-citations",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream("Hi")
+    var citations: [Citation] = []
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      let candidate = try XCTUnwrap(content.candidates.first)
+      XCTAssertEqual(candidate.finishReason, .stop)
+      if let sources = candidate.citationMetadata?.citationSources {
+        citations.append(contentsOf: sources)
+      }
+    }
+
+    XCTAssertEqual(citations.count, 8)
+    XCTAssertTrue(citations
+      .contains(where: { $0.startIndex == 574 && $0.endIndex == 705 && !$0.uri.isEmpty }))
+    XCTAssertTrue(citations
+      .contains(where: { $0.startIndex == 899 && $0.endIndex == 1026 && !$0.uri.isEmpty }))
+  }
+
+  func testGenerateContentStream_errorMidStream() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "streaming-failure-error-mid-stream",
+      withExtension: "txt"
+    )
+
+    var responseCount = 0
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await content in stream {
+        XCTAssertNotNil(content.text)
+        responseCount += 1
+      }
+    } catch let GenerateContentError.internalError(rpcError as RPCError) {
+      XCTAssertEqual(rpcError.httpResponseCode, 499)
+      XCTAssertEqual(rpcError.status, .cancelled)
+
+      // Check the content count is correct.
+      XCTAssertEqual(responseCount, 2)
+      return
+    }
+
+    XCTFail("Expected an internalError with an RPCError.")
+  }
+
+  func testGenerateContentStream_nonHTTPResponse() async throws {
+    MockURLProtocol.requestHandler = try nonHTTPRequestHandler()
+
+    let stream = model.generateContentStream("Hi")
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch let GenerateContentError.internalError(underlying) {
+      XCTAssertEqual(underlying.localizedDescription, "Response was not an HTTP response.")
+      return
+    }
+
+    XCTFail("Expected an internal error.")
+  }
+
+  func testGenerateContentStream_invalidResponse() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-invalid-json",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch let GenerateContentError.internalError(underlying as DecodingError) {
+      guard case let .dataCorrupted(context) = underlying else {
+        XCTFail("Not a data corrupted error: \(underlying)")
+        return
+      }
+      XCTAssert(context.debugDescription.hasPrefix("Failed to decode GenerateContentResponse"))
+      return
+    }
+
+    XCTFail("Expected an internal error.")
+  }
+
+  func testGenerateContentStream_malformedContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-malformed-content",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch let GenerateContentError.internalError(underlyingError as InvalidCandidateError) {
+      guard case let .malformedContent(contentError) = underlyingError else {
+        XCTFail("Not a malformed content error: \(underlyingError)")
+        return
+      }
+
+      XCTAssert(contentError is DecodingError)
+      return
+    }
+
+    XCTFail("Expected an internal decoding error.")
+  }
+
+  func testGenerateContentStream_failure_unsupportedUserLocation() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unsupported-user-location",
+        withExtension: "json",
+        statusCode: 400
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch GenerateContentError.unsupportedUserLocation {
+      return
+    }
+
+    XCTFail("Expected an unsupported user location error.")
+  }
+
+  func testGenerateContentStream_requestOptions_customTimeout() async throws {
+    let expectedTimeout = 150.0
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt",
+        timeout: expectedTimeout
+      )
+    let requestOptions = RequestOptions(timeout: expectedTimeout)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: requestOptions,
+      urlSession: urlSession
+    )
+
+    var responses = 0
+    let stream = model.generateContentStream(testPrompt)
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      responses += 1
+    }
+
+    XCTAssertEqual(responses, 1)
+  }
+
+  // MARK: - Count Tokens
+
+  func testCountTokens_succeeds() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "success-total-tokens",
+      withExtension: "json"
+    )
+
+    let response = try await model.countTokens("Why is the sky blue?")
+    XCTAssertEqual(response.totalTokens, 6)
+  }
+
+  func testCountTokens_modelNotFound() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "failure-model-not-found", withExtension: "json",
+      statusCode: 404
+    )
+
+    do {
+      _ = try await model.countTokens("Why is the sky blue?")
+      XCTFail("Request should not have succeeded.")
+    } catch let CountTokensError.internalError(rpcError as RPCError) {
+      XCTAssertEqual(rpcError.httpResponseCode, 404)
+      XCTAssertEqual(rpcError.status, .notFound)
+      XCTAssert(rpcError.message.hasPrefix("models/test-model-name is not found"))
+      return
+    }
+
+    XCTFail("Expected internal RPCError.")
+  }
+
+  func testCountTokens_requestOptions_customTimeout() async throws {
+    let expectedTimeout = 150.0
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "success-total-tokens",
+        withExtension: "json",
+        timeout: expectedTimeout
+      )
+    let requestOptions = RequestOptions(timeout: expectedTimeout)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: requestOptions,
+      urlSession: urlSession
+    )
+
+    let response = try await model.countTokens(testPrompt)
+
+    XCTAssertEqual(response.totalTokens, 6)
+  }
+
+  // MARK: - Model Resource Name
+
+  func testModelResourceName_noPrefix() async throws {
+    let modelName = "my-model"
+    let modelResourceName = "models/\(modelName)"
+
+    model = GenerativeModel(name: modelName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, modelResourceName)
+  }
+
+  func testModelResourceName_modelsPrefix() async throws {
+    let modelResourceName = "models/my-model"
+
+    model = GenerativeModel(name: modelResourceName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, modelResourceName)
+  }
+
+  func testModelResourceName_tunedModelsPrefix() async throws {
+    let tunedModelResourceName = "tunedModels/my-model"
+
+    model = GenerativeModel(name: tunedModelResourceName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
+  }
+
+  // MARK: - Helpers
+
+  private func nonHTTPRequestHandler() throws -> ((URLRequest) -> (
+    URLResponse,
+    AsyncLineSequence<URL.AsyncBytes>?
+  )) {
+    return { request in
+      // This is *not* an HTTPURLResponse
+      let response = URLResponse(
+        url: request.url!,
+        mimeType: nil,
+        expectedContentLength: 0,
+        textEncodingName: nil
+      )
+      return (response, nil)
+    }
+  }
+
+  private func httpRequestHandler(forResource name: String,
+                                  withExtension ext: String,
+                                  statusCode: Int = 200,
+                                  timeout: TimeInterval = URLRequest
+                                    .defaultTimeoutInterval()) throws -> ((URLRequest) throws -> (
+    URLResponse,
+    AsyncLineSequence<URL.AsyncBytes>?
+  )) {
+    let fileURL = try XCTUnwrap(Bundle.module.url(forResource: name, withExtension: ext))
+    return { request in
+      let requestURL = try XCTUnwrap(request.url)
+      XCTAssertEqual(requestURL.path.occurrenceCount(of: "models/"), 1)
+      XCTAssertEqual(request.timeoutInterval, timeout)
+      let response = try XCTUnwrap(HTTPURLResponse(
+        url: requestURL,
+        statusCode: statusCode,
+        httpVersion: nil,
+        headerFields: nil
+      ))
+      return (response, fileURL.lines)
+    }
+  }
+}
+
+private extension String {
+  /// Returns the number of occurrences of `substring` in the `String`.
+  func occurrenceCount(of substring: String) -> Int {
+    return components(separatedBy: substring).count - 1
+  }
+}
+
+private extension URLRequest {
+  /// Returns the default `timeoutInterval` for a `URLRequest`.
+  static func defaultTimeoutInterval() -> TimeInterval {
+    let placeholderURL = URL(string: "https://example.com")!
+    return URLRequest(url: placeholderURL).timeoutInterval
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift b/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
new file mode 100644
index 00000000000..cbc92527a1c
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
@@ -0,0 +1,175 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import GoogleGenerativeAI
+import XCTest
+#if canImport(AppKit)
+  import AppKit // For NSImage extensions.
+#elseif canImport(UIKit)
+  import UIKit // For UIImage extensions.
+#endif
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+final class GoogleGenerativeAITests: XCTestCase {
+  func codeSamples() async throws {
+    let config = GenerationConfig(temperature: 0.2,
+                                  topP: 0.1,
+                                  topK: 16,
+                                  candidateCount: 4,
+                                  maxOutputTokens: 256,
+                                  stopSequences: ["..."])
+    let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
+
+    // Permutations without optional arguments.
+    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY")
+    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", safetySettings: filters)
+    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", generationConfig: config)
+
+    // All arguments passed.
+    let genAI = GenerativeModel(name: "gemini-1.0-pro",
+                                apiKey: "API_KEY",
+                                generationConfig: config, // Optional
+                                safetySettings: filters // Optional
+    )
+    // Full Typed Usage
+    let pngData = Data() // ....
+    let contents = [ModelContent(role: "user",
+                                 parts: [
+                                   .text("Is it a cat?"),
+                                   .png(pngData),
+                                 ])]
+
+    do {
+      let response = try await genAI.generateContent(contents)
+      print(response.text ?? "Couldn't get text... check status")
+    } catch {
+      print("Error generating content: \(error)")
+    }
+
+    // Content input combinations.
+    let _ = try await genAI.generateContent("Constant String")
+    let str = "String Variable"
+    let _ = try await genAI.generateContent(str)
+    let _ = try await genAI.generateContent([str])
+    let _ = try await genAI.generateContent(str, "abc", "def")
+    #if canImport(UIKit)
+      _ = try await genAI.generateContent(UIImage())
+      _ = try await genAI.generateContent([UIImage()])
+      _ = try await genAI
+        .generateContent([str, UIImage(), ModelContent.Part.text(str)])
+      _ = try await genAI.generateContent(str, UIImage(), "def", UIImage())
+      _ = try await genAI.generateContent([str, UIImage(), "def", UIImage()])
+      _ = try await genAI.generateContent([ModelContent("def", UIImage()),
+                                           ModelContent("def", UIImage())])
+    #elseif canImport(AppKit)
+      _ = try await genAI.generateContent(NSImage())
+      _ = try await genAI.generateContent([NSImage()])
+      _ = try await genAI.generateContent(str, NSImage(), "def", NSImage())
+      _ = try await genAI.generateContent([str, NSImage(), "def", NSImage()])
+    #endif
+
+    // ThrowingPartsRepresentable combinations.
+    let _ = ModelContent(parts: [.text(str)])
+    let _ = ModelContent(role: "model", parts: [.text(str)])
+    let _ = ModelContent(parts: "Constant String")
+    let _ = ModelContent(parts: str)
+    // Note: This requires the `try` for some reason. Casting to explicit [PartsRepresentable] also
+    // doesn't work.
+    let _ = try ModelContent(parts: [str])
+    // Note: without `as [any ThrowingPartsRepresentable]` this will fail to compile with "Cannot
+    // convert value of type 'String' to expected element type
+    // 'Array<ModelContent.Part>.ArrayLiteralElement'. Not sure if there's a way we can get it to
+    // work.
+    let _ = try ModelContent(parts: [str, ModelContent.Part.data(
+      mimetype: "foo",
+      Data()
+    )] as [any ThrowingPartsRepresentable])
+    #if canImport(UIKit)
+      _ = try ModelContent(role: "user", parts: UIImage())
+      _ = try ModelContent(role: "user", parts: [UIImage()])
+      // Note: without `as [any ThrowingPartsRepresentable]` this will fail to compile with "Cannot
+      // convert
+      // value of type `[Any]` to expected type `[any ThrowingPartsRepresentable]`. Not sure if
+      // there's a
+      // way we can get it to work.
+      _ = try ModelContent(parts: [str, UIImage()] as [any ThrowingPartsRepresentable])
+      // Alternatively, you can explicitly declare the type in a variable and pass it in.
+      let representable2: [any ThrowingPartsRepresentable] = [str, UIImage()]
+      _ = try ModelContent(parts: representable2)
+      _ = try ModelContent(parts: [str, UIImage(),
+                                   ModelContent.Part.text(str)] as [any ThrowingPartsRepresentable])
+    #elseif canImport(AppKit)
+      _ = try ModelContent(role: "user", parts: NSImage())
+      _ = try ModelContent(role: "user", parts: [NSImage()])
+      // Note: without `as [any ThrowingPartsRepresentable]` this will fail to compile with "Cannot
+      // convert
+      // value of type `[Any]` to expected type `[any ThrowingPartsRepresentable]`. Not sure if
+      // there's a
+      // way we can get it to work.
+      _ = try ModelContent(parts: [str, NSImage()] as [any ThrowingPartsRepresentable])
+      // Alternatively, you can explicitly declare the type in a variable and pass it in.
+      let representable2: [any ThrowingPartsRepresentable] = [str, NSImage()]
+      _ = try ModelContent(parts: representable2)
+      _ =
+        try ModelContent(parts: [str, NSImage(),
+                                 ModelContent.Part.text(str)] as [any ThrowingPartsRepresentable])
+    #endif
+
+    // countTokens API
+    let _: CountTokensResponse = try await genAI.countTokens("What color is the Sky?")
+    #if canImport(UIKit)
+      let _: CountTokensResponse = try await genAI.countTokens("What color is the Sky?",
+                                                               UIImage())
+      let _: CountTokensResponse = try await genAI.countTokens([
+        ModelContent("What color is the Sky?", UIImage()),
+        ModelContent(UIImage(), "What color is the Sky?", UIImage()),
+      ])
+    #endif
+
+    // Chat
+    _ = genAI.startChat()
+    _ = genAI.startChat(history: [ModelContent(parts: "abc")])
+  }
+
+  // Result builder alternative
+
+  /*
+   let pngData = Data() // ....
+   let contents = [GenAIContent(role: "user",
+                                parts: [
+                                 .text("Is it a cat?"),
+                                 .png(pngData)
+                                ])]
+
+   // Turns into...
+
+   let contents = GenAIContent {
+     Role("user") {
+       Text("Is this a cat?")
+       Image(png: pngData)
+     }
+   }
+
+   GenAIContent {
+     ForEach(myInput) { input in
+       Role(input.role) {
+         input.contents
+       }
+     }
+   }
+
+   // Thoughts: this looks great from a code demo, but since I assume most content will be
+   // user generated, the result builder may not be the best API.
+   */
+}
diff --git a/FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift b/FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift
new file mode 100644
index 00000000000..30528cdf35a
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift
@@ -0,0 +1,62 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+class MockURLProtocol: URLProtocol {
+  static var requestHandler: ((URLRequest) throws -> (
+    URLResponse,
+    AsyncLineSequence<URL.AsyncBytes>?
+  ))?
+
+  override class func canInit(with request: URLRequest) -> Bool { return true }
+
+  override class func canonicalRequest(for request: URLRequest) -> URLRequest { return request }
+
+  override func startLoading() {
+    guard let requestHandler = MockURLProtocol.requestHandler else {
+      fatalError("`requestHandler` is nil.")
+    }
+    guard let client = client else {
+      fatalError("`client` is nil.")
+    }
+
+    Task {
+      let (response, stream) = try requestHandler(self.request)
+      client.urlProtocol(self, didReceive: response, cacheStoragePolicy: .notAllowed)
+      if let stream = stream {
+        do {
+          for try await line in stream {
+            guard let data = line.data(using: .utf8) else {
+              fatalError("Failed to convert \"\(line)\" to UTF8 data.")
+            }
+            client.urlProtocol(self, didLoad: data)
+            // Add a newline character since AsyncLineSequence strips them when reading line by
+            // line;
+            // without the following, the whole file is delivered as a single line.
+            client.urlProtocol(self, didLoad: "\n".data(using: .utf8)!)
+          }
+        } catch {
+          client.urlProtocol(self, didFailWithError: error)
+          XCTFail("Unexpected failure reading lines from stream: \(error.localizedDescription)")
+        }
+      }
+      client.urlProtocolDidFinishLoading(self)
+    }
+  }
+
+  override func stopLoading() {}
+}
diff --git a/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
new file mode 100644
index 00000000000..da2f35f727f
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
@@ -0,0 +1,138 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import CoreGraphics
+import CoreImage
+import GoogleGenerativeAI
+import XCTest
+#if canImport(UIKit)
+  import UIKit
+#else
+  import AppKit
+#endif
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+final class PartsRepresentableTests: XCTestCase {
+  func testModelContentFromCGImageIsNotEmpty() throws {
+    // adapted from https://forums.swift.org/t/creating-a-cgimage-from-color-array/18634/2
+    var srgbArray = [UInt32](repeating: 0xFFFF_FFFF, count: 8 * 8)
+    let image = srgbArray.withUnsafeMutableBytes { ptr -> CGImage in
+      let ctx = CGContext(
+        data: ptr.baseAddress,
+        width: 8,
+        height: 8,
+        bitsPerComponent: 8,
+        bytesPerRow: 4 * 8,
+        space: CGColorSpace(name: CGColorSpace.sRGB)!,
+        bitmapInfo: CGBitmapInfo.byteOrder32Little.rawValue +
+          CGImageAlphaInfo.premultipliedFirst.rawValue
+      )!
+      return ctx.makeImage()!
+    }
+    let modelContent = try image.tryPartsValue()
+    XCTAssert(modelContent.count > 0, "Expected non-empty model content for CGImage: \(image)")
+  }
+
+  func testModelContentFromCIImageIsNotEmpty() throws {
+    let image = CIImage(color: CIColor.red)
+      .cropped(to: CGRect(origin: CGPointZero, size: CGSize(width: 16, height: 16)))
+    let modelContent = try image.tryPartsValue()
+    XCTAssert(modelContent.count > 0, "Expected non-empty model content for CGImage: \(image)")
+  }
+
+  func testModelContentFromInvalidCIImageThrows() throws {
+    let image = CIImage.empty()
+    do {
+      let _ = try image.tryPartsValue()
+    } catch {
+      guard let imageError = (error as? ImageConversionError) else {
+        XCTFail("Got unexpected error type: \(error)")
+        return
+      }
+      switch imageError {
+      case let .couldNotConvertToJPEG(source):
+        // String(describing:) works around a type error.
+        XCTAssertEqual(String(describing: source), String(describing: image))
+        return
+      case _:
+        XCTFail("Expected image conversion error, got \(imageError) instead")
+        return
+      }
+    }
+    XCTFail("Expected model content from invlaid image to error")
+  }
+
+  #if canImport(UIKit)
+    func testModelContentFromInvalidUIImageThrows() throws {
+      let image = UIImage()
+      do {
+        _ = try image.tryPartsValue()
+      } catch {
+        guard let imageError = (error as? ImageConversionError) else {
+          XCTFail("Got unexpected error type: \(error)")
+          return
+        }
+        switch imageError {
+        case let .couldNotConvertToJPEG(source):
+          // String(describing:) works around a type error.
+          XCTAssertEqual(String(describing: source), String(describing: image))
+          return
+        case _:
+          XCTFail("Expected image conversion error, got \(imageError) instead")
+          return
+        }
+      }
+      XCTFail("Expected model content from invlaid image to error")
+    }
+
+    func testModelContentFromUIImageIsNotEmpty() throws {
+      let coreImage = CIImage(color: CIColor.red)
+        .cropped(to: CGRect(origin: CGPointZero, size: CGSize(width: 16, height: 16)))
+      let image = UIImage(ciImage: coreImage)
+      let modelContent = try image.tryPartsValue()
+      XCTAssert(modelContent.count > 0, "Expected non-empty model content for UIImage: \(image)")
+    }
+  #else
+    func testModelContentFromNSImageIsNotEmpty() throws {
+      let coreImage = CIImage(color: CIColor.red)
+        .cropped(to: CGRect(origin: CGPointZero, size: CGSize(width: 16, height: 16)))
+      let rep = NSCIImageRep(ciImage: coreImage)
+      let image = NSImage(size: rep.size)
+      image.addRepresentation(rep)
+      let modelContent = try image.tryPartsValue()
+      XCTAssert(modelContent.count > 0, "Expected non-empty model content for NSImage: \(image)")
+    }
+
+    func testModelContentFromInvalidNSImageThrows() throws {
+      let image = NSImage()
+      do {
+        _ = try image.tryPartsValue()
+      } catch {
+        guard let imageError = (error as? ImageConversionError) else {
+          XCTFail("Got unexpected error type: \(error)")
+          return
+        }
+        switch imageError {
+        case .invalidUnderlyingImage:
+          // Pass
+          return
+        case _:
+          XCTFail("Expected image conversion error, got \(imageError) instead")
+          return
+        }
+      }
+      XCTFail("Expected model content from invlaid image to error")
+    }
+  #endif
+}

From 1626fb795e0c4dab610744fd6c23f330cfa90c2a Mon Sep 17 00:00:00 2001
From: Morgan Chen <morganchen12@gmail.com>
Date: Tue, 19 Mar 2024 09:10:15 -0700
Subject: [PATCH 016/124] Add local App Check package to sample targets
 (#12576)

---
 .../project.pbxproj                           | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 20f07a5acca..1b4a69cafb0 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -56,6 +56,10 @@
 		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
 		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
 		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
+		8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */; };
+		8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */; };
+		8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */; };
+		8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E302BA8EED80049BFCA /* FirebaseAppCheck */; };
 		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
 /* End PBXBuildFile section */
 
@@ -97,6 +101,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */,
 				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
 				869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */,
 				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
@@ -108,6 +113,7 @@
 			buildActionMask = 2147483647;
 			files = (
 				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
+				8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */,
 				869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -116,6 +122,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */,
 				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
 				869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */,
 				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
@@ -126,6 +133,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */,
 				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
 				869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */,
 				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
@@ -335,6 +343,7 @@
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
 				869200AA2B86BE3100482873 /* FirebaseVertexAI */,
+				8D589E302BA8EED80049BFCA /* FirebaseAppCheck */,
 			);
 			productName = GenerativeAISample;
 			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
@@ -356,6 +365,7 @@
 			packageProductDependencies = (
 				88209C232B0FBE1700F64795 /* MarkdownUI */,
 				869200B02B86BFF500482873 /* FirebaseVertexAI */,
+				8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */,
 			);
 			productName = GenerativeAITextSample;
 			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
@@ -378,6 +388,7 @@
 				88B8A91D2B0FC55100424728 /* MarkdownUI */,
 				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
 				869200AE2B86BFBC00482873 /* FirebaseVertexAI */,
+				8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */,
 			);
 			productName = GenerativeAIMultimodalSample;
 			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
@@ -400,6 +411,7 @@
 				88D9474C2B14F27E008B5580 /* MarkdownUI */,
 				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
 				869200AC2B86BEA100482873 /* FirebaseVertexAI */,
+				8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */,
 			);
 			productName = ChatSample;
 			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
@@ -1043,6 +1055,22 @@
 			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
 			productName = MarkdownUI;
 		};
+		8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
+		8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
+		8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
+		8D589E302BA8EED80049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
 		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = GenerativeAIUIComponents;

From 3b4b436e910bbb30966b3773c82ec4b05939625e Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 19 Mar 2024 20:59:05 +0000
Subject: [PATCH 017/124] Use `generative-ai-swift` tests in Vertex AI (#12585)

---
 .github/workflows/vertexai.yml                |  7 +-
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   | 11 ++-
 .../streaming-success-citations2.txt          |  5 --
 .../unary-failure-api-key.json                |  2 +-
 ...e-finish-reason-recitation-no-content.json | 46 ----------
 ...ary-failure-unsupported-user-location.json | 13 ---
 .../Tests/Unit/GenerativeModelTests.swift     | 90 +++++++++----------
 .../Tests/Unit/PartsRepresentableTests.swift  |  2 +-
 ...leAITests.swift => VertexAIAPITests.swift} | 46 +++++++---
 Package.swift                                 |  9 ++
 .../FirebaseVertexAIUnit.xcscheme             | 77 ++++++++++++++++
 11 files changed, 178 insertions(+), 130 deletions(-)
 delete mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
 delete mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
 delete mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
 rename FirebaseVertexAI/Tests/Unit/{GoogleAITests.swift => VertexAIAPITests.swift} (84%)
 create mode 100644 scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
index 63cfd774e43..4f0880de320 100644
--- a/.github/workflows/vertexai.yml
+++ b/.github/workflows/vertexai.yml
@@ -31,7 +31,6 @@ jobs:
     - name: Xcode
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild
-      run: xcodebuild -list
-    # TODO: Add unit tests and switch from `spmbuildonly` to `spm`.
-    - name: Build
-      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAI ${{ matrix.target }} spmbuildonly
+      run: scripts/setup_spm_tests.sh
+    - name: Build and run tests
+      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAIUnit ${{ matrix.target }} spm
diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 668e3704122..046cce0e8bd 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -13,9 +13,10 @@
 // limitations under the License.
 
 import Foundation
-@testable import GoogleGenerativeAI
 import XCTest
 
+@testable import FirebaseVertexAI
+
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
 final class ChatTests: XCTestCase {
   var urlSession: URLSession!
@@ -46,7 +47,13 @@ final class ChatTests: XCTestCase {
       return (response, fileURL.lines)
     }
 
-    let model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+    let model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      urlSession: urlSession
+    )
     let chat = Chat(model: model, history: [])
     let input = "Test input"
     let stream = chat.sendMessageStream(input)
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
deleted file mode 100644
index 665e993ad75..00000000000
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
+++ /dev/null
@@ -1,5 +0,0 @@
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": "In the context of the science fiction comedy novel \"The Hitchhiker's Guide to the Galaxy\" by Douglas Adams, the answer to the \"Ultimate Question of"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.055720285,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.062674366},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03904829,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03339982},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12220858,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0540987},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05781161,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03149938}]}]}
-
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Life, the Universe, and Everything\" is given as 42. However, in real life, there is no universally accepted meaning of life. The meaning"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.05910154,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.038321976},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.049589027,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.019271139},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03986249},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.098946586,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04177388}],"citationMetadata": {"citations": [{"startIndex": 52,"endIndex": 181,"uri": "https://imcsteakhouse.com.au/faqs"}]}}]}
-
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " of life is a philosophical question that has been pondered by humans for centuries, and there is no single answer that is widely agreed upon."}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.044764314,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.028870905},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04240383,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.012576347},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.10650458,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034880884},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.11085559,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03191915}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 91,"totalTokenCount": 100}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
index ecf6f6b53fa..d3b5677366e 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -9,7 +9,7 @@
         "reason": "API_KEY_INVALID",
         "domain": "googleapis.com",
         "metadata": {
-          "service": "generativelanguage.googleapis.com"
+          "service": "staging-firebaseml.sandbox.googleapis.com"
         }
       },
       {
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
deleted file mode 100644
index 0d99cd71d51..00000000000
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
+++ /dev/null
@@ -1,46 +0,0 @@
-{
-  "candidates": [
-    {
-      "finishReason": "RECITATION",
-      "index": 0,
-      "safetyRatings": [
-        {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
-        },
-        {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "HIGH"
-        },
-        {
-          "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
-        },
-        {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
-        }
-      ]
-    }
-  ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
-  }
-}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
deleted file mode 100644
index c4c2ace4e20..00000000000
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
+++ /dev/null
@@ -1,13 +0,0 @@
-{
-  "error": {
-    "code": 400,
-    "message": "User location is not supported for the API use.",
-    "status": "FAILED_PRECONDITION",
-    "details": [
-      {
-        "@type": "type.googleapis.com/google.rpc.DebugInfo",
-        "detail": "[ORIGINAL ERROR] generic::failed_precondition: User location is not supported for the API use. [google.rpc.error_details_ext] { message: \"User location is not supported for the API use.\" }"
-      }
-    ]
-  }
-}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index b835a9ea05f..ac1980d93d4 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -12,9 +12,10 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-@testable import GoogleGenerativeAI
 import XCTest
 
+@testable import FirebaseVertexAI
+
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
 final class GenerativeModelTests: XCTestCase {
   let testPrompt = "What sorts of questions can I ask you?"
@@ -32,7 +33,13 @@ final class GenerativeModelTests: XCTestCase {
     let configuration = URLSessionConfiguration.default
     configuration.protocolClasses = [MockURLProtocol.self]
     urlSession = try XCTUnwrap(URLSession(configuration: configuration))
-    model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      urlSession: urlSession
+    )
   }
 
   override func tearDown() {
@@ -163,6 +170,8 @@ final class GenerativeModelTests: XCTestCase {
       // Model name is prefixed with "models/".
       name: "models/test-model",
       apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -181,10 +190,13 @@ final class GenerativeModelTests: XCTestCase {
     do {
       _ = try await model.generateContent(testPrompt)
       XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
-    } catch let GenerateContentError.invalidAPIKey(message) {
-      XCTAssertEqual(message, "API key not valid. Please pass a valid API key.")
+    } catch let GenerateContentError.internalError(error as RPCError) {
+      XCTAssertEqual(error.httpResponseCode, 400)
+      XCTAssertEqual(error.status, .invalidArgument)
+      XCTAssertEqual(error.message, "API key not valid. Please pass a valid API key.")
+      return
     } catch {
-      XCTFail("Should throw GenerateContentError.invalidAPIKey; error thrown: \(error)")
+      XCTFail("Should throw GenerateContentError.internalError(RPCError); error thrown: \(error)")
     }
   }
 
@@ -342,24 +354,6 @@ final class GenerativeModelTests: XCTestCase {
     }
   }
 
-  func testGenerateContent_failure_unsupportedUserLocation() async throws {
-    MockURLProtocol
-      .requestHandler = try httpRequestHandler(
-        forResource: "unary-failure-unsupported-user-location",
-        withExtension: "json",
-        statusCode: 400
-      )
-
-    do {
-      _ = try await model.generateContent(testPrompt)
-      XCTFail("Should throw GenerateContentError.unsupportedUserLocation; no error thrown.")
-    } catch GenerateContentError.unsupportedUserLocation {
-      return
-    }
-
-    XCTFail("Expected an unsupported user location error.")
-  }
-
   func testGenerateContent_failure_nonHTTPResponse() async throws {
     MockURLProtocol.requestHandler = try nonHTTPRequestHandler()
 
@@ -468,6 +462,7 @@ final class GenerativeModelTests: XCTestCase {
       name: "my-model",
       apiKey: "API_KEY",
       requestOptions: requestOptions,
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -490,8 +485,10 @@ final class GenerativeModelTests: XCTestCase {
       for try await _ in stream {
         XCTFail("No content is there, this shouldn't happen.")
       }
-    } catch GenerateContentError.invalidAPIKey {
-      // invalidAPIKey error is as expected, nothing else to check.
+    } catch let GenerateContentError.internalError(error as RPCError) {
+      XCTAssertEqual(error.httpResponseCode, 400)
+      XCTAssertEqual(error.status, .invalidArgument)
+      XCTAssertEqual(error.message, "API key not valid. Please pass a valid API key.")
       return
     }
 
@@ -747,26 +744,6 @@ final class GenerativeModelTests: XCTestCase {
     XCTFail("Expected an internal decoding error.")
   }
 
-  func testGenerateContentStream_failure_unsupportedUserLocation() async throws {
-    MockURLProtocol
-      .requestHandler = try httpRequestHandler(
-        forResource: "unary-failure-unsupported-user-location",
-        withExtension: "json",
-        statusCode: 400
-      )
-
-    let stream = model.generateContentStream(testPrompt)
-    do {
-      for try await content in stream {
-        XCTFail("Unexpected content in stream: \(content)")
-      }
-    } catch GenerateContentError.unsupportedUserLocation {
-      return
-    }
-
-    XCTFail("Expected an unsupported user location error.")
-  }
-
   func testGenerateContentStream_requestOptions_customTimeout() async throws {
     let expectedTimeout = 150.0
     MockURLProtocol
@@ -780,6 +757,7 @@ final class GenerativeModelTests: XCTestCase {
       name: "my-model",
       apiKey: "API_KEY",
       requestOptions: requestOptions,
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -837,6 +815,7 @@ final class GenerativeModelTests: XCTestCase {
       name: "my-model",
       apiKey: "API_KEY",
       requestOptions: requestOptions,
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -851,7 +830,12 @@ final class GenerativeModelTests: XCTestCase {
     let modelName = "my-model"
     let modelResourceName = "models/\(modelName)"
 
-    model = GenerativeModel(name: modelName, apiKey: "API_KEY")
+    model = GenerativeModel(
+      name: modelName,
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil
+    )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
   }
@@ -859,7 +843,12 @@ final class GenerativeModelTests: XCTestCase {
   func testModelResourceName_modelsPrefix() async throws {
     let modelResourceName = "models/my-model"
 
-    model = GenerativeModel(name: modelResourceName, apiKey: "API_KEY")
+    model = GenerativeModel(
+      name: modelResourceName,
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil
+    )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
   }
@@ -867,7 +856,12 @@ final class GenerativeModelTests: XCTestCase {
   func testModelResourceName_tunedModelsPrefix() async throws {
     let tunedModelResourceName = "tunedModels/my-model"
 
-    model = GenerativeModel(name: tunedModelResourceName, apiKey: "API_KEY")
+    model = GenerativeModel(
+      name: tunedModelResourceName,
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil
+    )
 
     XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
   }
diff --git a/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
index da2f35f727f..4243c2441be 100644
--- a/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
@@ -14,7 +14,7 @@
 
 import CoreGraphics
 import CoreImage
-import GoogleGenerativeAI
+import FirebaseVertexAI
 import XCTest
 #if canImport(UIKit)
   import UIKit
diff --git a/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
similarity index 84%
rename from FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
rename to FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index cbc92527a1c..4676b2e34d9 100644
--- a/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -12,7 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import GoogleGenerativeAI
+import FirebaseCore
+import FirebaseVertexAI
 import XCTest
 #if canImport(AppKit)
   import AppKit // For NSImage extensions.
@@ -21,8 +22,9 @@ import XCTest
 #endif
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-final class GoogleGenerativeAITests: XCTestCase {
+final class VertexAIAPITests: XCTestCase {
   func codeSamples() async throws {
+    let app = FirebaseApp.app()
     let config = GenerationConfig(temperature: 0.2,
                                   topP: 0.1,
                                   topK: 16,
@@ -32,16 +34,40 @@ final class GoogleGenerativeAITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Permutations without optional arguments.
-    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY")
-    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", safetySettings: filters)
-    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", generationConfig: config)
 
-    // All arguments passed.
-    let genAI = GenerativeModel(name: "gemini-1.0-pro",
-                                apiKey: "API_KEY",
-                                generationConfig: config, // Optional
-                                safetySettings: filters // Optional
+    // TODO: Change `genAI` to `_` when safetySettings and generationConfig are added to public API.
+    let genAI = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    let _ = VertexAI.generativeModel(
+      app: app!,
+      modelName: "gemini-1.0-pro",
+      location: "us-central1"
     )
+
+    // TODO: Add safetySettings to public API.
+    // TODO: Add permutation with `app` specified.
+    //  let _ = VertexAI.generativeModel(
+    //    modelName: "gemini-1.0-pro",
+    //    location: "us-central1",
+    //    safetySettings: filters
+    //  )
+    // TODO: Add generationConfig to public API.
+    // TODO: Add permutation with `app` specified.
+    // let _ = VertexAI.generativeModel(
+    //   modelName: "gemini-1.0-pro",
+    //   location: "us-central1",
+    //   generationConfig: config
+    // )
+
+    // All arguments passed.
+    // TODO: Add safetySettings and generationConfig to public API.
+    // TODO: Add permutation with `app` specified.
+    // let genAI = VertexAI.generativeModel(
+    //   modelName: "gemini-1.0-pro",
+    //   location: "us-central1",
+    //   generationConfig: config, // Optional
+    //   safetySettings: filters // Optional
+    // )
+
     // Full Typed Usage
     let pngData = Data() // ....
     let contents = [ModelContent(role: "user",
diff --git a/Package.swift b/Package.swift
index 71a2436591a..d3d99222cf7 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1363,6 +1363,15 @@ let package = Package(
       ],
       path: "FirebaseVertexAI/Sources"
     ),
+    .testTarget(
+      name: "FirebaseVertexAIUnit",
+      dependencies: ["FirebaseVertexAI"],
+      path: "FirebaseVertexAI/Tests/Unit",
+      resources: [
+        .process("CountTokenResponses"),
+        .process("GenerateContentResponses"),
+      ]
+    ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,
   cxxLanguageStandard: CXXLanguageStandard.gnucxx14
diff --git a/scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme b/scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme
new file mode 100644
index 00000000000..cb4b5adae36
--- /dev/null
+++ b/scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme
@@ -0,0 +1,77 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Scheme
+   LastUpgradeVersion = "1200"
+   version = "1.3">
+   <BuildAction
+      parallelizeBuildables = "YES"
+      buildImplicitDependencies = "YES">
+      <BuildActionEntries>
+         <BuildActionEntry
+            buildForTesting = "YES"
+            buildForRunning = "YES"
+            buildForProfiling = "NO"
+            buildForArchiving = "NO"
+            buildForAnalyzing = "YES">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "FirebaseVertexAIUnit"
+               BuildableName = "FirebaseVertexAIUnit"
+               BlueprintName = "FirebaseVertexAIUnit"
+               ReferencedContainer = "container:">
+            </BuildableReference>
+         </BuildActionEntry>
+      </BuildActionEntries>
+   </BuildAction>
+   <TestAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      shouldUseLaunchSchemeArgsEnv = "YES">
+      <Testables>
+         <TestableReference
+            skipped = "NO">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "FirebaseVertexAIUnit"
+               BuildableName = "FirebaseVertexAIUnit"
+               BlueprintName = "FirebaseVertexAIUnit"
+               ReferencedContainer = "container:">
+            </BuildableReference>
+         </TestableReference>
+      </Testables>
+   </TestAction>
+   <LaunchAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      launchStyle = "0"
+      useCustomWorkingDirectory = "NO"
+      ignoresPersistentStateOnLaunch = "NO"
+      debugDocumentVersioning = "YES"
+      debugServiceExtension = "internal"
+      allowLocationSimulation = "YES">
+   </LaunchAction>
+   <ProfileAction
+      buildConfiguration = "Release"
+      shouldUseLaunchSchemeArgsEnv = "YES"
+      savedToolIdentifier = ""
+      useCustomWorkingDirectory = "NO"
+      debugDocumentVersioning = "YES">
+      <MacroExpansion>
+         <BuildableReference
+            BuildableIdentifier = "primary"
+            BlueprintIdentifier = "FirebaseVertexAIUnit"
+            BuildableName = "FirebaseVertexAIUnit"
+            BlueprintName = "FirebaseVertexAIUnit"
+            ReferencedContainer = "container:">
+         </BuildableReference>
+      </MacroExpansion>
+   </ProfileAction>
+   <AnalyzeAction
+      buildConfiguration = "Debug">
+   </AnalyzeAction>
+   <ArchiveAction
+      buildConfiguration = "Release"
+      revealArchiveInOrganizer = "YES">
+   </ArchiveAction>
+</Scheme>

From f8d4278931f3440a8b3b75a75496ba098c1cf0c9 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 20 Mar 2024 14:11:28 +0000
Subject: [PATCH 018/124] Add App Check tests in `GenerativeModelTests`
 (#12590)

---
 .../Tests/Unit/GenerativeModelTests.swift     | 131 +++++++++++++++++-
 1 file changed, 129 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index ac1980d93d4..13c625036cc 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -12,6 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
 import XCTest
 
 @testable import FirebaseVertexAI
@@ -178,6 +179,43 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_appCheck_validToken() async throws {
+    let appCheckToken = "test-valid-token"
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(token: appCheckToken),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        appCheckToken: appCheckToken
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_appCheck_tokenRefreshError() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        appCheckToken: AppCheckInteropFake.placeholderTokenValue
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
   func testGenerateContent_failure_invalidAPIKey() async throws {
     let expectedStatusCode = 400
     MockURLProtocol
@@ -654,6 +692,45 @@ final class GenerativeModelTests: XCTestCase {
       .contains(where: { $0.startIndex == 899 && $0.endIndex == 1026 && !$0.uri.isEmpty }))
   }
 
+  func testGenerateContentStream_appCheck_validToken() async throws {
+    let appCheckToken = "test-valid-token"
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(token: appCheckToken),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt",
+        appCheckToken: appCheckToken
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    for try await _ in stream {}
+  }
+
+  func testGenerateContentStream_appCheck_tokenRefreshError() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt",
+        appCheckToken: AppCheckInteropFake.placeholderTokenValue
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    for try await _ in stream {}
+  }
+
   func testGenerateContentStream_errorMidStream() async throws {
     MockURLProtocol.requestHandler = try httpRequestHandler(
       forResource: "streaming-failure-error-mid-stream",
@@ -887,8 +964,8 @@ final class GenerativeModelTests: XCTestCase {
   private func httpRequestHandler(forResource name: String,
                                   withExtension ext: String,
                                   statusCode: Int = 200,
-                                  timeout: TimeInterval = URLRequest
-                                    .defaultTimeoutInterval()) throws -> ((URLRequest) throws -> (
+                                  timeout: TimeInterval = URLRequest.defaultTimeoutInterval(),
+                                  appCheckToken: String? = nil) throws -> ((URLRequest) throws -> (
     URLResponse,
     AsyncLineSequence<URL.AsyncBytes>?
   )) {
@@ -897,6 +974,7 @@ final class GenerativeModelTests: XCTestCase {
       let requestURL = try XCTUnwrap(request.url)
       XCTAssertEqual(requestURL.path.occurrenceCount(of: "models/"), 1)
       XCTAssertEqual(request.timeoutInterval, timeout)
+      XCTAssertEqual(request.value(forHTTPHeaderField: "X-Firebase-AppCheck"), appCheckToken)
       let response = try XCTUnwrap(HTTPURLResponse(
         url: requestURL,
         statusCode: statusCode,
@@ -922,3 +1000,52 @@ private extension URLRequest {
     return URLRequest(url: placeholderURL).timeoutInterval
   }
 }
+
+class AppCheckInteropFake: NSObject, AppCheckInterop {
+  /// The placeholder token value returned when an error occurs
+  static let placeholderTokenValue = "placeholder-token"
+
+  var token: String
+  var error: Error?
+
+  private init(token: String, error: Error?) {
+    self.token = token
+    self.error = error
+  }
+
+  convenience init(token: String) {
+    self.init(token: token, error: nil)
+  }
+
+  convenience init(error: Error) {
+    self.init(token: AppCheckInteropFake.placeholderTokenValue, error: error)
+  }
+
+  func getToken(forcingRefresh: Bool) async -> any FIRAppCheckTokenResultInterop {
+    return AppCheckTokenResultInteropFake(token: token, error: error)
+  }
+
+  func tokenDidChangeNotificationName() -> String {
+    fatalError("\(#function) not implemented.")
+  }
+
+  func notificationTokenKey() -> String {
+    fatalError("\(#function) not implemented.")
+  }
+
+  func notificationAppNameKey() -> String {
+    fatalError("\(#function) not implemented.")
+  }
+
+  private class AppCheckTokenResultInteropFake: NSObject, FIRAppCheckTokenResultInterop {
+    var token: String
+    var error: Error?
+
+    init(token: String, error: Error?) {
+      self.token = token
+      self.error = error
+    }
+  }
+}
+
+struct AppCheckErrorFake: Error {}

From 8d46fe36ba730a283e755388b77d3fd64775e87a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 21 Mar 2024 20:56:59 +0000
Subject: [PATCH 019/124] Make `generativeModel` an instance method of
 `VertexAI` (#12599)

---
 .../ViewModels/ConversationViewModel.swift    |  5 +-
 .../ViewModels/PhotoReasoningViewModel.swift  |  5 +-
 .../ViewModels/SummarizeViewModel.swift       |  5 +-
 FirebaseVertexAI/Sources/VertexAI.swift       | 81 +++++++++++--------
 .../Sources/VertexAIComponent.swift           | 10 +--
 .../Tests/Unit/VertexAIAPITests.swift         | 49 +++++------
 6 files changed, 85 insertions(+), 70 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 465789be387..05cbe11250f 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,7 +36,10 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1"
+    )
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index dc41e00444a..2f2ed88d4a1 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,10 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro-vision", location: "us-central1")
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro-vision",
+      location: "us-central1"
+    )
   }
 
   func reason() async {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index fb5b349ac82..0e3073d6da2 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,7 +32,10 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1"
+    )
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 0f79e1be99c..64e76d3b4f5 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -20,70 +20,81 @@ import Foundation
 @_implementationOnly import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-@objc(FIRVertexAI)
-open class VertexAI: NSObject {
+public class VertexAI: NSObject {
   // MARK: - Public APIs
 
-  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  /// The default `VertexAI` instance.
   ///
-  /// This instance is configured with the default `FirebaseApp`.
-  ///
-  /// TODO: Add RequestOptions to public API.
-  public static func generativeModel(modelName: String, location: String) -> GenerativeModel {
+  /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
+  public static func vertexAI() -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
-    return generativeModel(app: app, modelName: modelName, location: location)
+
+    return vertexAI(app: app)
   }
 
-  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
-  /// TODO: Add RequestOptions to public API.
-  public static func generativeModel(app: FirebaseApp, modelName: String,
-                                     location: String) -> GenerativeModel {
+  /// - Parameter app: The custom `FirebaseApp` used for initialization.
+  /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
+  public static func vertexAI(app: FirebaseApp) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
-    let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
-    let vertexAI = provider.vertexAI(location: location, modelResourceName: modelResourceName)
 
-    return vertexAI.model
+    return provider.vertexAI()
   }
 
-  // MARK: - Private
-
-  /// The `FirebaseApp` associated with this `VertexAI` instance.
-  private let app: FirebaseApp
-
-  private let appCheck: AppCheckInterop?
-
-  private let location: String
-
-  private let modelResouceName: String
+  /// Initializes a generative model with the given parameters.
+  ///
+  /// - Parameters:
+  ///   - modelName: The name of the model to use, e.g., `"gemini-1.0-pro"`; see
+  ///     [Gemini
+  ///     models](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#gemini-models)
+  ///     for a list of supported model names.
+  ///   - location: The location identifier, e.g., `us-central1`; see
+  ///     [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported locations.
+  ///   - generationConfig: The content generation parameters your model should use.
+  ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - requestOptions: Configuration parameters for sending requests to the backend.
+  public func generativeModel(modelName: String, location: String,
+                              generationConfig: GenerationConfig? = nil,
+                              safetySettings: [SafetySetting]? = nil,
+                              requestOptions: RequestOptions = RequestOptions())
+    -> GenerativeModel {
+    let modelResourceName = modelResourceName(modelName: modelName, location: location)
 
-  lazy var model: GenerativeModel = {
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
     }
+
     return GenerativeModel(
-      name: modelResouceName,
+      name: modelResourceName,
       apiKey: apiKey,
-      // TODO: Add RequestOptions to public API.
-      requestOptions: RequestOptions(),
+      generationConfig: generationConfig,
+      safetySettings: safetySettings,
+      requestOptions: requestOptions,
       appCheck: appCheck
     )
-  }()
+  }
+
+  // MARK: - Private
+
+  /// The `FirebaseApp` associated with this `VertexAI` instance.
+  private let app: FirebaseApp
+
+  private let appCheck: AppCheckInterop?
 
-  init(app: FirebaseApp, location: String, modelResourceName: String) {
+  init(app: FirebaseApp) {
     self.app = app
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
-    self.location = location
-    modelResouceName = modelResourceName
   }
 
-  private static func modelResourceName(app: FirebaseApp, modelName: String,
-                                        location: String) -> String {
+  private func modelResourceName(modelName: String, location: String) -> String {
     if modelName.contains("/") {
       return modelName
     }
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a8d6c177c74..1378f812626 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI(location: String, modelResourceName: String) -> VertexAI
+  @objc func vertexAI() -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -64,17 +64,17 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   // MARK: - VertexAIProvider conformance
 
-  func vertexAI(location: String, modelResourceName: String) -> VertexAI {
+  func vertexAI() -> VertexAI {
     os_unfair_lock_lock(&instancesLock)
 
     // Unlock before the function returns.
     defer { os_unfair_lock_unlock(&instancesLock) }
 
-    if let instance = instances[modelResourceName] {
+    if let instance = instances[app.name] {
       return instance
     }
-    let newInstance = VertexAI(app: app, location: location, modelResourceName: modelResourceName)
-    instances[modelResourceName] = newInstance
+    let newInstance = VertexAI(app: app)
+    instances[app.name] = newInstance
     return newInstance
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 4676b2e34d9..a1fd27ab4c5 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -33,40 +33,35 @@ final class VertexAIAPITests: XCTestCase {
                                   stopSequences: ["..."])
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
+    // Instantiate Vertex AI SDK - Default App
+    let vertexAI = VertexAI.vertexAI()
+
+    // Instantiate Vertex AI SDK - Custom App
+    let _ = VertexAI.vertexAI(app: app!)
+
     // Permutations without optional arguments.
 
-    // TODO: Change `genAI` to `_` when safetySettings and generationConfig are added to public API.
-    let genAI = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
-    let _ = VertexAI.generativeModel(
-      app: app!,
+    let _ = vertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+
+    let _ = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1"
+      location: "us-central1",
+      safetySettings: filters
     )
 
-    // TODO: Add safetySettings to public API.
-    // TODO: Add permutation with `app` specified.
-    //  let _ = VertexAI.generativeModel(
-    //    modelName: "gemini-1.0-pro",
-    //    location: "us-central1",
-    //    safetySettings: filters
-    //  )
-    // TODO: Add generationConfig to public API.
-    // TODO: Add permutation with `app` specified.
-    // let _ = VertexAI.generativeModel(
-    //   modelName: "gemini-1.0-pro",
-    //   location: "us-central1",
-    //   generationConfig: config
-    // )
+    let _ = vertexAI.generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1",
+      generationConfig: config
+    )
 
     // All arguments passed.
-    // TODO: Add safetySettings and generationConfig to public API.
-    // TODO: Add permutation with `app` specified.
-    // let genAI = VertexAI.generativeModel(
-    //   modelName: "gemini-1.0-pro",
-    //   location: "us-central1",
-    //   generationConfig: config, // Optional
-    //   safetySettings: filters // Optional
-    // )
+    let genAI = vertexAI.generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1",
+      generationConfig: config, // Optional
+      safetySettings: filters // Optional
+    )
 
     // Full Typed Usage
     let pngData = Data() // ....

From 38ddd25dd9ce1b897c40dae17fbf47501d570b6f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 22 Mar 2024 21:06:39 +0000
Subject: [PATCH 020/124] Update test JSON payloads with Vertex AI output
 (#12615)

---
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |  2 +-
 .../streaming-success-basic-reply-long.txt    | 20 +++----
 .../streaming-success-basic-reply-short.txt   |  3 +-
 .../unary-success-basic-reply-long.json       | 56 +++++++++----------
 .../unary-success-basic-reply-short.json      | 54 +++++++++---------
 .../Tests/Unit/GenerativeModelTests.swift     | 26 +++++----
 6 files changed, 75 insertions(+), 86 deletions(-)

diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 046cce0e8bd..5a7868fbe7a 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -66,7 +66,7 @@ final class ChatTests: XCTestCase {
     XCTAssertEqual(chat.history.count, 2)
     XCTAssertEqual(chat.history[0].parts[0].text, input)
 
-    let finalText = "1 2 3 4 5 6 7 8 9 10"
+    let finalText = "1 2 3 4 5 6 7 8"
     let assembledExpectation = ModelContent(role: "model", parts: finalText)
     XCTAssertEqual(chat.history[0].parts[0].text, input)
     XCTAssertEqual(chat.history[1], assembledExpectation)
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
index bca95140490..218cae0b985 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
@@ -1,19 +1,15 @@
-data: {"candidates": [{"content": {"parts": [{"text": "1 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "1 "}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "2 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "2 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.0394904,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04468087},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.034553625,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03890198},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.09401018,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.025809621},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.036562506,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.047691282}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "3 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "3 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.03507868,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045183755},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.027742893,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.043528143},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.026105914},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.06681233,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "4 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "4 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.037750278,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05089372},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.040087357,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05888469},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.071202725,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034100424},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.07613248,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.051749535}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "5 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "5 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.04672496,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059210256},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04977345,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05623635},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.083890386,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.08359067,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05975658}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "6 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "6 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07779744,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06052939},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.041930523,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.056756895},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05350215},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.09203286,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048676573}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "7 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "7 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.071202725,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05291181},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.031439852,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04509957},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.11417085,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04922211},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.09451043,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.052716404}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "8 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "9 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "10"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "8"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06221698,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045777276},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03085051,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04560694},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.0992954,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.040769264},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.100701615,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.061424047}]}],"usageMetadata": {"promptTokenCount": 6,"candidatesTokenCount": 326,"totalTokenCount": 332}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
index a7f5476954e..78f569ddecf 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
@@ -1,2 +1 @@
-data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
-
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Mountain View, California"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.02854415,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.052424565},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.24926445,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0996453},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.087096825,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.043123372},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.14402841,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.086169556}]}],"usageMetadata": {"promptTokenCount": 6,"candidatesTokenCount": 4,"totalTokenCount": 10}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
index 59b84de92fe..325d172b894 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
@@ -2,53 +2,49 @@
   "candidates": [
     {
       "content": {
+        "role": "model",
         "parts": [
           {
-            "text": "You can ask me a wide range of questions on various topics. Here are some examples:\n\n1. **General Knowledge:**\n   - What is the capital of France?\n   - Who painted the Mona Lisa?\n   - What is the largest ocean in the world?\n\n2. **Science and Technology:**\n   - How does a computer work?\n   - What is the difference between a virus and a bacteria?\n   - What are the latest advancements in artificial intelligence?\n\n3. **History and Culture:**\n   - Who was the first president of the United States?\n   - What is the significance of the Great Wall of China?\n   - What are some of the most famous works of Shakespeare?\n\n4. **Current Events:**\n   - What is the latest news about the COVID-19 pandemic?\n   - Who is the current president of Ukraine?\n   - What are the major issues being discussed in the upcoming election?\n\n5. **Personal Questions:**\n   - What are your hobbies?\n   - What is your favorite book or movie?\n   - What are your thoughts on the future of technology?\n\n6. **Fun and Games:**\n   - Can you tell me a joke?\n   - What is the answer to this riddle: \"I have keys but no locks. I have space but no room. You can enter, but can't go outside.\" (Answer: a keyboard)\n   - Let's play a game of 20 questions.\n\n7. **Hypothetical Questions:**\n   - What would you do if you won the lottery?\n   - What would happen if time travel were possible?\n   - What is the meaning of life?\n\n8. **Philosophical Questions:**\n   - What is the nature of reality?\n   - Does free will exist?\n   - What is the difference between right and wrong?\n\n9. **Creative Questions:**\n   - Write a poem about a sunset.\n   - Design a logo for a new company.\n   - Compose a song about your favorite season.\n\n10. **Technical Questions:**\n    - How can I improve the performance of my computer?\n    - What is the best way to troubleshoot a network issue?\n    - How do I create a website using HTML and CSS?\n\nRemember, I am still under development and may not be able to answer all questions perfectly. However, I will do my best to provide you with accurate and informative responses."
+            "text": "You can ask me a wide range of questions, including:\n\n* **General knowledge questions:** What is the capital of France? Who painted the Mona Lisa?\n* **Science questions:** What is the chemical formula for water? How does photosynthesis work?\n* **History questions:** When did World War II start? Who was the first president of the United States?\n* **Math questions:** What is the square root of 16? How do you solve for x in the equation x + 5 = 10?\n* **Current events questions:** What is happening in Ukraine? Who is the current president of the United States?\n* **Personal questions:** What are your hobbies? What are your favorite books?\n* **Hypothetical questions:** What would happen if all the ice caps melted? What if humans could live on Mars?\n* **Questions about me:** What is your name? How old are you? What are your capabilities?\n\n**Tips for asking questions:**\n\n* Be specific and clear in your questions.\n* Use correct grammar and spelling.\n* Try to ask open-ended questions that allow for multiple answers.\n* Be patient and wait for my response. I am still under development, so I may take a few seconds to process your question.\n\nPlease note that I am still under development and may not be able to answer all questions accurately. However, I will do my best to provide you with the most relevant information I have available."
           }
-        ],
-        "role": "model"
+        ]
       },
       "finishReason": "STOP",
-      "index": 0,
       "safetyRatings": [
         {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.047869004,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.050705366
         },
         {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.052134257,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.036288295
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.08464396,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.033907957
         },
         {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.06290424,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.050611436
         }
       ]
     }
   ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
+  "usageMetadata": {
+    "promptTokenCount": 6,
+    "candidatesTokenCount": 303,
+    "totalTokenCount": 309
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
index 40a9a6da58e..3e3ddb80fd6 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
@@ -2,53 +2,49 @@
   "candidates": [
     {
       "content": {
+        "role": "model",
         "parts": [
           {
             "text": "Mountain View, California, United States"
           }
-        ],
-        "role": "model"
+        ]
       },
       "finishReason": "STOP",
-      "index": 0,
       "safetyRatings": [
         {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.029035643,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.05613278
         },
         {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.2641685,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.082253955
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.087252244,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.04509957
         },
         {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.1431877,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.11027937
         }
       ]
     }
   ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
+  "usageMetadata": {
+    "promptTokenCount": 6,
+    "candidatesTokenCount": 7,
+    "totalTokenCount": 13
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 13c625036cc..c249613f9e9 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -25,7 +25,7 @@ final class GenerativeModelTests: XCTestCase {
     .init(category: .hateSpeech, probability: .negligible),
     .init(category: .harassment, probability: .negligible),
     .init(category: .dangerousContent, probability: .negligible),
-  ]
+  ].sorted()
 
   var urlSession: URLSession!
   var model: GenerativeModel!
@@ -62,15 +62,12 @@ final class GenerativeModelTests: XCTestCase {
     let candidate = try XCTUnwrap(response.candidates.first)
     let finishReason = try XCTUnwrap(candidate.finishReason)
     XCTAssertEqual(finishReason, .stop)
-    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.safetyRatings.sorted(), safetyRatingsNegligible)
     XCTAssertEqual(candidate.content.parts.count, 1)
     let part = try XCTUnwrap(candidate.content.parts.first)
     let partText = try XCTUnwrap(part.text)
     XCTAssertTrue(partText.hasPrefix("You can ask me a wide range of questions"))
     XCTAssertEqual(response.text, partText)
-    let promptFeedback = try XCTUnwrap(response.promptFeedback)
-    XCTAssertNil(promptFeedback.blockReason)
-    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
   }
 
   func testGenerateContent_success_basicReplyShort() async throws {
@@ -86,14 +83,11 @@ final class GenerativeModelTests: XCTestCase {
     let candidate = try XCTUnwrap(response.candidates.first)
     let finishReason = try XCTUnwrap(candidate.finishReason)
     XCTAssertEqual(finishReason, .stop)
-    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.safetyRatings.sorted(), safetyRatingsNegligible)
     XCTAssertEqual(candidate.content.parts.count, 1)
     let part = try XCTUnwrap(candidate.content.parts.first)
     XCTAssertEqual(part.text, "Mountain View, California, United States")
     XCTAssertEqual(response.text, part.text)
-    let promptFeedback = try XCTUnwrap(response.promptFeedback)
-    XCTAssertNil(promptFeedback.blockReason)
-    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
   }
 
   func testGenerateContent_success_citations() async throws {
@@ -131,7 +125,7 @@ final class GenerativeModelTests: XCTestCase {
     let candidate = try XCTUnwrap(response.candidates.first)
     let finishReason = try XCTUnwrap(candidate.finishReason)
     XCTAssertEqual(finishReason, .stop)
-    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.safetyRatings.sorted(), safetyRatingsNegligible)
     XCTAssertEqual(candidate.content.parts.count, 1)
     let part = try XCTUnwrap(candidate.content.parts.first)
     let partText = try XCTUnwrap(part.text)
@@ -139,7 +133,7 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(response.text, part.text)
     let promptFeedback = try XCTUnwrap(response.promptFeedback)
     XCTAssertNil(promptFeedback.blockReason)
-    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(promptFeedback.safetyRatings.sorted(), safetyRatingsNegligible)
   }
 
   func testGenerateContent_success_unknownEnum_safetyRatings() async throws {
@@ -627,7 +621,7 @@ final class GenerativeModelTests: XCTestCase {
       responses += 1
     }
 
-    XCTAssertEqual(responses, 10)
+    XCTAssertEqual(responses, 8)
   }
 
   func testGenerateContentStream_successBasicReplyShort() async throws {
@@ -1049,3 +1043,11 @@ class AppCheckInteropFake: NSObject, AppCheckInterop {
 }
 
 struct AppCheckErrorFake: Error {}
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+extension SafetyRating: Comparable {
+  public static func < (lhs: FirebaseVertexAI.SafetyRating,
+                        rhs: FirebaseVertexAI.SafetyRating) -> Bool {
+    return lhs.category.rawValue < rhs.category.rawValue
+  }
+}

From 237485f6745a60781851afe6c3eaa267275c603f Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Mon, 25 Mar 2024 08:21:58 -0700
Subject: [PATCH 021/124] Build Sample in CI (#12616)

---
 .github/workflows/vertexai.yml                | 42 ++++++++++++++++---
 .../Tests/Unit/GenerativeModelTests.swift     |  1 +
 scripts/build.sh                              |  8 ++++
 3 files changed, 46 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
index 4f0880de320..ebf58529f54 100644
--- a/.github/workflows/vertexai.yml
+++ b/.github/workflows/vertexai.yml
@@ -3,9 +3,9 @@ name: vertexai
 on:
   pull_request:
     paths:
-        - 'FirebaseVertexAI**'
-        - '.github/workflows/vertexai.yml'
-        - 'Gemfile*'
+    - 'FirebaseVertexAI**'
+    - '.github/workflows/vertexai.yml'
+    - 'Gemfile*'
   schedule:
     # Run every day at 11pm (PST) - cron uses UTC times
     - cron:  '0 7 * * *'
@@ -32,5 +32,37 @@ jobs:
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild
       run: scripts/setup_spm_tests.sh
-    - name: Build and run tests
-      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAIUnit ${{ matrix.target }} spm
+    - uses: nick-fields/retry@v3
+      with:
+        timeout_minutes: 120
+        max_attempts: 3
+        retry_on: error
+        retry_wait_seconds: 120
+        command: scripts/build.sh FirebaseVertexAIUnit ${{ matrix.target }} spm
+
+  sample:
+    strategy:
+      matrix:
+        # Test build with debug and release configs (whether or not DEBUG is set and optimization level)
+        build: [build]
+        include:
+          - os: macos-13
+            xcode: Xcode_15.0.1
+          - os: macos-14
+            xcode: Xcode_15.2
+    runs-on: ${{ matrix.os }}
+    steps:
+    - uses: actions/checkout@v4
+    - name: Xcode
+      run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
+    - name: Initialize xcodebuild
+      run: xcodebuild -list
+    - name: Placeholder GoogleService-Info.plist for build testing
+      run: cp FirebaseCore/Tests/Unit/Resources/GoogleService-Info.plist FirebaseVertexAI/Sample/
+    - uses: nick-fields/retry@v3
+      with:
+        timeout_minutes: 120
+        max_attempts: 3
+        retry_on: error
+        retry_wait_seconds: 120
+        command: scripts/build.sh VertexSample iOS
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index c249613f9e9..d2e4f95c0af 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -995,6 +995,7 @@ private extension URLRequest {
   }
 }
 
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
 class AppCheckInteropFake: NSObject, AppCheckInterop {
   /// The placeholder token value returned when an error occurs
   static let placeholderTokenValue = "placeholder-token"
diff --git a/scripts/build.sh b/scripts/build.sh
index d4caa635ebf..75935652c98 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -490,6 +490,14 @@ case "$product-$platform-$method" in
       build
     ;;
 
+  VertexSample-*-*)
+    RunXcodebuild \
+      -project 'FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj' \
+      -scheme "GenerativeAISample" \
+      "${xcb_flags[@]}" \
+      build
+    ;;
+
   Sessions-*-integration)
     # Perform "pod install" to install the relevant dependencies
     # ./FirebaseSessions/generate_testapp.sh

From 05f3d4f461e1f187af45c6c2ac24b5bb40103d20 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 25 Mar 2024 20:47:37 +0000
Subject: [PATCH 022/124] Fix `CitationMetadata` parsing in Vertex AI (#12626)

---
 .../Sources/GenerateContentResponse.swift     |  6 +-
 .../streaming-success-citations.txt           | 14 ++---
 .../unary-success-citations.json              | 63 +++++++++----------
 .../Tests/Unit/GenerativeModelTests.swift     | 25 +++++---
 4 files changed, 56 insertions(+), 52 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 03153da9ae7..4e02abb7f2c 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -155,6 +155,10 @@ extension CandidateResponse: Decodable {
 /// A collection of source attributions for a piece of content.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public struct CitationMetadata: Decodable {
+  enum CodingKeys: String, CodingKey {
+    case citationSources = "citations"
+  }
+
   /// A list of individual cited sources and the parts of the content to which they apply.
   public let citationSources: [Citation]
 }
@@ -172,7 +176,7 @@ public struct Citation: Decodable {
   public let uri: String
 
   /// The license the cited source work is distributed under.
-  public let license: String
+  public let license: String?
 }
 
 /// A value enumerating possible reasons for a model to terminate a content generation request.
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
index 4f50be7d32f..a530736b308 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
@@ -1,13 +1,9 @@
-data: {"candidates": [{"content": {"parts": [{"text": "Some information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Some information"}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": " More information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "  More information"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06632687,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.07477004,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048767097},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13695431,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059866417},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.046119746,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.036425155}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": ", Even more information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Some information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07850098,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.039416388},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.08035747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04885778},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12273335,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059646938},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.053206205,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04099903}],"citationMetadata": {"citations": [{"startIndex": 31,"endIndex": 187,"uri": "https://www.example.com/citation-1"}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": " Some information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-1","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-2","license": ""}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.044183318},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.094176665,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0575992},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13660839,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08035747},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.060197048,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.046464667}],"citationMetadata": {"citations": [{"startIndex": 73,"endIndex": 248,"uri": "https://www.example.com/citation-2"},{"startIndex": 133,"endIndex": 272,"uri": "https://www.example.com/citation-3", "license": "mit"}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "More information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-3","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-4","license": ""}]}}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "Even more information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-5","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-6","license": ""}]}}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "Physics (YouTube Channel)"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.google.com","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.google.com","license": ""}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.12147716,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0647717},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.11858909,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.14866412,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08479541},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05470151,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045015533}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 163,"totalTokenCount": 172}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
index ac99fcf0944..a0dacc3e883 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
@@ -2,63 +2,58 @@
   "candidates": [
     {
       "content": {
+        "role": "model",
         "parts": [
           {
             "text": "Some information cited from an external source"
           }
-        ],
-        "role": "model"
+        ]
       },
       "finishReason": "STOP",
-      "index": 0,
       "safetyRatings": [
         {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.16013464,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.074500255
         },
         {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.09687653,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.049313594
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.16817278,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.09451043
         },
         {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.05023736,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.034553625
         }
       ],
       "citationMetadata": {
-        "citationSources": [
+        "citations": [
           {
-            "startIndex": 574,
-            "endIndex": 705,
-            "uri": "https://www.example.com/some-citation",
-            "license": ""
+            "startIndex": 179,
+            "endIndex": 366,
+            "uri": "https://www.example.com/some-citation"
           }
         ]
       }
     }
   ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
+  "usageMetadata": {
+    "promptTokenCount": 11,
+    "candidatesTokenCount": 135,
+    "totalTokenCount": 146
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index d2e4f95c0af..41d95110340 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -107,9 +107,9 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(citationMetadata.citationSources.count, 1)
     let citationSource = try XCTUnwrap(citationMetadata.citationSources.first)
     XCTAssertEqual(citationSource.uri, "https://www.example.com/some-citation")
-    XCTAssertEqual(citationSource.startIndex, 574)
-    XCTAssertEqual(citationSource.endIndex, 705)
-    XCTAssertEqual(citationSource.license, "")
+    XCTAssertEqual(citationSource.startIndex, 179)
+    XCTAssertEqual(citationSource.endIndex, 366)
+    XCTAssertNil(citationSource.license)
   }
 
   func testGenerateContent_success_quoteReply() async throws {
@@ -669,21 +669,30 @@ final class GenerativeModelTests: XCTestCase {
       )
 
     let stream = model.generateContentStream("Hi")
-    var citations: [Citation] = []
+    var citations = [Citation]()
+    var responses = [GenerateContentResponse]()
     for try await content in stream {
+      responses.append(content)
       XCTAssertNotNil(content.text)
       let candidate = try XCTUnwrap(content.candidates.first)
-      XCTAssertEqual(candidate.finishReason, .stop)
       if let sources = candidate.citationMetadata?.citationSources {
         citations.append(contentsOf: sources)
       }
     }
 
-    XCTAssertEqual(citations.count, 8)
+    let lastCandidate = try XCTUnwrap(responses.last?.candidates.first)
+    XCTAssertEqual(lastCandidate.finishReason, .stop)
+    XCTAssertEqual(citations.count, 3)
     XCTAssertTrue(citations
-      .contains(where: { $0.startIndex == 574 && $0.endIndex == 705 && !$0.uri.isEmpty }))
+      .contains(where: {
+        $0.startIndex == 31 && $0.endIndex == 187 && $0
+          .uri == "https://www.example.com/citation-1" && $0.license == nil
+      }))
     XCTAssertTrue(citations
-      .contains(where: { $0.startIndex == 899 && $0.endIndex == 1026 && !$0.uri.isEmpty }))
+      .contains(where: {
+        $0.startIndex == 133 && $0.endIndex == 272 && $0
+          .uri == "https://www.example.com/citation-3" && $0.license == "mit"
+      }))
   }
 
   func testGenerateContentStream_appCheck_validToken() async throws {

From 4e3a4d108a5437c5e1f50960e5da4321f36b7048 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 26 Mar 2024 13:24:29 -0700
Subject: [PATCH 023/124] Manage location on VertexAI instead of model (#12630)

---
 .../ViewModels/ConversationViewModel.swift    |  5 +--
 .../ViewModels/PhotoReasoningViewModel.swift  |  5 +--
 .../ViewModels/SummarizeViewModel.swift       |  5 +--
 FirebaseVertexAI/Sources/VertexAI.swift       | 40 +++++++++++--------
 .../Sources/VertexAIComponent.swift           |  6 +--
 .../Tests/Unit/VertexAIAPITests.swift         |  9 ++---
 6 files changed, 33 insertions(+), 37 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 05cbe11250f..81667be2514 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,10 +36,7 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.vertexAI().generativeModel(
-      modelName: "gemini-1.0-pro",
-      location: "us-central1"
-    )
+    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index 2f2ed88d4a1..b4880bdc825 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,10 +44,7 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI().generativeModel(
-      modelName: "gemini-1.0-pro-vision",
-      location: "us-central1"
-    )
+    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func reason() async {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index 0e3073d6da2..e3c78d09060 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,10 +32,7 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI().generativeModel(
-      modelName: "gemini-1.0-pro",
-      location: "us-central1"
-    )
+    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 64e76d3b4f5..2abb4abf7b0 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,26 +25,35 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
+  ///  - Parameter region: The region identifier, e.g., `us-central1`; see
+  ///     [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI() -> VertexAI {
+  public static func vertexAI(region: String) -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app)
+    return vertexAI(app: app, region: region)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
-  /// - Parameter app: The custom `FirebaseApp` used for initialization.
+  ///  - Parameters:
+  ///   - app: The custom `FirebaseApp` used for initialization.
+  ///   - region: The region identifier, e.g., `us-central1`; see
+  ///     [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp, region: String) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI()
+    return provider.vertexAI(region)
   }
 
   /// Initializes a generative model with the given parameters.
@@ -54,19 +63,15 @@ public class VertexAI: NSObject {
   ///     [Gemini
   ///     models](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#gemini-models)
   ///     for a list of supported model names.
-  ///   - location: The location identifier, e.g., `us-central1`; see
-  ///     [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
-  ///     for a list of supported locations.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
-  public func generativeModel(modelName: String, location: String,
+  public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
-    let modelResourceName = modelResourceName(modelName: modelName, location: location)
+    let modelResourceName = modelResourceName(modelName: modelName, region: region)
 
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
@@ -89,26 +94,29 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
-  init(app: FirebaseApp) {
+  private let region: String
+
+  init(app: FirebaseApp, region: String) {
     self.app = app
+    self.region = region
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
   }
 
-  private func modelResourceName(modelName: String, location: String) -> String {
+  private func modelResourceName(modelName: String, region: String) -> String {
     if modelName.contains("/") {
       return modelName
     }
     guard let projectID = app.options.projectID else {
       fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
     }
-    guard !location.isEmpty else {
+    guard !region.isEmpty else {
       fatalError("""
-      No location specified; see
+      No region specified; see
       https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
       list of available regions.
       """)
     }
 
-    return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
+    return "projects/\(projectID)/locations/\(region)/publishers/google/models/\(modelName)"
   }
 }
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index 1378f812626..a9b7aa669a6 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI() -> VertexAI
+  @objc func vertexAI(_ location: String) -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -64,7 +64,7 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   // MARK: - VertexAIProvider conformance
 
-  func vertexAI() -> VertexAI {
+  func vertexAI(_ region: String) -> VertexAI {
     os_unfair_lock_lock(&instancesLock)
 
     // Unlock before the function returns.
@@ -73,7 +73,7 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
     if let instance = instances[app.name] {
       return instance
     }
-    let newInstance = VertexAI(app: app)
+    let newInstance = VertexAI(app: app, region: region)
     instances[app.name] = newInstance
     return newInstance
   }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index a1fd27ab4c5..a4983eb8834 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -34,31 +34,28 @@ final class VertexAIAPITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Instantiate Vertex AI SDK - Default App
-    let vertexAI = VertexAI.vertexAI()
+    let vertexAI = VertexAI.vertexAI(region: "my-region")
 
     // Instantiate Vertex AI SDK - Custom App
-    let _ = VertexAI.vertexAI(app: app!)
+    let _ = VertexAI.vertexAI(app: app!, region: "my-region")
 
     // Permutations without optional arguments.
 
-    let _ = vertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    let _ = vertexAI.generativeModel(modelName: "gemini-1.0-pro")
 
     let _ = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1",
       safetySettings: filters
     )
 
     let _ = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1",
       generationConfig: config
     )
 
     // All arguments passed.
     let genAI = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1",
       generationConfig: config, // Optional
       safetySettings: filters // Optional
     )

From a89956f87dc02465111aeb6e7c7a4f1436473332 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 26 Mar 2024 17:58:47 -0700
Subject: [PATCH 024/124] VertexAI Component tests; fix instance generation for
 different regions (#12631)

---
 FirebaseVertexAI/Sources/VertexAI.swift       |   2 +-
 .../Sources/VertexAIComponent.swift           |   8 +-
 .../Tests/Unit/VertexComponentTests.swift     | 115 ++++++++++++++++++
 Package.swift                                 |  15 ++-
 4 files changed, 132 insertions(+), 8 deletions(-)
 create mode 100644 FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 2abb4abf7b0..d1682745f98 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -94,7 +94,7 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
-  private let region: String
+  let region: String
 
   init(app: FirebaseApp, region: String) {
     self.app = app
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a9b7aa669a6..af8dbb56271 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI(_ location: String) -> VertexAI
+  @objc func vertexAI(_ region: String) -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -32,7 +32,7 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   /// The app associated with all `VertexAI` instances in this container.
   /// This is `unowned` instead of `weak` so it can be used without unwrapping in `vertexAI(...)`
-  private unowned let app: FirebaseApp
+  unowned let app: FirebaseApp
 
   /// A map of active  `VertexAI` instances for `app`, keyed by model resource names
   /// (e.g., "projects/my-project-id/locations/us-central1/publishers/google/models/gemini-pro").
@@ -70,11 +70,11 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
     // Unlock before the function returns.
     defer { os_unfair_lock_unlock(&instancesLock) }
 
-    if let instance = instances[app.name] {
+    if let instance = instances[region] {
       return instance
     }
     let newInstance = VertexAI(app: app, region: region)
-    instances[app.name] = newInstance
+    instances[region] = newInstance
     return newInstance
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
new file mode 100644
index 00000000000..7c046c5e949
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
@@ -0,0 +1,115 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+import FirebaseCore
+@testable import FirebaseVertexAI
+
+import SharedTestUtilities
+
+import XCTest
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+class VertexComponentTests: XCTestCase {
+  static var app: FirebaseApp!
+
+  override class func setUp() {
+    super.setUp()
+    if app == nil {
+      let options = FirebaseOptions(googleAppID: "0:0000000000000:ios:0000000000000000",
+                                    gcmSenderID: "00000000000000000-00000000000-000000000")
+      options.projectID = "myProjectID"
+      FirebaseApp.configure(options: options)
+      app = FirebaseApp(instanceWithName: "test", options: options)
+    }
+  }
+
+  // MARK: Interoperability Tests
+
+  /// Tests that the right number of components are being provided for the container.
+  func testComponentsBeingRegistered() throws {
+    let components = VertexAIComponent.componentsToRegister()
+    XCTAssert(components.count == 1)
+  }
+
+  /// Tests that a vertex instance can be created properly by the VertexAIComponent.
+  func testVertexInstanceCreation() throws {
+    let app = try XCTUnwrap(VertexComponentTests.app)
+    let component = VertexAIComponent(app: app)
+    let vertex = component.vertexAI("my-region")
+    XCTAssertNotNil(vertex)
+  }
+
+  /// Tests that the component container caches instances of VertexAIComponent.
+  func testMultipleComponentInstancesCreated() throws {
+    let registrants = NSMutableSet(array: [VertexAIComponent.self])
+    let container = FirebaseComponentContainer(
+      app: VertexComponentTests.app,
+      registrants: registrants
+    )
+
+    let provider1 = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                             in: container)
+    XCTAssertNotNil(provider1)
+
+    let provider2 = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                             in: container)
+    XCTAssertNotNil(provider2)
+
+    // Ensure they're the same instance.
+    XCTAssert(provider1 === provider2)
+  }
+
+  /// Tests that instances of vertex created are different.
+  func testMultipleVertexInstancesCreated() throws {
+    let app = try XCTUnwrap(VertexComponentTests.app)
+    let registrants = NSMutableSet(array: [VertexAIComponent.self])
+    let container = FirebaseComponentContainer(app: app, registrants: registrants)
+
+    let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                            in: container)
+    XCTAssertNotNil(provider)
+
+    let vertex1 = provider?.vertexAI("randomRegion")
+    let vertex2 = provider?.vertexAI("randomRegion")
+    XCTAssertNotNil(vertex1)
+
+    // Ensure they're the same instance.
+    XCTAssert(vertex1 === vertex2)
+
+    let vertex3 = provider?.vertexAI("differentRegion")
+    XCTAssertNotNil(vertex3)
+
+    XCTAssert(vertex1 !== vertex3)
+  }
+
+  /// Test that vertex instances get deallocated.
+  func testVertexLifecycle() throws {
+    weak var weakApp: FirebaseApp?
+    weak var weakVertex: VertexAI?
+    try autoreleasepool {
+      let options = FirebaseOptions(googleAppID: "0:0000000000000:ios:0000000000000000",
+                                    gcmSenderID: "00000000000000000-00000000000-000000000")
+      options.projectID = "myProjectID"
+      let app1 = FirebaseApp(instanceWithName: "transitory app", options: options)
+      weakApp = try XCTUnwrap(app1)
+      let vertex = VertexAI(app: app1, region: "transitory bucket")
+      weakVertex = vertex
+      XCTAssertNotNil(weakVertex)
+    }
+    XCTAssertNil(weakApp)
+    XCTAssertNil(weakVertex)
+  }
+}
diff --git a/Package.swift b/Package.swift
index d3d99222cf7..4d9f42f89db 100644
--- a/Package.swift
+++ b/Package.swift
@@ -170,10 +170,16 @@ let package = Package(
     ),
     abseilDependency(),
     grpcDependency(),
+    // TODO: restore OCMock when https://github.com/erikdoe/ocmock/pull/537
+    // gets merged to fix Xcode 15.3 builds.
     .package(
-      url: "https://github.com/erikdoe/ocmock.git",
-      revision: "c5eeaa6dde7c308a5ce48ae4d4530462dd3a1110"
+      url: "https://github.com/paulb777/ocmock.git",
+      revision: "173955e93e6ee6999a10729ab67e4b4efdd1db6d"
     ),
+//    .package(
+//      url: "https://github.com/erikdoe/ocmock.git",
+//      revision: "c5eeaa6dde7c308a5ce48ae4d4530462dd3a1110"
+//    ),
     .package(
       url: "https://github.com/firebase/leveldb.git",
       "1.22.2" ..< "1.23.0"
@@ -1365,11 +1371,14 @@ let package = Package(
     ),
     .testTarget(
       name: "FirebaseVertexAIUnit",
-      dependencies: ["FirebaseVertexAI"],
+      dependencies: ["FirebaseVertexAI", "SharedTestUtilities"],
       path: "FirebaseVertexAI/Tests/Unit",
       resources: [
         .process("CountTokenResponses"),
         .process("GenerateContentResponses"),
+      ],
+      cSettings: [
+        .headerSearchPath("../../../"),
       ]
     ),
   ] + firestoreTargets(),

From 19b2004c23892e0243fd01ab7f6fb1527b3f9a2b Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 14:34:49 +0000
Subject: [PATCH 025/124] Add function calling support in Vertex AI (#12633)

---
 FirebaseVertexAI/Sources/Chat.swift           |   2 +-
 .../Sources/FunctionCalling.swift             | 235 ++++++++++++++++++
 .../Sources/GenerateContentRequest.swift      |   2 +
 .../Sources/GenerativeModel.swift             |   8 +
 FirebaseVertexAI/Sources/JSONValue.swift      |  96 +++++++
 FirebaseVertexAI/Sources/ModelContent.swift   |  16 +-
 FirebaseVertexAI/Sources/VertexAI.swift       |   3 +
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |   1 +
 ...success-function-call-empty-arguments.json |  18 ++
 ...ry-success-function-call-no-arguments.json |  19 ++
 ...-success-function-call-with-arguments.json |  22 ++
 .../Tests/Unit/GenerativeModelTests.swift     |  79 ++++++
 .../Tests/Unit/JSONValueTests.swift           | 145 +++++++++++
 .../Tests/Unit/VertexComponentTests.swift     |   2 +-
 14 files changed, 645 insertions(+), 3 deletions(-)
 create mode 100644 FirebaseVertexAI/Sources/FunctionCalling.swift
 create mode 100644 FirebaseVertexAI/Sources/JSONValue.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/JSONValueTests.swift

diff --git a/FirebaseVertexAI/Sources/Chat.swift b/FirebaseVertexAI/Sources/Chat.swift
index c7cfb859fa8..e443e18bed8 100644
--- a/FirebaseVertexAI/Sources/Chat.swift
+++ b/FirebaseVertexAI/Sources/Chat.swift
@@ -153,7 +153,7 @@ public class Chat {
         case let .text(str):
           combinedText += str
 
-        case .data(mimetype: _, _):
+        case .data, .functionCall, .functionResponse:
           // Don't combine it, just add to the content. If there's any text pending, add that as
           // a part.
           if !combinedText.isEmpty {
diff --git a/FirebaseVertexAI/Sources/FunctionCalling.swift b/FirebaseVertexAI/Sources/FunctionCalling.swift
new file mode 100644
index 00000000000..7e4d84d857e
--- /dev/null
+++ b/FirebaseVertexAI/Sources/FunctionCalling.swift
@@ -0,0 +1,235 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A predicted function call returned from the model.
+public struct FunctionCall: Equatable, Encodable {
+  /// The name of the function to call.
+  public let name: String
+
+  /// The function parameters and values.
+  public let args: JSONObject
+}
+
+/// A `Schema` object allows the definition of input and output data types.
+///
+/// These types can be objects, but also primitives and arrays. Represents a select subset of an
+/// [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema).
+public class Schema: Encodable {
+  /// The data type.
+  let type: DataType
+
+  /// The format of the data.
+  let format: String?
+
+  /// A brief description of the parameter.
+  let description: String?
+
+  /// Indicates if the value may be null.
+  let nullable: Bool?
+
+  /// Possible values of the element of type ``DataType/string`` with "enum" format.
+  let enumValues: [String]?
+
+  /// Schema of the elements of type ``DataType/array``.
+  let items: Schema?
+
+  /// Properties of type ``DataType/object``.
+  let properties: [String: Schema]?
+
+  /// Required properties of type ``DataType/object``.
+  let requiredProperties: [String]?
+
+  enum CodingKeys: String, CodingKey {
+    case type
+    case format
+    case description
+    case nullable
+    case enumValues = "enum"
+    case items
+    case properties
+    case requiredProperties = "required"
+  }
+
+  /// Constructs a new `Schema`.
+  ///
+  /// - Parameters:
+  ///   - type: The data type.
+  ///   - format: The format of the data; used only for primitive datatypes.
+  ///     Supported formats:
+  ///     - ``DataType/integer``: int32, int64
+  ///     - ``DataType/number``: float, double
+  ///     - ``DataType/string``: enum
+  ///   - description: A brief description of the parameter; may be formatted as Markdown.
+  ///   - nullable: Indicates if the value may be null.
+  ///   - enumValues: Possible values of the element of type ``DataType/string`` with "enum" format.
+  ///     For example, an enum `Direction` may be defined as `["EAST", NORTH", "SOUTH", "WEST"]`.
+  ///   - items: Schema of the elements of type ``DataType/array``.
+  ///   - properties: Properties of type ``DataType/object``.
+  ///   - requiredProperties: Required properties of type ``DataType/object``.
+  public init(type: DataType, format: String? = nil, description: String? = nil,
+              nullable: Bool? = nil,
+              enumValues: [String]? = nil, items: Schema? = nil,
+              properties: [String: Schema]? = nil,
+              requiredProperties: [String]? = nil) {
+    self.type = type
+    self.format = format
+    self.description = description
+    self.nullable = nullable
+    self.enumValues = enumValues
+    self.items = items
+    self.properties = properties
+    self.requiredProperties = requiredProperties
+  }
+}
+
+/// A data type.
+///
+/// Contains the set of OpenAPI [data types](https://spec.openapis.org/oas/v3.0.3#data-types).
+public enum DataType: String, Encodable {
+  /// A `String` type.
+  case string = "STRING"
+
+  /// A floating-point number type.
+  case number = "NUMBER"
+
+  /// An integer type.
+  case integer = "INTEGER"
+
+  /// A boolean type.
+  case boolean = "BOOLEAN"
+
+  /// An array type.
+  case array = "ARRAY"
+
+  /// An object type.
+  case object = "OBJECT"
+}
+
+/// Structured representation of a function declaration.
+///
+/// This `FunctionDeclaration` is a representation of a block of code that can be used as a ``Tool``
+/// by the model and executed by the client.
+public struct FunctionDeclaration {
+  /// The name of the function.
+  let name: String
+
+  /// A brief description of the function.
+  let description: String
+
+  /// Describes the parameters to this function; must be of type ``DataType/object``.
+  let parameters: Schema?
+
+  /// Constructs a new `FunctionDeclaration`.
+  ///
+  /// - Parameters:
+  ///   - name: The name of the function; must be a-z, A-Z, 0-9, or contain underscores and dashes,
+  ///   with a maximum length of 63.
+  ///   - description: A brief description of the function.
+  ///   - parameters: Describes the parameters to this function; the keys are parameter names and
+  ///   the values are ``Schema`` objects describing them.
+  ///   - requiredParameters: A list of required parameters by name.
+  public init(name: String, description: String, parameters: [String: Schema]?,
+              requiredParameters: [String]?) {
+    self.name = name
+    self.description = description
+    self.parameters = Schema(
+      type: .object,
+      properties: parameters,
+      requiredProperties: requiredParameters
+    )
+  }
+}
+
+/// Helper tools that the model may use to generate response.
+///
+/// A `Tool` is a piece of code that enables the system to interact with external systems to
+/// perform an action, or set of actions, outside of knowledge and scope of the model.
+public struct Tool: Encodable {
+  /// A list of `FunctionDeclarations` available to the model.
+  let functionDeclarations: [FunctionDeclaration]?
+
+  /// Constructs a new `Tool`.
+  ///
+  /// - Parameters:
+  ///   - functionDeclarations: A list of `FunctionDeclarations` available to the model that can be
+  ///   used for function calling.
+  ///   The model or system does not execute the function. Instead the defined function may be
+  ///   returned as a ``FunctionCall`` in ``ModelContent/Part/functionCall(_:)`` with arguments to
+  ///   the client side for execution. The model may decide to call a subset of these functions by
+  ///   populating ``FunctionCall`` in the response. The next conversation turn may contain a
+  ///   ``FunctionResponse`` in ``ModelContent/Part/functionResponse(_:)`` with the
+  ///   ``ModelContent/role`` "function", providing generation context for the next model turn.
+  public init(functionDeclarations: [FunctionDeclaration]?) {
+    self.functionDeclarations = functionDeclarations
+  }
+}
+
+/// Result output from a ``FunctionCall``.
+///
+/// Contains a string representing the `FunctionDeclaration.name` and a structured JSON object
+/// containing any output from the function is used as context to the model. This should contain the
+/// result of a ``FunctionCall`` made based on model prediction.
+public struct FunctionResponse: Equatable, Encodable {
+  /// The name of the function that was called.
+  let name: String
+
+  /// The function's response.
+  let response: JSONObject
+
+  /// Constructs a new `FunctionResponse`.
+  ///
+  /// - Parameters:
+  ///   - name: The name of the function that was called.
+  ///   - response: The function's response.
+  public init(name: String, response: JSONObject) {
+    self.name = name
+    self.response = response
+  }
+}
+
+// MARK: - Codable Conformance
+
+extension FunctionCall: Decodable {
+  enum CodingKeys: CodingKey {
+    case name
+    case args
+  }
+
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    name = try container.decode(String.self, forKey: .name)
+    if let args = try container.decodeIfPresent(JSONObject.self, forKey: .args) {
+      self.args = args
+    } else {
+      args = JSONObject()
+    }
+  }
+}
+
+extension FunctionDeclaration: Encodable {
+  enum CodingKeys: String, CodingKey {
+    case name
+    case description
+    case parameters
+  }
+
+  public func encode(to encoder: Encoder) throws {
+    var container = encoder.container(keyedBy: CodingKeys.self)
+    try container.encode(name, forKey: .name)
+    try container.encode(description, forKey: .description)
+    try container.encode(parameters, forKey: .parameters)
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
index 417260bb700..535ac4da3ae 100644
--- a/FirebaseVertexAI/Sources/GenerateContentRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -21,6 +21,7 @@ struct GenerateContentRequest {
   let contents: [ModelContent]
   let generationConfig: GenerationConfig?
   let safetySettings: [SafetySetting]?
+  let tools: [Tool]?
   let isStreaming: Bool
   let options: RequestOptions
 }
@@ -31,6 +32,7 @@ extension GenerateContentRequest: Encodable {
     case contents
     case generationConfig
     case safetySettings
+    case tools
   }
 }
 
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index aad2b3e99c0..7eab63a1d7a 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -34,6 +34,9 @@ public final class GenerativeModel {
   /// The safety settings to be used for prompts.
   let safetySettings: [SafetySetting]?
 
+  /// A list of tools the model may use to generate the next response.
+  let tools: [Tool]?
+
   /// Configuration parameters for sending requests to the backend.
   let requestOptions: RequestOptions
 
@@ -45,12 +48,14 @@ public final class GenerativeModel {
   ///   - apiKey: The API key for your project.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
        apiKey: String,
        generationConfig: GenerationConfig? = nil,
        safetySettings: [SafetySetting]? = nil,
+       tools: [Tool]?,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
        urlSession: URLSession = .shared) {
@@ -62,6 +67,7 @@ public final class GenerativeModel {
     )
     self.generationConfig = generationConfig
     self.safetySettings = safetySettings
+    self.tools = tools
     self.requestOptions = requestOptions
 
     Logging.default.info("""
@@ -107,6 +113,7 @@ public final class GenerativeModel {
                                                               contents: content(),
                                                               generationConfig: generationConfig,
                                                               safetySettings: safetySettings,
+                                                              tools: tools,
                                                               isStreaming: false,
                                                               options: requestOptions)
       response = try await generativeAIService.loadRequest(request: generateContentRequest)
@@ -178,6 +185,7 @@ public final class GenerativeModel {
                                                         contents: evaluatedContent,
                                                         generationConfig: generationConfig,
                                                         safetySettings: safetySettings,
+                                                        tools: tools,
                                                         isStreaming: true,
                                                         options: requestOptions)
 
diff --git a/FirebaseVertexAI/Sources/JSONValue.swift b/FirebaseVertexAI/Sources/JSONValue.swift
new file mode 100644
index 00000000000..5ce52cd2af9
--- /dev/null
+++ b/FirebaseVertexAI/Sources/JSONValue.swift
@@ -0,0 +1,96 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A collection of name-value pairs representing a JSON object.
+///
+/// This may be decoded from, or encoded to, a
+/// [`google.protobuf.Struct`](https://protobuf.dev/reference/protobuf/google.protobuf/#struct).
+public typealias JSONObject = [String: JSONValue]
+
+/// Represents a value in one of JSON's data types.
+///
+/// This may be decoded from, or encoded to, a
+/// [`google.protobuf.Value`](https://protobuf.dev/reference/protobuf/google.protobuf/#value).
+public enum JSONValue {
+  /// A `null` value.
+  case null
+
+  /// A numeric value.
+  case number(Double)
+
+  /// A string value.
+  case string(String)
+
+  /// A boolean value.
+  case bool(Bool)
+
+  /// A JSON object.
+  case object(JSONObject)
+
+  /// An array of `JSONValue`s.
+  case array([JSONValue])
+}
+
+extension JSONValue: Decodable {
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.singleValueContainer()
+    if container.decodeNil() {
+      self = .null
+    } else if let numberValue = try? container.decode(Double.self) {
+      self = .number(numberValue)
+    } else if let stringValue = try? container.decode(String.self) {
+      self = .string(stringValue)
+    } else if let boolValue = try? container.decode(Bool.self) {
+      self = .bool(boolValue)
+    } else if let objectValue = try? container.decode(JSONObject.self) {
+      self = .object(objectValue)
+    } else if let arrayValue = try? container.decode([JSONValue].self) {
+      self = .array(arrayValue)
+    } else {
+      throw DecodingError.dataCorruptedError(
+        in: container,
+        debugDescription: "Failed to decode JSON value."
+      )
+    }
+  }
+}
+
+extension JSONValue: Encodable {
+  public func encode(to encoder: Encoder) throws {
+    var container = encoder.singleValueContainer()
+    switch self {
+    case .null:
+      try container.encodeNil()
+    case let .number(numberValue):
+      // Convert to `Decimal` before encoding for consistent floating-point serialization across
+      // platforms. E.g., `Double` serializes 3.14159 as 3.1415899999999999 in some cases and
+      // 3.14159 in others. See
+      // https://forums.swift.org/t/jsonencoder-encodable-floating-point-rounding-error/41390/4 for
+      // more details.
+      try container.encode(Decimal(numberValue))
+    case let .string(stringValue):
+      try container.encode(stringValue)
+    case let .bool(boolValue):
+      try container.encode(boolValue)
+    case let .object(objectValue):
+      try container.encode(objectValue)
+    case let .array(arrayValue):
+      try container.encode(arrayValue)
+    }
+  }
+}
+
+extension JSONValue: Equatable {}
diff --git a/FirebaseVertexAI/Sources/ModelContent.swift b/FirebaseVertexAI/Sources/ModelContent.swift
index 44648c57852..136fc1debe3 100644
--- a/FirebaseVertexAI/Sources/ModelContent.swift
+++ b/FirebaseVertexAI/Sources/ModelContent.swift
@@ -25,6 +25,8 @@ public struct ModelContent: Codable, Equatable {
     enum CodingKeys: String, CodingKey {
       case text
       case inlineData
+      case functionCall
+      case functionResponse
     }
 
     enum InlineDataKeys: String, CodingKey {
@@ -38,6 +40,12 @@ public struct ModelContent: Codable, Equatable {
     /// Data with a specified media type. Not all media types may be supported by the AI model.
     case data(mimetype: String, Data)
 
+    /// A predicted function call returned from the model.
+    case functionCall(FunctionCall)
+
+    /// A response to a function call.
+    case functionResponse(FunctionResponse)
+
     // MARK: Convenience Initializers
 
     /// Convenience function for populating a Part with JPEG data.
@@ -64,6 +72,10 @@ public struct ModelContent: Codable, Equatable {
         )
         try inlineDataContainer.encode(mimetype, forKey: .mimeType)
         try inlineDataContainer.encode(bytes, forKey: .bytes)
+      case let .functionCall(functionCall):
+        try container.encode(functionCall, forKey: .functionCall)
+      case let .functionResponse(functionResponse):
+        try container.encode(functionResponse, forKey: .functionResponse)
       }
     }
 
@@ -79,10 +91,12 @@ public struct ModelContent: Codable, Equatable {
         let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
         let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
         self = .data(mimetype: mimetype, bytes)
+      } else if values.contains(.functionCall) {
+        self = try .functionCall(values.decode(FunctionCall.self, forKey: .functionCall))
       } else {
         throw DecodingError.dataCorrupted(.init(
           codingPath: [CodingKeys.text, CodingKeys.inlineData],
-          debugDescription: "Neither text or inline data was found."
+          debugDescription: "No text, inline data or function call was found."
         ))
       }
     }
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index d1682745f98..a0df2ab015e 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -65,10 +65,12 @@ public class VertexAI: NSObject {
   ///     for a list of supported model names.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
+                              tools: [Tool]? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
     let modelResourceName = modelResourceName(modelName: modelName, region: region)
@@ -82,6 +84,7 @@ public class VertexAI: NSObject {
       apiKey: apiKey,
       generationConfig: generationConfig,
       safetySettings: safetySettings,
+      tools: tools,
       requestOptions: requestOptions,
       appCheck: appCheck
     )
diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 5a7868fbe7a..20897a8787b 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -50,6 +50,7 @@ final class ChatTests: XCTestCase {
     let model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
       urlSession: urlSession
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json
new file mode 100644
index 00000000000..dc0b75ad930
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json
@@ -0,0 +1,18 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "functionCall": {
+              "name": "current_time"
+            }
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json
new file mode 100644
index 00000000000..05f4f4d58f9
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json
@@ -0,0 +1,19 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "functionCall": {
+              "name": "current_time",
+              "args": {}
+            }
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json
new file mode 100644
index 00000000000..025735a3a39
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json
@@ -0,0 +1,22 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "functionCall": {
+              "name": "sum",
+              "args": {
+                "y": 5,
+                "x": 4
+              }
+            }
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 41d95110340..9a1210c123c 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -37,6 +37,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
       urlSession: urlSession
@@ -165,6 +166,7 @@ final class GenerativeModelTests: XCTestCase {
       // Model name is prefixed with "models/".
       name: "models/test-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
       urlSession: urlSession
@@ -173,11 +175,79 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_success_functionCall_emptyArguments() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-function-call-empty-arguments",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    guard case let .functionCall(functionCall) = part else {
+      XCTFail("Part is not a FunctionCall.")
+      return
+    }
+    XCTAssertEqual(functionCall.name, "current_time")
+    XCTAssertTrue(functionCall.args.isEmpty)
+  }
+
+  func testGenerateContent_success_functionCall_noArguments() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-function-call-no-arguments",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    guard case let .functionCall(functionCall) = part else {
+      XCTFail("Part is not a FunctionCall.")
+      return
+    }
+    XCTAssertEqual(functionCall.name, "current_time")
+    XCTAssertTrue(functionCall.args.isEmpty)
+  }
+
+  func testGenerateContent_success_functionCall_withArguments() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-function-call-with-arguments",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    guard case let .functionCall(functionCall) = part else {
+      XCTFail("Part is not a FunctionCall.")
+      return
+    }
+    XCTAssertEqual(functionCall.name, "sum")
+    XCTAssertEqual(functionCall.args.count, 2)
+    let argX = try XCTUnwrap(functionCall.args["x"])
+    XCTAssertEqual(argX, .number(4))
+    let argY = try XCTUnwrap(functionCall.args["y"])
+    XCTAssertEqual(argY, .number(5))
+  }
+
   func testGenerateContent_appCheck_validToken() async throws {
     let appCheckToken = "test-valid-token"
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
       urlSession: urlSession
@@ -196,6 +266,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
       urlSession: urlSession
@@ -493,6 +564,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
       urlSession: urlSession
@@ -700,6 +772,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
       urlSession: urlSession
@@ -719,6 +792,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
       urlSession: urlSession
@@ -836,6 +910,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
       urlSession: urlSession
@@ -894,6 +969,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
       urlSession: urlSession
@@ -913,6 +989,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: modelName,
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil
     )
@@ -926,6 +1003,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: modelResourceName,
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil
     )
@@ -939,6 +1017,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: tunedModelResourceName,
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil
     )
diff --git a/FirebaseVertexAI/Tests/Unit/JSONValueTests.swift b/FirebaseVertexAI/Tests/Unit/JSONValueTests.swift
new file mode 100644
index 00000000000..74196230887
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/JSONValueTests.swift
@@ -0,0 +1,145 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+import XCTest
+
+@testable import FirebaseVertexAI
+
+final class JSONValueTests: XCTestCase {
+  let decoder = JSONDecoder()
+  let encoder = JSONEncoder()
+
+  let numberKey = "pi"
+  let numberValue = 3.14159
+  let numberValueEncoded = "3.14159"
+  let stringKey = "hello"
+  let stringValue = "Hello, world!"
+
+  override func setUp() {
+    encoder.outputFormatting = .sortedKeys
+  }
+
+  func testDecodeNull() throws {
+    let jsonData = try XCTUnwrap("null".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .null)
+  }
+
+  func testDecodeNumber() throws {
+    let jsonData = try XCTUnwrap("\(numberValue)".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .number(numberValue))
+  }
+
+  func testDecodeString() throws {
+    let jsonData = try XCTUnwrap("\"\(stringValue)\"".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .string(stringValue))
+  }
+
+  func testDecodeBool() throws {
+    let expectedBool = true
+    let jsonData = try XCTUnwrap("\(expectedBool)".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .bool(expectedBool))
+  }
+
+  func testDecodeObject() throws {
+    let expectedObject: JSONObject = [
+      numberKey: .number(numberValue),
+      stringKey: .string(stringValue),
+    ]
+    let json = """
+    {
+      "\(numberKey)": \(numberValue),
+      "\(stringKey)": "\(stringValue)"
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .object(expectedObject))
+  }
+
+  func testDecodeArray() throws {
+    let expectedArray: [JSONValue] = [.null, .number(numberValue)]
+    let jsonData = try XCTUnwrap("[ null, \(numberValue) ]".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .array(expectedArray))
+  }
+
+  func testEncodeNull() throws {
+    let jsonData = try encoder.encode(JSONValue.null)
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "null")
+  }
+
+  func testEncodeNumber() throws {
+    let jsonData = try encoder.encode(JSONValue.number(numberValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "\(numberValue)")
+  }
+
+  func testEncodeString() throws {
+    let jsonData = try encoder.encode(JSONValue.string(stringValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "\"\(stringValue)\"")
+  }
+
+  func testEncodeBool() throws {
+    let boolValue = true
+
+    let jsonData = try encoder.encode(JSONValue.bool(boolValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "\(boolValue)")
+  }
+
+  func testEncodeObject() throws {
+    let objectValue: JSONObject = [
+      numberKey: .number(numberValue),
+      stringKey: .string(stringValue),
+    ]
+
+    let jsonData = try encoder.encode(JSONValue.object(objectValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(
+      json,
+      "{\"\(stringKey)\":\"\(stringValue)\",\"\(numberKey)\":\(numberValueEncoded)}"
+    )
+  }
+
+  func testEncodeArray() throws {
+    let arrayValue: [JSONValue] = [.null, .number(numberValue)]
+
+    let jsonData = try encoder.encode(JSONValue.array(arrayValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "[null,\(numberValueEncoded)]")
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
index 7c046c5e949..e28b747b65c 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
@@ -105,7 +105,7 @@ class VertexComponentTests: XCTestCase {
       options.projectID = "myProjectID"
       let app1 = FirebaseApp(instanceWithName: "transitory app", options: options)
       weakApp = try XCTUnwrap(app1)
-      let vertex = VertexAI(app: app1, region: "transitory bucket")
+      let vertex = VertexAI(app: app1, region: "transitory region")
       weakVertex = vertex
       XCTAssertNotNil(weakVertex)
     }

From 27617c4e2f050b55dd881e7e7c1d6fdb4e57b972 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 16:35:36 +0000
Subject: [PATCH 026/124] Add `-Preview` suffix to `FirebaseVertexAI` library
 name (#12641)

---
 .../project.pbxproj                           | 100 +++++++++---------
 Package.swift                                 |   2 +-
 2 files changed, 51 insertions(+), 51 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 1b4a69cafb0..9efade7342d 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -7,10 +7,14 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
-		869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AA2B86BE3100482873 /* FirebaseVertexAI */; };
-		869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AC2B86BEA100482873 /* FirebaseVertexAI */; };
-		869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AE2B86BFBC00482873 /* FirebaseVertexAI */; };
-		869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200B02B86BFF500482873 /* FirebaseVertexAI */; };
+		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
+		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
+		868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33692BB4771900304BB1 /* FirebaseAppCheck */; };
+		868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */; };
+		868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336D2BB4773600304BB1 /* FirebaseAppCheck */; };
+		868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */; };
+		868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33712BB4775D00304BB1 /* FirebaseAppCheck */; };
+		868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
 		869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
 		869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
@@ -56,10 +60,6 @@
 		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
 		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
 		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
-		8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */; };
-		8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */; };
-		8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */; };
-		8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E302BA8EED80049BFCA /* FirebaseAppCheck */; };
 		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
 /* End PBXBuildFile section */
 
@@ -101,9 +101,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */,
+				868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
-				869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */,
+				868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */,
 				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -112,9 +112,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
+				868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */,
 				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
-				8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */,
-				869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
@@ -122,9 +122,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */,
+				868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
-				869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */,
+				868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */,
 				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -133,9 +133,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */,
+				868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
-				869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */,
+				868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */,
 				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -342,8 +342,8 @@
 			packageProductDependencies = (
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
-				869200AA2B86BE3100482873 /* FirebaseVertexAI */,
-				8D589E302BA8EED80049BFCA /* FirebaseAppCheck */,
+				868A33652BB476FA00304BB1 /* FirebaseAppCheck */,
+				868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAISample;
 			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
@@ -364,8 +364,8 @@
 			name = GenerativeAITextSample;
 			packageProductDependencies = (
 				88209C232B0FBE1700F64795 /* MarkdownUI */,
-				869200B02B86BFF500482873 /* FirebaseVertexAI */,
-				8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */,
+				868A33692BB4771900304BB1 /* FirebaseAppCheck */,
+				868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAITextSample;
 			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
@@ -387,8 +387,8 @@
 			packageProductDependencies = (
 				88B8A91D2B0FC55100424728 /* MarkdownUI */,
 				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
-				869200AE2B86BFBC00482873 /* FirebaseVertexAI */,
-				8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */,
+				868A336D2BB4773600304BB1 /* FirebaseAppCheck */,
+				868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAIMultimodalSample;
 			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
@@ -410,8 +410,8 @@
 			packageProductDependencies = (
 				88D9474C2B14F27E008B5580 /* MarkdownUI */,
 				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
-				869200AC2B86BEA100482873 /* FirebaseVertexAI */,
-				8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */,
+				868A33712BB4775D00304BB1 /* FirebaseAppCheck */,
+				868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = ChatSample;
 			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
@@ -453,7 +453,7 @@
 			packageReferences = (
 				88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */,
 				DEA09AC32B1FCE22001962D9 /* XCRemoteSwiftPackageReference "NetworkImage" */,
-				869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */,
+				868A33642BB476FA00304BB1 /* XCLocalSwiftPackageReference "../.." */,
 			);
 			productRefGroup = 8848C8302B0D04BC007B434F /* Products */;
 			projectDirPath = "";
@@ -985,7 +985,7 @@
 /* End XCConfigurationList section */
 
 /* Begin XCLocalSwiftPackageReference section */
-		869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */ = {
+		868A33642BB476FA00304BB1 /* XCLocalSwiftPackageReference "../.." */ = {
 			isa = XCLocalSwiftPackageReference;
 			relativePath = ../..;
 		};
@@ -1011,21 +1011,37 @@
 /* End XCRemoteSwiftPackageReference section */
 
 /* Begin XCSwiftPackageProductDependency section */
-		869200AA2B86BE3100482873 /* FirebaseVertexAI */ = {
+		868A33652BB476FA00304BB1 /* FirebaseAppCheck */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = FirebaseAppCheck;
+		};
+		868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = "FirebaseVertexAI-Preview";
+		};
+		868A33692BB4771900304BB1 /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
 		};
-		869200AC2B86BEA100482873 /* FirebaseVertexAI */ = {
+		868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = "FirebaseVertexAI-Preview";
 		};
-		869200AE2B86BFBC00482873 /* FirebaseVertexAI */ = {
+		868A336D2BB4773600304BB1 /* FirebaseAppCheck */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = FirebaseAppCheck;
+		};
+		868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = "FirebaseVertexAI-Preview";
+		};
+		868A33712BB4775D00304BB1 /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
 		};
-		869200B02B86BFF500482873 /* FirebaseVertexAI */ = {
+		868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = "FirebaseVertexAI-Preview";
 		};
 		88209C232B0FBE1700F64795 /* MarkdownUI */ = {
 			isa = XCSwiftPackageProductDependency;
@@ -1055,22 +1071,6 @@
 			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
 			productName = MarkdownUI;
 		};
-		8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		8D589E302BA8EED80049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
 		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = GenerativeAIUIComponents;
diff --git a/Package.swift b/Package.swift
index 4d9f42f89db..e401abcb750 100644
--- a/Package.swift
+++ b/Package.swift
@@ -138,7 +138,7 @@ let package = Package(
       targets: ["FirebaseStorage"]
     ),
     .library(
-      name: "FirebaseVertexAI",
+      name: "FirebaseVertexAI-Preview",
       targets: ["FirebaseVertexAI"]
     ),
   ],

From 950c4c85118ca6128e0281430fc0b8e82bb6d795 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 20:33:37 +0000
Subject: [PATCH 027/124] Fix model name in Vertex AI multi-modal sample
 (#12645)

---
 .../ViewModels/PhotoReasoningViewModel.swift                   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index b4880bdc825..1fb0097a808 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,8 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    let vertexAI = VertexAI.vertexAI(region: "us-central1")
+    model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
   func reason() async {

From 98120553db5a016fe8a990f8a2416e7ea4f4440a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 20:37:29 +0000
Subject: [PATCH 028/124] Add `totalBillableCharacters` to
 `CountTokensResponse` in Vertex AI (#12646)

---
 FirebaseVertexAI/Sources/CountTokensRequest.swift              | 3 +++
 .../Tests/Unit/CountTokenResponses/success-total-tokens.json   | 3 ++-
 FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift         | 2 ++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
index 0a58d40acc0..d294212658c 100644
--- a/FirebaseVertexAI/Sources/CountTokensRequest.swift
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -42,4 +42,7 @@ extension CountTokensRequest: GenerativeAIRequest {
 public struct CountTokensResponse: Decodable {
   /// The total number of tokens in the input given to the model as a prompt.
   public let totalTokens: Int
+
+  /// The total number of billable characters in the input given to the model as a prompt.
+  public let totalBillableCharacters: Int
 }
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
index 0bccd9e7c3d..d2ad6e4ff30 100644
--- a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
@@ -1,3 +1,4 @@
 {
-  "totalTokens": 6
+  "totalTokens": 6,
+  "totalBillableCharacters": 16
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 9a1210c123c..123299ac1a7 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -935,7 +935,9 @@ final class GenerativeModelTests: XCTestCase {
     )
 
     let response = try await model.countTokens("Why is the sky blue?")
+
     XCTAssertEqual(response.totalTokens, 6)
+    XCTAssertEqual(response.totalBillableCharacters, 16)
   }
 
   func testCountTokens_modelNotFound() async throws {

From 39ea53a796dc71661ce4bc1be5aef23196b17dbe Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 28 Mar 2024 22:17:31 +0000
Subject: [PATCH 029/124] Rename the Vertex AI parameter `region` to `location`
 (#12656)

---
 .../ViewModels/ConversationViewModel.swift    |  2 +-
 .../ViewModels/PhotoReasoningViewModel.swift  |  2 +-
 .../ViewModels/SummarizeViewModel.swift       |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift       | 28 +++++++++----------
 .../Sources/VertexAIComponent.swift           | 10 +++----
 .../Tests/Unit/VertexAIAPITests.swift         |  4 +--
 .../Tests/Unit/VertexComponentTests.swift     | 10 +++----
 7 files changed, 29 insertions(+), 29 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 81667be2514..883cefb359f 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,7 +36,7 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index 1fb0097a808..bfa1cfa9028 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,7 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    let vertexAI = VertexAI.vertexAI(region: "us-central1")
+    let vertexAI = VertexAI.vertexAI(location: "us-central1")
     model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index e3c78d09060..a90e1cf15b8 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,7 +32,7 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index a0df2ab015e..3e5f5e67050 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,35 +25,35 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
-  ///  - Parameter region: The region identifier, e.g., `us-central1`; see
+  ///  - Parameter location: The region identifier, e.g., `us-central1`; see
   ///     [Vertex AI
   ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
   ///     for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI(region: String) -> VertexAI {
+  public static func vertexAI(location: String) -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app, region: region)
+    return vertexAI(app: app, location: location)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
-  ///   - region: The region identifier, e.g., `us-central1`; see
+  ///   - location: The region identifier, e.g., `us-central1`; see
   ///     [Vertex AI
   ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
   ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp, region: String) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp, location: String) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI(region)
+    return provider.vertexAI(location)
   }
 
   /// Initializes a generative model with the given parameters.
@@ -73,7 +73,7 @@ public class VertexAI: NSObject {
                               tools: [Tool]? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
-    let modelResourceName = modelResourceName(modelName: modelName, region: region)
+    let modelResourceName = modelResourceName(modelName: modelName, location: location)
 
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
@@ -97,29 +97,29 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
-  let region: String
+  let location: String
 
-  init(app: FirebaseApp, region: String) {
+  init(app: FirebaseApp, location: String) {
     self.app = app
-    self.region = region
+    self.location = location
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
   }
 
-  private func modelResourceName(modelName: String, region: String) -> String {
+  private func modelResourceName(modelName: String, location: String) -> String {
     if modelName.contains("/") {
       return modelName
     }
     guard let projectID = app.options.projectID else {
       fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
     }
-    guard !region.isEmpty else {
+    guard !location.isEmpty else {
       fatalError("""
-      No region specified; see
+      No location specified; see
       https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
       list of available regions.
       """)
     }
 
-    return "projects/\(projectID)/locations/\(region)/publishers/google/models/\(modelName)"
+    return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
   }
 }
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index af8dbb56271..a6d487e1208 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI(_ region: String) -> VertexAI
+  @objc func vertexAI(_ location: String) -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -64,17 +64,17 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   // MARK: - VertexAIProvider conformance
 
-  func vertexAI(_ region: String) -> VertexAI {
+  func vertexAI(_ location: String) -> VertexAI {
     os_unfair_lock_lock(&instancesLock)
 
     // Unlock before the function returns.
     defer { os_unfair_lock_unlock(&instancesLock) }
 
-    if let instance = instances[region] {
+    if let instance = instances[location] {
       return instance
     }
-    let newInstance = VertexAI(app: app, region: region)
-    instances[region] = newInstance
+    let newInstance = VertexAI(app: app, location: location)
+    instances[location] = newInstance
     return newInstance
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index a4983eb8834..80236f405de 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -34,10 +34,10 @@ final class VertexAIAPITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Instantiate Vertex AI SDK - Default App
-    let vertexAI = VertexAI.vertexAI(region: "my-region")
+    let vertexAI = VertexAI.vertexAI(location: "my-location")
 
     // Instantiate Vertex AI SDK - Custom App
-    let _ = VertexAI.vertexAI(app: app!, region: "my-region")
+    let _ = VertexAI.vertexAI(app: app!, location: "my-location")
 
     // Permutations without optional arguments.
 
diff --git a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
index e28b747b65c..20f2b15bee8 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
@@ -48,7 +48,7 @@ class VertexComponentTests: XCTestCase {
   func testVertexInstanceCreation() throws {
     let app = try XCTUnwrap(VertexComponentTests.app)
     let component = VertexAIComponent(app: app)
-    let vertex = component.vertexAI("my-region")
+    let vertex = component.vertexAI("my-location")
     XCTAssertNotNil(vertex)
   }
 
@@ -82,14 +82,14 @@ class VertexComponentTests: XCTestCase {
                                                             in: container)
     XCTAssertNotNil(provider)
 
-    let vertex1 = provider?.vertexAI("randomRegion")
-    let vertex2 = provider?.vertexAI("randomRegion")
+    let vertex1 = provider?.vertexAI("randomLocation")
+    let vertex2 = provider?.vertexAI("randomLocation")
     XCTAssertNotNil(vertex1)
 
     // Ensure they're the same instance.
     XCTAssert(vertex1 === vertex2)
 
-    let vertex3 = provider?.vertexAI("differentRegion")
+    let vertex3 = provider?.vertexAI("differentLocation")
     XCTAssertNotNil(vertex3)
 
     XCTAssert(vertex1 !== vertex3)
@@ -105,7 +105,7 @@ class VertexComponentTests: XCTestCase {
       options.projectID = "myProjectID"
       let app1 = FirebaseApp(instanceWithName: "transitory app", options: options)
       weakApp = try XCTUnwrap(app1)
-      let vertex = VertexAI(app: app1, region: "transitory region")
+      let vertex = VertexAI(app: app1, location: "transitory location")
       weakVertex = vertex
       XCTAssertNotNil(weakVertex)
     }

From c3ad6f1c5d2d0a7a922c2f64e4dc91e157d2577f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 2 Apr 2024 14:48:35 +0000
Subject: [PATCH 030/124] Update to prod endpoint in Vertex AI (#12657)

---
 FirebaseVertexAI/Sources/GenerativeAISwift.swift                | 2 +-
 .../Unit/GenerateContentResponses/unary-failure-api-key.json    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
index 0f0af2bc3e2..933a19c4759 100644
--- a/FirebaseVertexAI/Sources/GenerativeAISwift.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -21,5 +21,5 @@ import Foundation
 /// Constants associated with the GenerativeAISwift SDK
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public enum GenerativeAISwift {
-  static let baseURL = "https://staging-firebaseml.sandbox.googleapis.com"
+  static let baseURL = "https://firebaseml.googleapis.com"
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
index d3b5677366e..e9b472fd031 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -9,7 +9,7 @@
         "reason": "API_KEY_INVALID",
         "domain": "googleapis.com",
         "metadata": {
-          "service": "staging-firebaseml.sandbox.googleapis.com"
+          "service": "firebaseml.googleapis.com"
         }
       },
       {

From cbc3e8caaf7e4275e13761aac4b52c8ed05a46bd Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 2 Apr 2024 07:45:50 -0800
Subject: [PATCH 031/124] [VertexAI] Consolidate to single sample app (#12662)

---
 .../Sample/ChatSample/ChatSampleApp.swift     |  36 --
 .../GenerativeAIMultimodalSampleApp.swift     |  31 --
 .../project.pbxproj                           | 483 ------------------
 .../GenerativeAITextSampleApp.swift           |  31 --
 4 files changed, 581 deletions(-)
 delete mode 100644 FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
 delete mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
 delete mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift

diff --git a/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift b/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
deleted file mode 100644
index 7d7e02d7e3c..00000000000
--- a/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
+++ /dev/null
@@ -1,36 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import FirebaseCore
-import FirebaseVertexAI
-import SwiftUI
-
-@main
-struct ChatSampleApp: App {
-  @StateObject
-  var viewModel = ConversationViewModel()
-
-  init() {
-    FirebaseApp.configure()
-  }
-
-  var body: some Scene {
-    WindowGroup {
-      NavigationStack {
-        ConversationScreen()
-          .environmentObject(viewModel)
-      }
-    }
-  }
-}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
deleted file mode 100644
index 49565de52d1..00000000000
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
+++ /dev/null
@@ -1,31 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import FirebaseCore
-import SwiftUI
-
-@main
-struct GenerativeAIMultimodalSampleApp: App {
-  init() {
-    FirebaseApp.configure()
-  }
-
-  var body: some Scene {
-    WindowGroup {
-      NavigationStack {
-        PhotoReasoningScreen()
-      }
-    }
-  }
-}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 9efade7342d..ac930a76d95 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -9,34 +9,13 @@
 /* Begin PBXBuildFile section */
 		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
 		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
-		868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33692BB4771900304BB1 /* FirebaseAppCheck */; };
-		868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */; };
-		868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336D2BB4773600304BB1 /* FirebaseAppCheck */; };
-		868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */; };
-		868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33712BB4775D00304BB1 /* FirebaseAppCheck */; };
-		868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */; };
-		880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */; };
-		88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
-		88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
-		88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88209C232B0FBE1700F64795 /* MarkdownUI */; };
-		88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
 		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
 		8848C8352B0D04BC007B434F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8342B0D04BC007B434F /* ContentView.swift */; };
 		8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8362B0D04BD007B434F /* Assets.xcassets */; };
 		8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8392B0D04BD007B434F /* Preview Assets.xcassets */; };
-		8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */; };
-		8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84A2B0D051F007B434F /* Assets.xcassets */; };
-		8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84D2B0D051F007B434F /* Preview Assets.xcassets */; };
-		8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */; };
-		8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85C2B0D056D007B434F /* Assets.xcassets */; };
-		8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85F2B0D056D007B434F /* Preview Assets.xcassets */; };
 		886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
 		886F95D62B17BA010036F07A /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
 		886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95D72B17BA420036F07A /* MarkdownUI */; };
@@ -48,19 +27,6 @@
 		886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
 		886F95E12B17D5010036F07A /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
 		886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95E22B17D6630036F07A /* GenerativeAIUIComponents */; };
-		889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
-		88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A91D2B0FC55100424728 /* MarkdownUI */; };
-		88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */; };
-		88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88D9474C2B14F27E008B5580 /* MarkdownUI */; };
-		88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F442B110D5300C08E95 /* ChatSampleApp.swift */; };
-		88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F482B110D5400C08E95 /* Assets.xcassets */; };
-		88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */; };
-		88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
-		88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
-		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
-		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
-		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
-		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
 /* End PBXBuildFile section */
 
 /* Begin PBXFileReference section */
@@ -75,18 +41,12 @@
 		8848C8342B0D04BC007B434F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = "<group>"; };
 		8848C8362B0D04BD007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C8392B0D04BD007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
-		8848C8442B0D051E007B434F /* GenerativeAITextSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAITextSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAITextSampleApp.swift; sourceTree = "<group>"; };
 		8848C84A2B0D051F007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C84D2B0D051F007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
-		8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAIMultimodalSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAIMultimodalSampleApp.swift; sourceTree = "<group>"; };
 		8848C85C2B0D056D007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C85F2B0D056D007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
 		889873842B208563005B4896 /* ErrorDetailsView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ErrorDetailsView.swift; sourceTree = "<group>"; };
 		88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = GenerativeAIUIComponents; sourceTree = "<group>"; };
-		88E10F422B110D5300C08E95 /* ChatSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = ChatSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		88E10F442B110D5300C08E95 /* ChatSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatSampleApp.swift; sourceTree = "<group>"; };
 		88E10F482B110D5400C08E95 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
 		88E10F542B1112CA00C08E95 /* ConversationScreen.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ConversationScreen.swift; sourceTree = "<group>"; };
@@ -108,38 +68,6 @@
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
-		8848C8412B0D051E007B434F /* Frameworks */ = {
-			isa = PBXFrameworksBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
-				868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */,
-				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		8848C8532B0D056C007B434F /* Frameworks */ = {
-			isa = PBXFrameworksBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
-				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
-				868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */,
-				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		88E10F3F2B110D5300C08E95 /* Frameworks */ = {
-			isa = PBXFrameworksBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
-				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
-				868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */,
-				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
 /* End PBXFrameworksBuildPhase section */
 
 /* Begin PBXGroup section */
@@ -200,9 +128,6 @@
 			isa = PBXGroup;
 			children = (
 				8848C82F2B0D04BC007B434F /* GenerativeAISample.app */,
-				8848C8442B0D051E007B434F /* GenerativeAITextSample.app */,
-				8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */,
-				88E10F422B110D5300C08E95 /* ChatSample.app */,
 			);
 			name = Products;
 			sourceTree = "<group>";
@@ -231,7 +156,6 @@
 			children = (
 				88209C1C2B0FBDC300F64795 /* ViewModels */,
 				88209C1A2B0FBDC300F64795 /* Screens */,
-				8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */,
 				8848C84A2B0D051F007B434F /* Assets.xcassets */,
 				8848C84C2B0D051F007B434F /* Preview Content */,
 			);
@@ -251,7 +175,6 @@
 			children = (
 				8802666E2B0FC39000CF7CB6 /* ViewModels */,
 				880266742B0FC39000CF7CB6 /* Screens */,
-				8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */,
 				8848C85C2B0D056D007B434F /* Assets.xcassets */,
 				8848C85E2B0D056D007B434F /* Preview Content */,
 			);
@@ -273,7 +196,6 @@
 				88E10F502B11123600C08E95 /* ViewModels */,
 				88E10F512B11124100C08E95 /* Views */,
 				88E10F532B1112B900C08E95 /* Screens */,
-				88E10F442B110D5300C08E95 /* ChatSampleApp.swift */,
 				88E10F482B110D5400C08E95 /* Assets.xcassets */,
 				88E10F4A2B110D5400C08E95 /* Preview Content */,
 			);
@@ -349,74 +271,6 @@
 			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
 			productType = "com.apple.product-type.application";
 		};
-		8848C8432B0D051E007B434F /* GenerativeAITextSample */ = {
-			isa = PBXNativeTarget;
-			buildConfigurationList = 8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */;
-			buildPhases = (
-				8848C8402B0D051E007B434F /* Sources */,
-				8848C8412B0D051E007B434F /* Frameworks */,
-				8848C8422B0D051E007B434F /* Resources */,
-			);
-			buildRules = (
-			);
-			dependencies = (
-			);
-			name = GenerativeAITextSample;
-			packageProductDependencies = (
-				88209C232B0FBE1700F64795 /* MarkdownUI */,
-				868A33692BB4771900304BB1 /* FirebaseAppCheck */,
-				868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */,
-			);
-			productName = GenerativeAITextSample;
-			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
-			productType = "com.apple.product-type.application";
-		};
-		8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */ = {
-			isa = PBXNativeTarget;
-			buildConfigurationList = 8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */;
-			buildPhases = (
-				8848C8522B0D056C007B434F /* Sources */,
-				8848C8532B0D056C007B434F /* Frameworks */,
-				8848C8542B0D056C007B434F /* Resources */,
-			);
-			buildRules = (
-			);
-			dependencies = (
-			);
-			name = GenerativeAIMultimodalSample;
-			packageProductDependencies = (
-				88B8A91D2B0FC55100424728 /* MarkdownUI */,
-				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
-				868A336D2BB4773600304BB1 /* FirebaseAppCheck */,
-				868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */,
-			);
-			productName = GenerativeAIMultimodalSample;
-			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
-			productType = "com.apple.product-type.application";
-		};
-		88E10F412B110D5300C08E95 /* ChatSample */ = {
-			isa = PBXNativeTarget;
-			buildConfigurationList = 88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */;
-			buildPhases = (
-				88E10F3E2B110D5300C08E95 /* Sources */,
-				88E10F3F2B110D5300C08E95 /* Frameworks */,
-				88E10F402B110D5300C08E95 /* Resources */,
-			);
-			buildRules = (
-			);
-			dependencies = (
-			);
-			name = ChatSample;
-			packageProductDependencies = (
-				88D9474C2B14F27E008B5580 /* MarkdownUI */,
-				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
-				868A33712BB4775D00304BB1 /* FirebaseAppCheck */,
-				868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */,
-			);
-			productName = ChatSample;
-			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
-			productType = "com.apple.product-type.application";
-		};
 /* End PBXNativeTarget section */
 
 /* Begin PBXProject section */
@@ -430,15 +284,6 @@
 					8848C82E2B0D04BC007B434F = {
 						CreatedOnToolsVersion = 15.1;
 					};
-					8848C8432B0D051E007B434F = {
-						CreatedOnToolsVersion = 15.1;
-					};
-					8848C8552B0D056C007B434F = {
-						CreatedOnToolsVersion = 15.1;
-					};
-					88E10F412B110D5300C08E95 = {
-						CreatedOnToolsVersion = 15.1;
-					};
 				};
 			};
 			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */;
@@ -460,9 +305,6 @@
 			projectRoot = "";
 			targets = (
 				8848C82E2B0D04BC007B434F /* GenerativeAISample */,
-				8848C8432B0D051E007B434F /* GenerativeAITextSample */,
-				8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */,
-				88E10F412B110D5300C08E95 /* ChatSample */,
 			);
 		};
 /* End PBXProject section */
@@ -478,36 +320,6 @@
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
-		8848C8422B0D051E007B434F /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */,
-				8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */,
-				869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		8848C8542B0D056C007B434F /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */,
-				8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */,
-				869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		88E10F402B110D5300C08E95 /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */,
-				88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */,
-				869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
 /* End PBXResourcesBuildPhase section */
 
 /* Begin PBXSourcesBuildPhase section */
@@ -531,41 +343,6 @@
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
-		8848C8402B0D051E007B434F /* Sources */ = {
-			isa = PBXSourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */,
-				8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */,
-				88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		8848C8522B0D056C007B434F /* Sources */ = {
-			isa = PBXSourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */,
-				880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */,
-				8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		88E10F3E2B110D5300C08E95 /* Sources */ = {
-			isa = PBXSourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */,
-				88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */,
-				88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */,
-				88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */,
-				88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */,
-				88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */,
-				88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */,
-				889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
 /* End PBXSourcesBuildPhase section */
 
 /* Begin XCBuildConfiguration section */
@@ -748,192 +525,6 @@
 			};
 			name = Release;
 		};
-		8848C8502B0D051F007B434F /* Debug */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Debug;
-		};
-		8848C8512B0D051F007B434F /* Release */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Release;
-		};
-		8848C8622B0D056D007B434F /* Debug */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Debug;
-		};
-		8848C8632B0D056D007B434F /* Release */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Release;
-		};
-		88E10F4D2B110D5400C08E95 /* Debug */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Debug;
-		};
-		88E10F4E2B110D5400C08E95 /* Release */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Release;
-		};
 /* End XCBuildConfiguration section */
 
 /* Begin XCConfigurationList section */
@@ -955,33 +546,6 @@
 			defaultConfigurationIsVisible = 0;
 			defaultConfigurationName = Release;
 		};
-		8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */ = {
-			isa = XCConfigurationList;
-			buildConfigurations = (
-				8848C8502B0D051F007B434F /* Debug */,
-				8848C8512B0D051F007B434F /* Release */,
-			);
-			defaultConfigurationIsVisible = 0;
-			defaultConfigurationName = Release;
-		};
-		8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */ = {
-			isa = XCConfigurationList;
-			buildConfigurations = (
-				8848C8622B0D056D007B434F /* Debug */,
-				8848C8632B0D056D007B434F /* Release */,
-			);
-			defaultConfigurationIsVisible = 0;
-			defaultConfigurationName = Release;
-		};
-		88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */ = {
-			isa = XCConfigurationList;
-			buildConfigurations = (
-				88E10F4D2B110D5400C08E95 /* Debug */,
-				88E10F4E2B110D5400C08E95 /* Release */,
-			);
-			defaultConfigurationIsVisible = 0;
-			defaultConfigurationName = Release;
-		};
 /* End XCConfigurationList section */
 
 /* Begin XCLocalSwiftPackageReference section */
@@ -1019,35 +583,6 @@
 			isa = XCSwiftPackageProductDependency;
 			productName = "FirebaseVertexAI-Preview";
 		};
-		868A33692BB4771900304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = "FirebaseVertexAI-Preview";
-		};
-		868A336D2BB4773600304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = "FirebaseVertexAI-Preview";
-		};
-		868A33712BB4775D00304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = "FirebaseVertexAI-Preview";
-		};
-		88209C232B0FBE1700F64795 /* MarkdownUI */ = {
-			isa = XCSwiftPackageProductDependency;
-			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
-			productName = MarkdownUI;
-		};
 		886F95D72B17BA420036F07A /* MarkdownUI */ = {
 			isa = XCSwiftPackageProductDependency;
 			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
@@ -1057,24 +592,6 @@
 			isa = XCSwiftPackageProductDependency;
 			productName = GenerativeAIUIComponents;
 		};
-		88B8A91D2B0FC55100424728 /* MarkdownUI */ = {
-			isa = XCSwiftPackageProductDependency;
-			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
-			productName = MarkdownUI;
-		};
-		88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = GenerativeAIUIComponents;
-		};
-		88D9474C2B14F27E008B5580 /* MarkdownUI */ = {
-			isa = XCSwiftPackageProductDependency;
-			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
-			productName = MarkdownUI;
-		};
-		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = GenerativeAIUIComponents;
-		};
 /* End XCSwiftPackageProductDependency section */
 	};
 	rootObject = 8848C8272B0D04BC007B434F /* Project object */;
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
deleted file mode 100644
index c800e1f7be8..00000000000
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
+++ /dev/null
@@ -1,31 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import FirebaseCore
-import SwiftUI
-
-@main
-struct GenerativeAITextSampleApp: App {
-  init() {
-    FirebaseApp.configure()
-  }
-
-  var body: some Scene {
-    WindowGroup {
-      NavigationStack {
-        SummarizeScreen()
-      }
-    }
-  }
-}

From f1f233848ef7182e82f15c9a6c4b80e4e230862a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 2 Apr 2024 17:11:08 +0000
Subject: [PATCH 032/124] Add platform tags in `x-goog-api-client` (#12680)

---
 FirebaseCore/Sources/FIRApp.m                       |  2 +-
 FirebaseVertexAI/Sources/GenerativeAIService.swift  | 13 ++++++++++---
 .../Tests/Unit/GenerativeModelTests.swift           |  5 +++++
 3 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/FirebaseCore/Sources/FIRApp.m b/FirebaseCore/Sources/FIRApp.m
index a8cc38c12de..12afc134321 100644
--- a/FirebaseCore/Sources/FIRApp.m
+++ b/FirebaseCore/Sources/FIRApp.m
@@ -829,7 +829,7 @@ + (void)registerSwiftComponents {
     @"FIRSessions" : @"fire-ses",
     @"FIRFunctionsComponent" : @"fire-fun",
     @"FIRStorageComponent" : @"fire-str",
-    @"FIRVertexAIComponent" : @"fire-vtx",
+    @"FIRVertexAIComponent" : @"fire-vertex",
   };
   for (NSString *className in swiftComponents.allKeys) {
     Class klass = NSClassFromString(className);
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index e1df1010c06..b3687c3c013 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -18,6 +18,12 @@ import Foundation
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 struct GenerativeAIService {
+  /// The language of the SDK in the format "gl-<language>/<version>" where version may be blank.
+  static let languageTag = "gl-swift/"
+
+  /// The Firebase SDK version in the format "fire/<version>".
+  static let firebaseVersionTag = "fire/\(FirebaseVersion())"
+
   /// Gives permission to talk to the backend.
   private let apiKey: String
 
@@ -155,9 +161,10 @@ struct GenerativeAIService {
     var urlRequest = URLRequest(url: request.url)
     urlRequest.httpMethod = "POST"
     urlRequest.setValue(apiKey, forHTTPHeaderField: "x-goog-api-key")
-    // TODO: Determine the right client header to use.
-    //    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version))",
-    //                        forHTTPHeaderField: "x-goog-api-client")
+    urlRequest.setValue(
+      "\(GenerativeAIService.languageTag) \(GenerativeAIService.firebaseVersionTag)",
+      forHTTPHeaderField: "x-goog-api-client"
+    )
     urlRequest.setValue("application/json", forHTTPHeaderField: "Content-Type")
 
     if let appCheck {
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 123299ac1a7..8031fb95dcf 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseCore
 import XCTest
 
 @testable import FirebaseVertexAI
@@ -1058,6 +1059,10 @@ final class GenerativeModelTests: XCTestCase {
       let requestURL = try XCTUnwrap(request.url)
       XCTAssertEqual(requestURL.path.occurrenceCount(of: "models/"), 1)
       XCTAssertEqual(request.timeoutInterval, timeout)
+      let apiClientTags = try XCTUnwrap(request.value(forHTTPHeaderField: "x-goog-api-client"))
+        .components(separatedBy: " ")
+      XCTAssert(apiClientTags.contains(GenerativeAIService.languageTag))
+      XCTAssert(apiClientTags.contains(GenerativeAIService.firebaseVersionTag))
       XCTAssertEqual(request.value(forHTTPHeaderField: "X-Firebase-AppCheck"), appCheckToken)
       let response = try XCTUnwrap(HTTPURLResponse(
         url: requestURL,

From 1b67b2c3843526bd1ee3705a0458a59553179e4e Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 2 Apr 2024 09:52:57 -0800
Subject: [PATCH 033/124] [vertex-ai] Fix for SwiftFormat 0.53.5 (#12683)

---
 .../Sources/GenerativeAIUIComponents/InputField.swift            | 1 +
 1 file changed, 1 insertion(+)

diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
index 317baaaf3c1..3f12ea65d8a 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import SwiftUI
+
 public struct InputField<Label>: View where Label: View {
   @Binding
   private var text: String

From 07caa9dc0da6f8327435a0c2668345dd5ef0232c Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 3 Apr 2024 16:12:40 +0000
Subject: [PATCH 034/124] Add `functionCalls` accessor to
 `GenerateContentResponse` (#12690)

---
 .../Sources/GenerateContentResponse.swift           | 13 +++++++++++++
 .../Tests/Unit/GenerativeModelTests.swift           |  5 +++++
 2 files changed, 18 insertions(+)

diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 4e02abb7f2c..f43da308b13 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -37,6 +37,19 @@ public struct GenerateContentResponse {
     return text
   }
 
+  /// Returns function calls found in any `Part`s of the first candidate of the response, if any.
+  public var functionCalls: [FunctionCall] {
+    guard let candidate = candidates.first else {
+      return []
+    }
+    return candidate.content.parts.compactMap { part in
+      guard case let .functionCall(functionCall) = part else {
+        return nil
+      }
+      return functionCall
+    }
+  }
+
   /// Initializer for SwiftUI previews or tests.
   public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback?) {
     self.candidates = candidates
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 8031fb95dcf..13cbb6df94e 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -70,6 +70,7 @@ final class GenerativeModelTests: XCTestCase {
     let partText = try XCTUnwrap(part.text)
     XCTAssertTrue(partText.hasPrefix("You can ask me a wide range of questions"))
     XCTAssertEqual(response.text, partText)
+    XCTAssertEqual(response.functionCalls, [])
   }
 
   func testGenerateContent_success_basicReplyShort() async throws {
@@ -90,6 +91,7 @@ final class GenerativeModelTests: XCTestCase {
     let part = try XCTUnwrap(candidate.content.parts.first)
     XCTAssertEqual(part.text, "Mountain View, California, United States")
     XCTAssertEqual(response.text, part.text)
+    XCTAssertEqual(response.functionCalls, [])
   }
 
   func testGenerateContent_success_citations() async throws {
@@ -195,6 +197,7 @@ final class GenerativeModelTests: XCTestCase {
     }
     XCTAssertEqual(functionCall.name, "current_time")
     XCTAssertTrue(functionCall.args.isEmpty)
+    XCTAssertEqual(response.functionCalls, [functionCall])
   }
 
   func testGenerateContent_success_functionCall_noArguments() async throws {
@@ -216,6 +219,7 @@ final class GenerativeModelTests: XCTestCase {
     }
     XCTAssertEqual(functionCall.name, "current_time")
     XCTAssertTrue(functionCall.args.isEmpty)
+    XCTAssertEqual(response.functionCalls, [functionCall])
   }
 
   func testGenerateContent_success_functionCall_withArguments() async throws {
@@ -241,6 +245,7 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(argX, .number(4))
     let argY = try XCTUnwrap(functionCall.args["y"])
     XCTAssertEqual(argY, .number(5))
+    XCTAssertEqual(response.functionCalls, [functionCall])
   }
 
   func testGenerateContent_appCheck_validToken() async throws {

From 7b85605a588af4b515edfa5a9533d9cfcd2f550e Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Thu, 4 Apr 2024 06:59:30 -0800
Subject: [PATCH 035/124] Vertex AI preview versioning (#12691)

---
 FirebaseCore/Sources/FIROptions.m         |  2 +-
 FirebaseCore/Tests/Unit/FIROptionsTest.m  | 12 ++++++------
 Package.swift                             |  2 +-
 SwiftPMTests/swift-test/all-imports.swift |  3 +--
 4 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/FirebaseCore/Sources/FIROptions.m b/FirebaseCore/Sources/FIROptions.m
index d46b657501c..c0625a8bac9 100644
--- a/FirebaseCore/Sources/FIROptions.m
+++ b/FirebaseCore/Sources/FIROptions.m
@@ -284,7 +284,7 @@ - (NSString *)libraryVersionID {
     // The unit tests are set up to catch anything that does not properly convert.
     NSString *version = FIRFirebaseVersion();
     NSArray *components = [version componentsSeparatedByString:@"."];
-    NSString *major = [components objectAtIndex:0];
+    NSString *major = [NSString stringWithFormat:@"%02d", [[components objectAtIndex:0] intValue]];
     NSString *minor = [NSString stringWithFormat:@"%02d", [[components objectAtIndex:1] intValue]];
     NSString *patch = [NSString stringWithFormat:@"%02d", [[components objectAtIndex:2] intValue]];
     kFIRLibraryVersionID = [NSString stringWithFormat:@"%@%@%@000", major, minor, patch];
diff --git a/FirebaseCore/Tests/Unit/FIROptionsTest.m b/FirebaseCore/Tests/Unit/FIROptionsTest.m
index 3de059f4191..fd5c901598f 100644
--- a/FirebaseCore/Tests/Unit/FIROptionsTest.m
+++ b/FirebaseCore/Tests/Unit/FIROptionsTest.m
@@ -624,7 +624,7 @@ - (void)testVersionConsistency {
   int minor = (versionString[2] - '0') * 10 + versionString[3] - '0';
   int patch = (versionString[4] - '0') * 10 + versionString[5] - '0';
   NSString *str = [NSString stringWithFormat:@"%d.%d.%d", major, minor, patch];
-  XCTAssertEqualObjects(str, FIRFirebaseVersion());
+  XCTAssertTrue([FIRFirebaseVersion() hasPrefix:str]);
 }
 
 // Repeat test with more Objective-C.
@@ -638,11 +638,11 @@ - (void)testVersionConsistency2 {
   NSRange major = NSMakeRange(0, 2);
   NSRange minor = NSMakeRange(2, 2);
   NSRange patch = NSMakeRange(4, 2);
-  NSString *str =
-      [NSString stringWithFormat:@"%@.%d.%d", [kFIRLibraryVersionID substringWithRange:major],
-                                 [[kFIRLibraryVersionID substringWithRange:minor] intValue],
-                                 [[kFIRLibraryVersionID substringWithRange:patch] intValue]];
-  XCTAssertEqualObjects(str, FIRFirebaseVersion());
+  NSString *str = [NSString
+      stringWithFormat:@"%d.%d.%d", [[kFIRLibraryVersionID substringWithRange:major] intValue],
+                       [[kFIRLibraryVersionID substringWithRange:minor] intValue],
+                       [[kFIRLibraryVersionID substringWithRange:patch] intValue]];
+  XCTAssertTrue([FIRFirebaseVersion() hasPrefix:str]);
 }
 
 #pragma mark - Helpers
diff --git a/Package.swift b/Package.swift
index e401abcb750..ef6355a21d8 100644
--- a/Package.swift
+++ b/Package.swift
@@ -19,7 +19,7 @@
 import class Foundation.ProcessInfo
 import PackageDescription
 
-let firebaseVersion = "10.24.0"
+let firebaseVersion = "0.1.0-vertexai-preview"
 
 let package = Package(
   name: "Firebase",
diff --git a/SwiftPMTests/swift-test/all-imports.swift b/SwiftPMTests/swift-test/all-imports.swift
index 6486c1be833..c103f128d9d 100644
--- a/SwiftPMTests/swift-test/all-imports.swift
+++ b/SwiftPMTests/swift-test/all-imports.swift
@@ -76,9 +76,8 @@ class importTest: XCTestCase {
 
     let versionParts = FirebaseVersion().split(separator: ".")
     XCTAssert(versionParts.count == 3)
-    XCTAssertEqual(Int(versionParts[0]), 10)
+    XCTAssertNotNil(Int(versionParts[0]))
     XCTAssertNotNil(Int(versionParts[1]))
-    XCTAssertNotNil(Int(versionParts[2]))
 
     print("System version? Answer: \(GULAppEnvironmentUtil.systemVersion())")
   }

From 45c3c5985435504efbd56c826bd2bf540631cfbf Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 4 Apr 2024 19:15:09 +0000
Subject: [PATCH 036/124] Remove `location` from Vertex AI public API (#12706)

---
 .../ViewModels/ConversationViewModel.swift       |  2 +-
 .../ViewModels/PhotoReasoningViewModel.swift     |  3 +--
 .../ViewModels/SummarizeViewModel.swift          |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift          | 16 ++++------------
 .../Tests/Unit/VertexAIAPITests.swift            |  4 ++--
 5 files changed, 9 insertions(+), 18 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 883cefb359f..a7c3d21a960 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,7 +36,7 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.0-pro")
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index bfa1cfa9028..b27d518fe86 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,8 +44,7 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    let vertexAI = VertexAI.vertexAI(location: "us-central1")
-    model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
+    model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
   func reason() async {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index a90e1cf15b8..99b374efefa 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,7 +32,7 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 3e5f5e67050..1fcfd998b45 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,35 +25,27 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
-  ///  - Parameter location: The region identifier, e.g., `us-central1`; see
-  ///     [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
-  ///     for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI(location: String) -> VertexAI {
+  public static func vertexAI() -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app, location: location)
+    return vertexAI(app: app)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
-  ///   - location: The region identifier, e.g., `us-central1`; see
-  ///     [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
-  ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp, location: String) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI(location)
+    return provider.vertexAI("us-central1")
   }
 
   /// Initializes a generative model with the given parameters.
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 80236f405de..81b81414dcb 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -34,10 +34,10 @@ final class VertexAIAPITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Instantiate Vertex AI SDK - Default App
-    let vertexAI = VertexAI.vertexAI(location: "my-location")
+    let vertexAI = VertexAI.vertexAI()
 
     // Instantiate Vertex AI SDK - Custom App
-    let _ = VertexAI.vertexAI(app: app!, location: "my-location")
+    let _ = VertexAI.vertexAI(app: app!)
 
     // Permutations without optional arguments.
 

From 468013c8c6f9eb458dc3d1ba7f3d1422ff7479ea Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 4 Apr 2024 19:15:36 +0000
Subject: [PATCH 037/124] Remove `apiVersion` from `RequestOptions` in Vertex
 AI (#12707)

---
 FirebaseVertexAI/Sources/GenerativeAIRequest.swift | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
index 21f35b1b728..a3a198726d4 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
@@ -31,16 +31,14 @@ public struct RequestOptions {
   let timeout: TimeInterval?
 
   /// The API version to use in requests to the backend.
-  let apiVersion: String
+  let apiVersion = "v2beta"
 
   /// Initializes a request options object.
   ///
   /// - Parameters:
   ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
   ///   for a `URLRequest`.
-  ///   - apiVersion The API version to use in requests to the backend; defaults to "v2beta".
-  public init(timeout: TimeInterval? = nil, apiVersion: String = "v2beta") {
+  public init(timeout: TimeInterval? = nil) {
     self.timeout = timeout
-    self.apiVersion = apiVersion
   }
 }

From df34f3b7c38f4fa9918b2f853b47efe29ea4f5a7 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 8 Apr 2024 17:36:26 -0400
Subject: [PATCH 038/124] Add placeholder `GoogleService-Info.plist` to Vertex
 AI sample (#12724)

---
 .../project.pbxproj                           |  4 +-
 .../GenerativeAISampleApp.swift               |  8 ++++
 .../Sample/GoogleService-Info.plist           | 40 +++++++++++++++++++
 FirebaseVertexAI/Sample/README.md             | 16 ++++++++
 4 files changed, 66 insertions(+), 2 deletions(-)
 create mode 100644 FirebaseVertexAI/Sample/GoogleService-Info.plist
 create mode 100644 FirebaseVertexAI/Sample/README.md

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index ac930a76d95..01336b9cf19 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -487,7 +487,7 @@
 					"@executable_path/Frameworks",
 				);
 				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_BUNDLE_IDENTIFIER = com.google.firebase.VertexAISample;
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				SWIFT_EMIT_LOC_STRINGS = YES;
 				SWIFT_VERSION = 5.0;
@@ -517,7 +517,7 @@
 					"@executable_path/Frameworks",
 				);
 				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_BUNDLE_IDENTIFIER = com.google.firebase.VertexAISample;
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				SWIFT_EMIT_LOC_STRINGS = YES;
 				SWIFT_VERSION = 5.0;
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
index 7c709111b7f..509087a5e07 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
@@ -19,6 +19,14 @@ import SwiftUI
 struct GenerativeAISampleApp: App {
   init() {
     FirebaseApp.configure()
+
+    if let firebaseApp = FirebaseApp.app(), firebaseApp.options.projectID == "mockproject-1234" {
+      guard let bundleID = Bundle.main.bundleIdentifier else { fatalError() }
+      fatalError("You must create and/or download a valid `GoogleService-Info.plist` file for"
+        + " \(bundleID) from https://console.firebase.google.com to run this sample. Replace the"
+        + " existing `GoogleService-Info.plist` file in the `FirebaseVertexAI/Sample` directory"
+        + " with this new file.")
+    }
   }
 
   var body: some Scene {
diff --git a/FirebaseVertexAI/Sample/GoogleService-Info.plist b/FirebaseVertexAI/Sample/GoogleService-Info.plist
new file mode 100644
index 00000000000..b026413cc39
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GoogleService-Info.plist
@@ -0,0 +1,40 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>AD_UNIT_ID_FOR_BANNER_TEST</key>
+	<string>ca-app-pub-3940256099942544/2934735716</string>
+	<key>AD_UNIT_ID_FOR_INTERSTITIAL_TEST</key>
+	<string>ca-app-pub-3940256099942544/4411468910</string>
+	<key>API_KEY</key>
+	<string>AIzaSyAzlj4APqi5S58nFtE52Da0fYBOHA2MhaY</string>
+	<key>BUNDLE_ID</key>
+	<string>com.google.firebase.VertexAISample</string>
+	<key>CLIENT_ID</key>
+	<string>123456789000-hjugbg6ud799v4c49dim8ce2usclthar.apps.googleusercontent.com</string>
+	<key>DATABASE_URL</key>
+	<string>https://mockproject-1234.firebaseio.com</string>
+	<key>GCM_SENDER_ID</key>
+	<string>123456789000</string>
+	<key>GOOGLE_APP_ID</key>
+	<string>1:123456789000:ios:f1bf012572b04063</string>
+	<key>IS_ADS_ENABLED</key>
+	<true/>
+	<key>IS_ANALYTICS_ENABLED</key>
+	<true/>
+	<key>IS_APPINVITE_ENABLED</key>
+	<true/>
+	<key>IS_GCM_ENABLED</key>
+	<true/>
+	<key>IS_SIGNIN_ENABLED</key>
+	<true/>
+	<key>PLIST_VERSION</key>
+	<string>1</string>
+	<key>PROJECT_ID</key>
+	<string>mockproject-1234</string>
+	<key>REVERSED_CLIENT_ID</key>
+	<string>com.googleusercontent.apps.123456789000-hjugbg6ud799v4c49dim8ce2usclthar</string>
+	<key>STORAGE_BUCKET</key>
+	<string>mockproject-1234.appspot.com</string>
+</dict>
+</plist>
diff --git a/FirebaseVertexAI/Sample/README.md b/FirebaseVertexAI/Sample/README.md
new file mode 100644
index 00000000000..a4a8b458df3
--- /dev/null
+++ b/FirebaseVertexAI/Sample/README.md
@@ -0,0 +1,16 @@
+# Vertex AI for Firebase Sample App
+
+You can try out the SDK quickly, see a complete implementation of various use
+cases, or use the sample app if you don't have your own Apple platforms app. To
+use the sample app, you'll need to perform the following steps:
+1. Download
+   [vertexai-preview-0.1.0.zip](https://github.com/firebase/firebase-ios-sdk/archive/refs/heads/vertexai-preview-0.1.0.zip)
+   for the latest release of the SDK
+1. Extract the zip and open `GenerativeAISample.xcodeproj` in the
+   `FirebaseVertexAI/Sample` directory
+1. [Register the sample app](https://firebase.google.com/docs/ios/setup#register-app)
+   in your Firebase project
+   - The default bundle ID is `com.google.firebase.VertexAISample`
+1. [Download the `GoogleService-Info.plist`](https://firebase.google.com/docs/ios/setup#add-config-file)
+   to the `FirebaseVertexAI/Sample` directory, overwriting the placeholder file
+   with the same name

From 110f391462c39a850ab58f1183629f19f8ed8768 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 10 Apr 2024 14:27:48 -0400
Subject: [PATCH 039/124] Handle `totalBillableCharacters` optional decoding in
 Vertex AI (#12715)

---
 .../Sources/CountTokensRequest.swift          | 24 +++++++++++++++++--
 .../success-no-billable-characters.json       |  3 +++
 .../Tests/Unit/GenerativeModelTests.swift     | 15 ++++++++++++
 3 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100644 FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json

diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
index d294212658c..5072412cf9b 100644
--- a/FirebaseVertexAI/Sources/CountTokensRequest.swift
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -39,10 +39,30 @@ extension CountTokensRequest: GenerativeAIRequest {
 
 /// The model's response to a count tokens request.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct CountTokensResponse: Decodable {
+public struct CountTokensResponse {
   /// The total number of tokens in the input given to the model as a prompt.
   public let totalTokens: Int
 
-  /// The total number of billable characters in the input given to the model as a prompt.
+  /// The total number of billable characters in the text input given to the model as a prompt.
+  ///
+  /// > Important: This does not include billable image, video or other non-text input. See
+  /// [Vertex AI pricing](https://cloud.google.com/vertex-ai/generative-ai/pricing) for details.
   public let totalBillableCharacters: Int
 }
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case totalTokens
+    case totalBillableCharacters
+  }
+
+  public init(from decoder: any Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    totalTokens = try container.decode(Int.self, forKey: .totalTokens)
+    totalBillableCharacters = try container.decodeIfPresent(
+      Int.self,
+      forKey: .totalBillableCharacters
+    ) ?? 0
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json
new file mode 100644
index 00000000000..03425d51db0
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json
@@ -0,0 +1,3 @@
+{
+  "totalTokens": 258
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 13cbb6df94e..92585266891 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -946,6 +946,21 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(response.totalBillableCharacters, 16)
   }
 
+  func testCountTokens_succeeds_noBillableCharacters() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "success-no-billable-characters",
+      withExtension: "json"
+    )
+
+    let response = try await model.countTokens(ModelContent.Part.data(
+      mimetype: "image/jpeg",
+      Data()
+    ))
+
+    XCTAssertEqual(response.totalTokens, 258)
+    XCTAssertEqual(response.totalBillableCharacters, 0)
+  }
+
   func testCountTokens_modelNotFound() async throws {
     MockURLProtocol.requestHandler = try httpRequestHandler(
       forResource: "failure-model-not-found", withExtension: "json",

From 4c077572353f2f4e8a06710ba01ed27863690735 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 10 Apr 2024 16:30:24 -0400
Subject: [PATCH 040/124] Add function calling sample in Vertex AI (#12745)

---
 .../Screens/FunctionCallingScreen.swift       | 128 +++++++++
 .../ViewModels/FunctionCallingViewModel.swift | 264 ++++++++++++++++++
 .../project.pbxproj                           |  32 +++
 .../GenerativeAISample/ContentView.swift      |   8 +
 4 files changed, 432 insertions(+)
 create mode 100644 FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
new file mode 100644
index 00000000000..30d222e29bc
--- /dev/null
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
@@ -0,0 +1,128 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import GenerativeAIUIComponents
+import SwiftUI
+
+struct FunctionCallingScreen: View {
+  @EnvironmentObject
+  var viewModel: FunctionCallingViewModel
+
+  @State
+  private var userPrompt = "What is 100 Euros in U.S. Dollars?"
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      ScrollViewReader { scrollViewProxy in
+        List {
+          Text("Interact with a currency conversion API using function calling in Gemini.")
+          ForEach(viewModel.messages) { message in
+            MessageView(message: message)
+          }
+          if let error = viewModel.error {
+            ErrorView(error: error)
+              .tag("errorView")
+          }
+        }
+        .listStyle(.plain)
+        .onChange(of: viewModel.messages, perform: { newValue in
+          if viewModel.hasError {
+            // Wait for a short moment to make sure we can actually scroll to the bottom.
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo("errorView", anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          } else {
+            guard let lastMessage = viewModel.messages.last else { return }
+
+            // Wait for a short moment to make sure we can actually scroll to the bottom.
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo(lastMessage.id, anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          }
+        })
+      }
+      InputField("Message...", text: $userPrompt) {
+        Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
+          .font(.title)
+      }
+      .focused($focusedField, equals: .message)
+      .onSubmit { sendOrStop() }
+    }
+    .toolbar {
+      ToolbarItem(placement: .primaryAction) {
+        Button(action: newChat) {
+          Image(systemName: "square.and.pencil")
+        }
+      }
+    }
+    .navigationTitle("Function Calling")
+    .onAppear {
+      focusedField = .message
+    }
+  }
+
+  private func sendMessage() {
+    Task {
+      let prompt = userPrompt
+      userPrompt = ""
+      await viewModel.sendMessage(prompt, streaming: true)
+    }
+  }
+
+  private func sendOrStop() {
+    if viewModel.busy {
+      viewModel.stop()
+    } else {
+      sendMessage()
+    }
+  }
+
+  private func newChat() {
+    viewModel.startNewChat()
+  }
+}
+
+struct FunctionCallingScreen_Previews: PreviewProvider {
+  struct ContainerView: View {
+    @EnvironmentObject
+    var viewModel: FunctionCallingViewModel
+
+    var body: some View {
+      FunctionCallingScreen()
+        .onAppear {
+          viewModel.messages = ChatMessage.samples
+        }
+    }
+  }
+
+  static var previews: some View {
+    NavigationStack {
+      FunctionCallingScreen().environmentObject(FunctionCallingViewModel())
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
new file mode 100644
index 00000000000..f18c0e85fc1
--- /dev/null
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
@@ -0,0 +1,264 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import UIKit
+
+@MainActor
+class FunctionCallingViewModel: ObservableObject {
+  /// This array holds both the user's and the system's chat messages
+  @Published var messages = [ChatMessage]()
+
+  /// Indicates we're waiting for the model to finish
+  @Published var busy = false
+
+  @Published var error: Error?
+  var hasError: Bool {
+    return error != nil
+  }
+
+  /// Function calls pending processing
+  private var functionCalls = [FunctionCall]()
+
+  private var model: GenerativeModel
+  private var chat: Chat
+
+  private var chatTask: Task<Void, Never>?
+
+  init() {
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro",
+      tools: [Tool(functionDeclarations: [
+        FunctionDeclaration(
+          name: "get_exchange_rate",
+          description: "Get the exchange rate for currencies between countries",
+          parameters: [
+            "currency_from": Schema(
+              type: .string,
+              format: "enum",
+              description: "The currency to convert from in ISO 4217 format",
+              enumValues: ["USD", "EUR", "JPY", "GBP", "AUD", "CAD"]
+            ),
+            "currency_to": Schema(
+              type: .string,
+              format: "enum",
+              description: "The currency to convert to in ISO 4217 format",
+              enumValues: ["USD", "EUR", "JPY", "GBP", "AUD", "CAD"]
+            ),
+          ],
+          requiredParameters: ["currency_from", "currency_to"]
+        ),
+      ])]
+    )
+    chat = model.startChat()
+  }
+
+  func sendMessage(_ text: String, streaming: Bool = true) async {
+    error = nil
+    chatTask?.cancel()
+
+    chatTask = Task {
+      busy = true
+      defer {
+        busy = false
+      }
+
+      // first, add the user's message to the chat
+      let userMessage = ChatMessage(message: text, participant: .user)
+      messages.append(userMessage)
+
+      // add a pending message while we're waiting for a response from the backend
+      let systemMessage = ChatMessage.pending(participant: .system)
+      messages.append(systemMessage)
+
+      print(messages)
+      do {
+        repeat {
+          if streaming {
+            try await internalSendMessageStreaming(text)
+          } else {
+            try await internalSendMessage(text)
+          }
+        } while !functionCalls.isEmpty
+      } catch {
+        self.error = error
+        print(error.localizedDescription)
+        messages.removeLast()
+      }
+    }
+  }
+
+  func startNewChat() {
+    stop()
+    error = nil
+    chat = model.startChat()
+    messages.removeAll()
+  }
+
+  func stop() {
+    chatTask?.cancel()
+    error = nil
+  }
+
+  private func internalSendMessageStreaming(_ text: String) async throws {
+    let functionResponses = try await processFunctionCalls()
+    let responseStream: AsyncThrowingStream<GenerateContentResponse, Error>
+    if functionResponses.isEmpty {
+      responseStream = chat.sendMessageStream(text)
+    } else {
+      for functionResponse in functionResponses {
+        messages.insert(functionResponse.chatMessage(), at: messages.count - 1)
+      }
+      responseStream = chat.sendMessageStream(functionResponses.modelContent())
+    }
+    for try await chunk in responseStream {
+      processResponseContent(content: chunk)
+    }
+  }
+
+  private func internalSendMessage(_ text: String) async throws {
+    let functionResponses = try await processFunctionCalls()
+    let response: GenerateContentResponse
+    if functionResponses.isEmpty {
+      response = try await chat.sendMessage(text)
+    } else {
+      for functionResponse in functionResponses {
+        messages.insert(functionResponse.chatMessage(), at: messages.count - 1)
+      }
+      response = try await chat.sendMessage(functionResponses.modelContent())
+    }
+    processResponseContent(content: response)
+  }
+
+  func processResponseContent(content: GenerateContentResponse) {
+    guard let candidate = content.candidates.first else {
+      fatalError("No candidate.")
+    }
+
+    for part in candidate.content.parts {
+      switch part {
+      case let .text(text):
+        // replace pending message with backend response
+        messages[messages.count - 1].message += text
+        messages[messages.count - 1].pending = false
+      case let .functionCall(functionCall):
+        messages.insert(functionCall.chatMessage(), at: messages.count - 1)
+        functionCalls.append(functionCall)
+      case .data, .functionResponse:
+        fatalError("Unsupported response content.")
+      }
+    }
+  }
+
+  func processFunctionCalls() async throws -> [FunctionResponse] {
+    var functionResponses = [FunctionResponse]()
+    for functionCall in functionCalls {
+      switch functionCall.name {
+      case "get_exchange_rate":
+        let exchangeRates = getExchangeRate(args: functionCall.args)
+        functionResponses.append(FunctionResponse(
+          name: "get_exchange_rate",
+          response: exchangeRates
+        ))
+      default:
+        fatalError("Unknown function named \"\(functionCall.name)\".")
+      }
+    }
+    functionCalls = []
+
+    return functionResponses
+  }
+
+  // MARK: - Callable Functions
+
+  func getExchangeRate(args: JSONObject) -> JSONObject {
+    // 1. Validate and extract the parameters provided by the model (from a `FunctionCall`)
+    guard case let .string(from) = args["currency_from"] else {
+      fatalError("Missing `currency_from` parameter.")
+    }
+    guard case let .string(to) = args["currency_to"] else {
+      fatalError("Missing `currency_to` parameter.")
+    }
+
+    // 2. Get the exchange rate
+    let allRates: [String: [String: Double]] = [
+      "AUD": ["CAD": 0.89265, "EUR": 0.6072, "GBP": 0.51714, "JPY": 97.75, "USD": 0.66379],
+      "CAD": ["AUD": 1.1203, "EUR": 0.68023, "GBP": 0.57933, "JPY": 109.51, "USD": 0.74362],
+      "EUR": ["AUD": 1.6469, "CAD": 1.4701, "GBP": 0.85168, "JPY": 160.99, "USD": 1.0932],
+      "GBP": ["AUD": 1.9337, "CAD": 1.7261, "EUR": 1.1741, "JPY": 189.03, "USD": 1.2836],
+      "JPY": ["AUD": 0.01023, "CAD": 0.00913, "EUR": 0.00621, "GBP": 0.00529, "USD": 0.00679],
+      "USD": ["AUD": 1.5065, "CAD": 1.3448, "EUR": 0.91475, "GBP": 0.77907, "JPY": 147.26],
+    ]
+    guard let fromRates = allRates[from] else {
+      return ["error": .string("No data for currency \(from).")]
+    }
+    guard let toRate = fromRates[to] else {
+      return ["error": .string("No data for currency \(to).")]
+    }
+
+    // 3. Return the exchange rates as a JSON object (returned to the model in a `FunctionResponse`)
+    return ["rates": .number(toRate)]
+  }
+}
+
+private extension FunctionCall {
+  func chatMessage() -> ChatMessage {
+    let encoder = JSONEncoder()
+    encoder.outputFormatting = .prettyPrinted
+
+    let jsonData: Data
+    do {
+      jsonData = try encoder.encode(self)
+    } catch {
+      fatalError("JSON Encoding Failed: \(error.localizedDescription)")
+    }
+    guard let json = String(data: jsonData, encoding: .utf8) else {
+      fatalError("Failed to convert JSON data to a String.")
+    }
+    let messageText = "Function call requested by model:\n```\n\(json)\n```"
+
+    return ChatMessage(message: messageText, participant: .system)
+  }
+}
+
+private extension FunctionResponse {
+  func chatMessage() -> ChatMessage {
+    let encoder = JSONEncoder()
+    encoder.outputFormatting = .prettyPrinted
+
+    let jsonData: Data
+    do {
+      jsonData = try encoder.encode(self)
+    } catch {
+      fatalError("JSON Encoding Failed: \(error.localizedDescription)")
+    }
+    guard let json = String(data: jsonData, encoding: .utf8) else {
+      fatalError("Failed to convert JSON data to a String.")
+    }
+    let messageText = "Function response returned by app:\n```\n\(json)\n```"
+
+    return ChatMessage(message: messageText, participant: .user)
+  }
+}
+
+private extension [FunctionResponse] {
+  func modelContent() -> [ModelContent] {
+    return self.map { ModelContent(
+      role: "function",
+      parts: [ModelContent.Part.functionResponse($0)]
+    )
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 01336b9cf19..0e1bf8211f0 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -10,6 +10,8 @@
 		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
 		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */; };
+		86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */; };
 		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
 		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
@@ -31,6 +33,8 @@
 
 /* Begin PBXFileReference section */
 		869200B22B879C4F00482873 /* GoogleService-Info.plist */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text.plist.xml; path = "GoogleService-Info.plist"; sourceTree = "<group>"; };
+		86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = FunctionCallingScreen.swift; sourceTree = "<group>"; };
+		86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = FunctionCallingViewModel.swift; sourceTree = "<group>"; };
 		8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningViewModel.swift; sourceTree = "<group>"; };
 		880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningScreen.swift; sourceTree = "<group>"; };
 		88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeScreen.swift; sourceTree = "<group>"; };
@@ -71,6 +75,31 @@
 /* End PBXFrameworksBuildPhase section */
 
 /* Begin PBXGroup section */
+		86C1F47F2BC726150026816F /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+		86C1F4812BC726150026816F /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		86C1F4822BC726150026816F /* FunctionCallingSample */ = {
+			isa = PBXGroup;
+			children = (
+				86C1F4812BC726150026816F /* ViewModels */,
+				86C1F47F2BC726150026816F /* Screens */,
+			);
+			path = FunctionCallingSample;
+			sourceTree = "<group>";
+		};
 		8802666E2B0FC39000CF7CB6 /* ViewModels */ = {
 			isa = PBXGroup;
 			children = (
@@ -119,6 +148,7 @@
 				8848C8452B0D051E007B434F /* GenerativeAITextSample */,
 				8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */,
 				88E10F432B110D5300C08E95 /* ChatSample */,
+				86C1F4822BC726150026816F /* FunctionCallingSample */,
 				8848C8302B0D04BC007B434F /* Products */,
 				88209C222B0FBE1700F64795 /* Frameworks */,
 			);
@@ -327,7 +357,9 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */,
 				886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */,
+				86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */,
 				886F95DE2B17D5010036F07A /* ChatMessage.swift in Sources */,
 				88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */,
 				8848C8352B0D04BC007B434F /* ContentView.swift in Sources */,
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
index 34331bf6eb7..2841d634a6c 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
@@ -18,6 +18,9 @@ struct ContentView: View {
   @StateObject
   var viewModel = ConversationViewModel()
 
+  @StateObject
+  var functionCallingViewModel = FunctionCallingViewModel()
+
   var body: some View {
     NavigationStack {
       List {
@@ -37,6 +40,11 @@ struct ContentView: View {
         } label: {
           Label("Chat", systemImage: "ellipsis.message.fill")
         }
+        NavigationLink {
+          FunctionCallingScreen().environmentObject(functionCallingViewModel)
+        } label: {
+          Label("Function Calling", systemImage: "function")
+        }
       }
       .navigationTitle("Generative AI Samples")
     }

From 29c2248694cdf94541cd63890b9641bcb94b69c0 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 10 Apr 2024 20:46:45 -0400
Subject: [PATCH 041/124] Add forced function calling support in Vertex AI
 (#12748)

---
 .../Sources/FunctionCalling.swift             | 45 +++++++++++++++++++
 .../Sources/GenerateContentRequest.swift      |  2 +
 .../Sources/GenerativeModel.swift             |  8 ++++
 FirebaseVertexAI/Sources/VertexAI.swift       |  3 ++
 4 files changed, 58 insertions(+)

diff --git a/FirebaseVertexAI/Sources/FunctionCalling.swift b/FirebaseVertexAI/Sources/FunctionCalling.swift
index 7e4d84d857e..8860344c7cd 100644
--- a/FirebaseVertexAI/Sources/FunctionCalling.swift
+++ b/FirebaseVertexAI/Sources/FunctionCalling.swift
@@ -177,6 +177,51 @@ public struct Tool: Encodable {
   }
 }
 
+/// Configuration for specifying function calling behavior.
+public struct FunctionCallingConfig: Encodable {
+  /// Defines the execution behavior for function calling by defining the
+  /// execution mode.
+  public enum Mode: String, Encodable {
+    /// The default behavior for function calling. The model calls functions to answer queries at
+    /// its discretion.
+    case auto = "AUTO"
+
+    /// The model always predicts a provided function call to answer every query.
+    case any = "ANY"
+
+    /// The model will never predict a function call to answer a query. This can also be achieved by
+    /// not passing any tools to the model.
+    case none = "NONE"
+  }
+
+  /// Specifies the mode in which function calling should execute. If
+  /// unspecified, the default value will be set to AUTO.
+  let mode: Mode?
+
+  /// A set of function names that, when provided, limits the functions the model
+  /// will call.
+  ///
+  /// This should only be set when the Mode is ANY. Function names
+  /// should match [FunctionDeclaration.name]. With mode set to ANY, model will
+  /// predict a function call from the set of function names provided.
+  let allowedFunctionNames: [String]?
+
+  public init(mode: FunctionCallingConfig.Mode? = nil, allowedFunctionNames: [String]? = nil) {
+    self.mode = mode
+    self.allowedFunctionNames = allowedFunctionNames
+  }
+}
+
+/// Tool configuration for any `Tool` specified in the request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct ToolConfig: Encodable {
+  let functionCallingConfig: FunctionCallingConfig?
+
+  public init(functionCallingConfig: FunctionCallingConfig? = nil) {
+    self.functionCallingConfig = functionCallingConfig
+  }
+}
+
 /// Result output from a ``FunctionCall``.
 ///
 /// Contains a string representing the `FunctionDeclaration.name` and a structured JSON object
diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
index 535ac4da3ae..e0e9c2fe0c8 100644
--- a/FirebaseVertexAI/Sources/GenerateContentRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -22,6 +22,7 @@ struct GenerateContentRequest {
   let generationConfig: GenerationConfig?
   let safetySettings: [SafetySetting]?
   let tools: [Tool]?
+  let toolConfig: ToolConfig?
   let isStreaming: Bool
   let options: RequestOptions
 }
@@ -33,6 +34,7 @@ extension GenerateContentRequest: Encodable {
     case generationConfig
     case safetySettings
     case tools
+    case toolConfig
   }
 }
 
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index 7eab63a1d7a..124656dd639 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -37,6 +37,9 @@ public final class GenerativeModel {
   /// A list of tools the model may use to generate the next response.
   let tools: [Tool]?
 
+  /// Tool configuration for any `Tool` specified in the request.
+  let toolConfig: ToolConfig?
+
   /// Configuration parameters for sending requests to the backend.
   let requestOptions: RequestOptions
 
@@ -49,6 +52,7 @@ public final class GenerativeModel {
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
+  ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
@@ -56,6 +60,7 @@ public final class GenerativeModel {
        generationConfig: GenerationConfig? = nil,
        safetySettings: [SafetySetting]? = nil,
        tools: [Tool]?,
+       toolConfig: ToolConfig? = nil,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
        urlSession: URLSession = .shared) {
@@ -68,6 +73,7 @@ public final class GenerativeModel {
     self.generationConfig = generationConfig
     self.safetySettings = safetySettings
     self.tools = tools
+    self.toolConfig = toolConfig
     self.requestOptions = requestOptions
 
     Logging.default.info("""
@@ -114,6 +120,7 @@ public final class GenerativeModel {
                                                               generationConfig: generationConfig,
                                                               safetySettings: safetySettings,
                                                               tools: tools,
+                                                              toolConfig: toolConfig,
                                                               isStreaming: false,
                                                               options: requestOptions)
       response = try await generativeAIService.loadRequest(request: generateContentRequest)
@@ -186,6 +193,7 @@ public final class GenerativeModel {
                                                         generationConfig: generationConfig,
                                                         safetySettings: safetySettings,
                                                         tools: tools,
+                                                        toolConfig: toolConfig,
                                                         isStreaming: true,
                                                         options: requestOptions)
 
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 1fcfd998b45..843a3d09552 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -58,11 +58,13 @@ public class VertexAI: NSObject {
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
+  ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
                               tools: [Tool]? = nil,
+                              toolConfig: ToolConfig? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
     let modelResourceName = modelResourceName(modelName: modelName, location: location)
@@ -77,6 +79,7 @@ public class VertexAI: NSObject {
       generationConfig: generationConfig,
       safetySettings: safetySettings,
       tools: tools,
+      toolConfig: toolConfig,
       requestOptions: requestOptions,
       appCheck: appCheck
     )

From c7784f5f7967cfc77372e212a62cd1e01bf7bb23 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 11 Apr 2024 09:52:23 -0400
Subject: [PATCH 042/124] Add system instruction support in Vertex AI (#12749)

---
 FirebaseVertexAI/Sources/GenerateContentRequest.swift | 2 ++
 FirebaseVertexAI/Sources/GenerativeModel.swift        | 9 +++++++++
 FirebaseVertexAI/Sources/VertexAI.swift               | 4 ++++
 FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift    | 9 ++++++++-
 4 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
index e0e9c2fe0c8..05abadf54e1 100644
--- a/FirebaseVertexAI/Sources/GenerateContentRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -23,6 +23,7 @@ struct GenerateContentRequest {
   let safetySettings: [SafetySetting]?
   let tools: [Tool]?
   let toolConfig: ToolConfig?
+  let systemInstruction: ModelContent?
   let isStreaming: Bool
   let options: RequestOptions
 }
@@ -35,6 +36,7 @@ extension GenerateContentRequest: Encodable {
     case safetySettings
     case tools
     case toolConfig
+    case systemInstruction
   }
 }
 
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index 124656dd639..cdd6d175c0a 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -40,6 +40,9 @@ public final class GenerativeModel {
   /// Tool configuration for any `Tool` specified in the request.
   let toolConfig: ToolConfig?
 
+  /// Instructions that direct the model to behave a certain way.
+  let systemInstruction: ModelContent?
+
   /// Configuration parameters for sending requests to the backend.
   let requestOptions: RequestOptions
 
@@ -53,6 +56,8 @@ public final class GenerativeModel {
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
+  ///   - systemInstruction: Instructions that direct the model to behave a certain way; currently
+  ///     only text content is supported.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
@@ -61,6 +66,7 @@ public final class GenerativeModel {
        safetySettings: [SafetySetting]? = nil,
        tools: [Tool]?,
        toolConfig: ToolConfig? = nil,
+       systemInstruction: ModelContent? = nil,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
        urlSession: URLSession = .shared) {
@@ -74,6 +80,7 @@ public final class GenerativeModel {
     self.safetySettings = safetySettings
     self.tools = tools
     self.toolConfig = toolConfig
+    self.systemInstruction = systemInstruction
     self.requestOptions = requestOptions
 
     Logging.default.info("""
@@ -121,6 +128,7 @@ public final class GenerativeModel {
                                                               safetySettings: safetySettings,
                                                               tools: tools,
                                                               toolConfig: toolConfig,
+                                                              systemInstruction: systemInstruction,
                                                               isStreaming: false,
                                                               options: requestOptions)
       response = try await generativeAIService.loadRequest(request: generateContentRequest)
@@ -194,6 +202,7 @@ public final class GenerativeModel {
                                                         safetySettings: safetySettings,
                                                         tools: tools,
                                                         toolConfig: toolConfig,
+                                                        systemInstruction: systemInstruction,
                                                         isStreaming: true,
                                                         options: requestOptions)
 
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 843a3d09552..7a7865e86d2 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -59,12 +59,15 @@ public class VertexAI: NSObject {
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
+  ///   - systemInstruction: Instructions that direct the model to behave a certain way; currently
+  ///     only text content is supported.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
                               tools: [Tool]? = nil,
                               toolConfig: ToolConfig? = nil,
+                              systemInstruction: ModelContent? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
     let modelResourceName = modelResourceName(modelName: modelName, location: location)
@@ -80,6 +83,7 @@ public class VertexAI: NSObject {
       safetySettings: safetySettings,
       tools: tools,
       toolConfig: toolConfig,
+      systemInstruction: systemInstruction,
       requestOptions: requestOptions,
       appCheck: appCheck
     )
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 81b81414dcb..e9312edfee0 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -32,6 +32,7 @@ final class VertexAIAPITests: XCTestCase {
                                   maxOutputTokens: 256,
                                   stopSequences: ["..."])
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
+    let systemInstruction = ModelContent(role: "system", parts: [.text("Talk like a pirate.")])
 
     // Instantiate Vertex AI SDK - Default App
     let vertexAI = VertexAI.vertexAI()
@@ -53,11 +54,17 @@ final class VertexAIAPITests: XCTestCase {
       generationConfig: config
     )
 
+    let _ = vertexAI.generativeModel(
+      modelName: "gemini-1.0-pro",
+      systemInstruction: systemInstruction
+    )
+
     // All arguments passed.
     let genAI = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
       generationConfig: config, // Optional
-      safetySettings: filters // Optional
+      safetySettings: filters, // Optional
+      systemInstruction: systemInstruction // Optional
     )
 
     // Full Typed Usage

From fc0f8a4b6f9dd9b49c2497a34ae09a31b5ac50d3 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 11 Apr 2024 14:51:32 -0400
Subject: [PATCH 043/124] Remove FirebaseAppCheck module from Vertex AI sample
 app (#12754)

---
 .../Sample/GenerativeAISample.xcodeproj/project.pbxproj    | 7 -------
 .../Sample/GenerativeAISample/GenerativeAISampleApp.swift  | 3 +++
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 0e1bf8211f0..1914a9838e6 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -7,7 +7,6 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
-		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
 		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
 		86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */; };
@@ -67,7 +66,6 @@
 			files = (
 				868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
-				868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */,
 				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -294,7 +292,6 @@
 			packageProductDependencies = (
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
-				868A33652BB476FA00304BB1 /* FirebaseAppCheck */,
 				868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAISample;
@@ -607,10 +604,6 @@
 /* End XCRemoteSwiftPackageReference section */
 
 /* Begin XCSwiftPackageProductDependency section */
-		868A33652BB476FA00304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
 		868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = "FirebaseVertexAI-Preview";
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
index 509087a5e07..18723a459c5 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
@@ -18,6 +18,9 @@ import SwiftUI
 @main
 struct GenerativeAISampleApp: App {
   init() {
+    // Recommendation: Protect your Vertex AI API resources from abuse by preventing unauthorized
+    // clients using App Check; see https://firebase.google.com/docs/app-check#get_started.
+
     FirebaseApp.configure()
 
     if let firebaseApp = FirebaseApp.app(), firebaseApp.options.projectID == "mockproject-1234" {

From 113a100cbe6f42ac1bbb8a57a39c71c526e35d33 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 11 Apr 2024 17:51:09 -0400
Subject: [PATCH 044/124] Rename GenerativeAISample to VertexAISample (#12755)

---
 FirebaseVertexAI/Sample/README.md             |  2 +-
 .../project.pbxproj                           | 38 +++++++++----------
 .../AccentColor.colorset/Contents.json        |  0
 .../AppIcon.appiconset/Contents.json          |  0
 .../Assets.xcassets/Contents.json             |  0
 .../ContentView.swift                         |  0
 .../Preview Assets.xcassets/Contents.json     |  0
 .../VertexAISampleApp.swift}                  |  2 +-
 scripts/build.sh                              |  4 +-
 9 files changed, 23 insertions(+), 23 deletions(-)
 rename FirebaseVertexAI/Sample/{GenerativeAISample.xcodeproj => VertexAISample.xcodeproj}/project.pbxproj (94%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Assets.xcassets/AccentColor.colorset/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Assets.xcassets/AppIcon.appiconset/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Assets.xcassets/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/ContentView.swift (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Preview Content/Preview Assets.xcassets/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample/GenerativeAISampleApp.swift => VertexAISample/VertexAISampleApp.swift} (97%)

diff --git a/FirebaseVertexAI/Sample/README.md b/FirebaseVertexAI/Sample/README.md
index a4a8b458df3..8a7afa02977 100644
--- a/FirebaseVertexAI/Sample/README.md
+++ b/FirebaseVertexAI/Sample/README.md
@@ -6,7 +6,7 @@ use the sample app, you'll need to perform the following steps:
 1. Download
    [vertexai-preview-0.1.0.zip](https://github.com/firebase/firebase-ios-sdk/archive/refs/heads/vertexai-preview-0.1.0.zip)
    for the latest release of the SDK
-1. Extract the zip and open `GenerativeAISample.xcodeproj` in the
+1. Extract the zip and open `VertexAISample.xcodeproj` in the
    `FirebaseVertexAI/Sample` directory
 1. [Register the sample app](https://firebase.google.com/docs/ios/setup#register-app)
    in your Firebase project
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
similarity index 94%
rename from FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
rename to FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
index 1914a9838e6..a5c19152df7 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
@@ -13,7 +13,7 @@
 		86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */; };
 		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
-		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
+		8848C8332B0D04BC007B434F /* VertexAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* VertexAISampleApp.swift */; };
 		8848C8352B0D04BC007B434F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8342B0D04BC007B434F /* ContentView.swift */; };
 		8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8362B0D04BD007B434F /* Assets.xcassets */; };
 		8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8392B0D04BD007B434F /* Preview Assets.xcassets */; };
@@ -39,8 +39,8 @@
 		88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeScreen.swift; sourceTree = "<group>"; };
 		88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeViewModel.swift; sourceTree = "<group>"; };
 		88263BEE2B239BFE008AB09B /* ErrorView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = ErrorView.swift; sourceTree = "<group>"; };
-		8848C82F2B0D04BC007B434F /* GenerativeAISample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAISample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAISampleApp.swift; sourceTree = "<group>"; };
+		8848C82F2B0D04BC007B434F /* VertexAISample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = VertexAISample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8322B0D04BC007B434F /* VertexAISampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = VertexAISampleApp.swift; sourceTree = "<group>"; };
 		8848C8342B0D04BC007B434F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = "<group>"; };
 		8848C8362B0D04BD007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C8392B0D04BD007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
@@ -142,7 +142,7 @@
 			children = (
 				88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */,
 				869200B22B879C4F00482873 /* GoogleService-Info.plist */,
-				8848C8312B0D04BC007B434F /* GenerativeAISample */,
+				8848C8312B0D04BC007B434F /* VertexAISample */,
 				8848C8452B0D051E007B434F /* GenerativeAITextSample */,
 				8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */,
 				88E10F432B110D5300C08E95 /* ChatSample */,
@@ -155,20 +155,20 @@
 		8848C8302B0D04BC007B434F /* Products */ = {
 			isa = PBXGroup;
 			children = (
-				8848C82F2B0D04BC007B434F /* GenerativeAISample.app */,
+				8848C82F2B0D04BC007B434F /* VertexAISample.app */,
 			);
 			name = Products;
 			sourceTree = "<group>";
 		};
-		8848C8312B0D04BC007B434F /* GenerativeAISample */ = {
+		8848C8312B0D04BC007B434F /* VertexAISample */ = {
 			isa = PBXGroup;
 			children = (
-				8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */,
+				8848C8322B0D04BC007B434F /* VertexAISampleApp.swift */,
 				8848C8342B0D04BC007B434F /* ContentView.swift */,
 				8848C8362B0D04BD007B434F /* Assets.xcassets */,
 				8848C8382B0D04BD007B434F /* Preview Content */,
 			);
-			path = GenerativeAISample;
+			path = VertexAISample;
 			sourceTree = "<group>";
 		};
 		8848C8382B0D04BD007B434F /* Preview Content */ = {
@@ -276,9 +276,9 @@
 /* End PBXGroup section */
 
 /* Begin PBXNativeTarget section */
-		8848C82E2B0D04BC007B434F /* GenerativeAISample */ = {
+		8848C82E2B0D04BC007B434F /* VertexAISample */ = {
 			isa = PBXNativeTarget;
-			buildConfigurationList = 8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */;
+			buildConfigurationList = 8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "VertexAISample" */;
 			buildPhases = (
 				8848C82B2B0D04BC007B434F /* Sources */,
 				8848C82C2B0D04BC007B434F /* Frameworks */,
@@ -288,14 +288,14 @@
 			);
 			dependencies = (
 			);
-			name = GenerativeAISample;
+			name = VertexAISample;
 			packageProductDependencies = (
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
 				868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAISample;
-			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
+			productReference = 8848C82F2B0D04BC007B434F /* VertexAISample.app */;
 			productType = "com.apple.product-type.application";
 		};
 /* End PBXNativeTarget section */
@@ -313,7 +313,7 @@
 					};
 				};
 			};
-			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */;
+			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "VertexAISample" */;
 			compatibilityVersion = "Xcode 14.0";
 			developmentRegion = en;
 			hasScannedForEncodings = 0;
@@ -331,7 +331,7 @@
 			projectDirPath = "";
 			projectRoot = "";
 			targets = (
-				8848C82E2B0D04BC007B434F /* GenerativeAISample */,
+				8848C82E2B0D04BC007B434F /* VertexAISample */,
 			);
 		};
 /* End PBXProject section */
@@ -361,7 +361,7 @@
 				88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */,
 				8848C8352B0D04BC007B434F /* ContentView.swift in Sources */,
 				886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */,
-				8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */,
+				8848C8332B0D04BC007B434F /* VertexAISampleApp.swift in Sources */,
 				886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */,
 				886F95DD2B17D5010036F07A /* MessageView.swift in Sources */,
 				886F95DC2B17BAEF0036F07A /* PhotoReasoningScreen.swift in Sources */,
@@ -501,7 +501,7 @@
 				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
 				DEVELOPMENT_TEAM = "";
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
@@ -531,7 +531,7 @@
 				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
 				DEVELOPMENT_TEAM = "";
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
@@ -557,7 +557,7 @@
 /* End XCBuildConfiguration section */
 
 /* Begin XCConfigurationList section */
-		8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */ = {
+		8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "VertexAISample" */ = {
 			isa = XCConfigurationList;
 			buildConfigurations = (
 				8848C83B2B0D04BD007B434F /* Debug */,
@@ -566,7 +566,7 @@
 			defaultConfigurationIsVisible = 0;
 			defaultConfigurationName = Release;
 		};
-		8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */ = {
+		8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "VertexAISample" */ = {
 			isa = XCConfigurationList;
 			buildConfigurations = (
 				8848C83E2B0D04BD007B434F /* Debug */,
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AccentColor.colorset/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AccentColor.colorset/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift b/FirebaseVertexAI/Sample/VertexAISample/ContentView.swift
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
rename to FirebaseVertexAI/Sample/VertexAISample/ContentView.swift
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Preview Content/Preview Assets.xcassets/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Preview Content/Preview Assets.xcassets/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
similarity index 97%
rename from FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
rename to FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
index 18723a459c5..4772bcc2d7d 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
@@ -16,7 +16,7 @@ import FirebaseCore
 import SwiftUI
 
 @main
-struct GenerativeAISampleApp: App {
+struct VertexAISampleApp: App {
   init() {
     // Recommendation: Protect your Vertex AI API resources from abuse by preventing unauthorized
     // clients using App Check; see https://firebase.google.com/docs/app-check#get_started.
diff --git a/scripts/build.sh b/scripts/build.sh
index 75935652c98..5c9d3830cf5 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -492,8 +492,8 @@ case "$product-$platform-$method" in
 
   VertexSample-*-*)
     RunXcodebuild \
-      -project 'FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj' \
-      -scheme "GenerativeAISample" \
+      -project 'FirebaseVertexAI/Sample/VertexAISample.xcodeproj' \
+      -scheme "VertexAISample" \
       "${xcb_flags[@]}" \
       build
     ;;

From 9d304a1e70bba8b2feefcbd13917f6aa6febcbbb Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 12 Apr 2024 17:21:01 -0400
Subject: [PATCH 045/124] Add `usageMetadata` to `GenerateContentResponse` in
 Vertex AI (#12777)

---
 .../ChatSample/Views/ErrorDetailsView.swift   |  6 +--
 .../Sample/ChatSample/Views/ErrorView.swift   |  3 +-
 .../Sources/GenerateContentResponse.swift     | 38 ++++++++++++++++-
 .../Tests/Unit/GenerativeModelTests.swift     | 41 +++++++++++++++++++
 4 files changed, 81 insertions(+), 7 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
index 7b3b78753db..dc5ce8f9561 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -176,8 +176,7 @@ struct ErrorDetailsView: View {
       ],
       finishReason: FinishReason.maxTokens,
       citationMetadata: nil),
-    ],
-    promptFeedback: nil)
+    ])
   )
 
   return ErrorDetailsView(error: error)
@@ -200,8 +199,7 @@ struct ErrorDetailsView: View {
       ],
       finishReason: FinishReason.other,
       citationMetadata: nil),
-    ],
-    promptFeedback: nil)
+    ])
   )
 
   return ErrorDetailsView(error: error)
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
index 1307eee62d4..d4db2d67dc5 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
@@ -51,8 +51,7 @@ struct ErrorView: View {
         ],
         finishReason: FinishReason.other,
         citationMetadata: nil),
-      ],
-      promptFeedback: nil)
+      ])
     )
     List {
       MessageView(message: ChatMessage.samples[0])
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index f43da308b13..29c5b14f552 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -17,6 +17,18 @@ import Foundation
 /// The model's response to a generate content request.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public struct GenerateContentResponse {
+  /// Token usage metadata for processing the generate content request.
+  public struct UsageMetadata {
+    /// The number of tokens in the request prompt.
+    public let promptTokenCount: Int
+
+    /// The total number of tokens across the generated response candidates.
+    public let candidatesTokenCount: Int
+
+    /// The total number of tokens in both the request and response.
+    public let totalTokenCount: Int
+  }
+
   /// A list of candidate response content, ordered from best to worst.
   public let candidates: [CandidateResponse]
 
@@ -24,6 +36,9 @@ public struct GenerateContentResponse {
   /// reason for blocking the request.
   public let promptFeedback: PromptFeedback?
 
+  /// Token usage metadata for processing the generate content request.
+  public let usageMetadata: UsageMetadata?
+
   /// The response's content as text, if it exists.
   public var text: String? {
     guard let candidate = candidates.first else {
@@ -51,9 +66,11 @@ public struct GenerateContentResponse {
   }
 
   /// Initializer for SwiftUI previews or tests.
-  public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback?) {
+  public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback? = nil,
+              usageMetadata: UsageMetadata? = nil) {
     self.candidates = candidates
     self.promptFeedback = promptFeedback
+    self.usageMetadata = usageMetadata
   }
 }
 
@@ -62,6 +79,7 @@ extension GenerateContentResponse: Decodable {
   enum CodingKeys: CodingKey {
     case candidates
     case promptFeedback
+    case usageMetadata
   }
 
   public init(from decoder: Decoder) throws {
@@ -86,6 +104,7 @@ extension GenerateContentResponse: Decodable {
       candidates = []
     }
     promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
+    usageMetadata = try container.decodeIfPresent(UsageMetadata.self, forKey: .usageMetadata)
   }
 }
 
@@ -301,3 +320,20 @@ extension PromptFeedback: Decodable {
     }
   }
 }
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentResponse.UsageMetadata: Decodable {
+  enum CodingKeys: CodingKey {
+    case promptTokenCount
+    case candidatesTokenCount
+    case totalTokenCount
+  }
+
+  public init(from decoder: any Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    promptTokenCount = try container.decodeIfPresent(Int.self, forKey: .promptTokenCount) ?? 0
+    candidatesTokenCount = try container
+      .decodeIfPresent(Int.self, forKey: .candidatesTokenCount) ?? 0
+    totalTokenCount = try container.decodeIfPresent(Int.self, forKey: .totalTokenCount) ?? 0
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 92585266891..e7ea54f5dd9 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -287,6 +287,21 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_usageMetadata() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    let usageMetadata = try XCTUnwrap(response.usageMetadata)
+    XCTAssertEqual(usageMetadata.promptTokenCount, 6)
+    XCTAssertEqual(usageMetadata.candidatesTokenCount, 7)
+    XCTAssertEqual(usageMetadata.totalTokenCount, 13)
+  }
+
   func testGenerateContent_failure_invalidAPIKey() async throws {
     let expectedStatusCode = 400
     MockURLProtocol
@@ -814,6 +829,32 @@ final class GenerativeModelTests: XCTestCase {
     for try await _ in stream {}
   }
 
+  func testGenerateContentStream_usageMetadata() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt"
+      )
+    var responses = [GenerateContentResponse]()
+
+    let stream = model.generateContentStream(testPrompt)
+    for try await response in stream {
+      responses.append(response)
+    }
+
+    for (index, response) in responses.enumerated() {
+      if index == responses.endIndex - 1 {
+        let usageMetadata = try XCTUnwrap(response.usageMetadata)
+        XCTAssertEqual(usageMetadata.promptTokenCount, 6)
+        XCTAssertEqual(usageMetadata.candidatesTokenCount, 4)
+        XCTAssertEqual(usageMetadata.totalTokenCount, 10)
+      } else {
+        // Only the last streamed response contains usage metadata
+        XCTAssertNil(response.usageMetadata)
+      }
+    }
+  }
+
   func testGenerateContentStream_errorMidStream() async throws {
     MockURLProtocol.requestHandler = try httpRequestHandler(
       forResource: "streaming-failure-error-mid-stream",

From 2cbae6e502f4286e25d1d44296cd33827cb08291 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 15 Apr 2024 12:59:20 -0400
Subject: [PATCH 046/124] Add public API tests for `GenerateContentResponse` in
 Vertex AI (#12791)

---
 .../Tests/Unit/VertexAIAPITests.swift          | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index e9312edfee0..75fb8946d22 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -167,6 +167,24 @@ final class VertexAIAPITests: XCTestCase {
     _ = genAI.startChat(history: [ModelContent(parts: "abc")])
   }
 
+  // Public API tests for GenerateContentResponse.
+  func generateContentResponseAPI() {
+    let response = GenerateContentResponse(candidates: [])
+
+    let _: [CandidateResponse] = response.candidates
+    let _: PromptFeedback? = response.promptFeedback
+
+    // Usage Metadata
+    guard let usageMetadata = response.usageMetadata else { fatalError() }
+    let _: Int = usageMetadata.promptTokenCount
+    let _: Int = usageMetadata.candidatesTokenCount
+    let _: Int = usageMetadata.totalTokenCount
+
+    // Computed Properties
+    let _: String? = response.text
+    let _: [FunctionCall] = response.functionCalls
+  }
+
   // Result builder alternative
 
   /*

From a89224c028be0876afe6fb2cc0affcba958a715e Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 22 Apr 2024 12:50:01 -0400
Subject: [PATCH 047/124] [Vertex AI] Add Swift major version to language tag
 (#12829)

---
 FirebaseVertexAI/Sources/GenerativeAIService.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index b3687c3c013..60ca375c9f6 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -18,8 +18,8 @@ import Foundation
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 struct GenerativeAIService {
-  /// The language of the SDK in the format "gl-<language>/<version>" where version may be blank.
-  static let languageTag = "gl-swift/"
+  /// The language of the SDK in the format "gl-<language>/<version>".
+  static let languageTag = "gl-swift/5"
 
   /// The Firebase SDK version in the format "fire/<version>".
   static let firebaseVersionTag = "fire/\(FirebaseVersion())"

From b3ffeb1568cbcc1229edd072ca8686d04eebed79 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 22 Apr 2024 14:54:42 -0400
Subject: [PATCH 048/124] [Vertex AI] Update `Citation` decoding to handle
 optional `startIndex` (#12832)

---
 .../Sources/GenerateContentResponse.swift     | 24 +++++++++++-
 .../streaming-success-citations.txt           | 10 ++---
 .../unary-success-citations.json              | 38 ++++++++++++-------
 .../Tests/Unit/GenerativeModelTests.swift     | 32 +++++++++++-----
 4 files changed, 72 insertions(+), 32 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 29c5b14f552..48f4ff22413 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -197,7 +197,7 @@ public struct CitationMetadata: Decodable {
 
 /// A struct describing a source attribution.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct Citation: Decodable {
+public struct Citation {
   /// The inclusive beginning of a sequence in a model response that derives from a cited source.
   public let startIndex: Int
 
@@ -207,7 +207,7 @@ public struct Citation: Decodable {
   /// A link to the cited source.
   public let uri: String
 
-  /// The license the cited source work is distributed under.
+  /// The license the cited source work is distributed under, if specified.
   public let license: String?
 }
 
@@ -337,3 +337,23 @@ extension GenerateContentResponse.UsageMetadata: Decodable {
     totalTokenCount = try container.decodeIfPresent(Int.self, forKey: .totalTokenCount) ?? 0
   }
 }
+
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension Citation: Decodable {
+  enum CodingKeys: CodingKey {
+    case startIndex
+    case endIndex
+    case uri
+    case license
+  }
+
+  public init(from decoder: any Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    startIndex = try container.decodeIfPresent(Int.self, forKey: .startIndex) ?? 0
+    endIndex = try container.decode(Int.self, forKey: .endIndex)
+    uri = try container.decode(String.self, forKey: .uri)
+    license = try container.decodeIfPresent(String.self, forKey: .license)
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
index a530736b308..fdc1f20d8b6 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
@@ -1,9 +1,7 @@
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Some information"}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Some information"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.043204036,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.082549304},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.046291895,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.071461484},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.100701615,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06164962},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.13150747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.040087357}]}]}
 
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": "  More information"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06632687,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.07477004,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048767097},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13695431,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059866417},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.046119746,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.036425155}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "  Some information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.18982129,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.1337543},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.13637818,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.021906368},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.25404602,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.09073549},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.24202643,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06548521}],"citationMetadata": {"citations": [{"endIndex": 128,"uri": "https://www.example.com/citation-1"},{"startIndex": 130,"endIndex": 265,"uri": "https://www.example.com/citation-2"}]}}]}
 
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Some information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07850098,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.039416388},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.08035747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04885778},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12273335,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059646938},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.053206205,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04099903}],"citationMetadata": {"citations": [{"startIndex": 31,"endIndex": 187,"uri": "https://www.example.com/citation-1"}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07850098,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.039416388},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.08035747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04885778},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12273335,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059646938},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.053206205,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04099903}],"citationMetadata": {"citations": [{"startIndex": 272,"endIndex": 431,"uri": "https://www.example.com/citation-3","license": "mit"}]}}]}
 
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.044183318},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.094176665,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0575992},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13660839,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08035747},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.060197048,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.046464667}],"citationMetadata": {"citations": [{"startIndex": 73,"endIndex": 248,"uri": "https://www.example.com/citation-2"},{"startIndex": 133,"endIndex": 272,"uri": "https://www.example.com/citation-3", "license": "mit"}]}}]}
-
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.12147716,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0647717},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.11858909,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.14866412,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08479541},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05470151,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045015533}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 163,"totalTokenCount": 172}}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information "}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.16013464,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.11716747},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.10818896,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.021990221},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.2158462,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.07682221},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.19636348,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08021325}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 53,"totalTokenCount": 62}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
index a0dacc3e883..68fc9648773 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
@@ -14,46 +14,56 @@
         {
           "category": "HARM_CATEGORY_HATE_SPEECH",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.16013464,
+          "probabilityScore": 0.16926852,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.074500255
+          "severityScore": 0.08181271
         },
         {
           "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.09687653,
+          "probabilityScore": 0.15636235,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.049313594
+          "severityScore": 0.02981654
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.16817278,
+          "probabilityScore": 0.33133608,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.09451043
+          "severityScore": 0.10875559
         },
         {
           "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.05023736,
+          "probabilityScore": 0.17766814,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.034553625
+          "severityScore": 0.18787657
         }
       ],
       "citationMetadata": {
         "citations": [
           {
-            "startIndex": 179,
-            "endIndex": 366,
-            "uri": "https://www.example.com/some-citation"
+            "endIndex": 128,
+            "uri": "https://www.example.com/some-citation-1"
+          },
+          {
+            "startIndex": 130,
+            "endIndex": 265,
+            "uri": "https://www.example.com/some-citation-2"
+          },
+          {
+            "startIndex": 272,
+            "endIndex": 431,
+            "uri": "https://www.example.com/some-citation-3",
+            "license": "mit"
           }
         ]
       }
     }
   ],
   "usageMetadata": {
-    "promptTokenCount": 11,
-    "candidatesTokenCount": 135,
-    "totalTokenCount": 146
+    "promptTokenCount": 15,
+    "candidatesTokenCount": 253,
+    "totalTokenCount": 268
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index e7ea54f5dd9..c9442e05ba7 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -108,12 +108,22 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(candidate.content.parts.count, 1)
     XCTAssertEqual(response.text, "Some information cited from an external source")
     let citationMetadata = try XCTUnwrap(candidate.citationMetadata)
-    XCTAssertEqual(citationMetadata.citationSources.count, 1)
-    let citationSource = try XCTUnwrap(citationMetadata.citationSources.first)
-    XCTAssertEqual(citationSource.uri, "https://www.example.com/some-citation")
-    XCTAssertEqual(citationSource.startIndex, 179)
-    XCTAssertEqual(citationSource.endIndex, 366)
-    XCTAssertNil(citationSource.license)
+    XCTAssertEqual(citationMetadata.citationSources.count, 3)
+    let citationSource1 = try XCTUnwrap(citationMetadata.citationSources[0])
+    XCTAssertEqual(citationSource1.uri, "https://www.example.com/some-citation-1")
+    XCTAssertEqual(citationSource1.startIndex, 0)
+    XCTAssertEqual(citationSource1.endIndex, 128)
+    XCTAssertNil(citationSource1.license)
+    let citationSource2 = try XCTUnwrap(citationMetadata.citationSources[1])
+    XCTAssertEqual(citationSource2.uri, "https://www.example.com/some-citation-2")
+    XCTAssertEqual(citationSource2.startIndex, 130)
+    XCTAssertEqual(citationSource2.endIndex, 265)
+    XCTAssertNil(citationSource2.license)
+    let citationSource3 = try XCTUnwrap(citationMetadata.citationSources[2])
+    XCTAssertEqual(citationSource3.uri, "https://www.example.com/some-citation-3")
+    XCTAssertEqual(citationSource3.startIndex, 272)
+    XCTAssertEqual(citationSource3.endIndex, 431)
+    XCTAssertEqual(citationSource3.license, "mit")
   }
 
   func testGenerateContent_success_quoteReply() async throws {
@@ -778,13 +788,15 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(citations.count, 3)
     XCTAssertTrue(citations
       .contains(where: {
-        $0.startIndex == 31 && $0.endIndex == 187 && $0
-          .uri == "https://www.example.com/citation-1" && $0.license == nil
+        $0.startIndex == 0 && $0.endIndex == 128 && !$0.uri.isEmpty && $0.license == nil
       }))
     XCTAssertTrue(citations
       .contains(where: {
-        $0.startIndex == 133 && $0.endIndex == 272 && $0
-          .uri == "https://www.example.com/citation-3" && $0.license == "mit"
+        $0.startIndex == 130 && $0.endIndex == 265 && !$0.uri.isEmpty && $0.license == nil
+      }))
+    XCTAssertTrue(citations
+      .contains(where: {
+        $0.startIndex == 272 && $0.endIndex == 431 && !$0.uri.isEmpty && $0.license == "mit"
       }))
   }
 

From fc4d6e672505679886fe26b9ec2b0db0fefc8dd9 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 23 Apr 2024 09:45:11 -0400
Subject: [PATCH 049/124] [Vertex AI] Re-added `location` parameter to public
 API (#12835)

---
 FirebaseVertexAI/Sources/VertexAI.swift            | 14 ++++++++++----
 FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift |  2 ++
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 7a7865e86d2..44134db5b14 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,27 +25,33 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
+  /// - Parameter location: The region identifier, defaulting to `us-central1`; see [Vertex AI
+  ///   regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions) for a
+  ///   list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI() -> VertexAI {
+  public static func vertexAI(location: String = "us-central1") -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app)
+    return vertexAI(app: app, location: location)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
+  ///   - location: The region identifier, defaulting to `us-central1`; see [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp, location: String = "us-central1") -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI("us-central1")
+    return provider.vertexAI(location)
   }
 
   /// Initializes a generative model with the given parameters.
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 75fb8946d22..50410379b75 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -36,9 +36,11 @@ final class VertexAIAPITests: XCTestCase {
 
     // Instantiate Vertex AI SDK - Default App
     let vertexAI = VertexAI.vertexAI()
+    let _ = VertexAI.vertexAI(location: "my-location")
 
     // Instantiate Vertex AI SDK - Custom App
     let _ = VertexAI.vertexAI(app: app!)
+    let _ = VertexAI.vertexAI(app: app!, location: "my-location")
 
     // Permutations without optional arguments.
 

From a7f44fa2de34aadd1ff078e36b103b35647b99f9 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 23 Apr 2024 09:52:33 -0400
Subject: [PATCH 050/124] [Vertex AI] Remove public `Codable` conformance
 (#12833)

---
 .../Sources/CountTokensRequest.swift          |  16 +-
 .../Sources/FunctionCalling.swift             |  55 ++--
 .../Sources/GenerateContentResponse.swift     | 267 +++++++++---------
 .../Sources/GenerationConfig.swift            |   7 +-
 FirebaseVertexAI/Sources/ModelContent.swift   | 120 ++++----
 FirebaseVertexAI/Sources/Safety.swift         | 108 ++++---
 6 files changed, 310 insertions(+), 263 deletions(-)

diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
index 5072412cf9b..8af409329ad 100644
--- a/FirebaseVertexAI/Sources/CountTokensRequest.swift
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -21,13 +21,6 @@ struct CountTokensRequest {
   let options: RequestOptions
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension CountTokensRequest: Encodable {
-  enum CodingKeys: CodingKey {
-    case contents
-  }
-}
-
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 extension CountTokensRequest: GenerativeAIRequest {
   typealias Response = CountTokensResponse
@@ -50,6 +43,15 @@ public struct CountTokensResponse {
   public let totalBillableCharacters: Int
 }
 
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensRequest: Encodable {
+  enum CodingKeys: CodingKey {
+    case contents
+  }
+}
+
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 extension CountTokensResponse: Decodable {
   enum CodingKeys: CodingKey {
diff --git a/FirebaseVertexAI/Sources/FunctionCalling.swift b/FirebaseVertexAI/Sources/FunctionCalling.swift
index 8860344c7cd..f9f74ee2e99 100644
--- a/FirebaseVertexAI/Sources/FunctionCalling.swift
+++ b/FirebaseVertexAI/Sources/FunctionCalling.swift
@@ -15,7 +15,7 @@
 import Foundation
 
 /// A predicted function call returned from the model.
-public struct FunctionCall: Equatable, Encodable {
+public struct FunctionCall: Equatable {
   /// The name of the function to call.
   public let name: String
 
@@ -27,7 +27,7 @@ public struct FunctionCall: Equatable, Encodable {
 ///
 /// These types can be objects, but also primitives and arrays. Represents a select subset of an
 /// [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema).
-public class Schema: Encodable {
+public class Schema {
   /// The data type.
   let type: DataType
 
@@ -52,17 +52,6 @@ public class Schema: Encodable {
   /// Required properties of type ``DataType/object``.
   let requiredProperties: [String]?
 
-  enum CodingKeys: String, CodingKey {
-    case type
-    case format
-    case description
-    case nullable
-    case enumValues = "enum"
-    case items
-    case properties
-    case requiredProperties = "required"
-  }
-
   /// Constructs a new `Schema`.
   ///
   /// - Parameters:
@@ -98,7 +87,7 @@ public class Schema: Encodable {
 /// A data type.
 ///
 /// Contains the set of OpenAPI [data types](https://spec.openapis.org/oas/v3.0.3#data-types).
-public enum DataType: String, Encodable {
+public enum DataType: String {
   /// A `String` type.
   case string = "STRING"
 
@@ -157,7 +146,7 @@ public struct FunctionDeclaration {
 ///
 /// A `Tool` is a piece of code that enables the system to interact with external systems to
 /// perform an action, or set of actions, outside of knowledge and scope of the model.
-public struct Tool: Encodable {
+public struct Tool {
   /// A list of `FunctionDeclarations` available to the model.
   let functionDeclarations: [FunctionDeclaration]?
 
@@ -178,10 +167,10 @@ public struct Tool: Encodable {
 }
 
 /// Configuration for specifying function calling behavior.
-public struct FunctionCallingConfig: Encodable {
+public struct FunctionCallingConfig {
   /// Defines the execution behavior for function calling by defining the
   /// execution mode.
-  public enum Mode: String, Encodable {
+  public enum Mode: String {
     /// The default behavior for function calling. The model calls functions to answer queries at
     /// its discretion.
     case auto = "AUTO"
@@ -213,8 +202,7 @@ public struct FunctionCallingConfig: Encodable {
 }
 
 /// Tool configuration for any `Tool` specified in the request.
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct ToolConfig: Encodable {
+public struct ToolConfig {
   let functionCallingConfig: FunctionCallingConfig?
 
   public init(functionCallingConfig: FunctionCallingConfig? = nil) {
@@ -227,7 +215,7 @@ public struct ToolConfig: Encodable {
 /// Contains a string representing the `FunctionDeclaration.name` and a structured JSON object
 /// containing any output from the function is used as context to the model. This should contain the
 /// result of a ``FunctionCall`` made based on model prediction.
-public struct FunctionResponse: Equatable, Encodable {
+public struct FunctionResponse: Equatable {
   /// The name of the function that was called.
   let name: String
 
@@ -264,6 +252,8 @@ extension FunctionCall: Decodable {
   }
 }
 
+extension FunctionCall: Encodable {}
+
 extension FunctionDeclaration: Encodable {
   enum CodingKeys: String, CodingKey {
     case name
@@ -278,3 +268,28 @@ extension FunctionDeclaration: Encodable {
     try container.encode(parameters, forKey: .parameters)
   }
 }
+
+extension Schema: Encodable {
+  enum CodingKeys: String, CodingKey {
+    case type
+    case format
+    case description
+    case nullable
+    case enumValues = "enum"
+    case items
+    case properties
+    case requiredProperties = "required"
+  }
+}
+
+extension DataType: Encodable {}
+
+extension Tool: Encodable {}
+
+extension FunctionCallingConfig: Encodable {}
+
+extension FunctionCallingConfig.Mode: Encodable {}
+
+extension ToolConfig: Encodable {}
+
+extension FunctionResponse: Encodable {}
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 48f4ff22413..e89029c2d02 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -74,40 +74,6 @@ public struct GenerateContentResponse {
   }
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension GenerateContentResponse: Decodable {
-  enum CodingKeys: CodingKey {
-    case candidates
-    case promptFeedback
-    case usageMetadata
-  }
-
-  public init(from decoder: Decoder) throws {
-    let container = try decoder.container(keyedBy: CodingKeys.self)
-
-    guard container.contains(CodingKeys.candidates) || container
-      .contains(CodingKeys.promptFeedback) else {
-      let context = DecodingError.Context(
-        codingPath: [],
-        debugDescription: "Failed to decode GenerateContentResponse;" +
-          " missing keys 'candidates' and 'promptFeedback'."
-      )
-      throw DecodingError.dataCorrupted(context)
-    }
-
-    if let candidates = try container.decodeIfPresent(
-      [CandidateResponse].self,
-      forKey: .candidates
-    ) {
-      self.candidates = candidates
-    } else {
-      candidates = []
-    }
-    promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
-    usageMetadata = try container.decodeIfPresent(UsageMetadata.self, forKey: .usageMetadata)
-  }
-}
-
 /// A struct representing a possible reply to a content generation prompt. Each content generation
 /// prompt may produce multiple candidate responses.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -135,62 +101,9 @@ public struct CandidateResponse {
   }
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension CandidateResponse: Decodable {
-  enum CodingKeys: CodingKey {
-    case content
-    case safetyRatings
-    case finishReason
-    case finishMessage
-    case citationMetadata
-  }
-
-  /// Initializes a response from a decoder. Used for decoding server responses; not for public
-  /// use.
-  public init(from decoder: Decoder) throws {
-    let container = try decoder.container(keyedBy: CodingKeys.self)
-
-    do {
-      if let content = try container.decodeIfPresent(ModelContent.self, forKey: .content) {
-        self.content = content
-      } else {
-        content = ModelContent(parts: [])
-      }
-    } catch {
-      // Check if `content` can be decoded as an empty dictionary to detect the `"content": {}` bug.
-      if let content = try? container.decode([String: String].self, forKey: .content),
-         content.isEmpty {
-        throw InvalidCandidateError.emptyContent(underlyingError: error)
-      } else {
-        throw InvalidCandidateError.malformedContent(underlyingError: error)
-      }
-    }
-
-    if let safetyRatings = try container.decodeIfPresent(
-      [SafetyRating].self,
-      forKey: .safetyRatings
-    ) {
-      self.safetyRatings = safetyRatings
-    } else {
-      safetyRatings = []
-    }
-
-    finishReason = try container.decodeIfPresent(FinishReason.self, forKey: .finishReason)
-
-    citationMetadata = try container.decodeIfPresent(
-      CitationMetadata.self,
-      forKey: .citationMetadata
-    )
-  }
-}
-
 /// A collection of source attributions for a piece of content.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct CitationMetadata: Decodable {
-  enum CodingKeys: String, CodingKey {
-    case citationSources = "citations"
-  }
-
+public struct CitationMetadata {
   /// A list of individual cited sources and the parts of the content to which they apply.
   public let citationSources: [Citation]
 }
@@ -236,27 +149,11 @@ public enum FinishReason: String {
   case other = "OTHER"
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension FinishReason: Decodable {
-  /// Do not explicitly use. Initializer required for Decodable conformance.
-  public init(from decoder: Decoder) throws {
-    let value = try decoder.singleValueContainer().decode(String.self)
-    guard let decodedFinishReason = FinishReason(rawValue: value) else {
-      Logging.default
-        .error("[GoogleGenerativeAI] Unrecognized FinishReason with value \"\(value)\".")
-      self = .unknown
-      return
-    }
-
-    self = decodedFinishReason
-  }
-}
-
 /// A metadata struct containing any feedback the model had on the prompt it was provided.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public struct PromptFeedback {
   /// A type describing possible reasons to block a prompt.
-  public enum BlockReason: String, Decodable {
+  public enum BlockReason: String {
     /// The block reason is unknown.
     case unknown = "UNKNOWN"
 
@@ -268,19 +165,6 @@ public struct PromptFeedback {
 
     /// All other block reasons.
     case other = "OTHER"
-
-    /// Do not explicitly use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedBlockReason = BlockReason(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized BlockReason with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedBlockReason
-    }
   }
 
   /// The reason a prompt was blocked, if it was blocked.
@@ -296,28 +180,39 @@ public struct PromptFeedback {
   }
 }
 
+// MARK: - Codable Conformances
+
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension PromptFeedback: Decodable {
+extension GenerateContentResponse: Decodable {
   enum CodingKeys: CodingKey {
-    case blockReason
-    case safetyRatings
+    case candidates
+    case promptFeedback
+    case usageMetadata
   }
 
-  /// Do not explicitly use. Initializer required for Decodable conformance.
   public init(from decoder: Decoder) throws {
     let container = try decoder.container(keyedBy: CodingKeys.self)
-    blockReason = try container.decodeIfPresent(
-      PromptFeedback.BlockReason.self,
-      forKey: .blockReason
-    )
-    if let safetyRatings = try container.decodeIfPresent(
-      [SafetyRating].self,
-      forKey: .safetyRatings
+
+    guard container.contains(CodingKeys.candidates) || container
+      .contains(CodingKeys.promptFeedback) else {
+      let context = DecodingError.Context(
+        codingPath: [],
+        debugDescription: "Failed to decode GenerateContentResponse;" +
+          " missing keys 'candidates' and 'promptFeedback'."
+      )
+      throw DecodingError.dataCorrupted(context)
+    }
+
+    if let candidates = try container.decodeIfPresent(
+      [CandidateResponse].self,
+      forKey: .candidates
     ) {
-      self.safetyRatings = safetyRatings
+      self.candidates = candidates
     } else {
-      safetyRatings = []
+      candidates = []
     }
+    promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
+    usageMetadata = try container.decodeIfPresent(UsageMetadata.self, forKey: .usageMetadata)
   }
 }
 
@@ -338,7 +233,61 @@ extension GenerateContentResponse.UsageMetadata: Decodable {
   }
 }
 
-// MARK: - Codable Conformances
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CandidateResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case content
+    case safetyRatings
+    case finishReason
+    case finishMessage
+    case citationMetadata
+  }
+
+  /// Initializes a response from a decoder. Used for decoding server responses; not for public
+  /// use.
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+
+    do {
+      if let content = try container.decodeIfPresent(ModelContent.self, forKey: .content) {
+        self.content = content
+      } else {
+        content = ModelContent(parts: [])
+      }
+    } catch {
+      // Check if `content` can be decoded as an empty dictionary to detect the `"content": {}` bug.
+      if let content = try? container.decode([String: String].self, forKey: .content),
+         content.isEmpty {
+        throw InvalidCandidateError.emptyContent(underlyingError: error)
+      } else {
+        throw InvalidCandidateError.malformedContent(underlyingError: error)
+      }
+    }
+
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+
+    finishReason = try container.decodeIfPresent(FinishReason.self, forKey: .finishReason)
+
+    citationMetadata = try container.decodeIfPresent(
+      CitationMetadata.self,
+      forKey: .citationMetadata
+    )
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CitationMetadata: Decodable {
+  enum CodingKeys: String, CodingKey {
+    case citationSources = "citations"
+  }
+}
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 extension Citation: Decodable {
@@ -357,3 +306,57 @@ extension Citation: Decodable {
     license = try container.decodeIfPresent(String.self, forKey: .license)
   }
 }
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension FinishReason: Decodable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedFinishReason = FinishReason(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized FinishReason with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedFinishReason
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension PromptFeedback.BlockReason: Decodable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedBlockReason = PromptFeedback.BlockReason(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized BlockReason with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedBlockReason
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension PromptFeedback: Decodable {
+  enum CodingKeys: CodingKey {
+    case blockReason
+    case safetyRatings
+  }
+
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    blockReason = try container.decodeIfPresent(
+      PromptFeedback.BlockReason.self,
+      forKey: .blockReason
+    )
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerationConfig.swift b/FirebaseVertexAI/Sources/GenerationConfig.swift
index 2d1016c965c..18bf2bd1a52 100644
--- a/FirebaseVertexAI/Sources/GenerationConfig.swift
+++ b/FirebaseVertexAI/Sources/GenerationConfig.swift
@@ -17,7 +17,7 @@ import Foundation
 /// A struct defining model parameters to be used when sending generative AI
 /// requests to the backend model.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct GenerationConfig: Encodable {
+public struct GenerationConfig {
   /// A parameter controlling the degree of randomness in token selection. A
   /// temperature of zero is deterministic, always choosing the
   /// highest-probability response. Typical values are between 0 and 1
@@ -84,3 +84,8 @@ public struct GenerationConfig: Encodable {
     self.stopSequences = stopSequences
   }
 }
+
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerationConfig: Encodable {}
diff --git a/FirebaseVertexAI/Sources/ModelContent.swift b/FirebaseVertexAI/Sources/ModelContent.swift
index 136fc1debe3..bae04b13c18 100644
--- a/FirebaseVertexAI/Sources/ModelContent.swift
+++ b/FirebaseVertexAI/Sources/ModelContent.swift
@@ -18,22 +18,10 @@ import Foundation
 /// request or response contains an `Array` of ``ModelContent``s, and each ``ModelContent`` value
 /// may comprise multiple heterogeneous ``ModelContent/Part``s.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct ModelContent: Codable, Equatable {
+public struct ModelContent: Equatable {
   /// A discrete piece of data in a media format intepretable by an AI model. Within a single value
   /// of ``Part``, different data types may not mix.
-  public enum Part: Codable, Equatable {
-    enum CodingKeys: String, CodingKey {
-      case text
-      case inlineData
-      case functionCall
-      case functionResponse
-    }
-
-    enum InlineDataKeys: String, CodingKey {
-      case mimeType = "mime_type"
-      case bytes = "data"
-    }
-
+  public enum Part: Equatable {
     /// Text value.
     case text(String)
 
@@ -58,49 +46,6 @@ public struct ModelContent: Codable, Equatable {
       return .data(mimetype: "image/png", data)
     }
 
-    // MARK: Codable Conformance
-
-    public func encode(to encoder: Encoder) throws {
-      var container = encoder.container(keyedBy: ModelContent.Part.CodingKeys.self)
-      switch self {
-      case let .text(a0):
-        try container.encode(a0, forKey: .text)
-      case let .data(mimetype, bytes):
-        var inlineDataContainer = container.nestedContainer(
-          keyedBy: InlineDataKeys.self,
-          forKey: .inlineData
-        )
-        try inlineDataContainer.encode(mimetype, forKey: .mimeType)
-        try inlineDataContainer.encode(bytes, forKey: .bytes)
-      case let .functionCall(functionCall):
-        try container.encode(functionCall, forKey: .functionCall)
-      case let .functionResponse(functionResponse):
-        try container.encode(functionResponse, forKey: .functionResponse)
-      }
-    }
-
-    public init(from decoder: Decoder) throws {
-      let values = try decoder.container(keyedBy: CodingKeys.self)
-      if values.contains(.text) {
-        self = try .text(values.decode(String.self, forKey: .text))
-      } else if values.contains(.inlineData) {
-        let dataContainer = try values.nestedContainer(
-          keyedBy: InlineDataKeys.self,
-          forKey: .inlineData
-        )
-        let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
-        let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
-        self = .data(mimetype: mimetype, bytes)
-      } else if values.contains(.functionCall) {
-        self = try .functionCall(values.decode(FunctionCall.self, forKey: .functionCall))
-      } else {
-        throw DecodingError.dataCorrupted(.init(
-          codingPath: [CodingKeys.text, CodingKeys.inlineData],
-          debugDescription: "No text, inline data or function call was found."
-        ))
-      }
-    }
-
     /// Returns the text contents of this ``Part``, if it contains text.
     public var text: String? {
       switch self {
@@ -153,3 +98,64 @@ public struct ModelContent: Codable, Equatable {
     self.init(role: role, parts: content)
   }
 }
+
+// MARK: Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension ModelContent: Codable {}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension ModelContent.Part: Codable {
+  enum CodingKeys: String, CodingKey {
+    case text
+    case inlineData
+    case functionCall
+    case functionResponse
+  }
+
+  enum InlineDataKeys: String, CodingKey {
+    case mimeType = "mime_type"
+    case bytes = "data"
+  }
+
+  public func encode(to encoder: Encoder) throws {
+    var container = encoder.container(keyedBy: CodingKeys.self)
+    switch self {
+    case let .text(a0):
+      try container.encode(a0, forKey: .text)
+    case let .data(mimetype, bytes):
+      var inlineDataContainer = container.nestedContainer(
+        keyedBy: InlineDataKeys.self,
+        forKey: .inlineData
+      )
+      try inlineDataContainer.encode(mimetype, forKey: .mimeType)
+      try inlineDataContainer.encode(bytes, forKey: .bytes)
+    case let .functionCall(functionCall):
+      try container.encode(functionCall, forKey: .functionCall)
+    case let .functionResponse(functionResponse):
+      try container.encode(functionResponse, forKey: .functionResponse)
+    }
+  }
+
+  public init(from decoder: Decoder) throws {
+    let values = try decoder.container(keyedBy: CodingKeys.self)
+    if values.contains(.text) {
+      self = try .text(values.decode(String.self, forKey: .text))
+    } else if values.contains(.inlineData) {
+      let dataContainer = try values.nestedContainer(
+        keyedBy: InlineDataKeys.self,
+        forKey: .inlineData
+      )
+      let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
+      let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
+      self = .data(mimetype: mimetype, bytes)
+    } else if values.contains(.functionCall) {
+      self = try .functionCall(values.decode(FunctionCall.self, forKey: .functionCall))
+    } else {
+      throw DecodingError.dataCorrupted(.init(
+        codingPath: [CodingKeys.text, CodingKeys.inlineData],
+        debugDescription: "No text, inline data or function call was found."
+      ))
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/Safety.swift b/FirebaseVertexAI/Sources/Safety.swift
index d59511f2476..d2adc4a81e1 100644
--- a/FirebaseVertexAI/Sources/Safety.swift
+++ b/FirebaseVertexAI/Sources/Safety.swift
@@ -18,7 +18,7 @@ import Foundation
 /// of this type may be assigned to a category for every model-generated response, not just
 /// responses that exceed a certain threshold.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct SafetyRating: Decodable, Equatable, Hashable {
+public struct SafetyRating: Equatable, Hashable {
   /// The category describing the potential harm a piece of content may pose. See
   /// ``SafetySetting/HarmCategory`` for a list of possible values.
   public let category: SafetySetting.HarmCategory
@@ -38,7 +38,7 @@ public struct SafetyRating: Decodable, Equatable, Hashable {
 
   /// The probability that a given model output falls under a harmful content category. This does
   /// not indicate the severity of harm for a piece of content.
-  public enum HarmProbability: String, Codable {
+  public enum HarmProbability: String {
     /// Unknown. A new server value that isn't recognized by the SDK.
     case unknown = "UNKNOWN"
 
@@ -57,26 +57,12 @@ public struct SafetyRating: Decodable, Equatable, Hashable {
 
     /// The probability is high. The content described is very likely harmful.
     case high = "HIGH"
-
-    /// Initializes a new `SafetyRating` from a decoder.
-    /// Not for external use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedProbability = HarmProbability(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized HarmProbability with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedProbability
-    }
   }
 }
 
 /// Safety feedback for an entire request.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct SafetyFeedback: Decodable {
+public struct SafetyFeedback {
   /// Safety rating evaluated from content.
   public let rating: SafetyRating
 
@@ -93,10 +79,10 @@ public struct SafetyFeedback: Decodable {
 /// A type used to specify a threshold for harmful content, beyond which the model will return a
 /// fallback response instead of generated content.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct SafetySetting: Codable {
+public struct SafetySetting {
   /// A type describing safety attributes, which include harmful categories and topics that can
   /// be considered sensitive.
-  public enum HarmCategory: String, Codable {
+  public enum HarmCategory: String {
     /// Unknown. A new server value that isn't recognized by the SDK.
     case unknown = "HARM_CATEGORY_UNKNOWN"
 
@@ -114,23 +100,10 @@ public struct SafetySetting: Codable {
 
     /// Promotes or enables access to harmful goods, services, or activities.
     case dangerousContent = "HARM_CATEGORY_DANGEROUS_CONTENT"
-
-    /// Do not explicitly use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedCategory = HarmCategory(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized HarmCategory with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedCategory
-    }
   }
 
   /// Block at and beyond a specified ``SafetyRating/HarmProbability``.
-  public enum BlockThreshold: String, Codable {
+  public enum BlockThreshold: String {
     /// Unknown. A new server value that isn't recognized by the SDK.
     case unknown = "UNKNOWN"
 
@@ -148,19 +121,6 @@ public struct SafetySetting: Codable {
 
     /// All content will be allowed.
     case blockNone = "BLOCK_NONE"
-
-    /// Do not explicitly use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedThreshold = BlockThreshold(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized BlockThreshold with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedThreshold
-    }
   }
 
   enum CodingKeys: String, CodingKey {
@@ -180,3 +140,59 @@ public struct SafetySetting: Codable {
     self.threshold = threshold
   }
 }
+
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetyRating.HarmProbability: Codable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedProbability = SafetyRating.HarmProbability(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized HarmProbability with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedProbability
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetyRating: Decodable {}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetyFeedback: Decodable {}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetySetting.HarmCategory: Codable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedCategory = SafetySetting.HarmCategory(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized HarmCategory with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedCategory
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetySetting.BlockThreshold: Codable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedThreshold = SafetySetting.BlockThreshold(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized BlockThreshold with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedThreshold
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetySetting: Codable {}

From 6e1a429e51af7c2ee4bc727ed44e153abd61367b Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 25 Apr 2024 11:41:49 -0400
Subject: [PATCH 051/124] [Vertex AI] Add `location` checks for slashes and
 whitespace (#12849)

---
 .github/workflows/api_diff_report.yml   |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift | 18 ++++++++++--------
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/api_diff_report.yml b/.github/workflows/api_diff_report.yml
index 650f94f4727..8ca8c385b81 100644
--- a/.github/workflows/api_diff_report.yml
+++ b/.github/workflows/api_diff_report.yml
@@ -36,7 +36,7 @@ jobs:
       - name: Setup python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.7
+          python-version: '3.10'
 
       - name: Install Prerequisites
         run:  ~/api_diff_report/prerequisite.sh
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 44134db5b14..a8f4abbf04d 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -26,8 +26,9 @@ public class VertexAI: NSObject {
   /// The default `VertexAI` instance.
   ///
   /// - Parameter location: The region identifier, defaulting to `us-central1`; see [Vertex AI
-  ///   regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions) for a
-  ///   list of supported regions.
+  /// regions
+  ///   ](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions)
+  ///   for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
   public static func vertexAI(location: String = "us-central1") -> VertexAI {
     guard let app = FirebaseApp.app() else {
@@ -41,8 +42,8 @@ public class VertexAI: NSObject {
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
-  ///   - location: The region identifier, defaulting to `us-central1`; see [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///   - location: The region identifier, defaulting to `us-central1`; see [Vertex AI regions
+  ///     ](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions)
   ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
   public static func vertexAI(app: FirebaseApp, location: String = "us-central1") -> VertexAI {
@@ -117,11 +118,12 @@ public class VertexAI: NSObject {
     guard let projectID = app.options.projectID else {
       fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
     }
-    guard !location.isEmpty else {
+    guard !location.isEmpty && location
+      .allSatisfy({ !$0.isWhitespace && !$0.isNewline && $0 != "/" }) else {
       fatalError("""
-      No location specified; see
-      https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
-      list of available regions.
+      Invalid location "\(location)" specified; see \
+      https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for \
+      a list of available regions.
       """)
     }
 

From fb48938190bb5b691e0b6e7b4f91145ad97f7cb8 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 29 Apr 2024 13:49:15 -0400
Subject: [PATCH 052/124] Cherrypick CI fixes into vertex-ai branch (#12862)

Co-authored-by: Paul Beusterien <paulbeusterien@google.com>
---
 .github/workflows/functions.yml   | 119 +++++++++++++++---------------
 .github/workflows/performance.yml |   7 +-
 .github/workflows/prerelease.yml  |  81 ++++++++++----------
 .github/workflows/release.yml     |  82 ++++++++++----------
 .github/workflows/storage.yml     |   4 +-
 5 files changed, 148 insertions(+), 145 deletions(-)

diff --git a/.github/workflows/functions.yml b/.github/workflows/functions.yml
index 915b25118e1..bac9f2c2457 100644
--- a/.github/workflows/functions.yml
+++ b/.github/workflows/functions.yml
@@ -110,68 +110,69 @@ jobs:
     - name: Unit Tests
       run: scripts/third_party/travis/retry.sh ./scripts/build.sh FirebaseFunctionsUnit ${{ matrix.target }} spm
 
-  quickstart:
-    # Don't run on private repo unless it is a PR.
-    if: (github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule') || github.event_name == 'pull_request'
-    env:
-      plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      LEGACY: true
-    # TODO: Move to macos-14 and Xcode 15. The legacy quickstart uses material which doesn't build on Xcode 15.
-    runs-on: macos-12
+  # TODO: Move to macos-14 and Xcode 15. The legacy quickstart uses material which doesn't build on Xcode 15.
+  # quickstart:
+  #   # Don't run on private repo unless it is a PR.
+  #   if: (github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule') || github.event_name == 'pull_request'
+  #   env:
+  #     plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     LEGACY: true
+  #   # TODO: Move to macos-14 and Xcode 15. The legacy quickstart uses material which doesn't build on Xcode 15.
+  #   runs-on: macos-12
 
-    steps:
-    - uses: actions/checkout@v4
-    - uses: ruby/setup-ruby@v1
-    - name: Setup quickstart
-      run: scripts/setup_quickstart.sh functions
-    - name: install secret googleservice-info.plist
-      run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
-          quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
-    - name: Setup custom URL scheme
-      run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
-    - name: Test objc quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true)
-    - name: Test swift quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true swift)
+  #   steps:
+  #   - uses: actions/checkout@v4
+  #   - uses: ruby/setup-ruby@v1
+  #   - name: Setup quickstart
+  #     run: scripts/setup_quickstart.sh functions
+  #   - name: install secret googleservice-info.plist
+  #     run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
+  #         quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
+  #   - name: Setup custom URL scheme
+  #     run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
+  #   - name: Test objc quickstart
+  #     run: ([ -z $plist_secret ] ||
+  #           scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true)
+  #   - name: Test swift quickstart
+  #     run: ([ -z $plist_secret ] ||
+  #           scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true swift)
 
-  quickstart-ftl-cron-only:
-    # Don't run on private repo
-    if: github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule'
-    env:
-      plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      LEGACY: true
-    # TODO: Move to macos-14 and Xcode 15. The legacy quickstart uses material which doesn't build on Xcode 15.
-    runs-on: macos-12
+  # quickstart-ftl-cron-only:
+  #   # Don't run on private repo
+  #   if: github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule'
+  #   env:
+  #     plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     LEGACY: true
+  #   # TODO: Move to macos-14 and Xcode 15. The legacy quickstart uses material which doesn't build on Xcode 15.
+  #   runs-on: macos-12
 
-    steps:
-    - uses: actions/checkout@v4
-    - uses: ruby/setup-ruby@v1
-    - uses: actions/setup-python@v4
-      with:
-        python-version: '3.11'
-    - name: Setup quickstart
-      run: scripts/setup_quickstart.sh functions
-    - name: install secret googleservice-info.plist
-      run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
-          quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
-    - name: Setup custom URL scheme
-      run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
-    - name: Build objc quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart_ftl.sh Functions)
-    - name: Build swift quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart_ftl.sh Functions swift)
-    - id: ftl_test
-      uses: FirebaseExtended/github-actions/firebase-test-lab@v1.4
-      with:
-        credentials_json: ${{ secrets.FIREBASE_SERVICE_ACCOUNT_CREDENTIALS }}
-        testapp_dir: quickstart-ios/build-for-testing
-        test_type: "xctest"
+  #   steps:
+  #   - uses: actions/checkout@v4
+  #   - uses: ruby/setup-ruby@v1
+  #   - uses: actions/setup-python@v4
+  #     with:
+  #       python-version: '3.11'
+  #   - name: Setup quickstart
+  #     run: scripts/setup_quickstart.sh functions
+  #   - name: install secret googleservice-info.plist
+  #     run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
+  #         quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
+  #   - name: Setup custom URL scheme
+  #     run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
+  #   - name: Build objc quickstart
+  #     run: ([ -z $plist_secret ] ||
+  #           scripts/third_party/travis/retry.sh scripts/test_quickstart_ftl.sh Functions)
+  #   - name: Build swift quickstart
+  #     run: ([ -z $plist_secret ] ||
+  #           scripts/third_party/travis/retry.sh scripts/test_quickstart_ftl.sh Functions swift)
+  #   - id: ftl_test
+  #     uses: FirebaseExtended/github-actions/firebase-test-lab@v1.4
+  #     with:
+  #       credentials_json: ${{ secrets.FIREBASE_SERVICE_ACCOUNT_CREDENTIALS }}
+  #       testapp_dir: quickstart-ios/build-for-testing
+  #       test_type: "xctest"
 
   functions-cron-only:
     # Don't run on private repo.
diff --git a/.github/workflows/performance.yml b/.github/workflows/performance.yml
index 4e5ba16eb98..ce0a4ac8790 100644
--- a/.github/workflows/performance.yml
+++ b/.github/workflows/performance.yml
@@ -79,7 +79,7 @@ jobs:
     env:
       plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
       signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-    runs-on: macos-12  # TODO: the legacy ObjC quickstarts don't run with Xcode 15.
+    runs-on: macos-14
     steps:
     - uses: actions/checkout@v4
     - uses: ruby/setup-ruby@v1
@@ -90,8 +90,9 @@ jobs:
           quickstart-ios/performance/GoogleService-Info.plist "$plist_secret"
     - name: Test swift quickstart
       run: ([ -z $plist_secret ] || scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Performance true swift)
-    - name: Test objc quickstart
-      run: ([ -z $plist_secret ] || scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Performance true)
+    # TODO: The legacy ObjC quickstarts don't run with Xcode 15, re-able if we get these working.
+    # - name: Test objc quickstart
+    #   run: ([ -z $plist_secret ] || scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Performance true)
 
   quickstart-ftl-cron-only:
     if: github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule'
diff --git a/.github/workflows/prerelease.yml b/.github/workflows/prerelease.yml
index 6453cc4960a..6312d720461 100644
--- a/.github/workflows/prerelease.yml
+++ b/.github/workflows/prerelease.yml
@@ -434,47 +434,48 @@ jobs:
         name: quickstart_artifacts_firestore
         path: quickstart-ios/
 
-  functions_quickstart:
+  # TODO: The functions quickstart uses Material which isn't supported by Xcode 15
+  #functions_quickstart:
     # Don't run on private repo unless it is a PR.
-    if: (github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule') || github.event_name == 'workflow_dispatch'
-    needs: buildup_SpecsTesting_repo
-    env:
-      plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      bot_token_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      testing_repo_dir: "/tmp/test/"
-      testing_repo: "firebase-ios-sdk"
-      LEGACY: true
-    # TODO: The functions quickstart uses Material which isn't supported by Xcode 15
-    runs-on: macos-12
-    steps:
-    - uses: actions/checkout@v4
-    - uses: ruby/setup-ruby@v1
-    - name: Get token
-      run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/bot-access.txt.gpg \
-          bot-access.txt "$bot_token_secret"
-    - name: Setup testing repo and quickstart
-      run: |
-          botaccess=`cat bot-access.txt`
-          BOT_TOKEN="${botaccess}" scripts/setup_quickstart.sh functions prerelease_testing
-    - name: install secret googleservice-info.plist
-      run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
-          quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
-    - name: Setup custom URL scheme
-      run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
-    - name: Test objc quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true)
-    - name: Test swift quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true swift)
-    - name: Remove data before upload
-      run: scripts/remove_data.sh functions release_testing
-    - uses: actions/upload-artifact@v4
-      if: ${{ failure() }}
-      with:
-        name: quickstart_artifacts_functions
-        path: quickstart-ios/
+    # if: (github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule') || github.event_name == 'workflow_dispatch'
+    # needs: buildup_SpecsTesting_repo
+    # env:
+    #   plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+    #   signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+    #   bot_token_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+    #   testing_repo_dir: "/tmp/test/"
+    #   testing_repo: "firebase-ios-sdk"
+    #   LEGACY: true
+    # # TODO: The functions quickstart uses Material which isn't supported by Xcode 15
+    # runs-on: macos-12
+    # steps:
+    # - uses: actions/checkout@v4
+    # - uses: ruby/setup-ruby@v1
+    # - name: Get token
+    #   run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/bot-access.txt.gpg \
+    #       bot-access.txt "$bot_token_secret"
+    # - name: Setup testing repo and quickstart
+    #   run: |
+    #       botaccess=`cat bot-access.txt`
+    #       BOT_TOKEN="${botaccess}" scripts/setup_quickstart.sh functions prerelease_testing
+    # - name: install secret googleservice-info.plist
+    #   run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
+    #       quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
+    # - name: Setup custom URL scheme
+    #   run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
+    # - name: Test objc quickstart
+    #   run: ([ -z $plist_secret ] ||
+    #         scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true)
+    # - name: Test swift quickstart
+    #   run: ([ -z $plist_secret ] ||
+    #         scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true swift)
+    # - name: Remove data before upload
+    #   run: scripts/remove_data.sh functions release_testing
+    # - uses: actions/upload-artifact@v4
+    #   if: ${{ failure() }}
+    #   with:
+    #     name: quickstart_artifacts_functions
+    #     path: quickstart-ios/
 
   inappmessaging_quickstart:
     # Don't run on private repo unless it is a PR.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 0604a1de1d9..097f590b274 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -385,47 +385,47 @@ jobs:
         name: quickstart_artifacts_firestore
         path: quickstart-ios/
 
-  functions_quickstart:
-    # Don't run on private repo unless it is a PR.
-    if: (github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule') || github.event_name == 'pull_request' || github.event_name == 'workflow_dispatch'
-    needs: buildup_SpecsTesting_repo
-    env:
-      plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      bot_token_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
-      testing_repo_dir: "/tmp/test/"
-      testing_repo: "firebase-ios-sdk"
-      LEGACY: true
-    # TODO: The functions quickstart uses Material which isn't supported by Xcode 15
-    runs-on: macos-12
-    steps:
-    - uses: actions/checkout@v4
-    - uses: ruby/setup-ruby@v1
-    - name: Get token
-      run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/bot-access.txt.gpg \
-          bot-access.txt "$bot_token_secret"
-    - name: Setup testing repo and quickstart
-      run: |
-          botaccess=`cat bot-access.txt`
-          BOT_TOKEN="${botaccess}" scripts/setup_quickstart.sh functions nightly_release_testing
-    - name: install secret googleservice-info.plist
-      run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
-          quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
-    - name: Setup custom URL scheme
-      run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
-    - name: Test objc quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true)
-    - name: Test swift quickstart
-      run: ([ -z $plist_secret ] ||
-            scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true swift)
-    - name: Remove data before upload
-      run: scripts/remove_data.sh functions release_testing
-    - uses: actions/upload-artifact@v4
-      if: ${{ failure() }}
-      with:
-        name: quickstart_artifacts_functions
-        path: quickstart-ios/
+  # TODO: The functions quickstart uses Material which isn't supported by Xcode 15
+  # functions_quickstart:
+  #   # Don't run on private repo unless it is a PR.
+  #   if: (github.repository == 'Firebase/firebase-ios-sdk' && github.event_name == 'schedule') || github.event_name == 'pull_request' || github.event_name == 'workflow_dispatch'
+  #   needs: buildup_SpecsTesting_repo
+  #   env:
+  #     plist_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     signin_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     bot_token_secret: ${{ secrets.GHASecretsGPGPassphrase1 }}
+  #     testing_repo_dir: "/tmp/test/"
+  #     testing_repo: "firebase-ios-sdk"
+  #     LEGACY: true
+  #   runs-on: macos-12
+  #   steps:
+  #   - uses: actions/checkout@v4
+  #   - uses: ruby/setup-ruby@v1
+  #   - name: Get token
+  #     run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/bot-access.txt.gpg \
+  #         bot-access.txt "$bot_token_secret"
+  #   - name: Setup testing repo and quickstart
+  #     run: |
+  #         botaccess=`cat bot-access.txt`
+  #         BOT_TOKEN="${botaccess}" scripts/setup_quickstart.sh functions nightly_release_testing
+  #   - name: install secret googleservice-info.plist
+  #     run: scripts/decrypt_gha_secret.sh scripts/gha-encrypted/qs-functions.plist.gpg \
+  #         quickstart-ios/functions/GoogleService-Info.plist "$plist_secret"
+  #   - name: Setup custom URL scheme
+  #     run: sed -i '' 's/REVERSED_CLIENT_ID/com.googleusercontent.apps.1025801074639-6p6ebi8amuklcjrto20gvpe295smm8u6/' quickstart-ios/functions/LegacyFunctionsQuickstart/FunctionsExample/Info.plist
+  #   - name: Test objc quickstart
+  #     run: ([ -z $plist_secret ] ||
+  #           scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true)
+  #   - name: Test swift quickstart
+  #     run: ([ -z $plist_secret ] ||
+  #           scripts/third_party/travis/retry.sh scripts/test_quickstart.sh Functions true swift)
+  #   - name: Remove data before upload
+  #     run: scripts/remove_data.sh functions release_testing
+  #   - uses: actions/upload-artifact@v4
+  #     if: ${{ failure() }}
+  #     with:
+  #       name: quickstart_artifacts_functions
+  #       path: quickstart-ios/
 
   inappmessaging_quickstart:
     # Don't run on private repo unless it is a PR.
diff --git a/.github/workflows/storage.yml b/.github/workflows/storage.yml
index a651334bb99..4409dadc0ca 100644
--- a/.github/workflows/storage.yml
+++ b/.github/workflows/storage.yml
@@ -96,8 +96,8 @@ jobs:
     strategy:
       matrix:
         include:
-          - os: macos-13
-            xcode: Xcode_14.2 # TODO: the legacy ObjC quickstart doesn't build with Xcode 15.
+          #- os: macos-13
+          #  xcode: Xcode_14.2 # TODO: the legacy ObjC quickstart doesn't build with Xcode 15.
           - swift: swift
             os: macos-14
             xcode: Xcode_15.3

From 239ef8f039c658c0fa7d9ad8cc69e39ce4f9e667 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 29 Apr 2024 16:17:48 -0400
Subject: [PATCH 053/124] [Vertex AI] Integrate with AppCheckInterop (#12856)

---
 FirebaseAuth/Interop/FIRAuthInterop.h         |  4 +-
 .../Sources/GenerativeAIService.swift         | 10 +-
 .../Sources/GenerativeModel.swift             |  3 +
 FirebaseVertexAI/Sources/VertexAI.swift       |  7 +-
 .../Sources/VertexAIComponent.swift           |  3 +
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |  1 +
 .../Tests/Unit/Fakes/AuthInteropFake.swift    | 49 ++++++++++
 .../Tests/Unit/GenerativeModelTests.swift     | 95 ++++++++++++++++++-
 Package.swift                                 |  1 +
 9 files changed, 165 insertions(+), 8 deletions(-)
 create mode 100644 FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift

diff --git a/FirebaseAuth/Interop/FIRAuthInterop.h b/FirebaseAuth/Interop/FIRAuthInterop.h
index f710e39fad6..a50669da433 100644
--- a/FirebaseAuth/Interop/FIRAuthInterop.h
+++ b/FirebaseAuth/Interop/FIRAuthInterop.h
@@ -24,7 +24,7 @@ NS_ASSUME_NONNULL_BEGIN
 /** @typedef FIRTokenCallback
  @brief The type of block which gets called when a token is ready.
  */
-typedef void (^FIRTokenCallback)(NSString *_Nullable token, NSError *_Nullable error)
+typedef void (^FIRTokenCallback)(NSString *_Nullable_result token, NSError *_Nullable error)
     NS_SWIFT_UNAVAILABLE("Use Swift's closure syntax instead.");
 
 /// Common methods for Auth interoperability.
@@ -34,7 +34,7 @@ NS_SWIFT_NAME(AuthInterop)
 /// Retrieves the Firebase authentication token, possibly refreshing it if it has expired.
 - (void)getTokenForcingRefresh:(BOOL)forceRefresh
                   withCallback:
-                      (void (^)(NSString *_Nullable token, NSError *_Nullable error))callback
+                      (void (^)(NSString *_Nullable_result token, NSError *_Nullable error))callback
     NS_SWIFT_NAME(getToken(forcingRefresh:completion:));
 
 /// Get the current Auth user's UID. Returns nil if there is no user signed in.
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index 60ca375c9f6..dbf26c2c948 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import Foundation
 
@@ -29,11 +30,14 @@ struct GenerativeAIService {
 
   private let appCheck: AppCheckInterop?
 
+  private let auth: AuthInterop?
+
   private let urlSession: URLSession
 
-  init(apiKey: String, appCheck: AppCheckInterop?, urlSession: URLSession) {
+  init(apiKey: String, appCheck: AppCheckInterop?, auth: AuthInterop?, urlSession: URLSession) {
     self.apiKey = apiKey
     self.appCheck = appCheck
+    self.auth = auth
     self.urlSession = urlSession
   }
 
@@ -176,6 +180,10 @@ struct GenerativeAIService {
       }
     }
 
+    if let auth, let authToken = try await auth.getToken(forcingRefresh: false) {
+      urlRequest.setValue("Firebase \(authToken)", forHTTPHeaderField: "Authorization")
+    }
+
     let encoder = JSONEncoder()
     encoder.keyEncodingStrategy = .convertToSnakeCase
     urlRequest.httpBody = try encoder.encode(request)
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index cdd6d175c0a..0656edd6a7e 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import Foundation
 
 /// A type that represents a remote multimodal model (like Gemini), with the ability to generate
@@ -69,11 +70,13 @@ public final class GenerativeModel {
        systemInstruction: ModelContent? = nil,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
+       auth: AuthInterop?,
        urlSession: URLSession = .shared) {
     modelResourceName = GenerativeModel.modelResourceName(name: name)
     generativeAIService = GenerativeAIService(
       apiKey: apiKey,
       appCheck: appCheck,
+      auth: auth,
       urlSession: urlSession
     )
     self.generationConfig = generationConfig
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index a8f4abbf04d..53655eada79 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import Foundation
 
@@ -92,7 +93,8 @@ public class VertexAI: NSObject {
       toolConfig: toolConfig,
       systemInstruction: systemInstruction,
       requestOptions: requestOptions,
-      appCheck: appCheck
+      appCheck: appCheck,
+      auth: auth
     )
   }
 
@@ -103,12 +105,15 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
+  private let auth: AuthInterop?
+
   let location: String
 
   init(app: FirebaseApp, location: String) {
     self.app = app
     self.location = location
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
+    auth = ComponentType<AuthInterop>.instance(for: AuthInterop.self, in: app.container)
   }
 
   private func modelResourceName(modelName: String, location: String) -> String {
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a6d487e1208..c43cab5f1f6 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import Foundation
 
@@ -51,10 +52,12 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   static func componentsToRegister() -> [Component] {
     let appCheckInterop = Dependency(with: AppCheckInterop.self, isRequired: false)
+    let authInterop = Dependency(with: AuthInterop.self, isRequired: false)
     return [Component(VertexAIProvider.self,
                       instantiationTiming: .lazy,
                       dependencies: [
                         appCheckInterop,
+                        authInterop,
                       ]) { container, isCacheable in
         guard let app = container.app else { return nil }
         isCacheable.pointee = true
diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 20897a8787b..ea2633bdf40 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -53,6 +53,7 @@ final class ChatTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
     let chat = Chat(model: model, history: [])
diff --git a/FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift b/FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift
new file mode 100644
index 00000000000..6d0d33e1b41
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift
@@ -0,0 +1,49 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseAuthInterop
+import Foundation
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+class AuthInteropFake: NSObject, AuthInterop {
+  let token: String?
+  let error: Error?
+
+  func getToken(forcingRefresh forceRefresh: Bool) async throws -> String? {
+    if let error {
+      throw error
+    }
+
+    return token
+  }
+
+  func getUserID() -> String? {
+    fatalError("\(#function) not implemented.")
+  }
+
+  private init(token: String?, error: Error?) {
+    self.token = token
+    self.error = error
+  }
+
+  convenience init(error: Error) {
+    self.init(token: nil, error: error)
+  }
+
+  convenience init(token: String?) {
+    self.init(token: token, error: nil)
+  }
+}
+
+struct AuthErrorFake: Error {}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index c9442e05ba7..4431c9b7f1b 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import XCTest
 
@@ -41,6 +42,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
   }
@@ -182,6 +184,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -266,6 +269,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -285,6 +289,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -297,6 +302,74 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_auth_validAuthToken() async throws {
+    let authToken = "test-valid-token"
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      tools: nil,
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      auth: AuthInteropFake(token: authToken),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        authToken: authToken
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_auth_nilAuthToken() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      tools: nil,
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      auth: AuthInteropFake(token: nil),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        authToken: nil
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_auth_authTokenRefreshError() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      tools: nil,
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      auth: AuthInteropFake(error: AuthErrorFake()),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        authToken: nil
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw internalError(AuthErrorFake); no error.")
+    } catch GenerateContentError.internalError(_ as AuthErrorFake) {
+      //
+    } catch {
+      XCTFail("Should throw internalError(AuthErrorFake); error thrown: \(error)")
+    }
+  }
+
   func testGenerateContent_usageMetadata() async throws {
     MockURLProtocol
       .requestHandler = try httpRequestHandler(
@@ -598,6 +671,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -808,6 +882,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -828,6 +903,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -972,6 +1048,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -1048,6 +1125,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -1067,7 +1145,8 @@ final class GenerativeModelTests: XCTestCase {
       apiKey: "API_KEY",
       tools: nil,
       requestOptions: RequestOptions(),
-      appCheck: nil
+      appCheck: nil,
+      auth: nil
     )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
@@ -1081,7 +1160,8 @@ final class GenerativeModelTests: XCTestCase {
       apiKey: "API_KEY",
       tools: nil,
       requestOptions: RequestOptions(),
-      appCheck: nil
+      appCheck: nil,
+      auth: nil
     )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
@@ -1095,7 +1175,8 @@ final class GenerativeModelTests: XCTestCase {
       apiKey: "API_KEY",
       tools: nil,
       requestOptions: RequestOptions(),
-      appCheck: nil
+      appCheck: nil,
+      auth: nil
     )
 
     XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
@@ -1123,7 +1204,8 @@ final class GenerativeModelTests: XCTestCase {
                                   withExtension ext: String,
                                   statusCode: Int = 200,
                                   timeout: TimeInterval = URLRequest.defaultTimeoutInterval(),
-                                  appCheckToken: String? = nil) throws -> ((URLRequest) throws -> (
+                                  appCheckToken: String? = nil,
+                                  authToken: String? = nil) throws -> ((URLRequest) throws -> (
     URLResponse,
     AsyncLineSequence<URL.AsyncBytes>?
   )) {
@@ -1137,6 +1219,11 @@ final class GenerativeModelTests: XCTestCase {
       XCTAssert(apiClientTags.contains(GenerativeAIService.languageTag))
       XCTAssert(apiClientTags.contains(GenerativeAIService.firebaseVersionTag))
       XCTAssertEqual(request.value(forHTTPHeaderField: "X-Firebase-AppCheck"), appCheckToken)
+      if let authToken {
+        XCTAssertEqual(request.value(forHTTPHeaderField: "Authorization"), "Firebase \(authToken)")
+      } else {
+        XCTAssertNil(request.value(forHTTPHeaderField: "Authorization"))
+      }
       let response = try XCTUnwrap(HTTPURLResponse(
         url: requestURL,
         statusCode: statusCode,
diff --git a/Package.swift b/Package.swift
index 154bfd1e554..773abb0ef58 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1369,6 +1369,7 @@ let package = Package(
       name: "FirebaseVertexAI",
       dependencies: [
         "FirebaseAppCheckInterop",
+        "FirebaseAuthInterop",
         "FirebaseCore",
         "FirebaseCoreExtension",
       ],

From 35c206a3223204b701c884eeae41e3361b241d5c Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 5 Feb 2024 17:29:57 -0500
Subject: [PATCH 054/124] Add prototype for Firebase Vertex AI

---
 FirebaseCore/Sources/FIRApp.m                 |   1 +
 FirebaseVertexAI/Sources/VertexAI.swift       | 154 ++++++++++++++++++
 .../Sources/VertexAIComponent.swift           |  80 +++++++++
 Package.swift                                 |  23 ++-
 4 files changed, 257 insertions(+), 1 deletion(-)
 create mode 100644 FirebaseVertexAI/Sources/VertexAI.swift
 create mode 100644 FirebaseVertexAI/Sources/VertexAIComponent.swift

diff --git a/FirebaseCore/Sources/FIRApp.m b/FirebaseCore/Sources/FIRApp.m
index 8f4aefe4bc7..a8cc38c12de 100644
--- a/FirebaseCore/Sources/FIRApp.m
+++ b/FirebaseCore/Sources/FIRApp.m
@@ -829,6 +829,7 @@ + (void)registerSwiftComponents {
     @"FIRSessions" : @"fire-ses",
     @"FIRFunctionsComponent" : @"fire-fun",
     @"FIRStorageComponent" : @"fire-str",
+    @"FIRVertexAIComponent" : @"fire-vtx",
   };
   for (NSString *className in swiftComponents.allKeys) {
     Class klass = NSClassFromString(className);
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
new file mode 100644
index 00000000000..41aafaeee3d
--- /dev/null
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -0,0 +1,154 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+import FirebaseAppCheckInterop
+import FirebaseCore
+import GoogleGenerativeAI
+
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+@objc(FIRVertexAI)
+open class VertexAI: NSObject {
+  // MARK: - Public APIs
+
+  /// The default `VertexAI` instance.
+  ///
+  /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
+  public static func vertexAI(modelName: String, location: String) -> VertexAI {
+    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location)
+  }
+
+  public static func vertexAI(app: FirebaseApp, modelName: String, location: String) -> VertexAI {
+    let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                            in: app.container)
+    let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
+    return provider.vertexAI(location: location, modelResourceName: modelResourceName)
+  }
+
+  public func generateContentStream(prompt: String) async
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    return model.generateContentStream(prompt)
+  }
+
+  // MARK: - Private
+
+  /// The `FirebaseApp` associated with this `VertexAI` instance.
+  private let app: FirebaseApp
+
+  private let appCheck: AppCheckInterop?
+
+  private let location: String
+
+  private let modelResouceName: String
+
+  lazy var model: GenerativeModel = {
+    let options = RequestOptions(hooks: [
+      setVertexAIEndpoint,
+      addAccessTokenHeader,
+      addAppCheckHeader,
+    ])
+    return GenerativeModel(
+      name: modelResouceName,
+      apiKey: app.options.apiKey!,
+      requestOptions: options
+    )
+  }()
+
+  private static let accessTokenEnvKey = "FIRVertexAIAccessToken"
+
+  init(app: FirebaseApp, location: String, modelResourceName: String) {
+    self.app = app
+    appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
+    self.location = location
+    modelResouceName = modelResourceName
+  }
+
+  private static func modelResourceName(app: FirebaseApp, modelName: String,
+                                        location: String) -> String {
+    if modelName.contains("/") {
+      return modelName
+    }
+    guard let projectID = app.options.projectID else {
+      print("The FirebaseApp is missing a project ID.")
+      return modelName
+    }
+
+    return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
+  }
+
+  // MARK: Request Hooks
+
+  /// Replace the Labs endpoint with a Vertex AI endpoint in the provided request.
+  ///
+  /// This is temporary workaround until the Google Generative AI SDK supports setting an endpoint.
+  ///
+  /// - Parameter request: The `URLRequest` to modify with a Vertex AI hostname.
+  func setVertexAIEndpoint(request: inout URLRequest) {
+    guard let requestURL = request.url else {
+      return
+    }
+    guard var urlComponents = URLComponents(url: requestURL, resolvingAgainstBaseURL: false) else {
+      return
+    }
+    urlComponents.host = "\(location)-aiplatform.googleapis.com"
+
+    guard let componentsURL = urlComponents.url else {
+      return
+    }
+
+    request.url = componentsURL
+  }
+
+  /// Add a Google Cloud access token in an Authorization header in the provided request.
+  ///
+  /// This is a temporary workaround until Vertex AI can be called with an API key.
+  ///
+  /// - Parameter request: The `URLRequest` to modify by adding an access token.
+  func addAccessTokenHeader(request: inout URLRequest) {
+    // Remove the API key header, it is not supported by Vertex AI.
+    if var headers = request.allHTTPHeaderFields {
+      headers.removeValue(forKey: "x-goog-api-key")
+    }
+
+    guard let accessToken = ProcessInfo.processInfo.environment[VertexAI.accessTokenEnvKey] else {
+      print("""
+      Vertex AI requires an Access Token for authorization:
+      1. Get an access token by running `gcloud auth print-access-token`
+      2. Set it in the \(VertexAI.accessTokenEnvKey) environment variable.
+      """)
+      return
+    }
+
+    request.addValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
+  }
+
+  /// Adds an App Check token to the provided request, if App Check is included in the app.
+  ///
+  /// This demonstrates how an App Check token can be added to requests; it is currently ignored by
+  /// the backend.
+  ///
+  /// - Parameter request: The `URLRequest` to modify by adding an App Check token header.
+  func addAppCheckHeader(request: inout URLRequest) async {
+    guard let appCheck = appCheck else {
+      return
+    }
+
+    let tokenResult = await appCheck.getToken(forcingRefresh: false)
+    request.addValue(tokenResult.token, forHTTPHeaderField: "X-Firebase-AppCheck")
+  }
+}
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
new file mode 100644
index 00000000000..a8d6c177c74
--- /dev/null
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -0,0 +1,80 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseAppCheckInterop
+import FirebaseCore
+import Foundation
+
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+@objc(FIRVertexAIProvider)
+protocol VertexAIProvider {
+  @objc func vertexAI(location: String, modelResourceName: String) -> VertexAI
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+@objc(FIRVertexAIComponent)
+class VertexAIComponent: NSObject, Library, VertexAIProvider {
+  // MARK: - Private Variables
+
+  /// The app associated with all `VertexAI` instances in this container.
+  /// This is `unowned` instead of `weak` so it can be used without unwrapping in `vertexAI(...)`
+  private unowned let app: FirebaseApp
+
+  /// A map of active  `VertexAI` instances for `app`, keyed by model resource names
+  /// (e.g., "projects/my-project-id/locations/us-central1/publishers/google/models/gemini-pro").
+  private var instances: [String: VertexAI] = [:]
+
+  /// Lock to manage access to the `instances` array to avoid race conditions.
+  private var instancesLock: os_unfair_lock = .init()
+
+  // MARK: - Initializers
+
+  required init(app: FirebaseApp) {
+    self.app = app
+  }
+
+  // MARK: - Library conformance
+
+  static func componentsToRegister() -> [Component] {
+    let appCheckInterop = Dependency(with: AppCheckInterop.self, isRequired: false)
+    return [Component(VertexAIProvider.self,
+                      instantiationTiming: .lazy,
+                      dependencies: [
+                        appCheckInterop,
+                      ]) { container, isCacheable in
+        guard let app = container.app else { return nil }
+        isCacheable.pointee = true
+        return self.init(app: app)
+      }]
+  }
+
+  // MARK: - VertexAIProvider conformance
+
+  func vertexAI(location: String, modelResourceName: String) -> VertexAI {
+    os_unfair_lock_lock(&instancesLock)
+
+    // Unlock before the function returns.
+    defer { os_unfair_lock_unlock(&instancesLock) }
+
+    if let instance = instances[modelResourceName] {
+      return instance
+    }
+    let newInstance = VertexAI(app: app, location: location, modelResourceName: modelResourceName)
+    instances[modelResourceName] = newInstance
+    return newInstance
+  }
+}
diff --git a/Package.swift b/Package.swift
index 7423e659f66..9f66dde2ba7 100644
--- a/Package.swift
+++ b/Package.swift
@@ -137,6 +137,10 @@ let package = Package(
       name: "FirebaseStorage",
       targets: ["FirebaseStorage"]
     ),
+    .library(
+      name: "FirebaseVertexAI",
+      targets: ["FirebaseVertexAI"]
+    ),
   ],
   dependencies: [
     .package(
@@ -188,7 +192,11 @@ let package = Package(
       url: "https://github.com/google/interop-ios-for-google-sdks.git",
       "100.0.0" ..< "101.0.0"
     ),
-    .package(url: "https://github.com/google/app-check.git", "10.19.0" ..< "11.0.0"),
+    .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
+    .package(
+      url: "https://github.com/google/generative-ai-swift.git",
+      revision: "f59b91c7df6ae4d48ff85ab58e640b5084712fa6"
+    ),
   ],
   targets: [
     .target(
@@ -1358,6 +1366,19 @@ let package = Package(
         .headerSearchPath("../../.."),
       ]
     ),
+
+    // MARK: - Firebase Vertex AI
+
+    .target(
+      name: "FirebaseVertexAI",
+      dependencies: [
+        "FirebaseAppCheckInterop",
+        "FirebaseCore",
+        "FirebaseCoreExtension",
+        .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
+      ],
+      path: "FirebaseVertexAI/Sources"
+    ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,
   cxxLanguageStandard: CXXLanguageStandard.gnucxx14

From fec29fd60c6a35b5f5d2d3355495ad1691e04f36 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 5 Feb 2024 17:53:44 -0500
Subject: [PATCH 055/124] Update generative-ai-swift dependency

---
 Package.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Package.swift b/Package.swift
index 9f66dde2ba7..a59cf2d6b62 100644
--- a/Package.swift
+++ b/Package.swift
@@ -195,7 +195,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "f59b91c7df6ae4d48ff85ab58e640b5084712fa6"
+      revision: "efcee26db89d1948ebd581b3097c94186d615282"
     ),
   ],
   targets: [

From 7e2bfea501f871fd9a8a6af734003a846f056ba3 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 14 Feb 2024 16:01:53 -0500
Subject: [PATCH 056/124] Use Swift 5.9 `AccessLevelOnImport`

---
 FirebaseVertexAI/Sources/VertexAI.swift       | 29 ++++++++++++++-----
 .../Sources/VertexAIComponent.swift           |  8 ++---
 Package.swift                                 |  9 ++++--
 3 files changed, 32 insertions(+), 14 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 41aafaeee3d..429ad02491a 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -14,12 +14,16 @@
 
 import Foundation
 
+public import FirebaseCore
+public import class GoogleGenerativeAI.Chat
+public import protocol GoogleGenerativeAI.PartsRepresentable
+public import struct GoogleGenerativeAI.GenerateContentResponse
+public import struct GoogleGenerativeAI.ModelContent
+
+import FirebaseCoreExtension
 import FirebaseAppCheckInterop
-import FirebaseCore
-import GoogleGenerativeAI
 
-// Avoids exposing internal FirebaseCore APIs to Swift users.
-@_implementationOnly import FirebaseCoreExtension
+private import GoogleGenerativeAI
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAI)
@@ -40,9 +44,20 @@ open class VertexAI: NSObject {
     return provider.vertexAI(location: location, modelResourceName: modelResourceName)
   }
 
-  public func generateContentStream(prompt: String) async
-    -> AsyncThrowingStream<GenerateContentResponse, Error> {
-    return model.generateContentStream(prompt)
+  public func generateContentStream(_ parts: GoogleGenerativeAI
+    .PartsRepresentable...)
+    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
+    return model.generateContentStream([GoogleGenerativeAI.ModelContent(parts: parts)])
+  }
+
+  public func generateContentStream(_ content: [GoogleGenerativeAI.ModelContent])
+    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
+    return model.generateContentStream(content)
+  }
+
+  public func startChat(history: [GoogleGenerativeAI.ModelContent] = []) -> GoogleGenerativeAI
+    .Chat {
+    return model.startChat(history: history)
   }
 
   // MARK: - Private
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a8d6c177c74..68cc0220849 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -12,12 +12,12 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import FirebaseAppCheckInterop
-import FirebaseCore
 import Foundation
 
-// Avoids exposing internal FirebaseCore APIs to Swift users.
-@_implementationOnly import FirebaseCoreExtension
+public import FirebaseCore
+
+import FirebaseAppCheckInterop
+import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
diff --git a/Package.swift b/Package.swift
index a59cf2d6b62..1965fce1f3f 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1,4 +1,4 @@
-// swift-tools-version:5.7.1
+// swift-tools-version:5.9
 // The swift-tools-version declares the minimum version of Swift required to
 // build this package.
 
@@ -195,7 +195,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "efcee26db89d1948ebd581b3097c94186d615282"
+      revision: "4fe215743fd02c8182efdcfff5e7571873d2c158"
     ),
   ],
   targets: [
@@ -1377,7 +1377,10 @@ let package = Package(
         "FirebaseCoreExtension",
         .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources"
+      path: "FirebaseVertexAI/Sources",
+      swiftSettings: [
+        .enableExperimentalFeature("AccessLevelOnImport"),
+      ]
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From 090ce9cf16cc9d29db61cb64de5bf043324b64d6 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 20 Feb 2024 10:33:11 -0500
Subject: [PATCH 057/124] Set Vertex AI endpoint in `RequestOptions`

---
 FirebaseVertexAI/Sources/VertexAI.swift | 35 ++++++-------------------
 Package.swift                           |  2 +-
 2 files changed, 9 insertions(+), 28 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 429ad02491a..e0478ea3a33 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -20,8 +20,8 @@ public import protocol GoogleGenerativeAI.PartsRepresentable
 public import struct GoogleGenerativeAI.GenerateContentResponse
 public import struct GoogleGenerativeAI.ModelContent
 
-import FirebaseCoreExtension
 import FirebaseAppCheckInterop
+import FirebaseCoreExtension
 
 private import GoogleGenerativeAI
 
@@ -72,11 +72,13 @@ open class VertexAI: NSObject {
   private let modelResouceName: String
 
   lazy var model: GenerativeModel = {
-    let options = RequestOptions(hooks: [
-      setVertexAIEndpoint,
-      addAccessTokenHeader,
-      addAppCheckHeader,
-    ])
+    let options = RequestOptions(
+      endpoint: "\(location)-aiplatform.googleapis.com",
+      hooks: [
+        addAccessTokenHeader,
+        addAppCheckHeader,
+      ]
+    )
     return GenerativeModel(
       name: modelResouceName,
       apiKey: app.options.apiKey!,
@@ -108,27 +110,6 @@ open class VertexAI: NSObject {
 
   // MARK: Request Hooks
 
-  /// Replace the Labs endpoint with a Vertex AI endpoint in the provided request.
-  ///
-  /// This is temporary workaround until the Google Generative AI SDK supports setting an endpoint.
-  ///
-  /// - Parameter request: The `URLRequest` to modify with a Vertex AI hostname.
-  func setVertexAIEndpoint(request: inout URLRequest) {
-    guard let requestURL = request.url else {
-      return
-    }
-    guard var urlComponents = URLComponents(url: requestURL, resolvingAgainstBaseURL: false) else {
-      return
-    }
-    urlComponents.host = "\(location)-aiplatform.googleapis.com"
-
-    guard let componentsURL = urlComponents.url else {
-      return
-    }
-
-    request.url = componentsURL
-  }
-
   /// Add a Google Cloud access token in an Authorization header in the provided request.
   ///
   /// This is a temporary workaround until Vertex AI can be called with an API key.
diff --git a/Package.swift b/Package.swift
index 1965fce1f3f..6da0d8fd30e 100644
--- a/Package.swift
+++ b/Package.swift
@@ -195,7 +195,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "4fe215743fd02c8182efdcfff5e7571873d2c158"
+      revision: "46e56ae138f152528b7b50520a22777f3d8adb18"
     ),
   ],
   targets: [

From 3a3b570baf29929c353ed5a768483831981df40f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 20 Feb 2024 10:49:58 -0500
Subject: [PATCH 058/124] Revert "Use Swift 5.9 `AccessLevelOnImport`"

This reverts commit af7d7836bc5f4fc5576a06bc7af646f3b179a66a.

# Conflicts:
#	FirebaseVertexAI/Sources/VertexAI.swift
#	Package.swift
---
 FirebaseVertexAI/Sources/VertexAI.swift          | 12 ++++--------
 FirebaseVertexAI/Sources/VertexAIComponent.swift |  8 ++++----
 Package.swift                                    |  7 ++-----
 3 files changed, 10 insertions(+), 17 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index e0478ea3a33..9e3809c9828 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -14,16 +14,12 @@
 
 import Foundation
 
-public import FirebaseCore
-public import class GoogleGenerativeAI.Chat
-public import protocol GoogleGenerativeAI.PartsRepresentable
-public import struct GoogleGenerativeAI.GenerateContentResponse
-public import struct GoogleGenerativeAI.ModelContent
-
 import FirebaseAppCheckInterop
-import FirebaseCoreExtension
+import FirebaseCore
+import GoogleGenerativeAI
 
-private import GoogleGenerativeAI
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAI)
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index 68cc0220849..a8d6c177c74 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -12,12 +12,12 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
+import FirebaseCore
 import Foundation
 
-public import FirebaseCore
-
-import FirebaseAppCheckInterop
-import FirebaseCoreExtension
+// Avoids exposing internal FirebaseCore APIs to Swift users.
+@_implementationOnly import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
diff --git a/Package.swift b/Package.swift
index 6da0d8fd30e..78b681e8213 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1,4 +1,4 @@
-// swift-tools-version:5.9
+// swift-tools-version:5.7.1
 // The swift-tools-version declares the minimum version of Swift required to
 // build this package.
 
@@ -1377,10 +1377,7 @@ let package = Package(
         "FirebaseCoreExtension",
         .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources",
-      swiftSettings: [
-        .enableExperimentalFeature("AccessLevelOnImport"),
-      ]
+      path: "FirebaseVertexAI/Sources"
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From cb982076b687a138a9c5c1eb8347f3cad6656f74 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 20 Feb 2024 12:23:31 -0500
Subject: [PATCH 059/124] Add alternative API surface that returns
 `GenerativeModel` instances

---
 FirebaseVertexAI/Sources/VertexAI.swift | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 9e3809c9828..02fdb82807f 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -26,6 +26,27 @@ import GoogleGenerativeAI
 open class VertexAI: NSObject {
   // MARK: - Public APIs
 
+  // Option 1: Return `GoogleGenerativeAI.GenerativeModel` instances.
+
+  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  ///
+  /// This instance is configured with the default `FirebaseApp`.
+  public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
+    .GenerativeModel {
+    return vertexAI(modelName: modelName, location: location).model
+  }
+
+  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  public static func generativeModel(app: FirebaseApp, modelName: String,
+                                     location: String) -> GoogleGenerativeAI.GenerativeModel {
+    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location).model
+  }
+
+  // Option 2: Return `VertexAI` instances with a similar API surface to
+  // `GoogleGenerativeAI.GenerativeModel`.
+  // Some types are re-used from the Google AI SDK for requests and responses, e.g.,
+  // `GenerateContentResponse`.
+
   /// The default `VertexAI` instance.
   ///
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.

From 277c2893a4628cfc836dc795703d5157a8e74dd4 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 22 Feb 2024 11:01:44 -0500
Subject: [PATCH 060/124] Add sample apps using Vertex AI

---
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../ChatSample/Assets.xcassets/Contents.json  |    6 +
 .../Sample/ChatSample/ChatSampleApp.swift     |   36 +
 .../ChatSample/Models/ChatMessage.swift       |   64 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../Screens/ConversationScreen.swift          |  127 ++
 .../ViewModels/ConversationViewModel.swift    |  130 ++
 .../ChatSample/Views/BouncingDots.swift       |   77 ++
 .../ChatSample/Views/ErrorDetailsView.swift   |  247 ++++
 .../Sample/ChatSample/Views/ErrorView.swift   |   65 +
 .../Sample/ChatSample/Views/MessageView.swift |  108 ++
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../Assets.xcassets/Contents.json             |    6 +
 .../GenerativeAIMultimodalSampleApp.swift     |   31 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../Screens/PhotoReasoningScreen.swift        |   65 +
 .../ViewModels/PhotoReasoningViewModel.swift  |  119 ++
 .../project.pbxproj                           | 1053 +++++++++++++++++
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../Assets.xcassets/Contents.json             |    6 +
 .../GenerativeAISample/ContentView.swift      |   48 +
 .../GenerativeAISampleApp.swift               |   29 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../AccentColor.colorset/Contents.json        |   11 +
 .../AppIcon.appiconset/Contents.json          |   13 +
 .../Assets.xcassets/Contents.json             |    6 +
 .../GenerativeAITextSampleApp.swift           |   31 +
 .../Preview Assets.xcassets/Contents.json     |    6 +
 .../Screens/SummarizeScreen.swift             |   74 ++
 .../ViewModels/SummarizeViewModel.swift       |   68 ++
 .../GenerativeAIUIComponents/Package.swift    |   35 +
 .../GenerativeAIUIComponents/InputField.swift |   82 ++
 .../MultimodalInputField.swift                |  183 +++
 FirebaseVertexAI/Sources/VertexAI.swift       |   37 +-
 37 files changed, 2822 insertions(+), 31 deletions(-)
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
 create mode 100644 FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
 create mode 100644 FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift

diff --git a/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift b/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
new file mode 100644
index 00000000000..7d7e02d7e3c
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
@@ -0,0 +1,36 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import FirebaseVertexAI
+import SwiftUI
+
+@main
+struct ChatSampleApp: App {
+  @StateObject
+  var viewModel = ConversationViewModel()
+
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      NavigationStack {
+        ConversationScreen()
+          .environmentObject(viewModel)
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift b/FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift
new file mode 100644
index 00000000000..6f7ab321b12
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Models/ChatMessage.swift
@@ -0,0 +1,64 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+enum Participant {
+  case system
+  case user
+}
+
+struct ChatMessage: Identifiable, Equatable {
+  let id = UUID().uuidString
+  var message: String
+  let participant: Participant
+  var pending = false
+
+  static func pending(participant: Participant) -> ChatMessage {
+    Self(message: "", participant: participant, pending: true)
+  }
+}
+
+extension ChatMessage {
+  static var samples: [ChatMessage] = [
+    .init(message: "Hello. What can I do for you today?", participant: .system),
+    .init(message: "Show me a simple loop in Swift.", participant: .user),
+    .init(message: """
+    Sure, here is a simple loop in Swift:
+
+    # Example 1
+    ```
+    for i in 1...5 {
+      print("Hello, world!")
+    }
+    ```
+
+    This loop will print the string "Hello, world!" five times. The for loop iterates over a range of numbers,
+    in this case the numbers from 1 to 5. The variable i is assigned each number in the range, and the code inside the loop is executed.
+
+    **Here is another example of a simple loop in Swift:**
+    ```swift
+    var sum = 0
+    for i in 1...100 {
+      sum += i
+    }
+    print("The sum of the numbers from 1 to 100 is \\(sum).")
+    ```
+
+    This loop calculates the sum of the numbers from 1 to 100. The variable sum is initialized to 0, and then the for loop iterates over the range of numbers from 1 to 100. The variable i is assigned each number in the range, and the value of i is added to the sum variable. After the loop has finished executing, the value of sum is printed to the console.
+    """, participant: .system),
+  ]
+
+  static var sample = samples[0]
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
new file mode 100644
index 00000000000..a3044a1f430
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
@@ -0,0 +1,127 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import GenerativeAIUIComponents
+import SwiftUI
+
+struct ConversationScreen: View {
+  @EnvironmentObject
+  var viewModel: ConversationViewModel
+
+  @State
+  private var userPrompt = ""
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      ScrollViewReader { scrollViewProxy in
+        List {
+          ForEach(viewModel.messages) { message in
+            MessageView(message: message)
+          }
+          if let error = viewModel.error {
+            ErrorView(error: error)
+              .tag("errorView")
+          }
+        }
+        .listStyle(.plain)
+        .onChange(of: viewModel.messages, perform: { newValue in
+          if viewModel.hasError {
+            // wait for a short moment to make sure we can actually scroll to the bottom
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo("errorView", anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          } else {
+            guard let lastMessage = viewModel.messages.last else { return }
+
+            // wait for a short moment to make sure we can actually scroll to the bottom
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo(lastMessage.id, anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          }
+        })
+      }
+      InputField("Message...", text: $userPrompt) {
+        Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
+          .font(.title)
+      }
+      .focused($focusedField, equals: .message)
+      .onSubmit { sendOrStop() }
+    }
+    .toolbar {
+      ToolbarItem(placement: .primaryAction) {
+        Button(action: newChat) {
+          Image(systemName: "square.and.pencil")
+        }
+      }
+    }
+    .navigationTitle("Chat sample")
+    .onAppear {
+      focusedField = .message
+    }
+  }
+
+  private func sendMessage() {
+    Task {
+      let prompt = userPrompt
+      userPrompt = ""
+      await viewModel.sendMessage(prompt, streaming: true)
+    }
+  }
+
+  private func sendOrStop() {
+    if viewModel.busy {
+      viewModel.stop()
+    } else {
+      sendMessage()
+    }
+  }
+
+  private func newChat() {
+    viewModel.startNewChat()
+  }
+}
+
+struct ConversationScreen_Previews: PreviewProvider {
+  struct ContainerView: View {
+    @StateObject var viewModel = ConversationViewModel()
+
+    var body: some View {
+      ConversationScreen()
+        .environmentObject(viewModel)
+        .onAppear {
+          viewModel.messages = ChatMessage.samples
+        }
+    }
+  }
+
+  static var previews: some View {
+    NavigationStack {
+      ConversationScreen()
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
new file mode 100644
index 00000000000..465789be387
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -0,0 +1,130 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import UIKit
+
+@MainActor
+class ConversationViewModel: ObservableObject {
+  /// This array holds both the user's and the system's chat messages
+  @Published var messages = [ChatMessage]()
+
+  /// Indicates we're waiting for the model to finish
+  @Published var busy = false
+
+  @Published var error: Error?
+  var hasError: Bool {
+    return error != nil
+  }
+
+  private var model: GenerativeModel
+  private var chat: Chat
+  private var stopGenerating = false
+
+  private var chatTask: Task<Void, Never>?
+
+  init() {
+    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    chat = model.startChat()
+  }
+
+  func sendMessage(_ text: String, streaming: Bool = true) async {
+    error = nil
+    if streaming {
+      await internalSendMessageStreaming(text)
+    } else {
+      await internalSendMessage(text)
+    }
+  }
+
+  func startNewChat() {
+    stop()
+    error = nil
+    chat = model.startChat()
+    messages.removeAll()
+  }
+
+  func stop() {
+    chatTask?.cancel()
+    error = nil
+  }
+
+  private func internalSendMessageStreaming(_ text: String) async {
+    chatTask?.cancel()
+
+    chatTask = Task {
+      busy = true
+      defer {
+        busy = false
+      }
+
+      // first, add the user's message to the chat
+      let userMessage = ChatMessage(message: text, participant: .user)
+      messages.append(userMessage)
+
+      // add a pending message while we're waiting for a response from the backend
+      let systemMessage = ChatMessage.pending(participant: .system)
+      messages.append(systemMessage)
+
+      do {
+        let responseStream = chat.sendMessageStream(text)
+        for try await chunk in responseStream {
+          messages[messages.count - 1].pending = false
+          if let text = chunk.text {
+            messages[messages.count - 1].message += text
+          }
+        }
+      } catch {
+        self.error = error
+        print(error.localizedDescription)
+        messages.removeLast()
+      }
+    }
+  }
+
+  private func internalSendMessage(_ text: String) async {
+    chatTask?.cancel()
+
+    chatTask = Task {
+      busy = true
+      defer {
+        busy = false
+      }
+
+      // first, add the user's message to the chat
+      let userMessage = ChatMessage(message: text, participant: .user)
+      messages.append(userMessage)
+
+      // add a pending message while we're waiting for a response from the backend
+      let systemMessage = ChatMessage.pending(participant: .system)
+      messages.append(systemMessage)
+
+      do {
+        var response: GenerateContentResponse?
+        response = try await chat.sendMessage(text)
+
+        if let responseText = response?.text {
+          // replace pending message with backend response
+          messages[messages.count - 1].message = responseText
+          messages[messages.count - 1].pending = false
+        }
+      } catch {
+        self.error = error
+        print(error.localizedDescription)
+        messages.removeLast()
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift b/FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift
new file mode 100644
index 00000000000..6895e6723da
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/BouncingDots.swift
@@ -0,0 +1,77 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import SwiftUI
+
+struct BouncingDots: View {
+  @State
+  private var dot1YOffset: CGFloat = 0.0
+
+  @State
+  private var dot2YOffset: CGFloat = 0.0
+
+  @State
+  private var dot3YOffset: CGFloat = 0.0
+
+  let animation = Animation.easeInOut(duration: 0.8)
+    .repeatForever(autoreverses: true)
+
+  var body: some View {
+    HStack(spacing: 8) {
+      Circle()
+        .fill(Color.white)
+        .frame(width: 10, height: 10)
+        .offset(y: dot1YOffset)
+        .onAppear {
+          withAnimation(self.animation.delay(0.0)) {
+            self.dot1YOffset = -5
+          }
+        }
+      Circle()
+        .fill(Color.white)
+        .frame(width: 10, height: 10)
+        .offset(y: dot2YOffset)
+        .onAppear {
+          withAnimation(self.animation.delay(0.2)) {
+            self.dot2YOffset = -5
+          }
+        }
+      Circle()
+        .fill(Color.white)
+        .frame(width: 10, height: 10)
+        .offset(y: dot3YOffset)
+        .onAppear {
+          withAnimation(self.animation.delay(0.4)) {
+            self.dot3YOffset = -5
+          }
+        }
+    }
+    .onAppear {
+      let baseOffset: CGFloat = -2
+
+      self.dot1YOffset = baseOffset
+      self.dot2YOffset = baseOffset
+      self.dot3YOffset = baseOffset
+    }
+  }
+}
+
+struct BouncingDots_Previews: PreviewProvider {
+  static var previews: some View {
+    BouncingDots()
+      .frame(width: 200, height: 50)
+      .background(.blue)
+      .roundedCorner(10, corners: [.allCorners])
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
new file mode 100644
index 00000000000..e601876b2d9
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -0,0 +1,247 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import MarkdownUI
+import SwiftUI
+
+extension SafetySetting.HarmCategory: CustomStringConvertible {
+  public var description: String {
+    switch self {
+    case .dangerousContent: "Dangerous content"
+    case .harassment: "Harassment"
+    case .hateSpeech: "Hate speech"
+    case .sexuallyExplicit: "Sexually explicit"
+    case .unknown: "Unknown"
+    case .unspecified: "Unspecified"
+    }
+  }
+}
+
+extension SafetyRating.HarmProbability: CustomStringConvertible {
+  public var description: String {
+    switch self {
+    case .high: "High"
+    case .low: "Low"
+    case .medium: "Medium"
+    case .negligible: "Negligible"
+    case .unknown: "Unknown"
+    case .unspecified: "Unspecified"
+    }
+  }
+}
+
+private struct SubtitleFormRow: View {
+  var title: String
+  var value: String
+
+  var body: some View {
+    VStack(alignment: .leading) {
+      Text(title)
+        .font(.subheadline)
+      Text(value)
+    }
+  }
+}
+
+private struct SubtitleMarkdownFormRow: View {
+  var title: String
+  var value: String
+
+  var body: some View {
+    VStack(alignment: .leading) {
+      Text(title)
+        .font(.subheadline)
+      Markdown(value)
+    }
+  }
+}
+
+private struct SafetyRatingsSection: View {
+  var ratings: [SafetyRating]
+
+  var body: some View {
+    Section("Safety ratings") {
+      List(ratings, id: \.self) { rating in
+        HStack {
+          Text("\(String(describing: rating.category))")
+            .font(.subheadline)
+          Spacer()
+          Text("\(String(describing: rating.probability))")
+        }
+      }
+    }
+  }
+}
+
+struct ErrorDetailsView: View {
+  var error: Error
+
+  var body: some View {
+    NavigationView {
+      Form {
+        switch error {
+        case let GenerateContentError.internalError(underlying: underlyingError):
+          Section("Error Type") {
+            Text("Internal error")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description",
+                            value: underlyingError.localizedDescription)
+          }
+
+        case let GenerateContentError.promptBlocked(response: generateContentResponse):
+          Section("Error Type") {
+            Text("Your prompt was blocked")
+          }
+
+          Section("Details") {
+            if let reason = generateContentResponse.promptFeedback?.blockReason {
+              SubtitleFormRow(title: "Reason for blocking", value: reason.rawValue)
+            }
+
+            if let text = generateContentResponse.text {
+              SubtitleMarkdownFormRow(title: "Last chunk for the response", value: text)
+            }
+          }
+
+          if let ratings = generateContentResponse.candidates.first?.safetyRatings {
+            SafetyRatingsSection(ratings: ratings)
+          }
+
+        case let GenerateContentError.responseStoppedEarly(
+          reason: finishReason,
+          response: generateContentResponse
+        ):
+
+          Section("Error Type") {
+            Text("Response stopped early")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Reason for finishing early", value: finishReason.rawValue)
+
+            if let text = generateContentResponse.text {
+              SubtitleMarkdownFormRow(title: "Last chunk for the response", value: text)
+            }
+          }
+
+          if let ratings = generateContentResponse.candidates.first?.safetyRatings {
+            SafetyRatingsSection(ratings: ratings)
+          }
+
+        case GenerateContentError.invalidAPIKey:
+          Section("Error Type") {
+            Text("Invalid API Key")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
+            SubtitleMarkdownFormRow(
+              title: "Help",
+              value: """
+              Please provide a valid value for `API_KEY` in the `GenerativeAI-Info.plist` file.
+              """
+            )
+          }
+
+        case GenerateContentError.unsupportedUserLocation:
+          Section("Error Type") {
+            Text("Unsupported User Location")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
+            SubtitleMarkdownFormRow(
+              title: "Help",
+              value: """
+              The API is unsupported in your location (country / territory); please see the list of
+              [available regions](https://ai.google.dev/available_regions#available_regions).
+              """
+            )
+          }
+
+        default:
+          Section("Error Type") {
+            Text("Some other error")
+          }
+
+          Section("Details") {
+            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
+          }
+        }
+      }
+      .navigationTitle("Error details")
+      .navigationBarTitleDisplayMode(.inline)
+    }
+  }
+}
+
+#Preview("Response Stopped Early") {
+  let error = GenerateContentError.responseStoppedEarly(
+    reason: .maxTokens,
+    response: GenerateContentResponse(candidates: [
+      CandidateResponse(content: ModelContent(role: "model", [
+        """
+        A _hypothetical_ model response.
+        Cillum ex aliqua amet aliquip labore amet eiusmod consectetur reprehenderit sit commodo.
+        """,
+      ]),
+      safetyRatings: [
+        SafetyRating(category: .dangerousContent, probability: .high),
+        SafetyRating(category: .harassment, probability: .low),
+        SafetyRating(category: .hateSpeech, probability: .low),
+        SafetyRating(category: .sexuallyExplicit, probability: .low),
+      ],
+      finishReason: FinishReason.maxTokens,
+      citationMetadata: nil),
+    ],
+    promptFeedback: nil)
+  )
+
+  return ErrorDetailsView(error: error)
+}
+
+#Preview("Prompt Blocked") {
+  let error = GenerateContentError.promptBlocked(
+    response: GenerateContentResponse(candidates: [
+      CandidateResponse(content: ModelContent(role: "model", [
+        """
+        A _hypothetical_ model response.
+        Cillum ex aliqua amet aliquip labore amet eiusmod consectetur reprehenderit sit commodo.
+        """,
+      ]),
+      safetyRatings: [
+        SafetyRating(category: .dangerousContent, probability: .high),
+        SafetyRating(category: .harassment, probability: .low),
+        SafetyRating(category: .hateSpeech, probability: .low),
+        SafetyRating(category: .sexuallyExplicit, probability: .low),
+      ],
+      finishReason: FinishReason.other,
+      citationMetadata: nil),
+    ],
+    promptFeedback: nil)
+  )
+
+  return ErrorDetailsView(error: error)
+}
+
+#Preview("Invalid API Key") {
+  ErrorDetailsView(error: GenerateContentError.invalidAPIKey)
+}
+
+#Preview("Unsupported User Location") {
+  ErrorDetailsView(error: GenerateContentError.unsupportedUserLocation)
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
new file mode 100644
index 00000000000..1307eee62d4
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
@@ -0,0 +1,65 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import SwiftUI
+
+struct ErrorView: View {
+  var error: Error
+  @State private var isDetailsSheetPresented = false
+  var body: some View {
+    HStack {
+      Text("An error occurred.")
+      Button(action: { isDetailsSheetPresented.toggle() }) {
+        Image(systemName: "info.circle")
+      }
+    }
+    .frame(maxWidth: .infinity, alignment: .center)
+    .listRowSeparator(.hidden)
+    .sheet(isPresented: $isDetailsSheetPresented) {
+      ErrorDetailsView(error: error)
+    }
+  }
+}
+
+#Preview {
+  NavigationView {
+    let errorPromptBlocked = GenerateContentError.promptBlocked(
+      response: GenerateContentResponse(candidates: [
+        CandidateResponse(content: ModelContent(role: "model", [
+          """
+            A _hypothetical_ model response.
+            Cillum ex aliqua amet aliquip labore amet eiusmod consectetur reprehenderit sit commodo.
+          """,
+        ]),
+        safetyRatings: [
+          SafetyRating(category: .dangerousContent, probability: .high),
+          SafetyRating(category: .harassment, probability: .low),
+          SafetyRating(category: .hateSpeech, probability: .low),
+          SafetyRating(category: .sexuallyExplicit, probability: .low),
+        ],
+        finishReason: FinishReason.other,
+        citationMetadata: nil),
+      ],
+      promptFeedback: nil)
+    )
+    List {
+      MessageView(message: ChatMessage.samples[0])
+      MessageView(message: ChatMessage.samples[1])
+      ErrorView(error: errorPromptBlocked)
+    }
+    .listStyle(.plain)
+    .navigationTitle("Chat sample")
+  }
+}
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift
new file mode 100644
index 00000000000..79894503ffd
--- /dev/null
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/MessageView.swift
@@ -0,0 +1,108 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import MarkdownUI
+import SwiftUI
+
+struct RoundedCorner: Shape {
+  var radius: CGFloat = .infinity
+  var corners: UIRectCorner = .allCorners
+
+  func path(in rect: CGRect) -> Path {
+    let path = UIBezierPath(
+      roundedRect: rect,
+      byRoundingCorners: corners,
+      cornerRadii: CGSize(width: radius, height: radius)
+    )
+    return Path(path.cgPath)
+  }
+}
+
+extension View {
+  func roundedCorner(_ radius: CGFloat, corners: UIRectCorner) -> some View {
+    clipShape(RoundedCorner(radius: radius, corners: corners))
+  }
+}
+
+struct MessageContentView: View {
+  var message: ChatMessage
+
+  var body: some View {
+    if message.pending {
+      BouncingDots()
+    } else {
+      Markdown(message.message)
+        .markdownTextStyle {
+          FontFamilyVariant(.normal)
+          FontSize(.em(0.85))
+          ForegroundColor(message.participant == .system ? Color(UIColor.label) : .white)
+        }
+        .markdownBlockStyle(\.codeBlock) { configuration in
+          configuration.label
+            .relativeLineSpacing(.em(0.25))
+            .markdownTextStyle {
+              FontFamilyVariant(.monospaced)
+              FontSize(.em(0.85))
+              ForegroundColor(Color(.label))
+            }
+            .padding()
+            .background(Color(.secondarySystemBackground))
+            .clipShape(RoundedRectangle(cornerRadius: 8))
+            .markdownMargin(top: .zero, bottom: .em(0.8))
+        }
+    }
+  }
+}
+
+struct MessageView: View {
+  var message: ChatMessage
+
+  var body: some View {
+    HStack {
+      if message.participant == .user {
+        Spacer()
+      }
+      MessageContentView(message: message)
+        .padding(10)
+        .background(message.participant == .system
+          ? Color(UIColor.systemFill)
+          : Color(UIColor.systemBlue))
+        .roundedCorner(10,
+                       corners: [
+                         .topLeft,
+                         .topRight,
+                         message.participant == .system ? .bottomRight : .bottomLeft,
+                       ])
+      if message.participant == .system {
+        Spacer()
+      }
+    }
+    .listRowSeparator(.hidden)
+  }
+}
+
+struct MessageView_Previews: PreviewProvider {
+  static var previews: some View {
+    NavigationView {
+      List {
+        MessageView(message: ChatMessage.samples[0])
+        MessageView(message: ChatMessage.samples[1])
+        MessageView(message: ChatMessage.samples[2])
+        MessageView(message: ChatMessage(message: "Hello!", participant: .system, pending: true))
+      }
+      .listStyle(.plain)
+      .navigationTitle("Chat sample")
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
new file mode 100644
index 00000000000..49565de52d1
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
@@ -0,0 +1,31 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import SwiftUI
+
+@main
+struct GenerativeAIMultimodalSampleApp: App {
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      NavigationStack {
+        PhotoReasoningScreen()
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
new file mode 100644
index 00000000000..98f327585db
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
@@ -0,0 +1,65 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import GenerativeAIUIComponents
+import MarkdownUI
+import PhotosUI
+import SwiftUI
+
+struct PhotoReasoningScreen: View {
+  @StateObject var viewModel = PhotoReasoningViewModel()
+
+  var body: some View {
+    VStack {
+      MultimodalInputField(text: $viewModel.userInput, selection: $viewModel.selectedItems)
+        .onSubmit {
+          onSendTapped()
+        }
+
+      ScrollViewReader { scrollViewProxy in
+        List {
+          if let outputText = viewModel.outputText {
+            HStack(alignment: .top) {
+              if viewModel.inProgress {
+                ProgressView()
+              } else {
+                Image(systemName: "cloud.circle.fill")
+                  .font(.title2)
+              }
+
+              Markdown("\(outputText)")
+            }
+            .listRowSeparator(.hidden)
+          }
+        }
+        .listStyle(.plain)
+      }
+    }
+    .navigationTitle("Multimodal sample")
+  }
+
+  // MARK: - Actions
+
+  private func onSendTapped() {
+    Task {
+      await viewModel.reason()
+    }
+  }
+}
+
+#Preview {
+  NavigationStack {
+    PhotoReasoningScreen()
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
new file mode 100644
index 00000000000..aa646c127df
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -0,0 +1,119 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import OSLog
+import PhotosUI
+import SwiftUI
+
+@MainActor
+class PhotoReasoningViewModel: ObservableObject {
+  // Maximum value for the larger of the two image dimensions (height and width) in pixels. This is
+  // being used to reduce the image size in bytes.
+  private static let largestImageDimension = 768.0
+
+  private var logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "generative-ai")
+
+  @Published
+  var userInput: String = ""
+
+  @Published
+  var selectedItems = [PhotosPickerItem]()
+
+  @Published
+  var outputText: String? = nil
+
+  @Published
+  var errorMessage: String?
+
+  @Published
+  var inProgress = false
+
+  private var model: GenerativeModel?
+
+  init() {
+    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro-vision", location: "us-central1")
+  }
+
+  func reason() async {
+    defer {
+      inProgress = false
+    }
+    guard let model else {
+      return
+    }
+
+    do {
+      inProgress = true
+      errorMessage = nil
+      outputText = ""
+
+      let prompt = "Look at the image(s), and then answer the following question: \(userInput)"
+
+      var images = [PartsRepresentable]()
+      for item in selectedItems {
+        if let data = try? await item.loadTransferable(type: Data.self) {
+          guard let image = UIImage(data: data) else {
+            logger.error("Failed to parse data as an image, skipping.")
+            continue
+          }
+          if image.size.fits(largestDimension: PhotoReasoningViewModel.largestImageDimension) {
+            images.append(image)
+          } else {
+            guard let resizedImage = image
+              .preparingThumbnail(of: image.size
+                .aspectFit(largestDimension: PhotoReasoningViewModel.largestImageDimension)) else {
+              logger.error("Failed to resize image: \(image)")
+              continue
+            }
+
+            images.append(resizedImage)
+          }
+        }
+      }
+
+      let outputContentStream = model.generateContentStream(prompt, images)
+
+      // stream response
+      for try await outputContent in outputContentStream {
+        guard let line = outputContent.text else {
+          return
+        }
+
+        outputText = (outputText ?? "") + line
+      }
+    } catch {
+      logger.error("\(error.localizedDescription)")
+      errorMessage = error.localizedDescription
+    }
+  }
+}
+
+private extension CGSize {
+  func fits(largestDimension length: CGFloat) -> Bool {
+    return width <= length && height <= length
+  }
+
+  func aspectFit(largestDimension length: CGFloat) -> CGSize {
+    let aspectRatio = width / height
+    if width > height {
+      let width = min(self.width, length)
+      return CGSize(width: width, height: round(width / aspectRatio))
+    } else {
+      let height = min(self.height, length)
+      return CGSize(width: round(height * aspectRatio), height: height)
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
new file mode 100644
index 00000000000..20f07a5acca
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -0,0 +1,1053 @@
+// !$*UTF8*$!
+{
+	archiveVersion = 1;
+	classes = {
+	};
+	objectVersion = 60;
+	objects = {
+
+/* Begin PBXBuildFile section */
+		869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AA2B86BE3100482873 /* FirebaseVertexAI */; };
+		869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AC2B86BEA100482873 /* FirebaseVertexAI */; };
+		869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AE2B86BFBC00482873 /* FirebaseVertexAI */; };
+		869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200B02B86BFF500482873 /* FirebaseVertexAI */; };
+		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */; };
+		880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */; };
+		88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
+		88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
+		88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88209C232B0FBE1700F64795 /* MarkdownUI */; };
+		88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
+		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
+		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
+		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
+		8848C8352B0D04BC007B434F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8342B0D04BC007B434F /* ContentView.swift */; };
+		8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8362B0D04BD007B434F /* Assets.xcassets */; };
+		8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8392B0D04BD007B434F /* Preview Assets.xcassets */; };
+		8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */; };
+		8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84A2B0D051F007B434F /* Assets.xcassets */; };
+		8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84D2B0D051F007B434F /* Preview Assets.xcassets */; };
+		8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */; };
+		8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85C2B0D056D007B434F /* Assets.xcassets */; };
+		8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85F2B0D056D007B434F /* Preview Assets.xcassets */; };
+		886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
+		886F95D62B17BA010036F07A /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
+		886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95D72B17BA420036F07A /* MarkdownUI */; };
+		886F95DB2B17BAEF0036F07A /* PhotoReasoningViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */; };
+		886F95DC2B17BAEF0036F07A /* PhotoReasoningScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */; };
+		886F95DD2B17D5010036F07A /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
+		886F95DE2B17D5010036F07A /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
+		886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
+		886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
+		886F95E12B17D5010036F07A /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
+		886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95E22B17D6630036F07A /* GenerativeAIUIComponents */; };
+		889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
+		88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A91D2B0FC55100424728 /* MarkdownUI */; };
+		88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */; };
+		88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88D9474C2B14F27E008B5580 /* MarkdownUI */; };
+		88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F442B110D5300C08E95 /* ChatSampleApp.swift */; };
+		88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F482B110D5400C08E95 /* Assets.xcassets */; };
+		88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */; };
+		88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
+		88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
+		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
+		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
+		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
+		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
+/* End PBXBuildFile section */
+
+/* Begin PBXFileReference section */
+		869200B22B879C4F00482873 /* GoogleService-Info.plist */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text.plist.xml; path = "GoogleService-Info.plist"; sourceTree = "<group>"; };
+		8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningViewModel.swift; sourceTree = "<group>"; };
+		880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningScreen.swift; sourceTree = "<group>"; };
+		88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeScreen.swift; sourceTree = "<group>"; };
+		88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeViewModel.swift; sourceTree = "<group>"; };
+		88263BEE2B239BFE008AB09B /* ErrorView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = ErrorView.swift; sourceTree = "<group>"; };
+		8848C82F2B0D04BC007B434F /* GenerativeAISample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAISample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAISampleApp.swift; sourceTree = "<group>"; };
+		8848C8342B0D04BC007B434F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = "<group>"; };
+		8848C8362B0D04BD007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		8848C8392B0D04BD007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		8848C8442B0D051E007B434F /* GenerativeAITextSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAITextSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAITextSampleApp.swift; sourceTree = "<group>"; };
+		8848C84A2B0D051F007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		8848C84D2B0D051F007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAIMultimodalSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAIMultimodalSampleApp.swift; sourceTree = "<group>"; };
+		8848C85C2B0D056D007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		8848C85F2B0D056D007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		889873842B208563005B4896 /* ErrorDetailsView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ErrorDetailsView.swift; sourceTree = "<group>"; };
+		88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = GenerativeAIUIComponents; sourceTree = "<group>"; };
+		88E10F422B110D5300C08E95 /* ChatSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = ChatSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		88E10F442B110D5300C08E95 /* ChatSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatSampleApp.swift; sourceTree = "<group>"; };
+		88E10F482B110D5400C08E95 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
+		88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
+		88E10F542B1112CA00C08E95 /* ConversationScreen.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ConversationScreen.swift; sourceTree = "<group>"; };
+		88E10F562B1112F600C08E95 /* ConversationViewModel.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ConversationViewModel.swift; sourceTree = "<group>"; };
+		88E10F582B11131900C08E95 /* ChatMessage.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatMessage.swift; sourceTree = "<group>"; };
+		88E10F5A2B11133E00C08E95 /* MessageView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = MessageView.swift; sourceTree = "<group>"; };
+		88E10F5C2B11135000C08E95 /* BouncingDots.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = BouncingDots.swift; sourceTree = "<group>"; };
+/* End PBXFileReference section */
+
+/* Begin PBXFrameworksBuildPhase section */
+		8848C82C2B0D04BC007B434F /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
+				869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */,
+				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8412B0D051E007B434F /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
+				869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8532B0D056C007B434F /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
+				869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */,
+				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		88E10F3F2B110D5300C08E95 /* Frameworks */ = {
+			isa = PBXFrameworksBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
+				869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */,
+				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXFrameworksBuildPhase section */
+
+/* Begin PBXGroup section */
+		8802666E2B0FC39000CF7CB6 /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		880266742B0FC39000CF7CB6 /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+		88209C1A2B0FBDC300F64795 /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+		88209C1C2B0FBDC300F64795 /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		88209C222B0FBE1700F64795 /* Frameworks */ = {
+			isa = PBXGroup;
+			children = (
+			);
+			name = Frameworks;
+			sourceTree = "<group>";
+		};
+		8848C8262B0D04BC007B434F = {
+			isa = PBXGroup;
+			children = (
+				88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */,
+				869200B22B879C4F00482873 /* GoogleService-Info.plist */,
+				8848C8312B0D04BC007B434F /* GenerativeAISample */,
+				8848C8452B0D051E007B434F /* GenerativeAITextSample */,
+				8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */,
+				88E10F432B110D5300C08E95 /* ChatSample */,
+				8848C8302B0D04BC007B434F /* Products */,
+				88209C222B0FBE1700F64795 /* Frameworks */,
+			);
+			sourceTree = "<group>";
+		};
+		8848C8302B0D04BC007B434F /* Products */ = {
+			isa = PBXGroup;
+			children = (
+				8848C82F2B0D04BC007B434F /* GenerativeAISample.app */,
+				8848C8442B0D051E007B434F /* GenerativeAITextSample.app */,
+				8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */,
+				88E10F422B110D5300C08E95 /* ChatSample.app */,
+			);
+			name = Products;
+			sourceTree = "<group>";
+		};
+		8848C8312B0D04BC007B434F /* GenerativeAISample */ = {
+			isa = PBXGroup;
+			children = (
+				8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */,
+				8848C8342B0D04BC007B434F /* ContentView.swift */,
+				8848C8362B0D04BD007B434F /* Assets.xcassets */,
+				8848C8382B0D04BD007B434F /* Preview Content */,
+			);
+			path = GenerativeAISample;
+			sourceTree = "<group>";
+		};
+		8848C8382B0D04BD007B434F /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				8848C8392B0D04BD007B434F /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		8848C8452B0D051E007B434F /* GenerativeAITextSample */ = {
+			isa = PBXGroup;
+			children = (
+				88209C1C2B0FBDC300F64795 /* ViewModels */,
+				88209C1A2B0FBDC300F64795 /* Screens */,
+				8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */,
+				8848C84A2B0D051F007B434F /* Assets.xcassets */,
+				8848C84C2B0D051F007B434F /* Preview Content */,
+			);
+			path = GenerativeAITextSample;
+			sourceTree = "<group>";
+		};
+		8848C84C2B0D051F007B434F /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				8848C84D2B0D051F007B434F /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */ = {
+			isa = PBXGroup;
+			children = (
+				8802666E2B0FC39000CF7CB6 /* ViewModels */,
+				880266742B0FC39000CF7CB6 /* Screens */,
+				8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */,
+				8848C85C2B0D056D007B434F /* Assets.xcassets */,
+				8848C85E2B0D056D007B434F /* Preview Content */,
+			);
+			path = GenerativeAIMultimodalSample;
+			sourceTree = "<group>";
+		};
+		8848C85E2B0D056D007B434F /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				8848C85F2B0D056D007B434F /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		88E10F432B110D5300C08E95 /* ChatSample */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F522B11124A00C08E95 /* Models */,
+				88E10F502B11123600C08E95 /* ViewModels */,
+				88E10F512B11124100C08E95 /* Views */,
+				88E10F532B1112B900C08E95 /* Screens */,
+				88E10F442B110D5300C08E95 /* ChatSampleApp.swift */,
+				88E10F482B110D5400C08E95 /* Assets.xcassets */,
+				88E10F4A2B110D5400C08E95 /* Preview Content */,
+			);
+			path = ChatSample;
+			sourceTree = "<group>";
+		};
+		88E10F4A2B110D5400C08E95 /* Preview Content */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */,
+			);
+			path = "Preview Content";
+			sourceTree = "<group>";
+		};
+		88E10F502B11123600C08E95 /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F562B1112F600C08E95 /* ConversationViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		88E10F512B11124100C08E95 /* Views */ = {
+			isa = PBXGroup;
+			children = (
+				88263BEE2B239BFE008AB09B /* ErrorView.swift */,
+				88E10F5A2B11133E00C08E95 /* MessageView.swift */,
+				88E10F5C2B11135000C08E95 /* BouncingDots.swift */,
+				889873842B208563005B4896 /* ErrorDetailsView.swift */,
+			);
+			path = Views;
+			sourceTree = "<group>";
+		};
+		88E10F522B11124A00C08E95 /* Models */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F582B11131900C08E95 /* ChatMessage.swift */,
+			);
+			path = Models;
+			sourceTree = "<group>";
+		};
+		88E10F532B1112B900C08E95 /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				88E10F542B1112CA00C08E95 /* ConversationScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+/* End PBXGroup section */
+
+/* Begin PBXNativeTarget section */
+		8848C82E2B0D04BC007B434F /* GenerativeAISample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */;
+			buildPhases = (
+				8848C82B2B0D04BC007B434F /* Sources */,
+				8848C82C2B0D04BC007B434F /* Frameworks */,
+				8848C82D2B0D04BC007B434F /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = GenerativeAISample;
+			packageProductDependencies = (
+				886F95D72B17BA420036F07A /* MarkdownUI */,
+				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
+				869200AA2B86BE3100482873 /* FirebaseVertexAI */,
+			);
+			productName = GenerativeAISample;
+			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
+			productType = "com.apple.product-type.application";
+		};
+		8848C8432B0D051E007B434F /* GenerativeAITextSample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */;
+			buildPhases = (
+				8848C8402B0D051E007B434F /* Sources */,
+				8848C8412B0D051E007B434F /* Frameworks */,
+				8848C8422B0D051E007B434F /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = GenerativeAITextSample;
+			packageProductDependencies = (
+				88209C232B0FBE1700F64795 /* MarkdownUI */,
+				869200B02B86BFF500482873 /* FirebaseVertexAI */,
+			);
+			productName = GenerativeAITextSample;
+			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
+			productType = "com.apple.product-type.application";
+		};
+		8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */;
+			buildPhases = (
+				8848C8522B0D056C007B434F /* Sources */,
+				8848C8532B0D056C007B434F /* Frameworks */,
+				8848C8542B0D056C007B434F /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = GenerativeAIMultimodalSample;
+			packageProductDependencies = (
+				88B8A91D2B0FC55100424728 /* MarkdownUI */,
+				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
+				869200AE2B86BFBC00482873 /* FirebaseVertexAI */,
+			);
+			productName = GenerativeAIMultimodalSample;
+			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
+			productType = "com.apple.product-type.application";
+		};
+		88E10F412B110D5300C08E95 /* ChatSample */ = {
+			isa = PBXNativeTarget;
+			buildConfigurationList = 88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */;
+			buildPhases = (
+				88E10F3E2B110D5300C08E95 /* Sources */,
+				88E10F3F2B110D5300C08E95 /* Frameworks */,
+				88E10F402B110D5300C08E95 /* Resources */,
+			);
+			buildRules = (
+			);
+			dependencies = (
+			);
+			name = ChatSample;
+			packageProductDependencies = (
+				88D9474C2B14F27E008B5580 /* MarkdownUI */,
+				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
+				869200AC2B86BEA100482873 /* FirebaseVertexAI */,
+			);
+			productName = ChatSample;
+			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
+			productType = "com.apple.product-type.application";
+		};
+/* End PBXNativeTarget section */
+
+/* Begin PBXProject section */
+		8848C8272B0D04BC007B434F /* Project object */ = {
+			isa = PBXProject;
+			attributes = {
+				BuildIndependentTargetsInParallel = 1;
+				LastSwiftUpdateCheck = 1510;
+				LastUpgradeCheck = 1510;
+				TargetAttributes = {
+					8848C82E2B0D04BC007B434F = {
+						CreatedOnToolsVersion = 15.1;
+					};
+					8848C8432B0D051E007B434F = {
+						CreatedOnToolsVersion = 15.1;
+					};
+					8848C8552B0D056C007B434F = {
+						CreatedOnToolsVersion = 15.1;
+					};
+					88E10F412B110D5300C08E95 = {
+						CreatedOnToolsVersion = 15.1;
+					};
+				};
+			};
+			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */;
+			compatibilityVersion = "Xcode 14.0";
+			developmentRegion = en;
+			hasScannedForEncodings = 0;
+			knownRegions = (
+				en,
+				Base,
+			);
+			mainGroup = 8848C8262B0D04BC007B434F;
+			packageReferences = (
+				88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */,
+				DEA09AC32B1FCE22001962D9 /* XCRemoteSwiftPackageReference "NetworkImage" */,
+				869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */,
+			);
+			productRefGroup = 8848C8302B0D04BC007B434F /* Products */;
+			projectDirPath = "";
+			projectRoot = "";
+			targets = (
+				8848C82E2B0D04BC007B434F /* GenerativeAISample */,
+				8848C8432B0D051E007B434F /* GenerativeAITextSample */,
+				8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */,
+				88E10F412B110D5300C08E95 /* ChatSample */,
+			);
+		};
+/* End PBXProject section */
+
+/* Begin PBXResourcesBuildPhase section */
+		8848C82D2B0D04BC007B434F /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */,
+				8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */,
+				869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8422B0D051E007B434F /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */,
+				8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */,
+				869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8542B0D056C007B434F /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */,
+				8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */,
+				869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		88E10F402B110D5300C08E95 /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */,
+				88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */,
+				869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXResourcesBuildPhase section */
+
+/* Begin PBXSourcesBuildPhase section */
+		8848C82B2B0D04BC007B434F /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */,
+				886F95DE2B17D5010036F07A /* ChatMessage.swift in Sources */,
+				88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */,
+				8848C8352B0D04BC007B434F /* ContentView.swift in Sources */,
+				886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */,
+				8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */,
+				886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */,
+				886F95DD2B17D5010036F07A /* MessageView.swift in Sources */,
+				886F95DC2B17BAEF0036F07A /* PhotoReasoningScreen.swift in Sources */,
+				886F95DB2B17BAEF0036F07A /* PhotoReasoningViewModel.swift in Sources */,
+				886F95E12B17D5010036F07A /* ConversationScreen.swift in Sources */,
+				88263BF02B239C09008AB09B /* ErrorView.swift in Sources */,
+				886F95D62B17BA010036F07A /* SummarizeViewModel.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8402B0D051E007B434F /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */,
+				8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */,
+				88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		8848C8522B0D056C007B434F /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */,
+				880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */,
+				8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+		88E10F3E2B110D5300C08E95 /* Sources */ = {
+			isa = PBXSourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */,
+				88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */,
+				88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */,
+				88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */,
+				88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */,
+				88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */,
+				88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */,
+				889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXSourcesBuildPhase section */
+
+/* Begin XCBuildConfiguration section */
+		8848C83B2B0D04BD007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ALWAYS_SEARCH_USER_PATHS = NO;
+				ASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;
+				CLANG_ANALYZER_NONNULL = YES;
+				CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
+				CLANG_CXX_LANGUAGE_STANDARD = "gnu++20";
+				CLANG_ENABLE_MODULES = YES;
+				CLANG_ENABLE_OBJC_ARC = YES;
+				CLANG_ENABLE_OBJC_WEAK = YES;
+				CLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;
+				CLANG_WARN_BOOL_CONVERSION = YES;
+				CLANG_WARN_COMMA = YES;
+				CLANG_WARN_CONSTANT_CONVERSION = YES;
+				CLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;
+				CLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;
+				CLANG_WARN_DOCUMENTATION_COMMENTS = YES;
+				CLANG_WARN_EMPTY_BODY = YES;
+				CLANG_WARN_ENUM_CONVERSION = YES;
+				CLANG_WARN_INFINITE_RECURSION = YES;
+				CLANG_WARN_INT_CONVERSION = YES;
+				CLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;
+				CLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;
+				CLANG_WARN_OBJC_LITERAL_CONVERSION = YES;
+				CLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;
+				CLANG_WARN_QUOTED_INCLUDE_IN_FRAMEWORK_HEADER = YES;
+				CLANG_WARN_RANGE_LOOP_ANALYSIS = YES;
+				CLANG_WARN_STRICT_PROTOTYPES = YES;
+				CLANG_WARN_SUSPICIOUS_MOVE = YES;
+				CLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;
+				CLANG_WARN_UNREACHABLE_CODE = YES;
+				CLANG_WARN__DUPLICATE_METHOD_MATCH = YES;
+				COPY_PHASE_STRIP = NO;
+				DEBUG_INFORMATION_FORMAT = dwarf;
+				ENABLE_STRICT_OBJC_MSGSEND = YES;
+				ENABLE_TESTABILITY = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = YES;
+				GCC_C_LANGUAGE_STANDARD = gnu17;
+				GCC_DYNAMIC_NO_PIC = NO;
+				GCC_NO_COMMON_BLOCKS = YES;
+				GCC_OPTIMIZATION_LEVEL = 0;
+				GCC_PREPROCESSOR_DEFINITIONS = (
+					"DEBUG=1",
+					"$(inherited)",
+				);
+				GCC_WARN_64_TO_32_BIT_CONVERSION = YES;
+				GCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;
+				GCC_WARN_UNDECLARED_SELECTOR = YES;
+				GCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;
+				GCC_WARN_UNUSED_FUNCTION = YES;
+				GCC_WARN_UNUSED_VARIABLE = YES;
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LOCALIZATION_PREFERS_STRING_CATALOGS = YES;
+				MTL_ENABLE_DEBUG_INFO = INCLUDE_SOURCE;
+				MTL_FAST_MATH = YES;
+				ONLY_ACTIVE_ARCH = YES;
+				SDKROOT = iphoneos;
+				SWIFT_ACTIVE_COMPILATION_CONDITIONS = "DEBUG $(inherited)";
+				SWIFT_OPTIMIZATION_LEVEL = "-Onone";
+			};
+			name = Debug;
+		};
+		8848C83C2B0D04BD007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ALWAYS_SEARCH_USER_PATHS = NO;
+				ASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;
+				CLANG_ANALYZER_NONNULL = YES;
+				CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
+				CLANG_CXX_LANGUAGE_STANDARD = "gnu++20";
+				CLANG_ENABLE_MODULES = YES;
+				CLANG_ENABLE_OBJC_ARC = YES;
+				CLANG_ENABLE_OBJC_WEAK = YES;
+				CLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;
+				CLANG_WARN_BOOL_CONVERSION = YES;
+				CLANG_WARN_COMMA = YES;
+				CLANG_WARN_CONSTANT_CONVERSION = YES;
+				CLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;
+				CLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;
+				CLANG_WARN_DOCUMENTATION_COMMENTS = YES;
+				CLANG_WARN_EMPTY_BODY = YES;
+				CLANG_WARN_ENUM_CONVERSION = YES;
+				CLANG_WARN_INFINITE_RECURSION = YES;
+				CLANG_WARN_INT_CONVERSION = YES;
+				CLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;
+				CLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;
+				CLANG_WARN_OBJC_LITERAL_CONVERSION = YES;
+				CLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;
+				CLANG_WARN_QUOTED_INCLUDE_IN_FRAMEWORK_HEADER = YES;
+				CLANG_WARN_RANGE_LOOP_ANALYSIS = YES;
+				CLANG_WARN_STRICT_PROTOTYPES = YES;
+				CLANG_WARN_SUSPICIOUS_MOVE = YES;
+				CLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;
+				CLANG_WARN_UNREACHABLE_CODE = YES;
+				CLANG_WARN__DUPLICATE_METHOD_MATCH = YES;
+				COPY_PHASE_STRIP = NO;
+				DEBUG_INFORMATION_FORMAT = "dwarf-with-dsym";
+				ENABLE_NS_ASSERTIONS = NO;
+				ENABLE_STRICT_OBJC_MSGSEND = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = YES;
+				GCC_C_LANGUAGE_STANDARD = gnu17;
+				GCC_NO_COMMON_BLOCKS = YES;
+				GCC_WARN_64_TO_32_BIT_CONVERSION = YES;
+				GCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;
+				GCC_WARN_UNDECLARED_SELECTOR = YES;
+				GCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;
+				GCC_WARN_UNUSED_FUNCTION = YES;
+				GCC_WARN_UNUSED_VARIABLE = YES;
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LOCALIZATION_PREFERS_STRING_CATALOGS = YES;
+				MTL_ENABLE_DEBUG_INFO = NO;
+				MTL_FAST_MATH = YES;
+				SDKROOT = iphoneos;
+				SWIFT_COMPILATION_MODE = wholemodule;
+				VALIDATE_PRODUCT = YES;
+			};
+			name = Release;
+		};
+		8848C83E2B0D04BD007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		8848C83F2B0D04BD007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+		8848C8502B0D051F007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		8848C8512B0D051F007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+		8848C8622B0D056D007B434F /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		8848C8632B0D056D007B434F /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+		88E10F4D2B110D5400C08E95 /* Debug */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Debug;
+		};
+		88E10F4E2B110D5400C08E95 /* Release */ = {
+			isa = XCBuildConfiguration;
+			buildSettings = {
+				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
+				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
+				CODE_SIGN_STYLE = Automatic;
+				CURRENT_PROJECT_VERSION = 1;
+				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
+				DEVELOPMENT_TEAM = "";
+				ENABLE_PREVIEWS = YES;
+				ENABLE_USER_SCRIPT_SANDBOXING = NO;
+				GENERATE_INFOPLIST_FILE = YES;
+				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
+				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
+				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
+				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
+				LD_RUNPATH_SEARCH_PATHS = (
+					"$(inherited)",
+					"@executable_path/Frameworks",
+				);
+				MARKETING_VERSION = 1.0;
+				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
+				PRODUCT_NAME = "$(TARGET_NAME)";
+				SWIFT_EMIT_LOC_STRINGS = YES;
+				SWIFT_VERSION = 5.0;
+				TARGETED_DEVICE_FAMILY = "1,2";
+			};
+			name = Release;
+		};
+/* End XCBuildConfiguration section */
+
+/* Begin XCConfigurationList section */
+		8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C83B2B0D04BD007B434F /* Debug */,
+				8848C83C2B0D04BD007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C83E2B0D04BD007B434F /* Debug */,
+				8848C83F2B0D04BD007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C8502B0D051F007B434F /* Debug */,
+				8848C8512B0D051F007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				8848C8622B0D056D007B434F /* Debug */,
+				8848C8632B0D056D007B434F /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+		88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */ = {
+			isa = XCConfigurationList;
+			buildConfigurations = (
+				88E10F4D2B110D5400C08E95 /* Debug */,
+				88E10F4E2B110D5400C08E95 /* Release */,
+			);
+			defaultConfigurationIsVisible = 0;
+			defaultConfigurationName = Release;
+		};
+/* End XCConfigurationList section */
+
+/* Begin XCLocalSwiftPackageReference section */
+		869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */ = {
+			isa = XCLocalSwiftPackageReference;
+			relativePath = ../..;
+		};
+/* End XCLocalSwiftPackageReference section */
+
+/* Begin XCRemoteSwiftPackageReference section */
+		88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */ = {
+			isa = XCRemoteSwiftPackageReference;
+			repositoryURL = "https://github.com/gonzalezreal/swift-markdown-ui";
+			requirement = {
+				kind = revision;
+				revision = 5df8a4adedd6ae4eb2455ef60ff75183984daeb8;
+			};
+		};
+		DEA09AC32B1FCE22001962D9 /* XCRemoteSwiftPackageReference "NetworkImage" */ = {
+			isa = XCRemoteSwiftPackageReference;
+			repositoryURL = "https://github.com/gonzalezreal/NetworkImage";
+			requirement = {
+				kind = revision;
+				revision = 7aff8d1b31148d32c5933d75557d42f6323ee3d1;
+			};
+		};
+/* End XCRemoteSwiftPackageReference section */
+
+/* Begin XCSwiftPackageProductDependency section */
+		869200AA2B86BE3100482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		869200AC2B86BEA100482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		869200AE2B86BFBC00482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		869200B02B86BFF500482873 /* FirebaseVertexAI */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAI;
+		};
+		88209C232B0FBE1700F64795 /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		886F95D72B17BA420036F07A /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		886F95E22B17D6630036F07A /* GenerativeAIUIComponents */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = GenerativeAIUIComponents;
+		};
+		88B8A91D2B0FC55100424728 /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = GenerativeAIUIComponents;
+		};
+		88D9474C2B14F27E008B5580 /* MarkdownUI */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
+			productName = MarkdownUI;
+		};
+		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = GenerativeAIUIComponents;
+		};
+/* End XCSwiftPackageProductDependency section */
+	};
+	rootObject = 8848C8272B0D04BC007B434F /* Project object */;
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
new file mode 100644
index 00000000000..34331bf6eb7
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
@@ -0,0 +1,48 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import SwiftUI
+
+struct ContentView: View {
+  @StateObject
+  var viewModel = ConversationViewModel()
+
+  var body: some View {
+    NavigationStack {
+      List {
+        NavigationLink {
+          SummarizeScreen()
+        } label: {
+          Label("Text", systemImage: "doc.text")
+        }
+        NavigationLink {
+          PhotoReasoningScreen()
+        } label: {
+          Label("Multi-modal", systemImage: "doc.richtext")
+        }
+        NavigationLink {
+          ConversationScreen()
+            .environmentObject(viewModel)
+        } label: {
+          Label("Chat", systemImage: "ellipsis.message.fill")
+        }
+      }
+      .navigationTitle("Generative AI Samples")
+    }
+  }
+}
+
+#Preview {
+  ContentView()
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
new file mode 100644
index 00000000000..7c709111b7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
@@ -0,0 +1,29 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import SwiftUI
+
+@main
+struct GenerativeAISampleApp: App {
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      ContentView()
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json
new file mode 100644
index 00000000000..eb878970081
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AccentColor.colorset/Contents.json
@@ -0,0 +1,11 @@
+{
+  "colors" : [
+    {
+      "idiom" : "universal"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json
new file mode 100644
index 00000000000..13613e3ee1a
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/AppIcon.appiconset/Contents.json
@@ -0,0 +1,13 @@
+{
+  "images" : [
+    {
+      "idiom" : "universal",
+      "platform" : "ios",
+      "size" : "1024x1024"
+    }
+  ],
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Assets.xcassets/Contents.json
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
new file mode 100644
index 00000000000..c800e1f7be8
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
@@ -0,0 +1,31 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseCore
+import SwiftUI
+
+@main
+struct GenerativeAITextSampleApp: App {
+  init() {
+    FirebaseApp.configure()
+  }
+
+  var body: some Scene {
+    WindowGroup {
+      NavigationStack {
+        SummarizeScreen()
+      }
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json
new file mode 100644
index 00000000000..73c00596a7f
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Preview Content/Preview Assets.xcassets/Contents.json	
@@ -0,0 +1,6 @@
+{
+  "info" : {
+    "author" : "xcode",
+    "version" : 1
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
new file mode 100644
index 00000000000..8fbb89f4482
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -0,0 +1,74 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import MarkdownUI
+import SwiftUI
+
+struct SummarizeScreen: View {
+  @StateObject var viewModel = SummarizeViewModel()
+  @State var userInput = ""
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      Text("Enter some text, then tap on _Go_ to summarize it.")
+      HStack(alignment: .top) {
+        TextField("Enter text summarize", text: $userInput, axis: .vertical)
+          .textFieldStyle(.roundedBorder)
+          .onSubmit {
+            onSummarizeTapped()
+          }
+        Button("Go") {
+          onSummarizeTapped()
+        }
+        .padding(.top, 4)
+      }
+      .padding([.horizontal, .bottom])
+
+      List {
+        HStack(alignment: .top) {
+          if viewModel.inProgress {
+            ProgressView()
+          } else {
+            Image(systemName: "cloud.circle.fill")
+              .font(.title2)
+          }
+
+          Markdown("\(viewModel.outputText)")
+        }
+        .listRowSeparator(.hidden)
+      }
+      .listStyle(.plain)
+    }
+    .navigationTitle("Text sample")
+  }
+
+  private func onSummarizeTapped() {
+    Task {
+      await viewModel.summarize(inputText: userInput)
+    }
+  }
+}
+
+#Preview {
+  NavigationStack {
+    SummarizeScreen()
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
new file mode 100644
index 00000000000..fb5b349ac82
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -0,0 +1,68 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import OSLog
+
+@MainActor
+class SummarizeViewModel: ObservableObject {
+  private var logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "generative-ai")
+
+  @Published
+  var outputText = ""
+
+  @Published
+  var errorMessage: String?
+
+  @Published
+  var inProgress = false
+
+  private var model: GenerativeModel?
+
+  init() {
+    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+  }
+
+  func summarize(inputText: String) async {
+    defer {
+      inProgress = false
+    }
+    guard let model else {
+      return
+    }
+
+    do {
+      inProgress = true
+      errorMessage = nil
+      outputText = ""
+
+      let prompt = "Summarize the following text for me: \(inputText)"
+
+      let outputContentStream = model.generateContentStream(prompt)
+
+      // stream response
+      for try await outputContent in outputContentStream {
+        guard let line = outputContent.text else {
+          return
+        }
+
+        outputText = outputText + line
+      }
+    } catch {
+      logger.error("\(error.localizedDescription)")
+      errorMessage = error.localizedDescription
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift
new file mode 100644
index 00000000000..808f5f42a97
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Package.swift
@@ -0,0 +1,35 @@
+// swift-tools-version: 5.9
+// The swift-tools-version declares the minimum version of Swift required to build this package.
+
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+import PackageDescription
+
+let package = Package(
+  name: "GenerativeAIUIComponents",
+  platforms: [
+    .iOS(.v16),
+  ],
+  products: [
+    .library(
+      name: "GenerativeAIUIComponents",
+      targets: ["GenerativeAIUIComponents"]
+    ),
+  ],
+  targets: [
+    .target(
+      name: "GenerativeAIUIComponents"
+    ),
+  ]
+)
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
new file mode 100644
index 00000000000..317baaaf3c1
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
@@ -0,0 +1,82 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import SwiftUI
+public struct InputField<Label>: View where Label: View {
+  @Binding
+  private var text: String
+
+  private var title: String?
+  private var label: () -> Label
+
+  @Environment(\.submitHandler)
+  var submitHandler
+
+  private func submit() {
+    if let submitHandler {
+      submitHandler()
+    }
+  }
+
+  public init(_ title: String? = nil, text: Binding<String>,
+              @ViewBuilder label: @escaping () -> Label) {
+    self.title = title
+    _text = text
+    self.label = label
+  }
+
+  public var body: some View {
+    VStack(alignment: .leading) {
+      HStack(alignment: .bottom) {
+        VStack(alignment: .leading) {
+          TextField(
+            title ?? "",
+            text: $text,
+            axis: .vertical
+          )
+          .padding(.vertical, 4)
+          .onSubmit(submit)
+        }
+        .padding(.horizontal, 8)
+        .padding(.vertical, 4)
+        .overlay {
+          RoundedRectangle(
+            cornerRadius: 8,
+            style: .continuous
+          )
+          .stroke(Color(UIColor.systemFill), lineWidth: 1)
+        }
+
+        Button(action: submit, label: label)
+          .padding(.top, 4)
+      }
+    }
+    .padding(.horizontal)
+  }
+}
+
+#Preview {
+  struct Wrapper: View {
+    @State var userInput: String = ""
+
+    var body: some View {
+      InputField("Message", text: $userInput) {
+        Image(systemName: "arrow.up.circle.fill")
+          .font(.title)
+      }
+    }
+  }
+
+  return Wrapper()
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
new file mode 100644
index 00000000000..9519fb38a4b
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
@@ -0,0 +1,183 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import PhotosUI
+import SwiftUI
+
+struct MultimodalInputFieldSubmitHandler: EnvironmentKey {
+  static var defaultValue: (() -> Void)?
+}
+
+extension EnvironmentValues {
+  var submitHandler: (() -> Void)? {
+    get { self[MultimodalInputFieldSubmitHandler.self] }
+    set { self[MultimodalInputFieldSubmitHandler.self] = newValue }
+  }
+}
+
+public extension View {
+  func onSubmit(submitHandler: @escaping () -> Void) -> some View {
+    environment(\.submitHandler, submitHandler)
+  }
+}
+
+public struct MultimodalInputField: View {
+  @Binding public var text: String
+  @Binding public var selection: [PhotosPickerItem]
+
+  @Environment(\.submitHandler) var submitHandler
+
+  @State private var selectedImages = [Image]()
+
+  @State private var isChooseAttachmentTypePickerShowing = false
+  @State private var isAttachmentPickerShowing = false
+
+  private func showChooseAttachmentTypePicker() {
+    isChooseAttachmentTypePickerShowing.toggle()
+  }
+
+  private func showAttachmentPicker() {
+    isAttachmentPickerShowing.toggle()
+  }
+
+  private func submit() {
+    if let submitHandler {
+      submitHandler()
+    }
+  }
+
+  public init(text: Binding<String>,
+              selection: Binding<[PhotosPickerItem]>) {
+    _text = text
+    _selection = selection
+  }
+
+  public var body: some View {
+    VStack(alignment: .leading) {
+      HStack(alignment: .top) {
+        Button(action: showChooseAttachmentTypePicker) {
+          Image(systemName: "plus")
+        }
+        .padding(.top, 4)
+
+        VStack(alignment: .leading) {
+          TextField(
+            "Upload an image, and then ask a question about it",
+            text: $text,
+            axis: .vertical
+          )
+          .padding(.vertical, 4)
+          .onSubmit(submit)
+
+          if selectedImages.count > 0 {
+            ScrollView(.horizontal) {
+              LazyHStack {
+                ForEach(0 ..< selectedImages.count, id: \.self) { i in
+                  HStack {
+                    selectedImages[i]
+                      .resizable()
+                      .scaledToFill()
+                      .frame(width: 50, height: 50)
+                      .cornerRadius(8)
+                  }
+                }
+              }
+            }
+            .frame(height: 50)
+          }
+        }
+        .padding(.horizontal, 8)
+        .padding(.vertical, 4)
+        .overlay {
+          RoundedRectangle(
+            cornerRadius: 8,
+            style: .continuous
+          )
+          .stroke(Color(UIColor.systemFill), lineWidth: 1)
+        }
+
+        Button(action: submit) {
+          Text("Go")
+        }
+        .padding(.top, 4)
+      }
+    }
+    .padding(.horizontal)
+    .confirmationDialog(
+      "Select an image",
+      isPresented: $isChooseAttachmentTypePickerShowing,
+      titleVisibility: .hidden
+    ) {
+      Button(action: showAttachmentPicker) {
+        Text("Photo & Video Library")
+      }
+    }
+    .photosPicker(isPresented: $isAttachmentPickerShowing, selection: $selection)
+    .onChange(of: selection) { _ in
+      Task {
+        selectedImages.removeAll()
+
+        for item in selection {
+          if let data = try? await item.loadTransferable(type: Data.self) {
+            if let uiImage = UIImage(data: data) {
+              let image = Image(uiImage: uiImage)
+              selectedImages.append(image)
+            }
+          }
+        }
+      }
+    }
+  }
+}
+
+#Preview {
+  struct Wrapper: View {
+    @State var userInput: String = ""
+    @State var selectedItems = [PhotosPickerItem]()
+
+    @State private var selectedImages = [Image]()
+
+    var body: some View {
+      MultimodalInputField(text: $userInput, selection: $selectedItems)
+        .onChange(of: selectedItems) { _ in
+          Task {
+            selectedImages.removeAll()
+
+            for item in selectedItems {
+              if let data = try? await item.loadTransferable(type: Data.self) {
+                if let uiImage = UIImage(data: data) {
+                  let image = Image(uiImage: uiImage)
+                  selectedImages.append(image)
+                }
+              }
+            }
+          }
+        }
+
+      List {
+        ForEach(0 ..< $selectedImages.count, id: \.self) { i in
+          HStack {
+            selectedImages[i]
+              .resizable()
+              .scaledToFill()
+              .frame(width: .infinity)
+              .cornerRadius(8)
+          }
+        }
+      }
+    }
+  }
+
+  return Wrapper()
+}
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 02fdb82807f..8a5b7b05415 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -16,7 +16,9 @@ import Foundation
 
 import FirebaseAppCheckInterop
 import FirebaseCore
-import GoogleGenerativeAI
+
+// Exports the GoogleGenerativeAI module to users of the SDK.
+@_exported import GoogleGenerativeAI
 
 // Avoids exposing internal FirebaseCore APIs to Swift users.
 @_implementationOnly import FirebaseCoreExtension
@@ -90,11 +92,9 @@ open class VertexAI: NSObject {
 
   lazy var model: GenerativeModel = {
     let options = RequestOptions(
-      endpoint: "\(location)-aiplatform.googleapis.com",
-      hooks: [
-        addAccessTokenHeader,
-        addAppCheckHeader,
-      ]
+      apiVersion: "v2beta",
+      endpoint: "staging-firebaseml.sandbox.googleapis.com",
+      hooks: [addAppCheckHeader]
     )
     return GenerativeModel(
       name: modelResouceName,
@@ -103,8 +103,6 @@ open class VertexAI: NSObject {
     )
   }()
 
-  private static let accessTokenEnvKey = "FIRVertexAIAccessToken"
-
   init(app: FirebaseApp, location: String, modelResourceName: String) {
     self.app = app
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
@@ -127,29 +125,6 @@ open class VertexAI: NSObject {
 
   // MARK: Request Hooks
 
-  /// Add a Google Cloud access token in an Authorization header in the provided request.
-  ///
-  /// This is a temporary workaround until Vertex AI can be called with an API key.
-  ///
-  /// - Parameter request: The `URLRequest` to modify by adding an access token.
-  func addAccessTokenHeader(request: inout URLRequest) {
-    // Remove the API key header, it is not supported by Vertex AI.
-    if var headers = request.allHTTPHeaderFields {
-      headers.removeValue(forKey: "x-goog-api-key")
-    }
-
-    guard let accessToken = ProcessInfo.processInfo.environment[VertexAI.accessTokenEnvKey] else {
-      print("""
-      Vertex AI requires an Access Token for authorization:
-      1. Get an access token by running `gcloud auth print-access-token`
-      2. Set it in the \(VertexAI.accessTokenEnvKey) environment variable.
-      """)
-      return
-    }
-
-    request.addValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
-  }
-
   /// Adds an App Check token to the provided request, if App Check is included in the app.
   ///
   /// This demonstrates how an App Check token can be added to requests; it is currently ignored by

From 6caba0d00757a460a9720972415f0fde1ef13725 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 22 Feb 2024 11:26:08 -0500
Subject: [PATCH 061/124] Update generative-ai-swift dependency

---
 FirebaseVertexAI/Sources/VertexAI.swift | 38 ++-----------------------
 Package.swift                           |  2 +-
 2 files changed, 3 insertions(+), 37 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 8a5b7b05415..850ed74a226 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -28,55 +28,21 @@ import FirebaseCore
 open class VertexAI: NSObject {
   // MARK: - Public APIs
 
-  // Option 1: Return `GoogleGenerativeAI.GenerativeModel` instances.
-
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   ///
   /// This instance is configured with the default `FirebaseApp`.
   public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
     .GenerativeModel {
-    return vertexAI(modelName: modelName, location: location).model
+    return generativeModel(app: FirebaseApp.app()!, modelName: modelName, location: location)
   }
 
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   public static func generativeModel(app: FirebaseApp, modelName: String,
                                      location: String) -> GoogleGenerativeAI.GenerativeModel {
-    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location).model
-  }
-
-  // Option 2: Return `VertexAI` instances with a similar API surface to
-  // `GoogleGenerativeAI.GenerativeModel`.
-  // Some types are re-used from the Google AI SDK for requests and responses, e.g.,
-  // `GenerateContentResponse`.
-
-  /// The default `VertexAI` instance.
-  ///
-  /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI(modelName: String, location: String) -> VertexAI {
-    return vertexAI(app: FirebaseApp.app()!, modelName: modelName, location: location)
-  }
-
-  public static func vertexAI(app: FirebaseApp, modelName: String, location: String) -> VertexAI {
     let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                             in: app.container)
     let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
-    return provider.vertexAI(location: location, modelResourceName: modelResourceName)
-  }
-
-  public func generateContentStream(_ parts: GoogleGenerativeAI
-    .PartsRepresentable...)
-    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
-    return model.generateContentStream([GoogleGenerativeAI.ModelContent(parts: parts)])
-  }
-
-  public func generateContentStream(_ content: [GoogleGenerativeAI.ModelContent])
-    -> AsyncThrowingStream<GoogleGenerativeAI.GenerateContentResponse, Error> {
-    return model.generateContentStream(content)
-  }
-
-  public func startChat(history: [GoogleGenerativeAI.ModelContent] = []) -> GoogleGenerativeAI
-    .Chat {
-    return model.startChat(history: history)
+    return provider.vertexAI(location: location, modelResourceName: modelResourceName).model
   }
 
   // MARK: - Private
diff --git a/Package.swift b/Package.swift
index 78b681e8213..b77185fb42d 100644
--- a/Package.swift
+++ b/Package.swift
@@ -195,7 +195,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "46e56ae138f152528b7b50520a22777f3d8adb18"
+      revision: "ae0c9b580d2ea5d6003739b1e14cc5c62f2ffa62"
     ),
   ],
   targets: [

From 57aabb62a41496418fd33fb4cc10950c6cbd0d5a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 22 Feb 2024 11:47:55 -0500
Subject: [PATCH 062/124] Fix optional VertexAIProvider unwrapping

---
 FirebaseVertexAI/Sources/VertexAI.swift | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 850ed74a226..9d1b8b4016c 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -39,10 +39,14 @@ open class VertexAI: NSObject {
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   public static func generativeModel(app: FirebaseApp, modelName: String,
                                      location: String) -> GoogleGenerativeAI.GenerativeModel {
-    let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
-                                                            in: app.container)
+    guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                                  in: app.container) else {
+      fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
+    }
     let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
-    return provider.vertexAI(location: location, modelResourceName: modelResourceName).model
+    let vertexAI = provider.vertexAI(location: location, modelResourceName: modelResourceName)
+
+    return vertexAI.model
   }
 
   // MARK: - Private

From 202919b62225495015e50c8baed994941bb4b920 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 23 Feb 2024 14:36:12 -0500
Subject: [PATCH 063/124] Update `generative-ai-swift` dependency

---
 Package.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Package.swift b/Package.swift
index b77185fb42d..d6eaca85e57 100644
--- a/Package.swift
+++ b/Package.swift
@@ -195,7 +195,7 @@ let package = Package(
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
     .package(
       url: "https://github.com/google/generative-ai-swift.git",
-      revision: "ae0c9b580d2ea5d6003739b1e14cc5c62f2ffa62"
+      revision: "c9f2c4913bc65aa267815962c7e91358c2d8463f"
     ),
   ],
   targets: [

From e2003e2e1c8bbd2d029f8d9c4244bc4dbc0b15f9 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 4 Mar 2024 16:50:37 -0500
Subject: [PATCH 064/124] Prototype and sample fixes

---
 .../ChatSample/Views/ErrorDetailsView.swift   | 19 ++-------------
 .../ViewModels/PhotoReasoningViewModel.swift  |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift       | 24 ++++++++++++++-----
 3 files changed, 21 insertions(+), 24 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
index e601876b2d9..399777ad1e8 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -152,23 +152,8 @@ struct ErrorDetailsView: View {
             SubtitleMarkdownFormRow(
               title: "Help",
               value: """
-              Please provide a valid value for `API_KEY` in the `GenerativeAI-Info.plist` file.
-              """
-            )
-          }
-
-        case GenerateContentError.unsupportedUserLocation:
-          Section("Error Type") {
-            Text("Unsupported User Location")
-          }
-
-          Section("Details") {
-            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
-            SubtitleMarkdownFormRow(
-              title: "Help",
-              value: """
-              The API is unsupported in your location (country / territory); please see the list of
-              [available regions](https://ai.google.dev/available_regions#available_regions).
+              The `API_KEY` provided in the `GoogleService-Info.plist` file is invalid. Download a
+              new copy of the file from the [Firebase Console](https://console.firebase.google.com).
               """
             )
           }
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index aa646c127df..dc41e00444a 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -62,7 +62,7 @@ class PhotoReasoningViewModel: ObservableObject {
 
       let prompt = "Look at the image(s), and then answer the following question: \(userInput)"
 
-      var images = [PartsRepresentable]()
+      var images = [any ThrowingPartsRepresentable]()
       for item in selectedItems {
         if let data = try? await item.loadTransferable(type: Data.self) {
           guard let image = UIImage(data: data) else {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 9d1b8b4016c..8fb0f9e8c68 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -33,7 +33,10 @@ open class VertexAI: NSObject {
   /// This instance is configured with the default `FirebaseApp`.
   public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
     .GenerativeModel {
-    return generativeModel(app: FirebaseApp.app()!, modelName: modelName, location: location)
+    guard let app = FirebaseApp.app() else {
+      fatalError("No instance of the default Firebase app was found.")
+    }
+    return generativeModel(app: app, modelName: modelName, location: location)
   }
 
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
@@ -66,9 +69,12 @@ open class VertexAI: NSObject {
       endpoint: "staging-firebaseml.sandbox.googleapis.com",
       hooks: [addAppCheckHeader]
     )
+    guard let apiKey = app.options.apiKey else {
+      fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
+    }
     return GenerativeModel(
       name: modelResouceName,
-      apiKey: app.options.apiKey!,
+      apiKey: apiKey,
       requestOptions: options
     )
   }()
@@ -86,8 +92,14 @@ open class VertexAI: NSObject {
       return modelName
     }
     guard let projectID = app.options.projectID else {
-      print("The FirebaseApp is missing a project ID.")
-      return modelName
+      fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
+    }
+    guard !location.isEmpty else {
+      fatalError("""
+      No location specified; see
+      https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
+      list of available regions.
+      """)
     }
 
     return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
@@ -95,14 +107,14 @@ open class VertexAI: NSObject {
 
   // MARK: Request Hooks
 
-  /// Adds an App Check token to the provided request, if App Check is included in the app.
+  /// Adds an App Check token to the provided request if App Check is included in the app.
   ///
   /// This demonstrates how an App Check token can be added to requests; it is currently ignored by
   /// the backend.
   ///
   /// - Parameter request: The `URLRequest` to modify by adding an App Check token header.
   func addAppCheckHeader(request: inout URLRequest) async {
-    guard let appCheck = appCheck else {
+    guard let appCheck else {
       return
     }
 

From ecec18d985f7b717cd4d53242df6491ee64026b2 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 18 Mar 2024 19:27:53 +0000
Subject: [PATCH 065/124] Fork `google-generative-ai` for Firebase (#12564)

---
 FirebaseVertexAI/Sources/Chat.swift           | 184 +++++++++++
 .../Sources/CountTokensRequest.swift          |  45 +++
 FirebaseVertexAI/Sources/Errors.swift         | 189 +++++++++++
 .../Sources/GenerateContentError.swift        |  44 +++
 .../Sources/GenerateContentRequest.swift      |  49 +++
 .../Sources/GenerateContentResponse.swift     | 286 +++++++++++++++++
 .../Sources/GenerationConfig.swift            |  86 +++++
 .../Sources/GenerativeAIRequest.swift         |  46 +++
 .../Sources/GenerativeAIService.swift         | 257 +++++++++++++++
 .../Sources/GenerativeAISwift.swift           |  26 ++
 .../Sources/GenerativeModel.swift             | 298 ++++++++++++++++++
 FirebaseVertexAI/Sources/Logging.swift        |  56 ++++
 FirebaseVertexAI/Sources/ModelContent.swift   | 141 +++++++++
 .../Sources/PartsRepresentable+Image.swift    | 107 +++++++
 .../Sources/PartsRepresentable.swift          |  66 ++++
 FirebaseVertexAI/Sources/Safety.swift         | 182 +++++++++++
 Package.swift                                 |   6 +-
 17 files changed, 2067 insertions(+), 1 deletion(-)
 create mode 100644 FirebaseVertexAI/Sources/Chat.swift
 create mode 100644 FirebaseVertexAI/Sources/CountTokensRequest.swift
 create mode 100644 FirebaseVertexAI/Sources/Errors.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerateContentError.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerateContentRequest.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerateContentResponse.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerationConfig.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeAIRequest.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeAIService.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeAISwift.swift
 create mode 100644 FirebaseVertexAI/Sources/GenerativeModel.swift
 create mode 100644 FirebaseVertexAI/Sources/Logging.swift
 create mode 100644 FirebaseVertexAI/Sources/ModelContent.swift
 create mode 100644 FirebaseVertexAI/Sources/PartsRepresentable+Image.swift
 create mode 100644 FirebaseVertexAI/Sources/PartsRepresentable.swift
 create mode 100644 FirebaseVertexAI/Sources/Safety.swift

diff --git a/FirebaseVertexAI/Sources/Chat.swift b/FirebaseVertexAI/Sources/Chat.swift
new file mode 100644
index 00000000000..c7cfb859fa8
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Chat.swift
@@ -0,0 +1,184 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// An object that represents a back-and-forth chat with a model, capturing the history and saving
+/// the context in memory between each message sent.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public class Chat {
+  private let model: GenerativeModel
+
+  /// Initializes a new chat representing a 1:1 conversation between model and user.
+  init(model: GenerativeModel, history: [ModelContent]) {
+    self.model = model
+    self.history = history
+  }
+
+  /// The previous content from the chat that has been successfully sent and received from the
+  /// model. This will be provided to the model for each message sent as context for the discussion.
+  public var history: [ModelContent]
+
+  /// See ``sendMessage(_:)-3ify5``.
+  public func sendMessage(_ parts: any ThrowingPartsRepresentable...) async throws
+    -> GenerateContentResponse {
+    return try await sendMessage([ModelContent(parts: parts)])
+  }
+
+  /// Sends a message using the existing history of this chat as context. If successful, the message
+  /// and response will be added to the history. If unsuccessful, history will remain unchanged.
+  /// - Parameter content: The new content to send as a single chat message.
+  /// - Returns: The model's response if no error occurred.
+  /// - Throws: A ``GenerateContentError`` if an error occurred.
+  public func sendMessage(_ content: @autoclosure () throws -> [ModelContent]) async throws
+    -> GenerateContentResponse {
+    // Ensure that the new content has the role set.
+    let newContent: [ModelContent]
+    do {
+      newContent = try content().map(populateContentRole(_:))
+    } catch let underlying {
+      if let contentError = underlying as? ImageConversionError {
+        throw GenerateContentError.promptImageContentError(underlying: contentError)
+      } else {
+        throw GenerateContentError.internalError(underlying: underlying)
+      }
+    }
+
+    // Send the history alongside the new message as context.
+    let request = history + newContent
+    let result = try await model.generateContent(request)
+    guard let reply = result.candidates.first?.content else {
+      let error = NSError(domain: "com.google.generative-ai",
+                          code: -1,
+                          userInfo: [
+                            NSLocalizedDescriptionKey: "No candidates with content available.",
+                          ])
+      throw GenerateContentError.internalError(underlying: error)
+    }
+
+    // Make sure we inject the role into the content received.
+    let toAdd = ModelContent(role: "model", parts: reply.parts)
+
+    // Append the request and successful result to history, then return the value.
+    history.append(contentsOf: newContent)
+    history.append(toAdd)
+    return result
+  }
+
+  /// See ``sendMessageStream(_:)-4abs3``.
+  @available(macOS 12.0, *)
+  public func sendMessageStream(_ parts: any ThrowingPartsRepresentable...)
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    return try sendMessageStream([ModelContent(parts: parts)])
+  }
+
+  /// Sends a message using the existing history of this chat as context. If successful, the message
+  /// and response will be added to the history. If unsuccessful, history will remain unchanged.
+  /// - Parameter content: The new content to send as a single chat message.
+  /// - Returns: A stream containing the model's response or an error if an error occurred.
+  @available(macOS 12.0, *)
+  public func sendMessageStream(_ content: @autoclosure () throws -> [ModelContent])
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    let resolvedContent: [ModelContent]
+    do {
+      resolvedContent = try content()
+    } catch let underlying {
+      return AsyncThrowingStream { continuation in
+        let error: Error
+        if let contentError = underlying as? ImageConversionError {
+          error = GenerateContentError.promptImageContentError(underlying: contentError)
+        } else {
+          error = GenerateContentError.internalError(underlying: underlying)
+        }
+        continuation.finish(throwing: error)
+      }
+    }
+
+    return AsyncThrowingStream { continuation in
+      Task {
+        var aggregatedContent: [ModelContent] = []
+
+        // Ensure that the new content has the role set.
+        let newContent: [ModelContent] = resolvedContent.map(populateContentRole(_:))
+
+        // Send the history alongside the new message as context.
+        let request = history + newContent
+        let stream = model.generateContentStream(request)
+        do {
+          for try await chunk in stream {
+            // Capture any content that's streaming. This should be populated if there's no error.
+            if let chunkContent = chunk.candidates.first?.content {
+              aggregatedContent.append(chunkContent)
+            }
+
+            // Pass along the chunk.
+            continuation.yield(chunk)
+          }
+        } catch {
+          // Rethrow the error that the underlying stream threw. Don't add anything to history.
+          continuation.finish(throwing: error)
+          return
+        }
+
+        // Save the request.
+        history.append(contentsOf: newContent)
+
+        // Aggregate the content to add it to the history before we finish.
+        let aggregated = aggregatedChunks(aggregatedContent)
+        history.append(aggregated)
+
+        continuation.finish()
+      }
+    }
+  }
+
+  private func aggregatedChunks(_ chunks: [ModelContent]) -> ModelContent {
+    var parts: [ModelContent.Part] = []
+    var combinedText = ""
+    for aggregate in chunks {
+      // Loop through all the parts, aggregating the text and adding the images.
+      for part in aggregate.parts {
+        switch part {
+        case let .text(str):
+          combinedText += str
+
+        case .data(mimetype: _, _):
+          // Don't combine it, just add to the content. If there's any text pending, add that as
+          // a part.
+          if !combinedText.isEmpty {
+            parts.append(.text(combinedText))
+            combinedText = ""
+          }
+
+          parts.append(part)
+        }
+      }
+    }
+
+    if !combinedText.isEmpty {
+      parts.append(.text(combinedText))
+    }
+
+    return ModelContent(role: "model", parts: parts)
+  }
+
+  /// Populates the `role` field with `user` if it doesn't exist. Required in chat sessions.
+  private func populateContentRole(_ content: ModelContent) -> ModelContent {
+    if content.role != nil {
+      return content
+    } else {
+      return ModelContent(role: "user", parts: content.parts)
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
new file mode 100644
index 00000000000..0a58d40acc0
--- /dev/null
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -0,0 +1,45 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct CountTokensRequest {
+  let model: String
+  let contents: [ModelContent]
+  let options: RequestOptions
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensRequest: Encodable {
+  enum CodingKeys: CodingKey {
+    case contents
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensRequest: GenerativeAIRequest {
+  typealias Response = CountTokensResponse
+
+  var url: URL {
+    URL(string: "\(GenerativeAISwift.baseURL)/\(options.apiVersion)/\(model):countTokens")!
+  }
+}
+
+/// The model's response to a count tokens request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct CountTokensResponse: Decodable {
+  /// The total number of tokens in the input given to the model as a prompt.
+  public let totalTokens: Int
+}
diff --git a/FirebaseVertexAI/Sources/Errors.swift b/FirebaseVertexAI/Sources/Errors.swift
new file mode 100644
index 00000000000..35e7e6b813b
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Errors.swift
@@ -0,0 +1,189 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+struct RPCError: Error {
+  let httpResponseCode: Int
+  let message: String
+  let status: RPCStatus
+  let details: [ErrorDetails]
+
+  private var errorInfo: ErrorDetails? {
+    return details.first { $0.isErrorInfo() }
+  }
+
+  init(httpResponseCode: Int, message: String, status: RPCStatus, details: [ErrorDetails]) {
+    self.httpResponseCode = httpResponseCode
+    self.message = message
+    self.status = status
+    self.details = details
+  }
+
+  func isInvalidAPIKeyError() -> Bool {
+    return errorInfo?.reason == "API_KEY_INVALID"
+  }
+
+  func isUnsupportedUserLocationError() -> Bool {
+    return message == RPCErrorMessage.unsupportedUserLocation.rawValue
+  }
+}
+
+extension RPCError: Decodable {
+  enum CodingKeys: CodingKey {
+    case error
+  }
+
+  init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    let status = try container.decode(ErrorStatus.self, forKey: .error)
+
+    if let code = status.code {
+      httpResponseCode = code
+    } else {
+      httpResponseCode = -1
+    }
+
+    if let message = status.message {
+      self.message = message
+    } else {
+      message = "Unknown error."
+    }
+
+    if let rpcStatus = status.status {
+      self.status = rpcStatus
+    } else {
+      self.status = .unknown
+    }
+
+    details = status.details
+  }
+}
+
+struct ErrorStatus {
+  let code: Int?
+  let message: String?
+  let status: RPCStatus?
+  let details: [ErrorDetails]
+}
+
+struct ErrorDetails {
+  static let errorInfoType = "type.googleapis.com/google.rpc.ErrorInfo"
+
+  let type: String
+  let reason: String?
+  let domain: String?
+
+  func isErrorInfo() -> Bool {
+    return type == ErrorDetails.errorInfoType
+  }
+}
+
+extension ErrorDetails: Decodable, Equatable {
+  enum CodingKeys: String, CodingKey {
+    case type = "@type"
+    case reason
+    case domain
+  }
+}
+
+extension ErrorStatus: Decodable {
+  enum CodingKeys: CodingKey {
+    case code
+    case message
+    case status
+    case details
+  }
+
+  init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    code = try container.decodeIfPresent(Int.self, forKey: .code)
+    message = try container.decodeIfPresent(String.self, forKey: .message)
+    do {
+      status = try container.decodeIfPresent(RPCStatus.self, forKey: .status)
+    } catch {
+      status = .unknown
+    }
+    if container.contains(.details) {
+      details = try container.decode([ErrorDetails].self, forKey: .details)
+    } else {
+      details = []
+    }
+  }
+}
+
+enum RPCStatus: String, Decodable {
+  // Not an error; returned on success.
+  case ok = "OK"
+
+  // The operation was cancelled, typically by the caller.
+  case cancelled = "CANCELLED"
+
+  // Unknown error.
+  case unknown = "UNKNOWN"
+
+  // The client specified an invalid argument.
+  case invalidArgument = "INVALID_ARGUMENT"
+
+  // The deadline expired before the operation could complete.
+  case deadlineExceeded = "DEADLINE_EXCEEDED"
+
+  // Some requested entity (e.g., file or directory) was not found.
+  case notFound = "NOT_FOUND"
+
+  // The entity that a client attempted to create (e.g., file or directory) already exists.
+  case alreadyExists = "ALREADY_EXISTS"
+
+  // The caller does not have permission to execute the specified operation.
+  case permissionDenied = "PERMISSION_DENIED"
+
+  // The request does not have valid authentication credentials for the operation.
+  case unauthenticated = "UNAUTHENTICATED"
+
+  // Some resource has been exhausted, perhaps a per-user quota, or perhaps the entire file system
+  // is out of space.
+  case resourceExhausted = "RESOURCE_EXHAUSTED"
+
+  // The operation was rejected because the system is not in a state required for the operation's
+  // execution.
+  case failedPrecondition = "FAILED_PRECONDITION"
+
+  // The operation was aborted, typically due to a concurrency issue such as a sequencer check
+  // failure or transaction abort.
+  case aborted = "ABORTED"
+
+  // The operation was attempted past the valid range.
+  case outOfRange = "OUT_OF_RANGE"
+
+  // The operation is not implemented or is not supported/enabled in this service.
+  case unimplemented = "UNIMPLEMENTED"
+
+  // Internal errors.
+  case internalError = "INTERNAL"
+
+  // The service is currently unavailable.
+  case unavailable = "UNAVAILABLE"
+
+  // Unrecoverable data loss or corruption.
+  case dataLoss = "DATA_LOSS"
+}
+
+enum RPCErrorMessage: String {
+  case unsupportedUserLocation = "User location is not supported for the API use."
+}
+
+enum InvalidCandidateError: Error {
+  case emptyContent(underlyingError: Error)
+  case malformedContent(underlyingError: Error)
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentError.swift b/FirebaseVertexAI/Sources/GenerateContentError.swift
new file mode 100644
index 00000000000..055928e5f1b
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerateContentError.swift
@@ -0,0 +1,44 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// Errors that occur when generating content from a model.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum GenerateContentError: Error {
+  /// An error occurred when constructing the prompt. Examine the related error for details.
+  case promptImageContentError(underlying: ImageConversionError)
+
+  /// An internal error occurred. See the underlying error for more context.
+  case internalError(underlying: Error)
+
+  /// A prompt was blocked. See the response's `promptFeedback.blockReason` for more information.
+  case promptBlocked(response: GenerateContentResponse)
+
+  /// A response didn't fully complete. See the `FinishReason` for more information.
+  case responseStoppedEarly(reason: FinishReason, response: GenerateContentResponse)
+
+  /// The provided API key is invalid.
+  case invalidAPIKey(message: String)
+
+  /// The user's location (region) is not supported by the API.
+  ///
+  /// See the Google documentation for a
+  /// [list of regions](https://ai.google.dev/available_regions#available_regions)
+  /// (countries and territories) where the API is available.
+  ///
+  /// - Important: The API is only available in
+  /// [specific regions](https://ai.google.dev/available_regions#available_regions).
+  case unsupportedUserLocation
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
new file mode 100644
index 00000000000..417260bb700
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -0,0 +1,49 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct GenerateContentRequest {
+  /// Model name.
+  let model: String
+  let contents: [ModelContent]
+  let generationConfig: GenerationConfig?
+  let safetySettings: [SafetySetting]?
+  let isStreaming: Bool
+  let options: RequestOptions
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentRequest: Encodable {
+  enum CodingKeys: String, CodingKey {
+    case contents
+    case generationConfig
+    case safetySettings
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentRequest: GenerativeAIRequest {
+  typealias Response = GenerateContentResponse
+
+  var url: URL {
+    let modelURL = "\(GenerativeAISwift.baseURL)/\(options.apiVersion)/\(model)"
+    if isStreaming {
+      return URL(string: "\(modelURL):streamGenerateContent?alt=sse")!
+    } else {
+      return URL(string: "\(modelURL):generateContent")!
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
new file mode 100644
index 00000000000..03153da9ae7
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -0,0 +1,286 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// The model's response to a generate content request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct GenerateContentResponse {
+  /// A list of candidate response content, ordered from best to worst.
+  public let candidates: [CandidateResponse]
+
+  /// A value containing the safety ratings for the response, or, if the request was blocked, a
+  /// reason for blocking the request.
+  public let promptFeedback: PromptFeedback?
+
+  /// The response's content as text, if it exists.
+  public var text: String? {
+    guard let candidate = candidates.first else {
+      Logging.default.error("Could not get text from a response that had no candidates.")
+      return nil
+    }
+    guard let text = candidate.content.parts.first?.text else {
+      Logging.default.error("Could not get a text part from the first candidate.")
+      return nil
+    }
+    return text
+  }
+
+  /// Initializer for SwiftUI previews or tests.
+  public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback?) {
+    self.candidates = candidates
+    self.promptFeedback = promptFeedback
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case candidates
+    case promptFeedback
+  }
+
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+
+    guard container.contains(CodingKeys.candidates) || container
+      .contains(CodingKeys.promptFeedback) else {
+      let context = DecodingError.Context(
+        codingPath: [],
+        debugDescription: "Failed to decode GenerateContentResponse;" +
+          " missing keys 'candidates' and 'promptFeedback'."
+      )
+      throw DecodingError.dataCorrupted(context)
+    }
+
+    if let candidates = try container.decodeIfPresent(
+      [CandidateResponse].self,
+      forKey: .candidates
+    ) {
+      self.candidates = candidates
+    } else {
+      candidates = []
+    }
+    promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
+  }
+}
+
+/// A struct representing a possible reply to a content generation prompt. Each content generation
+/// prompt may produce multiple candidate responses.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct CandidateResponse {
+  /// The response's content.
+  public let content: ModelContent
+
+  /// The safety rating of the response content.
+  public let safetyRatings: [SafetyRating]
+
+  /// The reason the model stopped generating content, if it exists; for example, if the model
+  /// generated a predefined stop sequence.
+  public let finishReason: FinishReason?
+
+  /// Cited works in the model's response content, if it exists.
+  public let citationMetadata: CitationMetadata?
+
+  /// Initializer for SwiftUI previews or tests.
+  public init(content: ModelContent, safetyRatings: [SafetyRating], finishReason: FinishReason?,
+              citationMetadata: CitationMetadata?) {
+    self.content = content
+    self.safetyRatings = safetyRatings
+    self.finishReason = finishReason
+    self.citationMetadata = citationMetadata
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CandidateResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case content
+    case safetyRatings
+    case finishReason
+    case finishMessage
+    case citationMetadata
+  }
+
+  /// Initializes a response from a decoder. Used for decoding server responses; not for public
+  /// use.
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+
+    do {
+      if let content = try container.decodeIfPresent(ModelContent.self, forKey: .content) {
+        self.content = content
+      } else {
+        content = ModelContent(parts: [])
+      }
+    } catch {
+      // Check if `content` can be decoded as an empty dictionary to detect the `"content": {}` bug.
+      if let content = try? container.decode([String: String].self, forKey: .content),
+         content.isEmpty {
+        throw InvalidCandidateError.emptyContent(underlyingError: error)
+      } else {
+        throw InvalidCandidateError.malformedContent(underlyingError: error)
+      }
+    }
+
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+
+    finishReason = try container.decodeIfPresent(FinishReason.self, forKey: .finishReason)
+
+    citationMetadata = try container.decodeIfPresent(
+      CitationMetadata.self,
+      forKey: .citationMetadata
+    )
+  }
+}
+
+/// A collection of source attributions for a piece of content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct CitationMetadata: Decodable {
+  /// A list of individual cited sources and the parts of the content to which they apply.
+  public let citationSources: [Citation]
+}
+
+/// A struct describing a source attribution.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct Citation: Decodable {
+  /// The inclusive beginning of a sequence in a model response that derives from a cited source.
+  public let startIndex: Int
+
+  /// The exclusive end of a sequence in a model response that derives from a cited source.
+  public let endIndex: Int
+
+  /// A link to the cited source.
+  public let uri: String
+
+  /// The license the cited source work is distributed under.
+  public let license: String
+}
+
+/// A value enumerating possible reasons for a model to terminate a content generation request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum FinishReason: String {
+  case unknown = "FINISH_REASON_UNKNOWN"
+
+  case unspecified = "FINISH_REASON_UNSPECIFIED"
+
+  /// Natural stop point of the model or provided stop sequence.
+  case stop = "STOP"
+
+  /// The maximum number of tokens as specified in the request was reached.
+  case maxTokens = "MAX_TOKENS"
+
+  /// The token generation was stopped because the response was flagged for safety reasons.
+  /// NOTE: When streaming, the Candidate.content will be empty if content filters blocked the
+  /// output.
+  case safety = "SAFETY"
+
+  /// The token generation was stopped because the response was flagged for unauthorized citations.
+  case recitation = "RECITATION"
+
+  /// All other reasons that stopped token generation.
+  case other = "OTHER"
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension FinishReason: Decodable {
+  /// Do not explicitly use. Initializer required for Decodable conformance.
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedFinishReason = FinishReason(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized FinishReason with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedFinishReason
+  }
+}
+
+/// A metadata struct containing any feedback the model had on the prompt it was provided.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct PromptFeedback {
+  /// A type describing possible reasons to block a prompt.
+  public enum BlockReason: String, Decodable {
+    /// The block reason is unknown.
+    case unknown = "UNKNOWN"
+
+    /// The block reason was not specified in the server response.
+    case unspecified = "BLOCK_REASON_UNSPECIFIED"
+
+    /// The prompt was blocked because it was deemed unsafe.
+    case safety = "SAFETY"
+
+    /// All other block reasons.
+    case other = "OTHER"
+
+    /// Do not explicitly use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedBlockReason = BlockReason(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized BlockReason with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedBlockReason
+    }
+  }
+
+  /// The reason a prompt was blocked, if it was blocked.
+  public let blockReason: BlockReason?
+
+  /// The safety ratings of the prompt.
+  public let safetyRatings: [SafetyRating]
+
+  /// Initializer for SwiftUI previews or tests.
+  public init(blockReason: BlockReason?, safetyRatings: [SafetyRating]) {
+    self.blockReason = blockReason
+    self.safetyRatings = safetyRatings
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension PromptFeedback: Decodable {
+  enum CodingKeys: CodingKey {
+    case blockReason
+    case safetyRatings
+  }
+
+  /// Do not explicitly use. Initializer required for Decodable conformance.
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    blockReason = try container.decodeIfPresent(
+      PromptFeedback.BlockReason.self,
+      forKey: .blockReason
+    )
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerationConfig.swift b/FirebaseVertexAI/Sources/GenerationConfig.swift
new file mode 100644
index 00000000000..2d1016c965c
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerationConfig.swift
@@ -0,0 +1,86 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A struct defining model parameters to be used when sending generative AI
+/// requests to the backend model.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct GenerationConfig: Encodable {
+  /// A parameter controlling the degree of randomness in token selection. A
+  /// temperature of zero is deterministic, always choosing the
+  /// highest-probability response. Typical values are between 0 and 1
+  /// inclusive. Defaults to 0 if unspecified.
+  public let temperature: Float?
+
+  /// The `topP` parameter changes how the model selects tokens for output.
+  /// Tokens are selected from the most to least probable until the sum of
+  /// their probabilities equals the `topP` value. For example, if tokens A, B,
+  /// and C have probabilities of 0.3, 0.2, and 0.1 respectively and the topP
+  /// value is 0.5, then the model will select either A or B as the next token
+  /// by using the `temperature` and exclude C as a candidate.
+  /// Defaults to 0.95 if unset.
+  public let topP: Float?
+
+  /// The `topK` parameter changes how the model selects tokens for output. A
+  /// `topK` of 1 means the selected token is the most probable among all the
+  /// tokens in the model's vocabulary, while a `topK` of 3 means that the next
+  /// token is selected from among the 3 most probable using the `temperature`.
+  /// For each token selection step, the `topK` tokens with the highest
+  /// probabilities are sampled. Tokens are then further filtered based on
+  /// `topP` with the final token selected using `temperature` sampling.
+  /// Defaults to 40 if unspecified.
+  public let topK: Int?
+
+  /// The maximum number of generated response messages to return. This value
+  /// must be between [1, 8], inclusive. If unset, this will default to 1.
+  ///
+  /// - Note: Only unique candidates are returned. Higher temperatures are more
+  ///     likely to produce unique candidates. Setting `temperature` to 0 will
+  ///     always produce exactly one candidate regardless of the
+  ///     `candidateCount`.
+  public let candidateCount: Int?
+
+  /// Specifies the maximum number of tokens that can be generated in the
+  /// response. The number of tokens per word varies depending on the
+  /// language outputted. The maximum value is capped at 1024. Defaults to 0
+  /// (unbounded).
+  public let maxOutputTokens: Int?
+
+  /// A set of up to 5 `String`s that will stop output generation. If
+  /// specified, the API will stop at the first appearance of a stop sequence.
+  /// The stop sequence will not be included as part of the response.
+  public let stopSequences: [String]?
+
+  /// Creates a new `GenerationConfig` value.
+  ///
+  /// - Parameter temperature: See ``temperature``
+  /// - Parameter topP: See ``topP``
+  /// - Parameter topK: See ``topK``
+  /// - Parameter candidateCount: See ``candidateCount``
+  /// - Parameter maxOutputTokens: See ``maxOutputTokens``
+  /// - Parameter stopSequences: See ``stopSequences``
+  public init(temperature: Float? = nil, topP: Float? = nil, topK: Int? = nil,
+              candidateCount: Int? = nil, maxOutputTokens: Int? = nil,
+              stopSequences: [String]? = nil) {
+    // Explicit init because otherwise if we re-arrange the above variables it changes the API
+    // surface.
+    self.temperature = temperature
+    self.topP = topP
+    self.topK = topK
+    self.candidateCount = candidateCount
+    self.maxOutputTokens = maxOutputTokens
+    self.stopSequences = stopSequences
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
new file mode 100644
index 00000000000..2bf94503b1a
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
@@ -0,0 +1,46 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+protocol GenerativeAIRequest: Encodable {
+  associatedtype Response: Decodable
+
+  var url: URL { get }
+
+  var options: RequestOptions { get }
+}
+
+/// Configuration parameters for sending requests to the backend.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct RequestOptions {
+  /// The request’s timeout interval in seconds; if not specified uses the default value for a
+  /// `URLRequest`.
+  let timeout: TimeInterval?
+
+  /// The API version to use in requests to the backend.
+  let apiVersion: String
+
+  /// Initializes a request options object.
+  ///
+  /// - Parameters:
+  ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
+  ///   for a `URLRequest`.
+  ///   - apiVersion The API version to use in requests to the backend; defaults to "v1".
+  public init(timeout: TimeInterval? = nil, apiVersion: String = "v1") {
+    self.timeout = timeout
+    self.apiVersion = apiVersion
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
new file mode 100644
index 00000000000..8d904738927
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -0,0 +1,257 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct GenerativeAIService {
+  /// Gives permission to talk to the backend.
+  private let apiKey: String
+
+  private let urlSession: URLSession
+
+  init(apiKey: String, urlSession: URLSession) {
+    self.apiKey = apiKey
+    self.urlSession = urlSession
+  }
+
+  func loadRequest<T: GenerativeAIRequest>(request: T) async throws -> T.Response {
+    let urlRequest = try urlRequest(request: request)
+
+    #if DEBUG
+      printCURLCommand(from: urlRequest)
+    #endif
+
+    let data: Data
+    let rawResponse: URLResponse
+    (data, rawResponse) = try await urlSession.data(for: urlRequest)
+
+    let response = try httpResponse(urlResponse: rawResponse)
+
+    // Verify the status code is 200
+    guard response.statusCode == 200 else {
+      Logging.default.error("[GoogleGenerativeAI] The server responded with an error: \(response)")
+      if let responseString = String(data: data, encoding: .utf8) {
+        Logging.network.error("[GoogleGenerativeAI] Response payload: \(responseString)")
+      }
+
+      throw parseError(responseData: data)
+    }
+
+    return try parseResponse(T.Response.self, from: data)
+  }
+
+  @available(macOS 12.0, *)
+  func loadRequestStream<T: GenerativeAIRequest>(request: T)
+    -> AsyncThrowingStream<T.Response, Error> {
+    return AsyncThrowingStream { continuation in
+      Task {
+        let urlRequest: URLRequest
+        do {
+          urlRequest = try self.urlRequest(request: request)
+        } catch {
+          continuation.finish(throwing: error)
+          return
+        }
+
+        #if DEBUG
+          printCURLCommand(from: urlRequest)
+        #endif
+
+        let stream: URLSession.AsyncBytes
+        let rawResponse: URLResponse
+        do {
+          (stream, rawResponse) = try await urlSession.bytes(for: urlRequest)
+        } catch {
+          continuation.finish(throwing: error)
+          return
+        }
+
+        // Verify the status code is 200
+        let response: HTTPURLResponse
+        do {
+          response = try httpResponse(urlResponse: rawResponse)
+        } catch {
+          continuation.finish(throwing: error)
+          return
+        }
+
+        // Verify the status code is 200
+        guard response.statusCode == 200 else {
+          Logging.default
+            .error("[GoogleGenerativeAI] The server responded with an error: \(response)")
+          var responseBody = ""
+          for try await line in stream.lines {
+            responseBody += line + "\n"
+          }
+
+          Logging.network.error("[GoogleGenerativeAI] Response payload: \(responseBody)")
+          continuation.finish(throwing: parseError(responseBody: responseBody))
+
+          return
+        }
+
+        // Received lines that are not server-sent events (SSE); these are not prefixed with "data:"
+        var extraLines: String = ""
+
+        let decoder = JSONDecoder()
+        decoder.keyDecodingStrategy = .convertFromSnakeCase
+        for try await line in stream.lines {
+          Logging.network.debug("[GoogleGenerativeAI] Stream response: \(line)")
+
+          if line.hasPrefix("data:") {
+            // We can assume 5 characters since it's utf-8 encoded, removing `data:`.
+            let jsonText = String(line.dropFirst(5))
+            let data: Data
+            do {
+              data = try jsonData(jsonText: jsonText)
+            } catch {
+              continuation.finish(throwing: error)
+              return
+            }
+
+            // Handle the content.
+            do {
+              let content = try parseResponse(T.Response.self, from: data)
+              continuation.yield(content)
+            } catch {
+              continuation.finish(throwing: error)
+              return
+            }
+          } else {
+            extraLines += line
+          }
+        }
+
+        if extraLines.count > 0 {
+          continuation.finish(throwing: parseError(responseBody: extraLines))
+          return
+        }
+
+        continuation.finish(throwing: nil)
+      }
+    }
+  }
+
+  // MARK: - Private Helpers
+
+  private func urlRequest<T: GenerativeAIRequest>(request: T) throws -> URLRequest {
+    var urlRequest = URLRequest(url: request.url)
+    urlRequest.httpMethod = "POST"
+    urlRequest.setValue(apiKey, forHTTPHeaderField: "x-goog-api-key")
+    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version)",
+                        forHTTPHeaderField: "x-goog-api-client")
+    urlRequest.setValue("application/json", forHTTPHeaderField: "Content-Type")
+    let encoder = JSONEncoder()
+    encoder.keyEncodingStrategy = .convertToSnakeCase
+    urlRequest.httpBody = try encoder.encode(request)
+
+    if let timeoutInterval = request.options.timeout {
+      urlRequest.timeoutInterval = timeoutInterval
+    }
+
+    return urlRequest
+  }
+
+  private func httpResponse(urlResponse: URLResponse) throws -> HTTPURLResponse {
+    // Verify the status code is 200
+    guard let response = urlResponse as? HTTPURLResponse else {
+      Logging.default
+        .error(
+          "[GoogleGenerativeAI] Response wasn't an HTTP response, internal error \(urlResponse)"
+        )
+      throw NSError(
+        domain: "com.google.generative-ai",
+        code: -1,
+        userInfo: [NSLocalizedDescriptionKey: "Response was not an HTTP response."]
+      )
+    }
+
+    return response
+  }
+
+  private func jsonData(jsonText: String) throws -> Data {
+    guard let data = jsonText.data(using: .utf8) else {
+      let error = NSError(
+        domain: "com.google.generative-ai",
+        code: -1,
+        userInfo: [NSLocalizedDescriptionKey: "Could not parse response as UTF8."]
+      )
+      throw error
+    }
+
+    return data
+  }
+
+  private func parseError(responseBody: String) -> Error {
+    do {
+      let data = try jsonData(jsonText: responseBody)
+      return parseError(responseData: data)
+    } catch {
+      return error
+    }
+  }
+
+  private func parseError(responseData: Data) -> Error {
+    do {
+      return try JSONDecoder().decode(RPCError.self, from: responseData)
+    } catch {
+      // TODO: Return an error about an unrecognized error payload with the response body
+      return error
+    }
+  }
+
+  private func parseResponse<T: Decodable>(_ type: T.Type, from data: Data) throws -> T {
+    do {
+      return try JSONDecoder().decode(type, from: data)
+    } catch {
+      if let json = String(data: data, encoding: .utf8) {
+        Logging.network.error("[GoogleGenerativeAI] JSON response: \(json)")
+      }
+      Logging.default.error("[GoogleGenerativeAI] Error decoding server JSON: \(error)")
+      throw error
+    }
+  }
+
+  #if DEBUG
+    private func cURLCommand(from request: URLRequest) -> String {
+      var returnValue = "curl "
+      if let allHeaders = request.allHTTPHeaderFields {
+        for (key, value) in allHeaders {
+          returnValue += "-H '\(key): \(value)' "
+        }
+      }
+
+      guard let url = request.url else { return "" }
+      returnValue += "'\(url.absoluteString)' "
+
+      guard let body = request.httpBody,
+            let jsonStr = String(bytes: body, encoding: .utf8) else { return "" }
+      let escapedJSON = jsonStr.replacingOccurrences(of: "'", with: "'\\''")
+      returnValue += "-d '\(escapedJSON)'"
+
+      return returnValue
+    }
+
+    private func printCURLCommand(from request: URLRequest) {
+      let command = cURLCommand(from: request)
+      Logging.verbose.debug("""
+      [GoogleGenerativeAI] Creating request with the equivalent cURL command:
+      ----- cURL command -----
+      \(command, privacy: .private)
+      ------------------------
+      """)
+    }
+  #endif // DEBUG
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
new file mode 100644
index 00000000000..4146b60d989
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -0,0 +1,26 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+import Foundation
+
+#if !os(macOS) && !os(iOS)
+  #warning("Only iOS, macOS, and Catalyst targets are currently fully supported.")
+#endif
+
+/// Constants associated with the GenerativeAISwift SDK
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum GenerativeAISwift {
+  /// String value of the SDK version
+  public static let version = "0.4.8"
+  static let baseURL = "https://generativelanguage.googleapis.com"
+}
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
new file mode 100644
index 00000000000..3617cd3a251
--- /dev/null
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -0,0 +1,298 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A type that represents a remote multimodal model (like Gemini), with the ability to generate
+/// content based on various input types.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public final class GenerativeModel {
+  // The prefix for a model resource in the Gemini API.
+  private static let modelResourcePrefix = "models/"
+
+  /// The resource name of the model in the backend; has the format "models/model-name".
+  let modelResourceName: String
+
+  /// The backing service responsible for sending and receiving model requests to the backend.
+  let generativeAIService: GenerativeAIService
+
+  /// Configuration parameters used for the MultiModalModel.
+  let generationConfig: GenerationConfig?
+
+  /// The safety settings to be used for prompts.
+  let safetySettings: [SafetySetting]?
+
+  /// Configuration parameters for sending requests to the backend.
+  let requestOptions: RequestOptions
+
+  /// Initializes a new remote model with the given parameters.
+  ///
+  /// - Parameters:
+  ///   - name: The name of the model to use, e.g., `"gemini-1.0-pro"`; see
+  ///     [Gemini models](https://ai.google.dev/models/gemini) for a list of supported model names.
+  ///   - apiKey: The API key for your project.
+  ///   - generationConfig: The content generation parameters your model should use.
+  ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - requestOptions Configuration parameters for sending requests to the backend.
+  public convenience init(name: String,
+                          apiKey: String,
+                          generationConfig: GenerationConfig? = nil,
+                          safetySettings: [SafetySetting]? = nil,
+                          requestOptions: RequestOptions = RequestOptions()) {
+    self.init(
+      name: name,
+      apiKey: apiKey,
+      generationConfig: generationConfig,
+      safetySettings: safetySettings,
+      requestOptions: requestOptions,
+      urlSession: .shared
+    )
+  }
+
+  /// The designated initializer for this class.
+  init(name: String,
+       apiKey: String,
+       generationConfig: GenerationConfig? = nil,
+       safetySettings: [SafetySetting]? = nil,
+       requestOptions: RequestOptions = RequestOptions(),
+       urlSession: URLSession) {
+    modelResourceName = GenerativeModel.modelResourceName(name: name)
+    generativeAIService = GenerativeAIService(apiKey: apiKey, urlSession: urlSession)
+    self.generationConfig = generationConfig
+    self.safetySettings = safetySettings
+    self.requestOptions = requestOptions
+
+    Logging.default.info("""
+    [GoogleGenerativeAI] Model \(
+      name,
+      privacy: .public
+    ) initialized. To enable additional logging, add \
+    `\(Logging.enableArgumentKey, privacy: .public)` as a launch argument in Xcode.
+    """)
+    Logging.verbose.debug("[GoogleGenerativeAI] Verbose logging enabled.")
+  }
+
+  /// Generates content from String and/or image inputs, given to the model as a prompt, that are
+  /// representable as one or more ``ModelContent/Part``s.
+  ///
+  /// Since ``ModelContent/Part``s do not specify a role, this method is intended for generating
+  /// content from
+  /// [zero-shot](https://developers.google.com/machine-learning/glossary/generative#zero-shot-prompting)
+  /// or "direct" prompts. For
+  /// [few-shot](https://developers.google.com/machine-learning/glossary/generative#few-shot-prompting)
+  /// prompts, see ``generateContent(_:)-58rm0``.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt (see
+  /// ``ThrowingPartsRepresentable``
+  /// for conforming types).
+  /// - Returns: The content generated by the model.
+  /// - Throws: A ``GenerateContentError`` if the request failed.
+  public func generateContent(_ parts: any ThrowingPartsRepresentable...)
+    async throws -> GenerateContentResponse {
+    return try await generateContent([ModelContent(parts: parts)])
+  }
+
+  /// Generates new content from input content given to the model as a prompt.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt.
+  /// - Returns: The generated content response from the model.
+  /// - Throws: A ``GenerateContentError`` if the request failed.
+  public func generateContent(_ content: @autoclosure () throws -> [ModelContent]) async throws
+    -> GenerateContentResponse {
+    let response: GenerateContentResponse
+    do {
+      let generateContentRequest = try GenerateContentRequest(model: modelResourceName,
+                                                              contents: content(),
+                                                              generationConfig: generationConfig,
+                                                              safetySettings: safetySettings,
+                                                              isStreaming: false,
+                                                              options: requestOptions)
+      response = try await generativeAIService.loadRequest(request: generateContentRequest)
+    } catch {
+      if let imageError = error as? ImageConversionError {
+        throw GenerateContentError.promptImageContentError(underlying: imageError)
+      }
+      throw GenerativeModel.generateContentError(from: error)
+    }
+
+    // Check the prompt feedback to see if the prompt was blocked.
+    if response.promptFeedback?.blockReason != nil {
+      throw GenerateContentError.promptBlocked(response: response)
+    }
+
+    // Check to see if an error should be thrown for stop reason.
+    if let reason = response.candidates.first?.finishReason, reason != .stop {
+      throw GenerateContentError.responseStoppedEarly(reason: reason, response: response)
+    }
+
+    return response
+  }
+
+  /// Generates content from String and/or image inputs, given to the model as a prompt, that are
+  /// representable as one or more ``ModelContent/Part``s.
+  ///
+  /// Since ``ModelContent/Part``s do not specify a role, this method is intended for generating
+  /// content from
+  /// [zero-shot](https://developers.google.com/machine-learning/glossary/generative#zero-shot-prompting)
+  /// or "direct" prompts. For
+  /// [few-shot](https://developers.google.com/machine-learning/glossary/generative#few-shot-prompting)
+  /// prompts, see ``generateContent(_:)-58rm0``.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt (see
+  /// ``ThrowingPartsRepresentable``
+  /// for conforming types).
+  /// - Returns: A stream wrapping content generated by the model or a ``GenerateContentError``
+  ///     error if an error occurred.
+  @available(macOS 12.0, *)
+  public func generateContentStream(_ parts: any ThrowingPartsRepresentable...)
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    return try generateContentStream([ModelContent(parts: parts)])
+  }
+
+  /// Generates new content from input content given to the model as a prompt.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt.
+  /// - Returns: A stream wrapping content generated by the model or a ``GenerateContentError``
+  ///     error if an error occurred.
+  @available(macOS 12.0, *)
+  public func generateContentStream(_ content: @autoclosure () throws -> [ModelContent])
+    -> AsyncThrowingStream<GenerateContentResponse, Error> {
+    let evaluatedContent: [ModelContent]
+    do {
+      evaluatedContent = try content()
+    } catch let underlying {
+      return AsyncThrowingStream { continuation in
+        let error: Error
+        if let contentError = underlying as? ImageConversionError {
+          error = GenerateContentError.promptImageContentError(underlying: contentError)
+        } else {
+          error = GenerateContentError.internalError(underlying: underlying)
+        }
+        continuation.finish(throwing: error)
+      }
+    }
+
+    let generateContentRequest = GenerateContentRequest(model: modelResourceName,
+                                                        contents: evaluatedContent,
+                                                        generationConfig: generationConfig,
+                                                        safetySettings: safetySettings,
+                                                        isStreaming: true,
+                                                        options: requestOptions)
+
+    var responseIterator = generativeAIService.loadRequestStream(request: generateContentRequest)
+      .makeAsyncIterator()
+    return AsyncThrowingStream {
+      let response: GenerateContentResponse?
+      do {
+        response = try await responseIterator.next()
+      } catch {
+        throw GenerativeModel.generateContentError(from: error)
+      }
+
+      // The responseIterator will return `nil` when it's done.
+      guard let response = response else {
+        // This is the end of the stream! Signal it by sending `nil`.
+        return nil
+      }
+
+      // Check the prompt feedback to see if the prompt was blocked.
+      if response.promptFeedback?.blockReason != nil {
+        throw GenerateContentError.promptBlocked(response: response)
+      }
+
+      // If the stream ended early unexpectedly, throw an error.
+      if let finishReason = response.candidates.first?.finishReason, finishReason != .stop {
+        throw GenerateContentError.responseStoppedEarly(reason: finishReason, response: response)
+      } else {
+        // Response was valid content, pass it along and continue.
+        return response
+      }
+    }
+  }
+
+  /// Creates a new chat conversation using this model with the provided history.
+  public func startChat(history: [ModelContent] = []) -> Chat {
+    return Chat(model: self, history: history)
+  }
+
+  /// Runs the model's tokenizer on String and/or image inputs that are representable as one or more
+  /// ``ModelContent/Part``s.
+  ///
+  /// Since ``ModelContent/Part``s do not specify a role, this method is intended for tokenizing
+  /// [zero-shot](https://developers.google.com/machine-learning/glossary/generative#zero-shot-prompting)
+  /// or "direct" prompts. For
+  /// [few-shot](https://developers.google.com/machine-learning/glossary/generative#few-shot-prompting)
+  /// input, see ``countTokens(_:)-9spwl``.
+  ///
+  /// - Parameter content: The input(s) given to the model as a prompt (see
+  /// ``ThrowingPartsRepresentable``
+  /// for conforming types).
+  /// - Returns: The results of running the model's tokenizer on the input; contains
+  /// ``CountTokensResponse/totalTokens``.
+  /// - Throws: A ``CountTokensError`` if the tokenization request failed.
+  public func countTokens(_ parts: any ThrowingPartsRepresentable...) async throws
+    -> CountTokensResponse {
+    return try await countTokens([ModelContent(parts: parts)])
+  }
+
+  /// Runs the model's tokenizer on the input content and returns the token count.
+  ///
+  /// - Parameter content: The input given to the model as a prompt.
+  /// - Returns: The results of running the model's tokenizer on the input; contains
+  /// ``CountTokensResponse/totalTokens``.
+  /// - Throws: A ``CountTokensError`` if the tokenization request failed or the input content was
+  /// invalid.
+  public func countTokens(_ content: @autoclosure () throws -> [ModelContent]) async throws
+    -> CountTokensResponse {
+    do {
+      let countTokensRequest = try CountTokensRequest(
+        model: modelResourceName,
+        contents: content(),
+        options: requestOptions
+      )
+      return try await generativeAIService.loadRequest(request: countTokensRequest)
+    } catch {
+      throw CountTokensError.internalError(underlying: error)
+    }
+  }
+
+  /// Returns a model resource name of the form "models/model-name" based on `name`.
+  private static func modelResourceName(name: String) -> String {
+    if name.contains("/") {
+      return name
+    } else {
+      return modelResourcePrefix + name
+    }
+  }
+
+  /// Returns a `GenerateContentError` (for public consumption) from an internal error.
+  ///
+  /// If `error` is already a `GenerateContentError` the error is returned unchanged.
+  private static func generateContentError(from error: Error) -> GenerateContentError {
+    if let error = error as? GenerateContentError {
+      return error
+    } else if let error = error as? RPCError, error.isInvalidAPIKeyError() {
+      return GenerateContentError.invalidAPIKey(message: error.message)
+    } else if let error = error as? RPCError, error.isUnsupportedUserLocationError() {
+      return GenerateContentError.unsupportedUserLocation
+    }
+    return GenerateContentError.internalError(underlying: error)
+  }
+}
+
+/// See ``GenerativeModel/countTokens(_:)-9spwl``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public enum CountTokensError: Error {
+  case internalError(underlying: Error)
+}
diff --git a/FirebaseVertexAI/Sources/Logging.swift b/FirebaseVertexAI/Sources/Logging.swift
new file mode 100644
index 00000000000..458c34ed18d
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Logging.swift
@@ -0,0 +1,56 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+import OSLog
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+struct Logging {
+  /// Subsystem that should be used for all Loggers.
+  static let subsystem = "com.google.generative-ai"
+
+  /// Default category used for most loggers, unless specialized.
+  static let defaultCategory = ""
+
+  /// The argument required to enable additional logging.
+  static let enableArgumentKey = "-GoogleGenerativeAIDebugLogEnabled"
+
+  // No initializer available.
+  @available(*, unavailable)
+  private init() {}
+
+  /// The default logger that is visible for all users. Note: we shouldn't be using anything lower
+  /// than `.notice`.
+  static var `default` = Logger(subsystem: subsystem, category: defaultCategory)
+
+  /// A non default
+  static var network: Logger = {
+    if ProcessInfo.processInfo.arguments.contains(enableArgumentKey) {
+      return Logger(subsystem: subsystem, category: "NetworkResponse")
+    } else {
+      // Return a valid logger that's using `OSLog.disabled` as the logger, hiding everything.
+      return Logger(.disabled)
+    }
+  }()
+
+  ///
+  static var verbose: Logger = {
+    if ProcessInfo.processInfo.arguments.contains(enableArgumentKey) {
+      return Logger(subsystem: subsystem, category: defaultCategory)
+    } else {
+      // Return a valid logger that's using `OSLog.disabled` as the logger, hiding everything.
+      return Logger(.disabled)
+    }
+  }()
+}
diff --git a/FirebaseVertexAI/Sources/ModelContent.swift b/FirebaseVertexAI/Sources/ModelContent.swift
new file mode 100644
index 00000000000..44648c57852
--- /dev/null
+++ b/FirebaseVertexAI/Sources/ModelContent.swift
@@ -0,0 +1,141 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A type describing data in media formats interpretable by an AI model. Each generative AI
+/// request or response contains an `Array` of ``ModelContent``s, and each ``ModelContent`` value
+/// may comprise multiple heterogeneous ``ModelContent/Part``s.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct ModelContent: Codable, Equatable {
+  /// A discrete piece of data in a media format intepretable by an AI model. Within a single value
+  /// of ``Part``, different data types may not mix.
+  public enum Part: Codable, Equatable {
+    enum CodingKeys: String, CodingKey {
+      case text
+      case inlineData
+    }
+
+    enum InlineDataKeys: String, CodingKey {
+      case mimeType = "mime_type"
+      case bytes = "data"
+    }
+
+    /// Text value.
+    case text(String)
+
+    /// Data with a specified media type. Not all media types may be supported by the AI model.
+    case data(mimetype: String, Data)
+
+    // MARK: Convenience Initializers
+
+    /// Convenience function for populating a Part with JPEG data.
+    public static func jpeg(_ data: Data) -> Self {
+      return .data(mimetype: "image/jpeg", data)
+    }
+
+    /// Convenience function for populating a Part with PNG data.
+    public static func png(_ data: Data) -> Self {
+      return .data(mimetype: "image/png", data)
+    }
+
+    // MARK: Codable Conformance
+
+    public func encode(to encoder: Encoder) throws {
+      var container = encoder.container(keyedBy: ModelContent.Part.CodingKeys.self)
+      switch self {
+      case let .text(a0):
+        try container.encode(a0, forKey: .text)
+      case let .data(mimetype, bytes):
+        var inlineDataContainer = container.nestedContainer(
+          keyedBy: InlineDataKeys.self,
+          forKey: .inlineData
+        )
+        try inlineDataContainer.encode(mimetype, forKey: .mimeType)
+        try inlineDataContainer.encode(bytes, forKey: .bytes)
+      }
+    }
+
+    public init(from decoder: Decoder) throws {
+      let values = try decoder.container(keyedBy: CodingKeys.self)
+      if values.contains(.text) {
+        self = try .text(values.decode(String.self, forKey: .text))
+      } else if values.contains(.inlineData) {
+        let dataContainer = try values.nestedContainer(
+          keyedBy: InlineDataKeys.self,
+          forKey: .inlineData
+        )
+        let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
+        let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
+        self = .data(mimetype: mimetype, bytes)
+      } else {
+        throw DecodingError.dataCorrupted(.init(
+          codingPath: [CodingKeys.text, CodingKeys.inlineData],
+          debugDescription: "Neither text or inline data was found."
+        ))
+      }
+    }
+
+    /// Returns the text contents of this ``Part``, if it contains text.
+    public var text: String? {
+      switch self {
+      case let .text(contents): return contents
+      default: return nil
+      }
+    }
+  }
+
+  /// The role of the entity creating the ``ModelContent``. For user-generated client requests,
+  /// for example, the role is `user`.
+  public let role: String?
+
+  /// The data parts comprising this ``ModelContent`` value.
+  public let parts: [Part]
+
+  /// Creates a new value from any data or `Array` of data interpretable as a
+  /// ``Part``. See ``ThrowingPartsRepresentable`` for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", parts: some ThrowingPartsRepresentable) throws {
+    self.role = role
+    try self.parts = parts.tryPartsValue()
+  }
+
+  /// Creates a new value from any data or `Array` of data interpretable as a
+  /// ``Part``. See ``ThrowingPartsRepresentable`` for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", parts: some PartsRepresentable) {
+    self.role = role
+    self.parts = parts.partsValue
+  }
+
+  /// Creates a new value from a list of ``Part``s.
+  public init(role: String? = "user", parts: [Part]) {
+    self.role = role
+    self.parts = parts
+  }
+
+  /// Creates a new value from any data interpretable as a ``Part``. See
+  /// ``ThrowingPartsRepresentable``
+  /// for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", _ parts: any ThrowingPartsRepresentable...) throws {
+    let content = try parts.flatMap { try $0.tryPartsValue() }
+    self.init(role: role, parts: content)
+  }
+
+  /// Creates a new value from any data interpretable as a ``Part``. See
+  /// ``ThrowingPartsRepresentable``
+  /// for types that can be interpreted as `Part`s.
+  public init(role: String? = "user", _ parts: [PartsRepresentable]) {
+    let content = parts.flatMap { $0.partsValue }
+    self.init(role: role, parts: content)
+  }
+}
diff --git a/FirebaseVertexAI/Sources/PartsRepresentable+Image.swift b/FirebaseVertexAI/Sources/PartsRepresentable+Image.swift
new file mode 100644
index 00000000000..052a004e52f
--- /dev/null
+++ b/FirebaseVertexAI/Sources/PartsRepresentable+Image.swift
@@ -0,0 +1,107 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import UniformTypeIdentifiers
+#if canImport(UIKit)
+  import UIKit // For UIImage extensions.
+#elseif canImport(AppKit)
+  import AppKit // For NSImage extensions.
+#endif
+
+private let imageCompressionQuality: CGFloat = 0.8
+
+/// An enum describing failures that can occur when converting image types to model content data.
+/// For some image types like `CIImage`, creating valid model content requires creating a JPEG
+/// representation of the image that may not yet exist, which may be computationally expensive.
+public enum ImageConversionError: Error {
+  /// The image (the receiver of the call `toModelContentParts()`) was invalid.
+  case invalidUnderlyingImage
+
+  /// A valid image destination could not be allocated.
+  case couldNotAllocateDestination
+
+  /// JPEG image data conversion failed, accompanied by the original image, which may be an
+  /// instance of `NSImageRep`, `UIImage`, `CGImage`, or `CIImage`.
+  case couldNotConvertToJPEG(Any)
+}
+
+#if canImport(UIKit)
+  /// Enables images to be representable as ``ThrowingPartsRepresentable``.
+  @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+  extension UIImage: ThrowingPartsRepresentable {
+    public func tryPartsValue() throws -> [ModelContent.Part] {
+      guard let data = jpegData(compressionQuality: imageCompressionQuality) else {
+        throw ImageConversionError.couldNotConvertToJPEG(self)
+      }
+      return [ModelContent.Part.data(mimetype: "image/jpeg", data)]
+    }
+  }
+
+#elseif canImport(AppKit)
+  /// Enables images to be representable as ``ThrowingPartsRepresentable``.
+  @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+  extension NSImage: ThrowingPartsRepresentable {
+    public func tryPartsValue() throws -> [ModelContent.Part] {
+      guard let cgImage = cgImage(forProposedRect: nil, context: nil, hints: nil) else {
+        throw ImageConversionError.invalidUnderlyingImage
+      }
+      let bmp = NSBitmapImageRep(cgImage: cgImage)
+      guard let data = bmp.representation(using: .jpeg, properties: [.compressionFactor: 0.8])
+      else {
+        throw ImageConversionError.couldNotConvertToJPEG(bmp)
+      }
+      return [ModelContent.Part.data(mimetype: "image/jpeg", data)]
+    }
+  }
+#endif
+
+/// Enables `CGImages` to be representable as model content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CGImage: ThrowingPartsRepresentable {
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    let output = NSMutableData()
+    guard let imageDestination = CGImageDestinationCreateWithData(
+      output, UTType.jpeg.identifier as CFString, 1, nil
+    ) else {
+      throw ImageConversionError.couldNotAllocateDestination
+    }
+    CGImageDestinationAddImage(imageDestination, self, nil)
+    CGImageDestinationSetProperties(imageDestination, [
+      kCGImageDestinationLossyCompressionQuality: imageCompressionQuality,
+    ] as CFDictionary)
+    if CGImageDestinationFinalize(imageDestination) {
+      return [.data(mimetype: "image/jpeg", output as Data)]
+    }
+    throw ImageConversionError.couldNotConvertToJPEG(self)
+  }
+}
+
+/// Enables `CIImages` to be representable as model content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CIImage: ThrowingPartsRepresentable {
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    let context = CIContext()
+    let jpegData = (colorSpace ?? CGColorSpace(name: CGColorSpace.sRGB))
+      .flatMap {
+        // The docs specify kCGImageDestinationLossyCompressionQuality as a supported option, but
+        // Swift's type system does not allow this.
+        // [kCGImageDestinationLossyCompressionQuality: imageCompressionQuality]
+        context.jpegRepresentation(of: self, colorSpace: $0, options: [:])
+      }
+    if let jpegData = jpegData {
+      return [.data(mimetype: "image/jpeg", jpegData)]
+    }
+    throw ImageConversionError.couldNotConvertToJPEG(self)
+  }
+}
diff --git a/FirebaseVertexAI/Sources/PartsRepresentable.swift b/FirebaseVertexAI/Sources/PartsRepresentable.swift
new file mode 100644
index 00000000000..05ba0d9dabc
--- /dev/null
+++ b/FirebaseVertexAI/Sources/PartsRepresentable.swift
@@ -0,0 +1,66 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A protocol describing any data that could be serialized to model-interpretable input data,
+/// where the serialization process might fail with an error.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public protocol ThrowingPartsRepresentable {
+  func tryPartsValue() throws -> [ModelContent.Part]
+}
+
+/// A protocol describing any data that could be serialized to model-interpretable input data,
+/// where the serialization process cannot fail with an error. For a failable conversion, see
+/// ``ThrowingPartsRepresentable``
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public protocol PartsRepresentable: ThrowingPartsRepresentable {
+  var partsValue: [ModelContent.Part] { get }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public extension PartsRepresentable {
+  func tryPartsValue() throws -> [ModelContent.Part] {
+    return partsValue
+  }
+}
+
+/// Enables a ``ModelContent.Part`` to be passed in as ``ThrowingPartsRepresentable``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension ModelContent.Part: ThrowingPartsRepresentable {
+  public typealias ErrorType = Never
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    return [self]
+  }
+}
+
+/// Enable an `Array` of ``ThrowingPartsRepresentable`` values to be passed in as a single
+/// ``ThrowingPartsRepresentable``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension [ThrowingPartsRepresentable]: ThrowingPartsRepresentable {
+  public func tryPartsValue() throws -> [ModelContent.Part] {
+    return try compactMap { element in
+      try element.tryPartsValue()
+    }
+    .flatMap { $0 }
+  }
+}
+
+/// Enables a `String` to be passed in as ``ThrowingPartsRepresentable``.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension String: PartsRepresentable {
+  public var partsValue: [ModelContent.Part] {
+    return [.text(self)]
+  }
+}
diff --git a/FirebaseVertexAI/Sources/Safety.swift b/FirebaseVertexAI/Sources/Safety.swift
new file mode 100644
index 00000000000..d59511f2476
--- /dev/null
+++ b/FirebaseVertexAI/Sources/Safety.swift
@@ -0,0 +1,182 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A type defining potentially harmful media categories and their model-assigned ratings. A value
+/// of this type may be assigned to a category for every model-generated response, not just
+/// responses that exceed a certain threshold.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct SafetyRating: Decodable, Equatable, Hashable {
+  /// The category describing the potential harm a piece of content may pose. See
+  /// ``SafetySetting/HarmCategory`` for a list of possible values.
+  public let category: SafetySetting.HarmCategory
+
+  /// The model-generated probability that a given piece of content falls under the harm category
+  /// described in ``category``. This does not
+  /// indiciate the severity of harm for a piece of content. See ``HarmProbability`` for a list of
+  /// possible values.
+  public let probability: HarmProbability
+
+  /// Initializes a new `SafetyRating` instance with the given category and probability.
+  /// Use this initializer for SwiftUI previews or tests.
+  public init(category: SafetySetting.HarmCategory, probability: HarmProbability) {
+    self.category = category
+    self.probability = probability
+  }
+
+  /// The probability that a given model output falls under a harmful content category. This does
+  /// not indicate the severity of harm for a piece of content.
+  public enum HarmProbability: String, Codable {
+    /// Unknown. A new server value that isn't recognized by the SDK.
+    case unknown = "UNKNOWN"
+
+    /// The probability was not specified in the server response.
+    case unspecified = "HARM_PROBABILITY_UNSPECIFIED"
+
+    /// The probability is zero or close to zero. For benign content, the probability across all
+    /// categories will be this value.
+    case negligible = "NEGLIGIBLE"
+
+    /// The probability is small but non-zero.
+    case low = "LOW"
+
+    /// The probability is moderate.
+    case medium = "MEDIUM"
+
+    /// The probability is high. The content described is very likely harmful.
+    case high = "HIGH"
+
+    /// Initializes a new `SafetyRating` from a decoder.
+    /// Not for external use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedProbability = HarmProbability(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized HarmProbability with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedProbability
+    }
+  }
+}
+
+/// Safety feedback for an entire request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct SafetyFeedback: Decodable {
+  /// Safety rating evaluated from content.
+  public let rating: SafetyRating
+
+  /// Safety settings applied to the request.
+  public let setting: SafetySetting
+
+  /// Internal initializer.
+  init(rating: SafetyRating, setting: SafetySetting) {
+    self.rating = rating
+    self.setting = setting
+  }
+}
+
+/// A type used to specify a threshold for harmful content, beyond which the model will return a
+/// fallback response instead of generated content.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct SafetySetting: Codable {
+  /// A type describing safety attributes, which include harmful categories and topics that can
+  /// be considered sensitive.
+  public enum HarmCategory: String, Codable {
+    /// Unknown. A new server value that isn't recognized by the SDK.
+    case unknown = "HARM_CATEGORY_UNKNOWN"
+
+    /// Unspecified by the server.
+    case unspecified = "HARM_CATEGORY_UNSPECIFIED"
+
+    /// Harassment content.
+    case harassment = "HARM_CATEGORY_HARASSMENT"
+
+    /// Negative or harmful comments targeting identity and/or protected attributes.
+    case hateSpeech = "HARM_CATEGORY_HATE_SPEECH"
+
+    /// Contains references to sexual acts or other lewd content.
+    case sexuallyExplicit = "HARM_CATEGORY_SEXUALLY_EXPLICIT"
+
+    /// Promotes or enables access to harmful goods, services, or activities.
+    case dangerousContent = "HARM_CATEGORY_DANGEROUS_CONTENT"
+
+    /// Do not explicitly use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedCategory = HarmCategory(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized HarmCategory with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedCategory
+    }
+  }
+
+  /// Block at and beyond a specified ``SafetyRating/HarmProbability``.
+  public enum BlockThreshold: String, Codable {
+    /// Unknown. A new server value that isn't recognized by the SDK.
+    case unknown = "UNKNOWN"
+
+    /// Threshold is unspecified.
+    case unspecified = "HARM_BLOCK_THRESHOLD_UNSPECIFIED"
+
+    // Content with `.negligible` will be allowed.
+    case blockLowAndAbove = "BLOCK_LOW_AND_ABOVE"
+
+    /// Content with `.negligible` and `.low` will be allowed.
+    case blockMediumAndAbove = "BLOCK_MEDIUM_AND_ABOVE"
+
+    /// Content with `.negligible`, `.low`, and `.medium` will be allowed.
+    case blockOnlyHigh = "BLOCK_ONLY_HIGH"
+
+    /// All content will be allowed.
+    case blockNone = "BLOCK_NONE"
+
+    /// Do not explicitly use. Initializer required for Decodable conformance.
+    public init(from decoder: Decoder) throws {
+      let value = try decoder.singleValueContainer().decode(String.self)
+      guard let decodedThreshold = BlockThreshold(rawValue: value) else {
+        Logging.default
+          .error("[GoogleGenerativeAI] Unrecognized BlockThreshold with value \"\(value)\".")
+        self = .unknown
+        return
+      }
+
+      self = decodedThreshold
+    }
+  }
+
+  enum CodingKeys: String, CodingKey {
+    case harmCategory = "category"
+    case threshold
+  }
+
+  /// The category this safety setting should be applied to.
+  public let harmCategory: HarmCategory
+
+  /// The threshold describing what content should be blocked.
+  public let threshold: BlockThreshold
+
+  /// Initializes a new safety setting with the given category and threshold.
+  public init(harmCategory: HarmCategory, threshold: BlockThreshold) {
+    self.harmCategory = harmCategory
+    self.threshold = threshold
+  }
+}
diff --git a/Package.swift b/Package.swift
index d6eaca85e57..3d32a7fb9a5 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1377,7 +1377,11 @@ let package = Package(
         "FirebaseCoreExtension",
         .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources"
+      path: "FirebaseVertexAI/Sources",
+      sources: [
+        "VertexAI.swift",
+        "VertexAIComponent.swift",
+      ]
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From e5efafde5d30e672c945e00472993895a60e997d Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 18 Mar 2024 21:16:22 +0000
Subject: [PATCH 066/124] Use forked `google-generative-ai` in Vertex AI
 samples (#12574)

---
 .../ChatSample/Views/ErrorDetailsView.swift   | 24 -----------
 FirebaseVertexAI/Sources/Errors.swift         | 12 ------
 .../Sources/GenerateContentError.swift        | 13 ------
 .../Sources/GenerativeAIRequest.swift         |  4 +-
 .../Sources/GenerativeAIService.swift         | 28 ++++++++++---
 .../Sources/GenerativeAISwift.swift           |  5 +--
 .../Sources/GenerativeModel.swift             | 35 +++++-----------
 FirebaseVertexAI/Sources/VertexAI.swift       | 41 +++++--------------
 Package.swift                                 | 11 +----
 9 files changed, 48 insertions(+), 125 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
index 399777ad1e8..7b3b78753db 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -142,22 +142,6 @@ struct ErrorDetailsView: View {
             SafetyRatingsSection(ratings: ratings)
           }
 
-        case GenerateContentError.invalidAPIKey:
-          Section("Error Type") {
-            Text("Invalid API Key")
-          }
-
-          Section("Details") {
-            SubtitleFormRow(title: "Error description", value: error.localizedDescription)
-            SubtitleMarkdownFormRow(
-              title: "Help",
-              value: """
-              The `API_KEY` provided in the `GoogleService-Info.plist` file is invalid. Download a
-              new copy of the file from the [Firebase Console](https://console.firebase.google.com).
-              """
-            )
-          }
-
         default:
           Section("Error Type") {
             Text("Some other error")
@@ -222,11 +206,3 @@ struct ErrorDetailsView: View {
 
   return ErrorDetailsView(error: error)
 }
-
-#Preview("Invalid API Key") {
-  ErrorDetailsView(error: GenerateContentError.invalidAPIKey)
-}
-
-#Preview("Unsupported User Location") {
-  ErrorDetailsView(error: GenerateContentError.unsupportedUserLocation)
-}
diff --git a/FirebaseVertexAI/Sources/Errors.swift b/FirebaseVertexAI/Sources/Errors.swift
index 35e7e6b813b..0b1b8848411 100644
--- a/FirebaseVertexAI/Sources/Errors.swift
+++ b/FirebaseVertexAI/Sources/Errors.swift
@@ -30,14 +30,6 @@ struct RPCError: Error {
     self.status = status
     self.details = details
   }
-
-  func isInvalidAPIKeyError() -> Bool {
-    return errorInfo?.reason == "API_KEY_INVALID"
-  }
-
-  func isUnsupportedUserLocationError() -> Bool {
-    return message == RPCErrorMessage.unsupportedUserLocation.rawValue
-  }
 }
 
 extension RPCError: Decodable {
@@ -179,10 +171,6 @@ enum RPCStatus: String, Decodable {
   case dataLoss = "DATA_LOSS"
 }
 
-enum RPCErrorMessage: String {
-  case unsupportedUserLocation = "User location is not supported for the API use."
-}
-
 enum InvalidCandidateError: Error {
   case emptyContent(underlyingError: Error)
   case malformedContent(underlyingError: Error)
diff --git a/FirebaseVertexAI/Sources/GenerateContentError.swift b/FirebaseVertexAI/Sources/GenerateContentError.swift
index 055928e5f1b..31501a7744c 100644
--- a/FirebaseVertexAI/Sources/GenerateContentError.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentError.swift
@@ -28,17 +28,4 @@ public enum GenerateContentError: Error {
 
   /// A response didn't fully complete. See the `FinishReason` for more information.
   case responseStoppedEarly(reason: FinishReason, response: GenerateContentResponse)
-
-  /// The provided API key is invalid.
-  case invalidAPIKey(message: String)
-
-  /// The user's location (region) is not supported by the API.
-  ///
-  /// See the Google documentation for a
-  /// [list of regions](https://ai.google.dev/available_regions#available_regions)
-  /// (countries and territories) where the API is available.
-  ///
-  /// - Important: The API is only available in
-  /// [specific regions](https://ai.google.dev/available_regions#available_regions).
-  case unsupportedUserLocation
 }
diff --git a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
index 2bf94503b1a..21f35b1b728 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
@@ -38,8 +38,8 @@ public struct RequestOptions {
   /// - Parameters:
   ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
   ///   for a `URLRequest`.
-  ///   - apiVersion The API version to use in requests to the backend; defaults to "v1".
-  public init(timeout: TimeInterval? = nil, apiVersion: String = "v1") {
+  ///   - apiVersion The API version to use in requests to the backend; defaults to "v2beta".
+  public init(timeout: TimeInterval? = nil, apiVersion: String = "v2beta") {
     self.timeout = timeout
     self.apiVersion = apiVersion
   }
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index 8d904738927..e1df1010c06 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -12,6 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
+import FirebaseCore
 import Foundation
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -19,15 +21,18 @@ struct GenerativeAIService {
   /// Gives permission to talk to the backend.
   private let apiKey: String
 
+  private let appCheck: AppCheckInterop?
+
   private let urlSession: URLSession
 
-  init(apiKey: String, urlSession: URLSession) {
+  init(apiKey: String, appCheck: AppCheckInterop?, urlSession: URLSession) {
     self.apiKey = apiKey
+    self.appCheck = appCheck
     self.urlSession = urlSession
   }
 
   func loadRequest<T: GenerativeAIRequest>(request: T) async throws -> T.Response {
-    let urlRequest = try urlRequest(request: request)
+    let urlRequest = try await urlRequest(request: request)
 
     #if DEBUG
       printCURLCommand(from: urlRequest)
@@ -59,7 +64,7 @@ struct GenerativeAIService {
       Task {
         let urlRequest: URLRequest
         do {
-          urlRequest = try self.urlRequest(request: request)
+          urlRequest = try await self.urlRequest(request: request)
         } catch {
           continuation.finish(throwing: error)
           return
@@ -146,13 +151,24 @@ struct GenerativeAIService {
 
   // MARK: - Private Helpers
 
-  private func urlRequest<T: GenerativeAIRequest>(request: T) throws -> URLRequest {
+  private func urlRequest<T: GenerativeAIRequest>(request: T) async throws -> URLRequest {
     var urlRequest = URLRequest(url: request.url)
     urlRequest.httpMethod = "POST"
     urlRequest.setValue(apiKey, forHTTPHeaderField: "x-goog-api-key")
-    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version)",
-                        forHTTPHeaderField: "x-goog-api-client")
+    // TODO: Determine the right client header to use.
+    //    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version))",
+    //                        forHTTPHeaderField: "x-goog-api-client")
     urlRequest.setValue("application/json", forHTTPHeaderField: "Content-Type")
+
+    if let appCheck {
+      let tokenResult = await appCheck.getToken(forcingRefresh: false)
+      urlRequest.setValue(tokenResult.token, forHTTPHeaderField: "X-Firebase-AppCheck")
+      if let error = tokenResult.error {
+        Logging.default
+          .debug("[GoogleGenerativeAI] Failed to fetch AppCheck token. Error: \(error)")
+      }
+    }
+
     let encoder = JSONEncoder()
     encoder.keyEncodingStrategy = .convertToSnakeCase
     urlRequest.httpBody = try encoder.encode(request)
diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
index 4146b60d989..0f0af2bc3e2 100644
--- a/FirebaseVertexAI/Sources/GenerativeAISwift.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -11,6 +11,7 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
+
 import Foundation
 
 #if !os(macOS) && !os(iOS)
@@ -20,7 +21,5 @@ import Foundation
 /// Constants associated with the GenerativeAISwift SDK
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public enum GenerativeAISwift {
-  /// String value of the SDK version
-  public static let version = "0.4.8"
-  static let baseURL = "https://generativelanguage.googleapis.com"
+  static let baseURL = "https://staging-firebaseml.sandbox.googleapis.com"
 }
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index 3617cd3a251..aad2b3e99c0 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -12,6 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
 import Foundation
 
 /// A type that represents a remote multimodal model (like Gemini), with the ability to generate
@@ -44,31 +45,21 @@ public final class GenerativeModel {
   ///   - apiKey: The API key for your project.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
-  ///   - requestOptions Configuration parameters for sending requests to the backend.
-  public convenience init(name: String,
-                          apiKey: String,
-                          generationConfig: GenerationConfig? = nil,
-                          safetySettings: [SafetySetting]? = nil,
-                          requestOptions: RequestOptions = RequestOptions()) {
-    self.init(
-      name: name,
-      apiKey: apiKey,
-      generationConfig: generationConfig,
-      safetySettings: safetySettings,
-      requestOptions: requestOptions,
-      urlSession: .shared
-    )
-  }
-
-  /// The designated initializer for this class.
+  ///   - requestOptions: Configuration parameters for sending requests to the backend.
+  ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
        apiKey: String,
        generationConfig: GenerationConfig? = nil,
        safetySettings: [SafetySetting]? = nil,
-       requestOptions: RequestOptions = RequestOptions(),
-       urlSession: URLSession) {
+       requestOptions: RequestOptions,
+       appCheck: AppCheckInterop?,
+       urlSession: URLSession = .shared) {
     modelResourceName = GenerativeModel.modelResourceName(name: name)
-    generativeAIService = GenerativeAIService(apiKey: apiKey, urlSession: urlSession)
+    generativeAIService = GenerativeAIService(
+      apiKey: apiKey,
+      appCheck: appCheck,
+      urlSession: urlSession
+    )
     self.generationConfig = generationConfig
     self.safetySettings = safetySettings
     self.requestOptions = requestOptions
@@ -282,10 +273,6 @@ public final class GenerativeModel {
   private static func generateContentError(from error: Error) -> GenerateContentError {
     if let error = error as? GenerateContentError {
       return error
-    } else if let error = error as? RPCError, error.isInvalidAPIKeyError() {
-      return GenerateContentError.invalidAPIKey(message: error.message)
-    } else if let error = error as? RPCError, error.isUnsupportedUserLocationError() {
-      return GenerateContentError.unsupportedUserLocation
     }
     return GenerateContentError.internalError(underlying: error)
   }
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 8fb0f9e8c68..0f79e1be99c 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -12,13 +12,9 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import Foundation
-
 import FirebaseAppCheckInterop
 import FirebaseCore
-
-// Exports the GoogleGenerativeAI module to users of the SDK.
-@_exported import GoogleGenerativeAI
+import Foundation
 
 // Avoids exposing internal FirebaseCore APIs to Swift users.
 @_implementationOnly import FirebaseCoreExtension
@@ -31,8 +27,9 @@ open class VertexAI: NSObject {
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
   ///
   /// This instance is configured with the default `FirebaseApp`.
-  public static func generativeModel(modelName: String, location: String) -> GoogleGenerativeAI
-    .GenerativeModel {
+  ///
+  /// TODO: Add RequestOptions to public API.
+  public static func generativeModel(modelName: String, location: String) -> GenerativeModel {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
@@ -40,8 +37,10 @@ open class VertexAI: NSObject {
   }
 
   /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  ///
+  /// TODO: Add RequestOptions to public API.
   public static func generativeModel(app: FirebaseApp, modelName: String,
-                                     location: String) -> GoogleGenerativeAI.GenerativeModel {
+                                     location: String) -> GenerativeModel {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
@@ -64,18 +63,15 @@ open class VertexAI: NSObject {
   private let modelResouceName: String
 
   lazy var model: GenerativeModel = {
-    let options = RequestOptions(
-      apiVersion: "v2beta",
-      endpoint: "staging-firebaseml.sandbox.googleapis.com",
-      hooks: [addAppCheckHeader]
-    )
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
     }
     return GenerativeModel(
       name: modelResouceName,
       apiKey: apiKey,
-      requestOptions: options
+      // TODO: Add RequestOptions to public API.
+      requestOptions: RequestOptions(),
+      appCheck: appCheck
     )
   }()
 
@@ -104,21 +100,4 @@ open class VertexAI: NSObject {
 
     return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
   }
-
-  // MARK: Request Hooks
-
-  /// Adds an App Check token to the provided request if App Check is included in the app.
-  ///
-  /// This demonstrates how an App Check token can be added to requests; it is currently ignored by
-  /// the backend.
-  ///
-  /// - Parameter request: The `URLRequest` to modify by adding an App Check token header.
-  func addAppCheckHeader(request: inout URLRequest) async {
-    guard let appCheck else {
-      return
-    }
-
-    let tokenResult = await appCheck.getToken(forcingRefresh: false)
-    request.addValue(tokenResult.token, forHTTPHeaderField: "X-Firebase-AppCheck")
-  }
 }
diff --git a/Package.swift b/Package.swift
index 3d32a7fb9a5..bcbbb20422d 100644
--- a/Package.swift
+++ b/Package.swift
@@ -193,10 +193,6 @@ let package = Package(
       "100.0.0" ..< "101.0.0"
     ),
     .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
-    .package(
-      url: "https://github.com/google/generative-ai-swift.git",
-      revision: "c9f2c4913bc65aa267815962c7e91358c2d8463f"
-    ),
   ],
   targets: [
     .target(
@@ -1375,13 +1371,8 @@ let package = Package(
         "FirebaseAppCheckInterop",
         "FirebaseCore",
         "FirebaseCoreExtension",
-        .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
       ],
-      path: "FirebaseVertexAI/Sources",
-      sources: [
-        "VertexAI.swift",
-        "VertexAIComponent.swift",
-      ]
+      path: "FirebaseVertexAI/Sources"
     ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,

From 38c5724554115f2c5d3951b6900f8591bf90e735 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 18 Mar 2024 22:30:24 +0000
Subject: [PATCH 067/124] Add CI workflow for Vertex AI (#12577)

---
 .github/workflows/vertexai.yml | 37 ++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 .github/workflows/vertexai.yml

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
new file mode 100644
index 00000000000..63cfd774e43
--- /dev/null
+++ b/.github/workflows/vertexai.yml
@@ -0,0 +1,37 @@
+name: vertexai
+
+on:
+  pull_request:
+    paths:
+        - 'FirebaseVertexAI**'
+        - '.github/workflows/vertexai.yml'
+        - 'Gemfile*'
+  schedule:
+    # Run every day at 11pm (PST) - cron uses UTC times
+    - cron:  '0 7 * * *'
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  spm:
+    strategy:
+      matrix:
+        target: [iOS, macOS, catalyst]
+        os: [macos-13]
+        include:
+          - os: macos-13
+            xcode: Xcode_15.2
+    runs-on: ${{ matrix.os }}
+    env:
+      FIREBASECI_USE_LATEST_GOOGLEAPPMEASUREMENT: 1
+    steps:
+    - uses: actions/checkout@v4
+    - name: Xcode
+      run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
+    - name: Initialize xcodebuild
+      run: xcodebuild -list
+    # TODO: Add unit tests and switch from `spmbuildonly` to `spm`.
+    - name: Build
+      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAI ${{ matrix.target }} spmbuildonly

From ed1c3a67ace2afae66476435b544d52b3c661c0e Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 19 Mar 2024 15:44:55 +0000
Subject: [PATCH 068/124] Fork `generative-ai-swift` tests for Vertex AI
 (#12584)

---
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |  67 ++
 .../failure-model-not-found.json              |  13 +
 .../success-total-tokens.json                 |   3 +
 .../streaming-failure-empty-content.txt       |   1 +
 .../streaming-failure-error-mid-stream.txt    |  17 +
 ...streaming-failure-finish-reason-safety.txt |   2 +
 .../streaming-failure-invalid-json.txt        |   1 +
 .../streaming-failure-malformed-content.txt   |   1 +
 ...treaming-failure-prompt-blocked-safety.txt |   2 +
 ...treaming-failure-recitation-no-content.txt |   6 +
 .../streaming-failure-unknown-finish-enum.txt |  11 +
 .../streaming-success-basic-reply-long.txt    |  19 +
 .../streaming-success-basic-reply-short.txt   |   2 +
 .../streaming-success-citations.txt           |  13 +
 .../streaming-success-citations2.txt          |   5 +
 .../streaming-success-unknown-safety-enum.txt |   2 +
 .../unary-failure-api-key.json                |  21 +
 .../unary-failure-empty-content.json          |  28 +
 ...e-finish-reason-recitation-no-content.json |  46 +
 ...ilure-finish-reason-safety-no-content.json |  46 +
 .../unary-failure-finish-reason-safety.json   |  53 +
 .../unary-failure-image-rejected.json         |  13 +
 .../unary-failure-invalid-response.json       |  14 +
 .../unary-failure-malformed-content.json      |  30 +
 .../unary-failure-prompt-blocked-safety.json  |  23 +
 ...ry-failure-unknown-enum-finish-reason.json |  53 +
 ...y-failure-unknown-enum-prompt-blocked.json |  23 +
 .../unary-failure-unknown-model.json          |  13 +
 ...ary-failure-unsupported-user-location.json |  13 +
 .../unary-success-basic-reply-long.json       |  54 +
 .../unary-success-basic-reply-short.json      |  54 +
 .../unary-success-citations.json              |  64 ++
 .../unary-success-missing-safety-ratings.json |  16 +
 .../unary-success-quote-reply.json            |  54 +
 ...y-success-unknown-enum-safety-ratings.json |  46 +
 .../Tests/Unit/GenerativeModelTests.swift     | 930 ++++++++++++++++++
 .../Tests/Unit/GoogleAITests.swift            | 175 ++++
 .../Tests/Unit/MockURLProtocol.swift          |  62 ++
 .../Tests/Unit/PartsRepresentableTests.swift  | 138 +++
 39 files changed, 2134 insertions(+)
 create mode 100644 FirebaseVertexAI/Tests/Unit/ChatTests.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift

diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
new file mode 100644
index 00000000000..668e3704122
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -0,0 +1,67 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+@testable import GoogleGenerativeAI
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+final class ChatTests: XCTestCase {
+  var urlSession: URLSession!
+
+  override func setUp() {
+    let configuration = URLSessionConfiguration.default
+    configuration.protocolClasses = [MockURLProtocol.self]
+    urlSession = URLSession(configuration: configuration)
+  }
+
+  override func tearDown() {
+    MockURLProtocol.requestHandler = nil
+  }
+
+  func testMergingText() async throws {
+    let fileURL = try XCTUnwrap(Bundle.module.url(
+      forResource: "streaming-success-basic-reply-long",
+      withExtension: "txt"
+    ))
+
+    MockURLProtocol.requestHandler = { request in
+      let response = HTTPURLResponse(
+        url: request.url!,
+        statusCode: 200,
+        httpVersion: nil,
+        headerFields: nil
+      )!
+      return (response, fileURL.lines)
+    }
+
+    let model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+    let chat = Chat(model: model, history: [])
+    let input = "Test input"
+    let stream = chat.sendMessageStream(input)
+
+    // Ensure the values are parsed correctly
+    for try await value in stream {
+      XCTAssertNotNil(value.text)
+    }
+
+    XCTAssertEqual(chat.history.count, 2)
+    XCTAssertEqual(chat.history[0].parts[0].text, input)
+
+    let finalText = "1 2 3 4 5 6 7 8 9 10"
+    let assembledExpectation = ModelContent(role: "model", parts: finalText)
+    XCTAssertEqual(chat.history[0].parts[0].text, input)
+    XCTAssertEqual(chat.history[1], assembledExpectation)
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json
new file mode 100644
index 00000000000..50fcb725667
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/failure-model-not-found.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 404,
+    "message": "models/test-model-name is not found for API version v1beta, or is not supported for countTokens. Call ListModels to see the list of available models and their supported methods.",
+    "status": "NOT_FOUND",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::not_found: models/test-model-name is not found for API version v1beta, or is not supported for countTokens. Call ListModels to see the list of available models and their supported methods. [google.rpc.error_details_ext] { message: \"models/test-model-name is not found for API version v1beta, or is not supported for countTokens. Call ListModels to see the list of available models and their supported methods.\" }"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
new file mode 100644
index 00000000000..0bccd9e7c3d
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
@@ -0,0 +1,3 @@
+{
+  "totalTokens": 6
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt
new file mode 100644
index 00000000000..5762b515325
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-empty-content.txt
@@ -0,0 +1 @@
+data: {"candidates": [{"content": {},"index": 0}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt
new file mode 100644
index 00000000000..aeb4eb09389
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-error-mid-stream.txt
@@ -0,0 +1,17 @@
+data: {"candidates": [{"content": {"parts": [{"text": "First "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Second "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+{
+  "error": {
+    "code": 499,
+    "message": "The operation was cancelled.",
+    "status": "CANCELLED",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::cancelled: "
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt
new file mode 100644
index 00000000000..b73c75cf505
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-finish-reason-safety.txt
@@ -0,0 +1,2 @@
+data: {"candidates": [{"content": {"parts": [{"text": "No"}]},"finishReason": "SAFETY","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "HIGH"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt
new file mode 100644
index 00000000000..3aea78474cb
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-invalid-json.txt
@@ -0,0 +1 @@
+data: {"this": [{"is": {"not": [{"a": "valid"}]}, "response": {}}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt
new file mode 100644
index 00000000000..273b6cdc5ca
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-malformed-content.txt
@@ -0,0 +1 @@
+data: {"candidates": [{"content": {"missing-parts": true},"index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_TOXICITY","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_VIOLENCE","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_TOXICITY","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_VIOLENCE","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt
new file mode 100644
index 00000000000..58c914af08e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-prompt-blocked-safety.txt
@@ -0,0 +1,2 @@
+data: {"promptFeedback": {"blockReason": "SAFETY","safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "HIGH"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt
new file mode 100644
index 00000000000..60ec91d6dfc
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-recitation-no-content.txt
@@ -0,0 +1,6 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Some information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Some information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "LOW"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 30,"endIndex": 179,"uri": "https://www.example.com/some-citation","license": ""}]}}]}
+
+data: {"candidates": [{"finishReason": "RECITATION","index": 0}]}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt
new file mode 100644
index 00000000000..6194abd8957
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-failure-unknown-finish-enum.txt
@@ -0,0 +1,11 @@
+data: {"candidates": [{"content": {"parts": [{"text": "**Cats:**\n\n- **Physical Characteristics:**\n  - Size: Cats come"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": " in a wide range of sizes, from small breeds like the Singapura to large breeds like the Maine Coon.\n  - Fur: Cats have soft, furry coats"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": " that can vary in length and texture depending on the breed.\n  - Eyes: Cats have large, expressive eyes that can be various colors, including green, blue, yellow, and hazel.\n  - Ears: Cats have pointed, erect ears that are sensitive to sound.\n  - Tail: Cats have long"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": ", flexible tails that they use for balance and communication.\n\n- **Behavior and Personality:**\n  - Independent: Cats are often described as independent animals that enjoy spending time alone.\n  - Affectionate: Despite their independent nature, cats can be very affectionate and form strong bonds with their owners.\n  - Playful: Cats are naturally playful and enjoy engaging in activities such as chasing toys, climbing, and pouncing.\n  - Curious: Cats are curious creatures that love to explore their surroundings.\n  - Vocal: Cats communicate through a variety of vocalizations, including meows, purrs, hisses, and grow"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "ls.\n\n- **Health and Care:**\n  - Diet: Cats are obligate carnivores, meaning they require animal-based protein for optimal health.\n  - Grooming: Cats spend a significant amount of time grooming themselves to keep their fur clean and free of mats.\n  - Exercise: Cats need regular exercise to stay healthy and active. This can be achieved through play sessions or access to outdoor space.\n  - Veterinary Care: Regular veterinary checkups are essential for maintaining a cat's health and detecting any potential health issues early on.\n\n**Dogs:**\n\n- **Physical Characteristics:**\n  - Size: Dogs come in a wide range of sizes, from small breeds like the Chihuahua to giant breeds like the Great Dane.\n  - Fur: Dogs have fur coats that can vary in length, texture, and color depending on the breed.\n  - Eyes: Dogs have expressive eyes that can be various colors, including brown, blue, green, and hazel.\n  - Ears: Dogs have floppy or erect ears that are sensitive to sound.\n  - Tail: Dogs have long, wagging tails that they use for communication and expressing emotions.\n\n- **Behavior and Personality:**\n  - Loyal: Dogs are known for their loyalty and"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": " devotion to their owners.\n  - Friendly: Dogs are generally friendly and outgoing animals that enjoy interacting with people and other animals.\n  - Playful: Dogs are playful and energetic creatures that love to engage in activities such as fetching, running, and playing with toys.\n  - Trainable: Dogs are highly trainable and can learn a variety of commands and tricks.\n  - Vocal: Dogs communicate through a variety of vocalizations, including barking, howling, whining, and growling.\n\n- **Health and Care:**\n  - Diet: Dogs are omnivores and can eat a variety of foods, including meat, vegetables, and grains.\n  - Grooming: Dogs require regular grooming to keep their fur clean and free of mats. The frequency of grooming depends on the breed and coat type.\n  - Exercise: Dogs need regular exercise to stay healthy and active. The amount of exercise required varies depending on the breed and age of the dog.\n  - Veterinary Care: Regular veterinary checkups are essential for maintaining a dog's health and detecting any potential health issues early on."}]},"finishReason": "FAKE_ENUM","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT_NEW_ENUM","probability": "NEGLIGIBLE_UNKNOWN_ENUM"}]}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
new file mode 100644
index 00000000000..bca95140490
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
@@ -0,0 +1,19 @@
+data: {"candidates": [{"content": {"parts": [{"text": "1 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": "2 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "3 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "4 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "5 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "6 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "7 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "8 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "9 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "10"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
new file mode 100644
index 00000000000..a7f5476954e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
@@ -0,0 +1,2 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
new file mode 100644
index 00000000000..4f50be7d32f
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
@@ -0,0 +1,13 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Some information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
+data: {"candidates": [{"content": {"parts": [{"text": " More information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": ", Even more information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": " Some information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-1","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-2","license": ""}]}}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "More information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-3","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-4","license": ""}]}}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Even more information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-5","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-6","license": ""}]}}]}
+
+data: {"candidates": [{"content": {"parts": [{"text": "Physics (YouTube Channel)"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.google.com","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.google.com","license": ""}]}}]}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
new file mode 100644
index 00000000000..665e993ad75
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
@@ -0,0 +1,5 @@
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "In the context of the science fiction comedy novel \"The Hitchhiker's Guide to the Galaxy\" by Douglas Adams, the answer to the \"Ultimate Question of"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.055720285,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.062674366},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03904829,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03339982},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12220858,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0540987},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05781161,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03149938}]}]}
+
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Life, the Universe, and Everything\" is given as 42. However, in real life, there is no universally accepted meaning of life. The meaning"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.05910154,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.038321976},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.049589027,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.019271139},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03986249},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.098946586,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04177388}],"citationMetadata": {"citations": [{"startIndex": 52,"endIndex": 181,"uri": "https://imcsteakhouse.com.au/faqs"}]}}]}
+
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " of life is a philosophical question that has been pondered by humans for centuries, and there is no single answer that is widely agreed upon."}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.044764314,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.028870905},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04240383,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.012576347},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.10650458,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034880884},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.11085559,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03191915}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 91,"totalTokenCount": 100}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt
new file mode 100644
index 00000000000..8f7a8ff3c1c
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-unknown-safety-enum.txt
@@ -0,0 +1,2 @@
+data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_SOMETHING_NEW","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
new file mode 100644
index 00000000000..ecf6f6b53fa
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -0,0 +1,21 @@
+{
+  "error": {
+    "code": 400,
+    "message": "API key not valid. Please pass a valid API key.",
+    "status": "INVALID_ARGUMENT",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.ErrorInfo",
+        "reason": "API_KEY_INVALID",
+        "domain": "googleapis.com",
+        "metadata": {
+          "service": "generativelanguage.googleapis.com"
+        }
+      },
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "Invalid API key: AIzv00G7VmUCUeC-5OglO3hcXM"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json
new file mode 100644
index 00000000000..4e1889660f2
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-empty-content.json
@@ -0,0 +1,28 @@
+{
+  "candidates": [
+    {
+      "content": {},
+      "index": 0
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
new file mode 100644
index 00000000000..0d99cd71d51
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
@@ -0,0 +1,46 @@
+{
+  "candidates": [
+    {
+      "finishReason": "RECITATION",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "HIGH"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json
new file mode 100644
index 00000000000..03958d4afc1
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety-no-content.json
@@ -0,0 +1,46 @@
+{
+    "candidates": [
+        {
+            "finishReason": "SAFETY",
+            "index": 0,
+            "safetyRatings": [
+                {
+                    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_HATE_SPEECH",
+                    "probability": "HIGH"
+                },
+                {
+                    "category": "HARM_CATEGORY_HARASSMENT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                    "probability": "NEGLIGIBLE"
+                }
+            ]
+        }
+    ],
+    "promptFeedback": {
+        "safetyRatings": [
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "probability": "NEGLIGIBLE"
+            }
+        ]
+    }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json
new file mode 100644
index 00000000000..3249c739cf8
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-safety.json
@@ -0,0 +1,53 @@
+{
+    "candidates": [
+        {
+            "content": {
+                "parts": [
+                    {
+                        "text": "No"
+                    }
+                ]
+            },
+            "finishReason": "SAFETY",
+            "index": 0,
+            "safetyRatings": [
+                {
+                    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_HATE_SPEECH",
+                    "probability": "HIGH"
+                },
+                {
+                    "category": "HARM_CATEGORY_HARASSMENT",
+                    "probability": "NEGLIGIBLE"
+                },
+                {
+                    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                    "probability": "NEGLIGIBLE"
+                }
+            ]
+        }
+    ],
+    "promptFeedback": {
+        "safetyRatings": [
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "probability": "NEGLIGIBLE"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "probability": "NEGLIGIBLE"
+            }
+        ]
+    }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json
new file mode 100644
index 00000000000..9dacdc71e7a
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-image-rejected.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 400,
+    "message": "Request contains an invalid argument.",
+    "status": "INVALID_ARGUMENT",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::invalid_argument: invalid status photos.thumbnailer.Status.Code::5: Source image 0 too short"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json
new file mode 100644
index 00000000000..49d05e1840b
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-invalid-response.json
@@ -0,0 +1,14 @@
+{
+  "this": [
+    {
+      "is": {
+        "not": [
+          {
+            "a": "valid"
+          }
+        ]
+      },
+      "response": {}
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json
new file mode 100644
index 00000000000..737f2e08548
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-malformed-content.json
@@ -0,0 +1,30 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "invalid-field": true
+      },
+      "index": 0
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json
new file mode 100644
index 00000000000..9d2abbb23d6
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-prompt-blocked-safety.json
@@ -0,0 +1,23 @@
+{
+  "promptFeedback": {
+    "blockReason": "SAFETY",
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "HIGH"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json
new file mode 100644
index 00000000000..1adbc70cae4
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-finish-reason.json
@@ -0,0 +1,53 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Some text"
+          }
+        ]
+      },
+      "finishReason": "FAKE_NEW_FINISH_REASON",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json
new file mode 100644
index 00000000000..a00cbb72cfc
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-enum-prompt-blocked.json
@@ -0,0 +1,23 @@
+{
+  "promptFeedback": {
+    "blockReason": "FAKE_NEW_BLOCK_REASON",
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json
new file mode 100644
index 00000000000..60b3f55c978
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unknown-model.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 404,
+    "message": "models/unknown is not found for API version v1, or is not supported for GenerateContent. Call ListModels to see the list of available models and their supported methods.",
+    "status": "NOT_FOUND",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::not_found: models/unknown is not found for API version v1, or is not supported for GenerateContent. Call ListModels to see the list of available models and their supported methods. [google.rpc.error_details_ext] { message: \"models/unknown is not found for API version v1, or is not supported for GenerateContent. Call ListModels to see the list of available models and their supported methods.\" }"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
new file mode 100644
index 00000000000..c4c2ace4e20
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
@@ -0,0 +1,13 @@
+{
+  "error": {
+    "code": 400,
+    "message": "User location is not supported for the API use.",
+    "status": "FAILED_PRECONDITION",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.DebugInfo",
+        "detail": "[ORIGINAL ERROR] generic::failed_precondition: User location is not supported for the API use. [google.rpc.error_details_ext] { message: \"User location is not supported for the API use.\" }"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
new file mode 100644
index 00000000000..59b84de92fe
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
@@ -0,0 +1,54 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "You can ask me a wide range of questions on various topics. Here are some examples:\n\n1. **General Knowledge:**\n   - What is the capital of France?\n   - Who painted the Mona Lisa?\n   - What is the largest ocean in the world?\n\n2. **Science and Technology:**\n   - How does a computer work?\n   - What is the difference between a virus and a bacteria?\n   - What are the latest advancements in artificial intelligence?\n\n3. **History and Culture:**\n   - Who was the first president of the United States?\n   - What is the significance of the Great Wall of China?\n   - What are some of the most famous works of Shakespeare?\n\n4. **Current Events:**\n   - What is the latest news about the COVID-19 pandemic?\n   - Who is the current president of Ukraine?\n   - What are the major issues being discussed in the upcoming election?\n\n5. **Personal Questions:**\n   - What are your hobbies?\n   - What is your favorite book or movie?\n   - What are your thoughts on the future of technology?\n\n6. **Fun and Games:**\n   - Can you tell me a joke?\n   - What is the answer to this riddle: \"I have keys but no locks. I have space but no room. You can enter, but can't go outside.\" (Answer: a keyboard)\n   - Let's play a game of 20 questions.\n\n7. **Hypothetical Questions:**\n   - What would you do if you won the lottery?\n   - What would happen if time travel were possible?\n   - What is the meaning of life?\n\n8. **Philosophical Questions:**\n   - What is the nature of reality?\n   - Does free will exist?\n   - What is the difference between right and wrong?\n\n9. **Creative Questions:**\n   - Write a poem about a sunset.\n   - Design a logo for a new company.\n   - Compose a song about your favorite season.\n\n10. **Technical Questions:**\n    - How can I improve the performance of my computer?\n    - What is the best way to troubleshoot a network issue?\n    - How do I create a website using HTML and CSS?\n\nRemember, I am still under development and may not be able to answer all questions perfectly. However, I will do my best to provide you with accurate and informative responses."
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
new file mode 100644
index 00000000000..40a9a6da58e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
@@ -0,0 +1,54 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Mountain View, California, United States"
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
new file mode 100644
index 00000000000..ac99fcf0944
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
@@ -0,0 +1,64 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Some information cited from an external source"
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ],
+      "citationMetadata": {
+        "citationSources": [
+          {
+            "startIndex": 574,
+            "endIndex": 705,
+            "uri": "https://www.example.com/some-citation",
+            "license": ""
+          }
+        ]
+      }
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json
new file mode 100644
index 00000000000..645d3e6425e
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-missing-safety-ratings.json
@@ -0,0 +1,16 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "This is the generated content."
+          }
+        ]
+      },
+      "index": 0
+    }
+  ],
+  "promptFeedback": {
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json
new file mode 100644
index 00000000000..e5bf4f0b497
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-quote-reply.json
@@ -0,0 +1,54 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Google's mission is to \"organize the world's information and make it universally accessible and useful.\""
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "NEGLIGIBLE"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "NEGLIGIBLE"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "NEGLIGIBLE"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json
new file mode 100644
index 00000000000..3c47a21322c
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-unknown-enum-safety-ratings.json
@@ -0,0 +1,46 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "text": "Some text"
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0,
+      "safetyRatings": [
+        {
+          "category": "HARM_CATEGORY_HARASSMENT",
+          "probability": "MEDIUM"
+        },
+        {
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "FAKE_NEW_HARM_PROBABILITY"
+        },
+        {
+          "category": "FAKE_NEW_HARM_CATEGORY",
+          "probability": "HIGH"
+        }
+      ]
+    }
+  ],
+  "promptFeedback": {
+    "safetyRatings": [
+      {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "probability": "MEDIUM"
+      },
+      {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "probability": "FAKE_NEW_HARM_PROBABILITY"
+      },
+      {
+        "category": "FAKE_NEW_HARM_CATEGORY",
+        "probability": "HIGH"
+      }
+    ]
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
new file mode 100644
index 00000000000..b835a9ea05f
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -0,0 +1,930 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+@testable import GoogleGenerativeAI
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+final class GenerativeModelTests: XCTestCase {
+  let testPrompt = "What sorts of questions can I ask you?"
+  let safetyRatingsNegligible: [SafetyRating] = [
+    .init(category: .sexuallyExplicit, probability: .negligible),
+    .init(category: .hateSpeech, probability: .negligible),
+    .init(category: .harassment, probability: .negligible),
+    .init(category: .dangerousContent, probability: .negligible),
+  ]
+
+  var urlSession: URLSession!
+  var model: GenerativeModel!
+
+  override func setUp() async throws {
+    let configuration = URLSessionConfiguration.default
+    configuration.protocolClasses = [MockURLProtocol.self]
+    urlSession = try XCTUnwrap(URLSession(configuration: configuration))
+    model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+  }
+
+  override func tearDown() {
+    MockURLProtocol.requestHandler = nil
+  }
+
+  // MARK: - Generate Content
+
+  func testGenerateContent_success_basicReplyLong() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-long",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    let finishReason = try XCTUnwrap(candidate.finishReason)
+    XCTAssertEqual(finishReason, .stop)
+    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    let partText = try XCTUnwrap(part.text)
+    XCTAssertTrue(partText.hasPrefix("You can ask me a wide range of questions"))
+    XCTAssertEqual(response.text, partText)
+    let promptFeedback = try XCTUnwrap(response.promptFeedback)
+    XCTAssertNil(promptFeedback.blockReason)
+    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+  }
+
+  func testGenerateContent_success_basicReplyShort() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    let finishReason = try XCTUnwrap(candidate.finishReason)
+    XCTAssertEqual(finishReason, .stop)
+    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    XCTAssertEqual(part.text, "Mountain View, California, United States")
+    XCTAssertEqual(response.text, part.text)
+    let promptFeedback = try XCTUnwrap(response.promptFeedback)
+    XCTAssertNil(promptFeedback.blockReason)
+    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+  }
+
+  func testGenerateContent_success_citations() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-citations",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    XCTAssertEqual(response.text, "Some information cited from an external source")
+    let citationMetadata = try XCTUnwrap(candidate.citationMetadata)
+    XCTAssertEqual(citationMetadata.citationSources.count, 1)
+    let citationSource = try XCTUnwrap(citationMetadata.citationSources.first)
+    XCTAssertEqual(citationSource.uri, "https://www.example.com/some-citation")
+    XCTAssertEqual(citationSource.startIndex, 574)
+    XCTAssertEqual(citationSource.endIndex, 705)
+    XCTAssertEqual(citationSource.license, "")
+  }
+
+  func testGenerateContent_success_quoteReply() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-quote-reply",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    let finishReason = try XCTUnwrap(candidate.finishReason)
+    XCTAssertEqual(finishReason, .stop)
+    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    let partText = try XCTUnwrap(part.text)
+    XCTAssertTrue(partText.hasPrefix("Google"))
+    XCTAssertEqual(response.text, part.text)
+    let promptFeedback = try XCTUnwrap(response.promptFeedback)
+    XCTAssertNil(promptFeedback.blockReason)
+    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+  }
+
+  func testGenerateContent_success_unknownEnum_safetyRatings() async throws {
+    let expectedSafetyRatings = [
+      SafetyRating(category: .harassment, probability: .medium),
+      SafetyRating(category: .dangerousContent, probability: .unknown),
+      SafetyRating(category: .unknown, probability: .high),
+    ]
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-unknown-enum-safety-ratings",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.text, "Some text")
+    XCTAssertEqual(response.candidates.first?.safetyRatings, expectedSafetyRatings)
+    XCTAssertEqual(response.promptFeedback?.safetyRatings, expectedSafetyRatings)
+  }
+
+  func testGenerateContent_success_prefixedModelName() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json"
+      )
+    let model = GenerativeModel(
+      // Model name is prefixed with "models/".
+      name: "models/test-model",
+      apiKey: "API_KEY",
+      urlSession: urlSession
+    )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_failure_invalidAPIKey() async throws {
+    let expectedStatusCode = 400
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-api-key",
+        withExtension: "json",
+        statusCode: expectedStatusCode
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError.invalidAPIKey(message) {
+      XCTAssertEqual(message, "API key not valid. Please pass a valid API key.")
+    } catch {
+      XCTFail("Should throw GenerateContentError.invalidAPIKey; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_emptyContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-empty-content",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError
+      .internalError(underlying: invalidCandidateError as InvalidCandidateError) {
+      guard case let .emptyContent(decodingError) = invalidCandidateError else {
+        XCTFail("Not an InvalidCandidateError.emptyContent error: \(invalidCandidateError)")
+        return
+      }
+      _ = try XCTUnwrap(decodingError as? DecodingError,
+                        "Not a DecodingError: \(decodingError)")
+    } catch {
+      XCTFail("Should throw GenerateContentError.internalError; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_finishReasonSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-finish-reason-safety",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.responseStoppedEarly(reason, response) {
+      XCTAssertEqual(reason, .safety)
+      XCTAssertEqual(response.text, "No")
+    } catch {
+      XCTFail("Should throw a responseStoppedEarly")
+    }
+  }
+
+  func testGenerateContent_failure_finishReasonSafety_noContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-finish-reason-safety-no-content",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.responseStoppedEarly(reason, response) {
+      XCTAssertEqual(reason, .safety)
+      XCTAssertNil(response.text)
+    } catch {
+      XCTFail("Should throw a responseStoppedEarly")
+    }
+  }
+
+  func testGenerateContent_failure_imageRejected() async throws {
+    let expectedStatusCode = 400
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-image-rejected",
+        withExtension: "json",
+        statusCode: 400
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError.internalError(underlying: rpcError as RPCError) {
+      XCTAssertEqual(rpcError.status, .invalidArgument)
+      XCTAssertEqual(rpcError.httpResponseCode, expectedStatusCode)
+      XCTAssertEqual(rpcError.message, "Request contains an invalid argument.")
+    } catch {
+      XCTFail("Should throw GenerateContentError.internalError; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_promptBlockedSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-prompt-blocked-safety",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.promptBlocked(response) {
+      XCTAssertNil(response.text)
+    } catch {
+      XCTFail("Should throw a promptBlocked")
+    }
+  }
+
+  func testGenerateContent_failure_unknownEnum_finishReason() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unknown-enum-finish-reason",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.responseStoppedEarly(reason, response) {
+      XCTAssertEqual(reason, .unknown)
+      XCTAssertEqual(response.text, "Some text")
+    } catch {
+      XCTFail("Should throw a responseStoppedEarly")
+    }
+  }
+
+  func testGenerateContent_failure_unknownEnum_promptBlocked() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unknown-enum-prompt-blocked",
+        withExtension: "json"
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw")
+    } catch let GenerateContentError.promptBlocked(response) {
+      let promptFeedback = try XCTUnwrap(response.promptFeedback)
+      XCTAssertEqual(promptFeedback.blockReason, .unknown)
+    } catch {
+      XCTFail("Should throw a promptBlocked")
+    }
+  }
+
+  func testGenerateContent_failure_unknownModel() async throws {
+    let expectedStatusCode = 404
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unknown-model",
+        withExtension: "json",
+        statusCode: 404
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
+    } catch let GenerateContentError.internalError(underlying: rpcError as RPCError) {
+      XCTAssertEqual(rpcError.status, .notFound)
+      XCTAssertEqual(rpcError.httpResponseCode, expectedStatusCode)
+      XCTAssertTrue(rpcError.message.hasPrefix("models/unknown is not found"))
+    } catch {
+      XCTFail("Should throw GenerateContentError.internalError; error thrown: \(error)")
+    }
+  }
+
+  func testGenerateContent_failure_unsupportedUserLocation() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unsupported-user-location",
+        withExtension: "json",
+        statusCode: 400
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw GenerateContentError.unsupportedUserLocation; no error thrown.")
+    } catch GenerateContentError.unsupportedUserLocation {
+      return
+    }
+
+    XCTFail("Expected an unsupported user location error.")
+  }
+
+  func testGenerateContent_failure_nonHTTPResponse() async throws {
+    MockURLProtocol.requestHandler = try nonHTTPRequestHandler()
+
+    var responseError: Error?
+    var content: GenerateContentResponse?
+    do {
+      content = try await model.generateContent(testPrompt)
+    } catch {
+      responseError = error
+    }
+
+    XCTAssertNil(content)
+    XCTAssertNotNil(responseError)
+    let generateContentError = try XCTUnwrap(responseError as? GenerateContentError)
+    guard case let .internalError(underlyingError) = generateContentError else {
+      XCTFail("Not an internal error: \(generateContentError)")
+      return
+    }
+    XCTAssertEqual(underlyingError.localizedDescription, "Response was not an HTTP response.")
+  }
+
+  func testGenerateContent_failure_invalidResponse() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "unary-failure-invalid-response",
+      withExtension: "json"
+    )
+
+    var responseError: Error?
+    var content: GenerateContentResponse?
+    do {
+      content = try await model.generateContent(testPrompt)
+    } catch {
+      responseError = error
+    }
+
+    XCTAssertNil(content)
+    XCTAssertNotNil(responseError)
+    let generateContentError = try XCTUnwrap(responseError as? GenerateContentError)
+    guard case let .internalError(underlyingError) = generateContentError else {
+      XCTFail("Not an internal error: \(generateContentError)")
+      return
+    }
+    let decodingError = try XCTUnwrap(underlyingError as? DecodingError)
+    guard case let .dataCorrupted(context) = decodingError else {
+      XCTFail("Not a data corrupted error: \(decodingError)")
+      return
+    }
+    XCTAssert(context.debugDescription.hasPrefix("Failed to decode GenerateContentResponse"))
+  }
+
+  func testGenerateContent_failure_malformedContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-malformed-content",
+        withExtension: "json"
+      )
+
+    var responseError: Error?
+    var content: GenerateContentResponse?
+    do {
+      content = try await model.generateContent(testPrompt)
+    } catch {
+      responseError = error
+    }
+
+    XCTAssertNil(content)
+    XCTAssertNotNil(responseError)
+    let generateContentError = try XCTUnwrap(responseError as? GenerateContentError)
+    guard case let .internalError(underlyingError) = generateContentError else {
+      XCTFail("Not an internal error: \(generateContentError)")
+      return
+    }
+    let invalidCandidateError = try XCTUnwrap(underlyingError as? InvalidCandidateError)
+    guard case let .malformedContent(malformedContentUnderlyingError) = invalidCandidateError else {
+      XCTFail("Not a malformed content error: \(invalidCandidateError)")
+      return
+    }
+    _ = try XCTUnwrap(
+      malformedContentUnderlyingError as? DecodingError,
+      "Not a decoding error: \(malformedContentUnderlyingError)"
+    )
+  }
+
+  func testGenerateContentMissingSafetyRatings() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "unary-success-missing-safety-ratings",
+      withExtension: "json"
+    )
+
+    let content = try await model.generateContent(testPrompt)
+    let promptFeedback = try XCTUnwrap(content.promptFeedback)
+    XCTAssertEqual(promptFeedback.safetyRatings.count, 0)
+    XCTAssertEqual(content.text, "This is the generated content.")
+  }
+
+  func testGenerateContent_requestOptions_customTimeout() async throws {
+    let expectedTimeout = 150.0
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        timeout: expectedTimeout
+      )
+    let requestOptions = RequestOptions(timeout: expectedTimeout)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: requestOptions,
+      urlSession: urlSession
+    )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+  }
+
+  // MARK: - Generate Content (Streaming)
+
+  func testGenerateContentStream_failureInvalidAPIKey() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-api-key",
+        withExtension: "json"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("No content is there, this shouldn't happen.")
+      }
+    } catch GenerateContentError.invalidAPIKey {
+      // invalidAPIKey error is as expected, nothing else to check.
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failureEmptyContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-empty-content",
+        withExtension: "txt"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("No content is there, this shouldn't happen.")
+      }
+    } catch GenerateContentError.internalError(_ as InvalidCandidateError) {
+      // Underlying error is as expected, nothing else to check.
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failureFinishReasonSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-finish-reason-safety",
+        withExtension: "txt"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("Content shouldn't be shown, this shouldn't happen.")
+      }
+    } catch let GenerateContentError.responseStoppedEarly(reason, _) {
+      XCTAssertEqual(reason, .safety)
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failurePromptBlockedSafety() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-prompt-blocked-safety",
+        withExtension: "txt"
+      )
+
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await _ in stream {
+        XCTFail("Content shouldn't be shown, this shouldn't happen.")
+      }
+    } catch let GenerateContentError.promptBlocked(response) {
+      XCTAssertEqual(response.promptFeedback?.blockReason, .safety)
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_failureUnknownFinishEnum() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-unknown-finish-enum",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream("Hi")
+    do {
+      for try await content in stream {
+        XCTAssertNotNil(content.text)
+      }
+    } catch let GenerateContentError.responseStoppedEarly(reason, _) {
+      XCTAssertEqual(reason, .unknown)
+      return
+    }
+
+    XCTFail("Should have caught an error.")
+  }
+
+  func testGenerateContentStream_successBasicReplyLong() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-long",
+        withExtension: "txt"
+      )
+
+    var responses = 0
+    let stream = model.generateContentStream("Hi")
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      responses += 1
+    }
+
+    XCTAssertEqual(responses, 10)
+  }
+
+  func testGenerateContentStream_successBasicReplyShort() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt"
+      )
+
+    var responses = 0
+    let stream = model.generateContentStream("Hi")
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      responses += 1
+    }
+
+    XCTAssertEqual(responses, 1)
+  }
+
+  func testGenerateContentStream_successUnknownSafetyEnum() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-unknown-safety-enum",
+        withExtension: "txt"
+      )
+
+    var hadUnknown = false
+    let stream = model.generateContentStream("Hi")
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      if let ratings = content.candidates.first?.safetyRatings,
+         ratings.contains(where: { $0.category == .unknown }) {
+        hadUnknown = true
+      }
+    }
+
+    XCTAssertTrue(hadUnknown)
+  }
+
+  func testGenerateContentStream_successWithCitations() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-citations",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream("Hi")
+    var citations: [Citation] = []
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      let candidate = try XCTUnwrap(content.candidates.first)
+      XCTAssertEqual(candidate.finishReason, .stop)
+      if let sources = candidate.citationMetadata?.citationSources {
+        citations.append(contentsOf: sources)
+      }
+    }
+
+    XCTAssertEqual(citations.count, 8)
+    XCTAssertTrue(citations
+      .contains(where: { $0.startIndex == 574 && $0.endIndex == 705 && !$0.uri.isEmpty }))
+    XCTAssertTrue(citations
+      .contains(where: { $0.startIndex == 899 && $0.endIndex == 1026 && !$0.uri.isEmpty }))
+  }
+
+  func testGenerateContentStream_errorMidStream() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "streaming-failure-error-mid-stream",
+      withExtension: "txt"
+    )
+
+    var responseCount = 0
+    do {
+      let stream = model.generateContentStream("Hi")
+      for try await content in stream {
+        XCTAssertNotNil(content.text)
+        responseCount += 1
+      }
+    } catch let GenerateContentError.internalError(rpcError as RPCError) {
+      XCTAssertEqual(rpcError.httpResponseCode, 499)
+      XCTAssertEqual(rpcError.status, .cancelled)
+
+      // Check the content count is correct.
+      XCTAssertEqual(responseCount, 2)
+      return
+    }
+
+    XCTFail("Expected an internalError with an RPCError.")
+  }
+
+  func testGenerateContentStream_nonHTTPResponse() async throws {
+    MockURLProtocol.requestHandler = try nonHTTPRequestHandler()
+
+    let stream = model.generateContentStream("Hi")
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch let GenerateContentError.internalError(underlying) {
+      XCTAssertEqual(underlying.localizedDescription, "Response was not an HTTP response.")
+      return
+    }
+
+    XCTFail("Expected an internal error.")
+  }
+
+  func testGenerateContentStream_invalidResponse() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-invalid-json",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch let GenerateContentError.internalError(underlying as DecodingError) {
+      guard case let .dataCorrupted(context) = underlying else {
+        XCTFail("Not a data corrupted error: \(underlying)")
+        return
+      }
+      XCTAssert(context.debugDescription.hasPrefix("Failed to decode GenerateContentResponse"))
+      return
+    }
+
+    XCTFail("Expected an internal error.")
+  }
+
+  func testGenerateContentStream_malformedContent() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-failure-malformed-content",
+        withExtension: "txt"
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch let GenerateContentError.internalError(underlyingError as InvalidCandidateError) {
+      guard case let .malformedContent(contentError) = underlyingError else {
+        XCTFail("Not a malformed content error: \(underlyingError)")
+        return
+      }
+
+      XCTAssert(contentError is DecodingError)
+      return
+    }
+
+    XCTFail("Expected an internal decoding error.")
+  }
+
+  func testGenerateContentStream_failure_unsupportedUserLocation() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-failure-unsupported-user-location",
+        withExtension: "json",
+        statusCode: 400
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    do {
+      for try await content in stream {
+        XCTFail("Unexpected content in stream: \(content)")
+      }
+    } catch GenerateContentError.unsupportedUserLocation {
+      return
+    }
+
+    XCTFail("Expected an unsupported user location error.")
+  }
+
+  func testGenerateContentStream_requestOptions_customTimeout() async throws {
+    let expectedTimeout = 150.0
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt",
+        timeout: expectedTimeout
+      )
+    let requestOptions = RequestOptions(timeout: expectedTimeout)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: requestOptions,
+      urlSession: urlSession
+    )
+
+    var responses = 0
+    let stream = model.generateContentStream(testPrompt)
+    for try await content in stream {
+      XCTAssertNotNil(content.text)
+      responses += 1
+    }
+
+    XCTAssertEqual(responses, 1)
+  }
+
+  // MARK: - Count Tokens
+
+  func testCountTokens_succeeds() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "success-total-tokens",
+      withExtension: "json"
+    )
+
+    let response = try await model.countTokens("Why is the sky blue?")
+    XCTAssertEqual(response.totalTokens, 6)
+  }
+
+  func testCountTokens_modelNotFound() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "failure-model-not-found", withExtension: "json",
+      statusCode: 404
+    )
+
+    do {
+      _ = try await model.countTokens("Why is the sky blue?")
+      XCTFail("Request should not have succeeded.")
+    } catch let CountTokensError.internalError(rpcError as RPCError) {
+      XCTAssertEqual(rpcError.httpResponseCode, 404)
+      XCTAssertEqual(rpcError.status, .notFound)
+      XCTAssert(rpcError.message.hasPrefix("models/test-model-name is not found"))
+      return
+    }
+
+    XCTFail("Expected internal RPCError.")
+  }
+
+  func testCountTokens_requestOptions_customTimeout() async throws {
+    let expectedTimeout = 150.0
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "success-total-tokens",
+        withExtension: "json",
+        timeout: expectedTimeout
+      )
+    let requestOptions = RequestOptions(timeout: expectedTimeout)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: requestOptions,
+      urlSession: urlSession
+    )
+
+    let response = try await model.countTokens(testPrompt)
+
+    XCTAssertEqual(response.totalTokens, 6)
+  }
+
+  // MARK: - Model Resource Name
+
+  func testModelResourceName_noPrefix() async throws {
+    let modelName = "my-model"
+    let modelResourceName = "models/\(modelName)"
+
+    model = GenerativeModel(name: modelName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, modelResourceName)
+  }
+
+  func testModelResourceName_modelsPrefix() async throws {
+    let modelResourceName = "models/my-model"
+
+    model = GenerativeModel(name: modelResourceName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, modelResourceName)
+  }
+
+  func testModelResourceName_tunedModelsPrefix() async throws {
+    let tunedModelResourceName = "tunedModels/my-model"
+
+    model = GenerativeModel(name: tunedModelResourceName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
+  }
+
+  // MARK: - Helpers
+
+  private func nonHTTPRequestHandler() throws -> ((URLRequest) -> (
+    URLResponse,
+    AsyncLineSequence<URL.AsyncBytes>?
+  )) {
+    return { request in
+      // This is *not* an HTTPURLResponse
+      let response = URLResponse(
+        url: request.url!,
+        mimeType: nil,
+        expectedContentLength: 0,
+        textEncodingName: nil
+      )
+      return (response, nil)
+    }
+  }
+
+  private func httpRequestHandler(forResource name: String,
+                                  withExtension ext: String,
+                                  statusCode: Int = 200,
+                                  timeout: TimeInterval = URLRequest
+                                    .defaultTimeoutInterval()) throws -> ((URLRequest) throws -> (
+    URLResponse,
+    AsyncLineSequence<URL.AsyncBytes>?
+  )) {
+    let fileURL = try XCTUnwrap(Bundle.module.url(forResource: name, withExtension: ext))
+    return { request in
+      let requestURL = try XCTUnwrap(request.url)
+      XCTAssertEqual(requestURL.path.occurrenceCount(of: "models/"), 1)
+      XCTAssertEqual(request.timeoutInterval, timeout)
+      let response = try XCTUnwrap(HTTPURLResponse(
+        url: requestURL,
+        statusCode: statusCode,
+        httpVersion: nil,
+        headerFields: nil
+      ))
+      return (response, fileURL.lines)
+    }
+  }
+}
+
+private extension String {
+  /// Returns the number of occurrences of `substring` in the `String`.
+  func occurrenceCount(of substring: String) -> Int {
+    return components(separatedBy: substring).count - 1
+  }
+}
+
+private extension URLRequest {
+  /// Returns the default `timeoutInterval` for a `URLRequest`.
+  static func defaultTimeoutInterval() -> TimeInterval {
+    let placeholderURL = URL(string: "https://example.com")!
+    return URLRequest(url: placeholderURL).timeoutInterval
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift b/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
new file mode 100644
index 00000000000..cbc92527a1c
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
@@ -0,0 +1,175 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import GoogleGenerativeAI
+import XCTest
+#if canImport(AppKit)
+  import AppKit // For NSImage extensions.
+#elseif canImport(UIKit)
+  import UIKit // For UIImage extensions.
+#endif
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+final class GoogleGenerativeAITests: XCTestCase {
+  func codeSamples() async throws {
+    let config = GenerationConfig(temperature: 0.2,
+                                  topP: 0.1,
+                                  topK: 16,
+                                  candidateCount: 4,
+                                  maxOutputTokens: 256,
+                                  stopSequences: ["..."])
+    let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
+
+    // Permutations without optional arguments.
+    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY")
+    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", safetySettings: filters)
+    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", generationConfig: config)
+
+    // All arguments passed.
+    let genAI = GenerativeModel(name: "gemini-1.0-pro",
+                                apiKey: "API_KEY",
+                                generationConfig: config, // Optional
+                                safetySettings: filters // Optional
+    )
+    // Full Typed Usage
+    let pngData = Data() // ....
+    let contents = [ModelContent(role: "user",
+                                 parts: [
+                                   .text("Is it a cat?"),
+                                   .png(pngData),
+                                 ])]
+
+    do {
+      let response = try await genAI.generateContent(contents)
+      print(response.text ?? "Couldn't get text... check status")
+    } catch {
+      print("Error generating content: \(error)")
+    }
+
+    // Content input combinations.
+    let _ = try await genAI.generateContent("Constant String")
+    let str = "String Variable"
+    let _ = try await genAI.generateContent(str)
+    let _ = try await genAI.generateContent([str])
+    let _ = try await genAI.generateContent(str, "abc", "def")
+    #if canImport(UIKit)
+      _ = try await genAI.generateContent(UIImage())
+      _ = try await genAI.generateContent([UIImage()])
+      _ = try await genAI
+        .generateContent([str, UIImage(), ModelContent.Part.text(str)])
+      _ = try await genAI.generateContent(str, UIImage(), "def", UIImage())
+      _ = try await genAI.generateContent([str, UIImage(), "def", UIImage()])
+      _ = try await genAI.generateContent([ModelContent("def", UIImage()),
+                                           ModelContent("def", UIImage())])
+    #elseif canImport(AppKit)
+      _ = try await genAI.generateContent(NSImage())
+      _ = try await genAI.generateContent([NSImage()])
+      _ = try await genAI.generateContent(str, NSImage(), "def", NSImage())
+      _ = try await genAI.generateContent([str, NSImage(), "def", NSImage()])
+    #endif
+
+    // ThrowingPartsRepresentable combinations.
+    let _ = ModelContent(parts: [.text(str)])
+    let _ = ModelContent(role: "model", parts: [.text(str)])
+    let _ = ModelContent(parts: "Constant String")
+    let _ = ModelContent(parts: str)
+    // Note: This requires the `try` for some reason. Casting to explicit [PartsRepresentable] also
+    // doesn't work.
+    let _ = try ModelContent(parts: [str])
+    // Note: without `as [any ThrowingPartsRepresentable]` this will fail to compile with "Cannot
+    // convert value of type 'String' to expected element type
+    // 'Array<ModelContent.Part>.ArrayLiteralElement'. Not sure if there's a way we can get it to
+    // work.
+    let _ = try ModelContent(parts: [str, ModelContent.Part.data(
+      mimetype: "foo",
+      Data()
+    )] as [any ThrowingPartsRepresentable])
+    #if canImport(UIKit)
+      _ = try ModelContent(role: "user", parts: UIImage())
+      _ = try ModelContent(role: "user", parts: [UIImage()])
+      // Note: without `as [any ThrowingPartsRepresentable]` this will fail to compile with "Cannot
+      // convert
+      // value of type `[Any]` to expected type `[any ThrowingPartsRepresentable]`. Not sure if
+      // there's a
+      // way we can get it to work.
+      _ = try ModelContent(parts: [str, UIImage()] as [any ThrowingPartsRepresentable])
+      // Alternatively, you can explicitly declare the type in a variable and pass it in.
+      let representable2: [any ThrowingPartsRepresentable] = [str, UIImage()]
+      _ = try ModelContent(parts: representable2)
+      _ = try ModelContent(parts: [str, UIImage(),
+                                   ModelContent.Part.text(str)] as [any ThrowingPartsRepresentable])
+    #elseif canImport(AppKit)
+      _ = try ModelContent(role: "user", parts: NSImage())
+      _ = try ModelContent(role: "user", parts: [NSImage()])
+      // Note: without `as [any ThrowingPartsRepresentable]` this will fail to compile with "Cannot
+      // convert
+      // value of type `[Any]` to expected type `[any ThrowingPartsRepresentable]`. Not sure if
+      // there's a
+      // way we can get it to work.
+      _ = try ModelContent(parts: [str, NSImage()] as [any ThrowingPartsRepresentable])
+      // Alternatively, you can explicitly declare the type in a variable and pass it in.
+      let representable2: [any ThrowingPartsRepresentable] = [str, NSImage()]
+      _ = try ModelContent(parts: representable2)
+      _ =
+        try ModelContent(parts: [str, NSImage(),
+                                 ModelContent.Part.text(str)] as [any ThrowingPartsRepresentable])
+    #endif
+
+    // countTokens API
+    let _: CountTokensResponse = try await genAI.countTokens("What color is the Sky?")
+    #if canImport(UIKit)
+      let _: CountTokensResponse = try await genAI.countTokens("What color is the Sky?",
+                                                               UIImage())
+      let _: CountTokensResponse = try await genAI.countTokens([
+        ModelContent("What color is the Sky?", UIImage()),
+        ModelContent(UIImage(), "What color is the Sky?", UIImage()),
+      ])
+    #endif
+
+    // Chat
+    _ = genAI.startChat()
+    _ = genAI.startChat(history: [ModelContent(parts: "abc")])
+  }
+
+  // Result builder alternative
+
+  /*
+   let pngData = Data() // ....
+   let contents = [GenAIContent(role: "user",
+                                parts: [
+                                 .text("Is it a cat?"),
+                                 .png(pngData)
+                                ])]
+
+   // Turns into...
+
+   let contents = GenAIContent {
+     Role("user") {
+       Text("Is this a cat?")
+       Image(png: pngData)
+     }
+   }
+
+   GenAIContent {
+     ForEach(myInput) { input in
+       Role(input.role) {
+         input.contents
+       }
+     }
+   }
+
+   // Thoughts: this looks great from a code demo, but since I assume most content will be
+   // user generated, the result builder may not be the best API.
+   */
+}
diff --git a/FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift b/FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift
new file mode 100644
index 00000000000..30528cdf35a
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/MockURLProtocol.swift
@@ -0,0 +1,62 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+class MockURLProtocol: URLProtocol {
+  static var requestHandler: ((URLRequest) throws -> (
+    URLResponse,
+    AsyncLineSequence<URL.AsyncBytes>?
+  ))?
+
+  override class func canInit(with request: URLRequest) -> Bool { return true }
+
+  override class func canonicalRequest(for request: URLRequest) -> URLRequest { return request }
+
+  override func startLoading() {
+    guard let requestHandler = MockURLProtocol.requestHandler else {
+      fatalError("`requestHandler` is nil.")
+    }
+    guard let client = client else {
+      fatalError("`client` is nil.")
+    }
+
+    Task {
+      let (response, stream) = try requestHandler(self.request)
+      client.urlProtocol(self, didReceive: response, cacheStoragePolicy: .notAllowed)
+      if let stream = stream {
+        do {
+          for try await line in stream {
+            guard let data = line.data(using: .utf8) else {
+              fatalError("Failed to convert \"\(line)\" to UTF8 data.")
+            }
+            client.urlProtocol(self, didLoad: data)
+            // Add a newline character since AsyncLineSequence strips them when reading line by
+            // line;
+            // without the following, the whole file is delivered as a single line.
+            client.urlProtocol(self, didLoad: "\n".data(using: .utf8)!)
+          }
+        } catch {
+          client.urlProtocol(self, didFailWithError: error)
+          XCTFail("Unexpected failure reading lines from stream: \(error.localizedDescription)")
+        }
+      }
+      client.urlProtocolDidFinishLoading(self)
+    }
+  }
+
+  override func stopLoading() {}
+}
diff --git a/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
new file mode 100644
index 00000000000..da2f35f727f
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
@@ -0,0 +1,138 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import CoreGraphics
+import CoreImage
+import GoogleGenerativeAI
+import XCTest
+#if canImport(UIKit)
+  import UIKit
+#else
+  import AppKit
+#endif
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+final class PartsRepresentableTests: XCTestCase {
+  func testModelContentFromCGImageIsNotEmpty() throws {
+    // adapted from https://forums.swift.org/t/creating-a-cgimage-from-color-array/18634/2
+    var srgbArray = [UInt32](repeating: 0xFFFF_FFFF, count: 8 * 8)
+    let image = srgbArray.withUnsafeMutableBytes { ptr -> CGImage in
+      let ctx = CGContext(
+        data: ptr.baseAddress,
+        width: 8,
+        height: 8,
+        bitsPerComponent: 8,
+        bytesPerRow: 4 * 8,
+        space: CGColorSpace(name: CGColorSpace.sRGB)!,
+        bitmapInfo: CGBitmapInfo.byteOrder32Little.rawValue +
+          CGImageAlphaInfo.premultipliedFirst.rawValue
+      )!
+      return ctx.makeImage()!
+    }
+    let modelContent = try image.tryPartsValue()
+    XCTAssert(modelContent.count > 0, "Expected non-empty model content for CGImage: \(image)")
+  }
+
+  func testModelContentFromCIImageIsNotEmpty() throws {
+    let image = CIImage(color: CIColor.red)
+      .cropped(to: CGRect(origin: CGPointZero, size: CGSize(width: 16, height: 16)))
+    let modelContent = try image.tryPartsValue()
+    XCTAssert(modelContent.count > 0, "Expected non-empty model content for CGImage: \(image)")
+  }
+
+  func testModelContentFromInvalidCIImageThrows() throws {
+    let image = CIImage.empty()
+    do {
+      let _ = try image.tryPartsValue()
+    } catch {
+      guard let imageError = (error as? ImageConversionError) else {
+        XCTFail("Got unexpected error type: \(error)")
+        return
+      }
+      switch imageError {
+      case let .couldNotConvertToJPEG(source):
+        // String(describing:) works around a type error.
+        XCTAssertEqual(String(describing: source), String(describing: image))
+        return
+      case _:
+        XCTFail("Expected image conversion error, got \(imageError) instead")
+        return
+      }
+    }
+    XCTFail("Expected model content from invlaid image to error")
+  }
+
+  #if canImport(UIKit)
+    func testModelContentFromInvalidUIImageThrows() throws {
+      let image = UIImage()
+      do {
+        _ = try image.tryPartsValue()
+      } catch {
+        guard let imageError = (error as? ImageConversionError) else {
+          XCTFail("Got unexpected error type: \(error)")
+          return
+        }
+        switch imageError {
+        case let .couldNotConvertToJPEG(source):
+          // String(describing:) works around a type error.
+          XCTAssertEqual(String(describing: source), String(describing: image))
+          return
+        case _:
+          XCTFail("Expected image conversion error, got \(imageError) instead")
+          return
+        }
+      }
+      XCTFail("Expected model content from invlaid image to error")
+    }
+
+    func testModelContentFromUIImageIsNotEmpty() throws {
+      let coreImage = CIImage(color: CIColor.red)
+        .cropped(to: CGRect(origin: CGPointZero, size: CGSize(width: 16, height: 16)))
+      let image = UIImage(ciImage: coreImage)
+      let modelContent = try image.tryPartsValue()
+      XCTAssert(modelContent.count > 0, "Expected non-empty model content for UIImage: \(image)")
+    }
+  #else
+    func testModelContentFromNSImageIsNotEmpty() throws {
+      let coreImage = CIImage(color: CIColor.red)
+        .cropped(to: CGRect(origin: CGPointZero, size: CGSize(width: 16, height: 16)))
+      let rep = NSCIImageRep(ciImage: coreImage)
+      let image = NSImage(size: rep.size)
+      image.addRepresentation(rep)
+      let modelContent = try image.tryPartsValue()
+      XCTAssert(modelContent.count > 0, "Expected non-empty model content for NSImage: \(image)")
+    }
+
+    func testModelContentFromInvalidNSImageThrows() throws {
+      let image = NSImage()
+      do {
+        _ = try image.tryPartsValue()
+      } catch {
+        guard let imageError = (error as? ImageConversionError) else {
+          XCTFail("Got unexpected error type: \(error)")
+          return
+        }
+        switch imageError {
+        case .invalidUnderlyingImage:
+          // Pass
+          return
+        case _:
+          XCTFail("Expected image conversion error, got \(imageError) instead")
+          return
+        }
+      }
+      XCTFail("Expected model content from invlaid image to error")
+    }
+  #endif
+}

From bc2f6afa4fb67329a588612517dfc096cb41070a Mon Sep 17 00:00:00 2001
From: Morgan Chen <morganchen12@gmail.com>
Date: Tue, 19 Mar 2024 09:10:15 -0700
Subject: [PATCH 069/124] Add local App Check package to sample targets
 (#12576)

---
 .../project.pbxproj                           | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 20f07a5acca..1b4a69cafb0 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -56,6 +56,10 @@
 		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
 		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
 		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
+		8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */; };
+		8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */; };
+		8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */; };
+		8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E302BA8EED80049BFCA /* FirebaseAppCheck */; };
 		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
 /* End PBXBuildFile section */
 
@@ -97,6 +101,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */,
 				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
 				869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */,
 				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
@@ -108,6 +113,7 @@
 			buildActionMask = 2147483647;
 			files = (
 				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
+				8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */,
 				869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -116,6 +122,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */,
 				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
 				869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */,
 				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
@@ -126,6 +133,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */,
 				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
 				869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */,
 				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
@@ -335,6 +343,7 @@
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
 				869200AA2B86BE3100482873 /* FirebaseVertexAI */,
+				8D589E302BA8EED80049BFCA /* FirebaseAppCheck */,
 			);
 			productName = GenerativeAISample;
 			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
@@ -356,6 +365,7 @@
 			packageProductDependencies = (
 				88209C232B0FBE1700F64795 /* MarkdownUI */,
 				869200B02B86BFF500482873 /* FirebaseVertexAI */,
+				8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */,
 			);
 			productName = GenerativeAITextSample;
 			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
@@ -378,6 +388,7 @@
 				88B8A91D2B0FC55100424728 /* MarkdownUI */,
 				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
 				869200AE2B86BFBC00482873 /* FirebaseVertexAI */,
+				8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */,
 			);
 			productName = GenerativeAIMultimodalSample;
 			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
@@ -400,6 +411,7 @@
 				88D9474C2B14F27E008B5580 /* MarkdownUI */,
 				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
 				869200AC2B86BEA100482873 /* FirebaseVertexAI */,
+				8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */,
 			);
 			productName = ChatSample;
 			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
@@ -1043,6 +1055,22 @@
 			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
 			productName = MarkdownUI;
 		};
+		8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
+		8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
+		8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
+		8D589E302BA8EED80049BFCA /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
+		};
 		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = GenerativeAIUIComponents;

From 17f09541021a576cac0423c7427e00e4531f1020 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 19 Mar 2024 20:59:05 +0000
Subject: [PATCH 070/124] Use `generative-ai-swift` tests in Vertex AI (#12585)

---
 .github/workflows/vertexai.yml                |  7 +-
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   | 11 ++-
 .../streaming-success-citations2.txt          |  5 --
 .../unary-failure-api-key.json                |  2 +-
 ...e-finish-reason-recitation-no-content.json | 46 ----------
 ...ary-failure-unsupported-user-location.json | 13 ---
 .../Tests/Unit/GenerativeModelTests.swift     | 90 +++++++++----------
 .../Tests/Unit/PartsRepresentableTests.swift  |  2 +-
 ...leAITests.swift => VertexAIAPITests.swift} | 46 +++++++---
 Package.swift                                 |  9 ++
 .../FirebaseVertexAIUnit.xcscheme             | 77 ++++++++++++++++
 11 files changed, 178 insertions(+), 130 deletions(-)
 delete mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
 delete mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
 delete mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
 rename FirebaseVertexAI/Tests/Unit/{GoogleAITests.swift => VertexAIAPITests.swift} (84%)
 create mode 100644 scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
index 63cfd774e43..4f0880de320 100644
--- a/.github/workflows/vertexai.yml
+++ b/.github/workflows/vertexai.yml
@@ -31,7 +31,6 @@ jobs:
     - name: Xcode
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild
-      run: xcodebuild -list
-    # TODO: Add unit tests and switch from `spmbuildonly` to `spm`.
-    - name: Build
-      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAI ${{ matrix.target }} spmbuildonly
+      run: scripts/setup_spm_tests.sh
+    - name: Build and run tests
+      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAIUnit ${{ matrix.target }} spm
diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 668e3704122..046cce0e8bd 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -13,9 +13,10 @@
 // limitations under the License.
 
 import Foundation
-@testable import GoogleGenerativeAI
 import XCTest
 
+@testable import FirebaseVertexAI
+
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
 final class ChatTests: XCTestCase {
   var urlSession: URLSession!
@@ -46,7 +47,13 @@ final class ChatTests: XCTestCase {
       return (response, fileURL.lines)
     }
 
-    let model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+    let model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      urlSession: urlSession
+    )
     let chat = Chat(model: model, history: [])
     let input = "Test input"
     let stream = chat.sendMessageStream(input)
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
deleted file mode 100644
index 665e993ad75..00000000000
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations2.txt
+++ /dev/null
@@ -1,5 +0,0 @@
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": "In the context of the science fiction comedy novel \"The Hitchhiker's Guide to the Galaxy\" by Douglas Adams, the answer to the \"Ultimate Question of"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.055720285,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.062674366},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03904829,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03339982},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12220858,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0540987},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05781161,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03149938}]}]}
-
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Life, the Universe, and Everything\" is given as 42. However, in real life, there is no universally accepted meaning of life. The meaning"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.05910154,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.038321976},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.049589027,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.019271139},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03986249},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.098946586,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04177388}],"citationMetadata": {"citations": [{"startIndex": 52,"endIndex": 181,"uri": "https://imcsteakhouse.com.au/faqs"}]}}]}
-
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " of life is a philosophical question that has been pondered by humans for centuries, and there is no single answer that is widely agreed upon."}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.044764314,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.028870905},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04240383,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.012576347},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.10650458,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034880884},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.11085559,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03191915}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 91,"totalTokenCount": 100}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
index ecf6f6b53fa..d3b5677366e 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -9,7 +9,7 @@
         "reason": "API_KEY_INVALID",
         "domain": "googleapis.com",
         "metadata": {
-          "service": "generativelanguage.googleapis.com"
+          "service": "staging-firebaseml.sandbox.googleapis.com"
         }
       },
       {
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
deleted file mode 100644
index 0d99cd71d51..00000000000
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-finish-reason-recitation-no-content.json
+++ /dev/null
@@ -1,46 +0,0 @@
-{
-  "candidates": [
-    {
-      "finishReason": "RECITATION",
-      "index": 0,
-      "safetyRatings": [
-        {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
-        },
-        {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "HIGH"
-        },
-        {
-          "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
-        },
-        {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
-        }
-      ]
-    }
-  ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
-  }
-}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
deleted file mode 100644
index c4c2ace4e20..00000000000
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-unsupported-user-location.json
+++ /dev/null
@@ -1,13 +0,0 @@
-{
-  "error": {
-    "code": 400,
-    "message": "User location is not supported for the API use.",
-    "status": "FAILED_PRECONDITION",
-    "details": [
-      {
-        "@type": "type.googleapis.com/google.rpc.DebugInfo",
-        "detail": "[ORIGINAL ERROR] generic::failed_precondition: User location is not supported for the API use. [google.rpc.error_details_ext] { message: \"User location is not supported for the API use.\" }"
-      }
-    ]
-  }
-}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index b835a9ea05f..ac1980d93d4 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -12,9 +12,10 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-@testable import GoogleGenerativeAI
 import XCTest
 
+@testable import FirebaseVertexAI
+
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
 final class GenerativeModelTests: XCTestCase {
   let testPrompt = "What sorts of questions can I ask you?"
@@ -32,7 +33,13 @@ final class GenerativeModelTests: XCTestCase {
     let configuration = URLSessionConfiguration.default
     configuration.protocolClasses = [MockURLProtocol.self]
     urlSession = try XCTUnwrap(URLSession(configuration: configuration))
-    model = GenerativeModel(name: "my-model", apiKey: "API_KEY", urlSession: urlSession)
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      urlSession: urlSession
+    )
   }
 
   override func tearDown() {
@@ -163,6 +170,8 @@ final class GenerativeModelTests: XCTestCase {
       // Model name is prefixed with "models/".
       name: "models/test-model",
       apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -181,10 +190,13 @@ final class GenerativeModelTests: XCTestCase {
     do {
       _ = try await model.generateContent(testPrompt)
       XCTFail("Should throw GenerateContentError.internalError; no error thrown.")
-    } catch let GenerateContentError.invalidAPIKey(message) {
-      XCTAssertEqual(message, "API key not valid. Please pass a valid API key.")
+    } catch let GenerateContentError.internalError(error as RPCError) {
+      XCTAssertEqual(error.httpResponseCode, 400)
+      XCTAssertEqual(error.status, .invalidArgument)
+      XCTAssertEqual(error.message, "API key not valid. Please pass a valid API key.")
+      return
     } catch {
-      XCTFail("Should throw GenerateContentError.invalidAPIKey; error thrown: \(error)")
+      XCTFail("Should throw GenerateContentError.internalError(RPCError); error thrown: \(error)")
     }
   }
 
@@ -342,24 +354,6 @@ final class GenerativeModelTests: XCTestCase {
     }
   }
 
-  func testGenerateContent_failure_unsupportedUserLocation() async throws {
-    MockURLProtocol
-      .requestHandler = try httpRequestHandler(
-        forResource: "unary-failure-unsupported-user-location",
-        withExtension: "json",
-        statusCode: 400
-      )
-
-    do {
-      _ = try await model.generateContent(testPrompt)
-      XCTFail("Should throw GenerateContentError.unsupportedUserLocation; no error thrown.")
-    } catch GenerateContentError.unsupportedUserLocation {
-      return
-    }
-
-    XCTFail("Expected an unsupported user location error.")
-  }
-
   func testGenerateContent_failure_nonHTTPResponse() async throws {
     MockURLProtocol.requestHandler = try nonHTTPRequestHandler()
 
@@ -468,6 +462,7 @@ final class GenerativeModelTests: XCTestCase {
       name: "my-model",
       apiKey: "API_KEY",
       requestOptions: requestOptions,
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -490,8 +485,10 @@ final class GenerativeModelTests: XCTestCase {
       for try await _ in stream {
         XCTFail("No content is there, this shouldn't happen.")
       }
-    } catch GenerateContentError.invalidAPIKey {
-      // invalidAPIKey error is as expected, nothing else to check.
+    } catch let GenerateContentError.internalError(error as RPCError) {
+      XCTAssertEqual(error.httpResponseCode, 400)
+      XCTAssertEqual(error.status, .invalidArgument)
+      XCTAssertEqual(error.message, "API key not valid. Please pass a valid API key.")
       return
     }
 
@@ -747,26 +744,6 @@ final class GenerativeModelTests: XCTestCase {
     XCTFail("Expected an internal decoding error.")
   }
 
-  func testGenerateContentStream_failure_unsupportedUserLocation() async throws {
-    MockURLProtocol
-      .requestHandler = try httpRequestHandler(
-        forResource: "unary-failure-unsupported-user-location",
-        withExtension: "json",
-        statusCode: 400
-      )
-
-    let stream = model.generateContentStream(testPrompt)
-    do {
-      for try await content in stream {
-        XCTFail("Unexpected content in stream: \(content)")
-      }
-    } catch GenerateContentError.unsupportedUserLocation {
-      return
-    }
-
-    XCTFail("Expected an unsupported user location error.")
-  }
-
   func testGenerateContentStream_requestOptions_customTimeout() async throws {
     let expectedTimeout = 150.0
     MockURLProtocol
@@ -780,6 +757,7 @@ final class GenerativeModelTests: XCTestCase {
       name: "my-model",
       apiKey: "API_KEY",
       requestOptions: requestOptions,
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -837,6 +815,7 @@ final class GenerativeModelTests: XCTestCase {
       name: "my-model",
       apiKey: "API_KEY",
       requestOptions: requestOptions,
+      appCheck: nil,
       urlSession: urlSession
     )
 
@@ -851,7 +830,12 @@ final class GenerativeModelTests: XCTestCase {
     let modelName = "my-model"
     let modelResourceName = "models/\(modelName)"
 
-    model = GenerativeModel(name: modelName, apiKey: "API_KEY")
+    model = GenerativeModel(
+      name: modelName,
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil
+    )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
   }
@@ -859,7 +843,12 @@ final class GenerativeModelTests: XCTestCase {
   func testModelResourceName_modelsPrefix() async throws {
     let modelResourceName = "models/my-model"
 
-    model = GenerativeModel(name: modelResourceName, apiKey: "API_KEY")
+    model = GenerativeModel(
+      name: modelResourceName,
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil
+    )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
   }
@@ -867,7 +856,12 @@ final class GenerativeModelTests: XCTestCase {
   func testModelResourceName_tunedModelsPrefix() async throws {
     let tunedModelResourceName = "tunedModels/my-model"
 
-    model = GenerativeModel(name: tunedModelResourceName, apiKey: "API_KEY")
+    model = GenerativeModel(
+      name: tunedModelResourceName,
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: nil
+    )
 
     XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
   }
diff --git a/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
index da2f35f727f..4243c2441be 100644
--- a/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/PartsRepresentableTests.swift
@@ -14,7 +14,7 @@
 
 import CoreGraphics
 import CoreImage
-import GoogleGenerativeAI
+import FirebaseVertexAI
 import XCTest
 #if canImport(UIKit)
   import UIKit
diff --git a/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
similarity index 84%
rename from FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
rename to FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index cbc92527a1c..4676b2e34d9 100644
--- a/FirebaseVertexAI/Tests/Unit/GoogleAITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -12,7 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import GoogleGenerativeAI
+import FirebaseCore
+import FirebaseVertexAI
 import XCTest
 #if canImport(AppKit)
   import AppKit // For NSImage extensions.
@@ -21,8 +22,9 @@ import XCTest
 #endif
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-final class GoogleGenerativeAITests: XCTestCase {
+final class VertexAIAPITests: XCTestCase {
   func codeSamples() async throws {
+    let app = FirebaseApp.app()
     let config = GenerationConfig(temperature: 0.2,
                                   topP: 0.1,
                                   topK: 16,
@@ -32,16 +34,40 @@ final class GoogleGenerativeAITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Permutations without optional arguments.
-    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY")
-    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", safetySettings: filters)
-    let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", generationConfig: config)
 
-    // All arguments passed.
-    let genAI = GenerativeModel(name: "gemini-1.0-pro",
-                                apiKey: "API_KEY",
-                                generationConfig: config, // Optional
-                                safetySettings: filters // Optional
+    // TODO: Change `genAI` to `_` when safetySettings and generationConfig are added to public API.
+    let genAI = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    let _ = VertexAI.generativeModel(
+      app: app!,
+      modelName: "gemini-1.0-pro",
+      location: "us-central1"
     )
+
+    // TODO: Add safetySettings to public API.
+    // TODO: Add permutation with `app` specified.
+    //  let _ = VertexAI.generativeModel(
+    //    modelName: "gemini-1.0-pro",
+    //    location: "us-central1",
+    //    safetySettings: filters
+    //  )
+    // TODO: Add generationConfig to public API.
+    // TODO: Add permutation with `app` specified.
+    // let _ = VertexAI.generativeModel(
+    //   modelName: "gemini-1.0-pro",
+    //   location: "us-central1",
+    //   generationConfig: config
+    // )
+
+    // All arguments passed.
+    // TODO: Add safetySettings and generationConfig to public API.
+    // TODO: Add permutation with `app` specified.
+    // let genAI = VertexAI.generativeModel(
+    //   modelName: "gemini-1.0-pro",
+    //   location: "us-central1",
+    //   generationConfig: config, // Optional
+    //   safetySettings: filters // Optional
+    // )
+
     // Full Typed Usage
     let pngData = Data() // ....
     let contents = [ModelContent(role: "user",
diff --git a/Package.swift b/Package.swift
index bcbbb20422d..15929c130cf 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1374,6 +1374,15 @@ let package = Package(
       ],
       path: "FirebaseVertexAI/Sources"
     ),
+    .testTarget(
+      name: "FirebaseVertexAIUnit",
+      dependencies: ["FirebaseVertexAI"],
+      path: "FirebaseVertexAI/Tests/Unit",
+      resources: [
+        .process("CountTokenResponses"),
+        .process("GenerateContentResponses"),
+      ]
+    ),
   ] + firestoreTargets(),
   cLanguageStandard: .c99,
   cxxLanguageStandard: CXXLanguageStandard.gnucxx14
diff --git a/scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme b/scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme
new file mode 100644
index 00000000000..cb4b5adae36
--- /dev/null
+++ b/scripts/spm_test_schemes/FirebaseVertexAIUnit.xcscheme
@@ -0,0 +1,77 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Scheme
+   LastUpgradeVersion = "1200"
+   version = "1.3">
+   <BuildAction
+      parallelizeBuildables = "YES"
+      buildImplicitDependencies = "YES">
+      <BuildActionEntries>
+         <BuildActionEntry
+            buildForTesting = "YES"
+            buildForRunning = "YES"
+            buildForProfiling = "NO"
+            buildForArchiving = "NO"
+            buildForAnalyzing = "YES">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "FirebaseVertexAIUnit"
+               BuildableName = "FirebaseVertexAIUnit"
+               BlueprintName = "FirebaseVertexAIUnit"
+               ReferencedContainer = "container:">
+            </BuildableReference>
+         </BuildActionEntry>
+      </BuildActionEntries>
+   </BuildAction>
+   <TestAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      shouldUseLaunchSchemeArgsEnv = "YES">
+      <Testables>
+         <TestableReference
+            skipped = "NO">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "FirebaseVertexAIUnit"
+               BuildableName = "FirebaseVertexAIUnit"
+               BlueprintName = "FirebaseVertexAIUnit"
+               ReferencedContainer = "container:">
+            </BuildableReference>
+         </TestableReference>
+      </Testables>
+   </TestAction>
+   <LaunchAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      launchStyle = "0"
+      useCustomWorkingDirectory = "NO"
+      ignoresPersistentStateOnLaunch = "NO"
+      debugDocumentVersioning = "YES"
+      debugServiceExtension = "internal"
+      allowLocationSimulation = "YES">
+   </LaunchAction>
+   <ProfileAction
+      buildConfiguration = "Release"
+      shouldUseLaunchSchemeArgsEnv = "YES"
+      savedToolIdentifier = ""
+      useCustomWorkingDirectory = "NO"
+      debugDocumentVersioning = "YES">
+      <MacroExpansion>
+         <BuildableReference
+            BuildableIdentifier = "primary"
+            BlueprintIdentifier = "FirebaseVertexAIUnit"
+            BuildableName = "FirebaseVertexAIUnit"
+            BlueprintName = "FirebaseVertexAIUnit"
+            ReferencedContainer = "container:">
+         </BuildableReference>
+      </MacroExpansion>
+   </ProfileAction>
+   <AnalyzeAction
+      buildConfiguration = "Debug">
+   </AnalyzeAction>
+   <ArchiveAction
+      buildConfiguration = "Release"
+      revealArchiveInOrganizer = "YES">
+   </ArchiveAction>
+</Scheme>

From 0ba884843d728bb896d771c5267c41da64f9f20f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 20 Mar 2024 14:11:28 +0000
Subject: [PATCH 071/124] Add App Check tests in `GenerativeModelTests`
 (#12590)

---
 .../Tests/Unit/GenerativeModelTests.swift     | 131 +++++++++++++++++-
 1 file changed, 129 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index ac1980d93d4..13c625036cc 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -12,6 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import FirebaseAppCheckInterop
 import XCTest
 
 @testable import FirebaseVertexAI
@@ -178,6 +179,43 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_appCheck_validToken() async throws {
+    let appCheckToken = "test-valid-token"
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(token: appCheckToken),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        appCheckToken: appCheckToken
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_appCheck_tokenRefreshError() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        appCheckToken: AppCheckInteropFake.placeholderTokenValue
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
   func testGenerateContent_failure_invalidAPIKey() async throws {
     let expectedStatusCode = 400
     MockURLProtocol
@@ -654,6 +692,45 @@ final class GenerativeModelTests: XCTestCase {
       .contains(where: { $0.startIndex == 899 && $0.endIndex == 1026 && !$0.uri.isEmpty }))
   }
 
+  func testGenerateContentStream_appCheck_validToken() async throws {
+    let appCheckToken = "test-valid-token"
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(token: appCheckToken),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt",
+        appCheckToken: appCheckToken
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    for try await _ in stream {}
+  }
+
+  func testGenerateContentStream_appCheck_tokenRefreshError() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      requestOptions: RequestOptions(),
+      appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt",
+        appCheckToken: AppCheckInteropFake.placeholderTokenValue
+      )
+
+    let stream = model.generateContentStream(testPrompt)
+    for try await _ in stream {}
+  }
+
   func testGenerateContentStream_errorMidStream() async throws {
     MockURLProtocol.requestHandler = try httpRequestHandler(
       forResource: "streaming-failure-error-mid-stream",
@@ -887,8 +964,8 @@ final class GenerativeModelTests: XCTestCase {
   private func httpRequestHandler(forResource name: String,
                                   withExtension ext: String,
                                   statusCode: Int = 200,
-                                  timeout: TimeInterval = URLRequest
-                                    .defaultTimeoutInterval()) throws -> ((URLRequest) throws -> (
+                                  timeout: TimeInterval = URLRequest.defaultTimeoutInterval(),
+                                  appCheckToken: String? = nil) throws -> ((URLRequest) throws -> (
     URLResponse,
     AsyncLineSequence<URL.AsyncBytes>?
   )) {
@@ -897,6 +974,7 @@ final class GenerativeModelTests: XCTestCase {
       let requestURL = try XCTUnwrap(request.url)
       XCTAssertEqual(requestURL.path.occurrenceCount(of: "models/"), 1)
       XCTAssertEqual(request.timeoutInterval, timeout)
+      XCTAssertEqual(request.value(forHTTPHeaderField: "X-Firebase-AppCheck"), appCheckToken)
       let response = try XCTUnwrap(HTTPURLResponse(
         url: requestURL,
         statusCode: statusCode,
@@ -922,3 +1000,52 @@ private extension URLRequest {
     return URLRequest(url: placeholderURL).timeoutInterval
   }
 }
+
+class AppCheckInteropFake: NSObject, AppCheckInterop {
+  /// The placeholder token value returned when an error occurs
+  static let placeholderTokenValue = "placeholder-token"
+
+  var token: String
+  var error: Error?
+
+  private init(token: String, error: Error?) {
+    self.token = token
+    self.error = error
+  }
+
+  convenience init(token: String) {
+    self.init(token: token, error: nil)
+  }
+
+  convenience init(error: Error) {
+    self.init(token: AppCheckInteropFake.placeholderTokenValue, error: error)
+  }
+
+  func getToken(forcingRefresh: Bool) async -> any FIRAppCheckTokenResultInterop {
+    return AppCheckTokenResultInteropFake(token: token, error: error)
+  }
+
+  func tokenDidChangeNotificationName() -> String {
+    fatalError("\(#function) not implemented.")
+  }
+
+  func notificationTokenKey() -> String {
+    fatalError("\(#function) not implemented.")
+  }
+
+  func notificationAppNameKey() -> String {
+    fatalError("\(#function) not implemented.")
+  }
+
+  private class AppCheckTokenResultInteropFake: NSObject, FIRAppCheckTokenResultInterop {
+    var token: String
+    var error: Error?
+
+    init(token: String, error: Error?) {
+      self.token = token
+      self.error = error
+    }
+  }
+}
+
+struct AppCheckErrorFake: Error {}

From 41f4474080df061259c73458e647649b3d628fea Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 21 Mar 2024 20:56:59 +0000
Subject: [PATCH 072/124] Make `generativeModel` an instance method of
 `VertexAI` (#12599)

---
 .../ViewModels/ConversationViewModel.swift    |  5 +-
 .../ViewModels/PhotoReasoningViewModel.swift  |  5 +-
 .../ViewModels/SummarizeViewModel.swift       |  5 +-
 FirebaseVertexAI/Sources/VertexAI.swift       | 81 +++++++++++--------
 .../Sources/VertexAIComponent.swift           | 10 +--
 .../Tests/Unit/VertexAIAPITests.swift         | 49 +++++------
 6 files changed, 85 insertions(+), 70 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 465789be387..05cbe11250f 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,7 +36,10 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1"
+    )
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index dc41e00444a..2f2ed88d4a1 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,10 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro-vision", location: "us-central1")
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro-vision",
+      location: "us-central1"
+    )
   }
 
   func reason() async {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index fb5b349ac82..0e3073d6da2 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,7 +32,10 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1"
+    )
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 0f79e1be99c..64e76d3b4f5 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -20,70 +20,81 @@ import Foundation
 @_implementationOnly import FirebaseCoreExtension
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-@objc(FIRVertexAI)
-open class VertexAI: NSObject {
+public class VertexAI: NSObject {
   // MARK: - Public APIs
 
-  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  /// The default `VertexAI` instance.
   ///
-  /// This instance is configured with the default `FirebaseApp`.
-  ///
-  /// TODO: Add RequestOptions to public API.
-  public static func generativeModel(modelName: String, location: String) -> GenerativeModel {
+  /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
+  public static func vertexAI() -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
-    return generativeModel(app: app, modelName: modelName, location: location)
+
+    return vertexAI(app: app)
   }
 
-  /// Returns an instance of `GoogleGenerativeAI.GenerativeModel` that uses the Vertex AI API.
+  /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
-  /// TODO: Add RequestOptions to public API.
-  public static func generativeModel(app: FirebaseApp, modelName: String,
-                                     location: String) -> GenerativeModel {
+  /// - Parameter app: The custom `FirebaseApp` used for initialization.
+  /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
+  public static func vertexAI(app: FirebaseApp) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
-    let modelResourceName = modelResourceName(app: app, modelName: modelName, location: location)
-    let vertexAI = provider.vertexAI(location: location, modelResourceName: modelResourceName)
 
-    return vertexAI.model
+    return provider.vertexAI()
   }
 
-  // MARK: - Private
-
-  /// The `FirebaseApp` associated with this `VertexAI` instance.
-  private let app: FirebaseApp
-
-  private let appCheck: AppCheckInterop?
-
-  private let location: String
-
-  private let modelResouceName: String
+  /// Initializes a generative model with the given parameters.
+  ///
+  /// - Parameters:
+  ///   - modelName: The name of the model to use, e.g., `"gemini-1.0-pro"`; see
+  ///     [Gemini
+  ///     models](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#gemini-models)
+  ///     for a list of supported model names.
+  ///   - location: The location identifier, e.g., `us-central1`; see
+  ///     [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported locations.
+  ///   - generationConfig: The content generation parameters your model should use.
+  ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - requestOptions: Configuration parameters for sending requests to the backend.
+  public func generativeModel(modelName: String, location: String,
+                              generationConfig: GenerationConfig? = nil,
+                              safetySettings: [SafetySetting]? = nil,
+                              requestOptions: RequestOptions = RequestOptions())
+    -> GenerativeModel {
+    let modelResourceName = modelResourceName(modelName: modelName, location: location)
 
-  lazy var model: GenerativeModel = {
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
     }
+
     return GenerativeModel(
-      name: modelResouceName,
+      name: modelResourceName,
       apiKey: apiKey,
-      // TODO: Add RequestOptions to public API.
-      requestOptions: RequestOptions(),
+      generationConfig: generationConfig,
+      safetySettings: safetySettings,
+      requestOptions: requestOptions,
       appCheck: appCheck
     )
-  }()
+  }
+
+  // MARK: - Private
+
+  /// The `FirebaseApp` associated with this `VertexAI` instance.
+  private let app: FirebaseApp
+
+  private let appCheck: AppCheckInterop?
 
-  init(app: FirebaseApp, location: String, modelResourceName: String) {
+  init(app: FirebaseApp) {
     self.app = app
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
-    self.location = location
-    modelResouceName = modelResourceName
   }
 
-  private static func modelResourceName(app: FirebaseApp, modelName: String,
-                                        location: String) -> String {
+  private func modelResourceName(modelName: String, location: String) -> String {
     if modelName.contains("/") {
       return modelName
     }
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a8d6c177c74..1378f812626 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI(location: String, modelResourceName: String) -> VertexAI
+  @objc func vertexAI() -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -64,17 +64,17 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   // MARK: - VertexAIProvider conformance
 
-  func vertexAI(location: String, modelResourceName: String) -> VertexAI {
+  func vertexAI() -> VertexAI {
     os_unfair_lock_lock(&instancesLock)
 
     // Unlock before the function returns.
     defer { os_unfair_lock_unlock(&instancesLock) }
 
-    if let instance = instances[modelResourceName] {
+    if let instance = instances[app.name] {
       return instance
     }
-    let newInstance = VertexAI(app: app, location: location, modelResourceName: modelResourceName)
-    instances[modelResourceName] = newInstance
+    let newInstance = VertexAI(app: app)
+    instances[app.name] = newInstance
     return newInstance
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 4676b2e34d9..a1fd27ab4c5 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -33,40 +33,35 @@ final class VertexAIAPITests: XCTestCase {
                                   stopSequences: ["..."])
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
+    // Instantiate Vertex AI SDK - Default App
+    let vertexAI = VertexAI.vertexAI()
+
+    // Instantiate Vertex AI SDK - Custom App
+    let _ = VertexAI.vertexAI(app: app!)
+
     // Permutations without optional arguments.
 
-    // TODO: Change `genAI` to `_` when safetySettings and generationConfig are added to public API.
-    let genAI = VertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
-    let _ = VertexAI.generativeModel(
-      app: app!,
+    let _ = vertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+
+    let _ = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1"
+      location: "us-central1",
+      safetySettings: filters
     )
 
-    // TODO: Add safetySettings to public API.
-    // TODO: Add permutation with `app` specified.
-    //  let _ = VertexAI.generativeModel(
-    //    modelName: "gemini-1.0-pro",
-    //    location: "us-central1",
-    //    safetySettings: filters
-    //  )
-    // TODO: Add generationConfig to public API.
-    // TODO: Add permutation with `app` specified.
-    // let _ = VertexAI.generativeModel(
-    //   modelName: "gemini-1.0-pro",
-    //   location: "us-central1",
-    //   generationConfig: config
-    // )
+    let _ = vertexAI.generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1",
+      generationConfig: config
+    )
 
     // All arguments passed.
-    // TODO: Add safetySettings and generationConfig to public API.
-    // TODO: Add permutation with `app` specified.
-    // let genAI = VertexAI.generativeModel(
-    //   modelName: "gemini-1.0-pro",
-    //   location: "us-central1",
-    //   generationConfig: config, // Optional
-    //   safetySettings: filters // Optional
-    // )
+    let genAI = vertexAI.generativeModel(
+      modelName: "gemini-1.0-pro",
+      location: "us-central1",
+      generationConfig: config, // Optional
+      safetySettings: filters // Optional
+    )
 
     // Full Typed Usage
     let pngData = Data() // ....

From d2ced3b108a2df8a1a1bb2c0f2429d035bacba70 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 22 Mar 2024 21:06:39 +0000
Subject: [PATCH 073/124] Update test JSON payloads with Vertex AI output
 (#12615)

---
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |  2 +-
 .../streaming-success-basic-reply-long.txt    | 20 +++----
 .../streaming-success-basic-reply-short.txt   |  3 +-
 .../unary-success-basic-reply-long.json       | 56 +++++++++----------
 .../unary-success-basic-reply-short.json      | 54 +++++++++---------
 .../Tests/Unit/GenerativeModelTests.swift     | 26 +++++----
 6 files changed, 75 insertions(+), 86 deletions(-)

diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 046cce0e8bd..5a7868fbe7a 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -66,7 +66,7 @@ final class ChatTests: XCTestCase {
     XCTAssertEqual(chat.history.count, 2)
     XCTAssertEqual(chat.history[0].parts[0].text, input)
 
-    let finalText = "1 2 3 4 5 6 7 8 9 10"
+    let finalText = "1 2 3 4 5 6 7 8"
     let assembledExpectation = ModelContent(role: "model", parts: finalText)
     XCTAssertEqual(chat.history[0].parts[0].text, input)
     XCTAssertEqual(chat.history[1], assembledExpectation)
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
index bca95140490..218cae0b985 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt
@@ -1,19 +1,15 @@
-data: {"candidates": [{"content": {"parts": [{"text": "1 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "1 "}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "2 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "2 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.0394904,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04468087},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.034553625,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03890198},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.09401018,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.025809621},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.036562506,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.047691282}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "3 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "3 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.03507868,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045183755},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.027742893,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.043528143},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.026105914},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.06681233,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "4 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "4 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.037750278,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05089372},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.040087357,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05888469},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.071202725,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034100424},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.07613248,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.051749535}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "5 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "5 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.04672496,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059210256},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04977345,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05623635},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.083890386,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.08359067,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05975658}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "6 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "6 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07779744,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06052939},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.041930523,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.056756895},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05350215},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.09203286,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048676573}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "7 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "7 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.071202725,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05291181},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.031439852,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04509957},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.11417085,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04922211},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.09451043,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.052716404}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "8 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "9 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "10"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "8"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06221698,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045777276},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03085051,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04560694},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.0992954,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.040769264},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.100701615,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.061424047}]}],"usageMetadata": {"promptTokenCount": 6,"candidatesTokenCount": 326,"totalTokenCount": 332}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
index a7f5476954e..78f569ddecf 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt
@@ -1,2 +1 @@
-data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
-
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Mountain View, California"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.02854415,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.052424565},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.24926445,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0996453},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.087096825,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.043123372},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.14402841,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.086169556}]}],"usageMetadata": {"promptTokenCount": 6,"candidatesTokenCount": 4,"totalTokenCount": 10}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
index 59b84de92fe..325d172b894 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json
@@ -2,53 +2,49 @@
   "candidates": [
     {
       "content": {
+        "role": "model",
         "parts": [
           {
-            "text": "You can ask me a wide range of questions on various topics. Here are some examples:\n\n1. **General Knowledge:**\n   - What is the capital of France?\n   - Who painted the Mona Lisa?\n   - What is the largest ocean in the world?\n\n2. **Science and Technology:**\n   - How does a computer work?\n   - What is the difference between a virus and a bacteria?\n   - What are the latest advancements in artificial intelligence?\n\n3. **History and Culture:**\n   - Who was the first president of the United States?\n   - What is the significance of the Great Wall of China?\n   - What are some of the most famous works of Shakespeare?\n\n4. **Current Events:**\n   - What is the latest news about the COVID-19 pandemic?\n   - Who is the current president of Ukraine?\n   - What are the major issues being discussed in the upcoming election?\n\n5. **Personal Questions:**\n   - What are your hobbies?\n   - What is your favorite book or movie?\n   - What are your thoughts on the future of technology?\n\n6. **Fun and Games:**\n   - Can you tell me a joke?\n   - What is the answer to this riddle: \"I have keys but no locks. I have space but no room. You can enter, but can't go outside.\" (Answer: a keyboard)\n   - Let's play a game of 20 questions.\n\n7. **Hypothetical Questions:**\n   - What would you do if you won the lottery?\n   - What would happen if time travel were possible?\n   - What is the meaning of life?\n\n8. **Philosophical Questions:**\n   - What is the nature of reality?\n   - Does free will exist?\n   - What is the difference between right and wrong?\n\n9. **Creative Questions:**\n   - Write a poem about a sunset.\n   - Design a logo for a new company.\n   - Compose a song about your favorite season.\n\n10. **Technical Questions:**\n    - How can I improve the performance of my computer?\n    - What is the best way to troubleshoot a network issue?\n    - How do I create a website using HTML and CSS?\n\nRemember, I am still under development and may not be able to answer all questions perfectly. However, I will do my best to provide you with accurate and informative responses."
+            "text": "You can ask me a wide range of questions, including:\n\n* **General knowledge questions:** What is the capital of France? Who painted the Mona Lisa?\n* **Science questions:** What is the chemical formula for water? How does photosynthesis work?\n* **History questions:** When did World War II start? Who was the first president of the United States?\n* **Math questions:** What is the square root of 16? How do you solve for x in the equation x + 5 = 10?\n* **Current events questions:** What is happening in Ukraine? Who is the current president of the United States?\n* **Personal questions:** What are your hobbies? What are your favorite books?\n* **Hypothetical questions:** What would happen if all the ice caps melted? What if humans could live on Mars?\n* **Questions about me:** What is your name? How old are you? What are your capabilities?\n\n**Tips for asking questions:**\n\n* Be specific and clear in your questions.\n* Use correct grammar and spelling.\n* Try to ask open-ended questions that allow for multiple answers.\n* Be patient and wait for my response. I am still under development, so I may take a few seconds to process your question.\n\nPlease note that I am still under development and may not be able to answer all questions accurately. However, I will do my best to provide you with the most relevant information I have available."
           }
-        ],
-        "role": "model"
+        ]
       },
       "finishReason": "STOP",
-      "index": 0,
       "safetyRatings": [
         {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.047869004,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.050705366
         },
         {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.052134257,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.036288295
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.08464396,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.033907957
         },
         {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.06290424,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.050611436
         }
       ]
     }
   ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
+  "usageMetadata": {
+    "promptTokenCount": 6,
+    "candidatesTokenCount": 303,
+    "totalTokenCount": 309
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
index 40a9a6da58e..3e3ddb80fd6 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json
@@ -2,53 +2,49 @@
   "candidates": [
     {
       "content": {
+        "role": "model",
         "parts": [
           {
             "text": "Mountain View, California, United States"
           }
-        ],
-        "role": "model"
+        ]
       },
       "finishReason": "STOP",
-      "index": 0,
       "safetyRatings": [
         {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.029035643,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.05613278
         },
         {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.2641685,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.082253955
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.087252244,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.04509957
         },
         {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.1431877,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.11027937
         }
       ]
     }
   ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
+  "usageMetadata": {
+    "promptTokenCount": 6,
+    "candidatesTokenCount": 7,
+    "totalTokenCount": 13
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 13c625036cc..c249613f9e9 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -25,7 +25,7 @@ final class GenerativeModelTests: XCTestCase {
     .init(category: .hateSpeech, probability: .negligible),
     .init(category: .harassment, probability: .negligible),
     .init(category: .dangerousContent, probability: .negligible),
-  ]
+  ].sorted()
 
   var urlSession: URLSession!
   var model: GenerativeModel!
@@ -62,15 +62,12 @@ final class GenerativeModelTests: XCTestCase {
     let candidate = try XCTUnwrap(response.candidates.first)
     let finishReason = try XCTUnwrap(candidate.finishReason)
     XCTAssertEqual(finishReason, .stop)
-    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.safetyRatings.sorted(), safetyRatingsNegligible)
     XCTAssertEqual(candidate.content.parts.count, 1)
     let part = try XCTUnwrap(candidate.content.parts.first)
     let partText = try XCTUnwrap(part.text)
     XCTAssertTrue(partText.hasPrefix("You can ask me a wide range of questions"))
     XCTAssertEqual(response.text, partText)
-    let promptFeedback = try XCTUnwrap(response.promptFeedback)
-    XCTAssertNil(promptFeedback.blockReason)
-    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
   }
 
   func testGenerateContent_success_basicReplyShort() async throws {
@@ -86,14 +83,11 @@ final class GenerativeModelTests: XCTestCase {
     let candidate = try XCTUnwrap(response.candidates.first)
     let finishReason = try XCTUnwrap(candidate.finishReason)
     XCTAssertEqual(finishReason, .stop)
-    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.safetyRatings.sorted(), safetyRatingsNegligible)
     XCTAssertEqual(candidate.content.parts.count, 1)
     let part = try XCTUnwrap(candidate.content.parts.first)
     XCTAssertEqual(part.text, "Mountain View, California, United States")
     XCTAssertEqual(response.text, part.text)
-    let promptFeedback = try XCTUnwrap(response.promptFeedback)
-    XCTAssertNil(promptFeedback.blockReason)
-    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
   }
 
   func testGenerateContent_success_citations() async throws {
@@ -131,7 +125,7 @@ final class GenerativeModelTests: XCTestCase {
     let candidate = try XCTUnwrap(response.candidates.first)
     let finishReason = try XCTUnwrap(candidate.finishReason)
     XCTAssertEqual(finishReason, .stop)
-    XCTAssertEqual(candidate.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(candidate.safetyRatings.sorted(), safetyRatingsNegligible)
     XCTAssertEqual(candidate.content.parts.count, 1)
     let part = try XCTUnwrap(candidate.content.parts.first)
     let partText = try XCTUnwrap(part.text)
@@ -139,7 +133,7 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(response.text, part.text)
     let promptFeedback = try XCTUnwrap(response.promptFeedback)
     XCTAssertNil(promptFeedback.blockReason)
-    XCTAssertEqual(promptFeedback.safetyRatings, safetyRatingsNegligible)
+    XCTAssertEqual(promptFeedback.safetyRatings.sorted(), safetyRatingsNegligible)
   }
 
   func testGenerateContent_success_unknownEnum_safetyRatings() async throws {
@@ -627,7 +621,7 @@ final class GenerativeModelTests: XCTestCase {
       responses += 1
     }
 
-    XCTAssertEqual(responses, 10)
+    XCTAssertEqual(responses, 8)
   }
 
   func testGenerateContentStream_successBasicReplyShort() async throws {
@@ -1049,3 +1043,11 @@ class AppCheckInteropFake: NSObject, AppCheckInterop {
 }
 
 struct AppCheckErrorFake: Error {}
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+extension SafetyRating: Comparable {
+  public static func < (lhs: FirebaseVertexAI.SafetyRating,
+                        rhs: FirebaseVertexAI.SafetyRating) -> Bool {
+    return lhs.category.rawValue < rhs.category.rawValue
+  }
+}

From c0bdcab576e1c931204fee516ef6f25bc0917366 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Mon, 25 Mar 2024 08:21:58 -0700
Subject: [PATCH 074/124] Build Sample in CI (#12616)

---
 .github/workflows/vertexai.yml                | 42 ++++++++++++++++---
 .../Tests/Unit/GenerativeModelTests.swift     |  1 +
 scripts/build.sh                              |  8 ++++
 3 files changed, 46 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
index 4f0880de320..ebf58529f54 100644
--- a/.github/workflows/vertexai.yml
+++ b/.github/workflows/vertexai.yml
@@ -3,9 +3,9 @@ name: vertexai
 on:
   pull_request:
     paths:
-        - 'FirebaseVertexAI**'
-        - '.github/workflows/vertexai.yml'
-        - 'Gemfile*'
+    - 'FirebaseVertexAI**'
+    - '.github/workflows/vertexai.yml'
+    - 'Gemfile*'
   schedule:
     # Run every day at 11pm (PST) - cron uses UTC times
     - cron:  '0 7 * * *'
@@ -32,5 +32,37 @@ jobs:
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild
       run: scripts/setup_spm_tests.sh
-    - name: Build and run tests
-      run: scripts/third_party/travis/retry.sh scripts/build.sh FirebaseVertexAIUnit ${{ matrix.target }} spm
+    - uses: nick-fields/retry@v3
+      with:
+        timeout_minutes: 120
+        max_attempts: 3
+        retry_on: error
+        retry_wait_seconds: 120
+        command: scripts/build.sh FirebaseVertexAIUnit ${{ matrix.target }} spm
+
+  sample:
+    strategy:
+      matrix:
+        # Test build with debug and release configs (whether or not DEBUG is set and optimization level)
+        build: [build]
+        include:
+          - os: macos-13
+            xcode: Xcode_15.0.1
+          - os: macos-14
+            xcode: Xcode_15.2
+    runs-on: ${{ matrix.os }}
+    steps:
+    - uses: actions/checkout@v4
+    - name: Xcode
+      run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
+    - name: Initialize xcodebuild
+      run: xcodebuild -list
+    - name: Placeholder GoogleService-Info.plist for build testing
+      run: cp FirebaseCore/Tests/Unit/Resources/GoogleService-Info.plist FirebaseVertexAI/Sample/
+    - uses: nick-fields/retry@v3
+      with:
+        timeout_minutes: 120
+        max_attempts: 3
+        retry_on: error
+        retry_wait_seconds: 120
+        command: scripts/build.sh VertexSample iOS
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index c249613f9e9..d2e4f95c0af 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -995,6 +995,7 @@ private extension URLRequest {
   }
 }
 
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
 class AppCheckInteropFake: NSObject, AppCheckInterop {
   /// The placeholder token value returned when an error occurs
   static let placeholderTokenValue = "placeholder-token"
diff --git a/scripts/build.sh b/scripts/build.sh
index f44d3d28a21..fcaba2a263d 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -490,6 +490,14 @@ case "$product-$platform-$method" in
       build
     ;;
 
+  VertexSample-*-*)
+    RunXcodebuild \
+      -project 'FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj' \
+      -scheme "GenerativeAISample" \
+      "${xcb_flags[@]}" \
+      build
+    ;;
+
   Sessions-*-integration)
     # Perform "pod install" to install the relevant dependencies
     # ./FirebaseSessions/generate_testapp.sh

From e8e48e7cd58aa8294d334fb94a23d093906726e2 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 25 Mar 2024 20:47:37 +0000
Subject: [PATCH 075/124] Fix `CitationMetadata` parsing in Vertex AI (#12626)

---
 .../Sources/GenerateContentResponse.swift     |  6 +-
 .../streaming-success-citations.txt           | 14 ++---
 .../unary-success-citations.json              | 63 +++++++++----------
 .../Tests/Unit/GenerativeModelTests.swift     | 25 +++++---
 4 files changed, 56 insertions(+), 52 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 03153da9ae7..4e02abb7f2c 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -155,6 +155,10 @@ extension CandidateResponse: Decodable {
 /// A collection of source attributions for a piece of content.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public struct CitationMetadata: Decodable {
+  enum CodingKeys: String, CodingKey {
+    case citationSources = "citations"
+  }
+
   /// A list of individual cited sources and the parts of the content to which they apply.
   public let citationSources: [Citation]
 }
@@ -172,7 +176,7 @@ public struct Citation: Decodable {
   public let uri: String
 
   /// The license the cited source work is distributed under.
-  public let license: String
+  public let license: String?
 }
 
 /// A value enumerating possible reasons for a model to terminate a content generation request.
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
index 4f50be7d32f..a530736b308 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
@@ -1,13 +1,9 @@
-data: {"candidates": [{"content": {"parts": [{"text": "Some information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Some information"}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": " More information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "  More information"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06632687,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.07477004,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048767097},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13695431,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059866417},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.046119746,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.036425155}]}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": ", Even more information"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Some information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07850098,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.039416388},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.08035747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04885778},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12273335,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059646938},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.053206205,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04099903}],"citationMetadata": {"citations": [{"startIndex": 31,"endIndex": 187,"uri": "https://www.example.com/citation-1"}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": " Some information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-1","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-2","license": ""}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.044183318},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.094176665,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0575992},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13660839,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08035747},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.060197048,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.046464667}],"citationMetadata": {"citations": [{"startIndex": 73,"endIndex": 248,"uri": "https://www.example.com/citation-2"},{"startIndex": 133,"endIndex": 272,"uri": "https://www.example.com/citation-3", "license": "mit"}]}}]}
 
-data: {"candidates": [{"content": {"parts": [{"text": "More information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-3","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-4","license": ""}]}}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "Even more information cited from an external source"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.example.com/citation-5","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.example.com/citation-6","license": ""}]}}]}
-
-data: {"candidates": [{"content": {"parts": [{"text": "Physics (YouTube Channel)"}],"role": "model"},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}],"citationMetadata": {"citationSources": [{"startIndex": 574,"endIndex": 705,"uri": "https://www.google.com","license": ""},{"startIndex": 899,"endIndex": 1026,"uri": "https://www.google.com","license": ""}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.12147716,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0647717},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.11858909,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.14866412,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08479541},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05470151,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045015533}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 163,"totalTokenCount": 172}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
index ac99fcf0944..a0dacc3e883 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
@@ -2,63 +2,58 @@
   "candidates": [
     {
       "content": {
+        "role": "model",
         "parts": [
           {
             "text": "Some information cited from an external source"
           }
-        ],
-        "role": "model"
+        ]
       },
       "finishReason": "STOP",
-      "index": 0,
       "safetyRatings": [
         {
-          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_HATE_SPEECH",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.16013464,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.074500255
         },
         {
-          "category": "HARM_CATEGORY_HATE_SPEECH",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.09687653,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.049313594
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
-          "probability": "NEGLIGIBLE"
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.16817278,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.09451043
         },
         {
-          "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-          "probability": "NEGLIGIBLE"
+          "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+          "probability": "NEGLIGIBLE",
+          "probabilityScore": 0.05023736,
+          "severity": "HARM_SEVERITY_NEGLIGIBLE",
+          "severityScore": 0.034553625
         }
       ],
       "citationMetadata": {
-        "citationSources": [
+        "citations": [
           {
-            "startIndex": 574,
-            "endIndex": 705,
-            "uri": "https://www.example.com/some-citation",
-            "license": ""
+            "startIndex": 179,
+            "endIndex": 366,
+            "uri": "https://www.example.com/some-citation"
           }
         ]
       }
     }
   ],
-  "promptFeedback": {
-    "safetyRatings": [
-      {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "probability": "NEGLIGIBLE"
-      },
-      {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "probability": "NEGLIGIBLE"
-      }
-    ]
+  "usageMetadata": {
+    "promptTokenCount": 11,
+    "candidatesTokenCount": 135,
+    "totalTokenCount": 146
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index d2e4f95c0af..41d95110340 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -107,9 +107,9 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(citationMetadata.citationSources.count, 1)
     let citationSource = try XCTUnwrap(citationMetadata.citationSources.first)
     XCTAssertEqual(citationSource.uri, "https://www.example.com/some-citation")
-    XCTAssertEqual(citationSource.startIndex, 574)
-    XCTAssertEqual(citationSource.endIndex, 705)
-    XCTAssertEqual(citationSource.license, "")
+    XCTAssertEqual(citationSource.startIndex, 179)
+    XCTAssertEqual(citationSource.endIndex, 366)
+    XCTAssertNil(citationSource.license)
   }
 
   func testGenerateContent_success_quoteReply() async throws {
@@ -669,21 +669,30 @@ final class GenerativeModelTests: XCTestCase {
       )
 
     let stream = model.generateContentStream("Hi")
-    var citations: [Citation] = []
+    var citations = [Citation]()
+    var responses = [GenerateContentResponse]()
     for try await content in stream {
+      responses.append(content)
       XCTAssertNotNil(content.text)
       let candidate = try XCTUnwrap(content.candidates.first)
-      XCTAssertEqual(candidate.finishReason, .stop)
       if let sources = candidate.citationMetadata?.citationSources {
         citations.append(contentsOf: sources)
       }
     }
 
-    XCTAssertEqual(citations.count, 8)
+    let lastCandidate = try XCTUnwrap(responses.last?.candidates.first)
+    XCTAssertEqual(lastCandidate.finishReason, .stop)
+    XCTAssertEqual(citations.count, 3)
     XCTAssertTrue(citations
-      .contains(where: { $0.startIndex == 574 && $0.endIndex == 705 && !$0.uri.isEmpty }))
+      .contains(where: {
+        $0.startIndex == 31 && $0.endIndex == 187 && $0
+          .uri == "https://www.example.com/citation-1" && $0.license == nil
+      }))
     XCTAssertTrue(citations
-      .contains(where: { $0.startIndex == 899 && $0.endIndex == 1026 && !$0.uri.isEmpty }))
+      .contains(where: {
+        $0.startIndex == 133 && $0.endIndex == 272 && $0
+          .uri == "https://www.example.com/citation-3" && $0.license == "mit"
+      }))
   }
 
   func testGenerateContentStream_appCheck_validToken() async throws {

From f5cdb6471fc1fb32ba10bec553416ab82b73b2bc Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 26 Mar 2024 13:24:29 -0700
Subject: [PATCH 076/124] Manage location on VertexAI instead of model (#12630)

---
 .../ViewModels/ConversationViewModel.swift    |  5 +--
 .../ViewModels/PhotoReasoningViewModel.swift  |  5 +--
 .../ViewModels/SummarizeViewModel.swift       |  5 +--
 FirebaseVertexAI/Sources/VertexAI.swift       | 40 +++++++++++--------
 .../Sources/VertexAIComponent.swift           |  6 +--
 .../Tests/Unit/VertexAIAPITests.swift         |  9 ++---
 6 files changed, 33 insertions(+), 37 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 05cbe11250f..81667be2514 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,10 +36,7 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.vertexAI().generativeModel(
-      modelName: "gemini-1.0-pro",
-      location: "us-central1"
-    )
+    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index 2f2ed88d4a1..b4880bdc825 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,10 +44,7 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI().generativeModel(
-      modelName: "gemini-1.0-pro-vision",
-      location: "us-central1"
-    )
+    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func reason() async {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index 0e3073d6da2..e3c78d09060 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,10 +32,7 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI().generativeModel(
-      modelName: "gemini-1.0-pro",
-      location: "us-central1"
-    )
+    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 64e76d3b4f5..2abb4abf7b0 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,26 +25,35 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
+  ///  - Parameter region: The region identifier, e.g., `us-central1`; see
+  ///     [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI() -> VertexAI {
+  public static func vertexAI(region: String) -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app)
+    return vertexAI(app: app, region: region)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
-  /// - Parameter app: The custom `FirebaseApp` used for initialization.
+  ///  - Parameters:
+  ///   - app: The custom `FirebaseApp` used for initialization.
+  ///   - region: The region identifier, e.g., `us-central1`; see
+  ///     [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp, region: String) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI()
+    return provider.vertexAI(region)
   }
 
   /// Initializes a generative model with the given parameters.
@@ -54,19 +63,15 @@ public class VertexAI: NSObject {
   ///     [Gemini
   ///     models](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#gemini-models)
   ///     for a list of supported model names.
-  ///   - location: The location identifier, e.g., `us-central1`; see
-  ///     [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
-  ///     for a list of supported locations.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
-  public func generativeModel(modelName: String, location: String,
+  public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
-    let modelResourceName = modelResourceName(modelName: modelName, location: location)
+    let modelResourceName = modelResourceName(modelName: modelName, region: region)
 
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
@@ -89,26 +94,29 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
-  init(app: FirebaseApp) {
+  private let region: String
+
+  init(app: FirebaseApp, region: String) {
     self.app = app
+    self.region = region
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
   }
 
-  private func modelResourceName(modelName: String, location: String) -> String {
+  private func modelResourceName(modelName: String, region: String) -> String {
     if modelName.contains("/") {
       return modelName
     }
     guard let projectID = app.options.projectID else {
       fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
     }
-    guard !location.isEmpty else {
+    guard !region.isEmpty else {
       fatalError("""
-      No location specified; see
+      No region specified; see
       https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
       list of available regions.
       """)
     }
 
-    return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
+    return "projects/\(projectID)/locations/\(region)/publishers/google/models/\(modelName)"
   }
 }
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index 1378f812626..a9b7aa669a6 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI() -> VertexAI
+  @objc func vertexAI(_ location: String) -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -64,7 +64,7 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   // MARK: - VertexAIProvider conformance
 
-  func vertexAI() -> VertexAI {
+  func vertexAI(_ region: String) -> VertexAI {
     os_unfair_lock_lock(&instancesLock)
 
     // Unlock before the function returns.
@@ -73,7 +73,7 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
     if let instance = instances[app.name] {
       return instance
     }
-    let newInstance = VertexAI(app: app)
+    let newInstance = VertexAI(app: app, region: region)
     instances[app.name] = newInstance
     return newInstance
   }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index a1fd27ab4c5..a4983eb8834 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -34,31 +34,28 @@ final class VertexAIAPITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Instantiate Vertex AI SDK - Default App
-    let vertexAI = VertexAI.vertexAI()
+    let vertexAI = VertexAI.vertexAI(region: "my-region")
 
     // Instantiate Vertex AI SDK - Custom App
-    let _ = VertexAI.vertexAI(app: app!)
+    let _ = VertexAI.vertexAI(app: app!, region: "my-region")
 
     // Permutations without optional arguments.
 
-    let _ = vertexAI.generativeModel(modelName: "gemini-1.0-pro", location: "us-central1")
+    let _ = vertexAI.generativeModel(modelName: "gemini-1.0-pro")
 
     let _ = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1",
       safetySettings: filters
     )
 
     let _ = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1",
       generationConfig: config
     )
 
     // All arguments passed.
     let genAI = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
-      location: "us-central1",
       generationConfig: config, // Optional
       safetySettings: filters // Optional
     )

From a67dfa698ad014b8748f94fb876fe289d9f87932 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 26 Mar 2024 17:58:47 -0700
Subject: [PATCH 077/124] VertexAI Component tests; fix instance generation for
 different regions (#12631)

---
 FirebaseVertexAI/Sources/VertexAI.swift       |   2 +-
 .../Sources/VertexAIComponent.swift           |   8 +-
 .../Tests/Unit/VertexComponentTests.swift     | 115 ++++++++++++++++++
 Package.swift                                 |   5 +-
 4 files changed, 124 insertions(+), 6 deletions(-)
 create mode 100644 FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 2abb4abf7b0..d1682745f98 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -94,7 +94,7 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
-  private let region: String
+  let region: String
 
   init(app: FirebaseApp, region: String) {
     self.app = app
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a9b7aa669a6..af8dbb56271 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI(_ location: String) -> VertexAI
+  @objc func vertexAI(_ region: String) -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -32,7 +32,7 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   /// The app associated with all `VertexAI` instances in this container.
   /// This is `unowned` instead of `weak` so it can be used without unwrapping in `vertexAI(...)`
-  private unowned let app: FirebaseApp
+  unowned let app: FirebaseApp
 
   /// A map of active  `VertexAI` instances for `app`, keyed by model resource names
   /// (e.g., "projects/my-project-id/locations/us-central1/publishers/google/models/gemini-pro").
@@ -70,11 +70,11 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
     // Unlock before the function returns.
     defer { os_unfair_lock_unlock(&instancesLock) }
 
-    if let instance = instances[app.name] {
+    if let instance = instances[region] {
       return instance
     }
     let newInstance = VertexAI(app: app, region: region)
-    instances[app.name] = newInstance
+    instances[region] = newInstance
     return newInstance
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
new file mode 100644
index 00000000000..7c046c5e949
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
@@ -0,0 +1,115 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+import FirebaseCore
+@testable import FirebaseVertexAI
+
+import SharedTestUtilities
+
+import XCTest
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+class VertexComponentTests: XCTestCase {
+  static var app: FirebaseApp!
+
+  override class func setUp() {
+    super.setUp()
+    if app == nil {
+      let options = FirebaseOptions(googleAppID: "0:0000000000000:ios:0000000000000000",
+                                    gcmSenderID: "00000000000000000-00000000000-000000000")
+      options.projectID = "myProjectID"
+      FirebaseApp.configure(options: options)
+      app = FirebaseApp(instanceWithName: "test", options: options)
+    }
+  }
+
+  // MARK: Interoperability Tests
+
+  /// Tests that the right number of components are being provided for the container.
+  func testComponentsBeingRegistered() throws {
+    let components = VertexAIComponent.componentsToRegister()
+    XCTAssert(components.count == 1)
+  }
+
+  /// Tests that a vertex instance can be created properly by the VertexAIComponent.
+  func testVertexInstanceCreation() throws {
+    let app = try XCTUnwrap(VertexComponentTests.app)
+    let component = VertexAIComponent(app: app)
+    let vertex = component.vertexAI("my-region")
+    XCTAssertNotNil(vertex)
+  }
+
+  /// Tests that the component container caches instances of VertexAIComponent.
+  func testMultipleComponentInstancesCreated() throws {
+    let registrants = NSMutableSet(array: [VertexAIComponent.self])
+    let container = FirebaseComponentContainer(
+      app: VertexComponentTests.app,
+      registrants: registrants
+    )
+
+    let provider1 = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                             in: container)
+    XCTAssertNotNil(provider1)
+
+    let provider2 = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                             in: container)
+    XCTAssertNotNil(provider2)
+
+    // Ensure they're the same instance.
+    XCTAssert(provider1 === provider2)
+  }
+
+  /// Tests that instances of vertex created are different.
+  func testMultipleVertexInstancesCreated() throws {
+    let app = try XCTUnwrap(VertexComponentTests.app)
+    let registrants = NSMutableSet(array: [VertexAIComponent.self])
+    let container = FirebaseComponentContainer(app: app, registrants: registrants)
+
+    let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
+                                                            in: container)
+    XCTAssertNotNil(provider)
+
+    let vertex1 = provider?.vertexAI("randomRegion")
+    let vertex2 = provider?.vertexAI("randomRegion")
+    XCTAssertNotNil(vertex1)
+
+    // Ensure they're the same instance.
+    XCTAssert(vertex1 === vertex2)
+
+    let vertex3 = provider?.vertexAI("differentRegion")
+    XCTAssertNotNil(vertex3)
+
+    XCTAssert(vertex1 !== vertex3)
+  }
+
+  /// Test that vertex instances get deallocated.
+  func testVertexLifecycle() throws {
+    weak var weakApp: FirebaseApp?
+    weak var weakVertex: VertexAI?
+    try autoreleasepool {
+      let options = FirebaseOptions(googleAppID: "0:0000000000000:ios:0000000000000000",
+                                    gcmSenderID: "00000000000000000-00000000000-000000000")
+      options.projectID = "myProjectID"
+      let app1 = FirebaseApp(instanceWithName: "transitory app", options: options)
+      weakApp = try XCTUnwrap(app1)
+      let vertex = VertexAI(app: app1, region: "transitory bucket")
+      weakVertex = vertex
+      XCTAssertNotNil(weakVertex)
+    }
+    XCTAssertNil(weakApp)
+    XCTAssertNil(weakVertex)
+  }
+}
diff --git a/Package.swift b/Package.swift
index 15929c130cf..19752536507 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1376,11 +1376,14 @@ let package = Package(
     ),
     .testTarget(
       name: "FirebaseVertexAIUnit",
-      dependencies: ["FirebaseVertexAI"],
+      dependencies: ["FirebaseVertexAI", "SharedTestUtilities"],
       path: "FirebaseVertexAI/Tests/Unit",
       resources: [
         .process("CountTokenResponses"),
         .process("GenerateContentResponses"),
+      ],
+      cSettings: [
+        .headerSearchPath("../../../"),
       ]
     ),
   ] + firestoreTargets(),

From d4c036d60c58b1b0c458abfe07a242baf2fdbe3b Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 14:34:49 +0000
Subject: [PATCH 078/124] Add function calling support in Vertex AI (#12633)

---
 FirebaseVertexAI/Sources/Chat.swift           |   2 +-
 .../Sources/FunctionCalling.swift             | 235 ++++++++++++++++++
 .../Sources/GenerateContentRequest.swift      |   2 +
 .../Sources/GenerativeModel.swift             |   8 +
 FirebaseVertexAI/Sources/JSONValue.swift      |  96 +++++++
 FirebaseVertexAI/Sources/ModelContent.swift   |  16 +-
 FirebaseVertexAI/Sources/VertexAI.swift       |   3 +
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |   1 +
 ...success-function-call-empty-arguments.json |  18 ++
 ...ry-success-function-call-no-arguments.json |  19 ++
 ...-success-function-call-with-arguments.json |  22 ++
 .../Tests/Unit/GenerativeModelTests.swift     |  79 ++++++
 .../Tests/Unit/JSONValueTests.swift           | 145 +++++++++++
 .../Tests/Unit/VertexComponentTests.swift     |   2 +-
 14 files changed, 645 insertions(+), 3 deletions(-)
 create mode 100644 FirebaseVertexAI/Sources/FunctionCalling.swift
 create mode 100644 FirebaseVertexAI/Sources/JSONValue.swift
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json
 create mode 100644 FirebaseVertexAI/Tests/Unit/JSONValueTests.swift

diff --git a/FirebaseVertexAI/Sources/Chat.swift b/FirebaseVertexAI/Sources/Chat.swift
index c7cfb859fa8..e443e18bed8 100644
--- a/FirebaseVertexAI/Sources/Chat.swift
+++ b/FirebaseVertexAI/Sources/Chat.swift
@@ -153,7 +153,7 @@ public class Chat {
         case let .text(str):
           combinedText += str
 
-        case .data(mimetype: _, _):
+        case .data, .functionCall, .functionResponse:
           // Don't combine it, just add to the content. If there's any text pending, add that as
           // a part.
           if !combinedText.isEmpty {
diff --git a/FirebaseVertexAI/Sources/FunctionCalling.swift b/FirebaseVertexAI/Sources/FunctionCalling.swift
new file mode 100644
index 00000000000..7e4d84d857e
--- /dev/null
+++ b/FirebaseVertexAI/Sources/FunctionCalling.swift
@@ -0,0 +1,235 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A predicted function call returned from the model.
+public struct FunctionCall: Equatable, Encodable {
+  /// The name of the function to call.
+  public let name: String
+
+  /// The function parameters and values.
+  public let args: JSONObject
+}
+
+/// A `Schema` object allows the definition of input and output data types.
+///
+/// These types can be objects, but also primitives and arrays. Represents a select subset of an
+/// [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema).
+public class Schema: Encodable {
+  /// The data type.
+  let type: DataType
+
+  /// The format of the data.
+  let format: String?
+
+  /// A brief description of the parameter.
+  let description: String?
+
+  /// Indicates if the value may be null.
+  let nullable: Bool?
+
+  /// Possible values of the element of type ``DataType/string`` with "enum" format.
+  let enumValues: [String]?
+
+  /// Schema of the elements of type ``DataType/array``.
+  let items: Schema?
+
+  /// Properties of type ``DataType/object``.
+  let properties: [String: Schema]?
+
+  /// Required properties of type ``DataType/object``.
+  let requiredProperties: [String]?
+
+  enum CodingKeys: String, CodingKey {
+    case type
+    case format
+    case description
+    case nullable
+    case enumValues = "enum"
+    case items
+    case properties
+    case requiredProperties = "required"
+  }
+
+  /// Constructs a new `Schema`.
+  ///
+  /// - Parameters:
+  ///   - type: The data type.
+  ///   - format: The format of the data; used only for primitive datatypes.
+  ///     Supported formats:
+  ///     - ``DataType/integer``: int32, int64
+  ///     - ``DataType/number``: float, double
+  ///     - ``DataType/string``: enum
+  ///   - description: A brief description of the parameter; may be formatted as Markdown.
+  ///   - nullable: Indicates if the value may be null.
+  ///   - enumValues: Possible values of the element of type ``DataType/string`` with "enum" format.
+  ///     For example, an enum `Direction` may be defined as `["EAST", NORTH", "SOUTH", "WEST"]`.
+  ///   - items: Schema of the elements of type ``DataType/array``.
+  ///   - properties: Properties of type ``DataType/object``.
+  ///   - requiredProperties: Required properties of type ``DataType/object``.
+  public init(type: DataType, format: String? = nil, description: String? = nil,
+              nullable: Bool? = nil,
+              enumValues: [String]? = nil, items: Schema? = nil,
+              properties: [String: Schema]? = nil,
+              requiredProperties: [String]? = nil) {
+    self.type = type
+    self.format = format
+    self.description = description
+    self.nullable = nullable
+    self.enumValues = enumValues
+    self.items = items
+    self.properties = properties
+    self.requiredProperties = requiredProperties
+  }
+}
+
+/// A data type.
+///
+/// Contains the set of OpenAPI [data types](https://spec.openapis.org/oas/v3.0.3#data-types).
+public enum DataType: String, Encodable {
+  /// A `String` type.
+  case string = "STRING"
+
+  /// A floating-point number type.
+  case number = "NUMBER"
+
+  /// An integer type.
+  case integer = "INTEGER"
+
+  /// A boolean type.
+  case boolean = "BOOLEAN"
+
+  /// An array type.
+  case array = "ARRAY"
+
+  /// An object type.
+  case object = "OBJECT"
+}
+
+/// Structured representation of a function declaration.
+///
+/// This `FunctionDeclaration` is a representation of a block of code that can be used as a ``Tool``
+/// by the model and executed by the client.
+public struct FunctionDeclaration {
+  /// The name of the function.
+  let name: String
+
+  /// A brief description of the function.
+  let description: String
+
+  /// Describes the parameters to this function; must be of type ``DataType/object``.
+  let parameters: Schema?
+
+  /// Constructs a new `FunctionDeclaration`.
+  ///
+  /// - Parameters:
+  ///   - name: The name of the function; must be a-z, A-Z, 0-9, or contain underscores and dashes,
+  ///   with a maximum length of 63.
+  ///   - description: A brief description of the function.
+  ///   - parameters: Describes the parameters to this function; the keys are parameter names and
+  ///   the values are ``Schema`` objects describing them.
+  ///   - requiredParameters: A list of required parameters by name.
+  public init(name: String, description: String, parameters: [String: Schema]?,
+              requiredParameters: [String]?) {
+    self.name = name
+    self.description = description
+    self.parameters = Schema(
+      type: .object,
+      properties: parameters,
+      requiredProperties: requiredParameters
+    )
+  }
+}
+
+/// Helper tools that the model may use to generate response.
+///
+/// A `Tool` is a piece of code that enables the system to interact with external systems to
+/// perform an action, or set of actions, outside of knowledge and scope of the model.
+public struct Tool: Encodable {
+  /// A list of `FunctionDeclarations` available to the model.
+  let functionDeclarations: [FunctionDeclaration]?
+
+  /// Constructs a new `Tool`.
+  ///
+  /// - Parameters:
+  ///   - functionDeclarations: A list of `FunctionDeclarations` available to the model that can be
+  ///   used for function calling.
+  ///   The model or system does not execute the function. Instead the defined function may be
+  ///   returned as a ``FunctionCall`` in ``ModelContent/Part/functionCall(_:)`` with arguments to
+  ///   the client side for execution. The model may decide to call a subset of these functions by
+  ///   populating ``FunctionCall`` in the response. The next conversation turn may contain a
+  ///   ``FunctionResponse`` in ``ModelContent/Part/functionResponse(_:)`` with the
+  ///   ``ModelContent/role`` "function", providing generation context for the next model turn.
+  public init(functionDeclarations: [FunctionDeclaration]?) {
+    self.functionDeclarations = functionDeclarations
+  }
+}
+
+/// Result output from a ``FunctionCall``.
+///
+/// Contains a string representing the `FunctionDeclaration.name` and a structured JSON object
+/// containing any output from the function is used as context to the model. This should contain the
+/// result of a ``FunctionCall`` made based on model prediction.
+public struct FunctionResponse: Equatable, Encodable {
+  /// The name of the function that was called.
+  let name: String
+
+  /// The function's response.
+  let response: JSONObject
+
+  /// Constructs a new `FunctionResponse`.
+  ///
+  /// - Parameters:
+  ///   - name: The name of the function that was called.
+  ///   - response: The function's response.
+  public init(name: String, response: JSONObject) {
+    self.name = name
+    self.response = response
+  }
+}
+
+// MARK: - Codable Conformance
+
+extension FunctionCall: Decodable {
+  enum CodingKeys: CodingKey {
+    case name
+    case args
+  }
+
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    name = try container.decode(String.self, forKey: .name)
+    if let args = try container.decodeIfPresent(JSONObject.self, forKey: .args) {
+      self.args = args
+    } else {
+      args = JSONObject()
+    }
+  }
+}
+
+extension FunctionDeclaration: Encodable {
+  enum CodingKeys: String, CodingKey {
+    case name
+    case description
+    case parameters
+  }
+
+  public func encode(to encoder: Encoder) throws {
+    var container = encoder.container(keyedBy: CodingKeys.self)
+    try container.encode(name, forKey: .name)
+    try container.encode(description, forKey: .description)
+    try container.encode(parameters, forKey: .parameters)
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
index 417260bb700..535ac4da3ae 100644
--- a/FirebaseVertexAI/Sources/GenerateContentRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -21,6 +21,7 @@ struct GenerateContentRequest {
   let contents: [ModelContent]
   let generationConfig: GenerationConfig?
   let safetySettings: [SafetySetting]?
+  let tools: [Tool]?
   let isStreaming: Bool
   let options: RequestOptions
 }
@@ -31,6 +32,7 @@ extension GenerateContentRequest: Encodable {
     case contents
     case generationConfig
     case safetySettings
+    case tools
   }
 }
 
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index aad2b3e99c0..7eab63a1d7a 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -34,6 +34,9 @@ public final class GenerativeModel {
   /// The safety settings to be used for prompts.
   let safetySettings: [SafetySetting]?
 
+  /// A list of tools the model may use to generate the next response.
+  let tools: [Tool]?
+
   /// Configuration parameters for sending requests to the backend.
   let requestOptions: RequestOptions
 
@@ -45,12 +48,14 @@ public final class GenerativeModel {
   ///   - apiKey: The API key for your project.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
        apiKey: String,
        generationConfig: GenerationConfig? = nil,
        safetySettings: [SafetySetting]? = nil,
+       tools: [Tool]?,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
        urlSession: URLSession = .shared) {
@@ -62,6 +67,7 @@ public final class GenerativeModel {
     )
     self.generationConfig = generationConfig
     self.safetySettings = safetySettings
+    self.tools = tools
     self.requestOptions = requestOptions
 
     Logging.default.info("""
@@ -107,6 +113,7 @@ public final class GenerativeModel {
                                                               contents: content(),
                                                               generationConfig: generationConfig,
                                                               safetySettings: safetySettings,
+                                                              tools: tools,
                                                               isStreaming: false,
                                                               options: requestOptions)
       response = try await generativeAIService.loadRequest(request: generateContentRequest)
@@ -178,6 +185,7 @@ public final class GenerativeModel {
                                                         contents: evaluatedContent,
                                                         generationConfig: generationConfig,
                                                         safetySettings: safetySettings,
+                                                        tools: tools,
                                                         isStreaming: true,
                                                         options: requestOptions)
 
diff --git a/FirebaseVertexAI/Sources/JSONValue.swift b/FirebaseVertexAI/Sources/JSONValue.swift
new file mode 100644
index 00000000000..5ce52cd2af9
--- /dev/null
+++ b/FirebaseVertexAI/Sources/JSONValue.swift
@@ -0,0 +1,96 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+
+/// A collection of name-value pairs representing a JSON object.
+///
+/// This may be decoded from, or encoded to, a
+/// [`google.protobuf.Struct`](https://protobuf.dev/reference/protobuf/google.protobuf/#struct).
+public typealias JSONObject = [String: JSONValue]
+
+/// Represents a value in one of JSON's data types.
+///
+/// This may be decoded from, or encoded to, a
+/// [`google.protobuf.Value`](https://protobuf.dev/reference/protobuf/google.protobuf/#value).
+public enum JSONValue {
+  /// A `null` value.
+  case null
+
+  /// A numeric value.
+  case number(Double)
+
+  /// A string value.
+  case string(String)
+
+  /// A boolean value.
+  case bool(Bool)
+
+  /// A JSON object.
+  case object(JSONObject)
+
+  /// An array of `JSONValue`s.
+  case array([JSONValue])
+}
+
+extension JSONValue: Decodable {
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.singleValueContainer()
+    if container.decodeNil() {
+      self = .null
+    } else if let numberValue = try? container.decode(Double.self) {
+      self = .number(numberValue)
+    } else if let stringValue = try? container.decode(String.self) {
+      self = .string(stringValue)
+    } else if let boolValue = try? container.decode(Bool.self) {
+      self = .bool(boolValue)
+    } else if let objectValue = try? container.decode(JSONObject.self) {
+      self = .object(objectValue)
+    } else if let arrayValue = try? container.decode([JSONValue].self) {
+      self = .array(arrayValue)
+    } else {
+      throw DecodingError.dataCorruptedError(
+        in: container,
+        debugDescription: "Failed to decode JSON value."
+      )
+    }
+  }
+}
+
+extension JSONValue: Encodable {
+  public func encode(to encoder: Encoder) throws {
+    var container = encoder.singleValueContainer()
+    switch self {
+    case .null:
+      try container.encodeNil()
+    case let .number(numberValue):
+      // Convert to `Decimal` before encoding for consistent floating-point serialization across
+      // platforms. E.g., `Double` serializes 3.14159 as 3.1415899999999999 in some cases and
+      // 3.14159 in others. See
+      // https://forums.swift.org/t/jsonencoder-encodable-floating-point-rounding-error/41390/4 for
+      // more details.
+      try container.encode(Decimal(numberValue))
+    case let .string(stringValue):
+      try container.encode(stringValue)
+    case let .bool(boolValue):
+      try container.encode(boolValue)
+    case let .object(objectValue):
+      try container.encode(objectValue)
+    case let .array(arrayValue):
+      try container.encode(arrayValue)
+    }
+  }
+}
+
+extension JSONValue: Equatable {}
diff --git a/FirebaseVertexAI/Sources/ModelContent.swift b/FirebaseVertexAI/Sources/ModelContent.swift
index 44648c57852..136fc1debe3 100644
--- a/FirebaseVertexAI/Sources/ModelContent.swift
+++ b/FirebaseVertexAI/Sources/ModelContent.swift
@@ -25,6 +25,8 @@ public struct ModelContent: Codable, Equatable {
     enum CodingKeys: String, CodingKey {
       case text
       case inlineData
+      case functionCall
+      case functionResponse
     }
 
     enum InlineDataKeys: String, CodingKey {
@@ -38,6 +40,12 @@ public struct ModelContent: Codable, Equatable {
     /// Data with a specified media type. Not all media types may be supported by the AI model.
     case data(mimetype: String, Data)
 
+    /// A predicted function call returned from the model.
+    case functionCall(FunctionCall)
+
+    /// A response to a function call.
+    case functionResponse(FunctionResponse)
+
     // MARK: Convenience Initializers
 
     /// Convenience function for populating a Part with JPEG data.
@@ -64,6 +72,10 @@ public struct ModelContent: Codable, Equatable {
         )
         try inlineDataContainer.encode(mimetype, forKey: .mimeType)
         try inlineDataContainer.encode(bytes, forKey: .bytes)
+      case let .functionCall(functionCall):
+        try container.encode(functionCall, forKey: .functionCall)
+      case let .functionResponse(functionResponse):
+        try container.encode(functionResponse, forKey: .functionResponse)
       }
     }
 
@@ -79,10 +91,12 @@ public struct ModelContent: Codable, Equatable {
         let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
         let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
         self = .data(mimetype: mimetype, bytes)
+      } else if values.contains(.functionCall) {
+        self = try .functionCall(values.decode(FunctionCall.self, forKey: .functionCall))
       } else {
         throw DecodingError.dataCorrupted(.init(
           codingPath: [CodingKeys.text, CodingKeys.inlineData],
-          debugDescription: "Neither text or inline data was found."
+          debugDescription: "No text, inline data or function call was found."
         ))
       }
     }
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index d1682745f98..a0df2ab015e 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -65,10 +65,12 @@ public class VertexAI: NSObject {
   ///     for a list of supported model names.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
+  ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
+                              tools: [Tool]? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
     let modelResourceName = modelResourceName(modelName: modelName, region: region)
@@ -82,6 +84,7 @@ public class VertexAI: NSObject {
       apiKey: apiKey,
       generationConfig: generationConfig,
       safetySettings: safetySettings,
+      tools: tools,
       requestOptions: requestOptions,
       appCheck: appCheck
     )
diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 5a7868fbe7a..20897a8787b 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -50,6 +50,7 @@ final class ChatTests: XCTestCase {
     let model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
       urlSession: urlSession
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json
new file mode 100644
index 00000000000..dc0b75ad930
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-empty-arguments.json
@@ -0,0 +1,18 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "functionCall": {
+              "name": "current_time"
+            }
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json
new file mode 100644
index 00000000000..05f4f4d58f9
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-no-arguments.json
@@ -0,0 +1,19 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "functionCall": {
+              "name": "current_time",
+              "args": {}
+            }
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json
new file mode 100644
index 00000000000..025735a3a39
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-function-call-with-arguments.json
@@ -0,0 +1,22 @@
+{
+  "candidates": [
+    {
+      "content": {
+        "parts": [
+          {
+            "functionCall": {
+              "name": "sum",
+              "args": {
+                "y": 5,
+                "x": 4
+              }
+            }
+          }
+        ],
+        "role": "model"
+      },
+      "finishReason": "STOP",
+      "index": 0
+    }
+  ]
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 41d95110340..9a1210c123c 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -37,6 +37,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
       urlSession: urlSession
@@ -165,6 +166,7 @@ final class GenerativeModelTests: XCTestCase {
       // Model name is prefixed with "models/".
       name: "models/test-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
       urlSession: urlSession
@@ -173,11 +175,79 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_success_functionCall_emptyArguments() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-function-call-empty-arguments",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    guard case let .functionCall(functionCall) = part else {
+      XCTFail("Part is not a FunctionCall.")
+      return
+    }
+    XCTAssertEqual(functionCall.name, "current_time")
+    XCTAssertTrue(functionCall.args.isEmpty)
+  }
+
+  func testGenerateContent_success_functionCall_noArguments() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-function-call-no-arguments",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    guard case let .functionCall(functionCall) = part else {
+      XCTFail("Part is not a FunctionCall.")
+      return
+    }
+    XCTAssertEqual(functionCall.name, "current_time")
+    XCTAssertTrue(functionCall.args.isEmpty)
+  }
+
+  func testGenerateContent_success_functionCall_withArguments() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-function-call-with-arguments",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    XCTAssertEqual(response.candidates.count, 1)
+    let candidate = try XCTUnwrap(response.candidates.first)
+    XCTAssertEqual(candidate.content.parts.count, 1)
+    let part = try XCTUnwrap(candidate.content.parts.first)
+    guard case let .functionCall(functionCall) = part else {
+      XCTFail("Part is not a FunctionCall.")
+      return
+    }
+    XCTAssertEqual(functionCall.name, "sum")
+    XCTAssertEqual(functionCall.args.count, 2)
+    let argX = try XCTUnwrap(functionCall.args["x"])
+    XCTAssertEqual(argX, .number(4))
+    let argY = try XCTUnwrap(functionCall.args["y"])
+    XCTAssertEqual(argY, .number(5))
+  }
+
   func testGenerateContent_appCheck_validToken() async throws {
     let appCheckToken = "test-valid-token"
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
       urlSession: urlSession
@@ -196,6 +266,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
       urlSession: urlSession
@@ -493,6 +564,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
       urlSession: urlSession
@@ -700,6 +772,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
       urlSession: urlSession
@@ -719,6 +792,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
       urlSession: urlSession
@@ -836,6 +910,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
       urlSession: urlSession
@@ -894,6 +969,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: "my-model",
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
       urlSession: urlSession
@@ -913,6 +989,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: modelName,
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil
     )
@@ -926,6 +1003,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: modelResourceName,
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil
     )
@@ -939,6 +1017,7 @@ final class GenerativeModelTests: XCTestCase {
     model = GenerativeModel(
       name: tunedModelResourceName,
       apiKey: "API_KEY",
+      tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil
     )
diff --git a/FirebaseVertexAI/Tests/Unit/JSONValueTests.swift b/FirebaseVertexAI/Tests/Unit/JSONValueTests.swift
new file mode 100644
index 00000000000..74196230887
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/JSONValueTests.swift
@@ -0,0 +1,145 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+import XCTest
+
+@testable import FirebaseVertexAI
+
+final class JSONValueTests: XCTestCase {
+  let decoder = JSONDecoder()
+  let encoder = JSONEncoder()
+
+  let numberKey = "pi"
+  let numberValue = 3.14159
+  let numberValueEncoded = "3.14159"
+  let stringKey = "hello"
+  let stringValue = "Hello, world!"
+
+  override func setUp() {
+    encoder.outputFormatting = .sortedKeys
+  }
+
+  func testDecodeNull() throws {
+    let jsonData = try XCTUnwrap("null".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .null)
+  }
+
+  func testDecodeNumber() throws {
+    let jsonData = try XCTUnwrap("\(numberValue)".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .number(numberValue))
+  }
+
+  func testDecodeString() throws {
+    let jsonData = try XCTUnwrap("\"\(stringValue)\"".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .string(stringValue))
+  }
+
+  func testDecodeBool() throws {
+    let expectedBool = true
+    let jsonData = try XCTUnwrap("\(expectedBool)".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .bool(expectedBool))
+  }
+
+  func testDecodeObject() throws {
+    let expectedObject: JSONObject = [
+      numberKey: .number(numberValue),
+      stringKey: .string(stringValue),
+    ]
+    let json = """
+    {
+      "\(numberKey)": \(numberValue),
+      "\(stringKey)": "\(stringValue)"
+    }
+    """
+    let jsonData = try XCTUnwrap(json.data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .object(expectedObject))
+  }
+
+  func testDecodeArray() throws {
+    let expectedArray: [JSONValue] = [.null, .number(numberValue)]
+    let jsonData = try XCTUnwrap("[ null, \(numberValue) ]".data(using: .utf8))
+
+    let jsonObject = try XCTUnwrap(decoder.decode(JSONValue.self, from: jsonData))
+
+    XCTAssertEqual(jsonObject, .array(expectedArray))
+  }
+
+  func testEncodeNull() throws {
+    let jsonData = try encoder.encode(JSONValue.null)
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "null")
+  }
+
+  func testEncodeNumber() throws {
+    let jsonData = try encoder.encode(JSONValue.number(numberValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "\(numberValue)")
+  }
+
+  func testEncodeString() throws {
+    let jsonData = try encoder.encode(JSONValue.string(stringValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "\"\(stringValue)\"")
+  }
+
+  func testEncodeBool() throws {
+    let boolValue = true
+
+    let jsonData = try encoder.encode(JSONValue.bool(boolValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "\(boolValue)")
+  }
+
+  func testEncodeObject() throws {
+    let objectValue: JSONObject = [
+      numberKey: .number(numberValue),
+      stringKey: .string(stringValue),
+    ]
+
+    let jsonData = try encoder.encode(JSONValue.object(objectValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(
+      json,
+      "{\"\(stringKey)\":\"\(stringValue)\",\"\(numberKey)\":\(numberValueEncoded)}"
+    )
+  }
+
+  func testEncodeArray() throws {
+    let arrayValue: [JSONValue] = [.null, .number(numberValue)]
+
+    let jsonData = try encoder.encode(JSONValue.array(arrayValue))
+
+    let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
+    XCTAssertEqual(json, "[null,\(numberValueEncoded)]")
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
index 7c046c5e949..e28b747b65c 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
@@ -105,7 +105,7 @@ class VertexComponentTests: XCTestCase {
       options.projectID = "myProjectID"
       let app1 = FirebaseApp(instanceWithName: "transitory app", options: options)
       weakApp = try XCTUnwrap(app1)
-      let vertex = VertexAI(app: app1, region: "transitory bucket")
+      let vertex = VertexAI(app: app1, region: "transitory region")
       weakVertex = vertex
       XCTAssertNotNil(weakVertex)
     }

From 336ccf632c9f4e74901516225c90bcf6aa730c40 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 16:35:36 +0000
Subject: [PATCH 079/124] Add `-Preview` suffix to `FirebaseVertexAI` library
 name (#12641)

---
 .../project.pbxproj                           | 100 +++++++++---------
 Package.swift                                 |   2 +-
 2 files changed, 51 insertions(+), 51 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 1b4a69cafb0..9efade7342d 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -7,10 +7,14 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
-		869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AA2B86BE3100482873 /* FirebaseVertexAI */; };
-		869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AC2B86BEA100482873 /* FirebaseVertexAI */; };
-		869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200AE2B86BFBC00482873 /* FirebaseVertexAI */; };
-		869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */ = {isa = PBXBuildFile; productRef = 869200B02B86BFF500482873 /* FirebaseVertexAI */; };
+		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
+		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
+		868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33692BB4771900304BB1 /* FirebaseAppCheck */; };
+		868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */; };
+		868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336D2BB4773600304BB1 /* FirebaseAppCheck */; };
+		868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */; };
+		868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33712BB4775D00304BB1 /* FirebaseAppCheck */; };
+		868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
 		869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
 		869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
@@ -56,10 +60,6 @@
 		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
 		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
 		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
-		8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */; };
-		8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */; };
-		8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */; };
-		8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 8D589E302BA8EED80049BFCA /* FirebaseAppCheck */; };
 		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
 /* End PBXBuildFile section */
 
@@ -101,9 +101,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				8D589E312BA8EED80049BFCA /* FirebaseAppCheck in Frameworks */,
+				868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
-				869200AB2B86BE3100482873 /* FirebaseVertexAI in Frameworks */,
+				868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */,
 				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -112,9 +112,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
+				868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */,
 				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
-				8D589E2F2BA8EEBD0049BFCA /* FirebaseAppCheck in Frameworks */,
-				869200B12B86BFF500482873 /* FirebaseVertexAI in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
@@ -122,9 +122,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				8D589E2B2BA8E87A0049BFCA /* FirebaseAppCheck in Frameworks */,
+				868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
-				869200AF2B86BFBC00482873 /* FirebaseVertexAI in Frameworks */,
+				868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */,
 				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -133,9 +133,9 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				8D589E2D2BA8EEA30049BFCA /* FirebaseAppCheck in Frameworks */,
+				868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
-				869200AD2B86BEA100482873 /* FirebaseVertexAI in Frameworks */,
+				868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */,
 				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -342,8 +342,8 @@
 			packageProductDependencies = (
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
-				869200AA2B86BE3100482873 /* FirebaseVertexAI */,
-				8D589E302BA8EED80049BFCA /* FirebaseAppCheck */,
+				868A33652BB476FA00304BB1 /* FirebaseAppCheck */,
+				868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAISample;
 			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
@@ -364,8 +364,8 @@
 			name = GenerativeAITextSample;
 			packageProductDependencies = (
 				88209C232B0FBE1700F64795 /* MarkdownUI */,
-				869200B02B86BFF500482873 /* FirebaseVertexAI */,
-				8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */,
+				868A33692BB4771900304BB1 /* FirebaseAppCheck */,
+				868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAITextSample;
 			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
@@ -387,8 +387,8 @@
 			packageProductDependencies = (
 				88B8A91D2B0FC55100424728 /* MarkdownUI */,
 				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
-				869200AE2B86BFBC00482873 /* FirebaseVertexAI */,
-				8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */,
+				868A336D2BB4773600304BB1 /* FirebaseAppCheck */,
+				868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAIMultimodalSample;
 			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
@@ -410,8 +410,8 @@
 			packageProductDependencies = (
 				88D9474C2B14F27E008B5580 /* MarkdownUI */,
 				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
-				869200AC2B86BEA100482873 /* FirebaseVertexAI */,
-				8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */,
+				868A33712BB4775D00304BB1 /* FirebaseAppCheck */,
+				868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = ChatSample;
 			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
@@ -453,7 +453,7 @@
 			packageReferences = (
 				88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */,
 				DEA09AC32B1FCE22001962D9 /* XCRemoteSwiftPackageReference "NetworkImage" */,
-				869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */,
+				868A33642BB476FA00304BB1 /* XCLocalSwiftPackageReference "../.." */,
 			);
 			productRefGroup = 8848C8302B0D04BC007B434F /* Products */;
 			projectDirPath = "";
@@ -985,7 +985,7 @@
 /* End XCConfigurationList section */
 
 /* Begin XCLocalSwiftPackageReference section */
-		869200A92B86BE3100482873 /* XCLocalSwiftPackageReference "../.." */ = {
+		868A33642BB476FA00304BB1 /* XCLocalSwiftPackageReference "../.." */ = {
 			isa = XCLocalSwiftPackageReference;
 			relativePath = ../..;
 		};
@@ -1011,21 +1011,37 @@
 /* End XCRemoteSwiftPackageReference section */
 
 /* Begin XCSwiftPackageProductDependency section */
-		869200AA2B86BE3100482873 /* FirebaseVertexAI */ = {
+		868A33652BB476FA00304BB1 /* FirebaseAppCheck */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = FirebaseAppCheck;
+		};
+		868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = "FirebaseVertexAI-Preview";
+		};
+		868A33692BB4771900304BB1 /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
 		};
-		869200AC2B86BEA100482873 /* FirebaseVertexAI */ = {
+		868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = "FirebaseVertexAI-Preview";
 		};
-		869200AE2B86BFBC00482873 /* FirebaseVertexAI */ = {
+		868A336D2BB4773600304BB1 /* FirebaseAppCheck */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = FirebaseAppCheck;
+		};
+		868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = "FirebaseVertexAI-Preview";
+		};
+		868A33712BB4775D00304BB1 /* FirebaseAppCheck */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseAppCheck;
 		};
-		869200B02B86BFF500482873 /* FirebaseVertexAI */ = {
+		868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */ = {
 			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseVertexAI;
+			productName = "FirebaseVertexAI-Preview";
 		};
 		88209C232B0FBE1700F64795 /* MarkdownUI */ = {
 			isa = XCSwiftPackageProductDependency;
@@ -1055,22 +1071,6 @@
 			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
 			productName = MarkdownUI;
 		};
-		8D589E2A2BA8E87A0049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		8D589E2C2BA8EEA30049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		8D589E2E2BA8EEBD0049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		8D589E302BA8EED80049BFCA /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
 		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = GenerativeAIUIComponents;
diff --git a/Package.swift b/Package.swift
index 19752536507..29aa768d34d 100644
--- a/Package.swift
+++ b/Package.swift
@@ -138,7 +138,7 @@ let package = Package(
       targets: ["FirebaseStorage"]
     ),
     .library(
-      name: "FirebaseVertexAI",
+      name: "FirebaseVertexAI-Preview",
       targets: ["FirebaseVertexAI"]
     ),
   ],

From 1089da948e88da8f4e0f3a203ddb7c81a955b082 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 20:33:37 +0000
Subject: [PATCH 080/124] Fix model name in Vertex AI multi-modal sample
 (#12645)

---
 .../ViewModels/PhotoReasoningViewModel.swift                   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index b4880bdc825..1fb0097a808 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,8 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    let vertexAI = VertexAI.vertexAI(region: "us-central1")
+    model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
   func reason() async {

From 07cdaa9a86dbcb6ec203d33d68e275d51fc34a7f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 20:37:29 +0000
Subject: [PATCH 081/124] Add `totalBillableCharacters` to
 `CountTokensResponse` in Vertex AI (#12646)

---
 FirebaseVertexAI/Sources/CountTokensRequest.swift              | 3 +++
 .../Tests/Unit/CountTokenResponses/success-total-tokens.json   | 3 ++-
 FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift         | 2 ++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
index 0a58d40acc0..d294212658c 100644
--- a/FirebaseVertexAI/Sources/CountTokensRequest.swift
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -42,4 +42,7 @@ extension CountTokensRequest: GenerativeAIRequest {
 public struct CountTokensResponse: Decodable {
   /// The total number of tokens in the input given to the model as a prompt.
   public let totalTokens: Int
+
+  /// The total number of billable characters in the input given to the model as a prompt.
+  public let totalBillableCharacters: Int
 }
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
index 0bccd9e7c3d..d2ad6e4ff30 100644
--- a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-total-tokens.json
@@ -1,3 +1,4 @@
 {
-  "totalTokens": 6
+  "totalTokens": 6,
+  "totalBillableCharacters": 16
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 9a1210c123c..123299ac1a7 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -935,7 +935,9 @@ final class GenerativeModelTests: XCTestCase {
     )
 
     let response = try await model.countTokens("Why is the sky blue?")
+
     XCTAssertEqual(response.totalTokens, 6)
+    XCTAssertEqual(response.totalBillableCharacters, 16)
   }
 
   func testCountTokens_modelNotFound() async throws {

From 1c1f720ed769ca609f9c416e968faae40c1e1e93 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 28 Mar 2024 22:17:31 +0000
Subject: [PATCH 082/124] Rename the Vertex AI parameter `region` to `location`
 (#12656)

---
 .../ViewModels/ConversationViewModel.swift    |  2 +-
 .../ViewModels/PhotoReasoningViewModel.swift  |  2 +-
 .../ViewModels/SummarizeViewModel.swift       |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift       | 28 +++++++++----------
 .../Sources/VertexAIComponent.swift           | 10 +++----
 .../Tests/Unit/VertexAIAPITests.swift         |  4 +--
 .../Tests/Unit/VertexComponentTests.swift     | 10 +++----
 7 files changed, 29 insertions(+), 29 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 81667be2514..883cefb359f 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,7 +36,7 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index 1fb0097a808..bfa1cfa9028 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,7 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    let vertexAI = VertexAI.vertexAI(region: "us-central1")
+    let vertexAI = VertexAI.vertexAI(location: "us-central1")
     model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index e3c78d09060..a90e1cf15b8 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,7 +32,7 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index a0df2ab015e..3e5f5e67050 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,35 +25,35 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
-  ///  - Parameter region: The region identifier, e.g., `us-central1`; see
+  ///  - Parameter location: The region identifier, e.g., `us-central1`; see
   ///     [Vertex AI
   ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
   ///     for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI(region: String) -> VertexAI {
+  public static func vertexAI(location: String) -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app, region: region)
+    return vertexAI(app: app, location: location)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
-  ///   - region: The region identifier, e.g., `us-central1`; see
+  ///   - location: The region identifier, e.g., `us-central1`; see
   ///     [Vertex AI
   ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
   ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp, region: String) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp, location: String) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI(region)
+    return provider.vertexAI(location)
   }
 
   /// Initializes a generative model with the given parameters.
@@ -73,7 +73,7 @@ public class VertexAI: NSObject {
                               tools: [Tool]? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
-    let modelResourceName = modelResourceName(modelName: modelName, region: region)
+    let modelResourceName = modelResourceName(modelName: modelName, location: location)
 
     guard let apiKey = app.options.apiKey else {
       fatalError("The Firebase app named \"\(app.name)\" has no API key in its configuration.")
@@ -97,29 +97,29 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
-  let region: String
+  let location: String
 
-  init(app: FirebaseApp, region: String) {
+  init(app: FirebaseApp, location: String) {
     self.app = app
-    self.region = region
+    self.location = location
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
   }
 
-  private func modelResourceName(modelName: String, region: String) -> String {
+  private func modelResourceName(modelName: String, location: String) -> String {
     if modelName.contains("/") {
       return modelName
     }
     guard let projectID = app.options.projectID else {
       fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
     }
-    guard !region.isEmpty else {
+    guard !location.isEmpty else {
       fatalError("""
-      No region specified; see
+      No location specified; see
       https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
       list of available regions.
       """)
     }
 
-    return "projects/\(projectID)/locations/\(region)/publishers/google/models/\(modelName)"
+    return "projects/\(projectID)/locations/\(location)/publishers/google/models/\(modelName)"
   }
 }
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index af8dbb56271..a6d487e1208 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -22,7 +22,7 @@ import Foundation
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 @objc(FIRVertexAIProvider)
 protocol VertexAIProvider {
-  @objc func vertexAI(_ region: String) -> VertexAI
+  @objc func vertexAI(_ location: String) -> VertexAI
 }
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -64,17 +64,17 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   // MARK: - VertexAIProvider conformance
 
-  func vertexAI(_ region: String) -> VertexAI {
+  func vertexAI(_ location: String) -> VertexAI {
     os_unfair_lock_lock(&instancesLock)
 
     // Unlock before the function returns.
     defer { os_unfair_lock_unlock(&instancesLock) }
 
-    if let instance = instances[region] {
+    if let instance = instances[location] {
       return instance
     }
-    let newInstance = VertexAI(app: app, region: region)
-    instances[region] = newInstance
+    let newInstance = VertexAI(app: app, location: location)
+    instances[location] = newInstance
     return newInstance
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index a4983eb8834..80236f405de 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -34,10 +34,10 @@ final class VertexAIAPITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Instantiate Vertex AI SDK - Default App
-    let vertexAI = VertexAI.vertexAI(region: "my-region")
+    let vertexAI = VertexAI.vertexAI(location: "my-location")
 
     // Instantiate Vertex AI SDK - Custom App
-    let _ = VertexAI.vertexAI(app: app!, region: "my-region")
+    let _ = VertexAI.vertexAI(app: app!, location: "my-location")
 
     // Permutations without optional arguments.
 
diff --git a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
index e28b747b65c..20f2b15bee8 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift
@@ -48,7 +48,7 @@ class VertexComponentTests: XCTestCase {
   func testVertexInstanceCreation() throws {
     let app = try XCTUnwrap(VertexComponentTests.app)
     let component = VertexAIComponent(app: app)
-    let vertex = component.vertexAI("my-region")
+    let vertex = component.vertexAI("my-location")
     XCTAssertNotNil(vertex)
   }
 
@@ -82,14 +82,14 @@ class VertexComponentTests: XCTestCase {
                                                             in: container)
     XCTAssertNotNil(provider)
 
-    let vertex1 = provider?.vertexAI("randomRegion")
-    let vertex2 = provider?.vertexAI("randomRegion")
+    let vertex1 = provider?.vertexAI("randomLocation")
+    let vertex2 = provider?.vertexAI("randomLocation")
     XCTAssertNotNil(vertex1)
 
     // Ensure they're the same instance.
     XCTAssert(vertex1 === vertex2)
 
-    let vertex3 = provider?.vertexAI("differentRegion")
+    let vertex3 = provider?.vertexAI("differentLocation")
     XCTAssertNotNil(vertex3)
 
     XCTAssert(vertex1 !== vertex3)
@@ -105,7 +105,7 @@ class VertexComponentTests: XCTestCase {
       options.projectID = "myProjectID"
       let app1 = FirebaseApp(instanceWithName: "transitory app", options: options)
       weakApp = try XCTUnwrap(app1)
-      let vertex = VertexAI(app: app1, region: "transitory region")
+      let vertex = VertexAI(app: app1, location: "transitory location")
       weakVertex = vertex
       XCTAssertNotNil(weakVertex)
     }

From 63293c171615864c306ba37b438bf96652239095 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 2 Apr 2024 14:48:35 +0000
Subject: [PATCH 083/124] Update to prod endpoint in Vertex AI (#12657)

---
 FirebaseVertexAI/Sources/GenerativeAISwift.swift                | 2 +-
 .../Unit/GenerateContentResponses/unary-failure-api-key.json    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
index 0f0af2bc3e2..933a19c4759 100644
--- a/FirebaseVertexAI/Sources/GenerativeAISwift.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -21,5 +21,5 @@ import Foundation
 /// Constants associated with the GenerativeAISwift SDK
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public enum GenerativeAISwift {
-  static let baseURL = "https://staging-firebaseml.sandbox.googleapis.com"
+  static let baseURL = "https://firebaseml.googleapis.com"
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
index d3b5677366e..e9b472fd031 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -9,7 +9,7 @@
         "reason": "API_KEY_INVALID",
         "domain": "googleapis.com",
         "metadata": {
-          "service": "staging-firebaseml.sandbox.googleapis.com"
+          "service": "firebaseml.googleapis.com"
         }
       },
       {

From 33ae50dc41818d6e1b019e85d4214a90c031f942 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 2 Apr 2024 07:45:50 -0800
Subject: [PATCH 084/124] [VertexAI] Consolidate to single sample app (#12662)

---
 .../Sample/ChatSample/ChatSampleApp.swift     |  36 --
 .../GenerativeAIMultimodalSampleApp.swift     |  31 --
 .../project.pbxproj                           | 483 ------------------
 .../GenerativeAITextSampleApp.swift           |  31 --
 4 files changed, 581 deletions(-)
 delete mode 100644 FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
 delete mode 100644 FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
 delete mode 100644 FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift

diff --git a/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift b/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
deleted file mode 100644
index 7d7e02d7e3c..00000000000
--- a/FirebaseVertexAI/Sample/ChatSample/ChatSampleApp.swift
+++ /dev/null
@@ -1,36 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import FirebaseCore
-import FirebaseVertexAI
-import SwiftUI
-
-@main
-struct ChatSampleApp: App {
-  @StateObject
-  var viewModel = ConversationViewModel()
-
-  init() {
-    FirebaseApp.configure()
-  }
-
-  var body: some Scene {
-    WindowGroup {
-      NavigationStack {
-        ConversationScreen()
-          .environmentObject(viewModel)
-      }
-    }
-  }
-}
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
deleted file mode 100644
index 49565de52d1..00000000000
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/GenerativeAIMultimodalSampleApp.swift
+++ /dev/null
@@ -1,31 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import FirebaseCore
-import SwiftUI
-
-@main
-struct GenerativeAIMultimodalSampleApp: App {
-  init() {
-    FirebaseApp.configure()
-  }
-
-  var body: some Scene {
-    WindowGroup {
-      NavigationStack {
-        PhotoReasoningScreen()
-      }
-    }
-  }
-}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 9efade7342d..ac930a76d95 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -9,34 +9,13 @@
 /* Begin PBXBuildFile section */
 		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
 		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
-		868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33692BB4771900304BB1 /* FirebaseAppCheck */; };
-		868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */; };
-		868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336D2BB4773600304BB1 /* FirebaseAppCheck */; };
-		868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */; };
-		868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33712BB4775D00304BB1 /* FirebaseAppCheck */; };
-		868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
-		880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */; };
-		880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */; };
-		88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
-		88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
-		88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88209C232B0FBE1700F64795 /* MarkdownUI */; };
-		88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
 		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
 		8848C8352B0D04BC007B434F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8342B0D04BC007B434F /* ContentView.swift */; };
 		8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8362B0D04BD007B434F /* Assets.xcassets */; };
 		8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8392B0D04BD007B434F /* Preview Assets.xcassets */; };
-		8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */; };
-		8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84A2B0D051F007B434F /* Assets.xcassets */; };
-		8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C84D2B0D051F007B434F /* Preview Assets.xcassets */; };
-		8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */; };
-		8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85C2B0D056D007B434F /* Assets.xcassets */; };
-		8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C85F2B0D056D007B434F /* Preview Assets.xcassets */; };
 		886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */; };
 		886F95D62B17BA010036F07A /* SummarizeViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */; };
 		886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95D72B17BA420036F07A /* MarkdownUI */; };
@@ -48,19 +27,6 @@
 		886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
 		886F95E12B17D5010036F07A /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
 		886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 886F95E22B17D6630036F07A /* GenerativeAIUIComponents */; };
-		889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
-		88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A91D2B0FC55100424728 /* MarkdownUI */; };
-		88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = 88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */; };
-		88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */ = {isa = PBXBuildFile; productRef = 88D9474C2B14F27E008B5580 /* MarkdownUI */; };
-		88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F442B110D5300C08E95 /* ChatSampleApp.swift */; };
-		88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F482B110D5400C08E95 /* Assets.xcassets */; };
-		88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */; };
-		88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F542B1112CA00C08E95 /* ConversationScreen.swift */; };
-		88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F562B1112F600C08E95 /* ConversationViewModel.swift */; };
-		88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F582B11131900C08E95 /* ChatMessage.swift */; };
-		88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5A2B11133E00C08E95 /* MessageView.swift */; };
-		88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88E10F5C2B11135000C08E95 /* BouncingDots.swift */; };
-		CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */ = {isa = PBXBuildFile; productRef = CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */; };
 /* End PBXBuildFile section */
 
 /* Begin PBXFileReference section */
@@ -75,18 +41,12 @@
 		8848C8342B0D04BC007B434F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = "<group>"; };
 		8848C8362B0D04BD007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C8392B0D04BD007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
-		8848C8442B0D051E007B434F /* GenerativeAITextSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAITextSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAITextSampleApp.swift; sourceTree = "<group>"; };
 		8848C84A2B0D051F007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C84D2B0D051F007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
-		8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAIMultimodalSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAIMultimodalSampleApp.swift; sourceTree = "<group>"; };
 		8848C85C2B0D056D007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C85F2B0D056D007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
 		889873842B208563005B4896 /* ErrorDetailsView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ErrorDetailsView.swift; sourceTree = "<group>"; };
 		88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = GenerativeAIUIComponents; sourceTree = "<group>"; };
-		88E10F422B110D5300C08E95 /* ChatSample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = ChatSample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		88E10F442B110D5300C08E95 /* ChatSampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatSampleApp.swift; sourceTree = "<group>"; };
 		88E10F482B110D5400C08E95 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		88E10F4B2B110D5400C08E95 /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
 		88E10F542B1112CA00C08E95 /* ConversationScreen.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ConversationScreen.swift; sourceTree = "<group>"; };
@@ -108,38 +68,6 @@
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
-		8848C8412B0D051E007B434F /* Frameworks */ = {
-			isa = PBXFrameworksBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				868A336C2BB4771900304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
-				868A336A2BB4771900304BB1 /* FirebaseAppCheck in Frameworks */,
-				88209C242B0FBE1700F64795 /* MarkdownUI in Frameworks */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		8848C8532B0D056C007B434F /* Frameworks */ = {
-			isa = PBXFrameworksBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				868A33702BB4773600304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
-				88B8A91E2B0FC55100424728 /* MarkdownUI in Frameworks */,
-				868A336E2BB4773600304BB1 /* FirebaseAppCheck in Frameworks */,
-				88B8A9372B0FCBE700424728 /* GenerativeAIUIComponents in Frameworks */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		88E10F3F2B110D5300C08E95 /* Frameworks */ = {
-			isa = PBXFrameworksBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				868A33742BB4775D00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
-				88D9474D2B14F27E008B5580 /* MarkdownUI in Frameworks */,
-				868A33722BB4775D00304BB1 /* FirebaseAppCheck in Frameworks */,
-				CB0DFFD52B2B4F08006E109D /* GenerativeAIUIComponents in Frameworks */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
 /* End PBXFrameworksBuildPhase section */
 
 /* Begin PBXGroup section */
@@ -200,9 +128,6 @@
 			isa = PBXGroup;
 			children = (
 				8848C82F2B0D04BC007B434F /* GenerativeAISample.app */,
-				8848C8442B0D051E007B434F /* GenerativeAITextSample.app */,
-				8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */,
-				88E10F422B110D5300C08E95 /* ChatSample.app */,
 			);
 			name = Products;
 			sourceTree = "<group>";
@@ -231,7 +156,6 @@
 			children = (
 				88209C1C2B0FBDC300F64795 /* ViewModels */,
 				88209C1A2B0FBDC300F64795 /* Screens */,
-				8848C8462B0D051E007B434F /* GenerativeAITextSampleApp.swift */,
 				8848C84A2B0D051F007B434F /* Assets.xcassets */,
 				8848C84C2B0D051F007B434F /* Preview Content */,
 			);
@@ -251,7 +175,6 @@
 			children = (
 				8802666E2B0FC39000CF7CB6 /* ViewModels */,
 				880266742B0FC39000CF7CB6 /* Screens */,
-				8848C8582B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift */,
 				8848C85C2B0D056D007B434F /* Assets.xcassets */,
 				8848C85E2B0D056D007B434F /* Preview Content */,
 			);
@@ -273,7 +196,6 @@
 				88E10F502B11123600C08E95 /* ViewModels */,
 				88E10F512B11124100C08E95 /* Views */,
 				88E10F532B1112B900C08E95 /* Screens */,
-				88E10F442B110D5300C08E95 /* ChatSampleApp.swift */,
 				88E10F482B110D5400C08E95 /* Assets.xcassets */,
 				88E10F4A2B110D5400C08E95 /* Preview Content */,
 			);
@@ -349,74 +271,6 @@
 			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
 			productType = "com.apple.product-type.application";
 		};
-		8848C8432B0D051E007B434F /* GenerativeAITextSample */ = {
-			isa = PBXNativeTarget;
-			buildConfigurationList = 8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */;
-			buildPhases = (
-				8848C8402B0D051E007B434F /* Sources */,
-				8848C8412B0D051E007B434F /* Frameworks */,
-				8848C8422B0D051E007B434F /* Resources */,
-			);
-			buildRules = (
-			);
-			dependencies = (
-			);
-			name = GenerativeAITextSample;
-			packageProductDependencies = (
-				88209C232B0FBE1700F64795 /* MarkdownUI */,
-				868A33692BB4771900304BB1 /* FirebaseAppCheck */,
-				868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */,
-			);
-			productName = GenerativeAITextSample;
-			productReference = 8848C8442B0D051E007B434F /* GenerativeAITextSample.app */;
-			productType = "com.apple.product-type.application";
-		};
-		8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */ = {
-			isa = PBXNativeTarget;
-			buildConfigurationList = 8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */;
-			buildPhases = (
-				8848C8522B0D056C007B434F /* Sources */,
-				8848C8532B0D056C007B434F /* Frameworks */,
-				8848C8542B0D056C007B434F /* Resources */,
-			);
-			buildRules = (
-			);
-			dependencies = (
-			);
-			name = GenerativeAIMultimodalSample;
-			packageProductDependencies = (
-				88B8A91D2B0FC55100424728 /* MarkdownUI */,
-				88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */,
-				868A336D2BB4773600304BB1 /* FirebaseAppCheck */,
-				868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */,
-			);
-			productName = GenerativeAIMultimodalSample;
-			productReference = 8848C8562B0D056C007B434F /* GenerativeAIMultimodalSample.app */;
-			productType = "com.apple.product-type.application";
-		};
-		88E10F412B110D5300C08E95 /* ChatSample */ = {
-			isa = PBXNativeTarget;
-			buildConfigurationList = 88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */;
-			buildPhases = (
-				88E10F3E2B110D5300C08E95 /* Sources */,
-				88E10F3F2B110D5300C08E95 /* Frameworks */,
-				88E10F402B110D5300C08E95 /* Resources */,
-			);
-			buildRules = (
-			);
-			dependencies = (
-			);
-			name = ChatSample;
-			packageProductDependencies = (
-				88D9474C2B14F27E008B5580 /* MarkdownUI */,
-				CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */,
-				868A33712BB4775D00304BB1 /* FirebaseAppCheck */,
-				868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */,
-			);
-			productName = ChatSample;
-			productReference = 88E10F422B110D5300C08E95 /* ChatSample.app */;
-			productType = "com.apple.product-type.application";
-		};
 /* End PBXNativeTarget section */
 
 /* Begin PBXProject section */
@@ -430,15 +284,6 @@
 					8848C82E2B0D04BC007B434F = {
 						CreatedOnToolsVersion = 15.1;
 					};
-					8848C8432B0D051E007B434F = {
-						CreatedOnToolsVersion = 15.1;
-					};
-					8848C8552B0D056C007B434F = {
-						CreatedOnToolsVersion = 15.1;
-					};
-					88E10F412B110D5300C08E95 = {
-						CreatedOnToolsVersion = 15.1;
-					};
 				};
 			};
 			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */;
@@ -460,9 +305,6 @@
 			projectRoot = "";
 			targets = (
 				8848C82E2B0D04BC007B434F /* GenerativeAISample */,
-				8848C8432B0D051E007B434F /* GenerativeAITextSample */,
-				8848C8552B0D056C007B434F /* GenerativeAIMultimodalSample */,
-				88E10F412B110D5300C08E95 /* ChatSample */,
 			);
 		};
 /* End PBXProject section */
@@ -478,36 +320,6 @@
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
-		8848C8422B0D051E007B434F /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				8848C84E2B0D051F007B434F /* Preview Assets.xcassets in Resources */,
-				8848C84B2B0D051F007B434F /* Assets.xcassets in Resources */,
-				869200B42B879C4F00482873 /* GoogleService-Info.plist in Resources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		8848C8542B0D056C007B434F /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				8848C8602B0D056D007B434F /* Preview Assets.xcassets in Resources */,
-				8848C85D2B0D056D007B434F /* Assets.xcassets in Resources */,
-				869200B52B879C4F00482873 /* GoogleService-Info.plist in Resources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		88E10F402B110D5300C08E95 /* Resources */ = {
-			isa = PBXResourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				88E10F4C2B110D5400C08E95 /* Preview Assets.xcassets in Resources */,
-				88E10F492B110D5400C08E95 /* Assets.xcassets in Resources */,
-				869200B62B879C4F00482873 /* GoogleService-Info.plist in Resources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
 /* End PBXResourcesBuildPhase section */
 
 /* Begin PBXSourcesBuildPhase section */
@@ -531,41 +343,6 @@
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
-		8848C8402B0D051E007B434F /* Sources */ = {
-			isa = PBXSourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				88209C1F2B0FBDC300F64795 /* SummarizeScreen.swift in Sources */,
-				8848C8472B0D051E007B434F /* GenerativeAITextSampleApp.swift in Sources */,
-				88209C202B0FBDC300F64795 /* SummarizeViewModel.swift in Sources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		8848C8522B0D056C007B434F /* Sources */ = {
-			isa = PBXSourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				880266762B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift in Sources */,
-				880266792B0FC39000CF7CB6 /* PhotoReasoningScreen.swift in Sources */,
-				8848C8592B0D056C007B434F /* GenerativeAIMultimodalSampleApp.swift in Sources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
-		88E10F3E2B110D5300C08E95 /* Sources */ = {
-			isa = PBXSourcesBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-				88E10F5B2B11133E00C08E95 /* MessageView.swift in Sources */,
-				88E10F572B1112F600C08E95 /* ConversationViewModel.swift in Sources */,
-				88E10F552B1112CA00C08E95 /* ConversationScreen.swift in Sources */,
-				88E10F592B11131900C08E95 /* ChatMessage.swift in Sources */,
-				88E10F5D2B11135000C08E95 /* BouncingDots.swift in Sources */,
-				88E10F452B110D5300C08E95 /* ChatSampleApp.swift in Sources */,
-				88263BEF2B239BFE008AB09B /* ErrorView.swift in Sources */,
-				889873852B208563005B4896 /* ErrorDetailsView.swift in Sources */,
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-		};
 /* End PBXSourcesBuildPhase section */
 
 /* Begin XCBuildConfiguration section */
@@ -748,192 +525,6 @@
 			};
 			name = Release;
 		};
-		8848C8502B0D051F007B434F /* Debug */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Debug;
-		};
-		8848C8512B0D051F007B434F /* Release */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAITextSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAITextSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Release;
-		};
-		8848C8622B0D056D007B434F /* Debug */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Debug;
-		};
-		8848C8632B0D056D007B434F /* Release */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAIMultimodalSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAIMultimodalSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Release;
-		};
-		88E10F4D2B110D5400C08E95 /* Debug */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Debug;
-		};
-		88E10F4E2B110D5400C08E95 /* Release */ = {
-			isa = XCBuildConfiguration;
-			buildSettings = {
-				ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
-				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
-				CODE_SIGN_STYLE = Automatic;
-				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"ChatSample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
-				ENABLE_PREVIEWS = YES;
-				ENABLE_USER_SCRIPT_SANDBOXING = NO;
-				GENERATE_INFOPLIST_FILE = YES;
-				INFOPLIST_KEY_UIApplicationSceneManifest_Generation = YES;
-				INFOPLIST_KEY_UIApplicationSupportsIndirectInputEvents = YES;
-				INFOPLIST_KEY_UILaunchScreen_Generation = YES;
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPad = "UIInterfaceOrientationPortrait UIInterfaceOrientationPortraitUpsideDown UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				INFOPLIST_KEY_UISupportedInterfaceOrientations_iPhone = "UIInterfaceOrientationPortrait UIInterfaceOrientationLandscapeLeft UIInterfaceOrientationLandscapeRight";
-				IPHONEOS_DEPLOYMENT_TARGET = 16.0;
-				LD_RUNPATH_SEARCH_PATHS = (
-					"$(inherited)",
-					"@executable_path/Frameworks",
-				);
-				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.ChatSample;
-				PRODUCT_NAME = "$(TARGET_NAME)";
-				SWIFT_EMIT_LOC_STRINGS = YES;
-				SWIFT_VERSION = 5.0;
-				TARGETED_DEVICE_FAMILY = "1,2";
-			};
-			name = Release;
-		};
 /* End XCBuildConfiguration section */
 
 /* Begin XCConfigurationList section */
@@ -955,33 +546,6 @@
 			defaultConfigurationIsVisible = 0;
 			defaultConfigurationName = Release;
 		};
-		8848C84F2B0D051F007B434F /* Build configuration list for PBXNativeTarget "GenerativeAITextSample" */ = {
-			isa = XCConfigurationList;
-			buildConfigurations = (
-				8848C8502B0D051F007B434F /* Debug */,
-				8848C8512B0D051F007B434F /* Release */,
-			);
-			defaultConfigurationIsVisible = 0;
-			defaultConfigurationName = Release;
-		};
-		8848C8612B0D056D007B434F /* Build configuration list for PBXNativeTarget "GenerativeAIMultimodalSample" */ = {
-			isa = XCConfigurationList;
-			buildConfigurations = (
-				8848C8622B0D056D007B434F /* Debug */,
-				8848C8632B0D056D007B434F /* Release */,
-			);
-			defaultConfigurationIsVisible = 0;
-			defaultConfigurationName = Release;
-		};
-		88E10F4F2B110D5400C08E95 /* Build configuration list for PBXNativeTarget "ChatSample" */ = {
-			isa = XCConfigurationList;
-			buildConfigurations = (
-				88E10F4D2B110D5400C08E95 /* Debug */,
-				88E10F4E2B110D5400C08E95 /* Release */,
-			);
-			defaultConfigurationIsVisible = 0;
-			defaultConfigurationName = Release;
-		};
 /* End XCConfigurationList section */
 
 /* Begin XCLocalSwiftPackageReference section */
@@ -1019,35 +583,6 @@
 			isa = XCSwiftPackageProductDependency;
 			productName = "FirebaseVertexAI-Preview";
 		};
-		868A33692BB4771900304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		868A336B2BB4771900304BB1 /* FirebaseVertexAI-Preview */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = "FirebaseVertexAI-Preview";
-		};
-		868A336D2BB4773600304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		868A336F2BB4773600304BB1 /* FirebaseVertexAI-Preview */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = "FirebaseVertexAI-Preview";
-		};
-		868A33712BB4775D00304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
-		868A33732BB4775D00304BB1 /* FirebaseVertexAI-Preview */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = "FirebaseVertexAI-Preview";
-		};
-		88209C232B0FBE1700F64795 /* MarkdownUI */ = {
-			isa = XCSwiftPackageProductDependency;
-			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
-			productName = MarkdownUI;
-		};
 		886F95D72B17BA420036F07A /* MarkdownUI */ = {
 			isa = XCSwiftPackageProductDependency;
 			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
@@ -1057,24 +592,6 @@
 			isa = XCSwiftPackageProductDependency;
 			productName = GenerativeAIUIComponents;
 		};
-		88B8A91D2B0FC55100424728 /* MarkdownUI */ = {
-			isa = XCSwiftPackageProductDependency;
-			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
-			productName = MarkdownUI;
-		};
-		88B8A9362B0FCBE700424728 /* GenerativeAIUIComponents */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = GenerativeAIUIComponents;
-		};
-		88D9474C2B14F27E008B5580 /* MarkdownUI */ = {
-			isa = XCSwiftPackageProductDependency;
-			package = 88209C212B0FBDF700F64795 /* XCRemoteSwiftPackageReference "swift-markdown-ui" */;
-			productName = MarkdownUI;
-		};
-		CB0DFFD42B2B4F08006E109D /* GenerativeAIUIComponents */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = GenerativeAIUIComponents;
-		};
 /* End XCSwiftPackageProductDependency section */
 	};
 	rootObject = 8848C8272B0D04BC007B434F /* Project object */;
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
deleted file mode 100644
index c800e1f7be8..00000000000
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/GenerativeAITextSampleApp.swift
+++ /dev/null
@@ -1,31 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import FirebaseCore
-import SwiftUI
-
-@main
-struct GenerativeAITextSampleApp: App {
-  init() {
-    FirebaseApp.configure()
-  }
-
-  var body: some Scene {
-    WindowGroup {
-      NavigationStack {
-        SummarizeScreen()
-      }
-    }
-  }
-}

From b96cf18823895730f68a032a5e92a3dfee6cf808 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 2 Apr 2024 17:11:08 +0000
Subject: [PATCH 085/124] Add platform tags in `x-goog-api-client` (#12680)

---
 FirebaseCore/Sources/FIRApp.m                       |  2 +-
 FirebaseVertexAI/Sources/GenerativeAIService.swift  | 13 ++++++++++---
 .../Tests/Unit/GenerativeModelTests.swift           |  5 +++++
 3 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/FirebaseCore/Sources/FIRApp.m b/FirebaseCore/Sources/FIRApp.m
index a8cc38c12de..12afc134321 100644
--- a/FirebaseCore/Sources/FIRApp.m
+++ b/FirebaseCore/Sources/FIRApp.m
@@ -829,7 +829,7 @@ + (void)registerSwiftComponents {
     @"FIRSessions" : @"fire-ses",
     @"FIRFunctionsComponent" : @"fire-fun",
     @"FIRStorageComponent" : @"fire-str",
-    @"FIRVertexAIComponent" : @"fire-vtx",
+    @"FIRVertexAIComponent" : @"fire-vertex",
   };
   for (NSString *className in swiftComponents.allKeys) {
     Class klass = NSClassFromString(className);
diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index e1df1010c06..b3687c3c013 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -18,6 +18,12 @@ import Foundation
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 struct GenerativeAIService {
+  /// The language of the SDK in the format "gl-<language>/<version>" where version may be blank.
+  static let languageTag = "gl-swift/"
+
+  /// The Firebase SDK version in the format "fire/<version>".
+  static let firebaseVersionTag = "fire/\(FirebaseVersion())"
+
   /// Gives permission to talk to the backend.
   private let apiKey: String
 
@@ -155,9 +161,10 @@ struct GenerativeAIService {
     var urlRequest = URLRequest(url: request.url)
     urlRequest.httpMethod = "POST"
     urlRequest.setValue(apiKey, forHTTPHeaderField: "x-goog-api-key")
-    // TODO: Determine the right client header to use.
-    //    urlRequest.setValue("genai-swift/\(GenerativeAISwift.version))",
-    //                        forHTTPHeaderField: "x-goog-api-client")
+    urlRequest.setValue(
+      "\(GenerativeAIService.languageTag) \(GenerativeAIService.firebaseVersionTag)",
+      forHTTPHeaderField: "x-goog-api-client"
+    )
     urlRequest.setValue("application/json", forHTTPHeaderField: "Content-Type")
 
     if let appCheck {
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 123299ac1a7..8031fb95dcf 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseCore
 import XCTest
 
 @testable import FirebaseVertexAI
@@ -1058,6 +1059,10 @@ final class GenerativeModelTests: XCTestCase {
       let requestURL = try XCTUnwrap(request.url)
       XCTAssertEqual(requestURL.path.occurrenceCount(of: "models/"), 1)
       XCTAssertEqual(request.timeoutInterval, timeout)
+      let apiClientTags = try XCTUnwrap(request.value(forHTTPHeaderField: "x-goog-api-client"))
+        .components(separatedBy: " ")
+      XCTAssert(apiClientTags.contains(GenerativeAIService.languageTag))
+      XCTAssert(apiClientTags.contains(GenerativeAIService.firebaseVersionTag))
       XCTAssertEqual(request.value(forHTTPHeaderField: "X-Firebase-AppCheck"), appCheckToken)
       let response = try XCTUnwrap(HTTPURLResponse(
         url: requestURL,

From f1eb164cea278050065ab734e9d05d08793619f4 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Tue, 2 Apr 2024 09:52:57 -0800
Subject: [PATCH 086/124] [vertex-ai] Fix for SwiftFormat 0.53.5 (#12683)

---
 .../Sources/GenerativeAIUIComponents/InputField.swift            | 1 +
 1 file changed, 1 insertion(+)

diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
index 317baaaf3c1..3f12ea65d8a 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import SwiftUI
+
 public struct InputField<Label>: View where Label: View {
   @Binding
   private var text: String

From 2362b33ed7868be931ee4da432da1361f7ac90ac Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 3 Apr 2024 16:12:40 +0000
Subject: [PATCH 087/124] Add `functionCalls` accessor to
 `GenerateContentResponse` (#12690)

---
 .../Sources/GenerateContentResponse.swift           | 13 +++++++++++++
 .../Tests/Unit/GenerativeModelTests.swift           |  5 +++++
 2 files changed, 18 insertions(+)

diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 4e02abb7f2c..f43da308b13 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -37,6 +37,19 @@ public struct GenerateContentResponse {
     return text
   }
 
+  /// Returns function calls found in any `Part`s of the first candidate of the response, if any.
+  public var functionCalls: [FunctionCall] {
+    guard let candidate = candidates.first else {
+      return []
+    }
+    return candidate.content.parts.compactMap { part in
+      guard case let .functionCall(functionCall) = part else {
+        return nil
+      }
+      return functionCall
+    }
+  }
+
   /// Initializer for SwiftUI previews or tests.
   public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback?) {
     self.candidates = candidates
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 8031fb95dcf..13cbb6df94e 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -70,6 +70,7 @@ final class GenerativeModelTests: XCTestCase {
     let partText = try XCTUnwrap(part.text)
     XCTAssertTrue(partText.hasPrefix("You can ask me a wide range of questions"))
     XCTAssertEqual(response.text, partText)
+    XCTAssertEqual(response.functionCalls, [])
   }
 
   func testGenerateContent_success_basicReplyShort() async throws {
@@ -90,6 +91,7 @@ final class GenerativeModelTests: XCTestCase {
     let part = try XCTUnwrap(candidate.content.parts.first)
     XCTAssertEqual(part.text, "Mountain View, California, United States")
     XCTAssertEqual(response.text, part.text)
+    XCTAssertEqual(response.functionCalls, [])
   }
 
   func testGenerateContent_success_citations() async throws {
@@ -195,6 +197,7 @@ final class GenerativeModelTests: XCTestCase {
     }
     XCTAssertEqual(functionCall.name, "current_time")
     XCTAssertTrue(functionCall.args.isEmpty)
+    XCTAssertEqual(response.functionCalls, [functionCall])
   }
 
   func testGenerateContent_success_functionCall_noArguments() async throws {
@@ -216,6 +219,7 @@ final class GenerativeModelTests: XCTestCase {
     }
     XCTAssertEqual(functionCall.name, "current_time")
     XCTAssertTrue(functionCall.args.isEmpty)
+    XCTAssertEqual(response.functionCalls, [functionCall])
   }
 
   func testGenerateContent_success_functionCall_withArguments() async throws {
@@ -241,6 +245,7 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(argX, .number(4))
     let argY = try XCTUnwrap(functionCall.args["y"])
     XCTAssertEqual(argY, .number(5))
+    XCTAssertEqual(response.functionCalls, [functionCall])
   }
 
   func testGenerateContent_appCheck_validToken() async throws {

From e8d741e768f0e07c0efddc3be33a6de0f54cd421 Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Thu, 4 Apr 2024 06:59:30 -0800
Subject: [PATCH 088/124] Vertex AI preview versioning (#12691)

---
 FirebaseCore/Sources/FIROptions.m         |  2 +-
 FirebaseCore/Tests/Unit/FIROptionsTest.m  | 12 ++++++------
 Package.swift                             |  2 +-
 SwiftPMTests/swift-test/all-imports.swift |  3 +--
 4 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/FirebaseCore/Sources/FIROptions.m b/FirebaseCore/Sources/FIROptions.m
index d46b657501c..c0625a8bac9 100644
--- a/FirebaseCore/Sources/FIROptions.m
+++ b/FirebaseCore/Sources/FIROptions.m
@@ -284,7 +284,7 @@ - (NSString *)libraryVersionID {
     // The unit tests are set up to catch anything that does not properly convert.
     NSString *version = FIRFirebaseVersion();
     NSArray *components = [version componentsSeparatedByString:@"."];
-    NSString *major = [components objectAtIndex:0];
+    NSString *major = [NSString stringWithFormat:@"%02d", [[components objectAtIndex:0] intValue]];
     NSString *minor = [NSString stringWithFormat:@"%02d", [[components objectAtIndex:1] intValue]];
     NSString *patch = [NSString stringWithFormat:@"%02d", [[components objectAtIndex:2] intValue]];
     kFIRLibraryVersionID = [NSString stringWithFormat:@"%@%@%@000", major, minor, patch];
diff --git a/FirebaseCore/Tests/Unit/FIROptionsTest.m b/FirebaseCore/Tests/Unit/FIROptionsTest.m
index 3de059f4191..fd5c901598f 100644
--- a/FirebaseCore/Tests/Unit/FIROptionsTest.m
+++ b/FirebaseCore/Tests/Unit/FIROptionsTest.m
@@ -624,7 +624,7 @@ - (void)testVersionConsistency {
   int minor = (versionString[2] - '0') * 10 + versionString[3] - '0';
   int patch = (versionString[4] - '0') * 10 + versionString[5] - '0';
   NSString *str = [NSString stringWithFormat:@"%d.%d.%d", major, minor, patch];
-  XCTAssertEqualObjects(str, FIRFirebaseVersion());
+  XCTAssertTrue([FIRFirebaseVersion() hasPrefix:str]);
 }
 
 // Repeat test with more Objective-C.
@@ -638,11 +638,11 @@ - (void)testVersionConsistency2 {
   NSRange major = NSMakeRange(0, 2);
   NSRange minor = NSMakeRange(2, 2);
   NSRange patch = NSMakeRange(4, 2);
-  NSString *str =
-      [NSString stringWithFormat:@"%@.%d.%d", [kFIRLibraryVersionID substringWithRange:major],
-                                 [[kFIRLibraryVersionID substringWithRange:minor] intValue],
-                                 [[kFIRLibraryVersionID substringWithRange:patch] intValue]];
-  XCTAssertEqualObjects(str, FIRFirebaseVersion());
+  NSString *str = [NSString
+      stringWithFormat:@"%d.%d.%d", [[kFIRLibraryVersionID substringWithRange:major] intValue],
+                       [[kFIRLibraryVersionID substringWithRange:minor] intValue],
+                       [[kFIRLibraryVersionID substringWithRange:patch] intValue]];
+  XCTAssertTrue([FIRFirebaseVersion() hasPrefix:str]);
 }
 
 #pragma mark - Helpers
diff --git a/Package.swift b/Package.swift
index 29aa768d34d..c10a21d6e95 100644
--- a/Package.swift
+++ b/Package.swift
@@ -19,7 +19,7 @@
 import class Foundation.ProcessInfo
 import PackageDescription
 
-let firebaseVersion = "10.27.0"
+let firebaseVersion = "0.1.0-vertexai-preview"
 
 let package = Package(
   name: "Firebase",
diff --git a/SwiftPMTests/swift-test/all-imports.swift b/SwiftPMTests/swift-test/all-imports.swift
index 6486c1be833..c103f128d9d 100644
--- a/SwiftPMTests/swift-test/all-imports.swift
+++ b/SwiftPMTests/swift-test/all-imports.swift
@@ -76,9 +76,8 @@ class importTest: XCTestCase {
 
     let versionParts = FirebaseVersion().split(separator: ".")
     XCTAssert(versionParts.count == 3)
-    XCTAssertEqual(Int(versionParts[0]), 10)
+    XCTAssertNotNil(Int(versionParts[0]))
     XCTAssertNotNil(Int(versionParts[1]))
-    XCTAssertNotNil(Int(versionParts[2]))
 
     print("System version? Answer: \(GULAppEnvironmentUtil.systemVersion())")
   }

From 9c3422bd419cc733e46bb3366afa19e9f1a75458 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 4 Apr 2024 19:15:09 +0000
Subject: [PATCH 089/124] Remove `location` from Vertex AI public API (#12706)

---
 .../ViewModels/ConversationViewModel.swift       |  2 +-
 .../ViewModels/PhotoReasoningViewModel.swift     |  3 +--
 .../ViewModels/SummarizeViewModel.swift          |  2 +-
 FirebaseVertexAI/Sources/VertexAI.swift          | 16 ++++------------
 .../Tests/Unit/VertexAIAPITests.swift            |  4 ++--
 5 files changed, 9 insertions(+), 18 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
index 883cefb359f..a7c3d21a960 100644
--- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift
@@ -36,7 +36,7 @@ class ConversationViewModel: ObservableObject {
   private var chatTask: Task<Void, Never>?
 
   init() {
-    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.0-pro")
     chat = model.startChat()
   }
 
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index bfa1cfa9028..b27d518fe86 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,8 +44,7 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    let vertexAI = VertexAI.vertexAI(location: "us-central1")
-    model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
+    model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
   func reason() async {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
index a90e1cf15b8..99b374efefa 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift
@@ -32,7 +32,7 @@ class SummarizeViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(location: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.0-pro")
   }
 
   func summarize(inputText: String) async {
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 3e5f5e67050..1fcfd998b45 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,35 +25,27 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
-  ///  - Parameter location: The region identifier, e.g., `us-central1`; see
-  ///     [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
-  ///     for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI(location: String) -> VertexAI {
+  public static func vertexAI() -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app, location: location)
+    return vertexAI(app: app)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
-  ///   - location: The region identifier, e.g., `us-central1`; see
-  ///     [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
-  ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp, location: String) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp) -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI(location)
+    return provider.vertexAI("us-central1")
   }
 
   /// Initializes a generative model with the given parameters.
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 80236f405de..81b81414dcb 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -34,10 +34,10 @@ final class VertexAIAPITests: XCTestCase {
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
 
     // Instantiate Vertex AI SDK - Default App
-    let vertexAI = VertexAI.vertexAI(location: "my-location")
+    let vertexAI = VertexAI.vertexAI()
 
     // Instantiate Vertex AI SDK - Custom App
-    let _ = VertexAI.vertexAI(app: app!, location: "my-location")
+    let _ = VertexAI.vertexAI(app: app!)
 
     // Permutations without optional arguments.
 

From 2544f779702bad31cbf29504edd42c0ef6106600 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 4 Apr 2024 19:15:36 +0000
Subject: [PATCH 090/124] Remove `apiVersion` from `RequestOptions` in Vertex
 AI (#12707)

---
 FirebaseVertexAI/Sources/GenerativeAIRequest.swift | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
index 21f35b1b728..a3a198726d4 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIRequest.swift
@@ -31,16 +31,14 @@ public struct RequestOptions {
   let timeout: TimeInterval?
 
   /// The API version to use in requests to the backend.
-  let apiVersion: String
+  let apiVersion = "v2beta"
 
   /// Initializes a request options object.
   ///
   /// - Parameters:
   ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
   ///   for a `URLRequest`.
-  ///   - apiVersion The API version to use in requests to the backend; defaults to "v2beta".
-  public init(timeout: TimeInterval? = nil, apiVersion: String = "v2beta") {
+  public init(timeout: TimeInterval? = nil) {
     self.timeout = timeout
-    self.apiVersion = apiVersion
   }
 }

From e94075887be6a29f3f5f04ed1e71ae91118ed587 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 8 Apr 2024 17:36:26 -0400
Subject: [PATCH 091/124] Add placeholder `GoogleService-Info.plist` to Vertex
 AI sample (#12724)

---
 .../project.pbxproj                           |  4 +-
 .../GenerativeAISampleApp.swift               |  8 ++++
 .../Sample/GoogleService-Info.plist           | 40 +++++++++++++++++++
 FirebaseVertexAI/Sample/README.md             | 16 ++++++++
 4 files changed, 66 insertions(+), 2 deletions(-)
 create mode 100644 FirebaseVertexAI/Sample/GoogleService-Info.plist
 create mode 100644 FirebaseVertexAI/Sample/README.md

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index ac930a76d95..01336b9cf19 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -487,7 +487,7 @@
 					"@executable_path/Frameworks",
 				);
 				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_BUNDLE_IDENTIFIER = com.google.firebase.VertexAISample;
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				SWIFT_EMIT_LOC_STRINGS = YES;
 				SWIFT_VERSION = 5.0;
@@ -517,7 +517,7 @@
 					"@executable_path/Frameworks",
 				);
 				MARKETING_VERSION = 1.0;
-				PRODUCT_BUNDLE_IDENTIFIER = com.example.google.generativeai.GenerativeAISample;
+				PRODUCT_BUNDLE_IDENTIFIER = com.google.firebase.VertexAISample;
 				PRODUCT_NAME = "$(TARGET_NAME)";
 				SWIFT_EMIT_LOC_STRINGS = YES;
 				SWIFT_VERSION = 5.0;
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
index 7c709111b7f..509087a5e07 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
@@ -19,6 +19,14 @@ import SwiftUI
 struct GenerativeAISampleApp: App {
   init() {
     FirebaseApp.configure()
+
+    if let firebaseApp = FirebaseApp.app(), firebaseApp.options.projectID == "mockproject-1234" {
+      guard let bundleID = Bundle.main.bundleIdentifier else { fatalError() }
+      fatalError("You must create and/or download a valid `GoogleService-Info.plist` file for"
+        + " \(bundleID) from https://console.firebase.google.com to run this sample. Replace the"
+        + " existing `GoogleService-Info.plist` file in the `FirebaseVertexAI/Sample` directory"
+        + " with this new file.")
+    }
   }
 
   var body: some Scene {
diff --git a/FirebaseVertexAI/Sample/GoogleService-Info.plist b/FirebaseVertexAI/Sample/GoogleService-Info.plist
new file mode 100644
index 00000000000..b026413cc39
--- /dev/null
+++ b/FirebaseVertexAI/Sample/GoogleService-Info.plist
@@ -0,0 +1,40 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>AD_UNIT_ID_FOR_BANNER_TEST</key>
+	<string>ca-app-pub-3940256099942544/2934735716</string>
+	<key>AD_UNIT_ID_FOR_INTERSTITIAL_TEST</key>
+	<string>ca-app-pub-3940256099942544/4411468910</string>
+	<key>API_KEY</key>
+	<string>AIzaSyAzlj4APqi5S58nFtE52Da0fYBOHA2MhaY</string>
+	<key>BUNDLE_ID</key>
+	<string>com.google.firebase.VertexAISample</string>
+	<key>CLIENT_ID</key>
+	<string>123456789000-hjugbg6ud799v4c49dim8ce2usclthar.apps.googleusercontent.com</string>
+	<key>DATABASE_URL</key>
+	<string>https://mockproject-1234.firebaseio.com</string>
+	<key>GCM_SENDER_ID</key>
+	<string>123456789000</string>
+	<key>GOOGLE_APP_ID</key>
+	<string>1:123456789000:ios:f1bf012572b04063</string>
+	<key>IS_ADS_ENABLED</key>
+	<true/>
+	<key>IS_ANALYTICS_ENABLED</key>
+	<true/>
+	<key>IS_APPINVITE_ENABLED</key>
+	<true/>
+	<key>IS_GCM_ENABLED</key>
+	<true/>
+	<key>IS_SIGNIN_ENABLED</key>
+	<true/>
+	<key>PLIST_VERSION</key>
+	<string>1</string>
+	<key>PROJECT_ID</key>
+	<string>mockproject-1234</string>
+	<key>REVERSED_CLIENT_ID</key>
+	<string>com.googleusercontent.apps.123456789000-hjugbg6ud799v4c49dim8ce2usclthar</string>
+	<key>STORAGE_BUCKET</key>
+	<string>mockproject-1234.appspot.com</string>
+</dict>
+</plist>
diff --git a/FirebaseVertexAI/Sample/README.md b/FirebaseVertexAI/Sample/README.md
new file mode 100644
index 00000000000..a4a8b458df3
--- /dev/null
+++ b/FirebaseVertexAI/Sample/README.md
@@ -0,0 +1,16 @@
+# Vertex AI for Firebase Sample App
+
+You can try out the SDK quickly, see a complete implementation of various use
+cases, or use the sample app if you don't have your own Apple platforms app. To
+use the sample app, you'll need to perform the following steps:
+1. Download
+   [vertexai-preview-0.1.0.zip](https://github.com/firebase/firebase-ios-sdk/archive/refs/heads/vertexai-preview-0.1.0.zip)
+   for the latest release of the SDK
+1. Extract the zip and open `GenerativeAISample.xcodeproj` in the
+   `FirebaseVertexAI/Sample` directory
+1. [Register the sample app](https://firebase.google.com/docs/ios/setup#register-app)
+   in your Firebase project
+   - The default bundle ID is `com.google.firebase.VertexAISample`
+1. [Download the `GoogleService-Info.plist`](https://firebase.google.com/docs/ios/setup#add-config-file)
+   to the `FirebaseVertexAI/Sample` directory, overwriting the placeholder file
+   with the same name

From 9b678e4325aec19433be2d9a135066d99d329824 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 10 Apr 2024 14:27:48 -0400
Subject: [PATCH 092/124] Handle `totalBillableCharacters` optional decoding in
 Vertex AI (#12715)

---
 .../Sources/CountTokensRequest.swift          | 24 +++++++++++++++++--
 .../success-no-billable-characters.json       |  3 +++
 .../Tests/Unit/GenerativeModelTests.swift     | 15 ++++++++++++
 3 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100644 FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json

diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
index d294212658c..5072412cf9b 100644
--- a/FirebaseVertexAI/Sources/CountTokensRequest.swift
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -39,10 +39,30 @@ extension CountTokensRequest: GenerativeAIRequest {
 
 /// The model's response to a count tokens request.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct CountTokensResponse: Decodable {
+public struct CountTokensResponse {
   /// The total number of tokens in the input given to the model as a prompt.
   public let totalTokens: Int
 
-  /// The total number of billable characters in the input given to the model as a prompt.
+  /// The total number of billable characters in the text input given to the model as a prompt.
+  ///
+  /// > Important: This does not include billable image, video or other non-text input. See
+  /// [Vertex AI pricing](https://cloud.google.com/vertex-ai/generative-ai/pricing) for details.
   public let totalBillableCharacters: Int
 }
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case totalTokens
+    case totalBillableCharacters
+  }
+
+  public init(from decoder: any Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    totalTokens = try container.decode(Int.self, forKey: .totalTokens)
+    totalBillableCharacters = try container.decodeIfPresent(
+      Int.self,
+      forKey: .totalBillableCharacters
+    ) ?? 0
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json
new file mode 100644
index 00000000000..03425d51db0
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/CountTokenResponses/success-no-billable-characters.json
@@ -0,0 +1,3 @@
+{
+  "totalTokens": 258
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 13cbb6df94e..92585266891 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -946,6 +946,21 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(response.totalBillableCharacters, 16)
   }
 
+  func testCountTokens_succeeds_noBillableCharacters() async throws {
+    MockURLProtocol.requestHandler = try httpRequestHandler(
+      forResource: "success-no-billable-characters",
+      withExtension: "json"
+    )
+
+    let response = try await model.countTokens(ModelContent.Part.data(
+      mimetype: "image/jpeg",
+      Data()
+    ))
+
+    XCTAssertEqual(response.totalTokens, 258)
+    XCTAssertEqual(response.totalBillableCharacters, 0)
+  }
+
   func testCountTokens_modelNotFound() async throws {
     MockURLProtocol.requestHandler = try httpRequestHandler(
       forResource: "failure-model-not-found", withExtension: "json",

From b283714fd893379280e9683bababe37183aef47f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 10 Apr 2024 16:30:24 -0400
Subject: [PATCH 093/124] Add function calling sample in Vertex AI (#12745)

---
 .../Screens/FunctionCallingScreen.swift       | 128 +++++++++
 .../ViewModels/FunctionCallingViewModel.swift | 264 ++++++++++++++++++
 .../project.pbxproj                           |  32 +++
 .../GenerativeAISample/ContentView.swift      |   8 +
 4 files changed, 432 insertions(+)
 create mode 100644 FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
 create mode 100644 FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
new file mode 100644
index 00000000000..30d222e29bc
--- /dev/null
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
@@ -0,0 +1,128 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import GenerativeAIUIComponents
+import SwiftUI
+
+struct FunctionCallingScreen: View {
+  @EnvironmentObject
+  var viewModel: FunctionCallingViewModel
+
+  @State
+  private var userPrompt = "What is 100 Euros in U.S. Dollars?"
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      ScrollViewReader { scrollViewProxy in
+        List {
+          Text("Interact with a currency conversion API using function calling in Gemini.")
+          ForEach(viewModel.messages) { message in
+            MessageView(message: message)
+          }
+          if let error = viewModel.error {
+            ErrorView(error: error)
+              .tag("errorView")
+          }
+        }
+        .listStyle(.plain)
+        .onChange(of: viewModel.messages, perform: { newValue in
+          if viewModel.hasError {
+            // Wait for a short moment to make sure we can actually scroll to the bottom.
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo("errorView", anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          } else {
+            guard let lastMessage = viewModel.messages.last else { return }
+
+            // Wait for a short moment to make sure we can actually scroll to the bottom.
+            DispatchQueue.main.asyncAfter(deadline: .now() + 0.05) {
+              withAnimation {
+                scrollViewProxy.scrollTo(lastMessage.id, anchor: .bottom)
+              }
+              focusedField = .message
+            }
+          }
+        })
+      }
+      InputField("Message...", text: $userPrompt) {
+        Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
+          .font(.title)
+      }
+      .focused($focusedField, equals: .message)
+      .onSubmit { sendOrStop() }
+    }
+    .toolbar {
+      ToolbarItem(placement: .primaryAction) {
+        Button(action: newChat) {
+          Image(systemName: "square.and.pencil")
+        }
+      }
+    }
+    .navigationTitle("Function Calling")
+    .onAppear {
+      focusedField = .message
+    }
+  }
+
+  private func sendMessage() {
+    Task {
+      let prompt = userPrompt
+      userPrompt = ""
+      await viewModel.sendMessage(prompt, streaming: true)
+    }
+  }
+
+  private func sendOrStop() {
+    if viewModel.busy {
+      viewModel.stop()
+    } else {
+      sendMessage()
+    }
+  }
+
+  private func newChat() {
+    viewModel.startNewChat()
+  }
+}
+
+struct FunctionCallingScreen_Previews: PreviewProvider {
+  struct ContainerView: View {
+    @EnvironmentObject
+    var viewModel: FunctionCallingViewModel
+
+    var body: some View {
+      FunctionCallingScreen()
+        .onAppear {
+          viewModel.messages = ChatMessage.samples
+        }
+    }
+  }
+
+  static var previews: some View {
+    NavigationStack {
+      FunctionCallingScreen().environmentObject(FunctionCallingViewModel())
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
new file mode 100644
index 00000000000..f18c0e85fc1
--- /dev/null
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
@@ -0,0 +1,264 @@
+// Copyright 2023 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseVertexAI
+import Foundation
+import UIKit
+
+@MainActor
+class FunctionCallingViewModel: ObservableObject {
+  /// This array holds both the user's and the system's chat messages
+  @Published var messages = [ChatMessage]()
+
+  /// Indicates we're waiting for the model to finish
+  @Published var busy = false
+
+  @Published var error: Error?
+  var hasError: Bool {
+    return error != nil
+  }
+
+  /// Function calls pending processing
+  private var functionCalls = [FunctionCall]()
+
+  private var model: GenerativeModel
+  private var chat: Chat
+
+  private var chatTask: Task<Void, Never>?
+
+  init() {
+    model = VertexAI.vertexAI().generativeModel(
+      modelName: "gemini-1.0-pro",
+      tools: [Tool(functionDeclarations: [
+        FunctionDeclaration(
+          name: "get_exchange_rate",
+          description: "Get the exchange rate for currencies between countries",
+          parameters: [
+            "currency_from": Schema(
+              type: .string,
+              format: "enum",
+              description: "The currency to convert from in ISO 4217 format",
+              enumValues: ["USD", "EUR", "JPY", "GBP", "AUD", "CAD"]
+            ),
+            "currency_to": Schema(
+              type: .string,
+              format: "enum",
+              description: "The currency to convert to in ISO 4217 format",
+              enumValues: ["USD", "EUR", "JPY", "GBP", "AUD", "CAD"]
+            ),
+          ],
+          requiredParameters: ["currency_from", "currency_to"]
+        ),
+      ])]
+    )
+    chat = model.startChat()
+  }
+
+  func sendMessage(_ text: String, streaming: Bool = true) async {
+    error = nil
+    chatTask?.cancel()
+
+    chatTask = Task {
+      busy = true
+      defer {
+        busy = false
+      }
+
+      // first, add the user's message to the chat
+      let userMessage = ChatMessage(message: text, participant: .user)
+      messages.append(userMessage)
+
+      // add a pending message while we're waiting for a response from the backend
+      let systemMessage = ChatMessage.pending(participant: .system)
+      messages.append(systemMessage)
+
+      print(messages)
+      do {
+        repeat {
+          if streaming {
+            try await internalSendMessageStreaming(text)
+          } else {
+            try await internalSendMessage(text)
+          }
+        } while !functionCalls.isEmpty
+      } catch {
+        self.error = error
+        print(error.localizedDescription)
+        messages.removeLast()
+      }
+    }
+  }
+
+  func startNewChat() {
+    stop()
+    error = nil
+    chat = model.startChat()
+    messages.removeAll()
+  }
+
+  func stop() {
+    chatTask?.cancel()
+    error = nil
+  }
+
+  private func internalSendMessageStreaming(_ text: String) async throws {
+    let functionResponses = try await processFunctionCalls()
+    let responseStream: AsyncThrowingStream<GenerateContentResponse, Error>
+    if functionResponses.isEmpty {
+      responseStream = chat.sendMessageStream(text)
+    } else {
+      for functionResponse in functionResponses {
+        messages.insert(functionResponse.chatMessage(), at: messages.count - 1)
+      }
+      responseStream = chat.sendMessageStream(functionResponses.modelContent())
+    }
+    for try await chunk in responseStream {
+      processResponseContent(content: chunk)
+    }
+  }
+
+  private func internalSendMessage(_ text: String) async throws {
+    let functionResponses = try await processFunctionCalls()
+    let response: GenerateContentResponse
+    if functionResponses.isEmpty {
+      response = try await chat.sendMessage(text)
+    } else {
+      for functionResponse in functionResponses {
+        messages.insert(functionResponse.chatMessage(), at: messages.count - 1)
+      }
+      response = try await chat.sendMessage(functionResponses.modelContent())
+    }
+    processResponseContent(content: response)
+  }
+
+  func processResponseContent(content: GenerateContentResponse) {
+    guard let candidate = content.candidates.first else {
+      fatalError("No candidate.")
+    }
+
+    for part in candidate.content.parts {
+      switch part {
+      case let .text(text):
+        // replace pending message with backend response
+        messages[messages.count - 1].message += text
+        messages[messages.count - 1].pending = false
+      case let .functionCall(functionCall):
+        messages.insert(functionCall.chatMessage(), at: messages.count - 1)
+        functionCalls.append(functionCall)
+      case .data, .functionResponse:
+        fatalError("Unsupported response content.")
+      }
+    }
+  }
+
+  func processFunctionCalls() async throws -> [FunctionResponse] {
+    var functionResponses = [FunctionResponse]()
+    for functionCall in functionCalls {
+      switch functionCall.name {
+      case "get_exchange_rate":
+        let exchangeRates = getExchangeRate(args: functionCall.args)
+        functionResponses.append(FunctionResponse(
+          name: "get_exchange_rate",
+          response: exchangeRates
+        ))
+      default:
+        fatalError("Unknown function named \"\(functionCall.name)\".")
+      }
+    }
+    functionCalls = []
+
+    return functionResponses
+  }
+
+  // MARK: - Callable Functions
+
+  func getExchangeRate(args: JSONObject) -> JSONObject {
+    // 1. Validate and extract the parameters provided by the model (from a `FunctionCall`)
+    guard case let .string(from) = args["currency_from"] else {
+      fatalError("Missing `currency_from` parameter.")
+    }
+    guard case let .string(to) = args["currency_to"] else {
+      fatalError("Missing `currency_to` parameter.")
+    }
+
+    // 2. Get the exchange rate
+    let allRates: [String: [String: Double]] = [
+      "AUD": ["CAD": 0.89265, "EUR": 0.6072, "GBP": 0.51714, "JPY": 97.75, "USD": 0.66379],
+      "CAD": ["AUD": 1.1203, "EUR": 0.68023, "GBP": 0.57933, "JPY": 109.51, "USD": 0.74362],
+      "EUR": ["AUD": 1.6469, "CAD": 1.4701, "GBP": 0.85168, "JPY": 160.99, "USD": 1.0932],
+      "GBP": ["AUD": 1.9337, "CAD": 1.7261, "EUR": 1.1741, "JPY": 189.03, "USD": 1.2836],
+      "JPY": ["AUD": 0.01023, "CAD": 0.00913, "EUR": 0.00621, "GBP": 0.00529, "USD": 0.00679],
+      "USD": ["AUD": 1.5065, "CAD": 1.3448, "EUR": 0.91475, "GBP": 0.77907, "JPY": 147.26],
+    ]
+    guard let fromRates = allRates[from] else {
+      return ["error": .string("No data for currency \(from).")]
+    }
+    guard let toRate = fromRates[to] else {
+      return ["error": .string("No data for currency \(to).")]
+    }
+
+    // 3. Return the exchange rates as a JSON object (returned to the model in a `FunctionResponse`)
+    return ["rates": .number(toRate)]
+  }
+}
+
+private extension FunctionCall {
+  func chatMessage() -> ChatMessage {
+    let encoder = JSONEncoder()
+    encoder.outputFormatting = .prettyPrinted
+
+    let jsonData: Data
+    do {
+      jsonData = try encoder.encode(self)
+    } catch {
+      fatalError("JSON Encoding Failed: \(error.localizedDescription)")
+    }
+    guard let json = String(data: jsonData, encoding: .utf8) else {
+      fatalError("Failed to convert JSON data to a String.")
+    }
+    let messageText = "Function call requested by model:\n```\n\(json)\n```"
+
+    return ChatMessage(message: messageText, participant: .system)
+  }
+}
+
+private extension FunctionResponse {
+  func chatMessage() -> ChatMessage {
+    let encoder = JSONEncoder()
+    encoder.outputFormatting = .prettyPrinted
+
+    let jsonData: Data
+    do {
+      jsonData = try encoder.encode(self)
+    } catch {
+      fatalError("JSON Encoding Failed: \(error.localizedDescription)")
+    }
+    guard let json = String(data: jsonData, encoding: .utf8) else {
+      fatalError("Failed to convert JSON data to a String.")
+    }
+    let messageText = "Function response returned by app:\n```\n\(json)\n```"
+
+    return ChatMessage(message: messageText, participant: .user)
+  }
+}
+
+private extension [FunctionResponse] {
+  func modelContent() -> [ModelContent] {
+    return self.map { ModelContent(
+      role: "function",
+      parts: [ModelContent.Part.functionResponse($0)]
+    )
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 01336b9cf19..0e1bf8211f0 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -10,6 +10,8 @@
 		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
 		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
+		86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */; };
+		86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */; };
 		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
 		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
@@ -31,6 +33,8 @@
 
 /* Begin PBXFileReference section */
 		869200B22B879C4F00482873 /* GoogleService-Info.plist */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text.plist.xml; path = "GoogleService-Info.plist"; sourceTree = "<group>"; };
+		86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = FunctionCallingScreen.swift; sourceTree = "<group>"; };
+		86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = FunctionCallingViewModel.swift; sourceTree = "<group>"; };
 		8802666F2B0FC39000CF7CB6 /* PhotoReasoningViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningViewModel.swift; sourceTree = "<group>"; };
 		880266752B0FC39000CF7CB6 /* PhotoReasoningScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PhotoReasoningScreen.swift; sourceTree = "<group>"; };
 		88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeScreen.swift; sourceTree = "<group>"; };
@@ -71,6 +75,31 @@
 /* End PBXFrameworksBuildPhase section */
 
 /* Begin PBXGroup section */
+		86C1F47F2BC726150026816F /* Screens */ = {
+			isa = PBXGroup;
+			children = (
+				86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */,
+			);
+			path = Screens;
+			sourceTree = "<group>";
+		};
+		86C1F4812BC726150026816F /* ViewModels */ = {
+			isa = PBXGroup;
+			children = (
+				86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */,
+			);
+			path = ViewModels;
+			sourceTree = "<group>";
+		};
+		86C1F4822BC726150026816F /* FunctionCallingSample */ = {
+			isa = PBXGroup;
+			children = (
+				86C1F4812BC726150026816F /* ViewModels */,
+				86C1F47F2BC726150026816F /* Screens */,
+			);
+			path = FunctionCallingSample;
+			sourceTree = "<group>";
+		};
 		8802666E2B0FC39000CF7CB6 /* ViewModels */ = {
 			isa = PBXGroup;
 			children = (
@@ -119,6 +148,7 @@
 				8848C8452B0D051E007B434F /* GenerativeAITextSample */,
 				8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */,
 				88E10F432B110D5300C08E95 /* ChatSample */,
+				86C1F4822BC726150026816F /* FunctionCallingSample */,
 				8848C8302B0D04BC007B434F /* Products */,
 				88209C222B0FBE1700F64795 /* Frameworks */,
 			);
@@ -327,7 +357,9 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */,
 				886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */,
+				86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */,
 				886F95DE2B17D5010036F07A /* ChatMessage.swift in Sources */,
 				88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */,
 				8848C8352B0D04BC007B434F /* ContentView.swift in Sources */,
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
index 34331bf6eb7..2841d634a6c 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
@@ -18,6 +18,9 @@ struct ContentView: View {
   @StateObject
   var viewModel = ConversationViewModel()
 
+  @StateObject
+  var functionCallingViewModel = FunctionCallingViewModel()
+
   var body: some View {
     NavigationStack {
       List {
@@ -37,6 +40,11 @@ struct ContentView: View {
         } label: {
           Label("Chat", systemImage: "ellipsis.message.fill")
         }
+        NavigationLink {
+          FunctionCallingScreen().environmentObject(functionCallingViewModel)
+        } label: {
+          Label("Function Calling", systemImage: "function")
+        }
       }
       .navigationTitle("Generative AI Samples")
     }

From e587733ebf7d9adf03f3de08e7afa0bf9e1793b0 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 10 Apr 2024 20:46:45 -0400
Subject: [PATCH 094/124] Add forced function calling support in Vertex AI
 (#12748)

---
 .../Sources/FunctionCalling.swift             | 45 +++++++++++++++++++
 .../Sources/GenerateContentRequest.swift      |  2 +
 .../Sources/GenerativeModel.swift             |  8 ++++
 FirebaseVertexAI/Sources/VertexAI.swift       |  3 ++
 4 files changed, 58 insertions(+)

diff --git a/FirebaseVertexAI/Sources/FunctionCalling.swift b/FirebaseVertexAI/Sources/FunctionCalling.swift
index 7e4d84d857e..8860344c7cd 100644
--- a/FirebaseVertexAI/Sources/FunctionCalling.swift
+++ b/FirebaseVertexAI/Sources/FunctionCalling.swift
@@ -177,6 +177,51 @@ public struct Tool: Encodable {
   }
 }
 
+/// Configuration for specifying function calling behavior.
+public struct FunctionCallingConfig: Encodable {
+  /// Defines the execution behavior for function calling by defining the
+  /// execution mode.
+  public enum Mode: String, Encodable {
+    /// The default behavior for function calling. The model calls functions to answer queries at
+    /// its discretion.
+    case auto = "AUTO"
+
+    /// The model always predicts a provided function call to answer every query.
+    case any = "ANY"
+
+    /// The model will never predict a function call to answer a query. This can also be achieved by
+    /// not passing any tools to the model.
+    case none = "NONE"
+  }
+
+  /// Specifies the mode in which function calling should execute. If
+  /// unspecified, the default value will be set to AUTO.
+  let mode: Mode?
+
+  /// A set of function names that, when provided, limits the functions the model
+  /// will call.
+  ///
+  /// This should only be set when the Mode is ANY. Function names
+  /// should match [FunctionDeclaration.name]. With mode set to ANY, model will
+  /// predict a function call from the set of function names provided.
+  let allowedFunctionNames: [String]?
+
+  public init(mode: FunctionCallingConfig.Mode? = nil, allowedFunctionNames: [String]? = nil) {
+    self.mode = mode
+    self.allowedFunctionNames = allowedFunctionNames
+  }
+}
+
+/// Tool configuration for any `Tool` specified in the request.
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+public struct ToolConfig: Encodable {
+  let functionCallingConfig: FunctionCallingConfig?
+
+  public init(functionCallingConfig: FunctionCallingConfig? = nil) {
+    self.functionCallingConfig = functionCallingConfig
+  }
+}
+
 /// Result output from a ``FunctionCall``.
 ///
 /// Contains a string representing the `FunctionDeclaration.name` and a structured JSON object
diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
index 535ac4da3ae..e0e9c2fe0c8 100644
--- a/FirebaseVertexAI/Sources/GenerateContentRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -22,6 +22,7 @@ struct GenerateContentRequest {
   let generationConfig: GenerationConfig?
   let safetySettings: [SafetySetting]?
   let tools: [Tool]?
+  let toolConfig: ToolConfig?
   let isStreaming: Bool
   let options: RequestOptions
 }
@@ -33,6 +34,7 @@ extension GenerateContentRequest: Encodable {
     case generationConfig
     case safetySettings
     case tools
+    case toolConfig
   }
 }
 
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index 7eab63a1d7a..124656dd639 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -37,6 +37,9 @@ public final class GenerativeModel {
   /// A list of tools the model may use to generate the next response.
   let tools: [Tool]?
 
+  /// Tool configuration for any `Tool` specified in the request.
+  let toolConfig: ToolConfig?
+
   /// Configuration parameters for sending requests to the backend.
   let requestOptions: RequestOptions
 
@@ -49,6 +52,7 @@ public final class GenerativeModel {
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
+  ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
@@ -56,6 +60,7 @@ public final class GenerativeModel {
        generationConfig: GenerationConfig? = nil,
        safetySettings: [SafetySetting]? = nil,
        tools: [Tool]?,
+       toolConfig: ToolConfig? = nil,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
        urlSession: URLSession = .shared) {
@@ -68,6 +73,7 @@ public final class GenerativeModel {
     self.generationConfig = generationConfig
     self.safetySettings = safetySettings
     self.tools = tools
+    self.toolConfig = toolConfig
     self.requestOptions = requestOptions
 
     Logging.default.info("""
@@ -114,6 +120,7 @@ public final class GenerativeModel {
                                                               generationConfig: generationConfig,
                                                               safetySettings: safetySettings,
                                                               tools: tools,
+                                                              toolConfig: toolConfig,
                                                               isStreaming: false,
                                                               options: requestOptions)
       response = try await generativeAIService.loadRequest(request: generateContentRequest)
@@ -186,6 +193,7 @@ public final class GenerativeModel {
                                                         generationConfig: generationConfig,
                                                         safetySettings: safetySettings,
                                                         tools: tools,
+                                                        toolConfig: toolConfig,
                                                         isStreaming: true,
                                                         options: requestOptions)
 
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 1fcfd998b45..843a3d09552 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -58,11 +58,13 @@ public class VertexAI: NSObject {
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
+  ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
                               tools: [Tool]? = nil,
+                              toolConfig: ToolConfig? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
     let modelResourceName = modelResourceName(modelName: modelName, location: location)
@@ -77,6 +79,7 @@ public class VertexAI: NSObject {
       generationConfig: generationConfig,
       safetySettings: safetySettings,
       tools: tools,
+      toolConfig: toolConfig,
       requestOptions: requestOptions,
       appCheck: appCheck
     )

From 14b5a0261b40b7e03db3ec9ba480bc28d8c13742 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 11 Apr 2024 09:52:23 -0400
Subject: [PATCH 095/124] Add system instruction support in Vertex AI (#12749)

---
 FirebaseVertexAI/Sources/GenerateContentRequest.swift | 2 ++
 FirebaseVertexAI/Sources/GenerativeModel.swift        | 9 +++++++++
 FirebaseVertexAI/Sources/VertexAI.swift               | 4 ++++
 FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift    | 9 ++++++++-
 4 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sources/GenerateContentRequest.swift b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
index e0e9c2fe0c8..05abadf54e1 100644
--- a/FirebaseVertexAI/Sources/GenerateContentRequest.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentRequest.swift
@@ -23,6 +23,7 @@ struct GenerateContentRequest {
   let safetySettings: [SafetySetting]?
   let tools: [Tool]?
   let toolConfig: ToolConfig?
+  let systemInstruction: ModelContent?
   let isStreaming: Bool
   let options: RequestOptions
 }
@@ -35,6 +36,7 @@ extension GenerateContentRequest: Encodable {
     case safetySettings
     case tools
     case toolConfig
+    case systemInstruction
   }
 }
 
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index 124656dd639..cdd6d175c0a 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -40,6 +40,9 @@ public final class GenerativeModel {
   /// Tool configuration for any `Tool` specified in the request.
   let toolConfig: ToolConfig?
 
+  /// Instructions that direct the model to behave a certain way.
+  let systemInstruction: ModelContent?
+
   /// Configuration parameters for sending requests to the backend.
   let requestOptions: RequestOptions
 
@@ -53,6 +56,8 @@ public final class GenerativeModel {
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
+  ///   - systemInstruction: Instructions that direct the model to behave a certain way; currently
+  ///     only text content is supported.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   ///   - urlSession: The `URLSession` to use for requests; defaults to `URLSession.shared`.
   init(name: String,
@@ -61,6 +66,7 @@ public final class GenerativeModel {
        safetySettings: [SafetySetting]? = nil,
        tools: [Tool]?,
        toolConfig: ToolConfig? = nil,
+       systemInstruction: ModelContent? = nil,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
        urlSession: URLSession = .shared) {
@@ -74,6 +80,7 @@ public final class GenerativeModel {
     self.safetySettings = safetySettings
     self.tools = tools
     self.toolConfig = toolConfig
+    self.systemInstruction = systemInstruction
     self.requestOptions = requestOptions
 
     Logging.default.info("""
@@ -121,6 +128,7 @@ public final class GenerativeModel {
                                                               safetySettings: safetySettings,
                                                               tools: tools,
                                                               toolConfig: toolConfig,
+                                                              systemInstruction: systemInstruction,
                                                               isStreaming: false,
                                                               options: requestOptions)
       response = try await generativeAIService.loadRequest(request: generateContentRequest)
@@ -194,6 +202,7 @@ public final class GenerativeModel {
                                                         safetySettings: safetySettings,
                                                         tools: tools,
                                                         toolConfig: toolConfig,
+                                                        systemInstruction: systemInstruction,
                                                         isStreaming: true,
                                                         options: requestOptions)
 
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 843a3d09552..7a7865e86d2 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -59,12 +59,15 @@ public class VertexAI: NSObject {
   ///   - safetySettings: A value describing what types of harmful content your model should allow.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - toolConfig: Tool configuration for any `Tool` specified in the request.
+  ///   - systemInstruction: Instructions that direct the model to behave a certain way; currently
+  ///     only text content is supported.
   ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func generativeModel(modelName: String,
                               generationConfig: GenerationConfig? = nil,
                               safetySettings: [SafetySetting]? = nil,
                               tools: [Tool]? = nil,
                               toolConfig: ToolConfig? = nil,
+                              systemInstruction: ModelContent? = nil,
                               requestOptions: RequestOptions = RequestOptions())
     -> GenerativeModel {
     let modelResourceName = modelResourceName(modelName: modelName, location: location)
@@ -80,6 +83,7 @@ public class VertexAI: NSObject {
       safetySettings: safetySettings,
       tools: tools,
       toolConfig: toolConfig,
+      systemInstruction: systemInstruction,
       requestOptions: requestOptions,
       appCheck: appCheck
     )
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 81b81414dcb..e9312edfee0 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -32,6 +32,7 @@ final class VertexAIAPITests: XCTestCase {
                                   maxOutputTokens: 256,
                                   stopSequences: ["..."])
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
+    let systemInstruction = ModelContent(role: "system", parts: [.text("Talk like a pirate.")])
 
     // Instantiate Vertex AI SDK - Default App
     let vertexAI = VertexAI.vertexAI()
@@ -53,11 +54,17 @@ final class VertexAIAPITests: XCTestCase {
       generationConfig: config
     )
 
+    let _ = vertexAI.generativeModel(
+      modelName: "gemini-1.0-pro",
+      systemInstruction: systemInstruction
+    )
+
     // All arguments passed.
     let genAI = vertexAI.generativeModel(
       modelName: "gemini-1.0-pro",
       generationConfig: config, // Optional
-      safetySettings: filters // Optional
+      safetySettings: filters, // Optional
+      systemInstruction: systemInstruction // Optional
     )
 
     // Full Typed Usage

From 1a8d228a8105ec521bbf3d3aec7f00740a3fe502 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 11 Apr 2024 14:51:32 -0400
Subject: [PATCH 096/124] Remove FirebaseAppCheck module from Vertex AI sample
 app (#12754)

---
 .../Sample/GenerativeAISample.xcodeproj/project.pbxproj    | 7 -------
 .../Sample/GenerativeAISample/GenerativeAISampleApp.swift  | 3 +++
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
index 0e1bf8211f0..1914a9838e6 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
@@ -7,7 +7,6 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
-		868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33652BB476FA00304BB1 /* FirebaseAppCheck */; };
 		868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */ = {isa = PBXBuildFile; productRef = 868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */; };
 		869200B32B879C4F00482873 /* GoogleService-Info.plist in Resources */ = {isa = PBXBuildFile; fileRef = 869200B22B879C4F00482873 /* GoogleService-Info.plist */; };
 		86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F47E2BC726150026816F /* FunctionCallingScreen.swift */; };
@@ -67,7 +66,6 @@
 			files = (
 				868A33682BB476FA00304BB1 /* FirebaseVertexAI-Preview in Frameworks */,
 				886F95D82B17BA420036F07A /* MarkdownUI in Frameworks */,
-				868A33662BB476FA00304BB1 /* FirebaseAppCheck in Frameworks */,
 				886F95E32B17D6630036F07A /* GenerativeAIUIComponents in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
@@ -294,7 +292,6 @@
 			packageProductDependencies = (
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
-				868A33652BB476FA00304BB1 /* FirebaseAppCheck */,
 				868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAISample;
@@ -607,10 +604,6 @@
 /* End XCRemoteSwiftPackageReference section */
 
 /* Begin XCSwiftPackageProductDependency section */
-		868A33652BB476FA00304BB1 /* FirebaseAppCheck */ = {
-			isa = XCSwiftPackageProductDependency;
-			productName = FirebaseAppCheck;
-		};
 		868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = "FirebaseVertexAI-Preview";
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
index 509087a5e07..18723a459c5 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
@@ -18,6 +18,9 @@ import SwiftUI
 @main
 struct GenerativeAISampleApp: App {
   init() {
+    // Recommendation: Protect your Vertex AI API resources from abuse by preventing unauthorized
+    // clients using App Check; see https://firebase.google.com/docs/app-check#get_started.
+
     FirebaseApp.configure()
 
     if let firebaseApp = FirebaseApp.app(), firebaseApp.options.projectID == "mockproject-1234" {

From fedd7c854dbab737fb7fd17fef29482306a8189a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 11 Apr 2024 17:51:09 -0400
Subject: [PATCH 097/124] Rename GenerativeAISample to VertexAISample (#12755)

---
 FirebaseVertexAI/Sample/README.md             |  2 +-
 .../project.pbxproj                           | 38 +++++++++----------
 .../AccentColor.colorset/Contents.json        |  0
 .../AppIcon.appiconset/Contents.json          |  0
 .../Assets.xcassets/Contents.json             |  0
 .../ContentView.swift                         |  0
 .../Preview Assets.xcassets/Contents.json     |  0
 .../VertexAISampleApp.swift}                  |  2 +-
 scripts/build.sh                              |  4 +-
 9 files changed, 23 insertions(+), 23 deletions(-)
 rename FirebaseVertexAI/Sample/{GenerativeAISample.xcodeproj => VertexAISample.xcodeproj}/project.pbxproj (94%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Assets.xcassets/AccentColor.colorset/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Assets.xcassets/AppIcon.appiconset/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Assets.xcassets/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/ContentView.swift (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample => VertexAISample}/Preview Content/Preview Assets.xcassets/Contents.json (100%)
 rename FirebaseVertexAI/Sample/{GenerativeAISample/GenerativeAISampleApp.swift => VertexAISample/VertexAISampleApp.swift} (97%)

diff --git a/FirebaseVertexAI/Sample/README.md b/FirebaseVertexAI/Sample/README.md
index a4a8b458df3..8a7afa02977 100644
--- a/FirebaseVertexAI/Sample/README.md
+++ b/FirebaseVertexAI/Sample/README.md
@@ -6,7 +6,7 @@ use the sample app, you'll need to perform the following steps:
 1. Download
    [vertexai-preview-0.1.0.zip](https://github.com/firebase/firebase-ios-sdk/archive/refs/heads/vertexai-preview-0.1.0.zip)
    for the latest release of the SDK
-1. Extract the zip and open `GenerativeAISample.xcodeproj` in the
+1. Extract the zip and open `VertexAISample.xcodeproj` in the
    `FirebaseVertexAI/Sample` directory
 1. [Register the sample app](https://firebase.google.com/docs/ios/setup#register-app)
    in your Firebase project
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
similarity index 94%
rename from FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
rename to FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
index 1914a9838e6..a5c19152df7 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
@@ -13,7 +13,7 @@
 		86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 86C1F4802BC726150026816F /* FunctionCallingViewModel.swift */; };
 		88263BF02B239C09008AB09B /* ErrorView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 88263BEE2B239BFE008AB09B /* ErrorView.swift */; };
 		88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 889873842B208563005B4896 /* ErrorDetailsView.swift */; };
-		8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */; };
+		8848C8332B0D04BC007B434F /* VertexAISampleApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8322B0D04BC007B434F /* VertexAISampleApp.swift */; };
 		8848C8352B0D04BC007B434F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8848C8342B0D04BC007B434F /* ContentView.swift */; };
 		8848C8372B0D04BD007B434F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8362B0D04BD007B434F /* Assets.xcassets */; };
 		8848C83A2B0D04BD007B434F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8848C8392B0D04BD007B434F /* Preview Assets.xcassets */; };
@@ -39,8 +39,8 @@
 		88209C1B2B0FBDC300F64795 /* SummarizeScreen.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeScreen.swift; sourceTree = "<group>"; };
 		88209C1D2B0FBDC300F64795 /* SummarizeViewModel.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SummarizeViewModel.swift; sourceTree = "<group>"; };
 		88263BEE2B239BFE008AB09B /* ErrorView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = ErrorView.swift; sourceTree = "<group>"; };
-		8848C82F2B0D04BC007B434F /* GenerativeAISample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = GenerativeAISample.app; sourceTree = BUILT_PRODUCTS_DIR; };
-		8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerativeAISampleApp.swift; sourceTree = "<group>"; };
+		8848C82F2B0D04BC007B434F /* VertexAISample.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = VertexAISample.app; sourceTree = BUILT_PRODUCTS_DIR; };
+		8848C8322B0D04BC007B434F /* VertexAISampleApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = VertexAISampleApp.swift; sourceTree = "<group>"; };
 		8848C8342B0D04BC007B434F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = "<group>"; };
 		8848C8362B0D04BD007B434F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		8848C8392B0D04BD007B434F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = "Preview Assets.xcassets"; sourceTree = "<group>"; };
@@ -142,7 +142,7 @@
 			children = (
 				88B8A9352B0FCBA700424728 /* GenerativeAIUIComponents */,
 				869200B22B879C4F00482873 /* GoogleService-Info.plist */,
-				8848C8312B0D04BC007B434F /* GenerativeAISample */,
+				8848C8312B0D04BC007B434F /* VertexAISample */,
 				8848C8452B0D051E007B434F /* GenerativeAITextSample */,
 				8848C8572B0D056C007B434F /* GenerativeAIMultimodalSample */,
 				88E10F432B110D5300C08E95 /* ChatSample */,
@@ -155,20 +155,20 @@
 		8848C8302B0D04BC007B434F /* Products */ = {
 			isa = PBXGroup;
 			children = (
-				8848C82F2B0D04BC007B434F /* GenerativeAISample.app */,
+				8848C82F2B0D04BC007B434F /* VertexAISample.app */,
 			);
 			name = Products;
 			sourceTree = "<group>";
 		};
-		8848C8312B0D04BC007B434F /* GenerativeAISample */ = {
+		8848C8312B0D04BC007B434F /* VertexAISample */ = {
 			isa = PBXGroup;
 			children = (
-				8848C8322B0D04BC007B434F /* GenerativeAISampleApp.swift */,
+				8848C8322B0D04BC007B434F /* VertexAISampleApp.swift */,
 				8848C8342B0D04BC007B434F /* ContentView.swift */,
 				8848C8362B0D04BD007B434F /* Assets.xcassets */,
 				8848C8382B0D04BD007B434F /* Preview Content */,
 			);
-			path = GenerativeAISample;
+			path = VertexAISample;
 			sourceTree = "<group>";
 		};
 		8848C8382B0D04BD007B434F /* Preview Content */ = {
@@ -276,9 +276,9 @@
 /* End PBXGroup section */
 
 /* Begin PBXNativeTarget section */
-		8848C82E2B0D04BC007B434F /* GenerativeAISample */ = {
+		8848C82E2B0D04BC007B434F /* VertexAISample */ = {
 			isa = PBXNativeTarget;
-			buildConfigurationList = 8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */;
+			buildConfigurationList = 8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "VertexAISample" */;
 			buildPhases = (
 				8848C82B2B0D04BC007B434F /* Sources */,
 				8848C82C2B0D04BC007B434F /* Frameworks */,
@@ -288,14 +288,14 @@
 			);
 			dependencies = (
 			);
-			name = GenerativeAISample;
+			name = VertexAISample;
 			packageProductDependencies = (
 				886F95D72B17BA420036F07A /* MarkdownUI */,
 				886F95E22B17D6630036F07A /* GenerativeAIUIComponents */,
 				868A33672BB476FA00304BB1 /* FirebaseVertexAI-Preview */,
 			);
 			productName = GenerativeAISample;
-			productReference = 8848C82F2B0D04BC007B434F /* GenerativeAISample.app */;
+			productReference = 8848C82F2B0D04BC007B434F /* VertexAISample.app */;
 			productType = "com.apple.product-type.application";
 		};
 /* End PBXNativeTarget section */
@@ -313,7 +313,7 @@
 					};
 				};
 			};
-			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */;
+			buildConfigurationList = 8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "VertexAISample" */;
 			compatibilityVersion = "Xcode 14.0";
 			developmentRegion = en;
 			hasScannedForEncodings = 0;
@@ -331,7 +331,7 @@
 			projectDirPath = "";
 			projectRoot = "";
 			targets = (
-				8848C82E2B0D04BC007B434F /* GenerativeAISample */,
+				8848C82E2B0D04BC007B434F /* VertexAISample */,
 			);
 		};
 /* End PBXProject section */
@@ -361,7 +361,7 @@
 				88263BF12B239C11008AB09B /* ErrorDetailsView.swift in Sources */,
 				8848C8352B0D04BC007B434F /* ContentView.swift in Sources */,
 				886F95D52B17BA010036F07A /* SummarizeScreen.swift in Sources */,
-				8848C8332B0D04BC007B434F /* GenerativeAISampleApp.swift in Sources */,
+				8848C8332B0D04BC007B434F /* VertexAISampleApp.swift in Sources */,
 				886F95E02B17D5010036F07A /* ConversationViewModel.swift in Sources */,
 				886F95DD2B17D5010036F07A /* MessageView.swift in Sources */,
 				886F95DC2B17BAEF0036F07A /* PhotoReasoningScreen.swift in Sources */,
@@ -501,7 +501,7 @@
 				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
 				DEVELOPMENT_TEAM = "";
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
@@ -531,7 +531,7 @@
 				ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_ASSET_PATHS = "\"GenerativeAISample/Preview Content\"";
+				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
 				DEVELOPMENT_TEAM = "";
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
@@ -557,7 +557,7 @@
 /* End XCBuildConfiguration section */
 
 /* Begin XCConfigurationList section */
-		8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "GenerativeAISample" */ = {
+		8848C82A2B0D04BC007B434F /* Build configuration list for PBXProject "VertexAISample" */ = {
 			isa = XCConfigurationList;
 			buildConfigurations = (
 				8848C83B2B0D04BD007B434F /* Debug */,
@@ -566,7 +566,7 @@
 			defaultConfigurationIsVisible = 0;
 			defaultConfigurationName = Release;
 		};
-		8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "GenerativeAISample" */ = {
+		8848C83D2B0D04BD007B434F /* Build configuration list for PBXNativeTarget "VertexAISample" */ = {
 			isa = XCConfigurationList;
 			buildConfigurations = (
 				8848C83E2B0D04BD007B434F /* Debug */,
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AccentColor.colorset/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AccentColor.colorset/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AccentColor.colorset/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/AppIcon.appiconset/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Assets.xcassets/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Assets.xcassets/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift b/FirebaseVertexAI/Sample/VertexAISample/ContentView.swift
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/ContentView.swift
rename to FirebaseVertexAI/Sample/VertexAISample/ContentView.swift
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json b/FirebaseVertexAI/Sample/VertexAISample/Preview Content/Preview Assets.xcassets/Contents.json
similarity index 100%
rename from FirebaseVertexAI/Sample/GenerativeAISample/Preview Content/Preview Assets.xcassets/Contents.json
rename to FirebaseVertexAI/Sample/VertexAISample/Preview Content/Preview Assets.xcassets/Contents.json
diff --git a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
similarity index 97%
rename from FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
rename to FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
index 18723a459c5..4772bcc2d7d 100644
--- a/FirebaseVertexAI/Sample/GenerativeAISample/GenerativeAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
@@ -16,7 +16,7 @@ import FirebaseCore
 import SwiftUI
 
 @main
-struct GenerativeAISampleApp: App {
+struct VertexAISampleApp: App {
   init() {
     // Recommendation: Protect your Vertex AI API resources from abuse by preventing unauthorized
     // clients using App Check; see https://firebase.google.com/docs/app-check#get_started.
diff --git a/scripts/build.sh b/scripts/build.sh
index fcaba2a263d..55d9b7cf39c 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -492,8 +492,8 @@ case "$product-$platform-$method" in
 
   VertexSample-*-*)
     RunXcodebuild \
-      -project 'FirebaseVertexAI/Sample/GenerativeAISample.xcodeproj' \
-      -scheme "GenerativeAISample" \
+      -project 'FirebaseVertexAI/Sample/VertexAISample.xcodeproj' \
+      -scheme "VertexAISample" \
       "${xcb_flags[@]}" \
       build
     ;;

From 46be683c437b016048333f726b50bad05ff61270 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Fri, 12 Apr 2024 17:21:01 -0400
Subject: [PATCH 098/124] Add `usageMetadata` to `GenerateContentResponse` in
 Vertex AI (#12777)

---
 .../ChatSample/Views/ErrorDetailsView.swift   |  6 +--
 .../Sample/ChatSample/Views/ErrorView.swift   |  3 +-
 .../Sources/GenerateContentResponse.swift     | 38 ++++++++++++++++-
 .../Tests/Unit/GenerativeModelTests.swift     | 41 +++++++++++++++++++
 4 files changed, 81 insertions(+), 7 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
index 7b3b78753db..dc5ce8f9561 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorDetailsView.swift
@@ -176,8 +176,7 @@ struct ErrorDetailsView: View {
       ],
       finishReason: FinishReason.maxTokens,
       citationMetadata: nil),
-    ],
-    promptFeedback: nil)
+    ])
   )
 
   return ErrorDetailsView(error: error)
@@ -200,8 +199,7 @@ struct ErrorDetailsView: View {
       ],
       finishReason: FinishReason.other,
       citationMetadata: nil),
-    ],
-    promptFeedback: nil)
+    ])
   )
 
   return ErrorDetailsView(error: error)
diff --git a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
index 1307eee62d4..d4db2d67dc5 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Views/ErrorView.swift
@@ -51,8 +51,7 @@ struct ErrorView: View {
         ],
         finishReason: FinishReason.other,
         citationMetadata: nil),
-      ],
-      promptFeedback: nil)
+      ])
     )
     List {
       MessageView(message: ChatMessage.samples[0])
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index f43da308b13..29c5b14f552 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -17,6 +17,18 @@ import Foundation
 /// The model's response to a generate content request.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public struct GenerateContentResponse {
+  /// Token usage metadata for processing the generate content request.
+  public struct UsageMetadata {
+    /// The number of tokens in the request prompt.
+    public let promptTokenCount: Int
+
+    /// The total number of tokens across the generated response candidates.
+    public let candidatesTokenCount: Int
+
+    /// The total number of tokens in both the request and response.
+    public let totalTokenCount: Int
+  }
+
   /// A list of candidate response content, ordered from best to worst.
   public let candidates: [CandidateResponse]
 
@@ -24,6 +36,9 @@ public struct GenerateContentResponse {
   /// reason for blocking the request.
   public let promptFeedback: PromptFeedback?
 
+  /// Token usage metadata for processing the generate content request.
+  public let usageMetadata: UsageMetadata?
+
   /// The response's content as text, if it exists.
   public var text: String? {
     guard let candidate = candidates.first else {
@@ -51,9 +66,11 @@ public struct GenerateContentResponse {
   }
 
   /// Initializer for SwiftUI previews or tests.
-  public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback?) {
+  public init(candidates: [CandidateResponse], promptFeedback: PromptFeedback? = nil,
+              usageMetadata: UsageMetadata? = nil) {
     self.candidates = candidates
     self.promptFeedback = promptFeedback
+    self.usageMetadata = usageMetadata
   }
 }
 
@@ -62,6 +79,7 @@ extension GenerateContentResponse: Decodable {
   enum CodingKeys: CodingKey {
     case candidates
     case promptFeedback
+    case usageMetadata
   }
 
   public init(from decoder: Decoder) throws {
@@ -86,6 +104,7 @@ extension GenerateContentResponse: Decodable {
       candidates = []
     }
     promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
+    usageMetadata = try container.decodeIfPresent(UsageMetadata.self, forKey: .usageMetadata)
   }
 }
 
@@ -301,3 +320,20 @@ extension PromptFeedback: Decodable {
     }
   }
 }
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerateContentResponse.UsageMetadata: Decodable {
+  enum CodingKeys: CodingKey {
+    case promptTokenCount
+    case candidatesTokenCount
+    case totalTokenCount
+  }
+
+  public init(from decoder: any Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    promptTokenCount = try container.decodeIfPresent(Int.self, forKey: .promptTokenCount) ?? 0
+    candidatesTokenCount = try container
+      .decodeIfPresent(Int.self, forKey: .candidatesTokenCount) ?? 0
+    totalTokenCount = try container.decodeIfPresent(Int.self, forKey: .totalTokenCount) ?? 0
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index 92585266891..e7ea54f5dd9 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -287,6 +287,21 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_usageMetadata() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json"
+      )
+
+    let response = try await model.generateContent(testPrompt)
+
+    let usageMetadata = try XCTUnwrap(response.usageMetadata)
+    XCTAssertEqual(usageMetadata.promptTokenCount, 6)
+    XCTAssertEqual(usageMetadata.candidatesTokenCount, 7)
+    XCTAssertEqual(usageMetadata.totalTokenCount, 13)
+  }
+
   func testGenerateContent_failure_invalidAPIKey() async throws {
     let expectedStatusCode = 400
     MockURLProtocol
@@ -814,6 +829,32 @@ final class GenerativeModelTests: XCTestCase {
     for try await _ in stream {}
   }
 
+  func testGenerateContentStream_usageMetadata() async throws {
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "streaming-success-basic-reply-short",
+        withExtension: "txt"
+      )
+    var responses = [GenerateContentResponse]()
+
+    let stream = model.generateContentStream(testPrompt)
+    for try await response in stream {
+      responses.append(response)
+    }
+
+    for (index, response) in responses.enumerated() {
+      if index == responses.endIndex - 1 {
+        let usageMetadata = try XCTUnwrap(response.usageMetadata)
+        XCTAssertEqual(usageMetadata.promptTokenCount, 6)
+        XCTAssertEqual(usageMetadata.candidatesTokenCount, 4)
+        XCTAssertEqual(usageMetadata.totalTokenCount, 10)
+      } else {
+        // Only the last streamed response contains usage metadata
+        XCTAssertNil(response.usageMetadata)
+      }
+    }
+  }
+
   func testGenerateContentStream_errorMidStream() async throws {
     MockURLProtocol.requestHandler = try httpRequestHandler(
       forResource: "streaming-failure-error-mid-stream",

From 14ffa02256bf0ed6597824e91fcdb7461bf6cc82 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 15 Apr 2024 12:59:20 -0400
Subject: [PATCH 099/124] Add public API tests for `GenerateContentResponse` in
 Vertex AI (#12791)

---
 .../Tests/Unit/VertexAIAPITests.swift          | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index e9312edfee0..75fb8946d22 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -167,6 +167,24 @@ final class VertexAIAPITests: XCTestCase {
     _ = genAI.startChat(history: [ModelContent(parts: "abc")])
   }
 
+  // Public API tests for GenerateContentResponse.
+  func generateContentResponseAPI() {
+    let response = GenerateContentResponse(candidates: [])
+
+    let _: [CandidateResponse] = response.candidates
+    let _: PromptFeedback? = response.promptFeedback
+
+    // Usage Metadata
+    guard let usageMetadata = response.usageMetadata else { fatalError() }
+    let _: Int = usageMetadata.promptTokenCount
+    let _: Int = usageMetadata.candidatesTokenCount
+    let _: Int = usageMetadata.totalTokenCount
+
+    // Computed Properties
+    let _: String? = response.text
+    let _: [FunctionCall] = response.functionCalls
+  }
+
   // Result builder alternative
 
   /*

From 1df22f898f40d26c3e27f8c86e72f2fbb2f25593 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 22 Apr 2024 12:50:01 -0400
Subject: [PATCH 100/124] [Vertex AI] Add Swift major version to language tag
 (#12829)

---
 FirebaseVertexAI/Sources/GenerativeAIService.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index b3687c3c013..60ca375c9f6 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -18,8 +18,8 @@ import Foundation
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 struct GenerativeAIService {
-  /// The language of the SDK in the format "gl-<language>/<version>" where version may be blank.
-  static let languageTag = "gl-swift/"
+  /// The language of the SDK in the format "gl-<language>/<version>".
+  static let languageTag = "gl-swift/5"
 
   /// The Firebase SDK version in the format "fire/<version>".
   static let firebaseVersionTag = "fire/\(FirebaseVersion())"

From 0a854325972d9dd033c231f57eb4665155b6656f Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 22 Apr 2024 14:54:42 -0400
Subject: [PATCH 101/124] [Vertex AI] Update `Citation` decoding to handle
 optional `startIndex` (#12832)

---
 .../Sources/GenerateContentResponse.swift     | 24 +++++++++++-
 .../streaming-success-citations.txt           | 10 ++---
 .../unary-success-citations.json              | 38 ++++++++++++-------
 .../Tests/Unit/GenerativeModelTests.swift     | 32 +++++++++++-----
 4 files changed, 72 insertions(+), 32 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 29c5b14f552..48f4ff22413 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -197,7 +197,7 @@ public struct CitationMetadata: Decodable {
 
 /// A struct describing a source attribution.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct Citation: Decodable {
+public struct Citation {
   /// The inclusive beginning of a sequence in a model response that derives from a cited source.
   public let startIndex: Int
 
@@ -207,7 +207,7 @@ public struct Citation: Decodable {
   /// A link to the cited source.
   public let uri: String
 
-  /// The license the cited source work is distributed under.
+  /// The license the cited source work is distributed under, if specified.
   public let license: String?
 }
 
@@ -337,3 +337,23 @@ extension GenerateContentResponse.UsageMetadata: Decodable {
     totalTokenCount = try container.decodeIfPresent(Int.self, forKey: .totalTokenCount) ?? 0
   }
 }
+
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension Citation: Decodable {
+  enum CodingKeys: CodingKey {
+    case startIndex
+    case endIndex
+    case uri
+    case license
+  }
+
+  public init(from decoder: any Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    startIndex = try container.decodeIfPresent(Int.self, forKey: .startIndex) ?? 0
+    endIndex = try container.decode(Int.self, forKey: .endIndex)
+    uri = try container.decode(String.self, forKey: .uri)
+    license = try container.decodeIfPresent(String.self, forKey: .license)
+  }
+}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
index a530736b308..fdc1f20d8b6 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-citations.txt
@@ -1,9 +1,7 @@
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Some information"}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Some information"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.043204036,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.082549304},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.046291895,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.071461484},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.100701615,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06164962},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.13150747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.040087357}]}]}
 
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": "  More information"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06632687,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.07477004,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048767097},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13695431,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059866417},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.046119746,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.036425155}]}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": "  Some information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.18982129,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.1337543},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.13637818,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.021906368},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.25404602,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.09073549},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.24202643,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06548521}],"citationMetadata": {"citations": [{"endIndex": 128,"uri": "https://www.example.com/citation-1"},{"startIndex": 130,"endIndex": 265,"uri": "https://www.example.com/citation-2"}]}}]}
 
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " Some information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07850098,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.039416388},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.08035747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04885778},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12273335,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059646938},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.053206205,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04099903}],"citationMetadata": {"citations": [{"startIndex": 31,"endIndex": 187,"uri": "https://www.example.com/citation-1"}]}}]}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07850098,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.039416388},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.08035747,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04885778},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12273335,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059646938},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.053206205,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04099903}],"citationMetadata": {"citations": [{"startIndex": 272,"endIndex": 431,"uri": "https://www.example.com/citation-3","license": "mit"}]}}]}
 
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information cited from an external source"}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.044183318},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.094176665,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0575992},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.13660839,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08035747},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.060197048,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.046464667}],"citationMetadata": {"citations": [{"startIndex": 73,"endIndex": 248,"uri": "https://www.example.com/citation-2"},{"startIndex": 133,"endIndex": 272,"uri": "https://www.example.com/citation-3", "license": "mit"}]}}]}
-
-data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.12147716,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0647717},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.11858909,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.14866412,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08479541},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.05470151,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045015533}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 163,"totalTokenCount": 172}}
+data: {"candidates": [{"content": {"role": "model","parts": [{"text": " More information "}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.16013464,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.11716747},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.10818896,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.021990221},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.2158462,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.07682221},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.19636348,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.08021325}]}],"usageMetadata": {"promptTokenCount": 9,"candidatesTokenCount": 53,"totalTokenCount": 62}}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
index a0dacc3e883..68fc9648773 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-citations.json
@@ -14,46 +14,56 @@
         {
           "category": "HARM_CATEGORY_HATE_SPEECH",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.16013464,
+          "probabilityScore": 0.16926852,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.074500255
+          "severityScore": 0.08181271
         },
         {
           "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.09687653,
+          "probabilityScore": 0.15636235,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.049313594
+          "severityScore": 0.02981654
         },
         {
           "category": "HARM_CATEGORY_HARASSMENT",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.16817278,
+          "probabilityScore": 0.33133608,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.09451043
+          "severityScore": 0.10875559
         },
         {
           "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
           "probability": "NEGLIGIBLE",
-          "probabilityScore": 0.05023736,
+          "probabilityScore": 0.17766814,
           "severity": "HARM_SEVERITY_NEGLIGIBLE",
-          "severityScore": 0.034553625
+          "severityScore": 0.18787657
         }
       ],
       "citationMetadata": {
         "citations": [
           {
-            "startIndex": 179,
-            "endIndex": 366,
-            "uri": "https://www.example.com/some-citation"
+            "endIndex": 128,
+            "uri": "https://www.example.com/some-citation-1"
+          },
+          {
+            "startIndex": 130,
+            "endIndex": 265,
+            "uri": "https://www.example.com/some-citation-2"
+          },
+          {
+            "startIndex": 272,
+            "endIndex": 431,
+            "uri": "https://www.example.com/some-citation-3",
+            "license": "mit"
           }
         ]
       }
     }
   ],
   "usageMetadata": {
-    "promptTokenCount": 11,
-    "candidatesTokenCount": 135,
-    "totalTokenCount": 146
+    "promptTokenCount": 15,
+    "candidatesTokenCount": 253,
+    "totalTokenCount": 268
   }
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index e7ea54f5dd9..c9442e05ba7 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -108,12 +108,22 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(candidate.content.parts.count, 1)
     XCTAssertEqual(response.text, "Some information cited from an external source")
     let citationMetadata = try XCTUnwrap(candidate.citationMetadata)
-    XCTAssertEqual(citationMetadata.citationSources.count, 1)
-    let citationSource = try XCTUnwrap(citationMetadata.citationSources.first)
-    XCTAssertEqual(citationSource.uri, "https://www.example.com/some-citation")
-    XCTAssertEqual(citationSource.startIndex, 179)
-    XCTAssertEqual(citationSource.endIndex, 366)
-    XCTAssertNil(citationSource.license)
+    XCTAssertEqual(citationMetadata.citationSources.count, 3)
+    let citationSource1 = try XCTUnwrap(citationMetadata.citationSources[0])
+    XCTAssertEqual(citationSource1.uri, "https://www.example.com/some-citation-1")
+    XCTAssertEqual(citationSource1.startIndex, 0)
+    XCTAssertEqual(citationSource1.endIndex, 128)
+    XCTAssertNil(citationSource1.license)
+    let citationSource2 = try XCTUnwrap(citationMetadata.citationSources[1])
+    XCTAssertEqual(citationSource2.uri, "https://www.example.com/some-citation-2")
+    XCTAssertEqual(citationSource2.startIndex, 130)
+    XCTAssertEqual(citationSource2.endIndex, 265)
+    XCTAssertNil(citationSource2.license)
+    let citationSource3 = try XCTUnwrap(citationMetadata.citationSources[2])
+    XCTAssertEqual(citationSource3.uri, "https://www.example.com/some-citation-3")
+    XCTAssertEqual(citationSource3.startIndex, 272)
+    XCTAssertEqual(citationSource3.endIndex, 431)
+    XCTAssertEqual(citationSource3.license, "mit")
   }
 
   func testGenerateContent_success_quoteReply() async throws {
@@ -778,13 +788,15 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(citations.count, 3)
     XCTAssertTrue(citations
       .contains(where: {
-        $0.startIndex == 31 && $0.endIndex == 187 && $0
-          .uri == "https://www.example.com/citation-1" && $0.license == nil
+        $0.startIndex == 0 && $0.endIndex == 128 && !$0.uri.isEmpty && $0.license == nil
       }))
     XCTAssertTrue(citations
       .contains(where: {
-        $0.startIndex == 133 && $0.endIndex == 272 && $0
-          .uri == "https://www.example.com/citation-3" && $0.license == "mit"
+        $0.startIndex == 130 && $0.endIndex == 265 && !$0.uri.isEmpty && $0.license == nil
+      }))
+    XCTAssertTrue(citations
+      .contains(where: {
+        $0.startIndex == 272 && $0.endIndex == 431 && !$0.uri.isEmpty && $0.license == "mit"
       }))
   }
 

From 4423b60eb5b68717d0dbf2aa4436e4fdcc39abd6 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 23 Apr 2024 09:45:11 -0400
Subject: [PATCH 102/124] [Vertex AI] Re-added `location` parameter to public
 API (#12835)

---
 FirebaseVertexAI/Sources/VertexAI.swift            | 14 ++++++++++----
 FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift |  2 ++
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 7a7865e86d2..44134db5b14 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -25,27 +25,33 @@ public class VertexAI: NSObject {
 
   /// The default `VertexAI` instance.
   ///
+  /// - Parameter location: The region identifier, defaulting to `us-central1`; see [Vertex AI
+  ///   regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions) for a
+  ///   list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
-  public static func vertexAI() -> VertexAI {
+  public static func vertexAI(location: String = "us-central1") -> VertexAI {
     guard let app = FirebaseApp.app() else {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    return vertexAI(app: app)
+    return vertexAI(app: app, location: location)
   }
 
   /// Creates an instance of `VertexAI` configured with a custom `FirebaseApp`.
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
+  ///   - location: The region identifier, defaulting to `us-central1`; see [Vertex AI
+  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
-  public static func vertexAI(app: FirebaseApp) -> VertexAI {
+  public static func vertexAI(app: FirebaseApp, location: String = "us-central1") -> VertexAI {
     guard let provider = ComponentType<VertexAIProvider>.instance(for: VertexAIProvider.self,
                                                                   in: app.container) else {
       fatalError("No \(VertexAIProvider.self) instance found for Firebase app: \(app.name)")
     }
 
-    return provider.vertexAI("us-central1")
+    return provider.vertexAI(location)
   }
 
   /// Initializes a generative model with the given parameters.
diff --git a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
index 75fb8946d22..50410379b75 100644
--- a/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
+++ b/FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift
@@ -36,9 +36,11 @@ final class VertexAIAPITests: XCTestCase {
 
     // Instantiate Vertex AI SDK - Default App
     let vertexAI = VertexAI.vertexAI()
+    let _ = VertexAI.vertexAI(location: "my-location")
 
     // Instantiate Vertex AI SDK - Custom App
     let _ = VertexAI.vertexAI(app: app!)
+    let _ = VertexAI.vertexAI(app: app!, location: "my-location")
 
     // Permutations without optional arguments.
 

From 42e1d5156ec39a0ce810f1a540f6ed3e79294b57 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 23 Apr 2024 09:52:33 -0400
Subject: [PATCH 103/124] [Vertex AI] Remove public `Codable` conformance
 (#12833)

---
 .../Sources/CountTokensRequest.swift          |  16 +-
 .../Sources/FunctionCalling.swift             |  55 ++--
 .../Sources/GenerateContentResponse.swift     | 267 +++++++++---------
 .../Sources/GenerationConfig.swift            |   7 +-
 FirebaseVertexAI/Sources/ModelContent.swift   | 120 ++++----
 FirebaseVertexAI/Sources/Safety.swift         | 108 ++++---
 6 files changed, 310 insertions(+), 263 deletions(-)

diff --git a/FirebaseVertexAI/Sources/CountTokensRequest.swift b/FirebaseVertexAI/Sources/CountTokensRequest.swift
index 5072412cf9b..8af409329ad 100644
--- a/FirebaseVertexAI/Sources/CountTokensRequest.swift
+++ b/FirebaseVertexAI/Sources/CountTokensRequest.swift
@@ -21,13 +21,6 @@ struct CountTokensRequest {
   let options: RequestOptions
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension CountTokensRequest: Encodable {
-  enum CodingKeys: CodingKey {
-    case contents
-  }
-}
-
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 extension CountTokensRequest: GenerativeAIRequest {
   typealias Response = CountTokensResponse
@@ -50,6 +43,15 @@ public struct CountTokensResponse {
   public let totalBillableCharacters: Int
 }
 
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CountTokensRequest: Encodable {
+  enum CodingKeys: CodingKey {
+    case contents
+  }
+}
+
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 extension CountTokensResponse: Decodable {
   enum CodingKeys: CodingKey {
diff --git a/FirebaseVertexAI/Sources/FunctionCalling.swift b/FirebaseVertexAI/Sources/FunctionCalling.swift
index 8860344c7cd..f9f74ee2e99 100644
--- a/FirebaseVertexAI/Sources/FunctionCalling.swift
+++ b/FirebaseVertexAI/Sources/FunctionCalling.swift
@@ -15,7 +15,7 @@
 import Foundation
 
 /// A predicted function call returned from the model.
-public struct FunctionCall: Equatable, Encodable {
+public struct FunctionCall: Equatable {
   /// The name of the function to call.
   public let name: String
 
@@ -27,7 +27,7 @@ public struct FunctionCall: Equatable, Encodable {
 ///
 /// These types can be objects, but also primitives and arrays. Represents a select subset of an
 /// [OpenAPI 3.0 schema object](https://spec.openapis.org/oas/v3.0.3#schema).
-public class Schema: Encodable {
+public class Schema {
   /// The data type.
   let type: DataType
 
@@ -52,17 +52,6 @@ public class Schema: Encodable {
   /// Required properties of type ``DataType/object``.
   let requiredProperties: [String]?
 
-  enum CodingKeys: String, CodingKey {
-    case type
-    case format
-    case description
-    case nullable
-    case enumValues = "enum"
-    case items
-    case properties
-    case requiredProperties = "required"
-  }
-
   /// Constructs a new `Schema`.
   ///
   /// - Parameters:
@@ -98,7 +87,7 @@ public class Schema: Encodable {
 /// A data type.
 ///
 /// Contains the set of OpenAPI [data types](https://spec.openapis.org/oas/v3.0.3#data-types).
-public enum DataType: String, Encodable {
+public enum DataType: String {
   /// A `String` type.
   case string = "STRING"
 
@@ -157,7 +146,7 @@ public struct FunctionDeclaration {
 ///
 /// A `Tool` is a piece of code that enables the system to interact with external systems to
 /// perform an action, or set of actions, outside of knowledge and scope of the model.
-public struct Tool: Encodable {
+public struct Tool {
   /// A list of `FunctionDeclarations` available to the model.
   let functionDeclarations: [FunctionDeclaration]?
 
@@ -178,10 +167,10 @@ public struct Tool: Encodable {
 }
 
 /// Configuration for specifying function calling behavior.
-public struct FunctionCallingConfig: Encodable {
+public struct FunctionCallingConfig {
   /// Defines the execution behavior for function calling by defining the
   /// execution mode.
-  public enum Mode: String, Encodable {
+  public enum Mode: String {
     /// The default behavior for function calling. The model calls functions to answer queries at
     /// its discretion.
     case auto = "AUTO"
@@ -213,8 +202,7 @@ public struct FunctionCallingConfig: Encodable {
 }
 
 /// Tool configuration for any `Tool` specified in the request.
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct ToolConfig: Encodable {
+public struct ToolConfig {
   let functionCallingConfig: FunctionCallingConfig?
 
   public init(functionCallingConfig: FunctionCallingConfig? = nil) {
@@ -227,7 +215,7 @@ public struct ToolConfig: Encodable {
 /// Contains a string representing the `FunctionDeclaration.name` and a structured JSON object
 /// containing any output from the function is used as context to the model. This should contain the
 /// result of a ``FunctionCall`` made based on model prediction.
-public struct FunctionResponse: Equatable, Encodable {
+public struct FunctionResponse: Equatable {
   /// The name of the function that was called.
   let name: String
 
@@ -264,6 +252,8 @@ extension FunctionCall: Decodable {
   }
 }
 
+extension FunctionCall: Encodable {}
+
 extension FunctionDeclaration: Encodable {
   enum CodingKeys: String, CodingKey {
     case name
@@ -278,3 +268,28 @@ extension FunctionDeclaration: Encodable {
     try container.encode(parameters, forKey: .parameters)
   }
 }
+
+extension Schema: Encodable {
+  enum CodingKeys: String, CodingKey {
+    case type
+    case format
+    case description
+    case nullable
+    case enumValues = "enum"
+    case items
+    case properties
+    case requiredProperties = "required"
+  }
+}
+
+extension DataType: Encodable {}
+
+extension Tool: Encodable {}
+
+extension FunctionCallingConfig: Encodable {}
+
+extension FunctionCallingConfig.Mode: Encodable {}
+
+extension ToolConfig: Encodable {}
+
+extension FunctionResponse: Encodable {}
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
index 48f4ff22413..e89029c2d02 100644
--- a/FirebaseVertexAI/Sources/GenerateContentResponse.swift
+++ b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -74,40 +74,6 @@ public struct GenerateContentResponse {
   }
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension GenerateContentResponse: Decodable {
-  enum CodingKeys: CodingKey {
-    case candidates
-    case promptFeedback
-    case usageMetadata
-  }
-
-  public init(from decoder: Decoder) throws {
-    let container = try decoder.container(keyedBy: CodingKeys.self)
-
-    guard container.contains(CodingKeys.candidates) || container
-      .contains(CodingKeys.promptFeedback) else {
-      let context = DecodingError.Context(
-        codingPath: [],
-        debugDescription: "Failed to decode GenerateContentResponse;" +
-          " missing keys 'candidates' and 'promptFeedback'."
-      )
-      throw DecodingError.dataCorrupted(context)
-    }
-
-    if let candidates = try container.decodeIfPresent(
-      [CandidateResponse].self,
-      forKey: .candidates
-    ) {
-      self.candidates = candidates
-    } else {
-      candidates = []
-    }
-    promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
-    usageMetadata = try container.decodeIfPresent(UsageMetadata.self, forKey: .usageMetadata)
-  }
-}
-
 /// A struct representing a possible reply to a content generation prompt. Each content generation
 /// prompt may produce multiple candidate responses.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
@@ -135,62 +101,9 @@ public struct CandidateResponse {
   }
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension CandidateResponse: Decodable {
-  enum CodingKeys: CodingKey {
-    case content
-    case safetyRatings
-    case finishReason
-    case finishMessage
-    case citationMetadata
-  }
-
-  /// Initializes a response from a decoder. Used for decoding server responses; not for public
-  /// use.
-  public init(from decoder: Decoder) throws {
-    let container = try decoder.container(keyedBy: CodingKeys.self)
-
-    do {
-      if let content = try container.decodeIfPresent(ModelContent.self, forKey: .content) {
-        self.content = content
-      } else {
-        content = ModelContent(parts: [])
-      }
-    } catch {
-      // Check if `content` can be decoded as an empty dictionary to detect the `"content": {}` bug.
-      if let content = try? container.decode([String: String].self, forKey: .content),
-         content.isEmpty {
-        throw InvalidCandidateError.emptyContent(underlyingError: error)
-      } else {
-        throw InvalidCandidateError.malformedContent(underlyingError: error)
-      }
-    }
-
-    if let safetyRatings = try container.decodeIfPresent(
-      [SafetyRating].self,
-      forKey: .safetyRatings
-    ) {
-      self.safetyRatings = safetyRatings
-    } else {
-      safetyRatings = []
-    }
-
-    finishReason = try container.decodeIfPresent(FinishReason.self, forKey: .finishReason)
-
-    citationMetadata = try container.decodeIfPresent(
-      CitationMetadata.self,
-      forKey: .citationMetadata
-    )
-  }
-}
-
 /// A collection of source attributions for a piece of content.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct CitationMetadata: Decodable {
-  enum CodingKeys: String, CodingKey {
-    case citationSources = "citations"
-  }
-
+public struct CitationMetadata {
   /// A list of individual cited sources and the parts of the content to which they apply.
   public let citationSources: [Citation]
 }
@@ -236,27 +149,11 @@ public enum FinishReason: String {
   case other = "OTHER"
 }
 
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension FinishReason: Decodable {
-  /// Do not explicitly use. Initializer required for Decodable conformance.
-  public init(from decoder: Decoder) throws {
-    let value = try decoder.singleValueContainer().decode(String.self)
-    guard let decodedFinishReason = FinishReason(rawValue: value) else {
-      Logging.default
-        .error("[GoogleGenerativeAI] Unrecognized FinishReason with value \"\(value)\".")
-      self = .unknown
-      return
-    }
-
-    self = decodedFinishReason
-  }
-}
-
 /// A metadata struct containing any feedback the model had on the prompt it was provided.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public struct PromptFeedback {
   /// A type describing possible reasons to block a prompt.
-  public enum BlockReason: String, Decodable {
+  public enum BlockReason: String {
     /// The block reason is unknown.
     case unknown = "UNKNOWN"
 
@@ -268,19 +165,6 @@ public struct PromptFeedback {
 
     /// All other block reasons.
     case other = "OTHER"
-
-    /// Do not explicitly use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedBlockReason = BlockReason(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized BlockReason with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedBlockReason
-    }
   }
 
   /// The reason a prompt was blocked, if it was blocked.
@@ -296,28 +180,39 @@ public struct PromptFeedback {
   }
 }
 
+// MARK: - Codable Conformances
+
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-extension PromptFeedback: Decodable {
+extension GenerateContentResponse: Decodable {
   enum CodingKeys: CodingKey {
-    case blockReason
-    case safetyRatings
+    case candidates
+    case promptFeedback
+    case usageMetadata
   }
 
-  /// Do not explicitly use. Initializer required for Decodable conformance.
   public init(from decoder: Decoder) throws {
     let container = try decoder.container(keyedBy: CodingKeys.self)
-    blockReason = try container.decodeIfPresent(
-      PromptFeedback.BlockReason.self,
-      forKey: .blockReason
-    )
-    if let safetyRatings = try container.decodeIfPresent(
-      [SafetyRating].self,
-      forKey: .safetyRatings
+
+    guard container.contains(CodingKeys.candidates) || container
+      .contains(CodingKeys.promptFeedback) else {
+      let context = DecodingError.Context(
+        codingPath: [],
+        debugDescription: "Failed to decode GenerateContentResponse;" +
+          " missing keys 'candidates' and 'promptFeedback'."
+      )
+      throw DecodingError.dataCorrupted(context)
+    }
+
+    if let candidates = try container.decodeIfPresent(
+      [CandidateResponse].self,
+      forKey: .candidates
     ) {
-      self.safetyRatings = safetyRatings
+      self.candidates = candidates
     } else {
-      safetyRatings = []
+      candidates = []
     }
+    promptFeedback = try container.decodeIfPresent(PromptFeedback.self, forKey: .promptFeedback)
+    usageMetadata = try container.decodeIfPresent(UsageMetadata.self, forKey: .usageMetadata)
   }
 }
 
@@ -338,7 +233,61 @@ extension GenerateContentResponse.UsageMetadata: Decodable {
   }
 }
 
-// MARK: - Codable Conformances
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CandidateResponse: Decodable {
+  enum CodingKeys: CodingKey {
+    case content
+    case safetyRatings
+    case finishReason
+    case finishMessage
+    case citationMetadata
+  }
+
+  /// Initializes a response from a decoder. Used for decoding server responses; not for public
+  /// use.
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+
+    do {
+      if let content = try container.decodeIfPresent(ModelContent.self, forKey: .content) {
+        self.content = content
+      } else {
+        content = ModelContent(parts: [])
+      }
+    } catch {
+      // Check if `content` can be decoded as an empty dictionary to detect the `"content": {}` bug.
+      if let content = try? container.decode([String: String].self, forKey: .content),
+         content.isEmpty {
+        throw InvalidCandidateError.emptyContent(underlyingError: error)
+      } else {
+        throw InvalidCandidateError.malformedContent(underlyingError: error)
+      }
+    }
+
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+
+    finishReason = try container.decodeIfPresent(FinishReason.self, forKey: .finishReason)
+
+    citationMetadata = try container.decodeIfPresent(
+      CitationMetadata.self,
+      forKey: .citationMetadata
+    )
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension CitationMetadata: Decodable {
+  enum CodingKeys: String, CodingKey {
+    case citationSources = "citations"
+  }
+}
 
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 extension Citation: Decodable {
@@ -357,3 +306,57 @@ extension Citation: Decodable {
     license = try container.decodeIfPresent(String.self, forKey: .license)
   }
 }
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension FinishReason: Decodable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedFinishReason = FinishReason(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized FinishReason with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedFinishReason
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension PromptFeedback.BlockReason: Decodable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedBlockReason = PromptFeedback.BlockReason(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized BlockReason with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedBlockReason
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension PromptFeedback: Decodable {
+  enum CodingKeys: CodingKey {
+    case blockReason
+    case safetyRatings
+  }
+
+  public init(from decoder: Decoder) throws {
+    let container = try decoder.container(keyedBy: CodingKeys.self)
+    blockReason = try container.decodeIfPresent(
+      PromptFeedback.BlockReason.self,
+      forKey: .blockReason
+    )
+    if let safetyRatings = try container.decodeIfPresent(
+      [SafetyRating].self,
+      forKey: .safetyRatings
+    ) {
+      self.safetyRatings = safetyRatings
+    } else {
+      safetyRatings = []
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/GenerationConfig.swift b/FirebaseVertexAI/Sources/GenerationConfig.swift
index 2d1016c965c..18bf2bd1a52 100644
--- a/FirebaseVertexAI/Sources/GenerationConfig.swift
+++ b/FirebaseVertexAI/Sources/GenerationConfig.swift
@@ -17,7 +17,7 @@ import Foundation
 /// A struct defining model parameters to be used when sending generative AI
 /// requests to the backend model.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct GenerationConfig: Encodable {
+public struct GenerationConfig {
   /// A parameter controlling the degree of randomness in token selection. A
   /// temperature of zero is deterministic, always choosing the
   /// highest-probability response. Typical values are between 0 and 1
@@ -84,3 +84,8 @@ public struct GenerationConfig: Encodable {
     self.stopSequences = stopSequences
   }
 }
+
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension GenerationConfig: Encodable {}
diff --git a/FirebaseVertexAI/Sources/ModelContent.swift b/FirebaseVertexAI/Sources/ModelContent.swift
index 136fc1debe3..bae04b13c18 100644
--- a/FirebaseVertexAI/Sources/ModelContent.swift
+++ b/FirebaseVertexAI/Sources/ModelContent.swift
@@ -18,22 +18,10 @@ import Foundation
 /// request or response contains an `Array` of ``ModelContent``s, and each ``ModelContent`` value
 /// may comprise multiple heterogeneous ``ModelContent/Part``s.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct ModelContent: Codable, Equatable {
+public struct ModelContent: Equatable {
   /// A discrete piece of data in a media format intepretable by an AI model. Within a single value
   /// of ``Part``, different data types may not mix.
-  public enum Part: Codable, Equatable {
-    enum CodingKeys: String, CodingKey {
-      case text
-      case inlineData
-      case functionCall
-      case functionResponse
-    }
-
-    enum InlineDataKeys: String, CodingKey {
-      case mimeType = "mime_type"
-      case bytes = "data"
-    }
-
+  public enum Part: Equatable {
     /// Text value.
     case text(String)
 
@@ -58,49 +46,6 @@ public struct ModelContent: Codable, Equatable {
       return .data(mimetype: "image/png", data)
     }
 
-    // MARK: Codable Conformance
-
-    public func encode(to encoder: Encoder) throws {
-      var container = encoder.container(keyedBy: ModelContent.Part.CodingKeys.self)
-      switch self {
-      case let .text(a0):
-        try container.encode(a0, forKey: .text)
-      case let .data(mimetype, bytes):
-        var inlineDataContainer = container.nestedContainer(
-          keyedBy: InlineDataKeys.self,
-          forKey: .inlineData
-        )
-        try inlineDataContainer.encode(mimetype, forKey: .mimeType)
-        try inlineDataContainer.encode(bytes, forKey: .bytes)
-      case let .functionCall(functionCall):
-        try container.encode(functionCall, forKey: .functionCall)
-      case let .functionResponse(functionResponse):
-        try container.encode(functionResponse, forKey: .functionResponse)
-      }
-    }
-
-    public init(from decoder: Decoder) throws {
-      let values = try decoder.container(keyedBy: CodingKeys.self)
-      if values.contains(.text) {
-        self = try .text(values.decode(String.self, forKey: .text))
-      } else if values.contains(.inlineData) {
-        let dataContainer = try values.nestedContainer(
-          keyedBy: InlineDataKeys.self,
-          forKey: .inlineData
-        )
-        let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
-        let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
-        self = .data(mimetype: mimetype, bytes)
-      } else if values.contains(.functionCall) {
-        self = try .functionCall(values.decode(FunctionCall.self, forKey: .functionCall))
-      } else {
-        throw DecodingError.dataCorrupted(.init(
-          codingPath: [CodingKeys.text, CodingKeys.inlineData],
-          debugDescription: "No text, inline data or function call was found."
-        ))
-      }
-    }
-
     /// Returns the text contents of this ``Part``, if it contains text.
     public var text: String? {
       switch self {
@@ -153,3 +98,64 @@ public struct ModelContent: Codable, Equatable {
     self.init(role: role, parts: content)
   }
 }
+
+// MARK: Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension ModelContent: Codable {}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension ModelContent.Part: Codable {
+  enum CodingKeys: String, CodingKey {
+    case text
+    case inlineData
+    case functionCall
+    case functionResponse
+  }
+
+  enum InlineDataKeys: String, CodingKey {
+    case mimeType = "mime_type"
+    case bytes = "data"
+  }
+
+  public func encode(to encoder: Encoder) throws {
+    var container = encoder.container(keyedBy: CodingKeys.self)
+    switch self {
+    case let .text(a0):
+      try container.encode(a0, forKey: .text)
+    case let .data(mimetype, bytes):
+      var inlineDataContainer = container.nestedContainer(
+        keyedBy: InlineDataKeys.self,
+        forKey: .inlineData
+      )
+      try inlineDataContainer.encode(mimetype, forKey: .mimeType)
+      try inlineDataContainer.encode(bytes, forKey: .bytes)
+    case let .functionCall(functionCall):
+      try container.encode(functionCall, forKey: .functionCall)
+    case let .functionResponse(functionResponse):
+      try container.encode(functionResponse, forKey: .functionResponse)
+    }
+  }
+
+  public init(from decoder: Decoder) throws {
+    let values = try decoder.container(keyedBy: CodingKeys.self)
+    if values.contains(.text) {
+      self = try .text(values.decode(String.self, forKey: .text))
+    } else if values.contains(.inlineData) {
+      let dataContainer = try values.nestedContainer(
+        keyedBy: InlineDataKeys.self,
+        forKey: .inlineData
+      )
+      let mimetype = try dataContainer.decode(String.self, forKey: .mimeType)
+      let bytes = try dataContainer.decode(Data.self, forKey: .bytes)
+      self = .data(mimetype: mimetype, bytes)
+    } else if values.contains(.functionCall) {
+      self = try .functionCall(values.decode(FunctionCall.self, forKey: .functionCall))
+    } else {
+      throw DecodingError.dataCorrupted(.init(
+        codingPath: [CodingKeys.text, CodingKeys.inlineData],
+        debugDescription: "No text, inline data or function call was found."
+      ))
+    }
+  }
+}
diff --git a/FirebaseVertexAI/Sources/Safety.swift b/FirebaseVertexAI/Sources/Safety.swift
index d59511f2476..d2adc4a81e1 100644
--- a/FirebaseVertexAI/Sources/Safety.swift
+++ b/FirebaseVertexAI/Sources/Safety.swift
@@ -18,7 +18,7 @@ import Foundation
 /// of this type may be assigned to a category for every model-generated response, not just
 /// responses that exceed a certain threshold.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct SafetyRating: Decodable, Equatable, Hashable {
+public struct SafetyRating: Equatable, Hashable {
   /// The category describing the potential harm a piece of content may pose. See
   /// ``SafetySetting/HarmCategory`` for a list of possible values.
   public let category: SafetySetting.HarmCategory
@@ -38,7 +38,7 @@ public struct SafetyRating: Decodable, Equatable, Hashable {
 
   /// The probability that a given model output falls under a harmful content category. This does
   /// not indicate the severity of harm for a piece of content.
-  public enum HarmProbability: String, Codable {
+  public enum HarmProbability: String {
     /// Unknown. A new server value that isn't recognized by the SDK.
     case unknown = "UNKNOWN"
 
@@ -57,26 +57,12 @@ public struct SafetyRating: Decodable, Equatable, Hashable {
 
     /// The probability is high. The content described is very likely harmful.
     case high = "HIGH"
-
-    /// Initializes a new `SafetyRating` from a decoder.
-    /// Not for external use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedProbability = HarmProbability(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized HarmProbability with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedProbability
-    }
   }
 }
 
 /// Safety feedback for an entire request.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct SafetyFeedback: Decodable {
+public struct SafetyFeedback {
   /// Safety rating evaluated from content.
   public let rating: SafetyRating
 
@@ -93,10 +79,10 @@ public struct SafetyFeedback: Decodable {
 /// A type used to specify a threshold for harmful content, beyond which the model will return a
 /// fallback response instead of generated content.
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public struct SafetySetting: Codable {
+public struct SafetySetting {
   /// A type describing safety attributes, which include harmful categories and topics that can
   /// be considered sensitive.
-  public enum HarmCategory: String, Codable {
+  public enum HarmCategory: String {
     /// Unknown. A new server value that isn't recognized by the SDK.
     case unknown = "HARM_CATEGORY_UNKNOWN"
 
@@ -114,23 +100,10 @@ public struct SafetySetting: Codable {
 
     /// Promotes or enables access to harmful goods, services, or activities.
     case dangerousContent = "HARM_CATEGORY_DANGEROUS_CONTENT"
-
-    /// Do not explicitly use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedCategory = HarmCategory(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized HarmCategory with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedCategory
-    }
   }
 
   /// Block at and beyond a specified ``SafetyRating/HarmProbability``.
-  public enum BlockThreshold: String, Codable {
+  public enum BlockThreshold: String {
     /// Unknown. A new server value that isn't recognized by the SDK.
     case unknown = "UNKNOWN"
 
@@ -148,19 +121,6 @@ public struct SafetySetting: Codable {
 
     /// All content will be allowed.
     case blockNone = "BLOCK_NONE"
-
-    /// Do not explicitly use. Initializer required for Decodable conformance.
-    public init(from decoder: Decoder) throws {
-      let value = try decoder.singleValueContainer().decode(String.self)
-      guard let decodedThreshold = BlockThreshold(rawValue: value) else {
-        Logging.default
-          .error("[GoogleGenerativeAI] Unrecognized BlockThreshold with value \"\(value)\".")
-        self = .unknown
-        return
-      }
-
-      self = decodedThreshold
-    }
   }
 
   enum CodingKeys: String, CodingKey {
@@ -180,3 +140,59 @@ public struct SafetySetting: Codable {
     self.threshold = threshold
   }
 }
+
+// MARK: - Codable Conformances
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetyRating.HarmProbability: Codable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedProbability = SafetyRating.HarmProbability(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized HarmProbability with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedProbability
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetyRating: Decodable {}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetyFeedback: Decodable {}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetySetting.HarmCategory: Codable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedCategory = SafetySetting.HarmCategory(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized HarmCategory with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedCategory
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetySetting.BlockThreshold: Codable {
+  public init(from decoder: Decoder) throws {
+    let value = try decoder.singleValueContainer().decode(String.self)
+    guard let decodedThreshold = SafetySetting.BlockThreshold(rawValue: value) else {
+      Logging.default
+        .error("[GoogleGenerativeAI] Unrecognized BlockThreshold with value \"\(value)\".")
+      self = .unknown
+      return
+    }
+
+    self = decodedThreshold
+  }
+}
+
+@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
+extension SafetySetting: Codable {}

From e4625bd0800ba4f5c9505846aca5e7b10a33b62a Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Thu, 25 Apr 2024 11:41:49 -0400
Subject: [PATCH 104/124] [Vertex AI] Add `location` checks for slashes and
 whitespace (#12849)

---
 FirebaseVertexAI/Sources/VertexAI.swift | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index 44134db5b14..a8f4abbf04d 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -26,8 +26,9 @@ public class VertexAI: NSObject {
   /// The default `VertexAI` instance.
   ///
   /// - Parameter location: The region identifier, defaulting to `us-central1`; see [Vertex AI
-  ///   regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions) for a
-  ///   list of supported regions.
+  /// regions
+  ///   ](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions)
+  ///   for a list of supported regions.
   /// - Returns: An instance of `VertexAI`, configured with the default `FirebaseApp`.
   public static func vertexAI(location: String = "us-central1") -> VertexAI {
     guard let app = FirebaseApp.app() else {
@@ -41,8 +42,8 @@ public class VertexAI: NSObject {
   ///
   ///  - Parameters:
   ///   - app: The custom `FirebaseApp` used for initialization.
-  ///   - location: The region identifier, defaulting to `us-central1`; see [Vertex AI
-  ///     regions](https://cloud.google.com/vertex-ai/docs/general/locations#vertex-ai-regions)
+  ///   - location: The region identifier, defaulting to `us-central1`; see [Vertex AI regions
+  ///     ](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions)
   ///     for a list of supported regions.
   /// - Returns: A `VertexAI` instance, configured with the custom `FirebaseApp`.
   public static func vertexAI(app: FirebaseApp, location: String = "us-central1") -> VertexAI {
@@ -117,11 +118,12 @@ public class VertexAI: NSObject {
     guard let projectID = app.options.projectID else {
       fatalError("The Firebase app named \"\(app.name)\" has no project ID in its configuration.")
     }
-    guard !location.isEmpty else {
+    guard !location.isEmpty && location
+      .allSatisfy({ !$0.isWhitespace && !$0.isNewline && $0 != "/" }) else {
       fatalError("""
-      No location specified; see
-      https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for a
-      list of available regions.
+      Invalid location "\(location)" specified; see \
+      https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions for \
+      a list of available regions.
       """)
     }
 

From 948df07adc4f6c06571bab60eab73f4ae4072b95 Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Mon, 29 Apr 2024 16:17:48 -0400
Subject: [PATCH 105/124] [Vertex AI] Integrate with AppCheckInterop (#12856)

---
 .../Sources/GenerativeAIService.swift         | 10 +-
 .../Sources/GenerativeModel.swift             |  3 +
 FirebaseVertexAI/Sources/VertexAI.swift       |  7 +-
 .../Sources/VertexAIComponent.swift           |  3 +
 FirebaseVertexAI/Tests/Unit/ChatTests.swift   |  1 +
 .../Tests/Unit/Fakes/AuthInteropFake.swift    | 49 ++++++++++
 .../Tests/Unit/GenerativeModelTests.swift     | 95 ++++++++++++++++++-
 Package.swift                                 |  1 +
 8 files changed, 163 insertions(+), 6 deletions(-)
 create mode 100644 FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift

diff --git a/FirebaseVertexAI/Sources/GenerativeAIService.swift b/FirebaseVertexAI/Sources/GenerativeAIService.swift
index 60ca375c9f6..dbf26c2c948 100644
--- a/FirebaseVertexAI/Sources/GenerativeAIService.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAIService.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import Foundation
 
@@ -29,11 +30,14 @@ struct GenerativeAIService {
 
   private let appCheck: AppCheckInterop?
 
+  private let auth: AuthInterop?
+
   private let urlSession: URLSession
 
-  init(apiKey: String, appCheck: AppCheckInterop?, urlSession: URLSession) {
+  init(apiKey: String, appCheck: AppCheckInterop?, auth: AuthInterop?, urlSession: URLSession) {
     self.apiKey = apiKey
     self.appCheck = appCheck
+    self.auth = auth
     self.urlSession = urlSession
   }
 
@@ -176,6 +180,10 @@ struct GenerativeAIService {
       }
     }
 
+    if let auth, let authToken = try await auth.getToken(forcingRefresh: false) {
+      urlRequest.setValue("Firebase \(authToken)", forHTTPHeaderField: "Authorization")
+    }
+
     let encoder = JSONEncoder()
     encoder.keyEncodingStrategy = .convertToSnakeCase
     urlRequest.httpBody = try encoder.encode(request)
diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift
index cdd6d175c0a..0656edd6a7e 100644
--- a/FirebaseVertexAI/Sources/GenerativeModel.swift
+++ b/FirebaseVertexAI/Sources/GenerativeModel.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import Foundation
 
 /// A type that represents a remote multimodal model (like Gemini), with the ability to generate
@@ -69,11 +70,13 @@ public final class GenerativeModel {
        systemInstruction: ModelContent? = nil,
        requestOptions: RequestOptions,
        appCheck: AppCheckInterop?,
+       auth: AuthInterop?,
        urlSession: URLSession = .shared) {
     modelResourceName = GenerativeModel.modelResourceName(name: name)
     generativeAIService = GenerativeAIService(
       apiKey: apiKey,
       appCheck: appCheck,
+      auth: auth,
       urlSession: urlSession
     )
     self.generationConfig = generationConfig
diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift
index a8f4abbf04d..53655eada79 100644
--- a/FirebaseVertexAI/Sources/VertexAI.swift
+++ b/FirebaseVertexAI/Sources/VertexAI.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import Foundation
 
@@ -92,7 +93,8 @@ public class VertexAI: NSObject {
       toolConfig: toolConfig,
       systemInstruction: systemInstruction,
       requestOptions: requestOptions,
-      appCheck: appCheck
+      appCheck: appCheck,
+      auth: auth
     )
   }
 
@@ -103,12 +105,15 @@ public class VertexAI: NSObject {
 
   private let appCheck: AppCheckInterop?
 
+  private let auth: AuthInterop?
+
   let location: String
 
   init(app: FirebaseApp, location: String) {
     self.app = app
     self.location = location
     appCheck = ComponentType<AppCheckInterop>.instance(for: AppCheckInterop.self, in: app.container)
+    auth = ComponentType<AuthInterop>.instance(for: AuthInterop.self, in: app.container)
   }
 
   private func modelResourceName(modelName: String, location: String) -> String {
diff --git a/FirebaseVertexAI/Sources/VertexAIComponent.swift b/FirebaseVertexAI/Sources/VertexAIComponent.swift
index a6d487e1208..c43cab5f1f6 100644
--- a/FirebaseVertexAI/Sources/VertexAIComponent.swift
+++ b/FirebaseVertexAI/Sources/VertexAIComponent.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import Foundation
 
@@ -51,10 +52,12 @@ class VertexAIComponent: NSObject, Library, VertexAIProvider {
 
   static func componentsToRegister() -> [Component] {
     let appCheckInterop = Dependency(with: AppCheckInterop.self, isRequired: false)
+    let authInterop = Dependency(with: AuthInterop.self, isRequired: false)
     return [Component(VertexAIProvider.self,
                       instantiationTiming: .lazy,
                       dependencies: [
                         appCheckInterop,
+                        authInterop,
                       ]) { container, isCacheable in
         guard let app = container.app else { return nil }
         isCacheable.pointee = true
diff --git a/FirebaseVertexAI/Tests/Unit/ChatTests.swift b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
index 20897a8787b..ea2633bdf40 100644
--- a/FirebaseVertexAI/Tests/Unit/ChatTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/ChatTests.swift
@@ -53,6 +53,7 @@ final class ChatTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
     let chat = Chat(model: model, history: [])
diff --git a/FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift b/FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift
new file mode 100644
index 00000000000..6d0d33e1b41
--- /dev/null
+++ b/FirebaseVertexAI/Tests/Unit/Fakes/AuthInteropFake.swift
@@ -0,0 +1,49 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import FirebaseAuthInterop
+import Foundation
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, *)
+class AuthInteropFake: NSObject, AuthInterop {
+  let token: String?
+  let error: Error?
+
+  func getToken(forcingRefresh forceRefresh: Bool) async throws -> String? {
+    if let error {
+      throw error
+    }
+
+    return token
+  }
+
+  func getUserID() -> String? {
+    fatalError("\(#function) not implemented.")
+  }
+
+  private init(token: String?, error: Error?) {
+    self.token = token
+    self.error = error
+  }
+
+  convenience init(error: Error) {
+    self.init(token: nil, error: error)
+  }
+
+  convenience init(token: String?) {
+    self.init(token: token, error: nil)
+  }
+}
+
+struct AuthErrorFake: Error {}
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
index c9442e05ba7..4431c9b7f1b 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
+++ b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import FirebaseAppCheckInterop
+import FirebaseAuthInterop
 import FirebaseCore
 import XCTest
 
@@ -41,6 +42,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
   }
@@ -182,6 +184,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -266,6 +269,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -285,6 +289,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -297,6 +302,74 @@ final class GenerativeModelTests: XCTestCase {
     _ = try await model.generateContent(testPrompt)
   }
 
+  func testGenerateContent_auth_validAuthToken() async throws {
+    let authToken = "test-valid-token"
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      tools: nil,
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      auth: AuthInteropFake(token: authToken),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        authToken: authToken
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_auth_nilAuthToken() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      tools: nil,
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      auth: AuthInteropFake(token: nil),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        authToken: nil
+      )
+
+    _ = try await model.generateContent(testPrompt)
+  }
+
+  func testGenerateContent_auth_authTokenRefreshError() async throws {
+    model = GenerativeModel(
+      name: "my-model",
+      apiKey: "API_KEY",
+      tools: nil,
+      requestOptions: RequestOptions(),
+      appCheck: nil,
+      auth: AuthInteropFake(error: AuthErrorFake()),
+      urlSession: urlSession
+    )
+    MockURLProtocol
+      .requestHandler = try httpRequestHandler(
+        forResource: "unary-success-basic-reply-short",
+        withExtension: "json",
+        authToken: nil
+      )
+
+    do {
+      _ = try await model.generateContent(testPrompt)
+      XCTFail("Should throw internalError(AuthErrorFake); no error.")
+    } catch GenerateContentError.internalError(_ as AuthErrorFake) {
+      //
+    } catch {
+      XCTFail("Should throw internalError(AuthErrorFake); error thrown: \(error)")
+    }
+  }
+
   func testGenerateContent_usageMetadata() async throws {
     MockURLProtocol
       .requestHandler = try httpRequestHandler(
@@ -598,6 +671,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -808,6 +882,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(token: appCheckToken),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -828,6 +903,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: RequestOptions(),
       appCheck: AppCheckInteropFake(error: AppCheckErrorFake()),
+      auth: nil,
       urlSession: urlSession
     )
     MockURLProtocol
@@ -972,6 +1048,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -1048,6 +1125,7 @@ final class GenerativeModelTests: XCTestCase {
       tools: nil,
       requestOptions: requestOptions,
       appCheck: nil,
+      auth: nil,
       urlSession: urlSession
     )
 
@@ -1067,7 +1145,8 @@ final class GenerativeModelTests: XCTestCase {
       apiKey: "API_KEY",
       tools: nil,
       requestOptions: RequestOptions(),
-      appCheck: nil
+      appCheck: nil,
+      auth: nil
     )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
@@ -1081,7 +1160,8 @@ final class GenerativeModelTests: XCTestCase {
       apiKey: "API_KEY",
       tools: nil,
       requestOptions: RequestOptions(),
-      appCheck: nil
+      appCheck: nil,
+      auth: nil
     )
 
     XCTAssertEqual(model.modelResourceName, modelResourceName)
@@ -1095,7 +1175,8 @@ final class GenerativeModelTests: XCTestCase {
       apiKey: "API_KEY",
       tools: nil,
       requestOptions: RequestOptions(),
-      appCheck: nil
+      appCheck: nil,
+      auth: nil
     )
 
     XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
@@ -1123,7 +1204,8 @@ final class GenerativeModelTests: XCTestCase {
                                   withExtension ext: String,
                                   statusCode: Int = 200,
                                   timeout: TimeInterval = URLRequest.defaultTimeoutInterval(),
-                                  appCheckToken: String? = nil) throws -> ((URLRequest) throws -> (
+                                  appCheckToken: String? = nil,
+                                  authToken: String? = nil) throws -> ((URLRequest) throws -> (
     URLResponse,
     AsyncLineSequence<URL.AsyncBytes>?
   )) {
@@ -1137,6 +1219,11 @@ final class GenerativeModelTests: XCTestCase {
       XCTAssert(apiClientTags.contains(GenerativeAIService.languageTag))
       XCTAssert(apiClientTags.contains(GenerativeAIService.firebaseVersionTag))
       XCTAssertEqual(request.value(forHTTPHeaderField: "X-Firebase-AppCheck"), appCheckToken)
+      if let authToken {
+        XCTAssertEqual(request.value(forHTTPHeaderField: "Authorization"), "Firebase \(authToken)")
+      } else {
+        XCTAssertNil(request.value(forHTTPHeaderField: "Authorization"))
+      }
       let response = try XCTUnwrap(HTTPURLResponse(
         url: requestURL,
         statusCode: statusCode,
diff --git a/Package.swift b/Package.swift
index c10a21d6e95..95bc3365e30 100644
--- a/Package.swift
+++ b/Package.swift
@@ -1369,6 +1369,7 @@ let package = Package(
       name: "FirebaseVertexAI",
       dependencies: [
         "FirebaseAppCheckInterop",
+        "FirebaseAuthInterop",
         "FirebaseCore",
         "FirebaseCoreExtension",
       ],

From 6a7dd8ea3f8ef0732c55d1c95968aba91eb35124 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 01:35:47 +0800
Subject: [PATCH 106/124] Align and Close Keyboard

---
 .../Screens/ConversationScreen.swift          |  2 +
 .../Screens/FunctionCallingScreen.swift       |  2 +
 .../Screens/SummarizeScreen.swift             | 30 ++++++++-----
 .../MultimodalInputField.swift                |  7 ++-
 .../Sample/GoogleService-Info.plist           | 44 +++++++------------
 .../VertexAISample.xcodeproj/project.pbxproj  |  4 +-
 6 files changed, 46 insertions(+), 43 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
index a3044a1f430..b66782c5c1c 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
@@ -68,9 +68,11 @@ struct ConversationScreen: View {
       InputField("Message...", text: $userPrompt) {
         Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
           .font(.title)
+          .padding(.bottom, 3)
       }
       .focused($focusedField, equals: .message)
       .onSubmit { sendOrStop() }
+      .padding(.bottom, 5)
     }
     .toolbar {
       ToolbarItem(placement: .primaryAction) {
diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
index 30d222e29bc..5f0342f34c1 100644
--- a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
@@ -69,9 +69,11 @@ struct FunctionCallingScreen: View {
       InputField("Message...", text: $userPrompt) {
         Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
           .font(.title)
+          .padding(.bottom, 3)
       }
       .focused($focusedField, equals: .message)
       .onSubmit { sendOrStop() }
+      .padding(.bottom, 5)
     }
     .toolbar {
       ToolbarItem(placement: .primaryAction) {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
index 8fbb89f4482..8d9306ed02b 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -28,19 +28,22 @@ struct SummarizeScreen: View {
 
   var body: some View {
     VStack {
-      Text("Enter some text, then tap on _Go_ to summarize it.")
-      HStack(alignment: .top) {
-        TextField("Enter text summarize", text: $userInput, axis: .vertical)
-          .textFieldStyle(.roundedBorder)
-          .onSubmit {
-            onSummarizeTapped()
-          }
-        Button("Go") {
-          onSummarizeTapped()
+        VStack(alignment: .leading) {
+            Text("Enter some text, then tap on _Go_ to summarize it.")
+                .padding(.horizontal, 6)
+            HStack(alignment: .top) {
+                TextField("Enter text summarize", text: $userInput, axis: .vertical)
+                    .textFieldStyle(.roundedBorder)
+                    .onSubmit {
+                        onSummarizeTapped()
+                    }
+                Button("Go") {
+                    onSummarizeTapped()
+                }
+                .padding(.top, 4)
+            }
         }
-        .padding(.top, 4)
-      }
-      .padding([.horizontal, .bottom])
+        .padding(.horizontal, 16)
 
       List {
         HStack(alignment: .top) {
@@ -61,6 +64,9 @@ struct SummarizeScreen: View {
   }
 
   private func onSummarizeTapped() {
+    // MARK: Close Keyboard
+    UIApplication.shared.sendAction(#selector(UIResponder.resignFirstResponder), to: nil, from: nil, for: nil)
+
     Task {
       await viewModel.summarize(inputText: userInput)
     }
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
index 9519fb38a4b..51ee624de74 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
@@ -52,6 +52,9 @@ public struct MultimodalInputField: View {
   }
 
   private func submit() {
+    // MARK: Close Keyboard
+    UIApplication.shared.sendAction(#selector(UIResponder.resignFirstResponder), to: nil, from: nil, for: nil)
+      
     if let submitHandler {
       submitHandler()
     }
@@ -69,7 +72,7 @@ public struct MultimodalInputField: View {
         Button(action: showChooseAttachmentTypePicker) {
           Image(systemName: "plus")
         }
-        .padding(.top, 4)
+        .padding(.top, 10)
 
         VStack(alignment: .leading) {
           TextField(
@@ -110,7 +113,7 @@ public struct MultimodalInputField: View {
         Button(action: submit) {
           Text("Go")
         }
-        .padding(.top, 4)
+        .padding(.top, 8)
       }
     }
     .padding(.horizontal)
diff --git a/FirebaseVertexAI/Sample/GoogleService-Info.plist b/FirebaseVertexAI/Sample/GoogleService-Info.plist
index b026413cc39..bee7e31f325 100644
--- a/FirebaseVertexAI/Sample/GoogleService-Info.plist
+++ b/FirebaseVertexAI/Sample/GoogleService-Info.plist
@@ -2,39 +2,29 @@
 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
 <plist version="1.0">
 <dict>
-	<key>AD_UNIT_ID_FOR_BANNER_TEST</key>
-	<string>ca-app-pub-3940256099942544/2934735716</string>
-	<key>AD_UNIT_ID_FOR_INTERSTITIAL_TEST</key>
-	<string>ca-app-pub-3940256099942544/4411468910</string>
 	<key>API_KEY</key>
-	<string>AIzaSyAzlj4APqi5S58nFtE52Da0fYBOHA2MhaY</string>
+	<string>AIzaSyDE8eTNF_85FVtJWgJjYZaI_q_sE3CDzKc</string>
+	<key>GCM_SENDER_ID</key>
+	<string>212738571313</string>
+	<key>PLIST_VERSION</key>
+	<string>1</string>
 	<key>BUNDLE_ID</key>
 	<string>com.google.firebase.VertexAISample</string>
-	<key>CLIENT_ID</key>
-	<string>123456789000-hjugbg6ud799v4c49dim8ce2usclthar.apps.googleusercontent.com</string>
-	<key>DATABASE_URL</key>
-	<string>https://mockproject-1234.firebaseio.com</string>
-	<key>GCM_SENDER_ID</key>
-	<string>123456789000</string>
-	<key>GOOGLE_APP_ID</key>
-	<string>1:123456789000:ios:f1bf012572b04063</string>
+	<key>PROJECT_ID</key>
+	<string>vertexai-286b9</string>
+	<key>STORAGE_BUCKET</key>
+	<string>vertexai-286b9.appspot.com</string>
 	<key>IS_ADS_ENABLED</key>
-	<true/>
+	<false></false>
 	<key>IS_ANALYTICS_ENABLED</key>
-	<true/>
+	<false></false>
 	<key>IS_APPINVITE_ENABLED</key>
-	<true/>
+	<true></true>
 	<key>IS_GCM_ENABLED</key>
-	<true/>
+	<true></true>
 	<key>IS_SIGNIN_ENABLED</key>
-	<true/>
-	<key>PLIST_VERSION</key>
-	<string>1</string>
-	<key>PROJECT_ID</key>
-	<string>mockproject-1234</string>
-	<key>REVERSED_CLIENT_ID</key>
-	<string>com.googleusercontent.apps.123456789000-hjugbg6ud799v4c49dim8ce2usclthar</string>
-	<key>STORAGE_BUCKET</key>
-	<string>mockproject-1234.appspot.com</string>
+	<true></true>
+	<key>GOOGLE_APP_ID</key>
+	<string>1:212738571313:ios:c6639330d47159aa59e910</string>
 </dict>
-</plist>
+</plist>
\ No newline at end of file
diff --git a/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
index a5c19152df7..6646b91eb5b 100644
--- a/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
@@ -502,7 +502,7 @@
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
+				DEVELOPMENT_TEAM = 9PAHLTG8AD;
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
 				GENERATE_INFOPLIST_FILE = YES;
@@ -532,7 +532,7 @@
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
-				DEVELOPMENT_TEAM = "";
+				DEVELOPMENT_TEAM = 9PAHLTG8AD;
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
 				GENERATE_INFOPLIST_FILE = YES;

From 0ef7e556ac689d1ec9a35c06306d7818cd1934ee Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 01:39:32 +0800
Subject: [PATCH 107/124] Revert Google Info

This reverts commit dd592a716f4bc70b33b13dabd8958925c5eaf7bc.
---
 .../Sample/GoogleService-Info.plist           | 44 ++++++++++++-------
 1 file changed, 27 insertions(+), 17 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GoogleService-Info.plist b/FirebaseVertexAI/Sample/GoogleService-Info.plist
index bee7e31f325..b026413cc39 100644
--- a/FirebaseVertexAI/Sample/GoogleService-Info.plist
+++ b/FirebaseVertexAI/Sample/GoogleService-Info.plist
@@ -2,29 +2,39 @@
 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
 <plist version="1.0">
 <dict>
+	<key>AD_UNIT_ID_FOR_BANNER_TEST</key>
+	<string>ca-app-pub-3940256099942544/2934735716</string>
+	<key>AD_UNIT_ID_FOR_INTERSTITIAL_TEST</key>
+	<string>ca-app-pub-3940256099942544/4411468910</string>
 	<key>API_KEY</key>
-	<string>AIzaSyDE8eTNF_85FVtJWgJjYZaI_q_sE3CDzKc</string>
-	<key>GCM_SENDER_ID</key>
-	<string>212738571313</string>
-	<key>PLIST_VERSION</key>
-	<string>1</string>
+	<string>AIzaSyAzlj4APqi5S58nFtE52Da0fYBOHA2MhaY</string>
 	<key>BUNDLE_ID</key>
 	<string>com.google.firebase.VertexAISample</string>
-	<key>PROJECT_ID</key>
-	<string>vertexai-286b9</string>
-	<key>STORAGE_BUCKET</key>
-	<string>vertexai-286b9.appspot.com</string>
+	<key>CLIENT_ID</key>
+	<string>123456789000-hjugbg6ud799v4c49dim8ce2usclthar.apps.googleusercontent.com</string>
+	<key>DATABASE_URL</key>
+	<string>https://mockproject-1234.firebaseio.com</string>
+	<key>GCM_SENDER_ID</key>
+	<string>123456789000</string>
+	<key>GOOGLE_APP_ID</key>
+	<string>1:123456789000:ios:f1bf012572b04063</string>
 	<key>IS_ADS_ENABLED</key>
-	<false></false>
+	<true/>
 	<key>IS_ANALYTICS_ENABLED</key>
-	<false></false>
+	<true/>
 	<key>IS_APPINVITE_ENABLED</key>
-	<true></true>
+	<true/>
 	<key>IS_GCM_ENABLED</key>
-	<true></true>
+	<true/>
 	<key>IS_SIGNIN_ENABLED</key>
-	<true></true>
-	<key>GOOGLE_APP_ID</key>
-	<string>1:212738571313:ios:c6639330d47159aa59e910</string>
+	<true/>
+	<key>PLIST_VERSION</key>
+	<string>1</string>
+	<key>PROJECT_ID</key>
+	<string>mockproject-1234</string>
+	<key>REVERSED_CLIENT_ID</key>
+	<string>com.googleusercontent.apps.123456789000-hjugbg6ud799v4c49dim8ce2usclthar</string>
+	<key>STORAGE_BUCKET</key>
+	<string>mockproject-1234.appspot.com</string>
 </dict>
-</plist>
\ No newline at end of file
+</plist>

From b0618f296adaac15e8d75cd8dd6298b159cad26d Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 01:40:38 +0800
Subject: [PATCH 108/124] Rm TeamID

---
 .../Sample/VertexAISample.xcodeproj/project.pbxproj           | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
index 6646b91eb5b..a5c19152df7 100644
--- a/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
+++ b/FirebaseVertexAI/Sample/VertexAISample.xcodeproj/project.pbxproj
@@ -502,7 +502,7 @@
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
-				DEVELOPMENT_TEAM = 9PAHLTG8AD;
+				DEVELOPMENT_TEAM = "";
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
 				GENERATE_INFOPLIST_FILE = YES;
@@ -532,7 +532,7 @@
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 1;
 				DEVELOPMENT_ASSET_PATHS = "\"VertexAISample/Preview Content\"";
-				DEVELOPMENT_TEAM = 9PAHLTG8AD;
+				DEVELOPMENT_TEAM = "";
 				ENABLE_PREVIEWS = YES;
 				ENABLE_USER_SCRIPT_SANDBOXING = NO;
 				GENERATE_INFOPLIST_FILE = YES;

From 968dac7265f8ad77ca24389ddd18951a37c76d0d Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 01:56:11 +0800
Subject: [PATCH 109/124] Update VertexAISampleApp.swift

---
 FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift | 1 +
 1 file changed, 1 insertion(+)

diff --git a/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
index 4772bcc2d7d..2a1518844f9 100644
--- a/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
@@ -38,3 +38,4 @@ struct VertexAISampleApp: App {
     }
   }
 }
+

From 419a7ac688a66737723e9c4ab4daa84b73c694d8 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 01:57:16 +0800
Subject: [PATCH 110/124] Update VertexAISampleApp.swift

---
 FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift | 1 -
 1 file changed, 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
index 2a1518844f9..4772bcc2d7d 100644
--- a/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
+++ b/FirebaseVertexAI/Sample/VertexAISample/VertexAISampleApp.swift
@@ -38,4 +38,3 @@ struct VertexAISampleApp: App {
     }
   }
 }
-

From 04239e104598410255292c2c458758c13e3722a8 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 02:23:46 +0800
Subject: [PATCH 111/124] Style

---
 .../Screens/SummarizeScreen.swift             | 35 +++++++++++--------
 .../MultimodalInputField.swift                | 10 ++++--
 2 files changed, 28 insertions(+), 17 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
index 8d9306ed02b..87d739bde7e 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -28,22 +28,22 @@ struct SummarizeScreen: View {
 
   var body: some View {
     VStack {
-        VStack(alignment: .leading) {
-            Text("Enter some text, then tap on _Go_ to summarize it.")
-                .padding(.horizontal, 6)
-            HStack(alignment: .top) {
-                TextField("Enter text summarize", text: $userInput, axis: .vertical)
-                    .textFieldStyle(.roundedBorder)
-                    .onSubmit {
-                        onSummarizeTapped()
-                    }
-                Button("Go") {
-                    onSummarizeTapped()
-                }
-                .padding(.top, 4)
+      VStack(alignment: .leading) {
+        Text("Enter some text, then tap on _Go_ to summarize it.")
+          .padding(.horizontal, 6)
+        HStack(alignment: .top) {
+          TextField("Enter text summarize", text: $userInput, axis: .vertical)
+            .textFieldStyle(.roundedBorder)
+            .onSubmit {
+              onSummarizeTapped()
             }
+          Button("Go") {
+            onSummarizeTapped()
+          }
+          .padding(.top, 4)
         }
-        .padding(.horizontal, 16)
+      }
+      .padding(.horizontal, 16)
 
       List {
         HStack(alignment: .top) {
@@ -65,7 +65,12 @@ struct SummarizeScreen: View {
 
   private func onSummarizeTapped() {
     // MARK: Close Keyboard
-    UIApplication.shared.sendAction(#selector(UIResponder.resignFirstResponder), to: nil, from: nil, for: nil)
+    UIApplication.shared.sendAction(
+      #selector(UIResponder.resignFirstResponder),
+      to: nil,
+      from: nil,
+      for: nil
+    )
 
     Task {
       await viewModel.summarize(inputText: userInput)
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
index 51ee624de74..8916eb652b3 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
@@ -53,8 +53,14 @@ public struct MultimodalInputField: View {
 
   private func submit() {
     // MARK: Close Keyboard
-    UIApplication.shared.sendAction(#selector(UIResponder.resignFirstResponder), to: nil, from: nil, for: nil)
-      
+
+    UIApplication.shared.sendAction(
+      #selector(UIResponder.resignFirstResponder),
+      to: nil,
+      from: nil,
+      for: nil
+    )
+
     if let submitHandler {
       submitHandler()
     }

From 871b9bd2640ab12976781832d8eb34defa28cc96 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 02:37:40 +0800
Subject: [PATCH 112/124] Update SummarizeScreen.swift

---
 .../Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
index 87d739bde7e..e18a4868e8c 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -65,6 +65,7 @@ struct SummarizeScreen: View {
 
   private func onSummarizeTapped() {
     // MARK: Close Keyboard
+
     UIApplication.shared.sendAction(
       #selector(UIResponder.resignFirstResponder),
       to: nil,

From 7d061c54496a3c4a21f50bbb9d6760eb3bd95d34 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 3 May 2024 19:13:37 +0800
Subject: [PATCH 113/124] Fix Keyboard Close

---
 .../Sample/ChatSample/Screens/ConversationScreen.swift   | 1 +
 .../GenerativeAITextSample/Screens/SummarizeScreen.swift | 8 ++------
 .../GenerativeAIUIComponents/MultimodalInputField.swift  | 9 ---------
 3 files changed, 3 insertions(+), 15 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
index b66782c5c1c..ad6acb3b7a0 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
@@ -96,6 +96,7 @@ struct ConversationScreen: View {
   }
 
   private func sendOrStop() {
+    focusedField = nil
     if viewModel.busy {
       viewModel.stop()
     } else {
diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
index e18a4868e8c..c570f8743cb 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -33,6 +33,7 @@ struct SummarizeScreen: View {
           .padding(.horizontal, 6)
         HStack(alignment: .top) {
           TextField("Enter text summarize", text: $userInput, axis: .vertical)
+            .focused($focusedField, equals: .message)
             .textFieldStyle(.roundedBorder)
             .onSubmit {
               onSummarizeTapped()
@@ -66,12 +67,7 @@ struct SummarizeScreen: View {
   private func onSummarizeTapped() {
     // MARK: Close Keyboard
 
-    UIApplication.shared.sendAction(
-      #selector(UIResponder.resignFirstResponder),
-      to: nil,
-      from: nil,
-      for: nil
-    )
+    focusedField = nil
 
     Task {
       await viewModel.summarize(inputText: userInput)
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
index 8916eb652b3..128da4a4e23 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/MultimodalInputField.swift
@@ -52,15 +52,6 @@ public struct MultimodalInputField: View {
   }
 
   private func submit() {
-    // MARK: Close Keyboard
-
-    UIApplication.shared.sendAction(
-      #selector(UIResponder.resignFirstResponder),
-      to: nil,
-      from: nil,
-      for: nil
-    )
-
     if let submitHandler {
       submitHandler()
     }

From d639c34257604164777b6a3d72ccbf101ad11c48 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:12:29 +0800
Subject: [PATCH 114/124] Update InputField

---
 .../Screens/ConversationScreen.swift          |  1 -
 .../GenerativeAIUIComponents/InputField.swift | 46 +++++++++----------
 2 files changed, 23 insertions(+), 24 deletions(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
index ad6acb3b7a0..ea9e066357c 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
@@ -68,7 +68,6 @@ struct ConversationScreen: View {
       InputField("Message...", text: $userPrompt) {
         Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
           .font(.title)
-          .padding(.bottom, 3)
       }
       .focused($focusedField, equals: .message)
       .onSubmit { sendOrStop() }
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
index 3f12ea65d8a..5897c6ac793 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
@@ -38,32 +38,32 @@ public struct InputField<Label>: View where Label: View {
   }
 
   public var body: some View {
-    VStack(alignment: .leading) {
-      HStack(alignment: .bottom) {
-        VStack(alignment: .leading) {
-          TextField(
-            title ?? "",
-            text: $text,
-            axis: .vertical
-          )
+      VStack(alignment: .leading) {
+        HStack(alignment: .bottom) {
+          VStack(alignment: .leading) {
+            TextField(
+              title ?? "",
+              text: $text,
+              axis: .vertical
+            )
+            .padding(.vertical, 4)
+            .onSubmit(submit)
+          }
+          .padding(.horizontal, 8)
           .padding(.vertical, 4)
-          .onSubmit(submit)
-        }
-        .padding(.horizontal, 8)
-        .padding(.vertical, 4)
-        .overlay {
-          RoundedRectangle(
-            cornerRadius: 8,
-            style: .continuous
-          )
-          .stroke(Color(UIColor.systemFill), lineWidth: 1)
-        }
+          .overlay {
+            RoundedRectangle(
+              cornerRadius: 8,
+              style: .continuous
+            )
+            .stroke(Color(UIColor.systemFill), lineWidth: 1)
+          }
 
-        Button(action: submit, label: label)
-          .padding(.top, 4)
+          Button(action: submit, label: label)
+            .padding(.bottom, 4)
+        }
       }
-    }
-    .padding(.horizontal)
+      .padding(8)
   }
 }
 

From b3f7ea0d1ad0096aff704ea19aa2097bbebeaab3 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:13:58 +0800
Subject: [PATCH 115/124] Remove Padding

---
 .../Sample/ChatSample/Screens/ConversationScreen.swift           | 1 -
 1 file changed, 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
index ea9e066357c..ae419e06b00 100644
--- a/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
+++ b/FirebaseVertexAI/Sample/ChatSample/Screens/ConversationScreen.swift
@@ -71,7 +71,6 @@ struct ConversationScreen: View {
       }
       .focused($focusedField, equals: .message)
       .onSubmit { sendOrStop() }
-      .padding(.bottom, 5)
     }
     .toolbar {
       ToolbarItem(placement: .primaryAction) {

From 689f651b064fbeadc08de991460f1223dac49a6b Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:15:49 +0800
Subject: [PATCH 116/124] Remove Padding

---
 .../FunctionCallingSample/Screens/FunctionCallingScreen.swift   | 2 --
 1 file changed, 2 deletions(-)

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
index 5f0342f34c1..30d222e29bc 100644
--- a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
@@ -69,11 +69,9 @@ struct FunctionCallingScreen: View {
       InputField("Message...", text: $userPrompt) {
         Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")
           .font(.title)
-          .padding(.bottom, 3)
       }
       .focused($focusedField, equals: .message)
       .onSubmit { sendOrStop() }
-      .padding(.bottom, 5)
     }
     .toolbar {
       ToolbarItem(placement: .primaryAction) {

From 97f97cba3ca2f0fc46c26b7d34dcf00f4172ce8f Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:18:38 +0800
Subject: [PATCH 117/124] Remove Comment and empty line

---
 .../Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift | 2 --
 1 file changed, 2 deletions(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
index c570f8743cb..748c1addd5f 100644
--- a/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/Screens/SummarizeScreen.swift
@@ -65,8 +65,6 @@ struct SummarizeScreen: View {
   }
 
   private func onSummarizeTapped() {
-    // MARK: Close Keyboard
-
     focusedField = nil
 
     Task {

From b524818982f7e89d619751c107ab19faa8486bc4 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:28:42 +0800
Subject: [PATCH 118/124] Added Focus to PhotoReasoningScreen

---
 .../Screens/PhotoReasoningScreen.swift              | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
index 98f327585db..4866f8b9695 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
@@ -19,10 +19,18 @@ import SwiftUI
 
 struct PhotoReasoningScreen: View {
   @StateObject var viewModel = PhotoReasoningViewModel()
+    
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
 
   var body: some View {
     VStack {
       MultimodalInputField(text: $viewModel.userInput, selection: $viewModel.selectedItems)
+        .focused($focusedField, equals: .message)
         .onSubmit {
           onSendTapped()
         }
@@ -47,11 +55,16 @@ struct PhotoReasoningScreen: View {
       }
     }
     .navigationTitle("Multimodal sample")
+    .onAppear {
+      focusedField = .message
+    }
   }
 
   // MARK: - Actions
 
   private func onSendTapped() {
+    focusedField = nil
+      
     Task {
       await viewModel.reason()
     }

From f382ea612cdfae3a81cd679df5c0efae77e50b73 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:39:02 +0800
Subject: [PATCH 119/124] Update FunctionCallingScreen.swift

---
 .../FunctionCallingSample/Screens/FunctionCallingScreen.swift  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
index 30d222e29bc..7f790c5d846 100644
--- a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
@@ -65,6 +65,9 @@ struct FunctionCallingScreen: View {
             }
           }
         })
+        .onTapGesture {
+            focusedField = nil
+        }
       }
       InputField("Message...", text: $userPrompt) {
         Image(systemName: viewModel.busy ? "stop.circle.fill" : "arrow.up.circle.fill")

From b2606799a730bde2d6e87781ba481c714fa63528 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 20:39:37 +0800
Subject: [PATCH 120/124] Change from user to system

---
 .../ViewModels/FunctionCallingViewModel.swift                   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
index f18c0e85fc1..91299dbf711 100644
--- a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
@@ -249,7 +249,7 @@ private extension FunctionResponse {
     }
     let messageText = "Function response returned by app:\n```\n\(json)\n```"
 
-    return ChatMessage(message: messageText, participant: .user)
+    return ChatMessage(message: messageText, participant: .system)
   }
 }
 

From e15bc9541a58a736cd45342e4735606f149c01e7 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Fri, 10 May 2024 23:54:35 +0800
Subject: [PATCH 121/124] Update Indent / Style

---
 .../Screens/FunctionCallingScreen.swift       |  2 +-
 .../Screens/PhotoReasoningScreen.swift        |  4 +-
 .../GenerativeAIUIComponents/InputField.swift | 46 +++++++++----------
 3 files changed, 26 insertions(+), 26 deletions(-)

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
index 7f790c5d846..f16da39e22f 100644
--- a/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/Screens/FunctionCallingScreen.swift
@@ -66,7 +66,7 @@ struct FunctionCallingScreen: View {
           }
         })
         .onTapGesture {
-            focusedField = nil
+          focusedField = nil
         }
       }
       InputField("Message...", text: $userPrompt) {
diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
index 4866f8b9695..930214770d8 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/Screens/PhotoReasoningScreen.swift
@@ -19,7 +19,7 @@ import SwiftUI
 
 struct PhotoReasoningScreen: View {
   @StateObject var viewModel = PhotoReasoningViewModel()
-    
+
   enum FocusedField: Hashable {
     case message
   }
@@ -64,7 +64,7 @@ struct PhotoReasoningScreen: View {
 
   private func onSendTapped() {
     focusedField = nil
-      
+
     Task {
       await viewModel.reason()
     }
diff --git a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
index 5897c6ac793..67941c370cb 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIUIComponents/Sources/GenerativeAIUIComponents/InputField.swift
@@ -38,32 +38,32 @@ public struct InputField<Label>: View where Label: View {
   }
 
   public var body: some View {
-      VStack(alignment: .leading) {
-        HStack(alignment: .bottom) {
-          VStack(alignment: .leading) {
-            TextField(
-              title ?? "",
-              text: $text,
-              axis: .vertical
-            )
-            .padding(.vertical, 4)
-            .onSubmit(submit)
-          }
-          .padding(.horizontal, 8)
+    VStack(alignment: .leading) {
+      HStack(alignment: .bottom) {
+        VStack(alignment: .leading) {
+          TextField(
+            title ?? "",
+            text: $text,
+            axis: .vertical
+          )
           .padding(.vertical, 4)
-          .overlay {
-            RoundedRectangle(
-              cornerRadius: 8,
-              style: .continuous
-            )
-            .stroke(Color(UIColor.systemFill), lineWidth: 1)
-          }
-
-          Button(action: submit, label: label)
-            .padding(.bottom, 4)
+          .onSubmit(submit)
+        }
+        .padding(.horizontal, 8)
+        .padding(.vertical, 4)
+        .overlay {
+          RoundedRectangle(
+            cornerRadius: 8,
+            style: .continuous
+          )
+          .stroke(Color(UIColor.systemFill), lineWidth: 1)
         }
+
+        Button(action: submit, label: label)
+          .padding(.bottom, 4)
       }
-      .padding(8)
+    }
+    .padding(8)
   }
 }
 

From 65b440f423a28780fbca48ea8057ab9d2e95cf30 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Tue, 14 May 2024 10:12:05 +0800
Subject: [PATCH 122/124] Revert "Change from user to system"

This reverts commit 6cddfe32708235cb32a11299954303ea6a47e458.
---
 .../ViewModels/FunctionCallingViewModel.swift                   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
index 91299dbf711..f18c0e85fc1 100644
--- a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
+++ b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift
@@ -249,7 +249,7 @@ private extension FunctionResponse {
     }
     let messageText = "Function response returned by app:\n```\n\(json)\n```"
 
-    return ChatMessage(message: messageText, participant: .system)
+    return ChatMessage(message: messageText, participant: .user)
   }
 }
 

From 4c87d464b65860d104e226573a222f820bc43fbc Mon Sep 17 00:00:00 2001
From: MING <54872601+1998code@users.noreply.github.com>
Date: Tue, 14 May 2024 14:26:04 +0800
Subject: [PATCH 123/124] Correct Package.swift

---
 Package.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Package.swift b/Package.swift
index 0ca1d5a4432..205639fb162 100644
--- a/Package.swift
+++ b/Package.swift
@@ -19,7 +19,7 @@
 import class Foundation.ProcessInfo
 import PackageDescription
 
-let firebaseVersion = "0.1.0-vertexai-preview"
+let firebaseVersion = "10.27.0"
 
 let package = Package(
   name: "Firebase",
@@ -191,7 +191,7 @@ let package = Package(
       url: "https://github.com/google/interop-ios-for-google-sdks.git",
       "100.0.0" ..< "101.0.0"
     ),
-    .package(url: "https://github.com/google/app-check.git", "10.18.0" ..< "11.0.0"),
+    .package(url: "https://github.com/google/app-check.git", "10.19.0" ..< "11.0.0"),
   ],
   targets: [
     .target(

From 3c27e70cf32755d171ab8a7661dace3b4ea19db5 Mon Sep 17 00:00:00 2001
From: 1998code <54872601+1998code@users.noreply.github.com>
Date: Tue, 14 May 2024 23:42:31 +0800
Subject: [PATCH 124/124] Delete GenerativeAISwift.swift

---
 .../Sources/GenerativeAISwift.swift           | 25 -------------------
 1 file changed, 25 deletions(-)
 delete mode 100644 FirebaseVertexAI/Sources/GenerativeAISwift.swift

diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
deleted file mode 100644
index 933a19c4759..00000000000
--- a/FirebaseVertexAI/Sources/GenerativeAISwift.swift
+++ /dev/null
@@ -1,25 +0,0 @@
-// Copyright 2023 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//      http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-import Foundation
-
-#if !os(macOS) && !os(iOS)
-  #warning("Only iOS, macOS, and Catalyst targets are currently fully supported.")
-#endif
-
-/// Constants associated with the GenerativeAISwift SDK
-@available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
-public enum GenerativeAISwift {
-  static let baseURL = "https://firebaseml.googleapis.com"
-}