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

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @ai-sdk/google-vertex
 
+## 5.0.0-beta.64
+
+### Patch Changes
+
+- e59c955: feat(vertex): add grok models to vertex provider
+- Updated dependencies [e59c955]
+  - @ai-sdk/openai-compatible@3.0.0-beta.36
+
 ## 5.0.0-beta.63
 
 ### Major Changes

package/dist/anthropic/edge/index.js

@@ -10,7 +10,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.0-beta.63" : "0.0.0-test";
+var VERSION = true ? "5.0.0-beta.64" : "0.0.0-test";
 
 // src/edge/google-vertex-auth-edge.ts
 var loadCredentials = async () => {

package/dist/edge/index.js

@@ -14,7 +14,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.0-beta.63" : "0.0.0-test";
+var VERSION = true ? "5.0.0-beta.64" : "0.0.0-test";
 
 // src/google-vertex-embedding-model.ts
 import {

package/dist/index.js

@@ -35,7 +35,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.0-beta.63" : "0.0.0-test";
+var VERSION = true ? "5.0.0-beta.64" : "0.0.0-test";
 
 // src/google-vertex-embedding-model.ts
 import {

package/dist/maas/edge/index.js

@@ -10,7 +10,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.0-beta.63" : "0.0.0-test";
+var VERSION = true ? "5.0.0-beta.64" : "0.0.0-test";
 
 // src/edge/google-vertex-auth-edge.ts
 var loadCredentials = async () => {

package/dist/xai/edge/index.d.ts

@@ -0,0 +1,92 @@
+import { ProviderV4, LanguageModelV4 } from '@ai-sdk/provider';
+import { Resolvable, FetchFunction } from '@ai-sdk/provider-utils';
+
+interface GoogleCredentials {
+    /**
+     * The client email for the Google Cloud service account. Defaults to the
+     * value of the `GOOGLE_CLIENT_EMAIL` environment variable.
+     */
+    clientEmail: string;
+    /**
+     * The private key for the Google Cloud service account. Defaults to the
+     * value of the `GOOGLE_PRIVATE_KEY` environment variable.
+     */
+    privateKey: string;
+    /**
+     * Optional. The private key ID for the Google Cloud service account. Defaults
+     * to the value of the `GOOGLE_PRIVATE_KEY_ID` environment variable.
+     */
+    privateKeyId?: string;
+}
+
+type GoogleVertexXaiModelId = 'xai/grok-4.20-reasoning' | 'xai/grok-4.20-non-reasoning' | 'xai/grok-4.1-fast-reasoning' | 'xai/grok-4.1-fast-non-reasoning' | (string & {});
+
+interface GoogleVertexXaiProvider extends ProviderV4 {
+    /**
+     * Creates a model for text generation.
+     */
+    (modelId: GoogleVertexXaiModelId): LanguageModelV4;
+    /**
+     * Creates a model for text generation.
+     */
+    languageModel(modelId: GoogleVertexXaiModelId): LanguageModelV4;
+    /**
+     * Creates a chat model for text generation.
+     */
+    chatModel(modelId: GoogleVertexXaiModelId): LanguageModelV4;
+    /**
+     * @deprecated Use `embeddingModel` instead.
+     */
+    textEmbeddingModel(modelId: string): never;
+}
+interface GoogleVertexXaiProviderSettings$1 {
+    /**
+     * 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;
+}
+
+interface GoogleVertexXaiProviderSettings extends GoogleVertexXaiProviderSettings$1 {
+    /**
+     * 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 xAI provider instance for Edge runtimes.
+ * Uses the OpenAI-compatible Chat Completions API for Grok partner models.
+ * Automatically handles Google Cloud authentication.
+ *
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/partner-models/grok
+ */
+declare function createGoogleVertexXai(options?: GoogleVertexXaiProviderSettings): GoogleVertexXaiProvider;
+/**
+ * Default Google Vertex AI xAI provider instance for Edge runtimes.
+ */
+declare const googleVertexXai: GoogleVertexXaiProvider;
+
+export { type GoogleVertexXaiModelId, type GoogleVertexXaiProvider, type GoogleVertexXaiProviderSettings, createGoogleVertexXai, googleVertexXai };

package/dist/xai/edge/index.js

@@ -0,0 +1,259 @@
+// src/xai/edge/google-vertex-xai-provider-edge.ts
+import { resolve } from "@ai-sdk/provider-utils";
+
+// src/edge/google-vertex-auth-edge.ts
+import {
+  loadOptionalSetting,
+  loadSetting,
+  withUserAgentSuffix,
+  getRuntimeEnvironmentUserAgent
+} from "@ai-sdk/provider-utils";
+
+// src/version.ts
+var VERSION = true ? "5.0.0-beta.64" : "0.0.0-test";
+
+// src/edge/google-vertex-auth-edge.ts
+var loadCredentials = async () => {
+  try {
+    return {
+      clientEmail: loadSetting({
+        settingValue: void 0,
+        settingName: "clientEmail",
+        environmentVariableName: "GOOGLE_CLIENT_EMAIL",
+        description: "Google client email"
+      }),
+      privateKey: loadSetting({
+        settingValue: void 0,
+        settingName: "privateKey",
+        environmentVariableName: "GOOGLE_PRIVATE_KEY",
+        description: "Google private key"
+      }),
+      privateKeyId: loadOptionalSetting({
+        settingValue: void 0,
+        environmentVariableName: "GOOGLE_PRIVATE_KEY_ID"
+      })
+    };
+  } catch (error) {
+    throw new Error(`Failed to load Google credentials: ${error.message}`);
+  }
+};
+var base64url = (str) => {
+  return btoa(str).replace(/\+/g, "-").replace(/\//g, "_").replace(/=/g, "");
+};
+var importPrivateKey = async (pemKey) => {
+  const pemHeader = "-----BEGIN PRIVATE KEY-----";
+  const pemFooter = "-----END PRIVATE KEY-----";
+  const pemContents = pemKey.replace(pemHeader, "").replace(pemFooter, "").replace(/\s/g, "");
+  const binaryString = atob(pemContents);
+  const binaryData = new Uint8Array(binaryString.length);
+  for (let i = 0; i < binaryString.length; i++) {
+    binaryData[i] = binaryString.charCodeAt(i);
+  }
+  return await crypto.subtle.importKey(
+    "pkcs8",
+    binaryData,
+    { name: "RSASSA-PKCS1-v1_5", hash: "SHA-256" },
+    true,
+    ["sign"]
+  );
+};
+var buildJwt = async (credentials) => {
+  const now = Math.floor(Date.now() / 1e3);
+  const header = {
+    alg: "RS256",
+    typ: "JWT"
+  };
+  if (credentials.privateKeyId) {
+    header.kid = credentials.privateKeyId;
+  }
+  const payload = {
+    iss: credentials.clientEmail,
+    scope: "https://www.googleapis.com/auth/cloud-platform",
+    aud: "https://oauth2.googleapis.com/token",
+    exp: now + 3600,
+    iat: now
+  };
+  const privateKey = await importPrivateKey(credentials.privateKey);
+  const signingInput = `${base64url(JSON.stringify(header))}.${base64url(
+    JSON.stringify(payload)
+  )}`;
+  const encoder = new TextEncoder();
+  const data = encoder.encode(signingInput);
+  const signature = await crypto.subtle.sign(
+    "RSASSA-PKCS1-v1_5",
+    privateKey,
+    data
+  );
+  const signatureBase64 = base64url(
+    String.fromCharCode(...new Uint8Array(signature))
+  );
+  return `${base64url(JSON.stringify(header))}.${base64url(
+    JSON.stringify(payload)
+  )}.${signatureBase64}`;
+};
+async function generateAuthToken(credentials) {
+  try {
+    const creds = credentials || await loadCredentials();
+    const jwt = await buildJwt(creds);
+    const response = await fetch("https://oauth2.googleapis.com/token", {
+      method: "POST",
+      headers: withUserAgentSuffix(
+        { "Content-Type": "application/x-www-form-urlencoded" },
+        `ai-sdk/google-vertex/${VERSION}`,
+        getRuntimeEnvironmentUserAgent()
+      ),
+      body: new URLSearchParams({
+        grant_type: "urn:ietf:params:oauth:grant-type:jwt-bearer",
+        assertion: jwt
+      })
+    });
+    if (!response.ok) {
+      throw new Error(`Token request failed: ${response.statusText}`);
+    }
+    const data = await response.json();
+    return data.access_token;
+  } catch (error) {
+    throw error;
+  }
+}
+
+// src/xai/google-vertex-xai-provider.ts
+import {
+  NoSuchModelError
+} from "@ai-sdk/provider";
+import {
+  createOpenAICompatible
+} from "@ai-sdk/openai-compatible";
+import {
+  loadOptionalSetting as loadOptionalSetting2,
+  loadSetting as loadSetting2,
+  withoutTrailingSlash
+} from "@ai-sdk/provider-utils";
+function convertGoogleVertexXaiUsage(usage) {
+  var _a, _b, _c, _d, _e, _f;
+  if (usage == null) {
+    return {
+      inputTokens: {
+        total: void 0,
+        noCache: void 0,
+        cacheRead: void 0,
+        cacheWrite: void 0
+      },
+      outputTokens: {
+        total: void 0,
+        text: void 0,
+        reasoning: void 0
+      },
+      raw: void 0
+    };
+  }
+  const promptTokens = (_a = usage.prompt_tokens) != null ? _a : 0;
+  const completionTokens = (_b = usage.completion_tokens) != null ? _b : 0;
+  const cacheReadTokens = (_d = (_c = usage.prompt_tokens_details) == null ? void 0 : _c.cached_tokens) != null ? _d : 0;
+  const reasoningTokens = (_f = (_e = usage.completion_tokens_details) == null ? void 0 : _e.reasoning_tokens) != null ? _f : 0;
+  return {
+    inputTokens: {
+      total: promptTokens,
+      noCache: promptTokens - cacheReadTokens,
+      cacheRead: cacheReadTokens,
+      cacheWrite: void 0
+    },
+    outputTokens: {
+      total: completionTokens + reasoningTokens,
+      text: completionTokens,
+      reasoning: reasoningTokens
+    },
+    raw: usage
+  };
+}
+function transformGoogleVertexXaiRequestBody(args) {
+  const { reasoning_effort: _reasoningEffort, ...rest } = args;
+  return rest;
+}
+function createGoogleVertexXai(options = {}) {
+  const loadLocation = () => loadOptionalSetting2({
+    settingValue: options.location,
+    environmentVariableName: "GOOGLE_VERTEX_LOCATION"
+  });
+  const loadProject = () => loadSetting2({
+    settingValue: options.project,
+    settingName: "project",
+    environmentVariableName: "GOOGLE_VERTEX_PROJECT",
+    description: "Google Vertex project"
+  });
+  const constructBaseURL = () => {
+    var _a;
+    const projectId = loadProject();
+    const location = (_a = loadLocation()) != null ? _a : "global";
+    return `https://aiplatform.googleapis.com/v1/projects/${projectId}/locations/${location}/endpoints/openapi`;
+  };
+  const loadBaseURL = () => {
+    var _a;
+    return withoutTrailingSlash((_a = options.baseURL) != null ? _a : "") || constructBaseURL();
+  };
+  let cachedProvider;
+  const getProvider = () => cachedProvider != null ? cachedProvider : cachedProvider = createOpenAICompatible({
+    name: "googleVertex.xai",
+    baseURL: loadBaseURL(),
+    fetch: options.fetch,
+    includeUsage: true,
+    supportsStructuredOutputs: true,
+    supportedUrls: () => ({
+      "image/*": [/^https?:\/\/.*$/]
+    }),
+    transformRequestBody: transformGoogleVertexXaiRequestBody,
+    convertUsage: convertGoogleVertexXaiUsage
+  });
+  const createChatModel = (modelId) => getProvider().languageModel(modelId);
+  const provider = function(modelId) {
+    if (new.target) {
+      throw new Error(
+        "The Google Vertex xAI model function cannot be called with the new keyword."
+      );
+    }
+    return createChatModel(modelId);
+  };
+  provider.specificationVersion = "v4";
+  provider.languageModel = createChatModel;
+  provider.chatModel = (modelId) => getProvider().chatModel(modelId);
+  provider.embeddingModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "embeddingModel" });
+  };
+  provider.textEmbeddingModel = provider.embeddingModel;
+  provider.imageModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "imageModel" });
+  };
+  return provider;
+}
+
+// src/xai/edge/google-vertex-xai-provider-edge.ts
+function createGoogleVertexXai2(options = {}) {
+  const customFetch = async (url, init) => {
+    var _a;
+    const token = await generateAuthToken(options.googleCredentials);
+    const resolvedHeaders = await resolve(options.headers);
+    const authHeaders = {
+      ...resolvedHeaders,
+      Authorization: `Bearer ${token}`
+    };
+    const fetchInit = {
+      ...init,
+      headers: {
+        ...init == null ? void 0 : init.headers,
+        ...authHeaders
+      }
+    };
+    return ((_a = options.fetch) != null ? _a : fetch)(url, fetchInit);
+  };
+  return createGoogleVertexXai({
+    ...options,
+    fetch: customFetch,
+    headers: void 0
+  });
+}
+var googleVertexXai = createGoogleVertexXai2();
+export {
+  createGoogleVertexXai2 as createGoogleVertexXai,
+  googleVertexXai
+};
+//# sourceMappingURL=index.js.map

