@ai-sdk/google-vertex 5.0.0-beta.62 → 5.0.0-beta.64

package/dist/index.js

@@ -1,4 +1,4 @@
-// src/google-vertex-provider-node.ts
+// src/google-vertex-provider.ts
 import { loadOptionalSetting as loadOptionalSetting2, resolve as resolve5 } from "@ai-sdk/provider-utils";
 
 // src/google-vertex-auth-google-auth-library.ts
@@ -22,7 +22,7 @@ async function generateAuthToken(options) {
   return (token == null ? void 0 : token.token) || null;
 }
 
-// src/google-vertex-provider.ts
+// src/google-vertex-provider-base.ts
 import { GoogleLanguageModel as GoogleLanguageModel2 } from "@ai-sdk/google/internal";
 import {
   generateId,
@@ -35,7 +35,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.0-beta.62" : "0.0.0-test";
+var VERSION = true ? "5.0.0-beta.64" : "0.0.0-test";
 
 // src/google-vertex-embedding-model.ts
 import {
@@ -70,7 +70,7 @@ var googleVertexFailedResponseHandler = createJsonErrorResponseHandler(
   }
 );
 
-// src/google-vertex-embedding-options.ts
+// src/google-vertex-embedding-model-options.ts
 import { z as z2 } from "zod/v4";
 var googleVertexEmbeddingModelOptions = z2.object({
   /**
@@ -141,10 +141,17 @@ var GoogleVertexEmbeddingModel = class _GoogleVertexEmbeddingModel {
     providerOptions
   }) {
     let googleOptions = await parseProviderOptions({
-      provider: "vertex",
+      provider: "googleVertex",
       providerOptions,
       schema: googleVertexEmbeddingModelOptions
     });
+    if (googleOptions == null) {
+      googleOptions = await parseProviderOptions({
+        provider: "vertex",
+        providerOptions,
+        schema: googleVertexEmbeddingModelOptions
+      });
+    }
     if (googleOptions == null) {
       googleOptions = await parseProviderOptions({
         provider: "google",
@@ -234,7 +241,68 @@ import {
   WORKFLOW_SERIALIZE as WORKFLOW_SERIALIZE2,
   WORKFLOW_DESERIALIZE as WORKFLOW_DESERIALIZE2
 } from "@ai-sdk/provider-utils";
+import { z as z5 } from "zod/v4";
+
+// src/google-vertex-image-model-options.ts
 import { z as z4 } from "zod/v4";
+var googleVertexImageModelOptionsSchema = z4.object({
+  negativePrompt: z4.string().nullish(),
+  personGeneration: z4.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+  safetySetting: z4.enum([
+    "block_low_and_above",
+    "block_medium_and_above",
+    "block_only_high",
+    "block_none"
+  ]).nullish(),
+  addWatermark: z4.boolean().nullish(),
+  storageUri: z4.string().nullish(),
+  sampleImageSize: z4.enum(["1K", "2K"]).nullish(),
+  /**
+   * Configuration for image editing operations
+   */
+  edit: z4.object({
+    /**
+     * An integer that represents the number of sampling steps.
+     * A higher value offers better image quality, a lower value offers better latency.
+     * Try 35 steps to start. If the quality doesn't meet your requirements,
+     * increase the value towards an upper limit of 75.
+     */
+    baseSteps: z4.number().nullish(),
+    // Edit mode options
+    // https://cloud.google.com/vertex-ai/generative-ai/docs/image/edit-insert-objects
+    mode: z4.enum([
+      "EDIT_MODE_INPAINT_INSERTION",
+      "EDIT_MODE_INPAINT_REMOVAL",
+      "EDIT_MODE_OUTPAINT",
+      "EDIT_MODE_CONTROLLED_EDITING",
+      "EDIT_MODE_PRODUCT_IMAGE",
+      "EDIT_MODE_BGSWAP"
+    ]).nullish(),
+    /**
+     * The mask mode to use.
+     * - `MASK_MODE_DEFAULT` - Default value for mask mode.
+     * - `MASK_MODE_USER_PROVIDED` - User provided mask. No segmentation needed.
+     * - `MASK_MODE_DETECTION_BOX` - Mask from detected bounding boxes.
+     * - `MASK_MODE_CLOTHING_AREA` - Masks from segmenting the clothing area with open-vocab segmentation.
+     * - `MASK_MODE_PARSED_PERSON` - Masks from segmenting the person body and clothing using the person-parsing model.
+     */
+    maskMode: z4.enum([
+      "MASK_MODE_DEFAULT",
+      "MASK_MODE_USER_PROVIDED",
+      "MASK_MODE_DETECTION_BOX",
+      "MASK_MODE_CLOTHING_AREA",
+      "MASK_MODE_PARSED_PERSON"
+    ]).nullish(),
+    /**
+     * Optional. A float value between 0 and 1, inclusive, that represents the
+     * percentage of the image width to grow the mask by. Using dilation helps
+     * compensate for imprecise masks. We recommend a value of 0.01.
+     */
+    maskDilation: z4.number().nullish()
+  }).nullish()
+});
+
+// src/google-vertex-image-model.ts
 var GoogleVertexImageModel = class _GoogleVertexImageModel {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -277,7 +345,7 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
     files,
     mask
   }) {
-    var _a, _b, _c, _d, _e, _f, _g;
+    var _a, _b, _c, _d, _e, _f;
     const warnings = [];
     if (size != null) {
       warnings.push({
@@ -286,12 +354,16 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
         details: "This model does not support the `size` option. Use `aspectRatio` instead."
       });
     }
-    const vertexImageOptions = await parseProviderOptions2({
+    const googleVertexImageOptions = (_a = await parseProviderOptions2({
+      provider: "googleVertex",
+      providerOptions,
+      schema: googleVertexImageModelOptionsSchema
+    })) != null ? _a : await parseProviderOptions2({
       provider: "vertex",
       providerOptions,
       schema: googleVertexImageModelOptionsSchema
     });
-    const { edit, ...otherOptions } = vertexImageOptions != null ? vertexImageOptions : {};
+    const { edit, ...otherOptions } = googleVertexImageOptions != null ? googleVertexImageOptions : {};
     const { mode: editMode, baseSteps, maskMode, maskDilation } = edit != null ? edit : {};
     const isEditMode = files != null && files.length > 0;
     let body;
@@ -347,7 +419,7 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
         }
       };
     }
