@neurocode-ai/core 1.18.8

package/test/plugin/provider-sap-ai-core.test.ts

@@ -0,0 +1,159 @@
+import { AISDK } from "@neurocode-ai/core/aisdk"
+import { describe, expect } from "bun:test"
+import { Effect } from "effect"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { PluginV2 } from "@neurocode-ai/core/plugin"
+import { PluginHost } from "@neurocode-ai/core/plugin/host"
+import { Npm } from "@neurocode-ai/core/npm"
+import { SapAICorePlugin } from "@neurocode-ai/core/plugin/provider/sap-ai-core"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import { testEffect } from "../lib/effect"
+import { PluginTestLayer } from "./fixture"
+
+const fixtureProvider = new URL("./fixtures/provider-factory.ts", import.meta.url).href
+const it = testEffect(PluginTestLayer)
+const npm = Npm.Service.of({
+  add: () => Effect.succeed({ directory: "", entrypoint: undefined }),
+  install: () => Effect.void,
+  which: () => Effect.succeed(undefined),
+})
+
+const addPlugin = Effect.fn(function* () {
+  const plugin = yield* PluginV2.Service
+  const aisdk = yield* AISDK.Service
+  const host = yield* PluginHost.make(plugin)
+  yield* SapAICorePlugin.effect(host).pipe(Effect.provideService(Npm.Service, npm))
+})
+
+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]]))
+      for (const [key, value] of Object.entries(vars)) {
+        if (value === undefined) delete process.env[key]
+        else process.env[key] = value
+      }
+      return previous
+    }),
+    effect,
+    (previous) =>
+      Effect.sync(() => {
+        for (const [key, value] of Object.entries(previous)) {
+          if (value === undefined) delete process.env[key]
+          else process.env[key] = value
+        }
+      }),
+  )
+}
+
+function model(providerID: string) {
+  return ModelV2.Info.make({
+    ...ModelV2.Info.empty(ProviderV2.ID.make(providerID), ModelV2.ID.make("sap-model")),
+    api: { id: ModelV2.ID.make("sap-model"), type: "aisdk", package: fixtureProvider },
+  })
+}
+
+describe("SapAICorePlugin", () => {
+  it.effect("copies serviceKey option into AICORE_SERVICE_KEY but keeps SDK options to deployment metadata", () =>
+    withEnv(
+      { AICORE_SERVICE_KEY: undefined, AICORE_DEPLOYMENT_ID: "deployment", AICORE_RESOURCE_GROUP: "resource-group" },
+      () =>
+        Effect.gen(function* () {
+          const plugin = yield* PluginV2.Service
+          const aisdk = yield* AISDK.Service
+          yield* addPlugin()
+          const sdk = yield* aisdk.runSDK({
+            model: model("sap-ai-core"),
+            package: fixtureProvider,
+            options: { name: "sap-ai-core", serviceKey: "service-key" },
+          })
+          expect(process.env.AICORE_SERVICE_KEY).toBe("service-key")
+          expect(sdk.sdk.options).toEqual({ deploymentId: "deployment", resourceGroup: "resource-group" })
+        }),
+    ),
+  )
+
+  it.effect("preserves existing AICORE_SERVICE_KEY over serviceKey option", () =>
+    withEnv(
+      {
+        AICORE_SERVICE_KEY: "env-service-key",
+        AICORE_DEPLOYMENT_ID: "deployment",
+        AICORE_RESOURCE_GROUP: "resource-group",
+      },
+      () =>
+        Effect.gen(function* () {
+          const plugin = yield* PluginV2.Service
+          const aisdk = yield* AISDK.Service
+          yield* addPlugin()
+          const sdk = yield* aisdk.runSDK({
+            model: model("sap-ai-core"),
+            package: fixtureProvider,
+            options: { name: "sap-ai-core", serviceKey: "option-service-key" },
+          })
+          expect(process.env.AICORE_SERVICE_KEY).toBe("env-service-key")
+          expect(sdk.sdk.options).toEqual({ deploymentId: "deployment", resourceGroup: "resource-group" })
+        }),
+    ),
+  )
+
+  it.effect("omits deployment and resourceGroup SDK options when no service key is available", () =>
+    withEnv(
+      { AICORE_SERVICE_KEY: undefined, AICORE_DEPLOYMENT_ID: "deployment", AICORE_RESOURCE_GROUP: "resource-group" },
+      () =>
+        Effect.gen(function* () {
+          const plugin = yield* PluginV2.Service
+          const aisdk = yield* AISDK.Service
+          yield* addPlugin()
+          const sdk = yield* aisdk.runSDK({
+            model: model("sap-ai-core"),
+            package: fixtureProvider,
+            options: { name: "sap-ai-core" },
+          })
+          expect(process.env.AICORE_SERVICE_KEY).toBeUndefined()
+          expect(sdk.sdk.options).toEqual({})
+        }),
+    ),
+  )
+
+  it.effect("uses the callable SDK for language selection", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const sdk = Object.assign((modelID: string) => ({ modelID, provider: "callable" }), {
+        languageModel() {
+          throw new Error("SAP AI Core should call the SDK directly")
+        },
+      })
+      const language = yield* aisdk.runLanguage({ model: model("sap-ai-core"), sdk, options: {} })
+      expect(language.language as unknown).toEqual({ modelID: "sap-model", provider: "callable" })
+    }),
+  )
+
+  it.effect("ignores non-SAP AI Core providers", () =>
+    withEnv(
+      { AICORE_SERVICE_KEY: undefined, AICORE_DEPLOYMENT_ID: "deployment", AICORE_RESOURCE_GROUP: "resource-group" },
+      () =>
+        Effect.gen(function* () {
+          const plugin = yield* PluginV2.Service
+          const aisdk = yield* AISDK.Service
+          yield* addPlugin()
+          const sdk = yield* aisdk.runSDK({
+            model: model("openai"),
+            package: fixtureProvider,
+            options: { name: "openai", serviceKey: "service-key" },
+          })
+          const language = yield* aisdk.runLanguage({
+            model: model("openai"),
+            sdk: () => {
+              throw new Error("SAP AI Core should ignore other providers")
+            },
+            options: {},
+          })
+          expect(process.env.AICORE_SERVICE_KEY).toBeUndefined()
+          expect(sdk.sdk).toBeUndefined()
+          expect(language.language).toBeUndefined()
+        }),
+    ),
+  )
+})

