@amanm/vertex-maas 0.0.1

package/CHANGELOG.md

@@ -0,0 +1,5 @@
+# Changelog
+
+## Unreleased
+
+- Initial release.

package/README.md

@@ -0,0 +1,35 @@
+# @ai-sdk/vertex-maas
+
+Vertex MaaS provider for the Vercel AI SDK.
+
+This provider targets **Vertex AI models exposed via OpenAI-compatible endpoints** and authenticates using **Google Application Default Credentials (ADC)** (service account JSON via `GOOGLE_APPLICATION_CREDENTIALS`, workload identity, etc.).
+
+## Requirements
+
+- Node.js runtime (not Edge)
+- ADC configured (commonly via `GOOGLE_APPLICATION_CREDENTIALS`)
+- Set `GOOGLE_VERTEX_PROJECT` and `GOOGLE_VERTEX_LOCATION` (or pass `project`/`location` to `createVertexMaaS`)
+
+## Usage
+
+```ts
+import { generateText } from 'ai';
+import { createVertexMaaS } from '@amanm/vertex-maas';
+
+const vertex = createVertexMaaS({
+  project: process.env.GOOGLE_VERTEX_PROJECT!,
+  location: process.env.GOOGLE_VERTEX_LOCATION!, // e.g. "global" or "us-central1"
+});
+
+const result = await generateText({
+  model: vertex('glm-4.7'),
+  prompt: 'Explain Amdahl\\'s Law in simple terms.',
+});
+```
+
+## Settings
+
+- `baseURL`: Optional. If set, used directly.
+- `project` / `location`: Used to compute the Vertex OpenAI-compatible base URL if `baseURL` is not provided.
+- `apiKey`: Optional. If set, uses a static `Authorization: Bearer <apiKey>` header (no ADC).
+- `googleAuthOptions`: Optional. Passed to `google-auth-library` `GoogleAuth`.

package/dist/index.d.mts

@@ -0,0 +1,64 @@
+import { MetadataExtractor, OpenAICompatibleProvider } from '@ai-sdk/openai-compatible';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { GoogleAuthOptions } from 'google-auth-library';
+
+interface VertexMaaSProviderSettings {
+    /**
+     * Base URL for the API calls. If not set, it is derived from `project` and
+     * `location`.
+     */
+    baseURL?: string;
+    /**
+     * Vertex project id. Defaults to `GOOGLE_VERTEX_PROJECT` if `baseURL` is not
+     * provided.
+     */
+    project?: string;
+    /**
+     * Vertex location. Defaults to `GOOGLE_VERTEX_LOCATION` if `baseURL` is not
+     * provided.
+     */
+    location?: string;
+    /**
+     * API version used when computing baseURL. Defaults to `v1beta1`.
+     */
+    apiVersion?: 'v1' | 'v1beta1';
+    /**
+     * Override the base host for Vertex API calls (e.g. "https://aiplatform.googleapis.com").
+     * If not set, it is computed from location.
+     */
+    apiBase?: string;
+    /**
+     * Optional. Use a static bearer token instead of ADC.
+     */
+    apiKey?: string;
+    /**
+     * Optional. Passed to google-auth-library for ADC token generation.
+     */
+    googleAuthOptions?: GoogleAuthOptions;
+    /**
+     * Optional custom headers.
+     */
+    headers?: Record<string, string>;
+    /**
+     * Optional custom url query parameters to include in request urls.
+     */
+    queryParams?: Record<string, string>;
+    /**
+     * Custom fetch implementation.
+     */
+    fetch?: FetchFunction;
+    includeUsage?: boolean;
+    supportsStructuredOutputs?: boolean;
+    transformRequestBody?: (args: Record<string, any>) => Record<string, any>;
+    metadataExtractor?: MetadataExtractor;
+}
+type VertexMaaSProvider = OpenAICompatibleProvider;
+declare function createVertexMaaS(options?: VertexMaaSProviderSettings): VertexMaaSProvider;
+/**
+ * Default Vertex MaaS provider instance.
+ */
+declare const vertexMaaS: VertexMaaSProvider;
+
+declare const VERSION: string;
+
+export { VERSION, type VertexMaaSProvider, type VertexMaaSProviderSettings, createVertexMaaS, vertexMaaS };

package/dist/index.d.ts

@@ -0,0 +1,64 @@
+import { MetadataExtractor, OpenAICompatibleProvider } from '@ai-sdk/openai-compatible';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { GoogleAuthOptions } from 'google-auth-library';
+
+interface VertexMaaSProviderSettings {
+    /**
+     * Base URL for the API calls. If not set, it is derived from `project` and
+     * `location`.
+     */
+    baseURL?: string;
+    /**
+     * Vertex project id. Defaults to `GOOGLE_VERTEX_PROJECT` if `baseURL` is not
+     * provided.
+     */
+    project?: string;
+    /**
+     * Vertex location. Defaults to `GOOGLE_VERTEX_LOCATION` if `baseURL` is not
+     * provided.
+     */
+    location?: string;
+    /**
+     * API version used when computing baseURL. Defaults to `v1beta1`.
+     */
+    apiVersion?: 'v1' | 'v1beta1';
+    /**
+     * Override the base host for Vertex API calls (e.g. "https://aiplatform.googleapis.com").
+     * If not set, it is computed from location.
+     */
+    apiBase?: string;
+    /**
+     * Optional. Use a static bearer token instead of ADC.
+     */
+    apiKey?: string;
+    /**
+     * Optional. Passed to google-auth-library for ADC token generation.
+     */
+    googleAuthOptions?: GoogleAuthOptions;
+    /**
+     * Optional custom headers.
+     */
+    headers?: Record<string, string>;
+    /**
+     * Optional custom url query parameters to include in request urls.
+     */
+    queryParams?: Record<string, string>;
+    /**
+     * Custom fetch implementation.
+     */
+    fetch?: FetchFunction;
+    includeUsage?: boolean;
+    supportsStructuredOutputs?: boolean;
+    transformRequestBody?: (args: Record<string, any>) => Record<string, any>;
+    metadataExtractor?: MetadataExtractor;
+}
+type VertexMaaSProvider = OpenAICompatibleProvider;
+declare function createVertexMaaS(options?: VertexMaaSProviderSettings): VertexMaaSProvider;
+/**
+ * Default Vertex MaaS provider instance.
+ */
+declare const vertexMaaS: VertexMaaSProvider;
+
+declare const VERSION: string;
+
+export { VERSION, type VertexMaaSProvider, type VertexMaaSProviderSettings, createVertexMaaS, vertexMaaS };

package/dist/index.js

