captify 0.1.17 → 0.1.19

package/dist/model-budget/hooks/use-model-budget.js

@@ -0,0 +1,93 @@
+"use client";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+import { getBudgetConfig, setBudgetConfig, deleteBudgetConfig, getModelUsage, modelBudgetKeys, } from "../api/model-budget.client";
+import { invariant } from "../../shared/utilities/invariant";
+/**
+ * Hook for managing model budget configuration and usage.
+ *
+ * Provides queries for fetching budget config and usage stats,
+ * and mutations for setting/deleting budget configuration.
+ *
+ * @param providerId - The provider ID (required)
+ * @param modelId - The model ID (required)
+ */
+export function useModelBudget(providerId, modelId) {
+    const queryClient = useQueryClient();
+    // Query for budget configuration
+    const configQuery = useQuery({
+        queryKey: modelBudgetKeys.config(providerId !== null && providerId !== void 0 ? providerId : "", modelId !== null && modelId !== void 0 ? modelId : ""),
+        queryFn: () => {
+            invariant(providerId, "Provider ID is required");
+            invariant(modelId, "Model ID is required");
+            return getBudgetConfig(providerId, modelId);
+        },
+        enabled: !!providerId && !!modelId,
+    });
+    // Query for current usage statistics
+    const usageQuery = useQuery({
+        queryKey: modelBudgetKeys.usage(providerId !== null && providerId !== void 0 ? providerId : "", modelId !== null && modelId !== void 0 ? modelId : ""),
+        queryFn: () => {
+            invariant(providerId, "Provider ID is required");
+            invariant(modelId, "Model ID is required");
+            return getModelUsage(providerId, modelId);
+        },
+        enabled: !!providerId && !!modelId,
+    });
+    // Mutation for setting budget configuration
+    const setConfigMutation = useMutation({
+        mutationFn: ({ providerId, modelId, config, }) => setBudgetConfig(providerId, modelId, config),
+        onSuccess: (_, variables) => {
+            queryClient.invalidateQueries({
+                queryKey: modelBudgetKeys.config(variables.providerId, variables.modelId),
+            });
+            queryClient.invalidateQueries({
+                queryKey: modelBudgetKeys.usage(variables.providerId, variables.modelId),
+            });
+            // Also invalidate the model and provider caches since budgetConfig is stored on the model
+            queryClient.invalidateQueries({ queryKey: ["llm-models", variables.providerId] });
+            queryClient.invalidateQueries({
+                queryKey: ["llm-model", variables.providerId, variables.modelId],
+            });
+            queryClient.invalidateQueries({ queryKey: ["llm-providers"] });
+        },
+    });
+    // Mutation for deleting budget configuration
+    const deleteConfigMutation = useMutation({
+        mutationFn: ({ providerId, modelId }) => deleteBudgetConfig(providerId, modelId),
+        onSuccess: (_, variables) => {
+            // Explicitly clear the config cache to prevent stale data
+            queryClient.setQueryData(modelBudgetKeys.config(variables.providerId, variables.modelId), null);
+            queryClient.invalidateQueries({
+                queryKey: modelBudgetKeys.config(variables.providerId, variables.modelId),
+            });
+            queryClient.invalidateQueries({
+                queryKey: modelBudgetKeys.usage(variables.providerId, variables.modelId),
+            });
+            // Also invalidate the model and provider caches
+            queryClient.invalidateQueries({ queryKey: ["llm-models", variables.providerId] });
+            queryClient.invalidateQueries({
+                queryKey: ["llm-model", variables.providerId, variables.modelId],
+            });
+            queryClient.invalidateQueries({ queryKey: ["llm-providers"] });
+        },
+    });
+    return {
+        // Budget configuration
+        budgetConfig: configQuery.data,
+        isLoadingConfig: configQuery.isLoading,
+        configError: configQuery.error,
+        // Usage statistics
+        usage: usageQuery.data,
+        isLoadingUsage: usageQuery.isLoading,
+        usageError: usageQuery.error,
+        // Mutations
+        setConfig: setConfigMutation.mutateAsync,
+        deleteConfig: deleteConfigMutation.mutateAsync,
+        isSettingConfig: setConfigMutation.isPending,
+        isDeletingConfig: deleteConfigMutation.isPending,
+        // Refetch functions
+        refetchConfig: configQuery.refetch,
+        refetchUsage: usageQuery.refetch,
+    };
+}
+//# sourceMappingURL=use-model-budget.js.map

