diff options
Diffstat (limited to 'packages/console/core/src/model.ts')
| -rw-r--r-- | packages/console/core/src/model.ts | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/packages/console/core/src/model.ts b/packages/console/core/src/model.ts index 018df5b54..300f92ed0 100644 --- a/packages/console/core/src/model.ts +++ b/packages/console/core/src/model.ts @@ -7,7 +7,7 @@ import { fn } from "./util/fn" import { Actor } from "./actor" import { Resource } from "@opencode-ai/console-resource" -export namespace ZenModel { +export namespace ZenData { const ModelCostSchema = z.object({ input: z.number(), output: z.number(), @@ -16,7 +16,7 @@ export namespace ZenModel { cacheWrite1h: z.number().optional(), }) - export const ModelSchema = z.object({ + const ModelSchema = z.object({ name: z.string(), cost: ModelCostSchema, cost200K: ModelCostSchema.optional(), @@ -24,19 +24,32 @@ export namespace ZenModel { providers: z.array( z.object({ id: z.string(), - api: z.string(), - apiKey: z.string(), model: z.string(), weight: z.number().optional(), - headerMappings: z.record(z.string(), z.string()).optional(), disabled: z.boolean().optional(), }), ), }) - export const ModelsSchema = z.record(z.string(), ModelSchema) + const ProviderSchema = z.object({ + api: z.string(), + apiKey: z.string(), + headerMappings: z.record(z.string(), z.string()).optional(), + }) + + const ModelsSchema = z.object({ + models: z.record(z.string(), ModelSchema), + providers: z.record(z.string(), ProviderSchema), + }) - export const list = fn(z.void(), () => ModelsSchema.parse(JSON.parse(Resource.ZEN_MODELS.value))) + export const validate = fn(ModelsSchema, (input) => { + return input + }) + + export const list = fn(z.void(), () => { + const json = JSON.parse(Resource.ZEN_MODELS.value) + return ModelsSchema.parse(json) + }) } export namespace Model { |
