66 lines
2.8 KiB
TypeScript
66 lines
2.8 KiB
TypeScript
import path from "path"
|
|
import { describe, expect } from "bun:test"
|
|
import { Effect, Layer } from "effect"
|
|
import { Catalog } from "@opencode-ai/core/catalog"
|
|
import { Connector } from "@opencode-ai/core/connector"
|
|
import { Credential } from "@opencode-ai/core/credential"
|
|
import { Database } from "@opencode-ai/core/database/database"
|
|
import { EventV2 } from "@opencode-ai/core/event"
|
|
import { Flag } from "@opencode-ai/core/flag/flag"
|
|
import { Location } from "@opencode-ai/core/location"
|
|
import { ModelsDev } from "@opencode-ai/core/models-dev"
|
|
import { PluginV2 } from "@opencode-ai/core/plugin"
|
|
import { ModelsDevPlugin } from "@opencode-ai/core/plugin/models-dev"
|
|
import { Policy } from "@opencode-ai/core/policy"
|
|
import { AbsolutePath } from "@opencode-ai/core/schema"
|
|
import { location } from "../fixture/location"
|
|
import { testEffect } from "../lib/effect"
|
|
|
|
const events = EventV2.defaultLayer
|
|
const locationLayer = Layer.succeed(
|
|
Location.Service,
|
|
Location.Service.of(location({ directory: AbsolutePath.make(import.meta.dir) })),
|
|
)
|
|
const plugins = PluginV2.layer.pipe(Layer.provide(events))
|
|
const policy = Policy.layer.pipe(Layer.provide(locationLayer))
|
|
const credentials = Credential.layer.pipe(Layer.provide(Database.layerFromPath(":memory:")), Layer.provide(events))
|
|
const catalog = Catalog.layer.pipe(Layer.provide(Layer.mergeAll(events, locationLayer, plugins, policy, credentials)))
|
|
const connectors = Connector.locationLayer.pipe(Layer.provide(credentials), Layer.provide(events))
|
|
const layer = Layer.mergeAll(catalog, connectors, credentials, events, locationLayer, plugins)
|
|
const it = testEffect(layer)
|
|
|
|
describe("ModelsDevPlugin", () => {
|
|
it.effect("registers key connectors for providers with environment variables", () =>
|
|
Effect.acquireUseRelease(
|
|
Effect.sync(() => {
|
|
const previous = {
|
|
path: Flag.OPENCODE_MODELS_PATH,
|
|
disabled: Flag.OPENCODE_DISABLE_MODELS_FETCH,
|
|
}
|
|
Flag.OPENCODE_MODELS_PATH = path.join(import.meta.dir, "fixtures", "models-dev.json")
|
|
Flag.OPENCODE_DISABLE_MODELS_FETCH = true
|
|
return previous
|
|
}),
|
|
() =>
|
|
Effect.gen(function* () {
|
|
yield* ModelsDevPlugin.effect
|
|
const connectors = yield* Connector.Service
|
|
expect(yield* connectors.list()).toEqual([
|
|
new Connector.Info({
|
|
id: Connector.ID.make("acme"),
|
|
name: "Acme",
|
|
methods: [
|
|
new Connector.KeyMethod({ id: Connector.MethodID.make("api-key"), type: "key", label: "API Key" }),
|
|
],
|
|
}),
|
|
])
|
|
}).pipe(Effect.provide(ModelsDev.defaultLayer)),
|
|
(previous) =>
|
|
Effect.sync(() => {
|
|
Flag.OPENCODE_MODELS_PATH = previous.path
|
|
Flag.OPENCODE_DISABLE_MODELS_FETCH = previous.disabled
|
|
}),
|
|
),
|
|
)
|
|
})
|