package/dist/model-budget/hooks/use-model-budget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-model-budget.js","sourceRoot":"","sources":["../../../lib/model-budget/hooks/use-model-budget.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC9E,OAAO,EACL,eAAe,EACf,eAAe,EACf,kBAAkB,EAClB,aAAa,EACb,eAAe,GAChB,MAAM,4CAA4C,CAAC;AAEpD,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAE7D;;;;;;;;GAQG;AACH,MAAM,UAAU,cAAc,CAAC,UAAyB,EAAE,OAAsB;IAC9E,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,iCAAiC;IACjC,MAAM,WAAW,GAAG,QAAQ,CAAC;QAC3B,QAAQ,EAAE,eAAe,CAAC,MAAM,CAAC,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,EAAE,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE,CAAC;QACjE,OAAO,EAAE,GAAG,EAAE;YACZ,SAAS,CAAC,UAAU,EAAE,yBAAyB,CAAC,CAAC;YACjD,SAAS,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;YAC3C,OAAO,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,EAAE,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,OAAO;KACnC,CAAC,CAAC;IAEH,qCAAqC;IACrC,MAAM,UAAU,GAAG,QAAQ,CAAC;QAC1B,QAAQ,EAAE,eAAe,CAAC,KAAK,CAAC,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,EAAE,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE,CAAC;QAChE,OAAO,EAAE,GAAG,EAAE;YACZ,SAAS,CAAC,UAAU,EAAE,yBAAyB,CAAC,CAAC;YACjD,SAAS,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;YAC3C,OAAO,aAAa,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,EAAE,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,OAAO;KACnC,CAAC,CAAC;IAEH,4CAA4C;IAC5C,MAAM,iBAAiB,GAAG,WAAW,CAAC;QACpC,UAAU,EAAE,CAAC,EACX,UAAU,EACV,OAAO,EACP,MAAM,GAKP,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,EAAE,MAAM,CAAC;QAClD,SAAS,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE;YAC1B,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,QAAQ,EAAE,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC;aAC1E,CAAC,CAAC;YACH,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,QAAQ,EAAE,eAAe,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC;aACzE,CAAC,CAAC;YACH,0FAA0F;YAC1F,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAClF,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,QAAQ,EAAE,CAAC,WAAW,EAAE,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC;aACjE,CAAC,CAAC;YACH,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC;KACF,CAAC,CAAC;IAEH,6CAA6C;IAC7C,MAAM,oBAAoB,GAAG,WAAW,CAAC;QACvC,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,OAAO,EAA2C,EAAE,EAAE,CAC/E,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC;QACzC,SAAS,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE;YAC1B,0DAA0D;YAC1D,WAAW,CAAC,YAAY,CACtB,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC,EAC/D,IAAI,CACL,CAAC;YACF,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,QAAQ,EAAE,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC;aAC1E,CAAC,CAAC;YACH,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,QAAQ,EAAE,eAAe,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC;aACzE,CAAC,CAAC;YACH,gDAAgD;YAChD,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAClF,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,QAAQ,EAAE,CAAC,WAAW,EAAE,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC;aACjE,CAAC,CAAC;YACH,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC;KACF,CAAC,CAAC;IAEH,OAAO;QACL,uBAAuB;QACvB,YAAY,EAAE,WAAW,CAAC,IAAI;QAC9B,eAAe,EAAE,WAAW,CAAC,SAAS;QACtC,WAAW,EAAE,WAAW,CAAC,KAAK;QAE9B,mBAAmB;QACnB,KAAK,EAAE,UAAU,CAAC,IAAI;QACtB,cAAc,EAAE,UAAU,CAAC,SAAS;QACpC,UAAU,EAAE,UAAU,CAAC,KAAK;QAE5B,YAAY;QACZ,SAAS,EAAE,iBAAiB,CAAC,WAAW;QACxC,YAAY,EAAE,oBAAoB,CAAC,WAAW;QAC9C,eAAe,EAAE,iBAAiB,CAAC,SAAS;QAC5C,gBAAgB,EAAE,oBAAoB,CAAC,SAAS;QAEhD,oBAAoB;QACpB,aAAa,EAAE,WAAW,CAAC,OAAO;QAClC,YAAY,EAAE,UAAU,CAAC,OAAO;KACjC,CAAC;AACJ,CAAC"}

package/dist/model-budget/index.d.ts

@@ -0,0 +1,42 @@
+/**
+ * Model Budget module
+ *
+ * Provides per-model budget and rate limit enforcement for LLM usage.
+ *
+ * Features:
+ * - Budget limits: Set spend caps per model with auto-reset periods
+ * - Rate limits: Control requests per minute per model
+ * - Token limits: Optional token quotas per period
+ * - Multi-tenant: All limits are scoped by tenant
+ *
+ * @example
+ * ```typescript
+ * import { getModelBudgetService, BudgetExceededError } from "@pitamber/captify-core/model-budget";
+ *
+ * const budgetService = getModelBudgetService();
+ *
+ * // Check limits before making a request
+ * const result = await budgetService.checkLimits({
+ *   tenantId: "tenant-123",
+ *   providerId: "provider-uuid",
+ *   modelId: "gpt-4o",
+ * });
+ *
+ * if (!result.allowed) {
+ *   if (result.reason === "budget_exceeded") {
+ *     throw new BudgetExceededError(result.budget);
+ *   }
+ * }
+ * ```
+ */
+export { BUDGET_DEFAULTS, BUDGET_ERROR_MESSAGES, BUDGET_ERROR_CODES, getCurrentMinuteTimestamp, getCurrentPeriodStart, getCurrentPeriodEnd, } from "./model-budget.constants";
+export type { ModelWithBudget, BudgetCheckResult, RateLimitCheckResult, LimitCheckResult, ModelBudgetParams, RecordSpendParams, CheckLimitsParams, RecordRequestParams, GetUsageParams, BudgetUsage, TokenUsageInfo, RateLimitInfo, ModelUsageResponse, BudgetExceededDetails, RateLimitExceededDetails, TokenLimitExceededDetails, CachedModelConfig, BudgetModelCardProps, UsageModelCardProps, BudgetUsageDisplayProps, BudgetConfigDialogProps, AddBudgetDialogProps, DuplicateBudgetDialogProps, BudgetSummaryStatsProps, UsageSummaryStatsProps, BudgetEmptyStateProps, BudgetFiltersProps, BudgetStatusFilter, ViolationsPanelProps, Violation, ViolationType, UsageTrackingParams, UsageData, UsageRecordResult, UsageTrackingServiceDeps, } from "./model-budget.interfaces";
+export { BudgetExceededError, RateLimitExceededError, TokenLimitExceededError, isBudgetExceededError, isRateLimitExceededError, isTokenLimitExceededError, isLimitExceededError, } from "./model-budget.errors";
+export { setBudgetConfigBodySchema, budgetUsageSchema, tokenUsageSchema, rateLimitInfoSchema, modelUsageResponseSchema, budgetExceededResponseSchema, rateLimitExceededResponseSchema, tokenLimitExceededResponseSchema, budgetConfigFormSchema, } from "./model-budget.schemas";
+export type { SetBudgetConfigBody, ModelUsageResponseSchema, BudgetConfigFormValues, } from "./model-budget.schemas";
+export { ModelBudgetService, getModelBudgetService, UsageTrackingService, getUsageTrackingService, } from "./services";
+export { getBudgetConfig, setBudgetConfig, deleteBudgetConfig, getUsage, modelBudgetHandlers, } from "./api";
+export { fetchBudgetConfig, saveBudgetConfig, removeBudgetConfig, getModelUsage, modelBudgetKeys, } from "./api";
+export { useModelBudget, useAllModelBudgets } from "./hooks";
+export { BudgetConfigForm, BudgetUsageDisplay, BudgetConfigDialog, BudgetModelCard, BudgetModelCardSkeleton, AddBudgetDialog, BudgetSummaryStats, BudgetEmptyState, BudgetFilters, UsageSummaryStats, UsageModelCard, UsageModelCardSkeleton, ViolationsPanel, DuplicateBudgetDialog, } from "./components";
+//# sourceMappingURL=index.d.ts.map

