wip: zen lite

1 files changed, 4 insertions, 3 deletions

diff --git a/packages/console/core/src/model.ts b/packages/console/core/src/model.ts
index 6011cac37..e868b176e 100644
--- a/packages/console/core/src/model.ts
+++ b/packages/console/core/src/model.ts
@@ -73,6 +73,7 @@ export namespace ZenData {
 
   const ModelsSchema = z.object({
     models: z.record(z.string(), z.union([ModelSchema, z.array(ModelSchema.extend({ formatFilter: FormatSchema }))])),
+    liteModels: z.record(z.string(), ModelSchema),
     providers: z.record(z.string(), ProviderSchema),
     providerFamilies: z.record(z.string(), ProviderFamilySchema),
   })
@@ -81,7 +82,7 @@ export namespace ZenData {
     return input
   })
 
-  export const list = fn(z.void(), () => {
+  export const list = fn(z.enum(["lite", "full"]), (modelList) => {
     const json = JSON.parse(
       Resource.ZEN_MODELS1.value +
         Resource.ZEN_MODELS2.value +
@@ -114,9 +115,9 @@ export namespace ZenData {
         Resource.ZEN_MODELS29.value +
         Resource.ZEN_MODELS30.value,
     )
-    const { models, providers, providerFamilies } = ModelsSchema.parse(json)
+    const { models, liteModels, providers, providerFamilies } = ModelsSchema.parse(json)
     return {
-      models,
+      models: modelList === "lite" ? liteModels : models,
       providers: Object.fromEntries(
         Object.entries(providers).map(([id, provider]) => [
           id,