1import { Type, Kind, type Static, type TObject, type TProperties } from "@sinclair/typebox";
 2
 3const CustomModelSchema = Type.Object({
 4  provider: Type.String(),
 5  api: Type.String(),
 6  base_url: Type.String(),
 7  id: Type.String(),
 8  name: Type.String(),
 9  reasoning: Type.Boolean(),
10  input: Type.Array(Type.Union([Type.Literal("text"), Type.Literal("image")])),
11  cost: Type.Object({
12    input: Type.Number(),
13    output: Type.Number(),
14    cache_read: Type.Optional(Type.Number()),
15    cache_write: Type.Optional(Type.Number()),
16  }),
17  context_window: Type.Number(),
18  max_tokens: Type.Number(),
19  api_key: Type.Optional(Type.String()),
20  headers: Type.Optional(Type.Record(Type.String(), Type.String())),
21  compat: Type.Optional(
22    Type.Object({
23      supports_store: Type.Optional(Type.Boolean()),
24      supports_developer_role: Type.Optional(Type.Boolean()),
25      supports_reasoning_effort: Type.Optional(Type.Boolean()),
26      supports_usage_in_streaming: Type.Optional(Type.Boolean()),
27      max_tokens_field: Type.Optional(Type.Union([Type.Literal("max_tokens"), Type.Literal("max_completion_tokens")])),
28      requires_tool_result_name: Type.Optional(Type.Boolean()),
29      requires_assistant_after_tool_result: Type.Optional(Type.Boolean()),
30      requires_thinking_as_text: Type.Optional(Type.Boolean()),
31      requires_mistral_tool_ids: Type.Optional(Type.Boolean()),
32      thinking_format: Type.Optional(Type.Union([Type.Literal("openai"), Type.Literal("zai")])),
33    })
34  ),
35});
36
37export const ConfigSchema = Type.Object({
38  defaults: Type.Object({
39    model: Type.String(),
40    cleanup: Type.Boolean(),
41    kagi_session_token: Type.Optional(Type.String()),
42    tabstack_api_key: Type.Optional(Type.String()),
43  }),
44  web: Type.Object({
45    model: Type.Optional(Type.String()),
46    system_prompt_path: Type.Optional(Type.String()),
47    kagi_session_token: Type.Optional(Type.String()),
48    tabstack_api_key: Type.Optional(Type.String()),
49  }),
50  repo: Type.Object({
51    model: Type.Optional(Type.String()),
52    system_prompt_path: Type.Optional(Type.String()),
53    default_depth: Type.Optional(Type.Number({ minimum: 1 })),
54    blob_limit: Type.Optional(Type.String()),
55  }),
56  custom_models: Type.Optional(Type.Record(Type.String(), CustomModelSchema)),
57});
58
59/** Deep-partial version of ConfigSchema for validating TOML override files. */
60export function partialObject<T extends TProperties>(schema: TObject<T>) {
61  const partial: Record<string, unknown> = {};
62  for (const [key, value] of Object.entries(schema.properties)) {
63    const v = value as any;
64    const inner = v[Kind] === 'Object' && v.properties ? partialObject(v) : v;
65    partial[key] = Type.Optional(inner as any);
66  }
67  return Type.Object(partial as any);
68}
69
70export const PartialConfigSchema = Type.Object({
71  defaults: Type.Optional(partialObject(ConfigSchema.properties.defaults)),
72  web: Type.Optional(partialObject(ConfigSchema.properties.web)),
73  repo: Type.Optional(partialObject(ConfigSchema.properties.repo)),
74  custom_models: Type.Optional(Type.Record(Type.String(), CustomModelSchema)),
75});
76
77export type RumiloConfig = Static<typeof ConfigSchema>;
78export type CustomModelConfig = Static<typeof CustomModelSchema>;