wip: zen

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 {