@@ -0,0 +1,190 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  VERSION: () => VERSION,
+  createVertexMaaS: () => createVertexMaaS,
+  vertexMaaS: () => vertexMaaS
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/vertex-maas-provider.ts
+var import_openai_compatible = require("@ai-sdk/openai-compatible");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
+
+// src/version.ts
+var VERSION = true ? "0.0.1" : "0.0.0-test";
+
+// src/vertex-maas-auth-fetch.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
+
+// src/vertex-maas-auth-google-auth-library.ts
+var import_google_auth_library = require("google-auth-library");
+var authInstance = null;
+var authOptions = null;
+function getAuth(options) {
+  if (!authInstance || options !== authOptions) {
+    authInstance = new import_google_auth_library.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/vertex-maas-auth-fetch.ts
+function createVertexMaaSAuthFetch({
+  customFetch,
+  googleAuthOptions
+}) {
+  return async (url, init) => {
+    const existingHeaders = (0, import_provider_utils.normalizeHeaders)(init == null ? void 0 : init.headers);
+    if (existingHeaders.authorization != null) {
+      return (customFetch != null ? customFetch : fetch)(url, init);
+    }
+    const token = await generateAuthToken(googleAuthOptions);
+    if (!token) {
+      throw new Error("Failed to get access token from ADC");
+    }
+    const modifiedInit = {
+      ...init,
+      headers: {
+        ...existingHeaders,
+        authorization: `Bearer ${token}`
+      }
+    };
+    return (customFetch != null ? customFetch : fetch)(url, modifiedInit);
+  };
+}
+
+// src/vertex-maas-base-url.ts
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+function getDefaultVertexApiBase(location) {
+  return location === "global" ? "https://aiplatform.googleapis.com" : `https://${location}-aiplatform.googleapis.com`;
+}
+function getVertexOpenAICompatibleBaseURL({
+  project,
+  location,
+  apiVersion,
+  apiBase
+}) {
+  const resolvedApiBase = (0, import_provider_utils2.withoutTrailingSlash)(
+    apiBase != null ? apiBase : getDefaultVertexApiBase(location)
+  );
+  return `${resolvedApiBase}/${apiVersion}/projects/${project}/locations/${location}/endpoints/openapi`;
+}
+
+// src/vertex-maas-provider.ts
+function createVertexMaaS(options = {}) {
+  const providerName = "vertexMaas";
+  let baseURLCache = null;
+  const getBaseURL = () => {
+    var _a, _b;
+    if (baseURLCache != null) {
+      return baseURLCache;
+    }
+    baseURLCache = (_b = (0, import_provider_utils3.withoutTrailingSlash)(options.baseURL)) != null ? _b : getVertexOpenAICompatibleBaseURL({
+      project: (0, import_provider_utils3.loadSetting)({
+        settingValue: options.project,
+        settingName: "project",
+        environmentVariableName: "GOOGLE_VERTEX_PROJECT",
+        description: "Google Vertex project"
+      }),
+      location: (0, import_provider_utils3.loadSetting)({
+        settingValue: options.location,
+        settingName: "location",
+        environmentVariableName: "GOOGLE_VERTEX_LOCATION",
+        description: "Google Vertex location"
+      }),
+      apiVersion: (_a = options.apiVersion) != null ? _a : "v1beta1",
+      apiBase: options.apiBase
+    });
+    return baseURLCache;
+  };
+  const staticHeaders = {
+    ...options.apiKey && { Authorization: `Bearer ${options.apiKey}` },
+    ...options.headers
+  };
+  const getHeaders = () => (0, import_provider_utils3.withUserAgentSuffix)(staticHeaders, `ai-sdk/vertex-maas/${VERSION}`);
+  const fetch2 = options.apiKey != null ? options.fetch : createVertexMaaSAuthFetch({
+    customFetch: options.fetch,
+    googleAuthOptions: options.googleAuthOptions
+  });
+  const getCommonModelConfig = (modelType) => ({
+    provider: `${providerName}.${modelType}`,
+    url: ({ path }) => {
+      const url = new URL(`${getBaseURL()}${path}`);
+      if (options.queryParams) {
+        url.search = new URLSearchParams(options.queryParams).toString();
+      }
+      return url.toString();
+    },
+    headers: getHeaders,
+    fetch: fetch2
+  });
+  const createLanguageModel = (modelId) => createChatModel(modelId);
+  const createChatModel = (modelId) => new import_openai_compatible.OpenAICompatibleChatLanguageModel(modelId, {
+    ...getCommonModelConfig("chat"),
+    includeUsage: options.includeUsage,
+    supportsStructuredOutputs: options.supportsStructuredOutputs,
+    transformRequestBody: options.transformRequestBody,
+    metadataExtractor: options.metadataExtractor
+  });
+  const createCompletionModel = (modelId) => new import_openai_compatible.OpenAICompatibleCompletionLanguageModel(modelId, {
+    ...getCommonModelConfig("completion"),
+    includeUsage: options.includeUsage
+  });
+  const createEmbeddingModel = (modelId) => new import_openai_compatible.OpenAICompatibleEmbeddingModel(modelId, {
+    ...getCommonModelConfig("embedding")
+  });
+  const createImageModel = (modelId) => new import_openai_compatible.OpenAICompatibleImageModel(modelId, getCommonModelConfig("image"));
+  const provider = (modelId) => createLanguageModel(modelId);
+  provider.specificationVersion = "v3";
+  provider.languageModel = (modelId, config) => new import_openai_compatible.OpenAICompatibleChatLanguageModel(modelId, {
+    ...getCommonModelConfig("chat"),
+    includeUsage: options.includeUsage,
+    supportsStructuredOutputs: options.supportsStructuredOutputs,
+    transformRequestBody: options.transformRequestBody,
+    metadataExtractor: options.metadataExtractor,
+    ...config
+  });
+  provider.chatModel = createChatModel;
+  provider.completionModel = createCompletionModel;
+  provider.embeddingModel = createEmbeddingModel;
+  provider.textEmbeddingModel = createEmbeddingModel;
+  provider.imageModel = createImageModel;
+  return provider;
+}
+var vertexMaaS = createVertexMaaS();
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  VERSION,
+  createVertexMaaS,
+  vertexMaaS
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/vertex-maas-provider.ts","../src/version.ts","../src/vertex-maas-auth-fetch.ts","../src/vertex-maas-auth-google-auth-library.ts","../src/vertex-maas-base-url.ts"],"sourcesContent":["export { createVertexMaaS, vertexMaaS } from './vertex-maas-provider';\nexport type {\n  VertexMaaSProvider,\n  VertexMaaSProviderSettings,\n} from './vertex-maas-provider';\nexport { VERSION } from './version';\n","import {\n  MetadataExtractor,\n  OpenAICompatibleProvider,\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  OpenAICompatibleImageModel,\n} from '@ai-sdk/openai-compatible';\nimport {\n  FetchFunction,\n  loadSetting,\n  withoutTrailingSlash,\n  withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport type { GoogleAuthOptions } from 'google-auth-library';\nimport { VERSION } from './version';\nimport { createVertexMaaSAuthFetch } from './vertex-maas-auth-fetch';\nimport { getVertexOpenAICompatibleBaseURL } from './vertex-maas-base-url';\nimport type {\n  EmbeddingModelV3,\n  ImageModelV3,\n  LanguageModelV3,\n} from '@ai-sdk/provider';\n\ntype OpenAICompatibleChatConfig = ConstructorParameters<\n  typeof OpenAICompatibleChatLanguageModel\n>[1];\n\nexport interface VertexMaaSProviderSettings {\n  /**\n   * Base URL for the API calls. If not set, it is derived from `project` and\n   * `location`.\n   */\n  baseURL?: string;\n\n  /**\n   * Vertex project id. Defaults to `GOOGLE_VERTEX_PROJECT` if `baseURL` is not\n   * provided.\n   */\n  project?: string;\n\n  /**\n   * Vertex location. Defaults to `GOOGLE_VERTEX_LOCATION` if `baseURL` is not\n   * provided.\n   */\n  location?: string;\n\n  /**\n   * API version used when computing baseURL. Defaults to `v1beta1`.\n   */\n  apiVersion?: 'v1' | 'v1beta1';\n\n  /**\n   * Override the base host for Vertex API calls (e.g. \"https://aiplatform.googleapis.com\").\n   * If not set, it is computed from location.\n   */\n  apiBase?: string;\n\n  /**\n   * Optional. Use a static bearer token instead of ADC.\n   */\n  apiKey?: string;\n\n  /**\n   * Optional. Passed to google-auth-library for ADC token generation.\n   */\n  googleAuthOptions?: GoogleAuthOptions;\n\n  /**\n   * Optional custom headers.\n   */\n  headers?: Record<string, string>;\n\n  /**\n   * Optional custom url query parameters to include in request urls.\n   */\n  queryParams?: Record<string, string>;\n\n  /**\n   * Custom fetch implementation.\n   */\n  fetch?: FetchFunction;\n\n  includeUsage?: boolean;\n  supportsStructuredOutputs?: boolean;\n  transformRequestBody?: (args: Record<string, any>) => Record<string, any>;\n  metadataExtractor?: MetadataExtractor;\n}\n\nexport type VertexMaaSProvider = OpenAICompatibleProvider;\n\nexport function createVertexMaaS(\n  options: VertexMaaSProviderSettings = {},\n): VertexMaaSProvider {\n  const providerName = 'vertexMaas';\n\n  let baseURLCache: string | null = null;\n  const getBaseURL = () => {\n    if (baseURLCache != null) {\n      return baseURLCache;\n    }\n\n    baseURLCache =\n      withoutTrailingSlash(options.baseURL) ??\n      getVertexOpenAICompatibleBaseURL({\n        project: loadSetting({\n          settingValue: options.project,\n          settingName: 'project',\n          environmentVariableName: 'GOOGLE_VERTEX_PROJECT',\n          description: 'Google Vertex project',\n        }),\n        location: loadSetting({\n          settingValue: options.location,\n          settingName: 'location',\n          environmentVariableName: 'GOOGLE_VERTEX_LOCATION',\n          description: 'Google Vertex location',\n        }),\n        apiVersion: options.apiVersion ?? 'v1beta1',\n        apiBase: options.apiBase,\n      });\n\n    return baseURLCache;\n  };\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const staticHeaders = {\n    ...(options.apiKey && { Authorization: `Bearer ${options.apiKey}` }),\n    ...options.headers,\n  };\n\n  const getHeaders = () =>\n    withUserAgentSuffix(staticHeaders, `ai-sdk/vertex-maas/${VERSION}`);\n\n  const fetch =\n    options.apiKey != null\n      ? options.fetch\n      : createVertexMaaSAuthFetch({\n          customFetch: options.fetch,\n          googleAuthOptions: options.googleAuthOptions,\n        });\n\n  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({\n    provider: `${providerName}.${modelType}`,\n    url: ({ path }) => {\n      const url = new URL(`${getBaseURL()}${path}`);\n      if (options.queryParams) {\n        url.search = new URLSearchParams(options.queryParams).toString();\n      }\n      return url.toString();\n    },\n    headers: getHeaders,\n    fetch,\n  });\n\n  const createLanguageModel = (modelId: string) => createChatModel(modelId);\n\n  const createChatModel = (modelId: string) =>\n    new OpenAICompatibleChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      includeUsage: options.includeUsage,\n      supportsStructuredOutputs: options.supportsStructuredOutputs,\n      transformRequestBody: options.transformRequestBody,\n      metadataExtractor: options.metadataExtractor,\n    });\n\n  const createCompletionModel = (modelId: string) =>\n    new OpenAICompatibleCompletionLanguageModel(modelId, {\n      ...getCommonModelConfig('completion'),\n      includeUsage: options.includeUsage,\n    });\n\n  const createEmbeddingModel = (modelId: string) =>\n    new OpenAICompatibleEmbeddingModel(modelId, {\n      ...getCommonModelConfig('embedding'),\n    });\n\n  const createImageModel = (modelId: string) =>\n    new OpenAICompatibleImageModel(modelId, getCommonModelConfig('image'));\n\n  const provider = (modelId: string) => createLanguageModel(modelId);\n\n  provider.specificationVersion = 'v3' as const;\n  provider.languageModel = (\n    modelId: string,\n    config?: Partial<OpenAICompatibleChatConfig>,\n  ): LanguageModelV3 =>\n    new OpenAICompatibleChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      includeUsage: options.includeUsage,\n      supportsStructuredOutputs: options.supportsStructuredOutputs,\n      transformRequestBody: options.transformRequestBody,\n      metadataExtractor: options.metadataExtractor,\n      ...config,\n    });\n  provider.chatModel = createChatModel;\n  provider.completionModel = createCompletionModel;\n  provider.embeddingModel = createEmbeddingModel as (\n    modelId: string,\n  ) => EmbeddingModelV3;\n  provider.textEmbeddingModel = createEmbeddingModel as (\n    modelId: string,\n  ) => EmbeddingModelV3;\n  provider.imageModel = createImageModel as (modelId: string) => ImageModelV3;\n\n  return provider as unknown as OpenAICompatibleProvider;\n}\n\n/**\n * Default Vertex MaaS provider instance.\n */\nexport const vertexMaaS = createVertexMaaS();\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 { FetchFunction, normalizeHeaders } from '@ai-sdk/provider-utils';\nimport type { GoogleAuthOptions } from 'google-auth-library';\nimport { generateAuthToken } from './vertex-maas-auth-google-auth-library';\n\nexport function createVertexMaaSAuthFetch({\n  customFetch,\n  googleAuthOptions,\n}: {\n  customFetch?: FetchFunction;\n  googleAuthOptions?: GoogleAuthOptions;\n}): FetchFunction {\n  return async (url, init) => {\n    const existingHeaders = normalizeHeaders(init?.headers as any);\n\n    // Respect caller-provided auth (including static api keys).\n    if (existingHeaders.authorization != null) {\n      return (customFetch ?? fetch)(url as any, init);\n    }\n\n    const token = await generateAuthToken(googleAuthOptions);\n    if (!token) {\n      throw new Error('Failed to get access token from ADC');\n    }\n\n    const modifiedInit: RequestInit = {\n      ...init,\n      headers: {\n        ...existingHeaders,\n        authorization: `Bearer ${token}`,\n      },\n    };\n\n    return (customFetch ?? fetch)(url as any, modifiedInit);\n  };\n}\n","import { GoogleAuth, GoogleAuthOptions } 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 { withoutTrailingSlash } from '@ai-sdk/provider-utils';\n\nfunction getDefaultVertexApiBase(location: string): string {\n  return location === 'global'\n    ? 'https://aiplatform.googleapis.com'\n    : `https://${location}-aiplatform.googleapis.com`;\n}\n\nexport function getVertexOpenAICompatibleBaseURL({\n  project,\n  location,\n  apiVersion,\n  apiBase,\n}: {\n  project: string;\n  location: string;\n  apiVersion: 'v1' | 'v1beta1';\n  apiBase?: string;\n}): string {\n  const resolvedApiBase = withoutTrailingSlash(\n    apiBase ?? getDefaultVertexApiBase(location),\n  );\n\n  return `${resolvedApiBase}/${apiVersion}/projects/${project}/locations/${location}/endpoints/openapi`;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAOO;AACP,IAAAA,yBAKO;;;ACXA,IAAM,UACX,OACI,UACA;;;ACLN,4BAAgD;;;ACAhD,iCAA8C;AAE9C,IAAI,eAAkC;AACtC,IAAI,cAAwC;AAE5C,SAAS,QAAQ,SAA4B;AAC3C,MAAI,CAAC,gBAAgB,YAAY,aAAa;AAC5C,mBAAe,IAAI,sCAAW;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;;;ADjBO,SAAS,0BAA0B;AAAA,EACxC;AAAA,EACA;AACF,GAGkB;AAChB,SAAO,OAAO,KAAK,SAAS;AAC1B,UAAM,sBAAkB,wCAAiB,6BAAM,OAAc;AAG7D,QAAI,gBAAgB,iBAAiB,MAAM;AACzC,cAAQ,oCAAe,OAAO,KAAY,IAAI;AAAA,IAChD;AAEA,UAAM,QAAQ,MAAM,kBAAkB,iBAAiB;AACvD,QAAI,CAAC,OAAO;AACV,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AAEA,UAAM,eAA4B;AAAA,MAChC,GAAG;AAAA,MACH,SAAS;AAAA,QACP,GAAG;AAAA,QACH,eAAe,UAAU,KAAK;AAAA,MAChC;AAAA,IACF;AAEA,YAAQ,oCAAe,OAAO,KAAY,YAAY;AAAA,EACxD;AACF;;;AElCA,IAAAC,yBAAqC;AAErC,SAAS,wBAAwB,UAA0B;AACzD,SAAO,aAAa,WAChB,sCACA,WAAW,QAAQ;AACzB;AAEO,SAAS,iCAAiC;AAAA,EAC/C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAKW;AACT,QAAM,sBAAkB;AAAA,IACtB,4BAAW,wBAAwB,QAAQ;AAAA,EAC7C;AAEA,SAAO,GAAG,eAAe,IAAI,UAAU,aAAa,OAAO,cAAc,QAAQ;AACnF;;;AJmEO,SAAS,iBACd,UAAsC,CAAC,GACnB;AACpB,QAAM,eAAe;AAErB,MAAI,eAA8B;AAClC,QAAM,aAAa,MAAM;AAjG3B;AAkGI,QAAI,gBAAgB,MAAM;AACxB,aAAO;AAAA,IACT;AAEA,oBACE,sDAAqB,QAAQ,OAAO,MAApC,YACA,iCAAiC;AAAA,MAC/B,aAAS,oCAAY;AAAA,QACnB,cAAc,QAAQ;AAAA,QACtB,aAAa;AAAA,QACb,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC;AAAA,MACD,cAAU,oCAAY;AAAA,QACpB,cAAc,QAAQ;AAAA,QACtB,aAAa;AAAA,QACb,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC;AAAA,MACD,aAAY,aAAQ,eAAR,YAAsB;AAAA,MAClC,SAAS,QAAQ;AAAA,IACnB,CAAC;AAEH,WAAO;AAAA,EACT;AASA,QAAM,gBAAgB;AAAA,IACpB,GAAI,QAAQ,UAAU,EAAE,eAAe,UAAU,QAAQ,MAAM,GAAG;AAAA,IAClE,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,aAAa,UACjB,4CAAoB,eAAe,sBAAsB,OAAO,EAAE;AAEpE,QAAMC,SACJ,QAAQ,UAAU,OACd,QAAQ,QACR,0BAA0B;AAAA,IACxB,aAAa,QAAQ;AAAA,IACrB,mBAAmB,QAAQ;AAAA,EAC7B,CAAC;AAEP,QAAM,uBAAuB,CAAC,eAA0C;AAAA,IACtE,UAAU,GAAG,YAAY,IAAI,SAAS;AAAA,IACtC,KAAK,CAAC,EAAE,KAAK,MAAM;AACjB,YAAM,MAAM,IAAI,IAAI,GAAG,WAAW,CAAC,GAAG,IAAI,EAAE;AAC5C,UAAI,QAAQ,aAAa;AACvB,YAAI,SAAS,IAAI,gBAAgB,QAAQ,WAAW,EAAE,SAAS;AAAA,MACjE;AACA,aAAO,IAAI,SAAS;AAAA,IACtB;AAAA,IACA,SAAS;AAAA,IACT,OAAAA;AAAA,EACF;AAEA,QAAM,sBAAsB,CAAC,YAAoB,gBAAgB,OAAO;AAExE,QAAM,kBAAkB,CAAC,YACvB,IAAI,2DAAkC,SAAS;AAAA,IAC7C,GAAG,qBAAqB,MAAM;AAAA,IAC9B,cAAc,QAAQ;AAAA,IACtB,2BAA2B,QAAQ;AAAA,IACnC,sBAAsB,QAAQ;AAAA,IAC9B,mBAAmB,QAAQ;AAAA,EAC7B,CAAC;AAEH,QAAM,wBAAwB,CAAC,YAC7B,IAAI,iEAAwC,SAAS;AAAA,IACnD,GAAG,qBAAqB,YAAY;AAAA,IACpC,cAAc,QAAQ;AAAA,EACxB,CAAC;AAEH,QAAM,uBAAuB,CAAC,YAC5B,IAAI,wDAA+B,SAAS;AAAA,IAC1C,GAAG,qBAAqB,WAAW;AAAA,EACrC,CAAC;AAEH,QAAM,mBAAmB,CAAC,YACxB,IAAI,oDAA2B,SAAS,qBAAqB,OAAO,CAAC;AAEvE,QAAM,WAAW,CAAC,YAAoB,oBAAoB,OAAO;AAEjE,WAAS,uBAAuB;AAChC,WAAS,gBAAgB,CACvB,SACA,WAEA,IAAI,2DAAkC,SAAS;AAAA,IAC7C,GAAG,qBAAqB,MAAM;AAAA,IAC9B,cAAc,QAAQ;AAAA,IACtB,2BAA2B,QAAQ;AAAA,IACnC,sBAAsB,QAAQ;AAAA,IAC9B,mBAAmB,QAAQ;AAAA,IAC3B,GAAG;AAAA,EACL,CAAC;AACH,WAAS,YAAY;AACrB,WAAS,kBAAkB;AAC3B,WAAS,iBAAiB;AAG1B,WAAS,qBAAqB;AAG9B,WAAS,aAAa;AAEtB,SAAO;AACT;AAKO,IAAM,aAAa,iBAAiB;","names":["import_provider_utils","import_provider_utils","fetch"]}

package/dist/index.mjs

@@ -0,0 +1,170 @@
+// src/vertex-maas-provider.ts
+import {
+  OpenAICompatibleChatLanguageModel,
+  OpenAICompatibleCompletionLanguageModel,
+  OpenAICompatibleEmbeddingModel,
+  OpenAICompatibleImageModel
+} from "@ai-sdk/openai-compatible";
+import {
+  loadSetting,
+  withoutTrailingSlash as withoutTrailingSlash2,
+  withUserAgentSuffix
+} from "@ai-sdk/provider-utils";
+
+// src/version.ts
+var VERSION = true ? "0.0.1" : "0.0.0-test";
+
+// src/vertex-maas-auth-fetch.ts
+import { normalizeHeaders } from "@ai-sdk/provider-utils";
+
+// src/vertex-maas-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/vertex-maas-auth-fetch.ts
+function createVertexMaaSAuthFetch({
+  customFetch,
+  googleAuthOptions
+}) {
+  return async (url, init) => {
+    const existingHeaders = normalizeHeaders(init == null ? void 0 : init.headers);
+    if (existingHeaders.authorization != null) {
+      return (customFetch != null ? customFetch : fetch)(url, init);
+    }
+    const token = await generateAuthToken(googleAuthOptions);
+    if (!token) {
+      throw new Error("Failed to get access token from ADC");
+    }
+    const modifiedInit = {
+      ...init,
+      headers: {
+        ...existingHeaders,
+        authorization: `Bearer ${token}`
+      }
+    };
+    return (customFetch != null ? customFetch : fetch)(url, modifiedInit);
+  };
+}
+
+// src/vertex-maas-base-url.ts
+import { withoutTrailingSlash } from "@ai-sdk/provider-utils";
+function getDefaultVertexApiBase(location) {
+  return location === "global" ? "https://aiplatform.googleapis.com" : `https://${location}-aiplatform.googleapis.com`;
+}
+function getVertexOpenAICompatibleBaseURL({
+  project,
+  location,
+  apiVersion,
+  apiBase
+}) {
+  const resolvedApiBase = withoutTrailingSlash(
+    apiBase != null ? apiBase : getDefaultVertexApiBase(location)
+  );
+  return `${resolvedApiBase}/${apiVersion}/projects/${project}/locations/${location}/endpoints/openapi`;
+}
+
+// src/vertex-maas-provider.ts
+function createVertexMaaS(options = {}) {
+  const providerName = "vertexMaas";
+  let baseURLCache = null;
+  const getBaseURL = () => {
+    var _a, _b;
+    if (baseURLCache != null) {
+      return baseURLCache;
+    }
+    baseURLCache = (_b = withoutTrailingSlash2(options.baseURL)) != null ? _b : getVertexOpenAICompatibleBaseURL({
+      project: loadSetting({
+        settingValue: options.project,
+        settingName: "project",
+        environmentVariableName: "GOOGLE_VERTEX_PROJECT",
+        description: "Google Vertex project"
+      }),
+      location: loadSetting({
+        settingValue: options.location,
+        settingName: "location",
+        environmentVariableName: "GOOGLE_VERTEX_LOCATION",
+        description: "Google Vertex location"
+      }),
+      apiVersion: (_a = options.apiVersion) != null ? _a : "v1beta1",
+      apiBase: options.apiBase
+    });
+    return baseURLCache;
+  };
+  const staticHeaders = {
+    ...options.apiKey && { Authorization: `Bearer ${options.apiKey}` },
+    ...options.headers
+  };
+  const getHeaders = () => withUserAgentSuffix(staticHeaders, `ai-sdk/vertex-maas/${VERSION}`);
+  const fetch2 = options.apiKey != null ? options.fetch : createVertexMaaSAuthFetch({
+    customFetch: options.fetch,
+    googleAuthOptions: options.googleAuthOptions
+  });
+  const getCommonModelConfig = (modelType) => ({
+    provider: `${providerName}.${modelType}`,
+    url: ({ path }) => {
+      const url = new URL(`${getBaseURL()}${path}`);
+      if (options.queryParams) {
+        url.search = new URLSearchParams(options.queryParams).toString();
+      }
+      return url.toString();
+    },
+    headers: getHeaders,
+    fetch: fetch2
+  });
+  const createLanguageModel = (modelId) => createChatModel(modelId);
+  const createChatModel = (modelId) => new OpenAICompatibleChatLanguageModel(modelId, {
+    ...getCommonModelConfig("chat"),
+    includeUsage: options.includeUsage,
+    supportsStructuredOutputs: options.supportsStructuredOutputs,
+    transformRequestBody: options.transformRequestBody,
+    metadataExtractor: options.metadataExtractor
+  });
+  const createCompletionModel = (modelId) => new OpenAICompatibleCompletionLanguageModel(modelId, {
+    ...getCommonModelConfig("completion"),
+    includeUsage: options.includeUsage
+  });
+  const createEmbeddingModel = (modelId) => new OpenAICompatibleEmbeddingModel(modelId, {
+    ...getCommonModelConfig("embedding")
+  });
+  const createImageModel = (modelId) => new OpenAICompatibleImageModel(modelId, getCommonModelConfig("image"));
+  const provider = (modelId) => createLanguageModel(modelId);
+  provider.specificationVersion = "v3";
+  provider.languageModel = (modelId, config) => new OpenAICompatibleChatLanguageModel(modelId, {
+    ...getCommonModelConfig("chat"),
+    includeUsage: options.includeUsage,
+    supportsStructuredOutputs: options.supportsStructuredOutputs,
+    transformRequestBody: options.transformRequestBody,
+    metadataExtractor: options.metadataExtractor,
+    ...config
+  });
+  provider.chatModel = createChatModel;
+  provider.completionModel = createCompletionModel;
+  provider.embeddingModel = createEmbeddingModel;
+  provider.textEmbeddingModel = createEmbeddingModel;
+  provider.imageModel = createImageModel;
+  return provider;
+}
+var vertexMaaS = createVertexMaaS();
+export {
+  VERSION,
+  createVertexMaaS,
+  vertexMaaS
+};
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/vertex-maas-provider.ts","../src/version.ts","../src/vertex-maas-auth-fetch.ts","../src/vertex-maas-auth-google-auth-library.ts","../src/vertex-maas-base-url.ts"],"sourcesContent":["import {\n  MetadataExtractor,\n  OpenAICompatibleProvider,\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  OpenAICompatibleImageModel,\n} from '@ai-sdk/openai-compatible';\nimport {\n  FetchFunction,\n  loadSetting,\n  withoutTrailingSlash,\n  withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport type { GoogleAuthOptions } from 'google-auth-library';\nimport { VERSION } from './version';\nimport { createVertexMaaSAuthFetch } from './vertex-maas-auth-fetch';\nimport { getVertexOpenAICompatibleBaseURL } from './vertex-maas-base-url';\nimport type {\n  EmbeddingModelV3,\n  ImageModelV3,\n  LanguageModelV3,\n} from '@ai-sdk/provider';\n\ntype OpenAICompatibleChatConfig = ConstructorParameters<\n  typeof OpenAICompatibleChatLanguageModel\n>[1];\n\nexport interface VertexMaaSProviderSettings {\n  /**\n   * Base URL for the API calls. If not set, it is derived from `project` and\n   * `location`.\n   */\n  baseURL?: string;\n\n  /**\n   * Vertex project id. Defaults to `GOOGLE_VERTEX_PROJECT` if `baseURL` is not\n   * provided.\n   */\n  project?: string;\n\n  /**\n   * Vertex location. Defaults to `GOOGLE_VERTEX_LOCATION` if `baseURL` is not\n   * provided.\n   */\n  location?: string;\n\n  /**\n   * API version used when computing baseURL. Defaults to `v1beta1`.\n   */\n  apiVersion?: 'v1' | 'v1beta1';\n\n  /**\n   * Override the base host for Vertex API calls (e.g. \"https://aiplatform.googleapis.com\").\n   * If not set, it is computed from location.\n   */\n  apiBase?: string;\n\n  /**\n   * Optional. Use a static bearer token instead of ADC.\n   */\n  apiKey?: string;\n\n  /**\n   * Optional. Passed to google-auth-library for ADC token generation.\n   */\n  googleAuthOptions?: GoogleAuthOptions;\n\n  /**\n   * Optional custom headers.\n   */\n  headers?: Record<string, string>;\n\n  /**\n   * Optional custom url query parameters to include in request urls.\n   */\n  queryParams?: Record<string, string>;\n\n  /**\n   * Custom fetch implementation.\n   */\n  fetch?: FetchFunction;\n\n  includeUsage?: boolean;\n  supportsStructuredOutputs?: boolean;\n  transformRequestBody?: (args: Record<string, any>) => Record<string, any>;\n  metadataExtractor?: MetadataExtractor;\n}\n\nexport type VertexMaaSProvider = OpenAICompatibleProvider;\n\nexport function createVertexMaaS(\n  options: VertexMaaSProviderSettings = {},\n): VertexMaaSProvider {\n  const providerName = 'vertexMaas';\n\n  let baseURLCache: string | null = null;\n  const getBaseURL = () => {\n    if (baseURLCache != null) {\n      return baseURLCache;\n    }\n\n    baseURLCache =\n      withoutTrailingSlash(options.baseURL) ??\n      getVertexOpenAICompatibleBaseURL({\n        project: loadSetting({\n          settingValue: options.project,\n          settingName: 'project',\n          environmentVariableName: 'GOOGLE_VERTEX_PROJECT',\n          description: 'Google Vertex project',\n        }),\n        location: loadSetting({\n          settingValue: options.location,\n          settingName: 'location',\n          environmentVariableName: 'GOOGLE_VERTEX_LOCATION',\n          description: 'Google Vertex location',\n        }),\n        apiVersion: options.apiVersion ?? 'v1beta1',\n        apiBase: options.apiBase,\n      });\n\n    return baseURLCache;\n  };\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const staticHeaders = {\n    ...(options.apiKey && { Authorization: `Bearer ${options.apiKey}` }),\n    ...options.headers,\n  };\n\n  const getHeaders = () =>\n    withUserAgentSuffix(staticHeaders, `ai-sdk/vertex-maas/${VERSION}`);\n\n  const fetch =\n    options.apiKey != null\n      ? options.fetch\n      : createVertexMaaSAuthFetch({\n          customFetch: options.fetch,\n          googleAuthOptions: options.googleAuthOptions,\n        });\n\n  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({\n    provider: `${providerName}.${modelType}`,\n    url: ({ path }) => {\n      const url = new URL(`${getBaseURL()}${path}`);\n      if (options.queryParams) {\n        url.search = new URLSearchParams(options.queryParams).toString();\n      }\n      return url.toString();\n    },\n    headers: getHeaders,\n    fetch,\n  });\n\n  const createLanguageModel = (modelId: string) => createChatModel(modelId);\n\n  const createChatModel = (modelId: string) =>\n    new OpenAICompatibleChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      includeUsage: options.includeUsage,\n      supportsStructuredOutputs: options.supportsStructuredOutputs,\n      transformRequestBody: options.transformRequestBody,\n      metadataExtractor: options.metadataExtractor,\n    });\n\n  const createCompletionModel = (modelId: string) =>\n    new OpenAICompatibleCompletionLanguageModel(modelId, {\n      ...getCommonModelConfig('completion'),\n      includeUsage: options.includeUsage,\n    });\n\n  const createEmbeddingModel = (modelId: string) =>\n    new OpenAICompatibleEmbeddingModel(modelId, {\n      ...getCommonModelConfig('embedding'),\n    });\n\n  const createImageModel = (modelId: string) =>\n    new OpenAICompatibleImageModel(modelId, getCommonModelConfig('image'));\n\n  const provider = (modelId: string) => createLanguageModel(modelId);\n\n  provider.specificationVersion = 'v3' as const;\n  provider.languageModel = (\n    modelId: string,\n    config?: Partial<OpenAICompatibleChatConfig>,\n  ): LanguageModelV3 =>\n    new OpenAICompatibleChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      includeUsage: options.includeUsage,\n      supportsStructuredOutputs: options.supportsStructuredOutputs,\n      transformRequestBody: options.transformRequestBody,\n      metadataExtractor: options.metadataExtractor,\n      ...config,\n    });\n  provider.chatModel = createChatModel;\n  provider.completionModel = createCompletionModel;\n  provider.embeddingModel = createEmbeddingModel as (\n    modelId: string,\n  ) => EmbeddingModelV3;\n  provider.textEmbeddingModel = createEmbeddingModel as (\n    modelId: string,\n  ) => EmbeddingModelV3;\n  provider.imageModel = createImageModel as (modelId: string) => ImageModelV3;\n\n  return provider as unknown as OpenAICompatibleProvider;\n}\n\n/**\n * Default Vertex MaaS provider instance.\n */\nexport const vertexMaaS = createVertexMaaS();\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 { FetchFunction, normalizeHeaders } from '@ai-sdk/provider-utils';\nimport type { GoogleAuthOptions } from 'google-auth-library';\nimport { generateAuthToken } from './vertex-maas-auth-google-auth-library';\n\nexport function createVertexMaaSAuthFetch({\n  customFetch,\n  googleAuthOptions,\n}: {\n  customFetch?: FetchFunction;\n  googleAuthOptions?: GoogleAuthOptions;\n}): FetchFunction {\n  return async (url, init) => {\n    const existingHeaders = normalizeHeaders(init?.headers as any);\n\n    // Respect caller-provided auth (including static api keys).\n    if (existingHeaders.authorization != null) {\n      return (customFetch ?? fetch)(url as any, init);\n    }\n\n    const token = await generateAuthToken(googleAuthOptions);\n    if (!token) {\n      throw new Error('Failed to get access token from ADC');\n    }\n\n    const modifiedInit: RequestInit = {\n      ...init,\n      headers: {\n        ...existingHeaders,\n        authorization: `Bearer ${token}`,\n      },\n    };\n\n    return (customFetch ?? fetch)(url as any, modifiedInit);\n  };\n}\n","import { GoogleAuth, GoogleAuthOptions } 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 { withoutTrailingSlash } from '@ai-sdk/provider-utils';\n\nfunction getDefaultVertexApiBase(location: string): string {\n  return location === 'global'\n    ? 'https://aiplatform.googleapis.com'\n    : `https://${location}-aiplatform.googleapis.com`;\n}\n\nexport function getVertexOpenAICompatibleBaseURL({\n  project,\n  location,\n  apiVersion,\n  apiBase,\n}: {\n  project: string;\n  location: string;\n  apiVersion: 'v1' | 'v1beta1';\n  apiBase?: string;\n}): string {\n  const resolvedApiBase = withoutTrailingSlash(\n    apiBase ?? getDefaultVertexApiBase(location),\n  );\n\n  return `${resolvedApiBase}/${apiVersion}/projects/${project}/locations/${location}/endpoints/openapi`;\n}\n"],"mappings":";AAAA;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA,wBAAAA;AAAA,EACA;AAAA,OACK;;;ACXA,IAAM,UACX,OACI,UACA;;;ACLN,SAAwB,wBAAwB;;;ACAhD,SAAS,kBAAqC;AAE9C,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;;;ADjBO,SAAS,0BAA0B;AAAA,EACxC;AAAA,EACA;AACF,GAGkB;AAChB,SAAO,OAAO,KAAK,SAAS;AAC1B,UAAM,kBAAkB,iBAAiB,6BAAM,OAAc;AAG7D,QAAI,gBAAgB,iBAAiB,MAAM;AACzC,cAAQ,oCAAe,OAAO,KAAY,IAAI;AAAA,IAChD;AAEA,UAAM,QAAQ,MAAM,kBAAkB,iBAAiB;AACvD,QAAI,CAAC,OAAO;AACV,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AAEA,UAAM,eAA4B;AAAA,MAChC,GAAG;AAAA,MACH,SAAS;AAAA,QACP,GAAG;AAAA,QACH,eAAe,UAAU,KAAK;AAAA,MAChC;AAAA,IACF;AAEA,YAAQ,oCAAe,OAAO,KAAY,YAAY;AAAA,EACxD;AACF;;;AElCA,SAAS,4BAA4B;AAErC,SAAS,wBAAwB,UAA0B;AACzD,SAAO,aAAa,WAChB,sCACA,WAAW,QAAQ;AACzB;AAEO,SAAS,iCAAiC;AAAA,EAC/C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAKW;AACT,QAAM,kBAAkB;AAAA,IACtB,4BAAW,wBAAwB,QAAQ;AAAA,EAC7C;AAEA,SAAO,GAAG,eAAe,IAAI,UAAU,aAAa,OAAO,cAAc,QAAQ;AACnF;;;AJmEO,SAAS,iBACd,UAAsC,CAAC,GACnB;AACpB,QAAM,eAAe;AAErB,MAAI,eAA8B;AAClC,QAAM,aAAa,MAAM;AAjG3B;AAkGI,QAAI,gBAAgB,MAAM;AACxB,aAAO;AAAA,IACT;AAEA,oBACE,KAAAC,sBAAqB,QAAQ,OAAO,MAApC,YACA,iCAAiC;AAAA,MAC/B,SAAS,YAAY;AAAA,QACnB,cAAc,QAAQ;AAAA,QACtB,aAAa;AAAA,QACb,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC;AAAA,MACD,UAAU,YAAY;AAAA,QACpB,cAAc,QAAQ;AAAA,QACtB,aAAa;AAAA,QACb,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC;AAAA,MACD,aAAY,aAAQ,eAAR,YAAsB;AAAA,MAClC,SAAS,QAAQ;AAAA,IACnB,CAAC;AAEH,WAAO;AAAA,EACT;AASA,QAAM,gBAAgB;AAAA,IACpB,GAAI,QAAQ,UAAU,EAAE,eAAe,UAAU,QAAQ,MAAM,GAAG;AAAA,IAClE,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,aAAa,MACjB,oBAAoB,eAAe,sBAAsB,OAAO,EAAE;AAEpE,QAAMC,SACJ,QAAQ,UAAU,OACd,QAAQ,QACR,0BAA0B;AAAA,IACxB,aAAa,QAAQ;AAAA,IACrB,mBAAmB,QAAQ;AAAA,EAC7B,CAAC;AAEP,QAAM,uBAAuB,CAAC,eAA0C;AAAA,IACtE,UAAU,GAAG,YAAY,IAAI,SAAS;AAAA,IACtC,KAAK,CAAC,EAAE,KAAK,MAAM;AACjB,YAAM,MAAM,IAAI,IAAI,GAAG,WAAW,CAAC,GAAG,IAAI,EAAE;AAC5C,UAAI,QAAQ,aAAa;AACvB,YAAI,SAAS,IAAI,gBAAgB,QAAQ,WAAW,EAAE,SAAS;AAAA,MACjE;AACA,aAAO,IAAI,SAAS;AAAA,IACtB;AAAA,IACA,SAAS;AAAA,IACT,OAAAA;AAAA,EACF;AAEA,QAAM,sBAAsB,CAAC,YAAoB,gBAAgB,OAAO;AAExE,QAAM,kBAAkB,CAAC,YACvB,IAAI,kCAAkC,SAAS;AAAA,IAC7C,GAAG,qBAAqB,MAAM;AAAA,IAC9B,cAAc,QAAQ;AAAA,IACtB,2BAA2B,QAAQ;AAAA,IACnC,sBAAsB,QAAQ;AAAA,IAC9B,mBAAmB,QAAQ;AAAA,EAC7B,CAAC;AAEH,QAAM,wBAAwB,CAAC,YAC7B,IAAI,wCAAwC,SAAS;AAAA,IACnD,GAAG,qBAAqB,YAAY;AAAA,IACpC,cAAc,QAAQ;AAAA,EACxB,CAAC;AAEH,QAAM,uBAAuB,CAAC,YAC5B,IAAI,+BAA+B,SAAS;AAAA,IAC1C,GAAG,qBAAqB,WAAW;AAAA,EACrC,CAAC;AAEH,QAAM,mBAAmB,CAAC,YACxB,IAAI,2BAA2B,SAAS,qBAAqB,OAAO,CAAC;AAEvE,QAAM,WAAW,CAAC,YAAoB,oBAAoB,OAAO;AAEjE,WAAS,uBAAuB;AAChC,WAAS,gBAAgB,CACvB,SACA,WAEA,IAAI,kCAAkC,SAAS;AAAA,IAC7C,GAAG,qBAAqB,MAAM;AAAA,IAC9B,cAAc,QAAQ;AAAA,IACtB,2BAA2B,QAAQ;AAAA,IACnC,sBAAsB,QAAQ;AAAA,IAC9B,mBAAmB,QAAQ;AAAA,IAC3B,GAAG;AAAA,EACL,CAAC;AACH,WAAS,YAAY;AACrB,WAAS,kBAAkB;AAC3B,WAAS,iBAAiB;AAG1B,WAAS,qBAAqB;AAG9B,WAAS,aAAa;AAEtB,SAAO;AACT;AAKO,IAAM,aAAa,iBAAiB;","names":["withoutTrailingSlash","withoutTrailingSlash","fetch"]}

package/package.json

@@ -0,0 +1,72 @@
+{
+  "name": "@amanm/vertex-maas",
+  "version": "0.0.1",
+  "license": "Apache-2.0",
+  "sideEffects": false,
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist/**/*",
+    "src",
+    "!src/**/*.test.ts",
+    "!src/**/*.test-d.ts",
+    "!src/**/__snapshots__",
+    "!src/**/__fixtures__",
+    "CHANGELOG.md",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
+    "build:watch": "pnpm clean && tsup --watch",
+    "clean": "del-cli dist *.tsbuildinfo",
+    "lint": "eslint \"./**/*.ts*\"",
+    "type-check": "tsc --build",
+    "prettier-check": "prettier --check \"./**/*.ts*\"",
+    "test": "vitest --config vitest.node.config.js --run",
+    "test:update": "vitest --config vitest.node.config.js -u",
+    "test:watch": "vitest --config vitest.node.config.js"
+  },
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "@ai-sdk/openai-compatible": "^2.0.30",
+    "@ai-sdk/provider": "^3.0.8",
+    "@ai-sdk/provider-utils": "^4.0.15",
+    "google-auth-library": "^10.5.0"
+  },
+  "devDependencies": {
+    "@ai-sdk/test-server": "workspace:*",
+    "@types/node": "20.17.24",
+    "@vercel/ai-tsconfig": "workspace:*",
+    "tsup": "^8",
+    "typescript": "5.8.3",
+    "zod": "3.25.76"
+  },
+  "peerDependencies": {
+    "zod": "^3.25.76 || ^4.1.8"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "homepage": "https://ai-sdk.dev/docs",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/vercel/ai.git"
+  },
+  "bugs": {
+    "url": "https://github.com/vercel/ai/issues"
+  },
+  "keywords": [
+    "ai"
+  ]
+}