-    const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
+    const currentDate = (_d = (_c = (_b = this.config._internal) == null ? void 0 : _b.currentDate) == null ? void 0 : _c.call(_b)) != null ? _d : /* @__PURE__ */ new Date();
     const { value: response, responseHeaders } = await postJsonToApi2({
       url: `${this.config.baseURL}/models/${this.modelId}:predict`,
       headers: combineHeaders2(
@@ -357,32 +429,34 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
       body,
       failedResponseHandler: googleVertexFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler2(
-        vertexImageResponseSchema
+        googleVertexImageResponseSchema
       ),
       abortSignal,
       fetch: this.config.fetch
     });
     return {
-      images: (_e = (_d = response.predictions) == null ? void 0 : _d.map(
+      images: (_f = (_e = response.predictions) == null ? void 0 : _e.map(
         ({ bytesBase64Encoded }) => bytesBase64Encoded
-      )) != null ? _e : [],
+      )) != null ? _f : [],
       warnings,
       response: {
         timestamp: currentDate,
         modelId: this.modelId,
         headers: responseHeaders
       },
-      providerMetadata: {
-        vertex: {
-          images: (_g = (_f = response.predictions) == null ? void 0 : _f.map((prediction) => {
+      providerMetadata: (() => {
+        var _a2, _b2;
+        const payload = {
+          images: (_b2 = (_a2 = response.predictions) == null ? void 0 : _a2.map((prediction) => {
             const {
               // normalize revised prompt property
               prompt: revisedPrompt
             } = prediction;
             return { ...revisedPrompt != null && { revisedPrompt } };
-          })) != null ? _g : []
-        }
-      }
+          })) != null ? _b2 : []
+        };
+        return { googleVertex: payload, vertex: payload };
+      })()
     };
   }
   async doGenerateGemini({
@@ -453,17 +527,20 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
         "*": [/^https?:\/\/.*$/, /^gs:\/\/.*$/]
       })
     });
