captify 0.1.18 → 1.1.0

package/dist/model-budget/api/model-budget.handlers.js

@@ -0,0 +1,272 @@
+/**
+ * Model Budget API handlers
+ *
+ * Handlers for managing per-model budget and rate limit configuration,
+ * and querying usage statistics.
+ */
+import { getLLMService } from "../../llm-providers/services";
+import { getUserAwsCredentials, handleApi, withAuth } from "../../shared/utilities/api";
+import { NotFoundError } from "../../shared/services/services.errors";
+import { getModelBudgetService } from "../services";
+import { batchUsageBodySchema, setBudgetConfigBodySchema, } from "../model-budget.schemas";
+import { ACTION, RESOURCE } from "../../permissions/permissions.constants";
+import { requirePermission } from "../../permissions/permissions.utilities";
+// ============================================================================
+// Budget Configuration Handlers
+// ============================================================================
+/**
+ * GET /api/llm-providers/:providerId/models/:modelId/budget
+ *
+ * Gets the budget configuration for a specific model.
+ *
+ * @param providerId - Provider ID from URL path
+ * @param modelId - Model ID from URL path
+ *
+ * @returns The model's budget configuration, or null if not configured
+ *
+ * @example
+ * GET /api/llm-providers/provider_123/models/gpt-4o/budget
+ * Response: {
+ *   "budgetAmount": 100.00,
+ *   "budgetPeriodDays": 30,
+ *   "budgetStartDate": 1704067200000,
+ *   "requestsPerMinute": 60
+ * }
+ */
+export const getBudgetConfig = withAuth(async (_req, { idToken, tenantId, userId, user }, { params }) => handleApi(async () => {
+    var _a;
+    const { providerId, modelId } = await params;
+    const userCredentials = await getUserAwsCredentials(idToken);
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.LLM_BUDGETS,
+        action: ACTION.READ,
+        groups: user.groups,
+        userCredentials,
+    });
+    const model = await getLLMService().getModel(tenantId, providerId, modelId, {
+        userCredentials,
+    });
+    if (!model) {
+        throw new NotFoundError(`Model ${modelId} not found`);
+    }
+    return (_a = model.budgetConfig) !== null && _a !== void 0 ? _a : null;
+}));
+/**
+ * PATCH /api/llm-providers/:providerId/models/:modelId/budget
+ *
+ * Sets or updates the budget configuration for a model.
+ *
+ * @param providerId - Provider ID from URL path
+ * @param modelId - Model ID from URL path
+ *
+ * @body budgetAmount - Budget limit in USD (required)
+ * @body budgetPeriodType - Period type (monthly, weekly, daily, custom)
+ * @body budgetPeriodDays - Period length in days (only for custom period type)
+ * @body requestsPerMinute - Rate limit (required, 0 = unlimited)
+ * @body tokenLimit - Optional token limit for the period
+ * @body isEnabled - Whether budget limits are actively enforced
+ *
+ * @returns The updated model object
+ *
+ * @example
+ * PATCH /api/llm-providers/provider_123/models/gpt-4o/budget
+ * Body: {
+ *   "budgetAmount": 100.00,
+ *   "budgetPeriodType": "monthly",
+ *   "requestsPerMinute": 60
+ * }
+ */
+export const setBudgetConfig = withAuth(async (request, { idToken, tenantId, userId, user }, { params }) => handleApi(async () => {
+    var _a, _b, _c;
+    const { providerId, modelId } = await params;
+    const body = setBudgetConfigBodySchema.parse(await request.json());
+    const userCredentials = await getUserAwsCredentials(idToken);
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.LLM_BUDGETS,
+        action: ACTION.UPDATE,
+        groups: user.groups,
+        userCredentials,
+    });
+    // Verify model exists
+    const existingModel = await getLLMService().getModel(tenantId, providerId, modelId, {
+        userCredentials,
+    });
+    if (!existingModel) {
+        throw new NotFoundError(`Model ${modelId} not found`);
+    }
+    // Build budget config with period type
+    // Only set budgetStartDate for custom periods, and only if creating new config
+    const budgetConfig = Object.assign(Object.assign({ budgetAmount: body.budgetAmount, budgetPeriodType: body.budgetPeriodType, requestsPerMinute: body.requestsPerMinute, isEnabled: (_a = body.isEnabled) !== null && _a !== void 0 ? _a : true }, (body.tokenLimit != null && { tokenLimit: body.tokenLimit })), (body.budgetPeriodType === "custom" && {
+        budgetPeriodDays: body.budgetPeriodDays,
+        budgetStartDate: (_c = (_b = existingModel.budgetConfig) === null || _b === void 0 ? void 0 : _b.budgetStartDate) !== null && _c !== void 0 ? _c : Date.now(),
+    }));
+    // Update model with new budget config
+    const updatedModel = await getLLMService().updateModel(tenantId, providerId, modelId, { budgetConfig }, { userCredentials });
+    // Clear the config cache to ensure immediate effect
+    getModelBudgetService().clearConfigCache({
+        tenantId,
+        providerId,
+        modelId,
+    });
+    return updatedModel;
+}));
+/**
+ * DELETE /api/llm-providers/:providerId/models/:modelId/budget
+ *
+ * Removes the budget configuration from a model.
+ *
+ * @param providerId - Provider ID from URL path
+ * @param modelId - Model ID from URL path
+ *
+ * @returns The updated model object (with budgetConfig removed)
+ *
+ * @example
+ * DELETE /api/llm-providers/provider_123/models/gpt-4o/budget
+ */
+export const deleteBudgetConfig = withAuth(async (_req, { idToken, tenantId, userId, user }, { params }) => handleApi(async () => {
+    const { providerId, modelId } = await params;
+    const userCredentials = await getUserAwsCredentials(idToken);
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.LLM_BUDGETS,
+        action: ACTION.DELETE,
+        groups: user.groups,
+        userCredentials,
+    });
+    // Verify model exists
+    const existingModel = await getLLMService().getModel(tenantId, providerId, modelId, {
+        userCredentials,
+    });
+    if (!existingModel) {
+        throw new NotFoundError(`Model ${modelId} not found`);
+    }
+    // Remove budget config from model using null marker
+    // This signals updateModel to use REMOVE operation instead of SET
+    const updatedModel = await getLLMService().updateModel(tenantId, providerId, modelId, { budgetConfig: null }, { userCredentials });
+    // Clear the config cache
+    getModelBudgetService().clearConfigCache({
+        tenantId,
+        providerId,
+        modelId,
+    });
+    return updatedModel;
+}));
+// ============================================================================
+// Usage Query Handlers
+// ============================================================================
+/**
+ * GET /api/llm-providers/:providerId/models/:modelId/usage
+ *
+ * Gets current usage statistics for a model.
+ *
+ * @param providerId - Provider ID from URL path
+ * @param modelId - Model ID from URL path
+ *
+ * @returns Usage statistics including budget spend and rate limit counts
+ *
+ * @example
+ * GET /api/llm-providers/provider_123/models/gpt-4o/usage
+ * Response: {
+ *   "modelId": "gpt-4o",
+ *   "providerId": "provider_123",
+ *   "tenantId": "tenant_456",
+ *   "budget": {
+ *     "limit": 100.00,
+ *     "spent": 45.23,
+ *     "remaining": 54.77,
+ *     "periodStart": "2025-01-01T00:00:00.000Z",
+ *     "periodEnd": "2025-01-31T00:00:00.000Z"
+ *   },
+ *   "rateLimit": {
+ *     "limit": 60,
+ *     "currentMinuteCount": 12,
+ *     "resetsIn": 45
+ *   },
+ *   "hasLimitsConfigured": true
+ * }
+ */
+export const getUsage = withAuth(async (_req, { idToken, tenantId, userId, user }, { params }) => handleApi(async () => {
+    const { providerId, modelId } = await params;
+    const userCredentials = await getUserAwsCredentials(idToken);
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.LLM_USAGE,
+        action: ACTION.READ,
+        groups: user.groups,
+        userCredentials,
+    });
+    // Verify model exists
+    const model = await getLLMService().getModel(tenantId, providerId, modelId, {
+        userCredentials,
+    });
+    if (!model) {
+        throw new NotFoundError(`Model ${modelId} not found`);
+    }
+    // Get usage from budget service
+    return getModelBudgetService().getUsage({
+        tenantId,
+        providerId,
+        modelId,
+        userCredentials,
+    });
+}));
+/**
+ * POST /api/llm-providers/usage/batch
+ *
+ * Gets usage statistics for multiple models in a single request.
+ * More efficient than calling individual usage endpoints.
+ *
+ * @body models - Array of { providerId, modelId } objects
+ *
+ * @returns Array of usage statistics for each model
+ *
+ * @example
+ * POST /api/llm-providers/usage/batch
+ * Body: {
+ *   "models": [
+ *     { "providerId": "provider_123", "modelId": "gpt-4o" },
+ *     { "providerId": "provider_123", "modelId": "gpt-4o-mini" }
+ *   ]
+ * }
+ */
+export const getBatchUsage = withAuth(async (request, { idToken, tenantId, userId, user }) => handleApi(async () => {
+    const body = await request.json();
+    // Validate and parse request body - strips extra properties
+    const { models } = batchUsageBodySchema.parse(body);
+    const userCredentials = await getUserAwsCredentials(idToken);
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.LLM_USAGE,
+        action: ACTION.LIST,
+        groups: user.groups,
+        userCredentials,
+    });
+    // Build params for batch query using validated values
+    const params = models.map(({ providerId, modelId }) => ({
+        tenantId,
+        providerId,
+        modelId,
+        userCredentials,
+    }));
+    // Get all usage in parallel
+    const usages = await getModelBudgetService().getBatchUsage(params);
+    return usages;
+}));
+// ============================================================================
+// Grouped Handlers Export
+// ============================================================================
+export const modelBudgetHandlers = {
+    getBudgetConfig,
+    setBudgetConfig,
+    deleteBudgetConfig,
+    getUsage,
+    getBatchUsage,
+};
+//# sourceMappingURL=model-budget.handlers.js.map