package/dist/xai/edge/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/xai/edge/google-vertex-xai-provider-edge.ts","../../../src/edge/google-vertex-auth-edge.ts","../../../src/version.ts","../../../src/xai/google-vertex-xai-provider.ts"],"sourcesContent":["import { resolve, type FetchFunction } from '@ai-sdk/provider-utils';\nimport {\n  generateAuthToken,\n  type GoogleCredentials,\n} from '../../edge/google-vertex-auth-edge';\nimport {\n  createGoogleVertexXai as createGoogleVertexXaiOriginal,\n  type GoogleVertexXaiProvider,\n  type GoogleVertexXaiProviderSettings as GoogleVertexXaiProviderSettingsOriginal,\n} from '../google-vertex-xai-provider';\nexport type { GoogleVertexXaiProvider };\n\nexport interface GoogleVertexXaiProviderSettings extends GoogleVertexXaiProviderSettingsOriginal {\n  /**\n   * Optional. The Google credentials for the Google Cloud service account. If\n   * not provided, the Google Vertex provider will use environment variables to\n   * load the credentials.\n   */\n  googleCredentials?: GoogleCredentials;\n}\n\n/**\n * Create a Google Vertex AI xAI provider instance for Edge runtimes.\n * Uses the OpenAI-compatible Chat Completions API for Grok partner models.\n * Automatically handles Google Cloud authentication.\n *\n * @see https://cloud.google.com/vertex-ai/generative-ai/docs/partner-models/grok\n */\nexport function createGoogleVertexXai(\n  options: GoogleVertexXaiProviderSettings = {},\n): GoogleVertexXaiProvider {\n  const customFetch: FetchFunction = async (url, init) => {\n    const token = await generateAuthToken(options.googleCredentials);\n    const resolvedHeaders = await resolve(options.headers);\n    const authHeaders = {\n      ...resolvedHeaders,\n      Authorization: `Bearer ${token}`,\n    };\n\n    const fetchInit = {\n      ...init,\n      headers: {\n        ...init?.headers,\n        ...authHeaders,\n      },\n    };\n\n    return (options.fetch ?? fetch)(url, fetchInit);\n  };\n\n  return createGoogleVertexXaiOriginal({\n    ...options,\n    fetch: customFetch,\n    headers: undefined,\n  });\n}\n\n/**\n * Default Google Vertex AI xAI provider instance for Edge runtimes.\n */\nexport const googleVertexXai = createGoogleVertexXai();\n","import {\n  loadOptionalSetting,\n  loadSetting,\n  withUserAgentSuffix,\n  getRuntimeEnvironmentUserAgent,\n} from '@ai-sdk/provider-utils';\nimport { VERSION } from '../version';\n\nexport interface GoogleCredentials {\n  /**\n   * The client email for the Google Cloud service account. Defaults to the\n   * value of the `GOOGLE_CLIENT_EMAIL` environment variable.\n   */\n  clientEmail: string;\n\n  /**\n   * The private key for the Google Cloud service account. Defaults to the\n   * value of the `GOOGLE_PRIVATE_KEY` environment variable.\n   */\n  privateKey: string;\n\n  /**\n   * Optional. The private key ID for the Google Cloud service account. Defaults\n   * to the value of the `GOOGLE_PRIVATE_KEY_ID` environment variable.\n   */\n  privateKeyId?: string;\n}\n\nconst loadCredentials = async (): Promise<GoogleCredentials> => {\n  try {\n    return {\n      clientEmail: loadSetting({\n        settingValue: undefined,\n        settingName: 'clientEmail',\n        environmentVariableName: 'GOOGLE_CLIENT_EMAIL',\n        description: 'Google client email',\n      }),\n      privateKey: loadSetting({\n        settingValue: undefined,\n        settingName: 'privateKey',\n        environmentVariableName: 'GOOGLE_PRIVATE_KEY',\n        description: 'Google private key',\n      }),\n      privateKeyId: loadOptionalSetting({\n        settingValue: undefined,\n        environmentVariableName: 'GOOGLE_PRIVATE_KEY_ID',\n      }),\n    };\n  } catch (error: any) {\n    throw new Error(`Failed to load Google credentials: ${error.message}`);\n  }\n};\n\n// Convert a string to base64url\nconst base64url = (str: string) => {\n  return btoa(str).replace(/\\+/g, '-').replace(/\\//g, '_').replace(/=/g, '');\n};\nconst importPrivateKey = async (pemKey: string) => {\n  const pemHeader = '-----BEGIN PRIVATE KEY-----';\n  const pemFooter = '-----END PRIVATE KEY-----';\n\n  // Remove header, footer, and any whitespace/newlines\n  const pemContents = pemKey\n    .replace(pemHeader, '')\n    .replace(pemFooter, '')\n    .replace(/\\s/g, '');\n\n  // Decode base64 to binary\n  const binaryString = atob(pemContents);\n\n  // Convert binary string to Uint8Array\n  const binaryData = new Uint8Array(binaryString.length);\n  for (let i = 0; i < binaryString.length; i++) {\n    binaryData[i] = binaryString.charCodeAt(i);\n  }\n\n  return await crypto.subtle.importKey(\n    'pkcs8',\n    binaryData,\n    { name: 'RSASSA-PKCS1-v1_5', hash: 'SHA-256' },\n    true,\n    ['sign'],\n  );\n};\n\nconst buildJwt = async (credentials: GoogleCredentials) => {\n  const now = Math.floor(Date.now() / 1000);\n\n  // Only include kid in header if privateKeyId is provided\n  const header: { alg: string; typ: string; kid?: string } = {\n    alg: 'RS256',\n    typ: 'JWT',\n  };\n\n  if (credentials.privateKeyId) {\n    header.kid = credentials.privateKeyId;\n  }\n\n  const payload = {\n    iss: credentials.clientEmail,\n    scope: 'https://www.googleapis.com/auth/cloud-platform',\n    aud: 'https://oauth2.googleapis.com/token',\n    exp: now + 3600,\n    iat: now,\n  };\n\n  const privateKey = await importPrivateKey(credentials.privateKey);\n\n  const signingInput = `${base64url(JSON.stringify(header))}.${base64url(\n    JSON.stringify(payload),\n  )}`;\n  const encoder = new TextEncoder();\n  const data = encoder.encode(signingInput);\n\n  const signature = await crypto.subtle.sign(\n    'RSASSA-PKCS1-v1_5',\n    privateKey,\n    data,\n  );\n\n  const signatureBase64 = base64url(\n    String.fromCharCode(...new Uint8Array(signature)),\n  );\n\n  return `${base64url(JSON.stringify(header))}.${base64url(\n    JSON.stringify(payload),\n  )}.${signatureBase64}`;\n};\n\n/**\n * Generate an authentication token for Google Vertex AI in a manner compatible\n * with the Edge runtime.\n */\nexport async function generateAuthToken(credentials?: GoogleCredentials) {\n  try {\n    const creds = credentials || (await loadCredentials());\n    const jwt = await buildJwt(creds);\n\n    const response = await fetch('https://oauth2.googleapis.com/token', {\n      method: 'POST',\n      headers: withUserAgentSuffix(\n        { 'Content-Type': 'application/x-www-form-urlencoded' },\n        `ai-sdk/google-vertex/${VERSION}`,\n        getRuntimeEnvironmentUserAgent(),\n      ),\n      body: new URLSearchParams({\n        grant_type: 'urn:ietf:params:oauth:grant-type:jwt-bearer',\n        assertion: jwt,\n      }),\n    });\n\n    if (!response.ok) {\n      throw new Error(`Token request failed: ${response.statusText}`);\n    }\n\n    const data = await response.json();\n    return data.access_token;\n  } catch (error) {\n    throw error;\n  }\n}\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n  typeof __PACKAGE_VERSION__ !== 'undefined'\n    ? __PACKAGE_VERSION__\n    : '0.0.0-test';\n","import {\n  NoSuchModelError,\n  type LanguageModelV4,\n  type LanguageModelV4Usage,\n  type ProviderV4,\n} from '@ai-sdk/provider';\nimport {\n  createOpenAICompatible,\n  type OpenAICompatibleProvider,\n} from '@ai-sdk/openai-compatible';\nimport {\n  loadOptionalSetting,\n  loadSetting,\n  withoutTrailingSlash,\n  type FetchFunction,\n  type Resolvable,\n} from '@ai-sdk/provider-utils';\nimport type { GoogleVertexXaiModelId } from './google-vertex-xai-options';\n\nexport interface GoogleVertexXaiProvider extends ProviderV4 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: GoogleVertexXaiModelId): LanguageModelV4;\n\n  /**\n   * Creates a model for text generation.\n   */\n  languageModel(modelId: GoogleVertexXaiModelId): LanguageModelV4;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: GoogleVertexXaiModelId): LanguageModelV4;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: string): never;\n}\n\nexport interface GoogleVertexXaiProviderSettings {\n  /**\n   * Google Cloud project ID. Defaults to the value of the `GOOGLE_VERTEX_PROJECT` environment variable.\n   */\n  project?: string;\n\n  /**\n   * Google Cloud location/region. Defaults to the value of the `GOOGLE_VERTEX_LOCATION` environment variable.\n   * Use 'global' for the global endpoint.\n   */\n  location?: string;\n\n  /**\n   * Base URL for the API calls. If not provided, will be constructed from project and location.\n   */\n  baseURL?: string;\n\n  /**\n   * Headers to use for requests. Can be:\n   * - A headers object\n   * - A Promise that resolves to a headers object\n   * - A function that returns a headers object\n   * - A function that returns a Promise of a headers object\n   */\n  headers?: Resolvable<Record<string, string | undefined>>;\n\n  /**\n   * Custom fetch implementation. You can use it as a middleware to intercept requests,\n   * or to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n}\n\ntype GoogleVertexXaiUsage =\n  | {\n      prompt_tokens?: number | null;\n      completion_tokens?: number | null;\n      prompt_tokens_details?: {\n        cached_tokens?: number | null;\n      } | null;\n      completion_tokens_details?: {\n        reasoning_tokens?: number | null;\n      } | null;\n    }\n  | undefined\n  | null;\n\nfunction convertGoogleVertexXaiUsage(\n  usage: GoogleVertexXaiUsage,\n): LanguageModelV4Usage {\n  if (usage == null) {\n    return {\n      inputTokens: {\n        total: undefined,\n        noCache: undefined,\n        cacheRead: undefined,\n        cacheWrite: undefined,\n      },\n      outputTokens: {\n        total: undefined,\n        text: undefined,\n        reasoning: undefined,\n      },\n      raw: undefined,\n    };\n  }\n\n  const promptTokens = usage.prompt_tokens ?? 0;\n  const completionTokens = usage.completion_tokens ?? 0;\n  const cacheReadTokens = usage.prompt_tokens_details?.cached_tokens ?? 0;\n  const reasoningTokens =\n    usage.completion_tokens_details?.reasoning_tokens ?? 0;\n\n  return {\n    inputTokens: {\n      total: promptTokens,\n      noCache: promptTokens - cacheReadTokens,\n      cacheRead: cacheReadTokens,\n      cacheWrite: undefined,\n    },\n    outputTokens: {\n      total: completionTokens + reasoningTokens,\n      text: completionTokens,\n      reasoning: reasoningTokens,\n    },\n    raw: usage,\n  };\n}\n\nfunction transformGoogleVertexXaiRequestBody(args: Record<string, any>) {\n  const { reasoning_effort: _reasoningEffort, ...rest } = args;\n  return rest;\n}\n\n/**\n * Create a Google Vertex AI xAI provider instance.\n * Uses the OpenAI-compatible Chat Completions API for Grok partner models.\n *\n * @see https://cloud.google.com/vertex-ai/generative-ai/docs/partner-models/grok\n */\nexport function createGoogleVertexXai(\n  options: GoogleVertexXaiProviderSettings = {},\n): GoogleVertexXaiProvider {\n  const loadLocation = () =>\n    loadOptionalSetting({\n      settingValue: options.location,\n      environmentVariableName: 'GOOGLE_VERTEX_LOCATION',\n    });\n\n  const loadProject = () =>\n    loadSetting({\n      settingValue: options.project,\n      settingName: 'project',\n      environmentVariableName: 'GOOGLE_VERTEX_PROJECT',\n      description: 'Google Vertex project',\n    });\n\n  const constructBaseURL = () => {\n    const projectId = loadProject();\n    const location = loadLocation() ?? 'global';\n\n    return `https://aiplatform.googleapis.com/v1/projects/${projectId}/locations/${location}/endpoints/openapi`;\n  };\n\n  const loadBaseURL = () =>\n    withoutTrailingSlash(options.baseURL ?? '') || constructBaseURL();\n\n  let cachedProvider:\n    | OpenAICompatibleProvider<GoogleVertexXaiModelId, string, string, string>\n    | undefined;\n  const getProvider = () =>\n    (cachedProvider ??= createOpenAICompatible({\n      name: 'googleVertex.xai',\n      baseURL: loadBaseURL(),\n      fetch: options.fetch,\n      includeUsage: true,\n      supportsStructuredOutputs: true,\n      supportedUrls: () => ({\n        'image/*': [/^https?:\\/\\/.*$/],\n      }),\n      transformRequestBody: transformGoogleVertexXaiRequestBody,\n      convertUsage: convertGoogleVertexXaiUsage,\n    }));\n\n  const createChatModel = (modelId: GoogleVertexXaiModelId) =>\n    getProvider().languageModel(modelId);\n\n  const provider = function (modelId: GoogleVertexXaiModelId) {\n    if (new.target) {\n      throw new Error(\n        'The Google Vertex xAI model function cannot be called with the new keyword.',\n      );\n    }\n\n    return createChatModel(modelId);\n  };\n\n  provider.specificationVersion = 'v4' as const;\n  provider.languageModel = createChatModel;\n  provider.chatModel = (modelId: GoogleVertexXaiModelId) =>\n    getProvider().chatModel(modelId);\n  provider.embeddingModel = (modelId: string): never => {\n    throw new NoSuchModelError({ modelId, modelType: 'embeddingModel' });\n  };\n  provider.textEmbeddingModel = provider.embeddingModel;\n  provider.imageModel = (modelId: string): never => {\n    throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n  };\n\n  return provider;\n}\n"],"mappings":";AAAA,SAAS,eAAmC;;;ACA5C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;;;ACHA,IAAM,UACX,OACI,kBACA;;;ADuBN,IAAM,kBAAkB,YAAwC;AAC9D,MAAI;AACF,WAAO;AAAA,MACL,aAAa,YAAY;AAAA,QACvB,cAAc;AAAA,QACd,aAAa;AAAA,QACb,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC;AAAA,MACD,YAAY,YAAY;AAAA,QACtB,cAAc;AAAA,QACd,aAAa;AAAA,QACb,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC;AAAA,MACD,cAAc,oBAAoB;AAAA,QAChC,cAAc;AAAA,QACd,yBAAyB;AAAA,MAC3B,CAAC;AAAA,IACH;AAAA,EACF,SAAS,OAAY;AACnB,UAAM,IAAI,MAAM,sCAAsC,MAAM,OAAO,EAAE;AAAA,EACvE;AACF;AAGA,IAAM,YAAY,CAAC,QAAgB;AACjC,SAAO,KAAK,GAAG,EAAE,QAAQ,OAAO,GAAG,EAAE,QAAQ,OAAO,GAAG,EAAE,QAAQ,MAAM,EAAE;AAC3E;AACA,IAAM,mBAAmB,OAAO,WAAmB;AACjD,QAAM,YAAY;AAClB,QAAM,YAAY;AAGlB,QAAM,cAAc,OACjB,QAAQ,WAAW,EAAE,EACrB,QAAQ,WAAW,EAAE,EACrB,QAAQ,OAAO,EAAE;AAGpB,QAAM,eAAe,KAAK,WAAW;AAGrC,QAAM,aAAa,IAAI,WAAW,aAAa,MAAM;AACrD,WAAS,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;AAC5C,eAAW,CAAC,IAAI,aAAa,WAAW,CAAC;AAAA,EAC3C;AAEA,SAAO,MAAM,OAAO,OAAO;AAAA,IACzB;AAAA,IACA;AAAA,IACA,EAAE,MAAM,qBAAqB,MAAM,UAAU;AAAA,IAC7C;AAAA,IACA,CAAC,MAAM;AAAA,EACT;AACF;AAEA,IAAM,WAAW,OAAO,gBAAmC;AACzD,QAAM,MAAM,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAGxC,QAAM,SAAqD;AAAA,IACzD,KAAK;AAAA,IACL,KAAK;AAAA,EACP;AAEA,MAAI,YAAY,cAAc;AAC5B,WAAO,MAAM,YAAY;AAAA,EAC3B;AAEA,QAAM,UAAU;AAAA,IACd,KAAK,YAAY;AAAA,IACjB,OAAO;AAAA,IACP,KAAK;AAAA,IACL,KAAK,MAAM;AAAA,IACX,KAAK;AAAA,EACP;AAEA,QAAM,aAAa,MAAM,iBAAiB,YAAY,UAAU;AAEhE,QAAM,eAAe,GAAG,UAAU,KAAK,UAAU,MAAM,CAAC,CAAC,IAAI;AAAA,IAC3D,KAAK,UAAU,OAAO;AAAA,EACxB,CAAC;AACD,QAAM,UAAU,IAAI,YAAY;AAChC,QAAM,OAAO,QAAQ,OAAO,YAAY;AAExC,QAAM,YAAY,MAAM,OAAO,OAAO;AAAA,IACpC;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,kBAAkB;AAAA,IACtB,OAAO,aAAa,GAAG,IAAI,WAAW,SAAS,CAAC;AAAA,EAClD;AAEA,SAAO,GAAG,UAAU,KAAK,UAAU,MAAM,CAAC,CAAC,IAAI;AAAA,IAC7C,KAAK,UAAU,OAAO;AAAA,EACxB,CAAC,IAAI,eAAe;AACtB;AAMA,eAAsB,kBAAkB,aAAiC;AACvE,MAAI;AACF,UAAM,QAAQ,eAAgB,MAAM,gBAAgB;AACpD,UAAM,MAAM,MAAM,SAAS,KAAK;AAEhC,UAAM,WAAW,MAAM,MAAM,uCAAuC;AAAA,MAClE,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,EAAE,gBAAgB,oCAAoC;AAAA,QACtD,wBAAwB,OAAO;AAAA,QAC/B,+BAA+B;AAAA,MACjC;AAAA,MACA,MAAM,IAAI,gBAAgB;AAAA,QACxB,YAAY;AAAA,QACZ,WAAW;AAAA,MACb,CAAC;AAAA,IACH,CAAC;AAED,QAAI,CAAC,SAAS,IAAI;AAChB,YAAM,IAAI,MAAM,yBAAyB,SAAS,UAAU,EAAE;AAAA,IAChE;AAEA,UAAM,OAAO,MAAM,SAAS,KAAK;AACjC,WAAO,KAAK;AAAA,EACd,SAAS,OAAO;AACd,UAAM;AAAA,EACR;AACF;;;AEhKA;AAAA,EACE;AAAA,OAIK;AACP;AAAA,EACE;AAAA,OAEK;AACP;AAAA,EACE,uBAAAA;AAAA,EACA,eAAAC;AAAA,EACA;AAAA,OAGK;AAwEP,SAAS,4BACP,OACsB;AA1FxB;AA2FE,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,MACL,aAAa;AAAA,QACX,OAAO;AAAA,QACP,SAAS;AAAA,QACT,WAAW;AAAA,QACX,YAAY;AAAA,MACd;AAAA,MACA,cAAc;AAAA,QACZ,OAAO;AAAA,QACP,MAAM;AAAA,QACN,WAAW;AAAA,MACb;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAEA,QAAM,gBAAe,WAAM,kBAAN,YAAuB;AAC5C,QAAM,oBAAmB,WAAM,sBAAN,YAA2B;AACpD,QAAM,mBAAkB,iBAAM,0BAAN,mBAA6B,kBAA7B,YAA8C;AACtE,QAAM,mBACJ,iBAAM,8BAAN,mBAAiC,qBAAjC,YAAqD;AAEvD,SAAO;AAAA,IACL,aAAa;AAAA,MACX,OAAO;AAAA,MACP,SAAS,eAAe;AAAA,MACxB,WAAW;AAAA,MACX,YAAY;AAAA,IACd;AAAA,IACA,cAAc;AAAA,MACZ,OAAO,mBAAmB;AAAA,MAC1B,MAAM;AAAA,MACN,WAAW;AAAA,IACb;AAAA,IACA,KAAK;AAAA,EACP;AACF;AAEA,SAAS,oCAAoC,MAA2B;AACtE,QAAM,EAAE,kBAAkB,kBAAkB,GAAG,KAAK,IAAI;AACxD,SAAO;AACT;AAQO,SAAS,sBACd,UAA2C,CAAC,GACnB;AACzB,QAAM,eAAe,MACnBD,qBAAoB;AAAA,IAClB,cAAc,QAAQ;AAAA,IACtB,yBAAyB;AAAA,EAC3B,CAAC;AAEH,QAAM,cAAc,MAClBC,aAAY;AAAA,IACV,cAAc,QAAQ;AAAA,IACtB,aAAa;AAAA,IACb,yBAAyB;AAAA,IACzB,aAAa;AAAA,EACf,CAAC;AAEH,QAAM,mBAAmB,MAAM;AA9JjC;AA+JI,UAAM,YAAY,YAAY;AAC9B,UAAM,YAAW,kBAAa,MAAb,YAAkB;AAEnC,WAAO,iDAAiD,SAAS,cAAc,QAAQ;AAAA,EACzF;AAEA,QAAM,cAAc,MAAG;AArKzB;AAsKI,iCAAqB,aAAQ,YAAR,YAAmB,EAAE,KAAK,iBAAiB;AAAA;AAElE,MAAI;AAGJ,QAAM,cAAc,MACjB,2DAAmB,uBAAuB;AAAA,IACzC,MAAM;AAAA,IACN,SAAS,YAAY;AAAA,IACrB,OAAO,QAAQ;AAAA,IACf,cAAc;AAAA,IACd,2BAA2B;AAAA,IAC3B,eAAe,OAAO;AAAA,MACpB,WAAW,CAAC,iBAAiB;AAAA,IAC/B;AAAA,IACA,sBAAsB;AAAA,IACtB,cAAc;AAAA,EAChB,CAAC;AAEH,QAAM,kBAAkB,CAAC,YACvB,YAAY,EAAE,cAAc,OAAO;AAErC,QAAM,WAAW,SAAU,SAAiC;AAC1D,QAAI,YAAY;AACd,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,WAAO,gBAAgB,OAAO;AAAA,EAChC;AAEA,WAAS,uBAAuB;AAChC,WAAS,gBAAgB;AACzB,WAAS,YAAY,CAAC,YACpB,YAAY,EAAE,UAAU,OAAO;AACjC,WAAS,iBAAiB,CAAC,YAA2B;AACpD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,iBAAiB,CAAC;AAAA,EACrE;AACA,WAAS,qBAAqB,SAAS;AACvC,WAAS,aAAa,CAAC,YAA2B;AAChD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AAEA,SAAO;AACT;;;AHvLO,SAASC,uBACd,UAA2C,CAAC,GACnB;AACzB,QAAM,cAA6B,OAAO,KAAK,SAAS;AA/B1D;AAgCI,UAAM,QAAQ,MAAM,kBAAkB,QAAQ,iBAAiB;AAC/D,UAAM,kBAAkB,MAAM,QAAQ,QAAQ,OAAO;AACrD,UAAM,cAAc;AAAA,MAClB,GAAG;AAAA,MACH,eAAe,UAAU,KAAK;AAAA,IAChC;AAEA,UAAM,YAAY;AAAA,MAChB,GAAG;AAAA,MACH,SAAS;AAAA,QACP,GAAG,6BAAM;AAAA,QACT,GAAG;AAAA,MACL;AAAA,IACF;AAEA,aAAQ,aAAQ,UAAR,YAAiB,OAAO,KAAK,SAAS;AAAA,EAChD;AAEA,SAAO,sBAA8B;AAAA,IACnC,GAAG;AAAA,IACH,OAAO;AAAA,IACP,SAAS;AAAA,EACX,CAAC;AACH;AAKO,IAAM,kBAAkBA,uBAAsB;","names":["loadOptionalSetting","loadSetting","createGoogleVertexXai"]}