package/test/plugin/provider-snowflake-cortex.test.ts

@@ -0,0 +1,261 @@
+import { AISDK } from "@neurocode-ai/core/aisdk"
+import { describe, expect, it as bun_it } from "bun:test"
+import { Effect } from "effect"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { PluginV2 } from "@neurocode-ai/core/plugin"
+import { PluginHost } from "@neurocode-ai/core/plugin/host"
+import { SnowflakeCortexPlugin, cortexFetch } from "@neurocode-ai/core/plugin/provider/snowflake-cortex"
+import { ProviderPlugins } from "@neurocode-ai/core/plugin/provider"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import { testEffect } from "../lib/effect"
+import { PluginTestLayer } from "./fixture"
+
+const it = testEffect(PluginTestLayer)
+
+const addPlugin = Effect.fn(function* () {
+  const plugin = yield* PluginV2.Service
+  const aisdk = yield* AISDK.Service
+  const host = yield* PluginHost.make(plugin)
+  yield* SnowflakeCortexPlugin.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
+        })
+      }),
+  )
+}
+
+describe("SnowflakeCortexPlugin", () => {
+  it.effect("is registered in ProviderPlugins before OpenAICompatiblePlugin", () =>
+    Effect.sync(() => {
+      expect(ProviderPlugins.map((item) => item.id)).toContain(PluginV2.ID.make("snowflake-cortex"))
+      const ids = ProviderPlugins.map((p) => p.id)
+      expect(ids.indexOf("snowflake-cortex")).toBeLessThan(ids.indexOf("openai-compatible"))
+    }),
+  )
+
+  it.effect("ignores non-snowflake-cortex providers", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("openai"), ModelV2.ID.make("gpt-4")),
+          api: { id: ModelV2.ID.make("gpt-4"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/openai",
+        options: { name: "openai" },
+      })
+      expect(result.sdk).toBeUndefined()
+    }),
+  )
+
+  it.effect("creates SDK for snowflake-cortex using SNOWFLAKE_CORTEX_PAT env var", () =>
+    withEnv({ SNOWFLAKE_CORTEX_PAT: "test-pat" }, () =>
+      Effect.gen(function* () {
+        const plugin = yield* PluginV2.Service
+        const aisdk = yield* AISDK.Service
+        yield* addPlugin()
+        const result = yield* aisdk.runSDK({
+          model: ModelV2.Info.make({
+            ...ModelV2.Info.empty(ProviderV2.ID.make("snowflake-cortex"), ModelV2.ID.make("claude-sonnet-4-6")),
+            api: { id: ModelV2.ID.make("claude-sonnet-4-6"), type: "aisdk", package: "test-provider" },
+          }),
+          package: "@ai-sdk/openai-compatible",
+          options: { name: "snowflake-cortex", baseURL: "https://test.snowflakecomputing.com/api/v2/cortex/v1" },
+        })
+        expect(result.sdk).toBeDefined()
+      }),
+    ),
+  )
+
+  it.effect("falls back to options.apiKey when SNOWFLAKE_CORTEX_PAT env var is absent", () =>
+    withEnv({ SNOWFLAKE_CORTEX_PAT: undefined }, () =>
+      Effect.gen(function* () {
+        const plugin = yield* PluginV2.Service
+        const aisdk = yield* AISDK.Service
+        yield* addPlugin()
+        const result = yield* aisdk.runSDK({
+          model: ModelV2.Info.make({
+            ...ModelV2.Info.empty(ProviderV2.ID.make("snowflake-cortex"), ModelV2.ID.make("claude-sonnet-4-6")),
+            api: { id: ModelV2.ID.make("claude-sonnet-4-6"), type: "aisdk", package: "test-provider" },
+          }),
+          package: "@ai-sdk/openai-compatible",
+          options: {
+            name: "snowflake-cortex",
+            baseURL: "https://test.snowflakecomputing.com/api/v2/cortex/v1",
+            apiKey: "options-pat",
+          },
+        })
+        expect(result.sdk).toBeDefined()
+      }),
+    ),
+  )
+
+  it.effect("uses SNOWFLAKE_CORTEX_TOKEN env var", () =>
+    withEnv({ SNOWFLAKE_CORTEX_TOKEN: "oauth-token", SNOWFLAKE_CORTEX_PAT: undefined }, () =>
+      Effect.gen(function* () {
+        const plugin = yield* PluginV2.Service
+        const aisdk = yield* AISDK.Service
+        yield* addPlugin()
+        const result = yield* aisdk.runSDK({
+          model: ModelV2.Info.make({
+            ...ModelV2.Info.empty(ProviderV2.ID.make("snowflake-cortex"), ModelV2.ID.make("claude-sonnet-4-6")),
+            api: { id: ModelV2.ID.make("claude-sonnet-4-6"), type: "aisdk", package: "test-provider" },
+          }),
+          package: "@ai-sdk/openai-compatible",
+          options: { name: "snowflake-cortex", baseURL: "https://test.snowflakecomputing.com/api/v2/cortex/v1" },
+        })
+        expect(result.sdk).toBeDefined()
+      }),
+    ),
+  )
+
+  it.effect("falls back to options.token when no Snowflake env token is set", () =>
+    withEnv({ SNOWFLAKE_CORTEX_TOKEN: undefined, SNOWFLAKE_CORTEX_PAT: undefined }, () =>
+      Effect.gen(function* () {
+        const plugin = yield* PluginV2.Service
+        const aisdk = yield* AISDK.Service
+        yield* addPlugin()
+        const result = yield* aisdk.runSDK({
+          model: ModelV2.Info.make({
+            ...ModelV2.Info.empty(ProviderV2.ID.make("snowflake-cortex"), ModelV2.ID.make("claude-sonnet-4-6")),
+            api: { id: ModelV2.ID.make("claude-sonnet-4-6"), type: "aisdk", package: "test-provider" },
+          }),
+          package: "@ai-sdk/openai-compatible",
+          options: {
+            name: "snowflake-cortex",
+            baseURL: "https://test.snowflakecomputing.com/api/v2/cortex/v1",
+            token: "options-token",
+          },
+        })
+        expect(result.sdk).toBeDefined()
+      }),
+    ),
+  )
+
+  it.effect("sets includeUsage on the SDK options", () =>
+    withEnv({ SNOWFLAKE_CORTEX_PAT: "test-pat" }, () =>
+      Effect.gen(function* () {
+        const plugin = yield* PluginV2.Service
+        const aisdk = yield* AISDK.Service
+        yield* addPlugin()
+        const result = yield* aisdk.runSDK({
+          model: ModelV2.Info.make({
+            ...ModelV2.Info.empty(ProviderV2.ID.make("snowflake-cortex"), ModelV2.ID.make("claude-sonnet-4-6")),
+            api: { id: ModelV2.ID.make("claude-sonnet-4-6"), type: "aisdk", package: "test-provider" },
+          }),
+          package: "@ai-sdk/openai-compatible",
+          options: { name: "snowflake-cortex", baseURL: "https://test.snowflakecomputing.com/api/v2/cortex/v1" },
+        })
+        expect(result.options.includeUsage).toBe(true)
+      }),
+    ),
+  )
+})
+
+type FetchLike = (url: string | URL | Request, init?: RequestInit) => Promise<Response>
+
+describe("cortexFetch", () => {
+  bun_it("rewrites max_tokens to max_completion_tokens", async () => {
+    const captured: RequestInit[] = []
+    const upstream: FetchLike = async (_url, init) => {
+      captured.push(init ?? {})
+      return new Response("{}", { status: 200 })
+    }
+    await cortexFetch(upstream)("https://test", {
+      method: "POST",
+      body: JSON.stringify({ model: "claude-sonnet-4-6", max_tokens: 1024 }),
+    })
+    const body = JSON.parse(captured[0].body as string)
+    expect(body.max_completion_tokens).toBe(1024)
+    expect(body.max_tokens).toBeUndefined()
+  })
+
+  bun_it("preserves body when max_tokens is absent", async () => {
+    const captured: RequestInit[] = []
+    const upstream: FetchLike = async (_url, init) => {
+      captured.push(init ?? {})
+      return new Response("{}", { status: 200 })
+    }
+    const original = JSON.stringify({ model: "claude-sonnet-4-6", temperature: 0.7 })
+    await cortexFetch(upstream)("https://test", { method: "POST", body: original })
+    expect(captured[0].body).toBe(original)
+  })
+
+  bun_it("treats 400 'conversation complete' as a stop response", async () => {
+    const upstream: FetchLike = async () =>
+      new Response(JSON.stringify({ message: "Conversation complete" }), {
+        status: 400,
+        headers: { "content-type": "application/json" },
+      })
+    const response = await cortexFetch(upstream)("https://test", {})
+    expect(response.status).toBe(200)
+    const data = (await response.json()) as { choices: { finish_reason: string }[] }
+    expect(data.choices[0].finish_reason).toBe("stop")
+  })
+
+  bun_it("passes through other 400 errors unchanged", async () => {
+    const upstream: FetchLike = async () =>
+      new Response(JSON.stringify({ message: "Invalid model" }), {
+        status: 400,
+        headers: { "content-type": "application/json" },
+      })
+    const response = await cortexFetch(upstream)("https://test", {})
+    expect(response.status).toBe(400)
+  })
+
+  bun_it("passes through non-400 errors unchanged", async () => {
+    const upstream: FetchLike = async () => new Response("Unauthorized", { status: 401 })
+    const response = await cortexFetch(upstream)("https://test", {})
+    expect(response.status).toBe(401)
+  })
+
+  bun_it("handles invalid JSON body gracefully without throwing", async () => {
+    const captured: RequestInit[] = []
+    const upstream: FetchLike = async (_url, init) => {
+      captured.push(init ?? {})
+      return new Response("{}", { status: 200 })
+    }
+    const invalidBody = "{ not json }"
+    await cortexFetch(upstream)("https://test", { method: "POST", body: invalidBody })
+    expect(captured[0].body).toBe(invalidBody)
+  })
+
+  bun_it("rewrites role:'' to role:'assistant' in streaming SSE chunks", async () => {
+    const chunk = `data: {"choices":[{"delta":{"role":"","content":"Hi"},"index":0}]}\n\n`
+    const upstream: FetchLike = async () =>
+      new Response(
+        new ReadableStream({
+          start: (ctrl) => {
+            ctrl.enqueue(new TextEncoder().encode(chunk))
+            ctrl.close()
+          },
+        }),
+        {
+          status: 200,
+          headers: { "content-type": "text/event-stream" },
+        },
+      )
+    const response = await cortexFetch(upstream)("https://test", {})
+    const text = await response.text()
+    expect(text).toContain('"role":"assistant"')
+    expect(text).not.toContain('"role":""')
+  })
+})