package/dist/model-budget/api/model-budget.handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-budget.handlers.js","sourceRoot":"","sources":["../../../lib/model-budget/api/model-budget.handlers.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACxF,OAAO,EAAE,aAAa,EAAE,MAAM,uCAAuC,CAAC;AACtE,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EACL,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,yCAAyC,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAE5E,+EAA+E;AAC/E,gCAAgC;AAChC,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,QAAQ,CACrC,KAAK,EACH,IAAI,EACJ,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EACnC,EAAE,MAAM,EAAgE,EACxE,EAAE,CACF,SAAS,CAAC,KAAK,IAAI,EAAE;;IACnB,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC;IAC7C,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,WAAW;QAC9B,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,eAAe;KAChB,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,MAAM,aAAa,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE;QAC1E,eAAe;KAChB,CAAC,CAAC;IAEH,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,aAAa,CAAC,SAAS,OAAO,YAAY,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,MAAA,KAAK,CAAC,YAAY,mCAAI,IAAI,CAAC;AACpC,CAAC,CAAC,CACL,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,QAAQ,CACrC,KAAK,EACH,OAAO,EACP,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EACnC,EAAE,MAAM,EAAgE,EACxE,EAAE,CACF,SAAS,CAAC,KAAK,IAAI,EAAE;;IACnB,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC;IAC7C,MAAM,IAAI,GAAG,yBAAyB,CAAC,KAAK,CAAC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,WAAW;QAC9B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,eAAe;KAChB,CAAC,CAAC;IAEH,sBAAsB;IACtB,MAAM,aAAa,GAAG,MAAM,aAAa,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE;QAClF,eAAe;KAChB,CAAC,CAAC;IAEH,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,aAAa,CAAC,SAAS,OAAO,YAAY,CAAC,CAAC;IACxD,CAAC;IAED,uCAAuC;IACvC,+EAA+E;IAC/E,MAAM,YAAY,iCAChB,YAAY,EAAE,IAAI,CAAC,YAAY,EAC/B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EACvC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,EACzC,SAAS,EAAE,MAAA,IAAI,CAAC,SAAS,mCAAI,IAAI,IAC9B,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,GAE5D,CAAC,IAAI,CAAC,gBAAgB,KAAK,QAAQ,IAAI;QACxC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;QACvC,eAAe,EAAE,MAAA,MAAA,aAAa,CAAC,YAAY,0CAAE,eAAe,mCAAI,IAAI,CAAC,GAAG,EAAE;KAC3E,CAAC,CACH,CAAC;IAEF,sCAAsC;IACtC,MAAM,YAAY,GAAG,MAAM,aAAa,EAAE,CAAC,WAAW,CACpD,QAAQ,EACR,UAAU,EACV,OAAO,EACP,EAAE,YAAY,EAAE,EAChB,EAAE,eAAe,EAAE,CACpB,CAAC;IAEF,oDAAoD;IACpD,qBAAqB,EAAE,CAAC,gBAAgB,CAAC;QACvC,QAAQ;QACR,UAAU;QACV,OAAO;KACR,CAAC,CAAC;IAEH,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC,CACL,CAAC;AAEF;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,QAAQ,CACxC,KAAK,EACH,IAAI,EACJ,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EACnC,EAAE,MAAM,EAAgE,EACxE,EAAE,CACF,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC;IAC7C,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,WAAW;QAC9B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,eAAe;KAChB,CAAC,CAAC;IAEH,sBAAsB;IACtB,MAAM,aAAa,GAAG,MAAM,aAAa,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE;QAClF,eAAe;KAChB,CAAC,CAAC;IAEH,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,aAAa,CAAC,SAAS,OAAO,YAAY,CAAC,CAAC;IACxD,CAAC;IAED,oDAAoD;IACpD,kEAAkE;IAClE,MAAM,YAAY,GAAG,MAAM,aAAa,EAAE,CAAC,WAAW,CACpD,QAAQ,EACR,UAAU,EACV,OAAO,EACP,EAAE,YAAY,EAAE,IAAI,EAAE,EACtB,EAAE,eAAe,EAAE,CACpB,CAAC;IAEF,yBAAyB;IACzB,qBAAqB,EAAE,CAAC,gBAAgB,CAAC;QACvC,QAAQ;QACR,UAAU;QACV,OAAO;KACR,CAAC,CAAC;IAEH,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC,CACL,CAAC;AAEF,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,QAAQ,CAC9B,KAAK,EACH,IAAI,EACJ,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EACnC,EAAE,MAAM,EAAgE,EACxE,EAAE,CACF,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC;IAC7C,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,SAAS;QAC5B,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,eAAe;KAChB,CAAC,CAAC;IAEH,sBAAsB;IACtB,MAAM,KAAK,GAAG,MAAM,aAAa,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE;QAC1E,eAAe;KAChB,CAAC,CAAC;IAEH,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,aAAa,CAAC,SAAS,OAAO,YAAY,CAAC,CAAC;IACxD,CAAC;IAED,gCAAgC;IAChC,OAAO,qBAAqB,EAAE,CAAC,QAAQ,CAAC;QACtC,QAAQ;QACR,UAAU;QACV,OAAO;QACP,eAAe;KAChB,CAAC,CAAC;AACL,CAAC,CAAC,CACL,CAAC;AAEF;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAC3F,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IAElC,4DAA4D;IAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAEpD,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,SAAS;QAC5B,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,eAAe;KAChB,CAAC,CAAC;IAEH,sDAAsD;IACtD,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACtD,QAAQ;QACR,UAAU;QACV,OAAO;QACP,eAAe;KAChB,CAAC,CAAC,CAAC;IAEJ,4BAA4B;IAC5B,MAAM,MAAM,GAAG,MAAM,qBAAqB,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAEnE,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC,CACH,CAAC;AAEF,+EAA+E;AAC/E,0BAA0B;AAC1B,+EAA+E;AAE/E,MAAM,CAAC,MAAM,mBAAmB,GAAG;IACjC,eAAe;IACf,eAAe;IACf,kBAAkB;IAClB,QAAQ;IACR,aAAa;CACd,CAAC"}

