captify 0.1.18 → 0.1.19

package/dist/llm-providers/llm-providers.schemas.d.ts

@@ -57,8 +57,9 @@ export declare const providerTypeSchema: z.ZodEnum<{
  * model capabilities (e.g., chat, embeddings, vision).
  */
 export declare const modelCapabilitySchema: z.ZodEnum<{
-    streaming: "streaming";
+    document: "document";
     chat: "chat";
+    streaming: "streaming";
     vision: "vision";
     thinking: "thinking";
     "function-calling": "function-calling";
@@ -151,7 +152,7 @@ export declare const descriptionSchema: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodN
  * baseUrlSchema.parse(""); // Valid - becomes null
  * baseUrlSchema.parse(null); // Valid
  */
-export declare const baseUrlSchema: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null, string | null | undefined>>, z.ZodNullable<z.ZodString>>;
+export declare const baseUrlSchema: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
 /**
  * Pricing schema for token costs.
  *
@@ -162,14 +163,52 @@ export declare const pricingSchema: z.ZodOptional<z.ZodObject<{
     inputPerMillion: z.ZodNumber;
     outputPerMillion: z.ZodNumber;
 }, z.core.$strip>>;
+/**
+ * Budget configuration schema for per-model spend and rate limits.
+ *
+ * @description Defines budget and rate limiting configuration for a model.
+ * Budgets auto-reset when the period expires based on budgetStartDate.
+ *
+ * This schema is designed to be extensible for future enhancements:
+ * - Team-based limits: Can be extended to support team/group scoping
+ * - Token quotas: tokenLimit field can be added for raw token limits
+ * - User limits: Can create user-level budget configs with same structure
+ *
+ * @example
+ * {
+ *   budgetAmount: 100.00,       // $100 budget
+ *   budgetPeriodDays: 30,       // 30-day period
+ *   budgetStartDate: 1704067200000, // When period started (Unix ms)
+ *   requestsPerMinute: 60,      // 60 requests per minute limit
+ *   tokenLimit: 1000000         // Optional: 1M token limit per period
+ * }
+ */
+export declare const budgetConfigSchema: z.ZodOptional<z.ZodObject<{
+    budgetAmount: z.ZodNumber;
+    budgetPeriodType: z.ZodDefault<z.ZodEnum<{
+        custom: "custom";
+        monthly: "monthly";
+        weekly: "weekly";
+        daily: "daily";
+    }>>;
+    budgetPeriodDays: z.ZodOptional<z.ZodNumber>;
+    budgetStartDate: z.ZodOptional<z.ZodNumber>;
+    requestsPerMinute: z.ZodNumber;
+    tokenLimit: z.ZodOptional<z.ZodNumber>;
+    isEnabled: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+}, z.core.$strip>>;
 /**
  * Provider schema for validating DynamoDB data.
  *
  * @description Validates the structure of LLM provider configurations
- * retrieved from DynamoDB.
+ * retrieved from DynamoDB. Extends baseEntitySchema for id, createdAt, updatedAt.
  */
 export declare const providerSchema: z.ZodObject<{
     id: z.ZodString;
+    createdAt: z.ZodNumber;
+    updatedAt: z.ZodNumber;
+    createdById: z.ZodString;
+    modifiedById: z.ZodOptional<z.ZodString>;
     name: z.ZodString;
     type: z.ZodEnum<{
         openai: "openai";
@@ -179,23 +218,27 @@ export declare const providerSchema: z.ZodObject<{
     description: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     baseUrl: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     isEnabled: z.ZodBoolean;
-    createdAt: z.ZodInt;
-    updatedAt: z.ZodInt;
 }, z.core.$strip>;
 /**
  * Model schema for validating DynamoDB data.
  *
  * @description Validates the structure of LLM model configurations
- * retrieved from DynamoDB.
+ * retrieved from DynamoDB. Extends baseEntitySchema for id, createdAt, updatedAt.
  */
 export declare const modelSchema: z.ZodObject<{
     id: z.ZodString;
+    createdAt: z.ZodNumber;
+    updatedAt: z.ZodNumber;
+    createdById: z.ZodString;
+    modifiedById: z.ZodOptional<z.ZodString>;
+    modelId: z.ZodString;
     providerId: z.ZodString;
     name: z.ZodString;
     description: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     capabilities: z.ZodDefault<z.ZodArray<z.ZodEnum<{
-        streaming: "streaming";
+        document: "document";
         chat: "chat";
+        streaming: "streaming";
         vision: "vision";
         thinking: "thinking";
         "function-calling": "function-calling";
@@ -203,10 +246,25 @@ export declare const modelSchema: z.ZodObject<{
     }>>>;
     contextWindow: z.ZodInt;
     maxOutputTokens: z.ZodInt;
+    thinkingBudgetTokens: z.ZodOptional<z.ZodNullable<z.ZodInt>>;
     pricing: z.ZodOptional<z.ZodObject<{
         inputPerMillion: z.ZodNumber;
         outputPerMillion: z.ZodNumber;
     }, z.core.$strip>>;
+    budgetConfig: z.ZodOptional<z.ZodObject<{
+        budgetAmount: z.ZodNumber;
+        budgetPeriodType: z.ZodDefault<z.ZodEnum<{
+            custom: "custom";
+            monthly: "monthly";
+            weekly: "weekly";
+            daily: "daily";
+        }>>;
+        budgetPeriodDays: z.ZodOptional<z.ZodNumber>;
+        budgetStartDate: z.ZodOptional<z.ZodNumber>;
+        requestsPerMinute: z.ZodNumber;
+        tokenLimit: z.ZodOptional<z.ZodNumber>;
+        isEnabled: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+    }, z.core.$strip>>;
     isAvailable: z.ZodBoolean;
     isBasicAssistantDefault: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
     parameterConstraints: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodObject<{
@@ -219,8 +277,6 @@ export declare const modelSchema: z.ZodObject<{
             frequencyPenalty: "frequencyPenalty";
         }>>>;
     }, z.core.$strip>>>>;
-    createdAt: z.ZodInt;
-    updatedAt: z.ZodInt;
 }, z.core.$strip>;
 /**
  * Schema for getProviderById() method - get a single provider.
@@ -244,6 +300,7 @@ export declare const getProviderByIdParamsSchema: z.ZodObject<{
 export declare const createProviderParamsSchema: z.ZodObject<{
     tenantId: z.ZodString;
     provider: z.ZodObject<{
+        createdById: z.ZodString;
         name: z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>;
         type: z.ZodEnum<{
             openai: "openai";
@@ -251,7 +308,7 @@ export declare const createProviderParamsSchema: z.ZodObject<{
             bedrock: "bedrock";
         }>;
         description: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
-        baseUrl: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null, string | null | undefined>>, z.ZodNullable<z.ZodString>>;
+        baseUrl: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
         isEnabled: z.ZodBoolean;
     }, z.core.$strip>;
 }, z.core.$strip>;
@@ -266,6 +323,7 @@ export declare const updateProviderParamsSchema: z.ZodObject<{
     tenantId: z.ZodString;
     providerId: z.ZodString;
     updates: z.ZodObject<{
+        modifiedById: z.ZodOptional<z.ZodString>;
         name: z.ZodOptional<z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>>;
         type: z.ZodOptional<z.ZodEnum<{
             openai: "openai";
@@ -273,7 +331,7 @@ export declare const updateProviderParamsSchema: z.ZodObject<{
             bedrock: "bedrock";
         }>>;
         description: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
-        baseUrl: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null, string | null | undefined>>, z.ZodNullable<z.ZodString>>;
+        baseUrl: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
         isEnabled: z.ZodOptional<z.ZodBoolean>;
     }, z.core.$strip>;
 }, z.core.$strip>;
@@ -313,13 +371,15 @@ export declare const getModelParamsSchema: z.ZodObject<{
 export declare const createModelParamsSchema: z.ZodObject<{
     tenantId: z.ZodString;
     model: z.ZodObject<{
-        id: z.ZodString;
+        createdById: z.ZodString;
+        modelId: z.ZodString;
         providerId: z.ZodString;
         name: z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>;
         description: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
         capabilities: z.ZodDefault<z.ZodArray<z.ZodEnum<{
-            streaming: "streaming";
+            document: "document";
             chat: "chat";
+            streaming: "streaming";
             vision: "vision";
             thinking: "thinking";
             "function-calling": "function-calling";
@@ -327,10 +387,25 @@ export declare const createModelParamsSchema: z.ZodObject<{
         }>>>;
         contextWindow: z.ZodInt;
         maxOutputTokens: z.ZodInt;
+        thinkingBudgetTokens: z.ZodOptional<z.ZodNullable<z.ZodInt>>;
         pricing: z.ZodOptional<z.ZodObject<{
             inputPerMillion: z.ZodNumber;
             outputPerMillion: z.ZodNumber;
         }, z.core.$strip>>;
+        budgetConfig: z.ZodOptional<z.ZodObject<{
+            budgetAmount: z.ZodNumber;
+            budgetPeriodType: z.ZodDefault<z.ZodEnum<{
+                custom: "custom";
+                monthly: "monthly";
+                weekly: "weekly";
+                daily: "daily";
+            }>>;
+            budgetPeriodDays: z.ZodOptional<z.ZodNumber>;
+            budgetStartDate: z.ZodOptional<z.ZodNumber>;
+            requestsPerMinute: z.ZodNumber;
+            tokenLimit: z.ZodOptional<z.ZodNumber>;
+            isEnabled: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+        }, z.core.$strip>>;
         isAvailable: z.ZodBoolean;
         isBasicAssistantDefault: z.ZodOptional<z.ZodBoolean>;
         parameterConstraints: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodObject<{
@@ -357,11 +432,14 @@ export declare const updateModelParamsSchema: z.ZodObject<{
     providerId: z.ZodString;
     modelId: z.ZodString;
     updates: z.ZodObject<{
+        modifiedById: z.ZodOptional<z.ZodString>;
+        modelId: z.ZodOptional<z.ZodString>;
         name: z.ZodOptional<z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>>;
         description: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
         capabilities: z.ZodOptional<z.ZodArray<z.ZodEnum<{
-            streaming: "streaming";
+            document: "document";
             chat: "chat";
+            streaming: "streaming";
             vision: "vision";
             thinking: "thinking";
             "function-calling": "function-calling";
@@ -369,10 +447,25 @@ export declare const updateModelParamsSchema: z.ZodObject<{
         }>>>;
         contextWindow: z.ZodOptional<z.ZodInt>;
         maxOutputTokens: z.ZodOptional<z.ZodInt>;
+        thinkingBudgetTokens: z.ZodOptional<z.ZodNullable<z.ZodInt>>;
         pricing: z.ZodOptional<z.ZodObject<{
             inputPerMillion: z.ZodNumber;
             outputPerMillion: z.ZodNumber;
         }, z.core.$strip>>;
+        budgetConfig: z.ZodNullable<z.ZodOptional<z.ZodObject<{
+            budgetAmount: z.ZodNumber;
+            budgetPeriodType: z.ZodDefault<z.ZodEnum<{
+                custom: "custom";
+                monthly: "monthly";
+                weekly: "weekly";
+                daily: "daily";
+            }>>;
+            budgetPeriodDays: z.ZodOptional<z.ZodNumber>;
+            budgetStartDate: z.ZodOptional<z.ZodNumber>;
+            requestsPerMinute: z.ZodNumber;
+            tokenLimit: z.ZodOptional<z.ZodNumber>;
+            isEnabled: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+        }, z.core.$strip>>>;
         isAvailable: z.ZodOptional<z.ZodBoolean>;
         isBasicAssistantDefault: z.ZodOptional<z.ZodBoolean>;
         parameterConstraints: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodObject<{
@@ -421,11 +514,13 @@ export type Model = z.infer<typeof modelSchema>;
 /** New provider data - represents a new provider to be created (without generated fields) */
 export type NewProvider = Omit<Provider, "id" | "createdAt" | "updatedAt">;
 /** New model data - represents a new model to be created (without generated fields) */
-export type NewModel = Omit<Model, "createdAt" | "updatedAt" | "isBasicAssistantDefault"> & {
+export type NewModel = Omit<Model, "id" | "createdAt" | "updatedAt" | "isBasicAssistantDefault"> & {
     isBasicAssistantDefault?: boolean;
 };
 /** Token pricing configuration (schema-derived) */
 export type PricingSchema = z.infer<typeof pricingSchema>;
+/** Budget and rate limit configuration for a model */
+export type BudgetConfig = z.infer<typeof budgetConfigSchema>;
 /** Single parameter constraint configuration */
 export type ParameterConstraint = z.infer<typeof parameterConstraintSchema>;
 /** Map of parameter names to their constraints */
@@ -444,6 +539,8 @@ export type GetModelParams = z.infer<typeof getModelParamsSchema>;
 export type CreateModelParams = z.infer<typeof createModelParamsSchema>;
 /** Parameters for updating an LLM model */
 export type UpdateModelParams = z.infer<typeof updateModelParamsSchema>;
+/** Model updates with nullable budgetConfig for removal support */
+export type ModelUpdates = UpdateModelParams["updates"];
 /** Parameters for deleting an LLM model */
 export type DeleteModelParams = z.infer<typeof deleteModelParamsSchema>;
 /** Parameters for LLM request routing (streamResponse/generateText) */
@@ -478,12 +575,13 @@ export declare const providerFormSchema: z.ZodObject<{
  * while the API allows flexibility for automated/programmatic use cases.
  */
 export declare const modelFormSchema: z.ZodObject<{
-    id: z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>;
+    modelId: z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>;
     name: z.ZodPipe<z.ZodPipe<z.ZodString, z.ZodTransform<string, string>>, z.ZodString>;
     description: z.ZodPipe<z.ZodPipe<z.ZodOptional<z.ZodNullable<z.ZodString>>, z.ZodTransform<string | null | undefined, string | null | undefined>>, z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     capabilities: z.ZodArray<z.ZodEnum<{
-        streaming: "streaming";
+        document: "document";
         chat: "chat";
+        streaming: "streaming";
         vision: "vision";
         thinking: "thinking";
         "function-calling": "function-calling";
@@ -491,6 +589,7 @@ export declare const modelFormSchema: z.ZodObject<{
     }>>;
     contextWindow: z.ZodCoercedNumber<unknown>;
     maxOutputTokens: z.ZodCoercedNumber<unknown>;
+    thinkingBudgetTokens: z.ZodPipe<z.ZodTransform<number | null, unknown>, z.ZodNullable<z.ZodNumber>>;
     pricing: z.ZodOptional<z.ZodObject<{
         inputPerMillion: z.ZodCoercedNumber<unknown>;
         outputPerMillion: z.ZodCoercedNumber<unknown>;

package/dist/llm-providers/llm-providers.schemas.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm-providers.schemas.d.ts","sourceRoot":"","sources":["../../lib/llm-providers/llm-providers.schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAsBxB;;;;GAIG;AACH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEhE;;;;GAIG;AACH,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAM1D;;;;;;;;;;;;GAYG;AACH,QAAA,MAAM,mBAAmB,aAK0B,CAAC;AAEpD;;;;;;;;;;;;GAYG;AACH,QAAA,MAAM,gBAAgB,aAKqB,CAAC;AAE5C;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB;;;;EAI8B,CAAC;AAE9D;;;;;GAKG;AACH,eAAO,MAAM,qBAAqB;;;;;;;EAI8B,CAAC;AAEjE;;;;;GAKG;AACH,eAAO,MAAM,oBAAoB;;;;;;EAE4B,CAAC;AAE9D;;;;;;;;;GASG;AACH,eAAO,MAAM,yBAAyB;;;;;;;;;iBAQ8B,CAAC;AAErE;;;;;;;;;;;;;;;;;GAiBG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;oBAYW,CAAC;AAMnD;;;;;GAKG;AACH,eAAO,MAAM,UAAU,gFAIrB,CAAC;AAEH;;;;;GAKG;AACH,eAAO,MAAM,iBAAiB,kLAG5B,CAAC;AAEH;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,aAAa,uJAuByB,CAAC;AAMpD;;;;;GAKG;AACH,eAAO,MAAM,aAAa;;;kBAMgB,CAAC;AAE3C;;;;;GAKG;AACH,eAAO,MAAM,cAAc;;;;;;;;;;;;;iBAWmB,CAAC;AAE/C;;;;;GAKG;AACH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsBmB,CAAC;AA0B5C;;;;;;GAMG;AACH,eAAO,MAAM,2BAA2B;;;iBAEvC,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;iBAYkB,CAAC;AAE1D;;;;;;GAMG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;iBAYe,CAAC;AAEvD;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B;;;iBAEtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,oBAAoB;;;;iBAEhC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA6BkB,CAAC;AAEvD;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA6Be,CAAC;AAEpD;;;;;;;GAOG;AACH,eAAO,MAAM,uBAAuB;;;;iBAEnC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,sBAAsB;;;;iBAElC,CAAC;AAOF,2FAA2F;AAC3F,MAAM,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAEtD,sEAAsE;AACtE,MAAM,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,WAAW,CAAC,CAAC;AAEhD,6FAA6F;AAC7F,MAAM,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,CAAC;AAE3E,uFAAuF;AACvF,MAAM,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,EAAE,WAAW,GAAG,WAAW,GAAG,yBAAyB,CAAC,GAAG;IAC1F,uBAAuB,CAAC,EAAE,OAAO,CAAC;CACnC,CAAC;AAEF,mDAAmD;AACnD,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAE1D,gDAAgD;AAChD,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,yBAAyB,CAAC,CAAC;AAE5E,kDAAkD;AAClD,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAG9E,iDAAiD;AACjD,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAEhF,iDAAiD;AACjD,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAE9E,8CAA8C;AAC9C,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAE9E,8CAA8C;AAC9C,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAG9E,8CAA8C;AAC9C,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAElE,8CAA8C;AAC9C,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,2CAA2C;AAC3C,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,2CAA2C;AAC3C,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,uEAAuE;AACvE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAiCtE;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB;;;;;;;;;;iBAe7B,CAAC;AAEH;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA4C1B,CAAC;AAEH,kDAAkD;AAClD,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAEpE,kDAAkD;AAClD,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAAC;AAE7D,6DAA6D;AAC7D,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,eAAe,CAAC,CAAC;AAM/D;;GAEG;AACH,eAAO,MAAM,wBAAwB;;;;;;;;;;iBAMnC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,wBAAwB;;;;;;;;;;iBAMnC,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAC1E,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC"}
+{"version":3,"file":"llm-providers.schemas.d.ts","sourceRoot":"","sources":["../../lib/llm-providers/llm-providers.schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAyBxB;;;;GAIG;AACH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEhE;;;;GAIG;AACH,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAM1D;;;;;;;;;;;;GAYG;AACH,QAAA,MAAM,mBAAmB,aAK0B,CAAC;AAsBpD;;;;;;;;;;;;GAYG;AACH,QAAA,MAAM,gBAAgB,aAKkD,CAAC;AAEzE;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB;;;;EAI8B,CAAC;AAE9D;;;;;GAKG;AACH,eAAO,MAAM,qBAAqB;;;;;;;;EAI8B,CAAC;AAEjE;;;;;GAKG;AACH,eAAO,MAAM,oBAAoB;;;;;;EAE4B,CAAC;AAE9D;;;;;;;;;GASG;AACH,eAAO,MAAM,yBAAyB;;;;;;;;;iBAQ8B,CAAC;AAErE;;;;;;;;;;;;;;;;;GAiBG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;oBAYW,CAAC;AAMnD;;;;;GAKG;AACH,eAAO,MAAM,UAAU,gFAIrB,CAAC;AAEH;;;;;GAKG;AACH,eAAO,MAAM,iBAAiB,kLAG5B,CAAC;AAEH;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,aAAa,kLA6ByB,CAAC;AAMpD;;;;;GAKG;AACH,eAAO,MAAM,aAAa;;;kBAMgB,CAAC;AAE3C;;;;;;;;;;;;;;;;;;;GAmBG;AACH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;kBAuDkC,CAAC;AAElE;;;;;GAKG;AACH,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;iBAQmB,CAAC;AAE/C;;;;;GAKG;AACH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA6BmB,CAAC;AA2B5C;;;;;;GAMG;AACH,eAAO,MAAM,2BAA2B;;;iBAEvC,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;iBAgBkB,CAAC;AAE1D;;;;;;GAMG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;iBAae,CAAC;AAEvD;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B;;;iBAEtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,oBAAoB;;;;iBAEhC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA4CkB,CAAC;AAEvD;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA0Ce,CAAC;AAEpD;;;;;;;GAOG;AACH,eAAO,MAAM,uBAAuB;;;;iBAEnC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,sBAAsB;;;;iBAElC,CAAC;AAOF,2FAA2F;AAC3F,MAAM,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAEtD,sEAAsE;AACtE,MAAM,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,WAAW,CAAC,CAAC;AAEhD,6FAA6F;AAC7F,MAAM,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,CAAC;AAE3E,uFAAuF;AACvF,MAAM,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,GAAG,yBAAyB,CAAC,GAAG;IACjG,uBAAuB,CAAC,EAAE,OAAO,CAAC;CACnC,CAAC;AAEF,mDAAmD;AACnD,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAE1D,sDAAsD;AACtD,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAE9D,gDAAgD;AAChD,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,yBAAyB,CAAC,CAAC;AAE5E,kDAAkD;AAClD,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAG9E,iDAAiD;AACjD,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAEhF,iDAAiD;AACjD,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAE9E,8CAA8C;AAC9C,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAE9E,8CAA8C;AAC9C,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAG9E,8CAA8C;AAC9C,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAElE,8CAA8C;AAC9C,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,2CAA2C;AAC3C,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,mEAAmE;AACnE,MAAM,MAAM,YAAY,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC;AAExD,2CAA2C;AAC3C,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAExE,uEAAuE;AACvE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAiCtE;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB;;;;;;;;;;iBAe7B,CAAC;AAEH;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAoEzB,CAAC;AAEJ,kDAAkD;AAClD,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAEpE,kDAAkD;AAClD,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAAC;AAE7D,6DAA6D;AAC7D,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,eAAe,CAAC,CAAC;AAM/D;;GAEG;AACH,eAAO,MAAM,wBAAwB;;;;;;;;;;iBAMnC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,wBAAwB;;;;;;;;;;iBAMnC,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAC1E,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC"}

package/dist/llm-providers/llm-providers.schemas.js

@@ -1,7 +1,8 @@
 import { z } from "zod";
-import { createOptionalStringSchema, createRequiredStringSchema, } from "../shared/shared.schemas";
+import { baseEntitySchema, createOptionalStringSchema, createRequiredStringSchema, } from "../shared/shared.schemas";
 import { baseTenantParamsSchema, createBaseModelParamsSchema, createBaseProviderParamsSchema, } from "../shared/services/services.schemas";
-import { LLM_ERROR_MESSAGES, LLM_LIMITS, MODEL_CAPABILITIES, MODEL_PARAMETERS, PROVIDER_TYPES, } from "./llm-providers.constants";
+import { LLM_ERROR_MESSAGES, LLM_LIMITS, MODEL_CAPABILITIES, MODEL_PARAMETERS, PROVIDER_TYPES, THINKING_BUDGET, } from "./llm-providers.constants";
+import { budgetPeriodTypeSchema } from "../model-budget/model-budget.schemas";
 // ============================================================================
 // Base Field Schemas
 // ============================================================================
@@ -24,6 +25,25 @@ const llmProviderIdSchema = z
     .min(1, LLM_ERROR_MESSAGES.PROVIDER_ID_REQUIRED)
     .uuid(LLM_ERROR_MESSAGES.PROVIDER_ID_INVALID_FORMAT)
     .describe("UUID identifier for the LLM provider");
+/**
+ * Model UUID schema - validates model record UUIDs.
+ *
+ * @description Validates model record identifiers (UUIDs) used for database lookups.
+ * These are auto-generated when creating a model and remain constant.
+ *
+ * @constraints
+ * - Required: Cannot be empty
+ * - Format: Must be a valid UUID
+ *
+ * @example
+ * llmModelUuidSchema.parse("550e8400-e29b-41d4-a716-446655440000"); // Valid
+ */
+const llmModelUuidSchema = z
+    .string()
+    .trim()
+    .min(1, LLM_ERROR_MESSAGES.MODEL_ID_REQUIRED)
+    .uuid("Model ID must be a valid UUID")
+    .describe("UUID identifier for the LLM model record");
 /**
  * Model ID schema - validates LLM model identifiers.
  *
@@ -42,7 +62,7 @@ const llmModelIdSchema = z
     .trim()
     .min(1, LLM_ERROR_MESSAGES.MODEL_ID_REQUIRED)
     .max(LLM_LIMITS.MAX_MODEL_NAME_LENGTH, LLM_ERROR_MESSAGES.MODEL_ID_TOO_LONG)
-    .describe("Identifier for the LLM model");
+    .describe("Identifier for the LLM model (e.g., gpt-4, claude-3-opus)");
 /**
  * Provider type schema - validates supported LLM provider types.
  *
@@ -158,13 +178,20 @@ export const descriptionSchema = createOptionalStringSchema({
 export const baseUrlSchema = z
     .string()
     .nullish()
-    .transform((val) => (val === null || val === void 0 ? void 0 : val.trim()) || null)
+    .transform((val) => {
+    // Preserve undefined (field not provided in partial updates)
+    if (val === undefined)
+        return undefined;
+    // Trim and convert empty to null (field explicitly cleared)
+    const trimmed = val === null || val === void 0 ? void 0 : val.trim();
+    return trimmed || null;
+})
     .pipe(z
     .string()
     .max(LLM_LIMITS.MAX_BASE_URL_LENGTH, LLM_ERROR_MESSAGES.BASE_URL_TOO_LONG)
     .refine((val) => {
-    // Allow null values (optional field)
-    if (val === null)
+    // Allow null/undefined values (optional field)
+    if (val === null || val === undefined)
         return true;
     try {
         new URL(val);
@@ -174,7 +201,7 @@ export const baseUrlSchema = z
         return false;
     }
 }, { message: LLM_ERROR_MESSAGES.BASE_URL_INVALID })
-    .nullable())
+    .nullish())
     .describe("Custom API base URL for the provider");
 // ============================================================================
 // Entity Schemas (for DynamoDB validation)
@@ -192,40 +219,114 @@ export const pricingSchema = z
 })
     .optional()
     .describe("Token pricing configuration");
+/**
+ * Budget configuration schema for per-model spend and rate limits.
+ *
+ * @description Defines budget and rate limiting configuration for a model.
+ * Budgets auto-reset when the period expires based on budgetStartDate.
+ *
+ * This schema is designed to be extensible for future enhancements:
+ * - Team-based limits: Can be extended to support team/group scoping
+ * - Token quotas: tokenLimit field can be added for raw token limits
+ * - User limits: Can create user-level budget configs with same structure
+ *
+ * @example
+ * {
+ *   budgetAmount: 100.00,       // $100 budget
+ *   budgetPeriodDays: 30,       // 30-day period
+ *   budgetStartDate: 1704067200000, // When period started (Unix ms)
+ *   requestsPerMinute: 60,      // 60 requests per minute limit
+ *   tokenLimit: 1000000         // Optional: 1M token limit per period
+ * }
+ */
+export const budgetConfigSchema = z
+    .object({
+    budgetAmount: z
+        .number()
+        .nonnegative()
+        .describe("Budget limit in USD for the period (0 = no spend allowed)"),
+    budgetPeriodType: budgetPeriodTypeSchema
+        .default("monthly")
+        .describe("Period type (monthly, weekly, daily, custom)"),
+    budgetPeriodDays: z
+        .number()
+        .int()
+        .positive()
+        .optional()
+        .describe("Custom period length in days (only used when budgetPeriodType is 'custom')"),
+    budgetStartDate: z
+        .number()
+        .int()
+        .positive()
+        .optional()
+        .describe("Unix timestamp (ms) when the budget was created (used for custom periods)"),
+    requestsPerMinute: z
+        .number()
+        .int()
+        .nonnegative()
+        .describe("Maximum requests per minute (0 = unlimited)"),
+    tokenLimit: z
+        .number()
+        .int()
+        .nonnegative()
+        .optional()
+        .describe("Optional token limit for the period (0 or undefined = unlimited)"),
+    isEnabled: z
+        .boolean()
+        .optional()
+        .default(true)
+        .describe("Whether budget limits are actively enforced (default: true)"),
+})
+    .refine((data) => {
+    if (data.budgetPeriodType === "custom") {
+        return (data.budgetPeriodDays !== undefined &&
+            Number.isInteger(data.budgetPeriodDays) &&
+            data.budgetPeriodDays > 0);
+    }
+    return true;
+}, {
+    message: "budgetPeriodDays must be a positive integer when budgetPeriodType is 'custom'",
+    path: ["budgetPeriodDays"],
+})
+    .optional()
+    .describe("Budget and rate limit configuration for this model");
 /**
  * Provider schema for validating DynamoDB data.
  *
  * @description Validates the structure of LLM provider configurations
- * retrieved from DynamoDB.
+ * retrieved from DynamoDB. Extends baseEntitySchema for id, createdAt, updatedAt.
  */
-export const providerSchema = z
-    .object({
-    id: z.string().min(1).describe("Provider UUID"),
+export const providerSchema = baseEntitySchema
+    .extend({
     name: z.string().min(1).describe("Display name"),
     type: providerTypeSchema.describe("Provider type"),
     description: z.string().nullish().describe("Optional description"),
     baseUrl: z.string().nullish().describe("Custom API base URL"),
     isEnabled: z.boolean().describe("Whether provider is enabled"),
-    createdAt: z.int().positive().describe("Creation timestamp (Unix ms)"),
-    updatedAt: z.int().positive().describe("Last update timestamp (Unix ms)"),
 })
     .describe("LLM provider stored in DynamoDB");
 /**
  * Model schema for validating DynamoDB data.
  *
  * @description Validates the structure of LLM model configurations
- * retrieved from DynamoDB.
+ * retrieved from DynamoDB. Extends baseEntitySchema for id, createdAt, updatedAt.
  */
-export const modelSchema = z
-    .object({
-    id: z.string().min(1).describe("Model identifier"),
+export const modelSchema = baseEntitySchema
+    .extend({
+    modelId: z.string().min(1).describe("Actual model identifier (e.g., gpt-4, claude-3-opus)"),
     providerId: z.string().min(1).describe("Parent provider UUID"),
     name: z.string().min(1).describe("Display name"),
     description: z.string().nullish().describe("Optional description"),
     capabilities: z.array(modelCapabilitySchema).default([]).describe("Model capabilities"),
     contextWindow: z.int().positive().describe("Max context window size (tokens)"),
     maxOutputTokens: z.int().positive().describe("Max output tokens per request"),
+    thinkingBudgetTokens: z
+        .int()
+        .min(THINKING_BUDGET.MIN_TOKENS, `Thinking budget must be at least ${THINKING_BUDGET.MIN_TOKENS} tokens`)
+        .nullish()
+        .describe("Token budget for thinking/reasoning (only for models with thinking capability)"),
     pricing: pricingSchema.describe("Token pricing"),
+    budgetConfig: budgetConfigSchema.describe("Budget and rate limit configuration"),
     isAvailable: z.boolean().describe("Whether model is available for use"),
     isBasicAssistantDefault: z
         .boolean()
@@ -233,8 +334,6 @@ export const modelSchema = z
         .default(false)
         .describe("Whether this model is the default for Basic Assistant tasks"),
     parameterConstraints: parameterConstraintsSchema.describe("Parameter support and exclusivity rules for this model"),
-    createdAt: z.int().positive().describe("Creation timestamp (Unix ms)"),
-    updatedAt: z.int().positive().describe("Last update timestamp (Unix ms)"),
 })
     .describe("LLM model stored in DynamoDB");
 // ============================================================================
@@ -250,10 +349,11 @@ const baseProviderParamsSchema = createBaseProviderParamsSchema(llmProviderIdSch
 /**
  * Base schema for model-scoped LLM operations.
  * Built using the shared createBaseModelParamsSchema factory.
+ * Uses UUID for model lookups (not the actual model identifier).
  *
  * @see createBaseModelParamsSchema
  */
-const baseModelParamsSchema = createBaseModelParamsSchema(llmProviderIdSchema, llmModelIdSchema);
+const baseModelParamsSchema = createBaseModelParamsSchema(llmProviderIdSchema, llmModelUuidSchema);
 // ============================================================================
 // Method Parameter Schemas
 // ============================================================================
@@ -277,6 +377,10 @@ export const createProviderParamsSchema = baseTenantParamsSchema
     .extend({
     provider: z
         .object({
+        createdById: z
+            .string()
+            .min(1, "createdById is required")
+            .describe("User ID who created the provider"),
         name: nameSchema.describe("Provider display name"),
         type: providerTypeSchema.describe("Provider type"),
         description: descriptionSchema.describe("Optional description"),
@@ -297,6 +401,7 @@ export const updateProviderParamsSchema = baseProviderParamsSchema
     .extend({
     updates: z
         .object({
+        modifiedById: z.string().min(1).optional().describe("User ID who modified the provider"),
         name: nameSchema.optional().describe("New display name"),
         type: providerTypeSchema.optional().describe("New provider type"),
         description: descriptionSchema.describe("New description"),
@@ -336,7 +441,11 @@ export const createModelParamsSchema = baseTenantParamsSchema
     .extend({
     model: z
         .object({
-        id: llmModelIdSchema.describe("Model identifier (e.g., gpt-4-turbo)"),
+        createdById: z
+            .string()
+            .min(1, "createdById is required")
+            .describe("User ID who created the model"),
+        modelId: llmModelIdSchema.describe("Model identifier (e.g., gpt-4-turbo)"),
         providerId: llmProviderIdSchema.describe("Parent provider UUID"),
         name: nameSchema.describe("Model display name"),
         description: descriptionSchema.describe("Optional description"),
@@ -349,7 +458,13 @@ export const createModelParamsSchema = baseTenantParamsSchema
             .int()
             .positive("maxOutputTokens must be a positive number")
             .describe("Max output tokens"),
+        thinkingBudgetTokens: z
+            .int()
+            .min(THINKING_BUDGET.MIN_TOKENS, `Thinking budget must be at least ${THINKING_BUDGET.MIN_TOKENS} tokens`)
+            .nullish()
+            .describe("Token budget for thinking/reasoning (only for models with thinking capability)"),
         pricing: pricingSchema.describe("Token pricing"),
+        budgetConfig: budgetConfigSchema.describe("Budget and rate limit configuration"),
         isAvailable: z.boolean().describe("Whether model is available"),
         isBasicAssistantDefault: z
             .boolean()
@@ -371,6 +486,8 @@ export const updateModelParamsSchema = baseModelParamsSchema
     .extend({
     updates: z
         .object({
+        modifiedById: z.string().min(1).optional().describe("User ID who modified the model"),
+        modelId: llmModelIdSchema.optional().describe("New model identifier (e.g., gpt-4-turbo)"),
         name: nameSchema.optional().describe("New display name"),
         description: descriptionSchema.describe("New description"),
         capabilities: z.array(modelCapabilitySchema).optional().describe("New capabilities"),
@@ -384,7 +501,15 @@ export const updateModelParamsSchema = baseModelParamsSchema
             .positive("maxOutputTokens must be a positive number")
             .optional()
             .describe("New max output tokens"),
+        thinkingBudgetTokens: z
+            .int()
+            .min(THINKING_BUDGET.MIN_TOKENS, `Thinking budget must be at least ${THINKING_BUDGET.MIN_TOKENS} tokens`)
+            .nullish()
+            .describe("Token budget for thinking/reasoning (null to remove)"),
         pricing: pricingSchema.describe("New pricing"),
+        budgetConfig: budgetConfigSchema
+            .nullable()
+            .describe("Budget and rate limit configuration (null to remove)"),
         isAvailable: z.boolean().optional().describe("Enable/disable model"),
         isBasicAssistantDefault: z
             .boolean()
@@ -471,8 +596,9 @@ export const providerFormSchema = z.object({
  * This is intentional - forms should guide users to select capabilities,
  * while the API allows flexibility for automated/programmatic use cases.
  */
-export const modelFormSchema = z.object({
-    id: z
+export const modelFormSchema = z
+    .object({
+    modelId: z
         .string()
         .transform((val) => val.trim())
         .pipe(z
@@ -504,6 +630,11 @@ export const modelFormSchema = z.object({
         error: "Max output tokens must be an integer",
     })
         .min(1, "Max output tokens must be at least 1"),
+    thinkingBudgetTokens: z.preprocess((val) => (val === "" || val === undefined || val === null ? null : Number(val)), z
+        .number()
+        .int({ error: "Thinking budget tokens must be an integer" })
+        .min(THINKING_BUDGET.MIN_TOKENS, `Thinking budget must be at least ${THINKING_BUDGET.MIN_TOKENS} tokens`)
+        .nullable()),
     pricing: z
         .object({
         inputPerMillion: z.coerce.number().min(0, "Price cannot be negative"),
@@ -513,6 +644,15 @@ export const modelFormSchema = z.object({
     parameterConstraints: parameterConstraintsFormSchema,
     isAvailable: z.boolean(),
     isBasicAssistantDefault: z.boolean().optional().default(false),
+})
+    .refine((data) => {
+    if (data.thinkingBudgetTokens !== null && data.thinkingBudgetTokens !== undefined) {
+        return data.thinkingBudgetTokens < data.maxOutputTokens;
+    }
+    return true;
+}, {
+    message: "Thinking budget must be less than max output tokens",
+    path: ["thinkingBudgetTokens"],
 });
 // ============================================================================
 // API Body Schemas