[Vertex AI] Add FirebaseVertexAIPreview SPM target for preview APIs

Author: andrewheard

FirebaseVertexAI/Sources/VertexAI+Preview.swift

@@ -0,0 +1,30 @@
+// Copyright 2025 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
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
+public extension VertexAI {
+  func imagenModel(modelName: String, modelConfig: ImagenModelConfig? = nil,
+                   safetySettings: ImagenSafetySettings? = nil,
+                   requestOptions: RequestOptions = RequestOptions()) -> ImagenModel {
+    return createImagenModel(
+      modelName: modelName,
+      modelConfig: modelConfig,
+      safetySettings: safetySettings,
+      requestOptions: requestOptions
+    )
+  }
+}

FirebaseVertexAI/Sources/VertexAI.swift

@@ -104,9 +104,11 @@ public class VertexAI {
     )
   }
 
-  public func imagenModel(modelName: String, modelConfig: ImagenModelConfig? = nil,
-                          safetySettings: ImagenSafetySettings? = nil,
-                          requestOptions: RequestOptions = RequestOptions()) -> ImagenModel {
+  // MARK: - Package APIs
+
+  package func createImagenModel(modelName: String, modelConfig: ImagenModelConfig?,
+                                 safetySettings: ImagenSafetySettings?,
+                                 requestOptions: RequestOptions) -> ImagenModel {
     return ImagenModel(
       name: modelResourceName(modelName: modelName),
       projectID: projectID,
@@ -119,12 +121,12 @@ public class VertexAI {
     )
   }
 
+  // MARK: - Private
+
   /// Class to enable VertexAI to register via the Objective-C based Firebase component system
   /// to include VertexAI in the userAgent.
   @objc(FIRVertexAIComponent) class FirebaseVertexAIComponent: NSObject {}
 
-  // MARK: - Private
-
   /// The `FirebaseApp` associated with this `VertexAI` instance.
   private let app: FirebaseApp
 

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

@@ -16,6 +16,7 @@ import FirebaseAuth
 import FirebaseCore
 import FirebaseStorage
 import FirebaseVertexAI
+import FirebaseVertexAIPreview
 import VertexAITestApp
 import XCTest
 

FirebaseVertexAI/Tests/TestApp/VertexAITestApp.xcodeproj/project.pbxproj

@@ -8,6 +8,7 @@
 
 /* Begin PBXBuildFile section */
 		862218812D04E098007ED2D4 /* IntegrationTestUtils.swift in Sources */ = {isa = PBXBuildFile; fileRef = 862218802D04E08D007ED2D4 /* IntegrationTestUtils.swift */; };
+		86401F692D31B86E0026CAED /* FirebaseVertexAIPreview in Frameworks */ = {isa = PBXBuildFile; productRef = 86401F682D31B86E0026CAED /* FirebaseVertexAIPreview */; };
 		8661385C2CC943DD00F4B78E /* TestApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8661385B2CC943DD00F4B78E /* TestApp.swift */; };
 		8661385E2CC943DD00F4B78E /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8661385D2CC943DD00F4B78E /* ContentView.swift */; };
 		8661386E2CC943DE00F4B78E /* IntegrationTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8661386D2CC943DE00F4B78E /* IntegrationTests.swift */; };
@@ -55,6 +56,7 @@
 			buildActionMask = 2147483647;
 			files = (
 				8692F2982CC9477800539E8F /* FirebaseAppCheck in Frameworks */,
+				86401F692D31B86E0026CAED /* FirebaseVertexAIPreview in Frameworks */,
 				8692F29E2CC9477800539E8F /* FirebaseVertexAI in Frameworks */,
 				8692F29A2CC9477800539E8F /* FirebaseAuth in Frameworks */,
 				8692F29C2CC9477800539E8F /* FirebaseStorage in Frameworks */,
@@ -167,6 +169,7 @@
 				8692F2992CC9477800539E8F /* FirebaseAuth */,
 				8692F29B2CC9477800539E8F /* FirebaseStorage */,
 				8692F29D2CC9477800539E8F /* FirebaseVertexAI */,
+				86401F682D31B86E0026CAED /* FirebaseVertexAIPreview */,
 			);
 			productName = VertexAITestApp;
 			productReference = 866138582CC943DD00F4B78E /* VertexAITestApp-SPM.app */;
@@ -560,6 +563,10 @@
 /* End XCLocalSwiftPackageReference section */
 
 /* Begin XCSwiftPackageProductDependency section */
+		86401F682D31B86E0026CAED /* FirebaseVertexAIPreview */ = {
+			isa = XCSwiftPackageProductDependency;
+			productName = FirebaseVertexAIPreview;
+		};
 		8692F2972CC9477800539E8F /* FirebaseAppCheck */ = {
 			isa = XCSwiftPackageProductDependency;
 			productName = FirebaseAppCheck;

Package.swift

@@ -125,6 +125,10 @@ let package = Package(
       name: "FirebaseVertexAI",
       targets: ["FirebaseVertexAI"]
     ),
+    .library(
+      name: "FirebaseVertexAIPreview",
+      targets: ["FirebaseVertexAIPreview"]
+    ),
   ],
   dependencies: [
     .package(
@@ -1306,11 +1310,27 @@ let package = Package(
         "FirebaseCore",
         "FirebaseCoreExtension",
       ],
-      path: "FirebaseVertexAI/Sources"
+      path: "FirebaseVertexAI/Sources",
+      exclude: [
+        "VertexAI+Preview.swift",
+      ]
+    ),
+    .target(
+      name: "FirebaseVertexAIPreview",
+      dependencies: [
+        "FirebaseVertexAI",
+      ],
+      path: "FirebaseVertexAI/Sources",
+      sources: [
+        "VertexAI+Preview.swift",
+      ]
     ),
     .testTarget(
       name: "FirebaseVertexAIUnit",
-      dependencies: ["FirebaseVertexAI"],
+      dependencies: [
+        "FirebaseVertexAI",
+        "FirebaseVertexAIPreview",
+      ],
       path: "FirebaseVertexAI/Tests/Unit",
       resources: [
         .process("vertexai-sdk-test-data/mock-responses"),