package/dist/model-budget/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../lib/model-budget/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AAGH,OAAO,EACL,eAAe,EACf,qBAAqB,EACrB,kBAAkB,EAClB,yBAAyB,EACzB,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAGlC,YAAY,EAEV,eAAe,EAEf,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAEhB,iBAAiB,EACjB,iBAAiB,EACjB,iBAAiB,EACjB,mBAAmB,EACnB,cAAc,EAEd,WAAW,EACX,cAAc,EACd,aAAa,EACb,kBAAkB,EAElB,qBAAqB,EACrB,wBAAwB,EACxB,yBAAyB,EAEzB,iBAAiB,EAEjB,oBAAoB,EACpB,mBAAmB,EACnB,uBAAuB,EACvB,uBAAuB,EACvB,oBAAoB,EACpB,0BAA0B,EAC1B,uBAAuB,EACvB,sBAAsB,EACtB,qBAAqB,EACrB,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,SAAS,EACT,aAAa,EAEb,mBAAmB,EACnB,SAAS,EACT,iBAAiB,EACjB,wBAAwB,GACzB,MAAM,2BAA2B,CAAC;AAGnC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,uBAAuB,EACvB,qBAAqB,EACrB,wBAAwB,EACxB,yBAAyB,EACzB,oBAAoB,GACrB,MAAM,uBAAuB,CAAC;AAG/B,OAAO,EACL,yBAAyB,EACzB,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,wBAAwB,EACxB,4BAA4B,EAC5B,+BAA+B,EAC/B,gCAAgC,EAChC,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAChC,YAAY,EACV,mBAAmB,EACnB,wBAAwB,EACxB,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,GACxB,MAAM,YAAY,CAAC;AAGpB,OAAO,EACL,eAAe,EACf,eAAe,EACf,kBAAkB,EAClB,QAAQ,EACR,mBAAmB,GACpB,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EAClB,aAAa,EACb,eAAe,GAChB,MAAM,OAAO,CAAC;AAGf,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAG7D,OAAO,EACL,gBAAgB,EAChB,kBAAkB,EAClB,kBAAkB,EAClB,eAAe,EACf,uBAAuB,EACvB,eAAe,EACf,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,EACb,iBAAiB,EACjB,cAAc,EACd,sBAAsB,EACtB,eAAe,EACf,qBAAqB,GACtB,MAAM,cAAc,CAAC"}

package/dist/model-budget/index.js

@@ -0,0 +1,48 @@
+/**
+ * Model Budget module
+ *
+ * Provides per-model budget and rate limit enforcement for LLM usage.
+ *
+ * Features:
+ * - Budget limits: Set spend caps per model with auto-reset periods
+ * - Rate limits: Control requests per minute per model
+ * - Token limits: Optional token quotas per period
+ * - Multi-tenant: All limits are scoped by tenant
+ *
+ * @example
+ * ```typescript
+ * import { getModelBudgetService, BudgetExceededError } from "@pitamber/captify-core/model-budget";
+ *
+ * const budgetService = getModelBudgetService();
+ *
+ * // Check limits before making a request
+ * const result = await budgetService.checkLimits({
+ *   tenantId: "tenant-123",
+ *   providerId: "provider-uuid",
+ *   modelId: "gpt-4o",
+ * });
+ *
+ * if (!result.allowed) {
+ *   if (result.reason === "budget_exceeded") {
+ *     throw new BudgetExceededError(result.budget);
+ *   }
+ * }
+ * ```
+ */
+// Constants
+export { BUDGET_DEFAULTS, BUDGET_ERROR_MESSAGES, BUDGET_ERROR_CODES, getCurrentMinuteTimestamp, getCurrentPeriodStart, getCurrentPeriodEnd, } from "./model-budget.constants";
+// Errors
+export { BudgetExceededError, RateLimitExceededError, TokenLimitExceededError, isBudgetExceededError, isRateLimitExceededError, isTokenLimitExceededError, isLimitExceededError, } from "./model-budget.errors";
+// Schemas
+export { setBudgetConfigBodySchema, budgetUsageSchema, tokenUsageSchema, rateLimitInfoSchema, modelUsageResponseSchema, budgetExceededResponseSchema, rateLimitExceededResponseSchema, tokenLimitExceededResponseSchema, budgetConfigFormSchema, } from "./model-budget.schemas";
+// Services
+export { ModelBudgetService, getModelBudgetService, UsageTrackingService, getUsageTrackingService, } from "./services";
+// API Handlers
+export { getBudgetConfig, setBudgetConfig, deleteBudgetConfig, getUsage, modelBudgetHandlers, } from "./api";
+// Client API Functions & Query Keys
+export { fetchBudgetConfig, saveBudgetConfig, removeBudgetConfig, getModelUsage, modelBudgetKeys, } from "./api";
+// Hooks (ModelWithBudget type is exported from interfaces section above)
+export { useModelBudget, useAllModelBudgets } from "./hooks";
+// Components
+export { BudgetConfigForm, BudgetUsageDisplay, BudgetConfigDialog, BudgetModelCard, BudgetModelCardSkeleton, AddBudgetDialog, BudgetSummaryStats, BudgetEmptyState, BudgetFilters, UsageSummaryStats, UsageModelCard, UsageModelCardSkeleton, ViolationsPanel, DuplicateBudgetDialog, } from "./components";
+//# sourceMappingURL=index.js.map