package/src/index.ts

@@ -0,0 +1,6 @@
+export { createVertexMaaS, vertexMaaS } from './vertex-maas-provider';
+export type {
+  VertexMaaSProvider,
+  VertexMaaSProviderSettings,
+} from './vertex-maas-provider';
+export { VERSION } from './version';

package/src/version.ts

@@ -0,0 +1,6 @@
+// Version string of this package injected at build time.
+declare const __PACKAGE_VERSION__: string | undefined;
+export const VERSION: string =
+  typeof __PACKAGE_VERSION__ !== 'undefined'
+    ? __PACKAGE_VERSION__
+    : '0.0.0-test';

package/src/vertex-maas-auth-fetch.ts

@@ -0,0 +1,35 @@
+import { FetchFunction, normalizeHeaders } from '@ai-sdk/provider-utils';
+import type { GoogleAuthOptions } from 'google-auth-library';
+import { generateAuthToken } from './vertex-maas-auth-google-auth-library';
+
+export function createVertexMaaSAuthFetch({
+  customFetch,
+  googleAuthOptions,
+}: {
+  customFetch?: FetchFunction;
+  googleAuthOptions?: GoogleAuthOptions;
+}): FetchFunction {
+  return async (url, init) => {
+    const existingHeaders = normalizeHeaders(init?.headers as any);
+
+    // Respect caller-provided auth (including static api keys).
+    if (existingHeaders.authorization != null) {
+      return (customFetch ?? fetch)(url as any, init);
+    }
+
+    const token = await generateAuthToken(googleAuthOptions);
+    if (!token) {
+      throw new Error('Failed to get access token from ADC');
+    }
+
+    const modifiedInit: RequestInit = {
+      ...init,
+      headers: {
+        ...existingHeaders,
+        authorization: `Bearer ${token}`,
+      },
+    };
+
+    return (customFetch ?? fetch)(url as any, modifiedInit);
+  };
+}