package/dist/model-budget/components/add-budget-dialog.d.ts

@@ -0,0 +1,7 @@
+import type { AddBudgetDialogProps } from "../model-budget.interfaces";
+/**
+ * Dialog for adding a new budget configuration.
+ * Combines model selection and configuration in a single streamlined flow.
+ */
+export declare function AddBudgetDialog({ open, onOpenChange, modelsWithoutBudgets, onSave, isSaving: _isSaving, }: AddBudgetDialogProps): import("react/jsx-runtime").JSX.Element;
+//# sourceMappingURL=add-budget-dialog.d.ts.map

package/dist/model-budget/components/add-budget-dialog.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"add-budget-dialog.d.ts","sourceRoot":"","sources":["../../../lib/model-budget/components/add-budget-dialog.tsx"],"names":[],"mappings":"AA+BA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AAYvF;;;GAGG;AACH,wBAAgB,eAAe,CAAC,EAC9B,IAAI,EACJ,YAAY,EACZ,oBAAoB,EACpB,MAAM,EACN,QAAQ,EAAE,SAAS,GACpB,EAAE,oBAAoB,2CAqKtB"}

package/dist/model-budget/components/add-budget-dialog.js

@@ -0,0 +1,77 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+import { zodResolver } from "@hookform/resolvers/zod";
+import { useEffect, useState } from "react";
+import { useForm } from "react-hook-form";
+import { Brain, Check, ChevronsUpDown } from "lucide-react";
+import { Dialog, DialogContent, DialogDescription, DialogHeader, DialogTitle, } from "../../shared/components/ui/dialog";
+import { Command, CommandEmpty, CommandGroup, CommandInput, CommandItem, CommandList, } from "../../shared/components/ui/command";
+import { Popover, PopoverContent, PopoverTrigger } from "../../shared/components/ui/popover";
+import { Button } from "../../shared/components/ui/button";
+import { Label } from "../../shared/components/ui/label";
+import { Separator } from "../../shared/components/ui/separator";
+import { cn } from "../../shared/utilities/cn";
+import { budgetConfigFormSchema, } from "../model-budget.schemas";
+import { BudgetConfigForm } from "./budget-config-form";
+const defaultValues = {
+    budgetAmount: 0,
+    budgetPeriodType: "monthly",
+    budgetPeriodDays: undefined,
+    requestsPerMinute: 0,
+    tokenLimit: null,
+    isEnabled: true,
+};
+/**
+ * Dialog for adding a new budget configuration.
+ * Combines model selection and configuration in a single streamlined flow.
+ */
+export function AddBudgetDialog({ open, onOpenChange, modelsWithoutBudgets, onSave, isSaving: _isSaving, }) {
+    const [comboboxOpen, setComboboxOpen] = useState(false);
+    const [selectedModelKey, setSelectedModelKey] = useState("");
+    const form = useForm({
+        resolver: zodResolver(budgetConfigFormSchema),
+        defaultValues,
+    });
+    // Reset form when dialog opens
+    useEffect(() => {
+        if (open) {
+            form.reset(defaultValues);
+        }
+    }, [open, form]);
+    // Group models by provider for better organization
+    const modelsByProvider = modelsWithoutBudgets.reduce((acc, model) => {
+        const providerName = model.providerName;
+        if (!acc[providerName]) {
+            acc[providerName] = [];
+        }
+        acc[providerName].push(model);
+        return acc;
+    }, {});
+    const selectedModel = modelsWithoutBudgets.find((m) => `${m.providerId}:${m.id}` === selectedModelKey);
+    const handleSelect = (modelKey) => {
+        setSelectedModelKey(modelKey);
+        setComboboxOpen(false);
+    };
+    const handleOpenChange = (newOpen) => {
+        if (!newOpen) {
+            setSelectedModelKey("");
+            setComboboxOpen(false);
+        }
+        onOpenChange(newOpen);
+    };
+    const handleSave = async (config) => {
+        if (selectedModel) {
+            await onSave(selectedModel, config);
+            setSelectedModelKey("");
+            onOpenChange(false);
+        }
+    };
+    const noModelsAvailable = modelsWithoutBudgets.length === 0;
+    return (_jsx(Dialog, { open: open, onOpenChange: handleOpenChange, children: _jsxs(DialogContent, { className: "flex max-h-[90vh] flex-col gap-0 overflow-hidden p-0 sm:max-w-[550px]", children: [_jsxs(DialogHeader, { className: "p-6 pb-4", children: [_jsx(DialogTitle, { children: "Add Budget Configuration" }), _jsx(DialogDescription, { children: noModelsAvailable
+                                ? "All models already have budget configurations."
+                                : "Select a model and configure its budget and rate limits." })] }), _jsxs("div", { className: "flex-1 overflow-y-auto p-6 pt-0", children: [_jsxs("div", { className: "space-y-2", children: [_jsx(Label, { children: "Model" }), _jsxs(Popover, { open: comboboxOpen, onOpenChange: setComboboxOpen, children: [_jsx(PopoverTrigger, { asChild: true, children: _jsxs(Button, { variant: "outline", role: "combobox", "aria-expanded": comboboxOpen, className: "w-full justify-between", disabled: noModelsAvailable, children: [selectedModel ? (_jsxs("span", { className: "flex items-center gap-2 truncate", children: [_jsx(Brain, { className: "h-4 w-4 shrink-0" }), _jsx("span", { className: "truncate", children: selectedModel.name }), _jsxs("span", { className: "text-muted-foreground truncate text-xs", children: ["(", selectedModel.providerName, ")"] })] })) : (_jsx("span", { className: "text-muted-foreground", children: noModelsAvailable ? "No models available" : "Select a model..." })), _jsx(ChevronsUpDown, { className: "ml-2 h-4 w-4 shrink-0 opacity-50" })] }) }), _jsx(PopoverContent, { className: "w-[--radix-popover-trigger-width] p-0", align: "start", children: _jsxs(Command, { children: [_jsx(CommandInput, { placeholder: "Search models..." }), _jsxs(CommandList, { children: [_jsx(CommandEmpty, { children: "No models found." }), Object.entries(modelsByProvider).map(([providerName, models]) => (_jsx(CommandGroup, { heading: providerName, children: models.map((model) => {
+                                                                    const modelKey = `${model.providerId}:${model.id}`;
+                                                                    return (_jsx(CommandItem, { value: `${model.name} ${providerName}`, onSelect: () => handleSelect(modelKey), children: _jsxs("div", { className: "flex w-full items-center gap-2", children: [_jsx(Brain, { className: "h-4 w-4 shrink-0" }), _jsx("span", { className: "flex-1 truncate", children: model.name }), _jsx(Check, { className: cn("h-4 w-4 shrink-0", selectedModelKey === modelKey ? "opacity-100" : "opacity-0") })] }) }, modelKey));
+                                                                }) }, providerName)))] })] }) })] })] }), selectedModel && (_jsxs(_Fragment, { children: [_jsx(Separator, { className: "my-6" }), _jsx(BudgetConfigForm, { form: form, onSubmit: handleSave, onCancel: () => handleOpenChange(false), showPresets: true })] })), !selectedModel && !noModelsAvailable && (_jsx("div", { className: "mt-6 flex justify-end", children: _jsx(Button, { variant: "outline", onClick: () => handleOpenChange(false), children: "Cancel" }) })), noModelsAvailable && (_jsx("div", { className: "mt-6 flex justify-end", children: _jsx(Button, { onClick: () => handleOpenChange(false), children: "Close" }) }))] })] }) }));
+}
+//# sourceMappingURL=add-budget-dialog.js.map

