@tanstack/ai-anthropic 0.16.7 → 0.17.0

package/src/vertex/auth.ts

@@ -0,0 +1,66 @@
+import type { AnthropicVertex } from '@anthropic-ai/vertex-sdk'
+
+export class AnthropicVertexAuthError extends Error {
+  constructor(message: string) {
+    super(message)
+    this.name = 'AnthropicVertexAuthError'
+  }
+}
+
+type VertexSdkOptions = NonNullable<
+  ConstructorParameters<typeof AnthropicVertex>[0]
+>
+
+/**
+ * Public Vertex config for Claude. `project` and `location` match the Gemini
+ * Vertex factories so one auth object works for both.
+ */
+export type AnthropicVertexConfig = Omit<
+  VertexSdkOptions,
+  'projectId' | 'region'
+> & {
+  project?: string
+  location?: string
+}
+
+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])
+}
+
+export function resolveAnthropicVertexOptions(
+  config: AnthropicVertexConfig = {},
+): VertexSdkOptions {
+  const { project, location, ...rest } = config
+  const projectId =
+    nonEmpty(project) ??
+    readEnv('GOOGLE_CLOUD_PROJECT') ??
+    readEnv('GOOGLE_VERTEX_PROJECT') ??
+    readEnv('ANTHROPIC_VERTEX_PROJECT_ID')
+  const region =
+    nonEmpty(location) ??
+    readEnv('GOOGLE_CLOUD_LOCATION') ??
+    readEnv('GOOGLE_VERTEX_LOCATION') ??
+    readEnv('CLOUD_ML_REGION')
+
+  if (region === undefined) {
+    throw new AnthropicVertexAuthError(
+      'Anthropic Vertex needs a location. Pass location on the factory, or set GOOGLE_CLOUD_LOCATION, GOOGLE_VERTEX_LOCATION, or CLOUD_ML_REGION.',
+    )
+  }
+
+  return {
+    ...rest,
+    projectId: projectId ?? null,
+    region,
+  }
+}

package/src/vertex/index.ts

@@ -0,0 +1,29 @@
+import { AnthropicVertex } from '@anthropic-ai/vertex-sdk'
+import { createAnthropicChatWithClient } from '../adapters/text'
+import { resolveAnthropicVertexOptions } from './auth'
+import type { AnthropicTextAdapter } from '../adapters/text'
+import type { AnthropicVertexChatModel } from '../model-meta'
+import type { AnthropicVertexConfig } from './auth'
+
+export {
+  AnthropicVertexAuthError,
+  resolveAnthropicVertexOptions,
+  type AnthropicVertexConfig,
+} from './auth'
+export {
+  ANTHROPIC_VERTEX_CHAT_MODELS,
+  type AnthropicVertexChatModel,
+} from '../model-meta'
+
+/**
+ * Creates an Anthropic chat adapter that talks to Claude on Vertex AI.
+ *
+ * Install `@anthropic-ai/vertex-sdk` next to `@tanstack/ai-anthropic`.
+ */
+export function anthropicVertexText<TModel extends AnthropicVertexChatModel>(
+  model: TModel,
+  config: AnthropicVertexConfig = {},
+): AnthropicTextAdapter<TModel> {
+  const client = new AnthropicVertex(resolveAnthropicVertexOptions(config))
+  return createAnthropicChatWithClient(model, client)
+}