package/dist/xai/index.d.ts

@@ -0,0 +1,76 @@
+import { GoogleAuthOptions } from 'google-auth-library';
+import { ProviderV4, LanguageModelV4 } from '@ai-sdk/provider';
+import { Resolvable, FetchFunction } from '@ai-sdk/provider-utils';
+
+type GoogleVertexXaiModelId = 'xai/grok-4.20-reasoning' | 'xai/grok-4.20-non-reasoning' | 'xai/grok-4.1-fast-reasoning' | 'xai/grok-4.1-fast-non-reasoning' | (string & {});
+
+interface GoogleVertexXaiProvider extends ProviderV4 {
+    /**
+     * Creates a model for text generation.
+     */
+    (modelId: GoogleVertexXaiModelId): LanguageModelV4;
+    /**
+     * Creates a model for text generation.
+     */
+    languageModel(modelId: GoogleVertexXaiModelId): LanguageModelV4;
+    /**
+     * Creates a chat model for text generation.
+     */
+    chatModel(modelId: GoogleVertexXaiModelId): LanguageModelV4;
+    /**
+     * @deprecated Use `embeddingModel` instead.
+     */
+    textEmbeddingModel(modelId: string): never;
+}
+interface GoogleVertexXaiProviderSettings$1 {
+    /**
+     * 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;
+}
+
+interface GoogleVertexXaiProviderSettings extends GoogleVertexXaiProviderSettings$1 {
+    /**
+     * 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 xAI provider instance for Node.js.
+ * Uses the OpenAI-compatible Chat Completions API for Grok partner models.
+ * Automatically handles Google Cloud authentication.
+ *
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/partner-models/grok
+ */
+declare function createGoogleVertexXai(options?: GoogleVertexXaiProviderSettings): GoogleVertexXaiProvider;
+/**
+ * Default Google Vertex AI xAI provider instance for Node.js.
+ */
+declare const googleVertexXai: GoogleVertexXaiProvider;
+
+export { type GoogleVertexXaiModelId, type GoogleVertexXaiProvider, type GoogleVertexXaiProviderSettings, createGoogleVertexXai, googleVertexXai };