package/dist/model-budget/components/add-budget-dialog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"add-budget-dialog.js","sourceRoot":"","sources":["../../../lib/model-budget/components/add-budget-dialog.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC1C,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,EACL,MAAM,EACN,aAAa,EACb,iBAAiB,EACjB,YAAY,EACZ,WAAW,GACZ,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EACL,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,YAAY,EACZ,WAAW,EACX,WAAW,GACZ,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAC7F,OAAO,EAAE,MAAM,EAAE,MAAM,mCAAmC,CAAC;AAC3D,OAAO,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AACzD,OAAO,EAAE,SAAS,EAAE,MAAM,sCAAsC,CAAC;AACjE,OAAO,EAAE,EAAE,EAAE,MAAM,2BAA2B,CAAC;AAE/C,OAAO,EACL,sBAAsB,GAEvB,MAAM,yCAAyC,CAAC;AAEjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kDAAkD,CAAC;AAEpF,MAAM,aAAa,GAA2B;IAC5C,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,SAAS;IAC3B,gBAAgB,EAAE,SAAS;IAC3B,iBAAiB,EAAE,CAAC;IACpB,UAAU,EAAE,IAAI;IAChB,SAAS,EAAE,IAAI;CAChB,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,EAC9B,IAAI,EACJ,YAAY,EACZ,oBAAoB,EACpB,MAAM,EACN,QAAQ,EAAE,SAAS,GACE;IACrB,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACxD,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CAAS,EAAE,CAAC,CAAC;IAErE,MAAM,IAAI,GAAG,OAAO,CAAyB;QAC3C,QAAQ,EAAE,WAAW,CAAC,sBAAsB,CAAC;QAC7C,aAAa;KACd,CAAC,CAAC;IAEH,+BAA+B;IAC/B,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IAEjB,mDAAmD;IACnD,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,MAAM,CAClD,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QACb,MAAM,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;YACvB,GAAG,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;QACzB,CAAC;QACD,GAAG,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,OAAO,GAAG,CAAC;IACb,CAAC,EACD,EAAuC,CACxC,CAAC;IAEF,MAAM,aAAa,GAAG,oBAAoB,CAAC,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,EAAE,EAAE,KAAK,gBAAgB,CACtD,CAAC;IAEF,MAAM,YAAY,GAAG,CAAC,QAAgB,EAAE,EAAE;QACxC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9B,eAAe,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC,CAAC;IAEF,MAAM,gBAAgB,GAAG,CAAC,OAAgB,EAAE,EAAE;QAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,mBAAmB,CAAC,EAAE,CAAC,CAAC;YACxB,eAAe,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QACD,YAAY,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,KAAK,EAAE,MAA8B,EAAE,EAAE;QAC1D,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YACpC,mBAAmB,CAAC,EAAE,CAAC,CAAC;YACxB,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,MAAM,KAAK,CAAC,CAAC;IAE5D,OAAO,CACL,KAAC,MAAM,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,gBAAgB,YAChD,MAAC,aAAa,IAAC,SAAS,EAAC,uEAAuE,aAC9F,MAAC,YAAY,IAAC,SAAS,EAAC,UAAU,aAChC,KAAC,WAAW,2CAAuC,EACnD,KAAC,iBAAiB,cACf,iBAAiB;gCAChB,CAAC,CAAC,gDAAgD;gCAClD,CAAC,CAAC,0DAA0D,GAC5C,IACP,EAEf,eAAK,SAAS,EAAC,iCAAiC,aAE9C,eAAK,SAAS,EAAC,WAAW,aACxB,KAAC,KAAK,wBAAc,EACpB,MAAC,OAAO,IAAC,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,aACxD,KAAC,cAAc,IAAC,OAAO,kBACrB,MAAC,MAAM,IACL,OAAO,EAAC,SAAS,EACjB,IAAI,EAAC,UAAU,mBACA,YAAY,EAC3B,SAAS,EAAC,wBAAwB,EAClC,QAAQ,EAAE,iBAAiB,aAE1B,aAAa,CAAC,CAAC,CAAC,CACf,gBAAM,SAAS,EAAC,kCAAkC,aAChD,KAAC,KAAK,IAAC,SAAS,EAAC,kBAAkB,GAAG,EACtC,eAAM,SAAS,EAAC,UAAU,YAAE,aAAa,CAAC,IAAI,GAAQ,EACtD,gBAAM,SAAS,EAAC,wCAAwC,kBACpD,aAAa,CAAC,YAAY,SACvB,IACF,CACR,CAAC,CAAC,CAAC,CACF,eAAM,SAAS,EAAC,uBAAuB,YACpC,iBAAiB,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,mBAAmB,GAC3D,CACR,EACD,KAAC,cAAc,IAAC,SAAS,EAAC,kCAAkC,GAAG,IACxD,GACM,EACjB,KAAC,cAAc,IAAC,SAAS,EAAC,uCAAuC,EAAC,KAAK,EAAC,OAAO,YAC7E,MAAC,OAAO,eACN,KAAC,YAAY,IAAC,WAAW,EAAC,kBAAkB,GAAG,EAC/C,MAAC,WAAW,eACV,KAAC,YAAY,mCAAgC,EAC5C,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAChE,KAAC,YAAY,IAAoB,OAAO,EAAE,YAAY,YACnD,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oEACpB,MAAM,QAAQ,GAAG,GAAG,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;oEACnD,OAAO,CACL,KAAC,WAAW,IAEV,KAAK,EAAE,GAAG,KAAK,CAAC,IAAI,IAAI,YAAY,EAAE,EACtC,QAAQ,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,YAEtC,eAAK,SAAS,EAAC,gCAAgC,aAC7C,KAAC,KAAK,IAAC,SAAS,EAAC,kBAAkB,GAAG,EACtC,eAAM,SAAS,EAAC,iBAAiB,YAAE,KAAK,CAAC,IAAI,GAAQ,EACrD,KAAC,KAAK,IACJ,SAAS,EAAE,EAAE,CACX,kBAAkB,EAClB,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAC5D,GACD,IACE,IAbD,QAAQ,CAcD,CACf,CAAC;gEACJ,CAAC,CAAC,IArBe,YAAY,CAsBhB,CAChB,CAAC,IACU,IACN,GACK,IACT,IACN,EAGL,aAAa,IAAI,CAChB,8BACE,KAAC,SAAS,IAAC,SAAS,EAAC,MAAM,GAAG,EAC9B,KAAC,gBAAgB,IACf,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,UAAU,EACpB,QAAQ,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,EACvC,WAAW,SACX,IACD,CACJ,EAGA,CAAC,aAAa,IAAI,CAAC,iBAAiB,IAAI,CACvC,cAAK,SAAS,EAAC,uBAAuB,YACpC,KAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,OAAO,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,uBAEvD,GACL,CACP,EAGA,iBAAiB,IAAI,CACpB,cAAK,SAAS,EAAC,uBAAuB,YACpC,KAAC,MAAM,IAAC,OAAO,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,sBAAgB,GAC1D,CACP,IACG,IACQ,GACT,CACV,CAAC;AACJ,CAAC"}

