|
|
@@ -1,271 +0,0 @@
|
|
|
-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 { Model } from "@opencode-ai/core/model"
|
|
|
-import { Plugin } 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 { Provider } 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* Plugin.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(Provider.ID.make("google-vertex-anthropic"), (provider) => {
|
|
|
- provider.package = Provider.aisdk("@ai-sdk/google-vertex/anthropic")
|
|
|
- }),
|
|
|
- )
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- expect((yield* catalog.provider.get(Provider.ID.make("google-vertex-anthropic")))?.settings?.project).toBe(
|
|
|
- "cloud-project",
|
|
|
- )
|
|
|
- expect((yield* catalog.provider.get(Provider.ID.make("google-vertex-anthropic")))?.settings?.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(Provider.ID.make("google-vertex-anthropic"), (provider) => {
|
|
|
- provider.package = Provider.aisdk("@ai-sdk/google-vertex/anthropic")
|
|
|
- provider.settings = { ...provider.settings, project: "configured-project", location: "configured-location" }
|
|
|
- }),
|
|
|
- )
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- expect((yield* catalog.provider.get(Provider.ID.make("google-vertex-anthropic")))?.settings?.project).toBe(
|
|
|
- "configured-project",
|
|
|
- )
|
|
|
- expect((yield* catalog.provider.get(Provider.ID.make("google-vertex-anthropic")))?.settings?.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* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- const result = yield* aisdk.runSDK({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(
|
|
|
- Provider.ID.make("google-vertex-anthropic"),
|
|
|
- Model.ID.make("claude-sonnet-4-5"),
|
|
|
- ),
|
|
|
- modelID: Model.ID.make("claude-sonnet-4-5"),
|
|
|
- package: "aisdk: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* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- const result = yield* aisdk.runSDK({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(
|
|
|
- Provider.ID.make("google-vertex-anthropic"),
|
|
|
- Model.ID.make("claude-sonnet-4-5"),
|
|
|
- ),
|
|
|
- modelID: Model.ID.make("claude-sonnet-4-5"),
|
|
|
- package: "aisdk: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* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- const result = yield* aisdk.runSDK({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(Provider.ID.make("google-vertex"), Model.ID.make("claude-sonnet-4-5")),
|
|
|
- modelID: Model.ID.make("claude-sonnet-4-5"),
|
|
|
- package: "aisdk: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* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- const result = yield* aisdk.runSDK({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(Provider.ID.make("google-vertex"), Model.ID.make("claude-sonnet-4-5")),
|
|
|
- modelID: Model.ID.make("claude-sonnet-4-5"),
|
|
|
- package: "aisdk: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 plugins", () =>
|
|
|
- Effect.gen(function* () {
|
|
|
- const plugin = yield* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- yield* addPlugin(GoogleVertexPlugin)
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- const sdkResult = yield* aisdk.runSDK({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(Provider.ID.make("google-vertex"), Model.ID.make(" claude-sonnet-4-5 ")),
|
|
|
- modelID: Model.ID.make(" claude-sonnet-4-5 "),
|
|
|
- package: "aisdk:test-provider",
|
|
|
- }),
|
|
|
- package: "@ai-sdk/google-vertex/anthropic",
|
|
|
- options: { name: "google-vertex", project: "project", location: "us" },
|
|
|
- })
|
|
|
- const languageResult = yield* aisdk.runLanguage({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(Provider.ID.make("google-vertex"), Model.ID.make(" claude-sonnet-4-5 ")),
|
|
|
- modelID: Model.ID.make(" claude-sonnet-4-5 "),
|
|
|
- package: "aisdk: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* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- const calls: string[] = []
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- yield* aisdk.runLanguage({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(Provider.ID.make("google-vertex-anthropic"), Model.ID.make(" claude-sonnet-4-5 ")),
|
|
|
- modelID: Model.ID.make(" claude-sonnet-4-5 "),
|
|
|
- package: "aisdk: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* Plugin.Service
|
|
|
- const aisdk = yield* AISDK.Service
|
|
|
- const calls: string[] = []
|
|
|
- yield* addPlugin(GoogleVertexAnthropicPlugin)
|
|
|
- const result = yield* aisdk.runLanguage({
|
|
|
- model: Model.Info.make({
|
|
|
- ...Model.Info.default(Provider.ID.make("google-vertex"), Model.ID.make("claude-sonnet-4-5")),
|
|
|
- modelID: Model.ID.make("claude-sonnet-4-5"),
|
|
|
- package: "aisdk:test-provider",
|
|
|
- }),
|
|
|
- sdk: { languageModel: selector(calls) },
|
|
|
- options: {},
|
|
|
- })
|
|
|
- expect(calls).toEqual([])
|
|
|
- expect(result.language).toBeUndefined()
|
|
|
- }),
|
|
|
- )
|
|
|
-})
|