@ai-sdk/google-vertex 5.0.0-beta.4 → 5.0.0-beta.40

package/src/google-vertex-embedding-model.ts

@@ -1,5 +1,5 @@
 import {
-  EmbeddingModelV3,
+  EmbeddingModelV4,
   TooManyEmbeddingValuesForCallError,
 } from '@ai-sdk/provider';
 import {
@@ -17,8 +17,8 @@ import {
 } from './google-vertex-embedding-options';
 import { GoogleVertexConfig } from './google-vertex-config';
 
-export class GoogleVertexEmbeddingModel implements EmbeddingModelV3 {
-  readonly specificationVersion = 'v3';
+export class GoogleVertexEmbeddingModel implements EmbeddingModelV4 {
+  readonly specificationVersion = 'v4';
   readonly modelId: GoogleVertexEmbeddingModelId;
   readonly maxEmbeddingsPerCall = 2048;
   readonly supportsParallelCalls = true;
@@ -42,8 +42,8 @@ export class GoogleVertexEmbeddingModel implements EmbeddingModelV3 {
     headers,
     abortSignal,
     providerOptions,
-  }: Parameters<EmbeddingModelV3['doEmbed']>[0]): Promise<
-    Awaited<ReturnType<EmbeddingModelV3['doEmbed']>>
+  }: Parameters<EmbeddingModelV4['doEmbed']>[0]): Promise<
+    Awaited<ReturnType<EmbeddingModelV4['doEmbed']>>
   > {
     let googleOptions = await parseProviderOptions({
       provider: 'vertex',

package/src/google-vertex-embedding-options.ts

@@ -10,6 +10,8 @@ export type GoogleVertexEmbeddingModelId =
   | 'text-multilingual-embedding-002'
   | 'text-embedding-004'
   | 'text-embedding-005'
+  | 'gemini-embedding-001'
+  | 'gemini-embedding-2-preview'
   | (string & {});
 
 export const googleVertexEmbeddingModelOptions = z.object({

package/src/google-vertex-image-model.ts

@@ -1,10 +1,10 @@
 import type { GoogleLanguageModelOptions } from '@ai-sdk/google';
 import { GoogleGenerativeAILanguageModel } from '@ai-sdk/google/internal';
 import {
-  ImageModelV3,
-  ImageModelV3File,
-  LanguageModelV3Prompt,
-  SharedV3Warning,
+  ImageModelV4,
+  ImageModelV4File,
+  LanguageModelV4Prompt,
+  SharedV4Warning,
 } from '@ai-sdk/provider';
 import {
   Resolvable,
@@ -33,8 +33,8 @@ interface GoogleVertexImageModelConfig {
 }
 
 // https://cloud.google.com/vertex-ai/generative-ai/docs/image/generate-images
-export class GoogleVertexImageModel implements ImageModelV3 {
-  readonly specificationVersion = 'v3';
+export class GoogleVertexImageModel implements ImageModelV4 {
+  readonly specificationVersion = 'v4';
 
   get maxImagesPerCall(): number {
     if (isGeminiModel(this.modelId)) {
@@ -53,8 +53,8 @@ export class GoogleVertexImageModel implements ImageModelV3 {
   ) {}
 
   async doGenerate(
-    options: Parameters<ImageModelV3['doGenerate']>[0],
-  ): Promise<Awaited<ReturnType<ImageModelV3['doGenerate']>>> {
+    options: Parameters<ImageModelV4['doGenerate']>[0],
+  ): Promise<Awaited<ReturnType<ImageModelV4['doGenerate']>>> {
     if (isGeminiModel(this.modelId)) {
       return this.doGenerateGemini(options);
     }
@@ -72,10 +72,10 @@ export class GoogleVertexImageModel implements ImageModelV3 {
     abortSignal,
     files,
     mask,
-  }: Parameters<ImageModelV3['doGenerate']>[0]): Promise<
-    Awaited<ReturnType<ImageModelV3['doGenerate']>>
+  }: Parameters<ImageModelV4['doGenerate']>[0]): Promise<
+    Awaited<ReturnType<ImageModelV4['doGenerate']>>
   > {
-    const warnings: Array<SharedV3Warning> = [];
+    const warnings: Array<SharedV4Warning> = [];
 
     if (size != null) {
       warnings.push({
@@ -212,10 +212,10 @@ export class GoogleVertexImageModel implements ImageModelV3 {
     abortSignal,
     files,
     mask,
-  }: Parameters<ImageModelV3['doGenerate']>[0]): Promise<
-    Awaited<ReturnType<ImageModelV3['doGenerate']>>
+  }: Parameters<ImageModelV4['doGenerate']>[0]): Promise<
+    Awaited<ReturnType<ImageModelV4['doGenerate']>>
   > {
-    const warnings: Array<SharedV3Warning> = [];
+    const warnings: Array<SharedV4Warning> = [];
 
     if (mask != null) {
       throw new Error(
@@ -268,7 +268,7 @@ export class GoogleVertexImageModel implements ImageModelV3 {
       }
     }
 
-    const languageModelPrompt: LanguageModelV3Prompt = [
+    const languageModelPrompt: LanguageModelV4Prompt = [
       { role: 'user', content: userContent },
     ];
 
@@ -433,9 +433,9 @@ export type GoogleVertexImageModelOptions = z.infer<
 >;
 
 /**
- * Helper to convert ImageModelV3File data to base64 string
+ * Helper to convert ImageModelV4File data to base64 string
  */
-function getBase64Data(file: ImageModelV3File): string {
+function getBase64Data(file: ImageModelV4File): string {
   if (file.type === 'url') {
     throw new Error(
       'URL-based images are not supported for Google Vertex image editing. Please provide the image data directly.',

package/src/google-vertex-provider-node.ts

@@ -7,8 +7,7 @@ import {
   GoogleVertexProviderSettings as GoogleVertexProviderSettingsOriginal,
 } from './google-vertex-provider';
 
-export interface GoogleVertexProviderSettings
-  extends GoogleVertexProviderSettingsOriginal {
+export interface GoogleVertexProviderSettings extends GoogleVertexProviderSettingsOriginal {
   /**
    * Optional. The Authentication options provided by google-auth-library.
    * Complete list of authentication options is documented in the

package/src/google-vertex-provider.ts

@@ -1,9 +1,9 @@
 import { GoogleGenerativeAILanguageModel } from '@ai-sdk/google/internal';
 import {
-  Experimental_VideoModelV3,
-  ImageModelV3,
-  LanguageModelV3,
-  ProviderV3,
+  Experimental_VideoModelV4,
+  ImageModelV4,
+  LanguageModelV4,
+  ProviderV4,
 } from '@ai-sdk/provider';
 import {
   FetchFunction,
@@ -47,23 +47,23 @@ function createExpressModeFetch(
   };
 }
 
-export interface GoogleVertexProvider extends ProviderV3 {
+export interface GoogleVertexProvider extends ProviderV4 {
   /**
    * Creates a model for text generation.
    */
-  (modelId: GoogleVertexModelId): LanguageModelV3;
+  (modelId: GoogleVertexModelId): LanguageModelV4;
 
-  languageModel: (modelId: GoogleVertexModelId) => LanguageModelV3;
+  languageModel: (modelId: GoogleVertexModelId) => LanguageModelV4;
 
   /**
    * Creates a model for image generation.
    */
-  image(modelId: GoogleVertexImageModelId): ImageModelV3;
+  image(modelId: GoogleVertexImageModelId): ImageModelV4;
 
   /**
    * Creates a model for image generation.
    */
-  imageModel(modelId: GoogleVertexImageModelId): ImageModelV3;
+  imageModel(modelId: GoogleVertexImageModelId): ImageModelV4;
 
   tools: typeof googleVertexTools;
 
@@ -77,12 +77,12 @@ export interface GoogleVertexProvider extends ProviderV3 {
   /**
    * Creates a model for video generation.
    */
-  video(modelId: GoogleVertexVideoModelId): Experimental_VideoModelV3;
+  video(modelId: GoogleVertexVideoModelId): Experimental_VideoModelV4;
 
   /**
    * Creates a model for video generation.
    */
-  videoModel(modelId: GoogleVertexVideoModelId): Experimental_VideoModelV3;
+  videoModel(modelId: GoogleVertexVideoModelId): Experimental_VideoModelV4;
 }
 
 export interface GoogleVertexProviderSettings {
@@ -231,7 +231,7 @@ export function createVertex(
     return createChatModel(modelId);
   };
 
-  provider.specificationVersion = 'v3' as const;
+  provider.specificationVersion = 'v4' as const;
   provider.languageModel = createChatModel;
   provider.embeddingModel = createEmbeddingModel;
   provider.textEmbeddingModel = createEmbeddingModel;

package/src/google-vertex-video-model.ts

@@ -1,7 +1,7 @@
 import {
   AISDKError,
-  type Experimental_VideoModelV3,
-  type SharedV3Warning,
+  type Experimental_VideoModelV4,
+  type SharedV4Warning,
 } from '@ai-sdk/provider';
 import {
   combineHeaders,
@@ -53,8 +53,8 @@ interface GoogleVertexVideoModelConfig {
   };
 }
 
-export class GoogleVertexVideoModel implements Experimental_VideoModelV3 {
-  readonly specificationVersion = 'v3';
+export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
+  readonly specificationVersion = 'v4';
 
   get provider(): string {
     return this.config.provider;
@@ -71,10 +71,10 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV3 {
   ) {}
 
   async doGenerate(
-    options: Parameters<Experimental_VideoModelV3['doGenerate']>[0],
-  ): Promise<Awaited<ReturnType<Experimental_VideoModelV3['doGenerate']>>> {
+    options: Parameters<Experimental_VideoModelV4['doGenerate']>[0],
+  ): Promise<Awaited<ReturnType<Experimental_VideoModelV4['doGenerate']>>> {
     const currentDate = this.config._internal?.currentDate?.() ?? new Date();
-    const warnings: SharedV3Warning[] = [];
+    const warnings: SharedV4Warning[] = [];
 
     const vertexOptions = (await parseProviderOptions({
       provider: 'vertex',

package/src/maas/edge/google-vertex-maas-provider-edge.ts

@@ -0,0 +1,65 @@
+import { FetchFunction, resolve } from '@ai-sdk/provider-utils';
+import {
+  generateAuthToken,
+  GoogleCredentials,
+} from '../../edge/google-vertex-auth-edge';
+import {
+  createVertexMaas as createVertexMaasOriginal,
+  GoogleVertexMaasProvider,
+  GoogleVertexMaasProviderSettings as GoogleVertexMaasProviderSettingsOriginal,
+} from '../google-vertex-maas-provider';
+
+export type { GoogleVertexMaasProvider };
+
+export interface GoogleVertexMaasProviderSettings extends GoogleVertexMaasProviderSettingsOriginal {
+  /**
+   * Optional. The Google credentials for the Google Cloud service account. If
+   * not provided, the Google Vertex provider will use environment variables to
+   * load the credentials.
+   */
+  googleCredentials?: GoogleCredentials;
+}
+
+/**
+ * Create a Google Vertex AI MaaS (Model as a Service) provider instance for Edge runtimes.
+ * Uses the OpenAI-compatible Chat Completions API for partner and open models.
+ * Automatically handles Google Cloud authentication.
+ *
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
+ */
+export function createVertexMaas(
+  options: GoogleVertexMaasProviderSettings = {},
+): GoogleVertexMaasProvider {
+  // Create a custom fetch wrapper that adds auth headers
+  const customFetch: FetchFunction = async (url, init) => {
+    const token = await generateAuthToken(options.googleCredentials);
+    const resolvedHeaders = await resolve(options.headers);
+    const authHeaders = {
+      ...resolvedHeaders,
+      Authorization: `Bearer ${token}`,
+    };
+
+    // Merge auth headers with existing headers from init
+    const fetchInit = {
+      ...init,
+      headers: {
+        ...init?.headers,
+        ...authHeaders,
+      },
+    };
+
+    // Call the original fetch or user's custom fetch
+    return (options.fetch ?? fetch)(url, fetchInit);
+  };
+
+  return createVertexMaasOriginal({
+    ...options,
+    fetch: customFetch,
+    headers: undefined, // Don't pass headers, we handle them in fetch
+  });
+}
+
+/**
+ * Default Google Vertex AI MaaS provider instance for Edge runtimes.
+ */
+export const vertexMaas = createVertexMaas();

package/src/maas/edge/index.ts

@@ -0,0 +1,9 @@
+export {
+  createVertexMaas,
+  vertexMaas,
+} from './google-vertex-maas-provider-edge';
+export type {
+  GoogleVertexMaasProvider,
+  GoogleVertexMaasProviderSettings,
+} from './google-vertex-maas-provider-edge';
+export type { GoogleVertexMaasModelId } from '../google-vertex-maas-options';

package/src/maas/google-vertex-maas-options.ts

@@ -0,0 +1,15 @@
+// https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
+export type GoogleVertexMaasModelId =
+  | 'deepseek-ai/deepseek-r1-0528-maas'
+  | 'deepseek-ai/deepseek-v3.1-maas'
+  | 'deepseek-ai/deepseek-v3.2-maas'
+  | 'openai/gpt-oss-120b-maas'
+  | 'openai/gpt-oss-20b-maas'
+  | 'meta/llama-4-maverick-17b-128e-instruct-maas'
+  | 'meta/llama-4-scout-17b-16e-instruct-maas'
+  | 'minimax/minimax-m2-maas'
+  | 'qwen/qwen3-coder-480b-a35b-instruct-maas'
+  | 'qwen/qwen3-next-80b-a3b-instruct-maas'
+  | 'qwen/qwen3-next-80b-a3b-thinking-maas'
+  | 'moonshotai/kimi-k2-thinking-maas'
+  | (string & {});

package/src/maas/google-vertex-maas-provider-node.ts

@@ -0,0 +1,64 @@
+import { FetchFunction, resolve } from '@ai-sdk/provider-utils';
+import { GoogleAuthOptions } from 'google-auth-library';
+import { generateAuthToken } from '../google-vertex-auth-google-auth-library';
+import {
+  createVertexMaas as createVertexMaasOriginal,
+  GoogleVertexMaasProvider,
+  GoogleVertexMaasProviderSettings as GoogleVertexMaasProviderSettingsOriginal,
+} from './google-vertex-maas-provider';
+
+export type { GoogleVertexMaasProvider };
+
+export interface GoogleVertexMaasProviderSettings extends GoogleVertexMaasProviderSettingsOriginal {
+  /**
+   * Optional. The Authentication options provided by google-auth-library.
+   * Complete list of authentication options is documented in the
+   * GoogleAuthOptions interface:
+   * https://github.com/googleapis/google-auth-library-nodejs/blob/main/src/auth/googleauth.ts.
+   */
+  googleAuthOptions?: GoogleAuthOptions;
+}
+
+/**
+ * Create a Google Vertex AI MaaS (Model as a Service) provider instance for Node.js.
+ * Uses the OpenAI-compatible Chat Completions API for partner and open models.
+ * Automatically handles Google Cloud authentication.
+ *
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
+ */
+export function createVertexMaas(
+  options: GoogleVertexMaasProviderSettings = {},
+): GoogleVertexMaasProvider {
+  // Create a custom fetch wrapper that adds auth headers
+  const customFetch: FetchFunction = async (url, init) => {
+    const token = await generateAuthToken(options.googleAuthOptions);
+    const resolvedHeaders = await resolve(options.headers);
+    const authHeaders = {
+      ...resolvedHeaders,
+      Authorization: `Bearer ${token}`,
+    };
+
+    // Merge auth headers with existing headers from init
+    const fetchInit = {
+      ...init,
+      headers: {
+        ...init?.headers,
+        ...authHeaders,
+      },
+    };
+
+    // Call the original fetch or user's custom fetch
+    return (options.fetch ?? fetch)(url, fetchInit);
+  };
+
+  return createVertexMaasOriginal({
+    ...options,
+    fetch: customFetch,
+    headers: undefined, // Don't pass headers, we handle them in fetch
+  });
+}
+
+/**
+ * Default Google Vertex AI MaaS provider instance for Node.js.
+ */
+export const vertexMaas = createVertexMaas();

package/src/maas/google-vertex-maas-provider.ts

@@ -0,0 +1,111 @@
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
+import type { OpenAICompatibleProvider } from '@ai-sdk/openai-compatible';
+import {
+  FetchFunction,
+  loadOptionalSetting,
+  loadSetting,
+  Resolvable,
+  withoutTrailingSlash,
+} from '@ai-sdk/provider-utils';
+import type { GoogleVertexMaasModelId } from './google-vertex-maas-options';
+
+export interface GoogleVertexMaasProvider extends OpenAICompatibleProvider<
+  GoogleVertexMaasModelId,
+  string,
+  string,
+  string
+> {}
+
+export interface GoogleVertexMaasProviderSettings {
+  /**
+   * Google Cloud project ID. Defaults to the value of the `GOOGLE_VERTEX_PROJECT` environment variable.
+   */
+  project?: string;
+
+  /**
+   * Google Cloud location/region. Defaults to the value of the `GOOGLE_VERTEX_LOCATION` environment variable.
+   * Use 'global' for the global endpoint.
+   */
+  location?: string;
+
+  /**
+   * Base URL for the API calls. If not provided, will be constructed from project and location.
+   */
+  baseURL?: string;
+
+  /**
+   * Headers to use for requests. Can be:
+   * - A headers object
+   * - A Promise that resolves to a headers object
+   * - A function that returns a headers object
+   * - A function that returns a Promise of a headers object
+   */
+  headers?: Resolvable<Record<string, string | undefined>>;
+
+  /**
+   * Custom fetch implementation. You can use it as a middleware to intercept requests,
+   * or to provide a custom fetch implementation for e.g. testing.
+   */
+  fetch?: FetchFunction;
+}
+
+/**
+ * Create a Google Vertex AI MaaS (Model as a Service) provider instance.
+ * Uses the OpenAI-compatible Chat Completions API for partner and open models.
+ *
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
+ */
+export function createVertexMaas(
+  options: GoogleVertexMaasProviderSettings = {},
+): GoogleVertexMaasProvider {
+  // Lazy-load settings to support loading from environment variables at runtime
+  const loadLocation = () =>
+    loadOptionalSetting({
+      settingValue: options.location,
+      environmentVariableName: 'GOOGLE_VERTEX_LOCATION',
+    });
+
+  const loadProject = () =>
+    loadSetting({
+      settingValue: options.project,
+      settingName: 'project',
+      environmentVariableName: 'GOOGLE_VERTEX_PROJECT',
+      description: 'Google Vertex project',
+    });
+
+  // Construct base URL: https://aiplatform.googleapis.com/v1/projects/{project}/locations/{location}/endpoints/openapi
+  const constructBaseURL = () => {
+    const projectId = loadProject();
+    const location = loadLocation() ?? 'global';
+
+    return `https://aiplatform.googleapis.com/v1/projects/${projectId}/locations/${location}/endpoints/openapi`;
+  };
+
+  const loadBaseURL = () =>
+    withoutTrailingSlash(options.baseURL ?? '') || constructBaseURL();
+
+  let cachedProvider: GoogleVertexMaasProvider | undefined;
+  const getProvider = () =>
+    (cachedProvider ??= createOpenAICompatible({
+      name: 'vertex.maas',
+      baseURL: loadBaseURL(),
+      fetch: options.fetch,
+    }));
+
+  const provider = (modelId: GoogleVertexMaasModelId) => getProvider()(modelId);
+
+  provider.specificationVersion = 'v4' as const;
+  provider.languageModel = (modelId: GoogleVertexMaasModelId) =>
+    getProvider().languageModel(modelId);
+  provider.chatModel = (modelId: GoogleVertexMaasModelId) =>
+    getProvider().chatModel(modelId);
+  provider.completionModel = (modelId: string) =>
+    getProvider().completionModel(modelId);
+  provider.embeddingModel = (modelId: string) =>
+    getProvider().embeddingModel(modelId);
+  provider.textEmbeddingModel = (modelId: string) =>
+    getProvider().textEmbeddingModel(modelId);
+  provider.imageModel = (modelId: string) => getProvider().imageModel(modelId);
+
+  return provider as GoogleVertexMaasProvider;
+}

package/src/maas/index.ts

@@ -0,0 +1,9 @@
+export {
+  createVertexMaas,
+  vertexMaas,
+} from './google-vertex-maas-provider-node';
+export type {
+  GoogleVertexMaasProvider,
+  GoogleVertexMaasProviderSettings,
+} from './google-vertex-maas-provider-node';
+export type { GoogleVertexMaasModelId } from './google-vertex-maas-options';