package/dist/model-budget/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../lib/model-budget/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AAEH,YAAY;AACZ,OAAO,EACL,eAAe,EACf,qBAAqB,EACrB,kBAAkB,EAClB,yBAAyB,EACzB,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAiDlC,SAAS;AACT,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,uBAAuB,EACvB,qBAAqB,EACrB,wBAAwB,EACxB,yBAAyB,EACzB,oBAAoB,GACrB,MAAM,uBAAuB,CAAC;AAE/B,UAAU;AACV,OAAO,EACL,yBAAyB,EACzB,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,wBAAwB,EACxB,4BAA4B,EAC5B,+BAA+B,EAC/B,gCAAgC,EAChC,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAOhC,WAAW;AACX,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,GACxB,MAAM,YAAY,CAAC;AAEpB,eAAe;AACf,OAAO,EACL,eAAe,EACf,eAAe,EACf,kBAAkB,EAClB,QAAQ,EACR,mBAAmB,GACpB,MAAM,OAAO,CAAC;AAEf,oCAAoC;AACpC,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EAClB,aAAa,EACb,eAAe,GAChB,MAAM,OAAO,CAAC;AAEf,yEAAyE;AACzE,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAE7D,aAAa;AACb,OAAO,EACL,gBAAgB,EAChB,kBAAkB,EAClB,kBAAkB,EAClB,eAAe,EACf,uBAAuB,EACvB,eAAe,EACf,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,EACb,iBAAiB,EACjB,cAAc,EACd,sBAAsB,EACtB,eAAe,EACf,qBAAqB,GACtB,MAAM,cAAc,CAAC"}

package/dist/model-budget/model-budget.constants.d.ts

@@ -0,0 +1,48 @@
+/**
+ * Model Budget module constants
+ *
+ * Constants for budget and rate limit tracking including
+ * error messages, and configuration defaults.
+ */
+/**
+ * Get the current minute timestamp (floored to the minute).
+ * Used for rate limit window tracking.
+ */
+export declare function getCurrentMinuteTimestamp(): number;
+export type BudgetPeriodType = "monthly" | "weekly" | "daily" | "custom";
+/**
+ * Calculate the current budget period start timestamp based on period type.
+ *
+ * @param periodType - The type of period (monthly, weekly, daily, custom)
+ * @param budgetStartDate - Original budget start date (only used for custom periods)
+ * @param budgetPeriodDays - Period length in days (only used for custom periods)
+ * @returns Current period start timestamp
+ */
+export declare function getCurrentPeriodStart(periodType?: BudgetPeriodType, budgetStartDate?: number, budgetPeriodDays?: number): number;
+/**
+ * Calculate the current budget period end timestamp based on period type.
+ *
+ * @param periodType - The type of period (monthly, weekly, daily, custom)
+ * @param budgetStartDate - Original budget start date (only used for custom periods)
+ * @param budgetPeriodDays - Period length in days (only used for custom periods)
+ * @returns Current period end timestamp
+ */
+export declare function getCurrentPeriodEnd(periodType?: BudgetPeriodType, budgetStartDate?: number, budgetPeriodDays?: number): number;
+export declare const BUDGET_DEFAULTS: {
+    /** Cache TTL for model config (60 seconds to meet 1-minute update requirement) */
+    readonly CONFIG_CACHE_TTL_MS: 60000;
+    /** TTL for rate limit entries in DynamoDB (2 minutes) */
+    readonly RATE_LIMIT_TTL_SECONDS: 120;
+};
+export declare const BUDGET_ERROR_MESSAGES: {
+    readonly BUDGET_EXCEEDED: "Budget limit exceeded for this model";
+    readonly RATE_LIMIT_EXCEEDED: "Rate limit exceeded for this model";
+    readonly MODEL_NOT_FOUND: "Model not found";
+    readonly INVALID_BUDGET_CONFIG: "Invalid budget configuration";
+};
+export declare const BUDGET_ERROR_CODES: {
+    readonly BUDGET_EXCEEDED: "budget_exceeded";
+    readonly RATE_LIMIT_EXCEEDED: "rate_limit_exceeded";
+    readonly TOKEN_LIMIT_EXCEEDED: "token_limit_exceeded";
+};
+//# sourceMappingURL=model-budget.constants.d.ts.map

