@ai-sdk/google-vertex 3.0.135 → 3.0.137

package/dist/xai/index.mjs

@@ -0,0 +1,151 @@
+// src/xai/google-vertex-xai-provider-node.ts
+import { resolve } from "@ai-sdk/provider-utils";
+
+// src/google-vertex-auth-google-auth-library.ts
+import { GoogleAuth } from "google-auth-library";
+var authInstance = null;
+var authOptions = null;
+function getAuth(options) {
+  if (!authInstance || options !== authOptions) {
+    authInstance = new GoogleAuth({
+      scopes: ["https://www.googleapis.com/auth/cloud-platform"],
+      ...options
+    });
+    authOptions = options;
+  }
+  return authInstance;
+}
+async function generateAuthToken(options) {
+  const auth = getAuth(options || {});
+  const client = await auth.getClient();
+  const token = await client.getAccessToken();
+  return (token == null ? void 0 : token.token) || null;
+}
+
+// src/xai/google-vertex-xai-provider.ts
+import {
+  NoSuchModelError
+} from "@ai-sdk/provider";
+import {
+  createOpenAICompatible
+} from "@ai-sdk/openai-compatible";
+import {
+  loadOptionalSetting,
+  loadSetting,
+  withoutTrailingSlash
+} from "@ai-sdk/provider-utils";
+function convertGoogleVertexXaiUsage(usage) {
+  var _a, _b, _c, _d, _e, _f, _g;
+  if (usage == null) {
+    return {
+      inputTokens: void 0,
+      outputTokens: void 0,
+      totalTokens: void 0,
+      reasoningTokens: void 0,
+      cachedInputTokens: void 0
+    };
+  }
+  const promptTokens = (_a = usage.prompt_tokens) != null ? _a : void 0;
+  const completionTokens = (_b = usage.completion_tokens) != null ? _b : void 0;
+  const cachedInputTokens = (_d = (_c = usage.prompt_tokens_details) == null ? void 0 : _c.cached_tokens) != null ? _d : void 0;
+  const reasoningTokens = (_f = (_e = usage.completion_tokens_details) == null ? void 0 : _e.reasoning_tokens) != null ? _f : void 0;
+  const outputTokens = completionTokens == null && reasoningTokens == null ? void 0 : (completionTokens != null ? completionTokens : 0) + (reasoningTokens != null ? reasoningTokens : 0);
+  return {
+    inputTokens: promptTokens,
+    outputTokens,
+    totalTokens: (_g = usage.total_tokens) != null ? _g : void 0,
+    reasoningTokens,
+    cachedInputTokens
+  };
+}
+function transformGoogleVertexXaiRequestBody(args) {
+  const transformedArgs = { ...args };
+  delete transformedArgs.reasoning_effort;
+  return transformedArgs;
+}
+function createGoogleVertexXai(options = {}) {
+  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"
+  });
+  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 = "v2";
+  provider.languageModel = createChatModel;
+  provider.chatModel = (modelId) => getProvider().chatModel(modelId);
+  provider.textEmbeddingModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
+  };
+  provider.imageModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "imageModel" });
+  };
+  return provider;
+}
+
+// src/xai/google-vertex-xai-provider-node.ts
+function createGoogleVertexXai2(options = {}) {
+  const customFetch = async (url, init) => {
+    var _a;
+    const token = await generateAuthToken(options.googleAuthOptions);
+    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.mjs.map

package/dist/xai/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/xai/google-vertex-xai-provider-node.ts","../../src/google-vertex-auth-google-auth-library.ts","../../src/xai/google-vertex-xai-provider.ts"],"sourcesContent":["import { resolve, type FetchFunction } from '@ai-sdk/provider-utils';\nimport type { GoogleAuthOptions } from 'google-auth-library';\nimport { generateAuthToken } from '../google-vertex-auth-google-auth-library';\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 Authentication options provided by google-auth-library.\n   * Complete list of authentication options is documented in the\n   * GoogleAuthOptions interface:\n   * https://github.com/googleapis/google-auth-library-nodejs/blob/main/src/auth/googleauth.ts.\n   */\n  googleAuthOptions?: GoogleAuthOptions;\n}\n\n/**\n * Create a Google Vertex AI xAI provider instance for Node.js.\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.googleAuthOptions);\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 Node.js.\n */\nexport const googleVertexXai = createGoogleVertexXai();\n","import { type GoogleAuthOptions, GoogleAuth } from 'google-auth-library';\n\nlet authInstance: GoogleAuth | null = null;\nlet authOptions: GoogleAuthOptions | null = null;\n\nfunction getAuth(options: GoogleAuthOptions) {\n  if (!authInstance || options !== authOptions) {\n    authInstance = new GoogleAuth({\n      scopes: ['https://www.googleapis.com/auth/cloud-platform'],\n      ...options,\n    });\n    authOptions = options;\n  }\n  return authInstance;\n}\n\nexport async function generateAuthToken(options?: GoogleAuthOptions) {\n  const auth = getAuth(options || {});\n  const client = await auth.getClient();\n  const token = await client.getAccessToken();\n  return token?.token || null;\n}\n\n// For testing purposes only\nexport function _resetAuthInstance() {\n  authInstance = null;\n}\n","import {\n  NoSuchModelError,\n  type LanguageModelV2,\n  type LanguageModelV2Usage,\n  type ProviderV2,\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 ProviderV2 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: GoogleVertexXaiModelId): LanguageModelV2;\n\n  /**\n   * Creates a model for text generation.\n   */\n  languageModel(modelId: GoogleVertexXaiModelId): LanguageModelV2;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: GoogleVertexXaiModelId): LanguageModelV2;\n\n  /**\n   * Creates a text embedding model.\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      total_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): LanguageModelV2Usage {\n  if (usage == null) {\n    return {\n      inputTokens: undefined,\n      outputTokens: undefined,\n      totalTokens: undefined,\n      reasoningTokens: undefined,\n      cachedInputTokens: undefined,\n    };\n  }\n\n  const promptTokens = usage.prompt_tokens ?? undefined;\n  const completionTokens = usage.completion_tokens ?? undefined;\n  const cachedInputTokens =\n    usage.prompt_tokens_details?.cached_tokens ?? undefined;\n  const reasoningTokens =\n    usage.completion_tokens_details?.reasoning_tokens ?? undefined;\n  const outputTokens =\n    completionTokens == null && reasoningTokens == null\n      ? undefined\n      : (completionTokens ?? 0) + (reasoningTokens ?? 0);\n\n  return {\n    inputTokens: promptTokens,\n    outputTokens,\n    totalTokens: usage.total_tokens ?? undefined,\n    reasoningTokens,\n    cachedInputTokens,\n  };\n}\n\nfunction transformGoogleVertexXaiRequestBody(args: Record<string, any>) {\n  const transformedArgs = { ...args };\n  delete transformedArgs.reasoning_effort;\n  return transformedArgs;\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 = 'v2' as const;\n  provider.languageModel = createChatModel;\n  provider.chatModel = (modelId: GoogleVertexXaiModelId) =>\n    getProvider().chatModel(modelId);\n  provider.textEmbeddingModel = (modelId: string): never => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\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,SAAiC,kBAAkB;AAEnD,IAAI,eAAkC;AACtC,IAAI,cAAwC;AAE5C,SAAS,QAAQ,SAA4B;AAC3C,MAAI,CAAC,gBAAgB,YAAY,aAAa;AAC5C,mBAAe,IAAI,WAAW;AAAA,MAC5B,QAAQ,CAAC,gDAAgD;AAAA,MACzD,GAAG;AAAA,IACL,CAAC;AACD,kBAAc;AAAA,EAChB;AACA,SAAO;AACT;AAEA,eAAsB,kBAAkB,SAA6B;AACnE,QAAM,OAAO,QAAQ,WAAW,CAAC,CAAC;AAClC,QAAM,SAAS,MAAM,KAAK,UAAU;AACpC,QAAM,QAAQ,MAAM,OAAO,eAAe;AAC1C,UAAO,+BAAO,UAAS;AACzB;;;ACrBA;AAAA,EACE;AAAA,OAIK;AACP;AAAA,EACE;AAAA,OAEK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAGK;AAyEP,SAAS,4BACP,OACsB;AA3FxB;AA4FE,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,MACL,aAAa;AAAA,MACb,cAAc;AAAA,MACd,aAAa;AAAA,MACb,iBAAiB;AAAA,MACjB,mBAAmB;AAAA,IACrB;AAAA,EACF;AAEA,QAAM,gBAAe,WAAM,kBAAN,YAAuB;AAC5C,QAAM,oBAAmB,WAAM,sBAAN,YAA2B;AACpD,QAAM,qBACJ,iBAAM,0BAAN,mBAA6B,kBAA7B,YAA8C;AAChD,QAAM,mBACJ,iBAAM,8BAAN,mBAAiC,qBAAjC,YAAqD;AACvD,QAAM,eACJ,oBAAoB,QAAQ,mBAAmB,OAC3C,UACC,8CAAoB,MAAM,4CAAmB;AAEpD,SAAO;AAAA,IACL,aAAa;AAAA,IACb;AAAA,IACA,cAAa,WAAM,iBAAN,YAAsB;AAAA,IACnC;AAAA,IACA;AAAA,EACF;AACF;AAEA,SAAS,oCAAoC,MAA2B;AACtE,QAAM,kBAAkB,EAAE,GAAG,KAAK;AAClC,SAAO,gBAAgB;AACvB,SAAO;AACT;AAQO,SAAS,sBACd,UAA2C,CAAC,GACnB;AACzB,QAAM,eAAe,MACnB,oBAAoB;AAAA,IAClB,cAAc,QAAQ;AAAA,IACtB,yBAAyB;AAAA,EAC3B,CAAC;AAEH,QAAM,cAAc,MAClB,YAAY;AAAA,IACV,cAAc,QAAQ;AAAA,IACtB,aAAa;AAAA,IACb,yBAAyB;AAAA,IACzB,aAAa;AAAA,EACf,CAAC;AAEH,QAAM,mBAAmB,MAAM;AAvJjC;AAwJI,UAAM,YAAY,YAAY;AAC9B,UAAM,YAAW,kBAAa,MAAb,YAAkB;AAEnC,WAAO,iDAAiD,SAAS,cAAc,QAAQ;AAAA,EACzF;AAEA,QAAM,cAAc,MAAG;AA9JzB;AA+JI,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,qBAAqB,CAAC,YAA2B;AACxD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AACA,WAAS,aAAa,CAAC,YAA2B;AAChD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AAEA,SAAO;AACT;;;AFhLO,SAASA,uBACd,UAA2C,CAAC,GACnB;AACzB,QAAM,cAA6B,OAAO,KAAK,SAAS;AA9B1D;AA+BI,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":["createGoogleVertexXai"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/google-vertex",
-  "version": "3.0.135",
+  "version": "3.0.137",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -13,7 +13,9 @@
     "anthropic/edge.d.ts",
     "anthropic/index.d.ts",
     "maas/edge.d.ts",
-    "maas/index.d.ts"
+    "maas/index.d.ts",
+    "xai/edge.d.ts",
+    "xai/index.d.ts"
   ],
   "exports": {
     "./package.json": "./package.json",
@@ -46,13 +48,23 @@
       "types": "./dist/maas/edge/index.d.ts",
       "import": "./dist/maas/edge/index.mjs",
       "require": "./dist/maas/edge/index.js"
+    },
+    "./xai": {
+      "types": "./dist/xai/index.d.ts",
+      "import": "./dist/xai/index.mjs",
+      "require": "./dist/xai/index.js"
+    },
+    "./xai/edge": {
+      "types": "./dist/xai/edge/index.d.ts",
+      "import": "./dist/xai/edge/index.mjs",
+      "require": "./dist/xai/edge/index.js"
     }
   },
   "dependencies": {
     "google-auth-library": "^10.5.0",
     "@ai-sdk/anthropic": "2.0.79",
     "@ai-sdk/google": "2.0.72",
-    "@ai-sdk/openai-compatible": "1.0.38",
+    "@ai-sdk/openai-compatible": "1.0.39",
     "@ai-sdk/provider": "2.0.3",
     "@ai-sdk/provider-utils": "3.0.25"
   },
@@ -61,8 +73,8 @@
     "tsup": "^8",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "0.0.4",
-    "@vercel/ai-tsconfig": "0.0.0"
+    "@vercel/ai-tsconfig": "0.0.0",
+    "@ai-sdk/test-server": "0.0.4"
   },
   "peerDependencies": {
     "zod": "^3.25.76 || ^4.1.8"

package/xai/edge.d.ts

@@ -0,0 +1 @@
+export * from '../dist/xai/edge';

package/xai/index.d.ts

@@ -0,0 +1 @@
+export * from '../dist/xai/index';