package/test/plugin/provider-togetherai.test.ts

@@ -0,0 +1,129 @@
+import { AISDK } from "@neurocode-ai/core/aisdk"
+import { describe, expect } from "bun:test"
+import type { LanguageModelV3 } from "@ai-sdk/provider"
+import { Effect } from "effect"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { PluginV2 } from "@neurocode-ai/core/plugin"
+import { PluginHost } from "@neurocode-ai/core/plugin/host"
+import { TogetherAIPlugin } from "@neurocode-ai/core/plugin/provider/togetherai"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import { testEffect } from "../lib/effect"
+import { PluginTestLayer } from "./fixture"
+
+const it = testEffect(PluginTestLayer)
+
+const addPlugin = Effect.fn(function* () {
+  const plugin = yield* PluginV2.Service
+  const aisdk = yield* AISDK.Service
+  const host = yield* PluginHost.make(plugin)
+  yield* TogetherAIPlugin.effect(host)
+})
+
+function fakeSelectorSdk(calls: string[]) {
+  const make = (method: string) => (id: string) => {
+    calls.push(`${method}:${id}`)
+    return { modelId: id, provider: method, specificationVersion: "v3" } as unknown as LanguageModelV3
+  }
+  return {
+    responses: make("responses"),
+    messages: make("messages"),
+    chat: make("chat"),
+    languageModel: make("languageModel"),
+  }
+}
+
+describe("TogetherAIPlugin", () => {
+  it.effect("creates a TogetherAI SDK for @ai-sdk/togetherai", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("togetherai"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/togetherai",
+        options: { name: "togetherai" },
+      })
+      expect(result.sdk).toBeDefined()
+    }),
+  )
+
+  it.effect("matches the old bundled provider package exactly", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+
+      const ignored = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("togetherai"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "file:///tmp/@ai-sdk/togetherai-provider.js",
+        options: { name: "togetherai" },
+      })
+      expect(ignored.sdk).toBeUndefined()
+
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("togetherai"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/togetherai",
+        options: { name: "togetherai" },
+      })
+      expect(result.sdk).toBeDefined()
+    }),
+  )
+
+  it.effect("creates bundled TogetherAI SDKs for custom provider IDs", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("custom-togetherai"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/togetherai",
+        options: { name: "custom-togetherai" },
+      })
+
+      expect(result.sdk.languageModel("model").provider).toBe("togetherai.chat")
+    }),
+  )
+
+  it.effect("defaults language selection to sdk.languageModel with the model API ID", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      const calls: string[] = []
+      yield* addPlugin()
+
+      const result = yield* aisdk.runLanguage({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(
+            ProviderV2.ID.make("togetherai"),
+            ModelV2.ID.make("meta-llama/Llama-3.3-70B-Instruct-Turbo"),
+          ),
+          api: {
+            id: ModelV2.ID.make("meta-llama/Llama-3.3-70B-Instruct-Turbo"),
+            type: "aisdk",
+            package: "test-provider",
+          },
+        }),
+        sdk: { languageModel: fakeSelectorSdk(calls).languageModel },
+        options: {},
+      })
+
+      expect(result.language).toBeUndefined()
+      expect(calls).toEqual([])
+      expect(result.language ?? fakeSelectorSdk(calls).languageModel(result.model.api.id)).toBeDefined()
+      expect(calls).toEqual(["languageModel:meta-llama/Llama-3.3-70B-Instruct-Turbo"])
+    }),
+  )
+})

