opencode-agent[bot]
61 lines
2.0 KiB
TypeScript
61 lines
2.0 KiB
TypeScript
import { describe, expect, test } from "bun:test"
|
|
import { Schema } from "effect"
|
|
import { Model } from "../src/model.js"
|
|
import { Provider } from "../src/provider.js"
|
|
|
|
describe("Model.Ref", () => {
|
|
test("parses model references with optional variants", () => {
|
|
const variant = Model.Ref.parse("openrouter/openai/gpt-5#high")
|
|
expect(String(variant.providerID)).toBe("openrouter")
|
|
expect(String(variant.id)).toBe("openai/gpt-5")
|
|
expect(String(variant.variant)).toBe("high")
|
|
|
|
const standard = Model.Ref.parse("anthropic/claude-sonnet")
|
|
expect(String(standard.providerID)).toBe("anthropic")
|
|
expect(String(standard.id)).toBe("claude-sonnet")
|
|
expect(standard.variant).toBeUndefined()
|
|
})
|
|
|
|
test("rejects malformed model references", () => {
|
|
expect(() => Model.Ref.parse("gpt-5")).toThrow()
|
|
expect(() => Model.Ref.parse("openai/gpt-5#")).toThrow()
|
|
expect(() => Model.Ref.parse("openai/gpt-5#high#extra")).toThrow()
|
|
})
|
|
})
|
|
|
|
describe("Model.ReasoningField", () => {
|
|
test("accepts suggested and custom fields", () => {
|
|
const decode = Schema.decodeUnknownSync(Model.ReasoningField)
|
|
|
|
for (const field of ["reasoning", "reasoning_content", "reasoning_text", "vendor_reasoning"])
|
|
expect(decode(field)).toBe(field)
|
|
})
|
|
})
|
|
|
|
describe("Model.Compatibility", () => {
|
|
test("decodes model compatibility overrides", () => {
|
|
const decode = Schema.decodeUnknownSync(Model.Compatibility)
|
|
|
|
expect(decode({})).toEqual({})
|
|
expect(
|
|
decode({
|
|
reasoningField: "vendor_reasoning",
|
|
maxTokensField: "max_completion_tokens",
|
|
requireFinishReason: false,
|
|
}),
|
|
).toEqual({
|
|
reasoningField: "vendor_reasoning",
|
|
maxTokensField: "max_completion_tokens",
|
|
requireFinishReason: false,
|
|
})
|
|
})
|
|
})
|
|
|
|
describe("Model.Info", () => {
|
|
test("uses practical token limits for unknown models", () => {
|
|
const model = Model.Info.default(Provider.ID.make("custom"), Model.ID.make("gpt-5.6"))
|
|
|
|
expect(model.limit).toEqual({ context: 200_000, output: 32_000 })
|
|
})
|
|
})
|