package/dist/model-budget/components/budget-config-dialog.d.ts

@@ -0,0 +1,6 @@
+import type { BudgetConfigDialogProps } from "../model-budget.interfaces";
+/**
+ * Dialog for configuring model budget and rate limits.
+ */
+export declare function BudgetConfigDialog({ open, onOpenChange, modelName, modelId, budgetConfig, onSave, onDelete, isDeleting, readOnly, }: BudgetConfigDialogProps): import("react/jsx-runtime").JSX.Element;
+//# sourceMappingURL=budget-config-dialog.d.ts.map

package/dist/model-budget/components/budget-config-dialog.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-config-dialog.d.ts","sourceRoot":"","sources":["../../../lib/model-budget/components/budget-config-dialog.tsx"],"names":[],"mappings":"AA4BA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,4CAA4C,CAAC;AAY1F;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,EACjC,IAAI,EACJ,YAAY,EACZ,SAAS,EACT,OAAO,EACP,YAAY,EACZ,MAAM,EACN,QAAQ,EACR,UAAU,EACV,QAAQ,GACT,EAAE,uBAAuB,2CA8GzB"}

package/dist/model-budget/components/budget-config-dialog.js

@@ -0,0 +1,58 @@
+"use client";
+import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
+import { zodResolver } from "@hookform/resolvers/zod";
+import { Trash2 } from "lucide-react";
+import { useEffect, useState } from "react";
+import { useForm } from "react-hook-form";
+import { Dialog, DialogContent, DialogDescription, DialogHeader, DialogTitle, } from "../../shared/components/ui/dialog";
+import { Button } from "../../shared/components/ui/button";
+import { AlertDialog, AlertDialogAction, AlertDialogCancel, AlertDialogContent, AlertDialogDescription, AlertDialogFooter, AlertDialogHeader, AlertDialogTitle, } from "../../shared/components/ui/alert-dialog";
+import { budgetConfigFormSchema, } from "../model-budget.schemas";
+import { BudgetConfigForm } from "./budget-config-form";
+const defaultValues = {
+    budgetAmount: 0,
+    budgetPeriodType: "monthly",
+    budgetPeriodDays: undefined,
+    requestsPerMinute: 0,
+    tokenLimit: null,
+    isEnabled: true,
+};
+/**
+ * Dialog for configuring model budget and rate limits.
+ */
+export function BudgetConfigDialog({ open, onOpenChange, modelName, modelId, budgetConfig, onSave, onDelete, isDeleting, readOnly, }) {
+    const [showDeleteConfirm, setShowDeleteConfirm] = useState(false);
+    const hasConfig = !!budgetConfig;
+    const form = useForm({
+        resolver: zodResolver(budgetConfigFormSchema),
+        defaultValues,
+    });
+    // Reset form when dialog opens or budgetConfig changes
+    useEffect(() => {
+        var _a, _b, _c;
+        if (budgetConfig) {
+            form.reset({
+                budgetAmount: budgetConfig.budgetAmount,
+                budgetPeriodType: (_a = budgetConfig.budgetPeriodType) !== null && _a !== void 0 ? _a : "monthly",
+                budgetPeriodDays: budgetConfig.budgetPeriodDays,
+                requestsPerMinute: budgetConfig.requestsPerMinute,
+                tokenLimit: (_b = budgetConfig.tokenLimit) !== null && _b !== void 0 ? _b : null,
+                isEnabled: (_c = budgetConfig.isEnabled) !== null && _c !== void 0 ? _c : true,
+            });
+        }
+        else {
+            form.reset(defaultValues);
+        }
+    }, [budgetConfig, open, form]);
+    const handleSave = async (values) => {
+        await onSave(values);
+        onOpenChange(false);
+    };
+    const handleDelete = async () => {
+        await onDelete();
+        setShowDeleteConfirm(false);
+        onOpenChange(false);
+    };
+    return (_jsxs(_Fragment, { children: [_jsx(Dialog, { open: open, onOpenChange: onOpenChange, children: _jsxs(DialogContent, { className: "flex max-h-[90vh] flex-col gap-0 overflow-hidden p-0 sm:max-w-[550px]", children: [_jsxs(DialogHeader, { className: "p-6 pb-4", children: [_jsx(DialogTitle, { children: readOnly ? "View Budget & Rate Limits" : "Budget & Rate Limits" }), _jsx(DialogDescription, { children: readOnly ? (_jsxs(_Fragment, { children: ["Viewing spending limits and rate limits for ", _jsx("strong", { children: modelName }), " (", modelId, ")"] })) : (_jsxs(_Fragment, { children: ["Configure spending limits and rate limits for ", _jsx("strong", { children: modelName }), " (", modelId, ")"] })) })] }), _jsxs("div", { className: "flex-1 overflow-y-auto p-6 pt-2", children: [hasConfig && !readOnly && (_jsx("div", { className: "mb-4 flex justify-end", children: _jsxs(Button, { variant: "outline", size: "sm", onClick: () => setShowDeleteConfirm(true), disabled: isDeleting, className: "text-destructive hover:text-destructive", children: [_jsx(Trash2, { className: "mr-1.5 h-3.5 w-3.5" }), "Remove Limits"] }) })), _jsx(BudgetConfigForm, { form: form, onSubmit: handleSave, onCancel: () => onOpenChange(false), showPresets: !hasConfig && !readOnly, isEditing: hasConfig, isDeleting: isDeleting, readOnly: readOnly })] })] }) }), _jsx(AlertDialog, { open: showDeleteConfirm, onOpenChange: setShowDeleteConfirm, children: _jsxs(AlertDialogContent, { children: [_jsxs(AlertDialogHeader, { children: [_jsx(AlertDialogTitle, { children: "Remove Budget Limits?" }), _jsxs(AlertDialogDescription, { children: ["This will remove all budget and rate limit configuration for", " ", _jsx("strong", { children: modelName }), ". The model will have no spending or rate restrictions."] })] }), _jsxs(AlertDialogFooter, { children: [_jsx(AlertDialogCancel, { disabled: isDeleting, children: "Cancel" }), _jsx(AlertDialogAction, { onClick: handleDelete, disabled: isDeleting, className: "bg-destructive text-destructive-foreground hover:bg-destructive/90", children: isDeleting ? "Removing..." : "Remove Limits" })] })] }) })] }));
+}
+//# sourceMappingURL=budget-config-dialog.js.map