package/test/plugin/provider-venice.test.ts

@@ -0,0 +1,115 @@
+import { AISDK } from "@neurocode-ai/core/aisdk"
+import { describe, expect } from "bun:test"
+import type { LanguageModelV3 } from "@ai-sdk/provider"
+import { Effect } from "effect"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { PluginV2 } from "@neurocode-ai/core/plugin"
+import { PluginHost } from "@neurocode-ai/core/plugin/host"
+import { VenicePlugin } from "@neurocode-ai/core/plugin/provider/venice"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import { testEffect } from "../lib/effect"
+import { PluginTestLayer } from "./fixture"
+
+const it = testEffect(PluginTestLayer)
+
+const addPlugin = Effect.fn(function* () {
+  const plugin = yield* PluginV2.Service
+  const aisdk = yield* AISDK.Service
+  const host = yield* PluginHost.make(plugin)
+  yield* VenicePlugin.effect(host)
+})
+
+function fakeSelectorSdk(calls: string[]) {
+  const make = (method: string) => (id: string) => {
+    calls.push(`${method}:${id}`)
+    return { modelId: id, provider: method, specificationVersion: "v3" } as unknown as LanguageModelV3
+  }
+  return {
+    responses: make("responses"),
+    messages: make("messages"),
+    chat: make("chat"),
+    languageModel: make("languageModel"),
+  }
+}
+
+describe("VenicePlugin", () => {
+  it.effect("creates a Venice SDK for venice-ai-sdk-provider", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("venice"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "venice-ai-sdk-provider",
+        options: { name: "venice" },
+      })
+      expect(result.sdk).toBeDefined()
+    }),
+  )
+
+  it.effect("uses the model provider ID as the bundled Venice SDK name", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("custom-venice"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "venice-ai-sdk-provider",
+        options: { name: "custom-venice", apiKey: "test" },
+      })
+      expect(result.sdk).toBeDefined()
+      expect(result.sdk.languageModel("model").provider).toBe("custom-venice.chat")
+    }),
+  )
+
+  it.effect("only handles the bundled venice-ai-sdk-provider package", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const similar = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("venice"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "file:///tmp/venice-ai-sdk-provider.js",
+        options: { name: "venice" },
+      })
+      const other = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("venice"), ModelV2.ID.make("model")),
+          api: { id: ModelV2.ID.make("model"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/openai-compatible",
+        options: { name: "venice" },
+      })
+      expect(similar.sdk).toBeUndefined()
+      expect(other.sdk).toBeUndefined()
+    }),
+  )
+
+  it.effect("leaves Venice language selection to the default languageModel fallback", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      const calls: string[] = []
+      yield* addPlugin()
+      const result = yield* aisdk.runLanguage({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("venice"), ModelV2.ID.make("alias")),
+          api: { id: ModelV2.ID.make("alias"), type: "aisdk", package: "test-provider" },
+        }),
+        sdk: fakeSelectorSdk(calls),
+        options: {},
+      })
+      expect(calls).toEqual([])
+      expect(result.language).toBeUndefined()
+    }),
+  )
+})