mirror of
https://github.com/anomalyco/opencode.git
synced 2026-07-21 10:16:03 +00:00
80 lines
3.3 KiB
TypeScript
80 lines
3.3 KiB
TypeScript
import { describe, expect } from "bun:test"
|
|
import { Catalog } from "@opencode-ai/core/catalog"
|
|
import { Credential } from "@opencode-ai/core/credential"
|
|
import { EventV2 } from "@opencode-ai/core/event"
|
|
import { Integration } from "@opencode-ai/core/integration"
|
|
import { Location } from "@opencode-ai/core/location"
|
|
import { ModelV2 } from "@opencode-ai/core/model"
|
|
import { Policy } from "@opencode-ai/core/policy"
|
|
import { VariantPlugin } from "@opencode-ai/core/plugin/variant"
|
|
import { ProviderV2 } from "@opencode-ai/core/provider"
|
|
import { AbsolutePath } from "@opencode-ai/core/schema"
|
|
import { Effect, Layer } from "effect"
|
|
import { location } from "../fixture/location"
|
|
import { testEffect } from "../lib/effect"
|
|
import { catalogHost, host } from "./host"
|
|
|
|
const events = EventV2.defaultLayer
|
|
const locationLayer = Layer.succeed(
|
|
Location.Service,
|
|
Location.Service.of(location({ directory: AbsolutePath.make(import.meta.dir) })),
|
|
)
|
|
const connections = Credential.defaultLayer.pipe(Layer.fresh)
|
|
const integrations = Integration.locationLayer.pipe(Layer.provide(events), Layer.provide(connections))
|
|
const catalog = Catalog.layer.pipe(
|
|
Layer.provide(
|
|
Layer.mergeAll(events, locationLayer, Policy.layer.pipe(Layer.provide(locationLayer)), connections, integrations),
|
|
),
|
|
)
|
|
const it = testEffect(
|
|
Layer.mergeAll(catalog.pipe(Layer.provide(connections)), integrations, connections, events, locationLayer),
|
|
)
|
|
|
|
describe("VariantPlugin", () => {
|
|
it.effect("adds GLM 5.2 variants after catalog sources", () =>
|
|
Effect.gen(function* () {
|
|
const service = yield* Catalog.Service
|
|
yield* service.transform((catalog) => {
|
|
catalog.provider.update(ProviderV2.ID.opencode, (provider) => {
|
|
provider.api = { type: "aisdk", package: "@ai-sdk/openai-compatible" }
|
|
})
|
|
catalog.model.update(ProviderV2.ID.opencode, ModelV2.ID.make("glm-5.2"), (model) => {
|
|
model.api = {
|
|
id: ModelV2.ID.make("glm-5.2"),
|
|
type: "aisdk",
|
|
package: "@ai-sdk/openai-compatible",
|
|
}
|
|
})
|
|
})
|
|
yield* VariantPlugin.Plugin.effect(host({ catalog: catalogHost(service) }))
|
|
|
|
expect((yield* service.model.get(ProviderV2.ID.opencode, ModelV2.ID.make("glm-5.2")))?.variants).toEqual([
|
|
expect.objectContaining({ id: "high", body: { reasoning_effort: "high" } }),
|
|
expect.objectContaining({ id: "max", body: { reasoning_effort: "max" } }),
|
|
])
|
|
}),
|
|
)
|
|
|
|
it.effect("keeps explicit variants over generated defaults", () =>
|
|
Effect.gen(function* () {
|
|
const service = yield* Catalog.Service
|
|
yield* service.transform((catalog) => {
|
|
catalog.model.update(ProviderV2.ID.opencode, ModelV2.ID.make("glm-5.2"), (model) => {
|
|
model.api = {
|
|
id: ModelV2.ID.make("glm-5.2"),
|
|
type: "aisdk",
|
|
package: "@ai-sdk/openai-compatible",
|
|
}
|
|
model.variants = [{ id: ModelV2.VariantID.make("high"), headers: { custom: "true" }, body: {} }]
|
|
})
|
|
})
|
|
yield* VariantPlugin.Plugin.effect(host({ catalog: catalogHost(service) }))
|
|
|
|
expect((yield* service.model.get(ProviderV2.ID.opencode, ModelV2.ID.make("glm-5.2")))?.variants).toEqual([
|
|
expect.objectContaining({ id: "high", headers: { custom: "true" } }),
|
|
expect.objectContaining({ id: "max", body: { reasoning_effort: "max" } }),
|
|
])
|
|
}),
|
|
)
|
|
})
|