package/dist/model-budget/model-budget.constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-budget.constants.d.ts","sourceRoot":"","sources":["../../lib/model-budget/model-budget.constants.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH;;;GAGG;AACH,wBAAgB,yBAAyB,IAAI,MAAM,CAElD;AAED,MAAM,MAAM,gBAAgB,GAAG,SAAS,GAAG,QAAQ,GAAG,OAAO,GAAG,QAAQ,CAAC;AAEzE;;;;;;;GAOG;AACH,wBAAgB,qBAAqB,CACnC,UAAU,GAAE,gBAA4B,EACxC,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,MAAM,GACxB,MAAM,CAkCR;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CACjC,UAAU,GAAE,gBAA4B,EACxC,eAAe,CAAC,EAAE,MAAM,EACxB,gBAAgB,CAAC,EAAE,MAAM,GACxB,MAAM,CA2BR;AAMD,eAAO,MAAM,eAAe;IAC1B,kFAAkF;;IAElF,yDAAyD;;CAEjD,CAAC;AAMX,eAAO,MAAM,qBAAqB;;;;;CAKxB,CAAC;AAMX,eAAO,MAAM,kBAAkB;;;;CAIrB,CAAC"}

package/dist/model-budget/model-budget.constants.js

@@ -0,0 +1,117 @@
+/**
+ * Model Budget module constants
+ *
+ * Constants for budget and rate limit tracking including
+ * error messages, and configuration defaults.
+ */
+/**
+ * Get the current minute timestamp (floored to the minute).
+ * Used for rate limit window tracking.
+ */
+export function getCurrentMinuteTimestamp() {
+    return Math.floor(Date.now() / 60000) * 60000;
+}
+/**
+ * Calculate the current budget period start timestamp based on period type.
+ *
+ * @param periodType - The type of period (monthly, weekly, daily, custom)
+ * @param budgetStartDate - Original budget start date (only used for custom periods)
+ * @param budgetPeriodDays - Period length in days (only used for custom periods)
+ * @returns Current period start timestamp
+ */
+export function getCurrentPeriodStart(periodType = "monthly", budgetStartDate, budgetPeriodDays) {
+    const now = new Date();
+    switch (periodType) {
+        case "monthly": {
+            // Start of current month at midnight
+            return new Date(now.getFullYear(), now.getMonth(), 1, 0, 0, 0, 0).getTime();
+        }
+        case "weekly": {
+            // Start of current week (Monday at midnight)
+            const dayOfWeek = now.getDay();
+            const daysFromMonday = dayOfWeek === 0 ? 6 : dayOfWeek - 1; // Sunday = 6 days from Monday
+            const monday = new Date(now);
+            monday.setDate(now.getDate() - daysFromMonday);
+            monday.setHours(0, 0, 0, 0);
+            return monday.getTime();
+        }
+        case "daily": {
+            // Start of current day at midnight
+            return new Date(now.getFullYear(), now.getMonth(), now.getDate(), 0, 0, 0, 0).getTime();
+        }
+        case "custom": {
+            // Custom period based on budgetStartDate and budgetPeriodDays
+            if (!budgetStartDate || !budgetPeriodDays) {
+                // Fallback to monthly if custom config is missing
+                return new Date(now.getFullYear(), now.getMonth(), 1, 0, 0, 0, 0).getTime();
+            }
+            const periodMs = budgetPeriodDays * 24 * 60 * 60 * 1000;
+            const periodsElapsed = Math.floor((now.getTime() - budgetStartDate) / periodMs);
+            return budgetStartDate + periodsElapsed * periodMs;
+        }
+        default:
+            return new Date(now.getFullYear(), now.getMonth(), 1, 0, 0, 0, 0).getTime();
+    }
+}
+/**
+ * Calculate the current budget period end timestamp based on period type.
+ *
+ * @param periodType - The type of period (monthly, weekly, daily, custom)
+ * @param budgetStartDate - Original budget start date (only used for custom periods)
+ * @param budgetPeriodDays - Period length in days (only used for custom periods)
+ * @returns Current period end timestamp
+ */
+export function getCurrentPeriodEnd(periodType = "monthly", budgetStartDate, budgetPeriodDays) {
+    const now = new Date();
+    switch (periodType) {
+        case "monthly": {
+            // Start of next month
+            return new Date(now.getFullYear(), now.getMonth() + 1, 1, 0, 0, 0, 0).getTime();
+        }
+        case "weekly": {
+            // Start of next week (next Monday)
+            const periodStart = getCurrentPeriodStart("weekly");
+            return periodStart + 7 * 24 * 60 * 60 * 1000;
+        }
+        case "daily": {
+            // Start of next day
+            return new Date(now.getFullYear(), now.getMonth(), now.getDate() + 1, 0, 0, 0, 0).getTime();
+        }
+        case "custom": {
+            if (!budgetStartDate || !budgetPeriodDays) {
+                return new Date(now.getFullYear(), now.getMonth() + 1, 1, 0, 0, 0, 0).getTime();
+            }
+            const periodStart = getCurrentPeriodStart("custom", budgetStartDate, budgetPeriodDays);
+            return periodStart + budgetPeriodDays * 24 * 60 * 60 * 1000;
+        }
+        default:
+            return new Date(now.getFullYear(), now.getMonth() + 1, 1, 0, 0, 0, 0).getTime();
+    }
+}
+// ============================================================================
+// Configuration Defaults
+// ============================================================================
+export const BUDGET_DEFAULTS = {
+    /** Cache TTL for model config (60 seconds to meet 1-minute update requirement) */
+    CONFIG_CACHE_TTL_MS: 60000,
+    /** TTL for rate limit entries in DynamoDB (2 minutes) */
+    RATE_LIMIT_TTL_SECONDS: 120,
+};
+// ============================================================================
+// Error Messages
+// ============================================================================
+export const BUDGET_ERROR_MESSAGES = {
+    BUDGET_EXCEEDED: "Budget limit exceeded for this model",
+    RATE_LIMIT_EXCEEDED: "Rate limit exceeded for this model",
+    MODEL_NOT_FOUND: "Model not found",
+    INVALID_BUDGET_CONFIG: "Invalid budget configuration",
+};
+// ============================================================================
+// Error Codes
+// ============================================================================
+export const BUDGET_ERROR_CODES = {
+    BUDGET_EXCEEDED: "budget_exceeded",
+    RATE_LIMIT_EXCEEDED: "rate_limit_exceeded",
+    TOKEN_LIMIT_EXCEEDED: "token_limit_exceeded",
+};
+//# sourceMappingURL=model-budget.constants.js.map

