| import { AISDK } from "@opencode-ai/core/aisdk" |
| import type { LanguageModelV3 } from "@ai-sdk/provider" |
| import { describe, expect } from "bun:test" |
| import { Effect } from "effect" |
| import { Catalog } from "@opencode-ai/core/catalog" |
| import { ModelV2 } from "@opencode-ai/core/model" |
| import { PluginV2 } from "@opencode-ai/core/plugin" |
| import { PluginHost } from "@opencode-ai/core/plugin/host" |
| import { GoogleVertexAnthropicPlugin, GoogleVertexPlugin } from "@opencode-ai/core/plugin/provider/google-vertex" |
| import { ProviderV2 } from "@opencode-ai/core/provider" |
| import { testEffect } from "../lib/effect" |
| import { PluginTestLayer } from "./fixture" |
|
|
| const it = testEffect(PluginTestLayer) |
|
|
| const addPlugin = Effect.fn(function* (definition: typeof GoogleVertexAnthropicPlugin | typeof GoogleVertexPlugin) { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| const host = yield* PluginHost.make(plugin) |
| yield* definition.effect(host) |
| }) |
|
|
| function withEnv<A, E, R>(vars: Record<string, string | undefined>, effect: () => Effect.Effect<A, E, R>) { |
| return Effect.acquireUseRelease( |
| Effect.sync(() => { |
| const previous = Object.fromEntries(Object.keys(vars).map((key) => [key, process.env[key]])) |
| Object.entries(vars).forEach(([key, value]) => { |
| if (value === undefined) delete process.env[key] |
| else process.env[key] = value |
| }) |
| return previous |
| }), |
| effect, |
| (previous) => |
| Effect.sync(() => { |
| Object.entries(previous).forEach(([key, value]) => { |
| if (value === undefined) delete process.env[key] |
| else process.env[key] = value |
| }) |
| }), |
| ) |
| } |
|
|
| function selector(calls: string[]) { |
| return (id: string) => { |
| calls.push(`languageModel:${id}`) |
| return { modelId: id, provider: "languageModel", specificationVersion: "v3" } as unknown as LanguageModelV3 |
| } |
| } |
|
|
| describe("GoogleVertexAnthropicPlugin", () => { |
| it.effect("resolves legacy project and location env on provider update", () => |
| withEnv( |
| { |
| GOOGLE_CLOUD_PROJECT: "cloud-project", |
| GCP_PROJECT: "gcp-project", |
| GCLOUD_PROJECT: "gcloud-project", |
| GOOGLE_CLOUD_LOCATION: "cloud-location", |
| VERTEX_LOCATION: "vertex-location", |
| GOOGLE_VERTEX_LOCATION: "google-vertex-location", |
| }, |
| () => |
| Effect.gen(function* () { |
| const catalog = yield* Catalog.Service |
| yield* catalog.transform((catalog) => |
| catalog.provider.update(ProviderV2.ID.make("google-vertex-anthropic"), (provider) => { |
| provider.api = { type: "aisdk", package: "@ai-sdk/google-vertex/anthropic" } |
| }), |
| ) |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| expect( |
| (yield* catalog.provider.get(ProviderV2.ID.make("google-vertex-anthropic")))?.request.body.project, |
| ).toBe("cloud-project") |
| expect( |
| (yield* catalog.provider.get(ProviderV2.ID.make("google-vertex-anthropic")))?.request.body.location, |
| ).toBe("cloud-location") |
| }), |
| ), |
| ) |
|
|
| it.effect("keeps configured project and location over env fallback", () => |
| withEnv({ GOOGLE_CLOUD_PROJECT: "env-project", GOOGLE_CLOUD_LOCATION: "env-location" }, () => |
| Effect.gen(function* () { |
| const catalog = yield* Catalog.Service |
| yield* catalog.transform((catalog) => |
| catalog.provider.update(ProviderV2.ID.make("google-vertex-anthropic"), (provider) => { |
| provider.api = { type: "aisdk", package: "@ai-sdk/google-vertex/anthropic" } |
| provider.request.body.project = "configured-project" |
| provider.request.body.location = "configured-location" |
| }), |
| ) |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| expect((yield* catalog.provider.get(ProviderV2.ID.make("google-vertex-anthropic")))?.request.body.project).toBe( |
| "configured-project", |
| ) |
| expect( |
| (yield* catalog.provider.get(ProviderV2.ID.make("google-vertex-anthropic")))?.request.body.location, |
| ).toBe("configured-location") |
| }), |
| ), |
| ) |
|
|
| it.effect("creates SDKs from legacy env fallback and default location", () => |
| withEnv( |
| { |
| GOOGLE_CLOUD_PROJECT: undefined, |
| GCP_PROJECT: "gcp-project", |
| GCLOUD_PROJECT: "gcloud-project", |
| GOOGLE_CLOUD_LOCATION: undefined, |
| VERTEX_LOCATION: undefined, |
| GOOGLE_VERTEX_LOCATION: "ignored-location", |
| }, |
| () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| const result = yield* aisdk.runSDK({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty( |
| ProviderV2.ID.make("google-vertex-anthropic"), |
| ModelV2.ID.make("claude-sonnet-4-5"), |
| ), |
| api: { id: ModelV2.ID.make("claude-sonnet-4-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/google-vertex/anthropic", |
| options: { name: "google-vertex-anthropic" }, |
| }) |
| expect(result.sdk.languageModel("claude-sonnet-4-5").config.baseURL).toBe( |
| "https://aiplatform.googleapis.com/v1/projects/gcp-project/locations/global/publishers/anthropic/models", |
| ) |
| }), |
| ), |
| ) |
|
|
| it.effect("uses GOOGLE_CLOUD_LOCATION before VERTEX_LOCATION when creating SDKs", () => |
| withEnv( |
| { GOOGLE_CLOUD_PROJECT: "project", GOOGLE_CLOUD_LOCATION: "cloud-location", VERTEX_LOCATION: "vertex-location" }, |
| () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| const result = yield* aisdk.runSDK({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty( |
| ProviderV2.ID.make("google-vertex-anthropic"), |
| ModelV2.ID.make("claude-sonnet-4-5"), |
| ), |
| api: { id: ModelV2.ID.make("claude-sonnet-4-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/google-vertex/anthropic", |
| options: { name: "google-vertex-anthropic" }, |
| }) |
| expect(result.sdk.languageModel("claude-sonnet-4-5").config.baseURL).toBe( |
| "https://cloud-location-aiplatform.googleapis.com/v1/projects/project/locations/cloud-location/publishers/anthropic/models", |
| ) |
| }), |
| ), |
| ) |
|
|
| it.effect("creates SDKs for google-vertex Anthropic models with multi-region endpoints", () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| const result = yield* aisdk.runSDK({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty(ProviderV2.ID.make("google-vertex"), ModelV2.ID.make("claude-sonnet-4-5")), |
| api: { id: ModelV2.ID.make("claude-sonnet-4-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/google-vertex/anthropic", |
| options: { name: "google-vertex", project: "project", location: "eu" }, |
| }) |
| expect(result.sdk.languageModel("claude-sonnet-4-5").config.baseURL).toBe( |
| "https://aiplatform.eu.rep.googleapis.com/v1/projects/project/locations/eu/publishers/anthropic/models", |
| ) |
| }), |
| ) |
|
|
| it.effect("keeps configured baseURL for google-vertex Anthropic models", () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| const result = yield* aisdk.runSDK({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty(ProviderV2.ID.make("google-vertex"), ModelV2.ID.make("claude-sonnet-4-5")), |
| api: { id: ModelV2.ID.make("claude-sonnet-4-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/google-vertex/anthropic", |
| options: { name: "google-vertex", project: "project", location: "eu", baseURL: "https://proxy.example/v1" }, |
| }) |
| expect(result.sdk.languageModel("claude-sonnet-4-5").config.baseURL).toBe("https://proxy.example/v1") |
| }), |
| ) |
|
|
| it.effect("selects google-vertex Anthropic language models through V2 plugins", () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| yield* addPlugin(GoogleVertexPlugin) |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| const sdkResult = yield* aisdk.runSDK({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty(ProviderV2.ID.make("google-vertex"), ModelV2.ID.make(" claude-sonnet-4-5 ")), |
| api: { id: ModelV2.ID.make(" claude-sonnet-4-5 "), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/google-vertex/anthropic", |
| options: { name: "google-vertex", project: "project", location: "us" }, |
| }) |
| const languageResult = yield* aisdk.runLanguage({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty(ProviderV2.ID.make("google-vertex"), ModelV2.ID.make(" claude-sonnet-4-5 ")), |
| api: { id: ModelV2.ID.make(" claude-sonnet-4-5 "), type: "aisdk", package: "test-provider" }, |
| }), |
| sdk: sdkResult.sdk, |
| options: {}, |
| }) |
| const language = languageResult.language as unknown as { config: { baseURL: string }; modelId: string } |
| expect(language.config.baseURL).toBe( |
| "https://aiplatform.us.rep.googleapis.com/v1/projects/project/locations/us/publishers/anthropic/models", |
| ) |
| expect(language.modelId).toBe("claude-sonnet-4-5") |
| }), |
| ) |
|
|
| it.effect("trims model IDs before selecting language models", () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| const calls: string[] = [] |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| yield* aisdk.runLanguage({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty(ProviderV2.ID.make("google-vertex-anthropic"), ModelV2.ID.make(" claude-sonnet-4-5 ")), |
| api: { id: ModelV2.ID.make(" claude-sonnet-4-5 "), type: "aisdk", package: "test-provider" }, |
| }), |
| sdk: { languageModel: selector(calls) }, |
| options: {}, |
| }) |
| expect(calls).toEqual(["languageModel:claude-sonnet-4-5"]) |
| }), |
| ) |
|
|
| it.effect("ignores non Vertex Anthropic providers for language selection", () => |
| Effect.gen(function* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| const calls: string[] = [] |
| yield* addPlugin(GoogleVertexAnthropicPlugin) |
| const result = yield* aisdk.runLanguage({ |
| model: ModelV2.Info.make({ |
| ...ModelV2.Info.empty(ProviderV2.ID.make("google-vertex"), ModelV2.ID.make("claude-sonnet-4-5")), |
| api: { id: ModelV2.ID.make("claude-sonnet-4-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| sdk: { languageModel: selector(calls) }, |
| options: {}, |
| }) |
| expect(calls).toEqual([]) |
| expect(result.language).toBeUndefined() |
| }), |
| ) |
| }) |
|
|