package/dist/model-budget/components/budget-config-dialog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-config-dialog.js","sourceRoot":"","sources":["../../../lib/model-budget/components/budget-config-dialog.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC1C,OAAO,EACL,MAAM,EACN,aAAa,EACb,iBAAiB,EACjB,YAAY,EACZ,WAAW,GACZ,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,mCAAmC,CAAC;AAC3D,OAAO,EACL,WAAW,EACX,iBAAiB,EACjB,iBAAiB,EACjB,kBAAkB,EAClB,sBAAsB,EACtB,iBAAiB,EACjB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EACL,sBAAsB,GAEvB,MAAM,yCAAyC,CAAC;AAEjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kDAAkD,CAAC;AAEpF,MAAM,aAAa,GAA2B;IAC5C,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,SAAS;IAC3B,gBAAgB,EAAE,SAAS;IAC3B,iBAAiB,EAAE,CAAC;IACpB,UAAU,EAAE,IAAI;IAChB,SAAS,EAAE,IAAI;CAChB,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,EACjC,IAAI,EACJ,YAAY,EACZ,SAAS,EACT,OAAO,EACP,YAAY,EACZ,MAAM,EACN,QAAQ,EACR,UAAU,EACV,QAAQ,GACgB;IACxB,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClE,MAAM,SAAS,GAAG,CAAC,CAAC,YAAY,CAAC;IAEjC,MAAM,IAAI,GAAG,OAAO,CAAyB;QAC3C,QAAQ,EAAE,WAAW,CAAC,sBAAsB,CAAC;QAC7C,aAAa;KACd,CAAC,CAAC;IAEH,uDAAuD;IACvD,SAAS,CAAC,GAAG,EAAE;;QACb,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,KAAK,CAAC;gBACT,YAAY,EAAE,YAAY,CAAC,YAAY;gBACvC,gBAAgB,EAAE,MAAA,YAAY,CAAC,gBAAgB,mCAAI,SAAS;gBAC5D,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;gBAC/C,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;gBACjD,UAAU,EAAE,MAAA,YAAY,CAAC,UAAU,mCAAI,IAAI;gBAC3C,SAAS,EAAE,MAAA,YAAY,CAAC,SAAS,mCAAI,IAAI;aAC1C,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IAE/B,MAAM,UAAU,GAAG,KAAK,EAAE,MAA8B,EAAE,EAAE;QAC1D,MAAM,MAAM,CAAC,MAAM,CAAC,CAAC;QACrB,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;QAC9B,MAAM,QAAQ,EAAE,CAAC;QACjB,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAC5B,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC,CAAC;IAEF,OAAO,CACL,8BACE,KAAC,MAAM,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,YAC5C,MAAC,aAAa,IAAC,SAAS,EAAC,uEAAuE,aAC9F,MAAC,YAAY,IAAC,SAAS,EAAC,UAAU,aAChC,KAAC,WAAW,cACT,QAAQ,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,sBAAsB,GACpD,EACd,KAAC,iBAAiB,cACf,QAAQ,CAAC,CAAC,CAAC,CACV,8EAC8C,2BAAS,SAAS,GAAU,QACvE,OAAO,SACP,CACJ,CAAC,CAAC,CAAC,CACF,gFACgD,2BAAS,SAAS,GAAU,QACzE,OAAO,SACP,CACJ,GACiB,IACP,EAEf,eAAK,SAAS,EAAC,iCAAiC,aAC7C,SAAS,IAAI,CAAC,QAAQ,IAAI,CACzB,cAAK,SAAS,EAAC,uBAAuB,YACpC,MAAC,MAAM,IACL,OAAO,EAAC,SAAS,EACjB,IAAI,EAAC,IAAI,EACT,OAAO,EAAE,GAAG,EAAE,CAAC,oBAAoB,CAAC,IAAI,CAAC,EACzC,QAAQ,EAAE,UAAU,EACpB,SAAS,EAAC,yCAAyC,aAEnD,KAAC,MAAM,IAAC,SAAS,EAAC,oBAAoB,GAAG,qBAElC,GACL,CACP,EACD,KAAC,gBAAgB,IACf,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,UAAU,EACpB,QAAQ,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,EACnC,WAAW,EAAE,CAAC,SAAS,IAAI,CAAC,QAAQ,EACpC,SAAS,EAAE,SAAS,EACpB,UAAU,EAAE,UAAU,EACtB,QAAQ,EAAE,QAAQ,GAClB,IACE,IACQ,GACT,EAET,KAAC,WAAW,IAAC,IAAI,EAAE,iBAAiB,EAAE,YAAY,EAAE,oBAAoB,YACtE,MAAC,kBAAkB,eACjB,MAAC,iBAAiB,eAChB,KAAC,gBAAgB,wCAAyC,EAC1D,MAAC,sBAAsB,+EACwC,GAAG,EAChE,2BAAS,SAAS,GAAU,+DACL,IACP,EACpB,MAAC,iBAAiB,eAChB,KAAC,iBAAiB,IAAC,QAAQ,EAAE,UAAU,uBAA4B,EACnE,KAAC,iBAAiB,IAChB,OAAO,EAAE,YAAY,EACrB,QAAQ,EAAE,UAAU,EACpB,SAAS,EAAC,oEAAoE,YAE7E,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,eAAe,GAC3B,IACF,IACD,GACT,IACb,CACJ,CAAC;AACJ,CAAC"}