package/dist/model-budget/model-budget.constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-budget.constants.js","sourceRoot":"","sources":["../../lib/model-budget/model-budget.constants.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH;;;GAGG;AACH,MAAM,UAAU,yBAAyB;IACvC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,GAAG,KAAK,CAAC;AAChD,CAAC;AAID;;;;;;;GAOG;AACH,MAAM,UAAU,qBAAqB,CACnC,aAA+B,SAAS,EACxC,eAAwB,EACxB,gBAAyB;IAEzB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,qCAAqC;YACrC,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9E,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,6CAA6C;YAC7C,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC;YAC/B,MAAM,cAAc,GAAG,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,8BAA8B;YAC1F,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;YAC7B,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,cAAc,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC5B,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;QAC1B,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,mCAAmC;YACnC,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAC1F,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,8DAA8D;YAC9D,IAAI,CAAC,eAAe,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC1C,kDAAkD;gBAClD,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC9E,CAAC;YACD,MAAM,QAAQ,GAAG,gBAAgB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;YACxD,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,eAAe,CAAC,GAAG,QAAQ,CAAC,CAAC;YAChF,OAAO,eAAe,GAAG,cAAc,GAAG,QAAQ,CAAC;QACrD,CAAC;QACD;YACE,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAChF,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CACjC,aAA+B,SAAS,EACxC,eAAwB,EACxB,gBAAyB;IAEzB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,sBAAsB;YACtB,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAClF,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,mCAAmC;YACnC,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACpD,OAAO,WAAW,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAC/C,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,oBAAoB;YACpB,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9F,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,IAAI,CAAC,eAAe,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC1C,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAClF,CAAC;YACD,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,EAAE,eAAe,EAAE,gBAAgB,CAAC,CAAC;YACvF,OAAO,WAAW,GAAG,gBAAgB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAC9D,CAAC;QACD;YACE,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IACpF,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,yBAAyB;AACzB,+EAA+E;AAE/E,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,kFAAkF;IAClF,mBAAmB,EAAE,KAAK;IAC1B,yDAAyD;IACzD,sBAAsB,EAAE,GAAG;CACnB,CAAC;AAEX,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,eAAe,EAAE,sCAAsC;IACvD,mBAAmB,EAAE,oCAAoC;IACzD,eAAe,EAAE,iBAAiB;IAClC,qBAAqB,EAAE,8BAA8B;CAC7C,CAAC;AAEX,+EAA+E;AAC/E,cAAc;AACd,+EAA+E;AAE/E,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,eAAe,EAAE,iBAAiB;IAClC,mBAAmB,EAAE,qBAAqB;IAC1C,oBAAoB,EAAE,sBAAsB;CACpC,CAAC"}

package/dist/model-budget/model-budget.errors.d.ts

@@ -0,0 +1,111 @@
+/**
+ * Model Budget module error classes
+ *
+ * Custom error classes for budget and rate limit enforcement.
+ * These errors are designed to map directly to HTTP status codes
+ * and provide detailed information for API responses.
+ */
+import { ServiceError } from "../shared/services/services.errors";
+import type { BudgetExceededDetails, RateLimitExceededDetails, TokenLimitExceededDetails } from "./model-budget.interfaces";
+/**
+ * Error thrown when a model's budget limit has been exceeded.
+ *
+ * Uses HTTP 402 (Payment Required) to indicate that the request cannot
+ * be fulfilled due to budget constraints.
+ *
+ * @example
+ * throw new BudgetExceededError({
+ *   modelId: "gpt-4o",
+ *   providerId: "provider-uuid",
+ *   spent: 100.50,
+ *   limit: 100.00,
+ *   periodEnds: 1738281600000, // Unix timestamp (milliseconds)
+ * });
+ */
+export declare class BudgetExceededError extends ServiceError {
+    readonly code: "budget_exceeded";
+    readonly details: BudgetExceededDetails;
+    constructor(details: BudgetExceededDetails);
+    /**
+     * Convert to JSON for API response.
+     */
+    toJSON(): {
+        error: "budget_exceeded";
+        message: string;
+        details: BudgetExceededDetails;
+    };
+}
+/**
+ * Error thrown when a model's rate limit has been exceeded.
+ *
+ * Uses HTTP 429 (Too Many Requests) with a Retry-After header value.
+ *
+ * @example
+ * throw new RateLimitExceededError({
+ *   modelId: "gpt-4o",
+ *   providerId: "provider-uuid",
+ *   current: 61,
+ *   limit: 60,
+ *   retryAfter: 45,
+ * });
+ */
+export declare class RateLimitExceededError extends ServiceError {
+    readonly code: "rate_limit_exceeded";
+    readonly details: RateLimitExceededDetails;
+    readonly retryAfter: number;
+    constructor(details: RateLimitExceededDetails);
+    /**
+     * Convert to JSON for API response.
+     */
+    toJSON(): {
+        error: "rate_limit_exceeded";
+        message: string;
+        details: RateLimitExceededDetails;
+        retryAfter: number;
+    };
+}
+/**
+ * Error thrown when a model's token limit has been exceeded.
+ *
+ * Uses HTTP 402 (Payment Required) to indicate that the request cannot
+ * be fulfilled due to token quota constraints.
+ *
+ * @example
+ * throw new TokenLimitExceededError({
+ *   modelId: "gpt-4o",
+ *   providerId: "provider-uuid",
+ *   used: 1000001,
+ *   limit: 1000000,
+ *   periodEnds: 1738281600000, // Unix timestamp (milliseconds)
+ * });
+ */
+export declare class TokenLimitExceededError extends ServiceError {
+    readonly code: "token_limit_exceeded";
+    readonly details: TokenLimitExceededDetails;
+    constructor(details: TokenLimitExceededDetails);
+    /**
+     * Convert to JSON for API response.
+     */
+    toJSON(): {
+        error: "token_limit_exceeded";
+        message: string;
+        details: TokenLimitExceededDetails;
+    };
+}
+/**
+ * Check if an error is a BudgetExceededError.
+ */
+export declare function isBudgetExceededError(err: unknown): err is BudgetExceededError;
+/**
+ * Check if an error is a RateLimitExceededError.
+ */
+export declare function isRateLimitExceededError(err: unknown): err is RateLimitExceededError;
+/**
+ * Check if an error is a TokenLimitExceededError.
+ */
+export declare function isTokenLimitExceededError(err: unknown): err is TokenLimitExceededError;
+/**
+ * Check if an error is any type of limit exceeded error.
+ */
+export declare function isLimitExceededError(err: unknown): err is BudgetExceededError | RateLimitExceededError | TokenLimitExceededError;
+//# sourceMappingURL=model-budget.errors.d.ts.map

