@tanstack/ai-vertex 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Tanner Linsley
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,32 @@
+# @tanstack/ai-vertex
+
+Google Vertex AI adapter for [TanStack AI](https://tanstack.com/ai).
+
+Use this package to run Gemini models on Vertex. That gives you regional
+endpoints and Google Cloud credentials (ADC, service accounts, or Vertex
+express API keys).
+
+Claude on Vertex lives in [`@tanstack/ai-anthropic/vertex`](https://tanstack.com/ai/latest/docs/adapters/anthropic).
+
+## Install
+
+```bash
+pnpm add @tanstack/ai @tanstack/ai-vertex
+```
+
+## Usage
+
+```ts
+import { chat } from '@tanstack/ai'
+import { vertexText } from '@tanstack/ai-vertex'
+
+const stream = chat({
+  adapter: vertexText('gemini-3.7-flash', {
+    project: 'my-project',
+    location: 'europe-west1',
+  }),
+  messages: [{ role: 'user', content: 'Hello' }],
+})
+```
+
+See the [Vertex adapter docs](https://tanstack.com/ai/latest/docs/adapters/vertex).

package/dist/esm/auth.d.ts

@@ -0,0 +1,12 @@
+import { GeminiClientConfig } from '@tanstack/ai-gemini';
+export type VertexClientConfig = Omit<GeminiClientConfig, 'vertexai' | 'enterprise'>;
+export type VertexVideoConfig = VertexClientConfig & {
+    allowUrlFetch?: boolean;
+};
+/**
+ * Resolves Vertex Gemini client options.
+ *
+ * Factory fields win. Then env. Then ADC inside `@google/genai`.
+ * Does not read `GEMINI_API_KEY` or `GOOGLE_API_KEY`.
+ */
+export declare function resolveVertexGeminiOptions(config?: VertexClientConfig): GeminiClientConfig;

package/dist/esm/auth.js

@@ -0,0 +1,33 @@
+import { VertexAuthError } from "./errors.js";
+//#region src/auth.ts
+function nonEmpty(value) {
+	if (value === void 0 || value.length === 0) return;
+	return value;
+}
+function readEnv(name) {
+	if (typeof process === "undefined" || process.env === void 0) return;
+	return nonEmpty(process.env[name]);
+}
+/**
+* Resolves Vertex Gemini client options.
+*
+* Factory fields win. Then env. Then ADC inside `@google/genai`.
+* Does not read `GEMINI_API_KEY` or `GOOGLE_API_KEY`.
+*/
+function resolveVertexGeminiOptions(config = {}) {
+	const project = nonEmpty(config.project) ?? readEnv("GOOGLE_CLOUD_PROJECT") ?? readEnv("GOOGLE_VERTEX_PROJECT");
+	const location = nonEmpty(config.location) ?? readEnv("GOOGLE_CLOUD_LOCATION") ?? readEnv("GOOGLE_VERTEX_LOCATION");
+	const apiKey = nonEmpty(config.apiKey) ?? readEnv("GOOGLE_VERTEX_API_KEY");
+	if (apiKey === void 0 && (project === void 0 || location === void 0)) throw new VertexAuthError("Vertex Gemini needs project and location, or an express apiKey. Pass project and location on the factory, or set GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION. For express mode, pass apiKey or set GOOGLE_VERTEX_API_KEY.");
+	return {
+		...config,
+		vertexai: true,
+		...project === void 0 ? {} : { project },
+		...location === void 0 ? {} : { location },
+		...apiKey === void 0 ? {} : { apiKey }
+	};
+}
+//#endregion
+export { resolveVertexGeminiOptions };
+
+//# sourceMappingURL=auth.js.map

package/dist/esm/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","names":[],"sources":["../../src/auth.ts"],"sourcesContent":["import { VertexAuthError } from './errors'\nimport type { GeminiClientConfig } from '@tanstack/ai-gemini'\n\nexport type VertexClientConfig = Omit<\n  GeminiClientConfig,\n  'vertexai' | 'enterprise'\n>\n\nexport type VertexVideoConfig = VertexClientConfig & {\n  allowUrlFetch?: boolean\n}\n\nfunction nonEmpty(value: string | undefined): string | undefined {\n  if (value === undefined || value.length === 0) {\n    return undefined\n  }\n  return value\n}\n\nfunction readEnv(name: string): string | undefined {\n  if (typeof process === 'undefined' || process.env === undefined) {\n    return undefined\n  }\n  return nonEmpty(process.env[name])\n}\n\n/**\n * Resolves Vertex Gemini client options.\n *\n * Factory fields win. Then env. Then ADC inside `@google/genai`.\n * Does not read `GEMINI_API_KEY` or `GOOGLE_API_KEY`.\n */\nexport function resolveVertexGeminiOptions(\n  config: VertexClientConfig = {},\n): GeminiClientConfig {\n  const project =\n    nonEmpty(config.project) ??\n    readEnv('GOOGLE_CLOUD_PROJECT') ??\n    readEnv('GOOGLE_VERTEX_PROJECT')\n  const location =\n    nonEmpty(config.location) ??\n    readEnv('GOOGLE_CLOUD_LOCATION') ??\n    readEnv('GOOGLE_VERTEX_LOCATION')\n  const apiKey = nonEmpty(config.apiKey) ?? readEnv('GOOGLE_VERTEX_API_KEY')\n\n  if (\n    apiKey === undefined &&\n    (project === undefined || location === undefined)\n  ) {\n    throw new VertexAuthError(\n      'Vertex Gemini needs project and location, or an express apiKey. Pass project and location on the factory, or set GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION. For express mode, pass apiKey or set GOOGLE_VERTEX_API_KEY.',\n    )\n  }\n\n  return {\n    ...config,\n    vertexai: true,\n    ...(project === undefined ? {} : { project }),\n    ...(location === undefined ? {} : { location }),\n    ...(apiKey === undefined ? {} : { apiKey }),\n  }\n}\n"],"mappings":";;AAYA,SAAS,SAAS,OAA+C;CAC/D,IAAI,UAAU,KAAA,KAAa,MAAM,WAAW,GAC1C;CAEF,OAAO;AACT;AAEA,SAAS,QAAQ,MAAkC;CACjD,IAAI,OAAO,YAAY,eAAe,QAAQ,QAAQ,KAAA,GACpD;CAEF,OAAO,SAAS,QAAQ,IAAI,KAAK;AACnC;;;;;;;AAQA,SAAgB,2BACd,SAA6B,CAAC,GACV;CACpB,MAAM,UACJ,SAAS,OAAO,OAAO,KACvB,QAAQ,sBAAsB,KAC9B,QAAQ,uBAAuB;CACjC,MAAM,WACJ,SAAS,OAAO,QAAQ,KACxB,QAAQ,uBAAuB,KAC/B,QAAQ,wBAAwB;CAClC,MAAM,SAAS,SAAS,OAAO,MAAM,KAAK,QAAQ,uBAAuB;CAEzE,IACE,WAAW,KAAA,MACV,YAAY,KAAA,KAAa,aAAa,KAAA,IAEvC,MAAM,IAAI,gBACR,8NACF;CAGF,OAAO;EACL,GAAG;EACH,UAAU;EACV,GAAI,YAAY,KAAA,IAAY,CAAC,IAAI,EAAE,QAAQ;EAC3C,GAAI,aAAa,KAAA,IAAY,CAAC,IAAI,EAAE,SAAS;EAC7C,GAAI,WAAW,KAAA,IAAY,CAAC,IAAI,EAAE,OAAO;CAC3C;AACF"}

package/dist/esm/errors.d.ts

@@ -0,0 +1,3 @@
+export declare class VertexAuthError extends Error {
+    constructor(message: string);
+}

package/dist/esm/errors.js

@@ -0,0 +1,11 @@
+//#region src/errors.ts
+var VertexAuthError = class extends Error {
+	constructor(message) {
+		super(message);
+		this.name = "VertexAuthError";
+	}
+};
+//#endregion
+export { VertexAuthError };
+
+//# sourceMappingURL=errors.js.map

package/dist/esm/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","names":[],"sources":["../../src/errors.ts"],"sourcesContent":["export class VertexAuthError extends Error {\n  constructor(message: string) {\n    super(message)\n    this.name = 'VertexAuthError'\n  }\n}\n"],"mappings":";AAAA,IAAa,kBAAb,cAAqC,MAAM;CACzC,YAAY,SAAiB;EAC3B,MAAM,OAAO;EACb,KAAK,OAAO;CACd;AACF"}

package/dist/esm/index.d.ts

@@ -0,0 +1,13 @@
+import { ChatStreamSummarizeAdapter } from '@tanstack/ai/adapters';
+import { GeminiAudioAdapter, GeminiEmbeddingAdapter, GeminiImageAdapter, GeminiTTSAdapter, GeminiTextAdapter, GeminiVideoAdapter, GeminiAudioModel, GeminiEmbeddingModel, GeminiImageModel, GeminiTTSModels, GeminiTextModel, GeminiVideoModel } from '@tanstack/ai-gemini';
+import { VertexClientConfig, VertexVideoConfig } from './auth.js';
+export { VertexAuthError } from './errors.js';
+export { resolveVertexGeminiOptions, type VertexClientConfig, type VertexVideoConfig, } from './auth.js';
+type GeminiTTSModel = (typeof GeminiTTSModels)[number];
+export declare function vertexText<TModel extends GeminiTextModel>(model: TModel, config?: VertexClientConfig): GeminiTextAdapter<TModel>;
+export declare function vertexSummarize<TModel extends GeminiTextModel>(model: TModel, config?: VertexClientConfig): ChatStreamSummarizeAdapter<TModel>;
+export declare function vertexImage<TModel extends GeminiImageModel>(model: TModel, config?: VertexClientConfig): GeminiImageAdapter<TModel>;
+export declare function vertexEmbedding<TModel extends GeminiEmbeddingModel>(model: TModel, config?: VertexClientConfig): GeminiEmbeddingAdapter<TModel>;
+export declare function vertexSpeech<TModel extends GeminiTTSModel>(model: TModel, config?: VertexClientConfig): GeminiTTSAdapter<TModel>;
+export declare function vertexAudio<TModel extends GeminiAudioModel>(model: TModel, config?: VertexClientConfig): GeminiAudioAdapter<TModel>;
+export declare function vertexVideo<TModel extends GeminiVideoModel>(model: TModel, config?: VertexVideoConfig): GeminiVideoAdapter<TModel>;

package/dist/esm/index.js

@@ -0,0 +1,60 @@
+import { VertexAuthError } from "./errors.js";
+import { resolveVertexGeminiOptions } from "./auth.js";
+import { ChatStreamSummarizeAdapter } from "@tanstack/ai/adapters";
+import { GeminiAudioAdapter, GeminiEmbeddingAdapter, GeminiImageAdapter, GeminiTTSAdapter, GeminiTextAdapter, GeminiVideoAdapter } from "@tanstack/ai-gemini";
+//#region src/index.ts
+function createVertex(config = {}) {
+	const resolved = resolveVertexGeminiOptions(config);
+	return {
+		text(model) {
+			return new GeminiTextAdapter(resolved, model);
+		},
+		summarize(model) {
+			return new ChatStreamSummarizeAdapter(new GeminiTextAdapter(resolved, model), model, "gemini");
+		},
+		image(model) {
+			return new GeminiImageAdapter(resolved, model);
+		},
+		embedding(model) {
+			return new GeminiEmbeddingAdapter(resolved, model);
+		},
+		speech(model) {
+			return new GeminiTTSAdapter(resolved, model);
+		},
+		audio(model) {
+			return new GeminiAudioAdapter(resolved, model);
+		},
+		video(model, videoConfig) {
+			return new GeminiVideoAdapter({
+				...resolved,
+				allowUrlFetch: videoConfig?.allowUrlFetch
+			}, model);
+		}
+	};
+}
+function vertexText(model, config = {}) {
+	return createVertex(config).text(model);
+}
+function vertexSummarize(model, config = {}) {
+	return createVertex(config).summarize(model);
+}
+function vertexImage(model, config = {}) {
+	return createVertex(config).image(model);
+}
+function vertexEmbedding(model, config = {}) {
+	return createVertex(config).embedding(model);
+}
+function vertexSpeech(model, config = {}) {
+	return createVertex(config).speech(model);
+}
+function vertexAudio(model, config = {}) {
+	return createVertex(config).audio(model);
+}
+function vertexVideo(model, config = {}) {
+	const { allowUrlFetch, ...client } = config;
+	return createVertex(client).video(model, { allowUrlFetch });
+}
+//#endregion
+export { VertexAuthError, resolveVertexGeminiOptions, vertexAudio, vertexEmbedding, vertexImage, vertexSpeech, vertexSummarize, vertexText, vertexVideo };
+
+//# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../../src/index.ts"],"sourcesContent":["import { ChatStreamSummarizeAdapter } from '@tanstack/ai/adapters'\nimport {\n  GeminiAudioAdapter,\n  GeminiEmbeddingAdapter,\n  GeminiImageAdapter,\n  GeminiTTSAdapter,\n  GeminiTextAdapter,\n  GeminiVideoAdapter,\n} from '@tanstack/ai-gemini'\nimport { resolveVertexGeminiOptions } from './auth'\nimport type {\n  GeminiAudioModel,\n  GeminiEmbeddingModel,\n  GeminiImageModel,\n  GeminiTTSModels,\n  GeminiTextModel,\n  GeminiVideoModel,\n} from '@tanstack/ai-gemini'\nimport type { VertexClientConfig, VertexVideoConfig } from './auth'\n\nexport { VertexAuthError } from './errors'\nexport {\n  resolveVertexGeminiOptions,\n  type VertexClientConfig,\n  type VertexVideoConfig,\n} from './auth'\n\ntype GeminiTTSModel = (typeof GeminiTTSModels)[number]\n\nfunction createVertex(config: VertexClientConfig = {}) {\n  const resolved = resolveVertexGeminiOptions(config)\n  return {\n    text<TModel extends GeminiTextModel>(model: TModel) {\n      return new GeminiTextAdapter(resolved, model)\n    },\n    summarize<TModel extends GeminiTextModel>(model: TModel) {\n      return new ChatStreamSummarizeAdapter(\n        new GeminiTextAdapter(resolved, model),\n        model,\n        'gemini',\n      )\n    },\n    image<TModel extends GeminiImageModel>(model: TModel) {\n      return new GeminiImageAdapter(resolved, model)\n    },\n    embedding<TModel extends GeminiEmbeddingModel>(model: TModel) {\n      return new GeminiEmbeddingAdapter(resolved, model)\n    },\n    speech<TModel extends GeminiTTSModel>(model: TModel) {\n      return new GeminiTTSAdapter(resolved, model)\n    },\n    audio<TModel extends GeminiAudioModel>(model: TModel) {\n      return new GeminiAudioAdapter(resolved, model)\n    },\n    video<TModel extends GeminiVideoModel>(\n      model: TModel,\n      videoConfig?: Pick<VertexVideoConfig, 'allowUrlFetch'>,\n    ) {\n      return new GeminiVideoAdapter(\n        { ...resolved, allowUrlFetch: videoConfig?.allowUrlFetch },\n        model,\n      )\n    },\n  }\n}\n\nexport function vertexText<TModel extends GeminiTextModel>(\n  model: TModel,\n  config: VertexClientConfig = {},\n): GeminiTextAdapter<TModel> {\n  return createVertex(config).text(model)\n}\n\nexport function vertexSummarize<TModel extends GeminiTextModel>(\n  model: TModel,\n  config: VertexClientConfig = {},\n): ChatStreamSummarizeAdapter<TModel> {\n  return createVertex(config).summarize(model)\n}\n\nexport function vertexImage<TModel extends GeminiImageModel>(\n  model: TModel,\n  config: VertexClientConfig = {},\n): GeminiImageAdapter<TModel> {\n  return createVertex(config).image(model)\n}\n\nexport function vertexEmbedding<TModel extends GeminiEmbeddingModel>(\n  model: TModel,\n  config: VertexClientConfig = {},\n): GeminiEmbeddingAdapter<TModel> {\n  return createVertex(config).embedding(model)\n}\n\nexport function vertexSpeech<TModel extends GeminiTTSModel>(\n  model: TModel,\n  config: VertexClientConfig = {},\n): GeminiTTSAdapter<TModel> {\n  return createVertex(config).speech(model)\n}\n\nexport function vertexAudio<TModel extends GeminiAudioModel>(\n  model: TModel,\n  config: VertexClientConfig = {},\n): GeminiAudioAdapter<TModel> {\n  return createVertex(config).audio(model)\n}\n\nexport function vertexVideo<TModel extends GeminiVideoModel>(\n  model: TModel,\n  config: VertexVideoConfig = {},\n): GeminiVideoAdapter<TModel> {\n  const { allowUrlFetch, ...client } = config\n  return createVertex(client).video(model, { allowUrlFetch })\n}\n"],"mappings":";;;;;AA6BA,SAAS,aAAa,SAA6B,CAAC,GAAG;CACrD,MAAM,WAAW,2BAA2B,MAAM;CAClD,OAAO;EACL,KAAqC,OAAe;GAClD,OAAO,IAAI,kBAAkB,UAAU,KAAK;EAC9C;EACA,UAA0C,OAAe;GACvD,OAAO,IAAI,2BACT,IAAI,kBAAkB,UAAU,KAAK,GACrC,OACA,QACF;EACF;EACA,MAAuC,OAAe;GACpD,OAAO,IAAI,mBAAmB,UAAU,KAAK;EAC/C;EACA,UAA+C,OAAe;GAC5D,OAAO,IAAI,uBAAuB,UAAU,KAAK;EACnD;EACA,OAAsC,OAAe;GACnD,OAAO,IAAI,iBAAiB,UAAU,KAAK;EAC7C;EACA,MAAuC,OAAe;GACpD,OAAO,IAAI,mBAAmB,UAAU,KAAK;EAC/C;EACA,MACE,OACA,aACA;GACA,OAAO,IAAI,mBACT;IAAE,GAAG;IAAU,eAAe,aAAa;GAAc,GACzD,KACF;EACF;CACF;AACF;AAEA,SAAgB,WACd,OACA,SAA6B,CAAC,GACH;CAC3B,OAAO,aAAa,MAAM,CAAC,CAAC,KAAK,KAAK;AACxC;AAEA,SAAgB,gBACd,OACA,SAA6B,CAAC,GACM;CACpC,OAAO,aAAa,MAAM,CAAC,CAAC,UAAU,KAAK;AAC7C;AAEA,SAAgB,YACd,OACA,SAA6B,CAAC,GACF;CAC5B,OAAO,aAAa,MAAM,CAAC,CAAC,MAAM,KAAK;AACzC;AAEA,SAAgB,gBACd,OACA,SAA6B,CAAC,GACE;CAChC,OAAO,aAAa,MAAM,CAAC,CAAC,UAAU,KAAK;AAC7C;AAEA,SAAgB,aACd,OACA,SAA6B,CAAC,GACJ;CAC1B,OAAO,aAAa,MAAM,CAAC,CAAC,OAAO,KAAK;AAC1C;AAEA,SAAgB,YACd,OACA,SAA6B,CAAC,GACF;CAC5B,OAAO,aAAa,MAAM,CAAC,CAAC,MAAM,KAAK;AACzC;AAEA,SAAgB,YACd,OACA,SAA4B,CAAC,GACD;CAC5B,MAAM,EAAE,eAAe,GAAG,WAAW;CACrC,OAAO,aAAa,MAAM,CAAC,CAAC,MAAM,OAAO,EAAE,cAAc,CAAC;AAC5D"}

package/package.json

@@ -0,0 +1,66 @@
+{
+  "name": "@tanstack/ai-vertex",
+  "version": "0.1.0",
+  "description": "Google Vertex AI adapter for TanStack AI. Runs Gemini models on Vertex with regional endpoints and Google Cloud credentials.",
+  "author": "Tanner Linsley",
+  "license": "MIT",
+  "homepage": "https://tanstack.com/ai",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/TanStack/ai.git",
+    "directory": "packages/ai-vertex"
+  },
+  "bugs": {
+    "url": "https://github.com/TanStack/ai/issues"
+  },
+  "funding": {
+    "type": "github",
+    "url": "https://github.com/sponsors/tannerlinsley"
+  },
+  "type": "module",
+  "module": "./dist/esm/index.js",
+  "types": "./dist/esm/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/esm/index.d.ts",
+      "import": "./dist/esm/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "keywords": [
+    "ai",
+    "ai-sdk",
+    "typescript",
+    "tanstack",
+    "vertex",
+    "vertex-ai",
+    "gemini",
+    "google",
+    "adapter"
+  ],
+  "dependencies": {
+    "@tanstack/ai-gemini": "^0.24.2"
+  },
+  "peerDependencies": {
+    "@tanstack/ai": "^0.47.2"
+  },
+  "devDependencies": {
+    "@google/genai": "^2.10.0",
+    "@vitest/coverage-v8": "4.1.10",
+    "vite": "^8.2.1",
+    "@tanstack/ai": "0.47.2"
+  },
+  "scripts": {
+    "build": "vite build",
+    "clean": "premove ./build ./dist",
+    "lint:fix": "oxlint src --type-aware --fix",
+    "test:build": "publint --strict",
+    "test:oxlint": "oxlint src --type-aware",
+    "test:lib": "vitest run",
+    "test:lib:dev": "pnpm test:lib --watch",
+    "test:types": "tsc"
+  }
+}

package/src/auth.ts

@@ -0,0 +1,62 @@
+import { VertexAuthError } from './errors'
+import type { GeminiClientConfig } from '@tanstack/ai-gemini'
+
+export type VertexClientConfig = Omit<
+  GeminiClientConfig,
+  'vertexai' | 'enterprise'
+>
+
+export type VertexVideoConfig = VertexClientConfig & {
+  allowUrlFetch?: boolean
+}
+
+function nonEmpty(value: string | undefined): string | undefined {
+  if (value === undefined || value.length === 0) {
+    return undefined
+  }
+  return value
+}
+
+function readEnv(name: string): string | undefined {
+  if (typeof process === 'undefined' || process.env === undefined) {
+    return undefined
+  }
+  return nonEmpty(process.env[name])
+}
+
+/**
+ * Resolves Vertex Gemini client options.
+ *
+ * Factory fields win. Then env. Then ADC inside `@google/genai`.
+ * Does not read `GEMINI_API_KEY` or `GOOGLE_API_KEY`.
+ */
+export function resolveVertexGeminiOptions(
+  config: VertexClientConfig = {},
+): GeminiClientConfig {
+  const project =
+    nonEmpty(config.project) ??
+    readEnv('GOOGLE_CLOUD_PROJECT') ??
+    readEnv('GOOGLE_VERTEX_PROJECT')
+  const location =
+    nonEmpty(config.location) ??
+    readEnv('GOOGLE_CLOUD_LOCATION') ??
+    readEnv('GOOGLE_VERTEX_LOCATION')
+  const apiKey = nonEmpty(config.apiKey) ?? readEnv('GOOGLE_VERTEX_API_KEY')
+
+  if (
+    apiKey === undefined &&
+    (project === undefined || location === undefined)
+  ) {
+    throw new VertexAuthError(
+      'Vertex Gemini needs project and location, or an express apiKey. Pass project and location on the factory, or set GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION. For express mode, pass apiKey or set GOOGLE_VERTEX_API_KEY.',
+    )
+  }
+
+  return {
+    ...config,
+    vertexai: true,
+    ...(project === undefined ? {} : { project }),
+    ...(location === undefined ? {} : { location }),
+    ...(apiKey === undefined ? {} : { apiKey }),
+  }
+}

package/src/errors.ts

@@ -0,0 +1,6 @@
+export class VertexAuthError extends Error {
+  constructor(message: string) {
+    super(message)
+    this.name = 'VertexAuthError'
+  }
+}

package/src/index.ts

@@ -0,0 +1,115 @@
+import { ChatStreamSummarizeAdapter } from '@tanstack/ai/adapters'
+import {
+  GeminiAudioAdapter,
+  GeminiEmbeddingAdapter,
+  GeminiImageAdapter,
+  GeminiTTSAdapter,
+  GeminiTextAdapter,
+  GeminiVideoAdapter,
+} from '@tanstack/ai-gemini'
+import { resolveVertexGeminiOptions } from './auth'
+import type {
+  GeminiAudioModel,
+  GeminiEmbeddingModel,
+  GeminiImageModel,
+  GeminiTTSModels,
+  GeminiTextModel,
+  GeminiVideoModel,
+} from '@tanstack/ai-gemini'
+import type { VertexClientConfig, VertexVideoConfig } from './auth'
+
+export { VertexAuthError } from './errors'
+export {
+  resolveVertexGeminiOptions,
+  type VertexClientConfig,
+  type VertexVideoConfig,
+} from './auth'
+
+type GeminiTTSModel = (typeof GeminiTTSModels)[number]
+
+function createVertex(config: VertexClientConfig = {}) {
+  const resolved = resolveVertexGeminiOptions(config)
+  return {
+    text<TModel extends GeminiTextModel>(model: TModel) {
+      return new GeminiTextAdapter(resolved, model)
+    },
+    summarize<TModel extends GeminiTextModel>(model: TModel) {
+      return new ChatStreamSummarizeAdapter(
+        new GeminiTextAdapter(resolved, model),
+        model,
+        'gemini',
+      )
+    },
+    image<TModel extends GeminiImageModel>(model: TModel) {
+      return new GeminiImageAdapter(resolved, model)
+    },
+    embedding<TModel extends GeminiEmbeddingModel>(model: TModel) {
+      return new GeminiEmbeddingAdapter(resolved, model)
+    },
+    speech<TModel extends GeminiTTSModel>(model: TModel) {
+      return new GeminiTTSAdapter(resolved, model)
+    },
+    audio<TModel extends GeminiAudioModel>(model: TModel) {
+      return new GeminiAudioAdapter(resolved, model)
+    },
+    video<TModel extends GeminiVideoModel>(
+      model: TModel,
+      videoConfig?: Pick<VertexVideoConfig, 'allowUrlFetch'>,
+    ) {
+      return new GeminiVideoAdapter(
+        { ...resolved, allowUrlFetch: videoConfig?.allowUrlFetch },
+        model,
+      )
+    },
+  }
+}
+
+export function vertexText<TModel extends GeminiTextModel>(
+  model: TModel,
+  config: VertexClientConfig = {},
+): GeminiTextAdapter<TModel> {
+  return createVertex(config).text(model)
+}
+
+export function vertexSummarize<TModel extends GeminiTextModel>(
+  model: TModel,
+  config: VertexClientConfig = {},
+): ChatStreamSummarizeAdapter<TModel> {
+  return createVertex(config).summarize(model)
+}
+
+export function vertexImage<TModel extends GeminiImageModel>(
+  model: TModel,
+  config: VertexClientConfig = {},
+): GeminiImageAdapter<TModel> {
+  return createVertex(config).image(model)
+}
+
+export function vertexEmbedding<TModel extends GeminiEmbeddingModel>(
+  model: TModel,
+  config: VertexClientConfig = {},
+): GeminiEmbeddingAdapter<TModel> {
+  return createVertex(config).embedding(model)
+}
+
+export function vertexSpeech<TModel extends GeminiTTSModel>(
+  model: TModel,
+  config: VertexClientConfig = {},
+): GeminiTTSAdapter<TModel> {
+  return createVertex(config).speech(model)
+}
+
+export function vertexAudio<TModel extends GeminiAudioModel>(
+  model: TModel,
+  config: VertexClientConfig = {},
+): GeminiAudioAdapter<TModel> {
+  return createVertex(config).audio(model)
+}
+
+export function vertexVideo<TModel extends GeminiVideoModel>(
+  model: TModel,
+  config: VertexVideoConfig = {},
+): GeminiVideoAdapter<TModel> {
+  const { allowUrlFetch, ...client } = config
+  return createVertex(client).video(model, { allowUrlFetch })
+}