package/src/vertex-maas-auth-google-auth-library.ts

@@ -0,0 +1,27 @@
+import { GoogleAuth, GoogleAuthOptions } from 'google-auth-library';
+
+let authInstance: GoogleAuth | null = null;
+let authOptions: GoogleAuthOptions | null = null;
+
+function getAuth(options: GoogleAuthOptions) {
+  if (!authInstance || options !== authOptions) {
+    authInstance = new GoogleAuth({
+      scopes: ['https://www.googleapis.com/auth/cloud-platform'],
+      ...options,
+    });
+    authOptions = options;
+  }
+  return authInstance;
+}
+
+export async function generateAuthToken(options?: GoogleAuthOptions) {
+  const auth = getAuth(options || {});
+  const client = await auth.getClient();
+  const token = await client.getAccessToken();
+  return token?.token || null;
+}
+
+// For testing purposes only
+export function _resetAuthInstance() {
+  authInstance = null;
+}

package/src/vertex-maas-base-url.ts

@@ -0,0 +1,25 @@
+import { withoutTrailingSlash } from '@ai-sdk/provider-utils';
+
+function getDefaultVertexApiBase(location: string): string {
+  return location === 'global'
+    ? 'https://aiplatform.googleapis.com'
+    : `https://${location}-aiplatform.googleapis.com`;
+}
+
+export function getVertexOpenAICompatibleBaseURL({
+  project,
+  location,
+  apiVersion,
+  apiBase,
+}: {
+  project: string;
+  location: string;
+  apiVersion: 'v1' | 'v1beta1';
+  apiBase?: string;
+}): string {
+  const resolvedApiBase = withoutTrailingSlash(
+    apiBase ?? getDefaultVertexApiBase(location),
+  );
+
+  return `${resolvedApiBase}/${apiVersion}/projects/${project}/locations/${location}/endpoints/openapi`;
+}