package/dist/model-budget/model-budget.errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-budget.errors.d.ts","sourceRoot":"","sources":["../../lib/model-budget/model-budget.errors.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AAKrE,OAAO,KAAK,EACV,qBAAqB,EACrB,wBAAwB,EACxB,yBAAyB,EAC1B,MAAM,4CAA4C,CAAC;AAMpD;;;;;;;;;;;;;;GAcG;AACH,qBAAa,mBAAoB,SAAQ,YAAY;IACnD,QAAQ,CAAC,IAAI,oBAAsC;IACnD,QAAQ,CAAC,OAAO,EAAE,qBAAqB,CAAC;gBAE5B,OAAO,EAAE,qBAAqB;IAO1C;;OAEG;IACH,MAAM;;;;;CAOP;AAMD;;;;;;;;;;;;;GAaG;AACH,qBAAa,sBAAuB,SAAQ,YAAY;IACtD,QAAQ,CAAC,IAAI,wBAA0C;IACvD,QAAQ,CAAC,OAAO,EAAE,wBAAwB,CAAC;IAC3C,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAEhB,OAAO,EAAE,wBAAwB;IAQ7C;;OAEG;IACH,MAAM;;;;;;CAQP;AAMD;;;;;;;;;;;;;;GAcG;AACH,qBAAa,uBAAwB,SAAQ,YAAY;IACvD,QAAQ,CAAC,IAAI,yBAA2C;IACxD,QAAQ,CAAC,OAAO,EAAE,yBAAyB,CAAC;gBAEhC,OAAO,EAAE,yBAAyB;IAO9C;;OAEG;IACH,MAAM;;;;;CAOP;AAMD;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,OAAO,GAAG,GAAG,IAAI,mBAAmB,CAE9E;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CAAC,GAAG,EAAE,OAAO,GAAG,GAAG,IAAI,sBAAsB,CAEpF;AAED;;GAEG;AACH,wBAAgB,yBAAyB,CAAC,GAAG,EAAE,OAAO,GAAG,GAAG,IAAI,uBAAuB,CAEtF;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,mBAAmB,GAAG,sBAAsB,GAAG,uBAAuB,CAI/E"}

package/dist/model-budget/model-budget.errors.js