package/dist/model-budget/components/budget-config-form.d.ts

@@ -0,0 +1,22 @@
+import type { UseFormReturn } from "react-hook-form";
+import { type BudgetConfigFormValues } from "../model-budget.schemas";
+interface BudgetConfigFormProps {
+    form: UseFormReturn<BudgetConfigFormValues>;
+    onSubmit: (values: BudgetConfigFormValues) => void | Promise<void>;
+    onCancel?: () => void;
+    showPresets?: boolean;
+    /** Whether editing an existing config (affects submit button label) */
+    isEditing?: boolean;
+    /** Whether a delete operation is in progress (disables submit) */
+    isDeleting?: boolean;
+    /** Whether the form is in read-only mode (disables editing and hides submit) */
+    readOnly?: boolean;
+}
+/**
+ * Form for configuring model budget and rate limits.
+ *
+ * Uses FormBuilder for consistent form rendering with the rest of the app.
+ */
+export declare function BudgetConfigForm({ form, onSubmit, onCancel, showPresets, isEditing, isDeleting, readOnly, }: BudgetConfigFormProps): import("react/jsx-runtime").JSX.Element;
+export {};
+//# sourceMappingURL=budget-config-form.d.ts.map

package/dist/model-budget/components/budget-config-form.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-config-form.d.ts","sourceRoot":"","sources":["../../../lib/model-budget/components/budget-config-form.tsx"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAIrD,OAAO,EAEL,KAAK,sBAAsB,EAE5B,MAAM,yCAAyC,CAAC;AA6CjD,UAAU,qBAAqB;IAC7B,IAAI,EAAE,aAAa,CAAC,sBAAsB,CAAC,CAAC;IAC5C,QAAQ,EAAE,CAAC,MAAM,EAAE,sBAAsB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,QAAQ,CAAC,EAAE,MAAM,IAAI,CAAC;IACtB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,uEAAuE;IACvE,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,kEAAkE;IAClE,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,gFAAgF;IAChF,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAAC,EAC/B,IAAI,EACJ,QAAQ,EACR,QAAQ,EACR,WAAmB,EACnB,SAAiB,EACjB,UAAkB,EAClB,QAAgB,GACjB,EAAE,qBAAqB,2CA+IvB"}