package/src/vertex-maas-provider.ts

@@ -0,0 +1,217 @@
+import {
+  MetadataExtractor,
+  OpenAICompatibleProvider,
+  OpenAICompatibleChatLanguageModel,
+  OpenAICompatibleCompletionLanguageModel,
+  OpenAICompatibleEmbeddingModel,
+  OpenAICompatibleImageModel,
+} from '@ai-sdk/openai-compatible';
+import {
+  FetchFunction,
+  loadSetting,
+  withoutTrailingSlash,
+  withUserAgentSuffix,
+} from '@ai-sdk/provider-utils';
+import type { GoogleAuthOptions } from 'google-auth-library';
+import { VERSION } from './version';
+import { createVertexMaaSAuthFetch } from './vertex-maas-auth-fetch';
+import { getVertexOpenAICompatibleBaseURL } from './vertex-maas-base-url';
+import type {
+  EmbeddingModelV3,
+  ImageModelV3,
+  LanguageModelV3,
+} from '@ai-sdk/provider';
+
+type OpenAICompatibleChatConfig = ConstructorParameters<
+  typeof OpenAICompatibleChatLanguageModel
+>[1];
+
+export interface VertexMaaSProviderSettings {
+  /**
+   * Base URL for the API calls. If not set, it is derived from `project` and
+   * `location`.
+   */
+  baseURL?: string;
+
+  /**
+   * Vertex project id. Defaults to `GOOGLE_VERTEX_PROJECT` if `baseURL` is not
+   * provided.
+   */
+  project?: string;
+
+  /**
+   * Vertex location. Defaults to `GOOGLE_VERTEX_LOCATION` if `baseURL` is not
+   * provided.
+   */
+  location?: string;
+
+  /**
+   * API version used when computing baseURL. Defaults to `v1beta1`.
+   */
+  apiVersion?: 'v1' | 'v1beta1';
+
+  /**
+   * Override the base host for Vertex API calls (e.g. "https://aiplatform.googleapis.com").
+   * If not set, it is computed from location.
+   */
+  apiBase?: string;
+
+  /**
+   * Optional. Use a static bearer token instead of ADC.
+   */
+  apiKey?: string;
+
+  /**
+   * Optional. Passed to google-auth-library for ADC token generation.
+   */
+  googleAuthOptions?: GoogleAuthOptions;
+
+  /**
+   * Optional custom headers.
+   */
+  headers?: Record<string, string>;
+
+  /**
+   * Optional custom url query parameters to include in request urls.
+   */
+  queryParams?: Record<string, string>;
+
+  /**
+   * Custom fetch implementation.
+   */
+  fetch?: FetchFunction;
+
+  includeUsage?: boolean;
+  supportsStructuredOutputs?: boolean;
+  transformRequestBody?: (args: Record<string, any>) => Record<string, any>;
+  metadataExtractor?: MetadataExtractor;
+}
+
+export type VertexMaaSProvider = OpenAICompatibleProvider;
+
+export function createVertexMaaS(
+  options: VertexMaaSProviderSettings = {},
+): VertexMaaSProvider {
+  const providerName = 'vertexMaas';
+
+  let baseURLCache: string | null = null;
+  const getBaseURL = () => {
+    if (baseURLCache != null) {
+      return baseURLCache;
+    }
+
+    baseURLCache =
+      withoutTrailingSlash(options.baseURL) ??
+      getVertexOpenAICompatibleBaseURL({
+        project: loadSetting({
+          settingValue: options.project,
+          settingName: 'project',
+          environmentVariableName: 'GOOGLE_VERTEX_PROJECT',
+          description: 'Google Vertex project',
+        }),
+        location: loadSetting({
+          settingValue: options.location,
+          settingName: 'location',
+          environmentVariableName: 'GOOGLE_VERTEX_LOCATION',
+          description: 'Google Vertex location',
+        }),
+        apiVersion: options.apiVersion ?? 'v1beta1',
+        apiBase: options.apiBase,
+      });
+
+    return baseURLCache;
+  };
+
+  interface CommonModelConfig {
+    provider: string;
+    url: ({ path }: { path: string }) => string;
+    headers: () => Record<string, string>;
+    fetch?: FetchFunction;
+  }
+
+  const staticHeaders = {
+    ...(options.apiKey && { Authorization: `Bearer ${options.apiKey}` }),
+    ...options.headers,
+  };
+
+  const getHeaders = () =>
+    withUserAgentSuffix(staticHeaders, `ai-sdk/vertex-maas/${VERSION}`);
+
+  const fetch =
+    options.apiKey != null
+      ? options.fetch
+      : createVertexMaaSAuthFetch({
+          customFetch: options.fetch,
+          googleAuthOptions: options.googleAuthOptions,
+        });
+
+  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({
+    provider: `${providerName}.${modelType}`,
+    url: ({ path }) => {
+      const url = new URL(`${getBaseURL()}${path}`);
+      if (options.queryParams) {
+        url.search = new URLSearchParams(options.queryParams).toString();
+      }
+      return url.toString();
+    },
+    headers: getHeaders,
+    fetch,
+  });
+
+  const createLanguageModel = (modelId: string) => createChatModel(modelId);
+
+  const createChatModel = (modelId: string) =>
+    new OpenAICompatibleChatLanguageModel(modelId, {
+      ...getCommonModelConfig('chat'),
+      includeUsage: options.includeUsage,
+      supportsStructuredOutputs: options.supportsStructuredOutputs,
+      transformRequestBody: options.transformRequestBody,
+      metadataExtractor: options.metadataExtractor,
+    });
+
+  const createCompletionModel = (modelId: string) =>
+    new OpenAICompatibleCompletionLanguageModel(modelId, {
+      ...getCommonModelConfig('completion'),
+      includeUsage: options.includeUsage,
+    });
+
+  const createEmbeddingModel = (modelId: string) =>
+    new OpenAICompatibleEmbeddingModel(modelId, {
+      ...getCommonModelConfig('embedding'),
+    });
+
+  const createImageModel = (modelId: string) =>
+    new OpenAICompatibleImageModel(modelId, getCommonModelConfig('image'));
+
+  const provider = (modelId: string) => createLanguageModel(modelId);
+
+  provider.specificationVersion = 'v3' as const;
+  provider.languageModel = (
+    modelId: string,
+    config?: Partial<OpenAICompatibleChatConfig>,
+  ): LanguageModelV3 =>
+    new OpenAICompatibleChatLanguageModel(modelId, {
+      ...getCommonModelConfig('chat'),
+      includeUsage: options.includeUsage,
+      supportsStructuredOutputs: options.supportsStructuredOutputs,
+      transformRequestBody: options.transformRequestBody,
+      metadataExtractor: options.metadataExtractor,
+      ...config,
+    });
+  provider.chatModel = createChatModel;
+  provider.completionModel = createCompletionModel;
+  provider.embeddingModel = createEmbeddingModel as (
+    modelId: string,
+  ) => EmbeddingModelV3;
+  provider.textEmbeddingModel = createEmbeddingModel as (
+    modelId: string,
+  ) => EmbeddingModelV3;
+  provider.imageModel = createImageModel as (modelId: string) => ImageModelV3;
+
+  return provider as unknown as OpenAICompatibleProvider;
+}
+
+/**
+ * Default Vertex MaaS provider instance.
+ */
+export const vertexMaaS = createVertexMaaS();