+    const userVertexOptions = (_c = providerOptions == null ? void 0 : providerOptions.googleVertex) != null ? _c : providerOptions == null ? void 0 : providerOptions.vertex;
+    const innerVertexOptions = {
+      responseModalities: ["IMAGE"],
+      imageConfig: aspectRatio ? {
+        aspectRatio
+      } : void 0,
+      ...userVertexOptions != null ? userVertexOptions : {}
+    };
     const result = await languageModel.doGenerate({
       prompt: languageModelPrompt,
       seed,
       providerOptions: {
-        vertex: {
-          responseModalities: ["IMAGE"],
-          imageConfig: aspectRatio ? {
-            aspectRatio
-          } : void 0,
-          ...(_c = providerOptions == null ? void 0 : providerOptions.vertex) != null ? _c : {}
-        }
+        googleVertex: innerVertexOptions,
+        vertex: innerVertexOptions
       },
       headers,
       abortSignal
@@ -475,13 +552,15 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
         images.push(convertToBase64(part.data.data));
       }
     }
+    const geminiPayload = {
+      images: images.map(() => ({}))
+    };
     return {
       images,
       warnings,
       providerMetadata: {
-        vertex: {
-          images: images.map(() => ({}))
-        }
+        googleVertex: geminiPayload,
+        vertex: geminiPayload
       },
       response: {
         timestamp: currentDate,
@@ -499,71 +578,15 @@ var GoogleVertexImageModel = class _GoogleVertexImageModel {
 function isGeminiModel(modelId) {
   return modelId.startsWith("gemini-");
 }
-var vertexImageResponseSchema = z4.object({
-  predictions: z4.array(
-    z4.object({
-      bytesBase64Encoded: z4.string(),
-      mimeType: z4.string(),
-      prompt: z4.string().nullish()
+var googleVertexImageResponseSchema = z5.object({
+  predictions: z5.array(
+    z5.object({
+      bytesBase64Encoded: z5.string(),
+      mimeType: z5.string(),
+      prompt: z5.string().nullish()
     })
   ).nullish()
 });
-var googleVertexImageModelOptionsSchema = z4.object({
-  negativePrompt: z4.string().nullish(),
-  personGeneration: z4.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-  safetySetting: z4.enum([
-    "block_low_and_above",
-    "block_medium_and_above",
-    "block_only_high",
-    "block_none"
-  ]).nullish(),
-  addWatermark: z4.boolean().nullish(),
-  storageUri: z4.string().nullish(),
-  sampleImageSize: z4.enum(["1K", "2K"]).nullish(),
-  /**
-   * Configuration for image editing operations
-   */
-  edit: z4.object({
-    /**
-     * An integer that represents the number of sampling steps.
-     * A higher value offers better image quality, a lower value offers better latency.
-     * Try 35 steps to start. If the quality doesn't meet your requirements,
-     * increase the value towards an upper limit of 75.
-     */
-    baseSteps: z4.number().nullish(),
-    // Edit mode options
-    // https://cloud.google.com/vertex-ai/generative-ai/docs/image/edit-insert-objects
-    mode: z4.enum([
-      "EDIT_MODE_INPAINT_INSERTION",
-      "EDIT_MODE_INPAINT_REMOVAL",
-      "EDIT_MODE_OUTPAINT",
-      "EDIT_MODE_CONTROLLED_EDITING",
-      "EDIT_MODE_PRODUCT_IMAGE",
-      "EDIT_MODE_BGSWAP"
-    ]).nullish(),
-    /**
-     * The mask mode to use.
-     * - `MASK_MODE_DEFAULT` - Default value for mask mode.
-     * - `MASK_MODE_USER_PROVIDED` - User provided mask. No segmentation needed.
-     * - `MASK_MODE_DETECTION_BOX` - Mask from detected bounding boxes.
-     * - `MASK_MODE_CLOTHING_AREA` - Masks from segmenting the clothing area with open-vocab segmentation.
-     * - `MASK_MODE_PARSED_PERSON` - Masks from segmenting the person body and clothing using the person-parsing model.
-     */
-    maskMode: z4.enum([
-      "MASK_MODE_DEFAULT",
-      "MASK_MODE_USER_PROVIDED",
-      "MASK_MODE_DETECTION_BOX",
-      "MASK_MODE_CLOTHING_AREA",
-      "MASK_MODE_PARSED_PERSON"
-    ]).nullish(),
-    /**
-     * Optional. A float value between 0 and 1, inclusive, that represents the
-     * percentage of the image width to grow the mask by. Using dilation helps
-     * compensate for imprecise masks. We recommend a value of 0.01.
-     */
-    maskDilation: z4.number().nullish()
-  }).nullish()
-});
 function getBase64Data(file) {
   if (file.type === "url") {
     throw new Error(
@@ -597,13 +620,35 @@ import {
   convertUint8ArrayToBase64 as convertUint8ArrayToBase642,
   createJsonResponseHandler as createJsonResponseHandler3,
   delay,
-  lazySchema,
   parseProviderOptions as parseProviderOptions3,
   postJsonToApi as postJsonToApi3,
-  resolve as resolve3,
-  zodSchema
+  resolve as resolve3
 } from "@ai-sdk/provider-utils";
-import { z as z5 } from "zod/v4";
+import { z as z7 } from "zod/v4";
+
+// src/google-vertex-video-model-options.ts
+import { lazySchema, zodSchema } from "@ai-sdk/provider-utils";
+import { z as z6 } from "zod/v4";
+var googleVertexVideoModelOptionsSchema = lazySchema(
+  () => zodSchema(
+    z6.object({
+      pollIntervalMs: z6.number().positive().nullish(),
+      pollTimeoutMs: z6.number().positive().nullish(),
+      personGeneration: z6.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+      negativePrompt: z6.string().nullish(),
+      generateAudio: z6.boolean().nullish(),
+      gcsOutputDirectory: z6.string().nullish(),
+      referenceImages: z6.array(
+        z6.object({
+          bytesBase64Encoded: z6.string().nullish(),
+          gcsUri: z6.string().nullish()
+        })
+      ).nullish()
+    }).passthrough()
+  )
+);
+
+// src/google-vertex-video-model.ts
 var GoogleVertexVideoModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -617,10 +662,14 @@ var GoogleVertexVideoModel = class {
     return 4;
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f;
+    var _a, _b, _c, _d, _e, _f, _g;
     const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
     const warnings = [];
-    const vertexOptions = await parseProviderOptions3({
+    const googleVertexOptions = (_d = await parseProviderOptions3({
+      provider: "googleVertex",
+      providerOptions: options.providerOptions,
+      schema: googleVertexVideoModelOptionsSchema
+    })) != null ? _d : await parseProviderOptions3({
       provider: "vertex",
       providerOptions: options.providerOptions,
       schema: googleVertexVideoModelOptionsSchema
@@ -645,8 +694,8 @@ var GoogleVertexVideoModel = class {
         };
       }
     }
-    if ((vertexOptions == null ? void 0 : vertexOptions.referenceImages) != null) {
-      instance.referenceImages = vertexOptions.referenceImages;
+    if ((googleVertexOptions == null ? void 0 : googleVertexOptions.referenceImages) != null) {
+      instance.referenceImages = googleVertexOptions.referenceImages;
     }
     const parameters = {
       sampleCount: options.n
@@ -668,8 +717,8 @@ var GoogleVertexVideoModel = class {
     if (options.seed) {
       parameters.seed = options.seed;
     }
-    if (vertexOptions != null) {
-      const opts = vertexOptions;
+    if (googleVertexOptions != null) {
+      const opts = googleVertexOptions;
       if (opts.personGeneration !== void 0 && opts.personGeneration !== null) {
         parameters.personGeneration = opts.personGeneration;
       }
@@ -707,7 +756,7 @@ var GoogleVertexVideoModel = class {
         parameters
       },
       successfulResponseHandler: createJsonResponseHandler3(
-        vertexOperationSchema
+        googleVertexOperationSchema
       ),
       failedResponseHandler: googleVertexFailedResponseHandler,
       abortSignal: options.abortSignal,
@@ -720,8 +769,8 @@ var GoogleVertexVideoModel = class {
         message: "No operation name returned from API"
       });
     }
-    const pollIntervalMs = (_d = vertexOptions == null ? void 0 : vertexOptions.pollIntervalMs) != null ? _d : 1e4;
-    const pollTimeoutMs = (_e = vertexOptions == null ? void 0 : vertexOptions.pollTimeoutMs) != null ? _e : 6e5;
+    const pollIntervalMs = (_e = googleVertexOptions == null ? void 0 : googleVertexOptions.pollIntervalMs) != null ? _e : 1e4;
+    const pollTimeoutMs = (_f = googleVertexOptions == null ? void 0 : googleVertexOptions.pollTimeoutMs) != null ? _f : 6e5;
     const startTime = Date.now();
     let finalOperation = operation;
     let responseHeaders;
@@ -733,7 +782,7 @@ var GoogleVertexVideoModel = class {
         });
       }
       await delay(pollIntervalMs);
-      if ((_f = options.abortSignal) == null ? void 0 : _f.aborted) {
+      if ((_g = options.abortSignal) == null ? void 0 : _g.aborted) {
         throw new AISDKError({
           name: "VERTEX_VIDEO_GENERATION_ABORTED",
           message: "Video generation request was aborted"
@@ -749,7 +798,7 @@ var GoogleVertexVideoModel = class {
           operationName
         },
         successfulResponseHandler: createJsonResponseHandler3(
-          vertexOperationSchema
+          googleVertexOperationSchema
         ),
         failedResponseHandler: googleVertexFailedResponseHandler,
         abortSignal: options.abortSignal,
@@ -809,53 +858,39 @@ var GoogleVertexVideoModel = class {
         modelId: this.modelId,
         headers: responseHeaders
       },
-      providerMetadata: {
-        "google-vertex": {
-          videos: videoMetadata
-        }
-      }
+      providerMetadata: /* @__PURE__ */ (() => {
+        const payload = { videos: videoMetadata };
+        return {
+          googleVertex: payload,
+          // Legacy keys preserved for backward compatibility.
+          "google-vertex": payload,
+          vertex: payload
+        };
+      })()
     };
   }
 };
-var vertexOperationSchema = z5.object({
-  name: z5.string().nullish(),
-  done: z5.boolean().nullish(),
-  error: z5.object({
-    code: z5.number().nullish(),
-    message: z5.string(),
-    status: z5.string().nullish()
+var googleVertexOperationSchema = z7.object({
+  name: z7.string().nullish(),
+  done: z7.boolean().nullish(),
+  error: z7.object({
+    code: z7.number().nullish(),
+    message: z7.string(),
+    status: z7.string().nullish()
   }).nullish(),
-  response: z5.object({
-    videos: z5.array(
-      z5.object({
-        bytesBase64Encoded: z5.string().nullish(),
-        gcsUri: z5.string().nullish(),
-        mimeType: z5.string().nullish()
+  response: z7.object({
+    videos: z7.array(
+      z7.object({
+        bytesBase64Encoded: z7.string().nullish(),
+        gcsUri: z7.string().nullish(),
+        mimeType: z7.string().nullish()
       })
     ).nullish(),
-    raiMediaFilteredCount: z5.number().nullish()
+    raiMediaFilteredCount: z7.number().nullish()
   }).nullish()
 });
-var googleVertexVideoModelOptionsSchema = lazySchema(
-  () => zodSchema(
-    z5.object({
-      pollIntervalMs: z5.number().positive().nullish(),
-      pollTimeoutMs: z5.number().positive().nullish(),
-      personGeneration: z5.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-      negativePrompt: z5.string().nullish(),
-      generateAudio: z5.boolean().nullish(),
-      gcsOutputDirectory: z5.string().nullish(),
-      referenceImages: z5.array(
-        z5.object({
-          bytesBase64Encoded: z5.string().nullish(),
-          gcsUri: z5.string().nullish()
-        })
-      ).nullish()
-    }).passthrough()
-  )
-);
 
-// src/google-vertex-provider.ts
+// src/google-vertex-provider-base.ts
 var EXPRESS_MODE_BASE_URL = "https://aiplatform.googleapis.com/v1/publishers/google";
 function createExpressModeFetch(apiKey, customFetch) {
   return async (url, init) => {
@@ -869,18 +904,18 @@ function createExpressModeFetch(apiKey, customFetch) {
     return (customFetch != null ? customFetch : fetch)(url.toString(), modifiedInit);
   };
 }
-function createVertex(options = {}) {
+function createGoogleVertex(options = {}) {
   const apiKey = loadOptionalSetting({
     settingValue: options.apiKey,
     environmentVariableName: "GOOGLE_VERTEX_API_KEY"
   });
-  const loadVertexProject = () => loadSetting({
+  const loadGoogleVertexProject = () => loadSetting({
     settingValue: options.project,
     settingName: "project",
     environmentVariableName: "GOOGLE_VERTEX_PROJECT",
     description: "Google Vertex project"
   });
-  const loadVertexLocation = () => loadSetting({
+  const loadGoogleVertexLocation = () => loadSetting({
     settingValue: options.location,
     settingName: "location",
     environmentVariableName: "GOOGLE_VERTEX_LOCATION",
@@ -891,8 +926,8 @@ function createVertex(options = {}) {
     if (apiKey) {
       return (_a = withoutTrailingSlash(options.baseURL)) != null ? _a : EXPRESS_MODE_BASE_URL;
     }
-    const region = loadVertexLocation();
-    const project = loadVertexProject();
+    const region = loadGoogleVertexLocation();
+    const project = loadGoogleVertexProject();
     const baseHost = `${region === "global" ? "" : region + "-"}aiplatform.googleapis.com`;
     return (_b = withoutTrailingSlash(options.baseURL)) != null ? _b : `https://${baseHost}/v1beta1/projects/${project}/locations/${region}/publishers/google`;
   };
@@ -962,16 +997,16 @@ function createVertex(options = {}) {
   return provider;
 }
 
-// src/google-vertex-provider-node.ts
-function createVertex2(options = {}) {
+// src/google-vertex-provider.ts
+function createGoogleVertex2(options = {}) {
   const apiKey = loadOptionalSetting2({
     settingValue: options.apiKey,
     environmentVariableName: "GOOGLE_VERTEX_API_KEY"
   });
   if (apiKey) {
-    return createVertex(options);
+    return createGoogleVertex(options);
   }
-  return createVertex({
+  return createGoogleVertex({
     ...options,
     headers: async () => ({
       Authorization: `Bearer ${await generateAuthToken(
@@ -981,10 +1016,12 @@ function createVertex2(options = {}) {
     })
   });
 }
-var vertex = createVertex2();
+var googleVertex = createGoogleVertex2();
 export {
   VERSION,
-  createVertex2 as createVertex,
-  vertex
+  createGoogleVertex2 as createGoogleVertex,
+  createGoogleVertex2 as createVertex,
+  googleVertex,
+  googleVertex as vertex
 };
 //# sourceMappingURL=index.js.map