package/dist/model-budget/components/budget-config-form.js

@@ -0,0 +1,144 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useEffect, useMemo } from "react";
+import { Sparkles } from "lucide-react";
+import { Button } from "../../shared/components/ui/button";
+import { FormBuilder } from "../../shared/components/form-builder";
+import { budgetConfigFormSchema, } from "../model-budget.schemas";
+const periodTypeOptions = [
+    { value: "monthly", label: "Monthly (resets on 1st)" },
+    { value: "weekly", label: "Weekly (resets on Monday)" },
+    { value: "daily", label: "Daily (resets at midnight)" },
+    { value: "custom", label: "Custom (specify days)" },
+];
+const presets = [
+    {
+        name: "Light",
+        description: "$50/mo, 30 req/min",
+        values: {
+            budgetAmount: 50,
+            budgetPeriodType: "monthly",
+            requestsPerMinute: 30,
+            tokenLimit: null,
+            isEnabled: true,
+        },
+    },
+    {
+        name: "Standard",
+        description: "$200/mo, 60 req/min",
+        values: {
+            budgetAmount: 200,
+            budgetPeriodType: "monthly",
+            requestsPerMinute: 60,
+            tokenLimit: null,
+            isEnabled: true,
+        },
+    },
+    {
+        name: "Heavy",
+        description: "$1000/mo, 120 req/min",
+        values: {
+            budgetAmount: 1000,
+            budgetPeriodType: "monthly",
+            requestsPerMinute: 120,
+            tokenLimit: null,
+            isEnabled: true,
+        },
+    },
+];
+/**
+ * Form for configuring model budget and rate limits.
+ *
+ * Uses FormBuilder for consistent form rendering with the rest of the app.
+ */
+export function BudgetConfigForm({ form, onSubmit, onCancel, showPresets = false, isEditing = false, isDeleting = false, readOnly = false, }) {
+    // Watch the period type to conditionally show custom days field
+    const periodType = form.watch("budgetPeriodType");
+    // Clear budgetPeriodDays when switching away from custom period type
+    useEffect(() => {
+        if (periodType !== "custom") {
+            form.setValue("budgetPeriodDays", undefined);
+        }
+    }, [periodType, form]);
+    const applyPreset = (preset) => {
+        form.reset(preset.values);
+    };
+    const sections = useMemo(() => [
+        {
+            fields: [
+                {
+                    name: "isEnabled",
+                    label: "Enable Budget Limits",
+                    type: "switch",
+                    description: "When disabled, limits are not enforced but configuration is preserved",
+                    disabled: readOnly,
+                },
+            ],
+        },
+        {
+            title: "Budget Limits",
+            description: "Set spending limits for this model",
+            fields: [
+                {
+                    name: "budgetAmount",
+                    label: "Budget Amount (USD)",
+                    type: "number",
+                    step: 0.000001,
+                    placeholder: "e.g. 100.00",
+                    description: "Maximum spend allowed per period",
+                    disabled: readOnly,
+                },
+                {
+                    name: "budgetPeriodType",
+                    label: "Reset Period",
+                    type: "select",
+                    options: periodTypeOptions,
+                    description: "When the budget resets",
+                    disabled: readOnly,
+                },
+                ...(periodType === "custom"
+                    ? [
+                        {
+                            name: "budgetPeriodDays",
+                            label: "Period Length (days)",
+                            type: "number",
+                            placeholder: "e.g. 14",
+                            description: "Budget resets after this many days",
+                            span: 2,
+                            disabled: readOnly,
+                        },
+                    ]
+                    : []),
+            ],
+            columns: 2,
+        },
+        {
+            title: "Rate Limits",
+            description: "Control request frequency",
+            fields: [
+                {
+                    name: "requestsPerMinute",
+                    label: "Requests per Minute",
+                    type: "number",
+                    placeholder: "e.g. 60",
+                    description: "Maximum requests allowed per minute (0 = unlimited)",
+                    disabled: readOnly,
+                },
+                {
+                    name: "tokenLimit",
+                    label: "Token Limit (optional)",
+                    type: "number",
+                    placeholder: "Leave empty for unlimited",
+                    description: "Maximum tokens per period (leave empty for unlimited)",
+                    disabled: readOnly,
+                },
+            ],
+            columns: 2,
+        },
+    ], [periodType, readOnly]);
+    const handleSubmit = async (values) => {
+        await onSubmit(values);
+    };
+    return (_jsxs("div", { className: "space-y-6", children: [showPresets && (_jsxs("div", { className: "space-y-3", children: [_jsxs("div", { className: "flex items-center gap-2 text-sm font-medium", children: [_jsx(Sparkles, { className: "h-4 w-4" }), "Quick Presets"] }), _jsx("div", { className: "grid grid-cols-3 gap-2", children: presets.map((preset) => (_jsxs(Button, { type: "button", variant: "outline", size: "sm", className: "h-auto flex-col py-2", onClick: () => applyPreset(preset), children: [_jsx("span", { className: "font-medium", children: preset.name }), _jsx("span", { className: "text-muted-foreground text-xs", children: preset.description })] }, preset.name))) })] })), _jsx(FormBuilder, { form: form, schema: budgetConfigFormSchema, sections: sections, onSubmit: handleSubmit, onCancel: onCancel, isEditing: isEditing, submitLabel: isEditing ? "Update Configuration" : "Set Configuration", submitDisabled: isDeleting, hideButtons: readOnly }), readOnly && onCancel && (_jsx("div", { className: "mt-6 flex justify-end pb-4", children: _jsx(Button, { type: "button", variant: "outline", onClick: onCancel, children: "Close" }) }))] }));
+}
+//# sourceMappingURL=budget-config-form.js.map