| import { AISDK } from "@opencode-ai/core/aisdk" |
| import { describe, expect } from "bun:test" |
| import type { LanguageModelV3 } from "@ai-sdk/provider" |
| import { Effect } from "effect" |
| import { Catalog } from "@opencode-ai/core/catalog" |
| import { Integration } from "@opencode-ai/core/integration" |
| import { ModelV2 } from "@opencode-ai/core/model" |
| import { PluginV2 } from "@opencode-ai/core/plugin" |
| import { PluginHost } from "@opencode-ai/core/plugin/host" |
| import { OpenAIPlugin } from "@opencode-ai/core/plugin/provider/openai" |
| 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* () { |
| const plugin = yield* PluginV2.Service |
| const aisdk = yield* AISDK.Service |
| const host = yield* PluginHost.make(plugin) |
| const integrations = yield* Integration.Service |
| yield* OpenAIPlugin.effect(host).pipe(Effect.provideService(Integration.Service, integrations)) |
| }) |
|
|
| function required<T>(value: T | undefined): T { |
| if (value === undefined) throw new Error("Expected value") |
| return value |
| } |
|
|
| 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("OpenAIPlugin", () => { |
| it.effect("registers browser and headless ChatGPT OAuth methods", () => |
| Effect.gen(function* () { |
| yield* addPlugin() |
| expect((yield* (yield* Integration.Service).get(Integration.ID.make("openai")))?.methods).toEqual([ |
| { |
| id: Integration.MethodID.make("chatgpt-browser"), |
| type: "oauth", |
| label: "ChatGPT Pro/Plus (browser)", |
| }, |
| { |
| id: Integration.MethodID.make("chatgpt-headless"), |
| type: "oauth", |
| label: "ChatGPT Pro/Plus (headless)", |
| }, |
| ]) |
| }), |
| ) |
|
|
| it.effect("creates an OpenAI SDK for @ai-sdk/openai using the provider ID as 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-openai"), ModelV2.ID.make("gpt-5")), |
| api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/openai", |
| options: { name: "custom-openai", apiKey: "test" }, |
| }) |
| expect(result.sdk?.responses("gpt-5").provider).toBe("custom-openai.responses") |
| }), |
| ) |
|
|
| it.effect("ignores non-OpenAI SDK packages", () => |
| 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.openai, ModelV2.ID.make("gpt-5")), |
| api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| package: "@ai-sdk/openai-compatible", |
| options: { name: "openai" }, |
| }) |
| expect(result.sdk).toBeUndefined() |
| }), |
| ) |
|
|
| it.effect("uses the Responses API for language models", () => |
| 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.openai, ModelV2.ID.make("alias")), |
| api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| sdk: fakeSelectorSdk(calls), |
| options: {}, |
| }) |
| expect(calls).toEqual(["responses:gpt-5"]) |
| expect(result.language).toBeDefined() |
| }), |
| ) |
|
|
| it.effect("ignores non-OpenAI providers", () => |
| 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.anthropic, ModelV2.ID.make("gpt-5")), |
| api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" }, |
| }), |
| sdk: fakeSelectorSdk(calls), |
| options: {}, |
| }) |
| expect(calls).toEqual([]) |
| expect(result.language).toBeUndefined() |
| }), |
| ) |
|
|
| it.effect("disables gpt-5-chat-latest during catalog transforms", () => |
| Effect.gen(function* () { |
| const catalog = yield* Catalog.Service |
| yield* catalog.transform((catalog) => { |
| const item = ProviderV2.Info.make({ |
| ...ProviderV2.Info.empty(ProviderV2.ID.openai), |
| api: { type: "aisdk", package: "@ai-sdk/openai" }, |
| }) |
| catalog.provider.update(item.id, (draft) => { |
| draft.api = item.api |
| }) |
| catalog.model.update(item.id, ModelV2.ID.make("gpt-5"), () => {}) |
| catalog.model.update(item.id, ModelV2.ID.make("gpt-5-chat-latest"), () => {}) |
| }) |
| yield* addPlugin() |
| expect(required(yield* catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-5"))).enabled).toBe(true) |
| expect( |
| required(yield* catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-5-chat-latest"))).enabled, |
| ).toBe(false) |
| }), |
| ) |
|
|
| it.effect("does not disable gpt-5-chat-latest for non-OpenAI providers", () => |
| Effect.gen(function* () { |
| const catalog = yield* Catalog.Service |
| yield* catalog.transform((catalog) => { |
| const item = ProviderV2.Info.make({ |
| ...ProviderV2.Info.empty(ProviderV2.ID.make("custom-openai")), |
| api: { type: "aisdk", package: "test-provider" }, |
| }) |
| catalog.provider.update(item.id, (draft) => { |
| draft.api = item.api |
| }) |
| catalog.model.update(item.id, ModelV2.ID.make("gpt-5-chat-latest"), () => {}) |
| }) |
| yield* addPlugin() |
| expect( |
| required(yield* catalog.model.get(ProviderV2.ID.make("custom-openai"), ModelV2.ID.make("gpt-5-chat-latest"))) |
| .enabled, |
| ).toBe(true) |
| }), |
| ) |
| }) |
|
|