@@ -0,0 +1,149 @@
+/**
+ * Model Budget module error classes
+ *
+ * Custom error classes for budget and rate limit enforcement.
+ * These errors are designed to map directly to HTTP status codes
+ * and provide detailed information for API responses.
+ */
+import { ServiceError } from "../shared/services/services.errors";
+import { BUDGET_ERROR_CODES, BUDGET_ERROR_MESSAGES, } from "./model-budget.constants";
+// ============================================================================
+// Budget Exceeded Error (HTTP 402 Payment Required)
+// ============================================================================
+/**
+ * Error thrown when a model's budget limit has been exceeded.
+ *
+ * Uses HTTP 402 (Payment Required) to indicate that the request cannot
+ * be fulfilled due to budget constraints.
+ *
+ * @example
+ * throw new BudgetExceededError({
+ *   modelId: "gpt-4o",
+ *   providerId: "provider-uuid",
+ *   spent: 100.50,
+ *   limit: 100.00,
+ *   periodEnds: 1738281600000, // Unix timestamp (milliseconds)
+ * });
+ */
+export class BudgetExceededError extends ServiceError {
+    constructor(details) {
+        const message = `${BUDGET_ERROR_MESSAGES.BUDGET_EXCEEDED}: spent $${details.spent.toFixed(6)} of $${details.limit.toFixed(6)} limit`;
+        super(402, message);
+        this.code = BUDGET_ERROR_CODES.BUDGET_EXCEEDED;
+        this.name = "BudgetExceededError";
+        this.details = details;
+    }
+    /**
+     * Convert to JSON for API response.
+     */
+    toJSON() {
+        return {
+            error: this.code,
+            message: this.message,
+            details: this.details,
+        };
+    }
+}
+// ============================================================================
+// Rate Limit Exceeded Error (HTTP 429 Too Many Requests)
+// ============================================================================
+/**
+ * Error thrown when a model's rate limit has been exceeded.
+ *
+ * Uses HTTP 429 (Too Many Requests) with a Retry-After header value.
+ *
+ * @example
+ * throw new RateLimitExceededError({
+ *   modelId: "gpt-4o",
+ *   providerId: "provider-uuid",
+ *   current: 61,
+ *   limit: 60,
+ *   retryAfter: 45,
+ * });
+ */
+export class RateLimitExceededError extends ServiceError {
+    constructor(details) {
+        const message = `${BUDGET_ERROR_MESSAGES.RATE_LIMIT_EXCEEDED}: ${details.current}/${details.limit} requests per minute`;
+        super(429, message);
+        this.code = BUDGET_ERROR_CODES.RATE_LIMIT_EXCEEDED;
+        this.name = "RateLimitExceededError";
+        this.details = details;
+        this.retryAfter = Math.ceil(details.retryAfter);
+    }
+    /**
+     * Convert to JSON for API response.
+     */
+    toJSON() {
+        return {
+            error: this.code,
+            message: this.message,
+            details: this.details,
+            retryAfter: this.retryAfter,
+        };
+    }
+}
+// ============================================================================
+// Token Limit Exceeded Error (HTTP 402 Payment Required)
+// ============================================================================
+/**
+ * Error thrown when a model's token limit has been exceeded.
+ *
+ * Uses HTTP 402 (Payment Required) to indicate that the request cannot
+ * be fulfilled due to token quota constraints.
+ *
+ * @example
+ * throw new TokenLimitExceededError({
+ *   modelId: "gpt-4o",
+ *   providerId: "provider-uuid",
+ *   used: 1000001,
+ *   limit: 1000000,
+ *   periodEnds: 1738281600000, // Unix timestamp (milliseconds)
+ * });
+ */
+export class TokenLimitExceededError extends ServiceError {
+    constructor(details) {
+        const message = `Token limit exceeded for this model: used ${details.used.toLocaleString()} of ${details.limit.toLocaleString()} tokens`;
+        super(402, message);
+        this.code = BUDGET_ERROR_CODES.TOKEN_LIMIT_EXCEEDED;
+        this.name = "TokenLimitExceededError";
+        this.details = details;
+    }
+    /**
+     * Convert to JSON for API response.
+     */
+    toJSON() {
+        return {
+            error: this.code,
+            message: this.message,
+            details: this.details,
+        };
+    }
+}
+// ============================================================================
+// Type Guards
+// ============================================================================
+/**
+ * Check if an error is a BudgetExceededError.
+ */
+export function isBudgetExceededError(err) {
+    return err instanceof BudgetExceededError;
+}
+/**
+ * Check if an error is a RateLimitExceededError.
+ */
+export function isRateLimitExceededError(err) {
+    return err instanceof RateLimitExceededError;
+}
+/**
+ * Check if an error is a TokenLimitExceededError.
+ */
+export function isTokenLimitExceededError(err) {
+    return err instanceof TokenLimitExceededError;
+}
+/**
+ * Check if an error is any type of limit exceeded error.
+ */
+export function isLimitExceededError(err) {
+    return (isBudgetExceededError(err) || isRateLimitExceededError(err) || isTokenLimitExceededError(err));
+}
+//# sourceMappingURL=model-budget.errors.js.map

package/dist/model-budget/model-budget.errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-budget.errors.js","sourceRoot":"","sources":["../../lib/model-budget/model-budget.errors.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AACrE,OAAO,EACL,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,2CAA2C,CAAC;AAOnD,+EAA+E;AAC/E,oDAAoD;AACpD,+EAA+E;AAE/E;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,mBAAoB,SAAQ,YAAY;IAInD,YAAY,OAA8B;QACxC,MAAM,OAAO,GAAG,GAAG,qBAAqB,CAAC,eAAe,YAAY,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC;QACrI,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QALb,SAAI,GAAG,kBAAkB,CAAC,eAAe,CAAC;QAMjD,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AAED,+EAA+E;AAC/E,yDAAyD;AACzD,+EAA+E;AAE/E;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAO,sBAAuB,SAAQ,YAAY;IAKtD,YAAY,OAAiC;QAC3C,MAAM,OAAO,GAAG,GAAG,qBAAqB,CAAC,mBAAmB,KAAK,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,KAAK,sBAAsB,CAAC;QACxH,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QANb,SAAI,GAAG,kBAAkB,CAAC,mBAAmB,CAAC;QAOrD,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC;CACF;AAED,+EAA+E;AAC/E,yDAAyD;AACzD,+EAA+E;AAE/E;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,uBAAwB,SAAQ,YAAY;IAIvD,YAAY,OAAkC;QAC5C,MAAM,OAAO,GAAG,6CAA6C,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,SAAS,CAAC;QACzI,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QALb,SAAI,GAAG,kBAAkB,CAAC,oBAAoB,CAAC;QAMtD,IAAI,CAAC,IAAI,GAAG,yBAAyB,CAAC;QACtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AAED,+EAA+E;AAC/E,cAAc;AACd,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,GAAY;IAChD,OAAO,GAAG,YAAY,mBAAmB,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,GAAY;IACnD,OAAO,GAAG,YAAY,sBAAsB,CAAC;AAC/C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB,CAAC,GAAY;IACpD,OAAO,GAAG,YAAY,uBAAuB,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,GAAY;IAEZ,OAAO,CACL,qBAAqB,CAAC,GAAG,CAAC,IAAI,wBAAwB,CAAC,GAAG,CAAC,IAAI,yBAAyB,CAAC,GAAG,CAAC,CAC9F,CAAC;AACJ,CAAC"}