openAgent

1import { Effect, Layer } from "effect"
2import { Provider } from "@/provider/provider"
3import { ProviderV2 } from "@opencode-ai/core/provider"
4import { ModelV2 } from "@opencode-ai/core/model"
5 
6export namespace ProviderTest {
7  export function model(override: Partial<Provider.Model> = {}): Provider.Model {
8    const id = override.id ?? ModelV2.ID.make("gpt-5.2")
9    const providerID = override.providerID ?? ProviderV2.ID.make("openai")
10    return {
11      id,
12      providerID,
13      name: "Test Model",
14      capabilities: {
15        toolcall: true,
16        attachment: false,
17        reasoning: false,
18        temperature: true,
19        interleaved: false,
20        input: { text: true, image: false, audio: false, video: false, pdf: false },
21        output: { text: true, image: false, audio: false, video: false, pdf: false },
22      },
23      api: { id, url: "https://example.com", npm: "@ai-sdk/openai" },
24      cost: { input: 0, output: 0, cache: { read: 0, write: 0 } },
25      limit: { context: 200_000, output: 10_000 },
26      status: "active",
27      options: {},
28      headers: {},
29      release_date: "2025-01-01",
30      ...override,
31    }
32  }
33 
34  export function info(override: Partial<Provider.Info> = {}, mdl = model()): Provider.Info {
35    const id = override.id ?? mdl.providerID
36    return {
37      id,
38      name: "Test Provider",
39      source: "config",
40      env: [],
41      options: {},
42      models: { [mdl.id]: mdl },
43      ...override,
44    }
45  }
46 
47  export function fake(override: Partial<Provider.Interface> & { model?: Provider.Model; info?: Provider.Info } = {}) {
48    const mdl = override.model ?? model()
49    const row = override.info ?? info({}, mdl)
50    return {
51      model: mdl,
52      info: row,
53      layer: Layer.succeed(
54        Provider.Service,
55        Provider.Service.of({
56          list: Effect.fn("TestProvider.list")(() => Effect.succeed({ [row.id]: row })),
57          getProvider: Effect.fn("TestProvider.getProvider")((providerID) => {
58            if (providerID === row.id) return Effect.succeed(row)
59            return Effect.die(new Error(`Unknown test provider: ${providerID}`))
60          }),
61          getModel: Effect.fn("TestProvider.getModel")((providerID, modelID) => {
62            if (providerID === row.id && modelID === mdl.id) return Effect.succeed(mdl)
63            return Effect.die(new Error(`Unknown test model: ${providerID}/${modelID}`))
64          }),
65          getLanguage: Effect.fn("TestProvider.getLanguage")(() =>
66            Effect.die(new Error("ProviderTest.getLanguage not configured")),
67          ),
68          closest: Effect.fn("TestProvider.closest")((providerID) =>
69            Effect.succeed(providerID === row.id ? { providerID: row.id, modelID: mdl.id } : undefined),
70          ),
71          getSmallModel: Effect.fn("TestProvider.getSmallModel")((providerID) =>
72            Effect.succeed(providerID === row.id ? mdl : undefined),
73          ),
74          defaultModel: Effect.fn("TestProvider.defaultModel")(() =>
75            Effect.succeed({ providerID: row.id, modelID: mdl.id }),
76          ),
77          ...override,
78        }),
79      ),
80    }
81  }
82}
83