captify 0.1.18 → 1.1.0

package/dist/model-budget/services/model-budget.service.js

@@ -0,0 +1,557 @@
+/**
+ * Model Budget Service
+ *
+ * Provides budget and rate limit enforcement for LLM model usage.
+ * Supports per-model spend limits, request rate limits, and optional token quotas.
+ *
+ * Features:
+ * - Budget tracking with auto-reset periods
+ * - Post-increment rate limiting (counts only successful completions)
+ * - Token limit tracking (optional)
+ * - In-memory config caching with 60-second TTL
+ * - Multi-tenant support
+ */
+import { BaseAwsService } from "../../shared/services/base-aws-service";
+import { DynamoDbInstances, DYNAMODB_ERROR_CODES, } from "../../shared/services/dynamodb";
+import { LoggerFactory } from "../../shared/services/logger";
+import { BaseAwsError } from "../../shared/services/services.errors";
+import { getLLMService } from "../../llm-providers/services";
+import { RateLimitExceededError } from "../model-budget.errors";
+import { modelUsageKeys } from "../../db/schemas";
+import { TABLES, USER_CONTEXT_WITH_METADATA } from "../../shared/shared.constants";
+import { BUDGET_DEFAULTS, getCurrentMinuteTimestamp, getCurrentPeriodStart, getCurrentPeriodEnd, } from "../model-budget.constants";
+// ============================================================================
+// Model Budget Service
+// ============================================================================
+export class ModelBudgetService extends BaseAwsService {
+    constructor(deps = {}) {
+        var _a;
+        super({ logger: deps.logger, serviceName: "model-budget" });
+        /**
+         * In-memory cache for model budget configurations.
+         * Keyed by `${tenantId}:${providerId}:${modelId}`.
+         */
+        this.configCache = new Map();
+        this.db = (_a = deps.db) !== null && _a !== void 0 ? _a : DynamoDbInstances.getInstance(TABLES.modelUsage);
+    }
+    // ==========================================================================
+    // Public Methods
+    // ==========================================================================
+    /**
+     * Check if a request is allowed based on budget and rate limits.
+     * This should be called BEFORE making the LLM request.
+     *
+     * @param params - Model identification params
+     * @returns LimitCheckResult indicating if request is allowed
+     *
+     * @example
+     * ```typescript
+     * const result = await budgetService.checkLimits({
+     *   tenantId: "tenant-123",
+     *   providerId: "provider-uuid",
+     *   modelId: "model-uuid",
+     * });
+     *
+     * if (!result.allowed) {
+     *   throw new BudgetExceededError(result.budget);
+     * }
+     * ```
+     */
+    async checkLimits(params) {
+        this.logger.debug("Checking budget limits", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+        });
+        const budgetConfig = await this.getCachedBudgetConfig(params);
+        // If no budget config, all requests are allowed
+        if (!budgetConfig) {
+            return { allowed: true };
+        }
+        // If budget is disabled, allow all requests without checking limits
+        if (budgetConfig.isEnabled === false) {
+            this.logger.debug("Budget is disabled, skipping limit checks", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+            });
+            return { allowed: true };
+        }
+        // Note: Rate limit is checked atomically in recordRequest() method
+        // This method only checks budget/token limits
+        // Check budget limit
+        const budgetResult = await this.checkBudget(params, budgetConfig);
+        if (!budgetResult.allowed) {
+            return {
+                allowed: false,
+                reason: budgetResult.reason,
+                budget: budgetResult,
+            };
+        }
+        return {
+            allowed: true,
+            budget: budgetResult,
+        };
+    }
+    /**
+     * Atomically check rate limit and record request.
+     * Uses conditional update to prevent race conditions.
+     *
+     * @param params - Model identification params
+     * @throws RateLimitExceededError if rate limit is exceeded
+     *
+     * @example
+     * ```typescript
+     * // Before making LLM call
+     * await budgetService.recordRequest({
+     *   tenantId: "tenant-123",
+     *   providerId: "provider-uuid",
+     *   modelId: "model-uuid",
+     * });
+     * ```
+     */
+    async recordRequest(params) {
+        this.logger.debug("Recording request with atomic rate limit check", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+        });
+        const budgetConfig = await this.getCachedBudgetConfig(params);
+        // No config means no limits to enforce
+        if (!budgetConfig) {
+            return;
+        }
+        // If budget is disabled, skip rate limiting
+        if (budgetConfig.isEnabled === false) {
+            return;
+        }
+        // If rate limit is 0 or not set, no rate limiting needed
+        const rateLimit = budgetConfig.requestsPerMinute;
+        if (!rateLimit || rateLimit <= 0) {
+            return;
+        }
+        const PK = modelUsageKeys.pk(params.tenantId, params.providerId, params.modelId);
+        const currentMinute = getCurrentMinuteTimestamp();
+        const SK = modelUsageKeys.rateLimitSk(currentMinute);
+        // TTL: 2 minutes from now (in seconds for DynamoDB)
+        const ttl = Math.floor(Date.now() / 1000) + BUDGET_DEFAULTS.RATE_LIMIT_TTL_SECONDS;
+        // Calculate seconds until this minute window resets
+        const now = Date.now();
+        const nextMinute = currentMinute + 60000;
+        const resetsIn = Math.ceil((nextMinute - now) / 1000);
+        try {
+            // Atomic check-and-increment: only succeeds if count < limit
+            await this.db.updateItem({ PK, SK }, "SET requestCount = if_not_exists(requestCount, :zero) + :one, #ttl = :ttl", { "#ttl": "ttl" }, { ":zero": 0, ":one": 1, ":ttl": ttl, ":limit": rateLimit }, {
+                conditionExpression: "attribute_not_exists(requestCount) OR requestCount < :limit",
+                userCredentials: params.userCredentials,
+            });
+            this.logger.debug("Request recorded successfully", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                minute: currentMinute,
+                rateLimit,
+            });
+        }
+        catch (error) {
+            // Check if this is a conditional check failure (rate limit exceeded)
+            const isConditionalCheckFailed = (error instanceof BaseAwsError &&
+                error.code === DYNAMODB_ERROR_CODES.CONDITIONAL_CHECK_FAILED) ||
+                (error instanceof Error &&
+                    error.message.includes(DYNAMODB_ERROR_CODES.CONDITIONAL_CHECK_FAILED));
+            if (isConditionalCheckFailed) {
+                this.logger.warn("Rate limit exceeded (atomic check)", {
+                    tenantId: params.tenantId,
+                    providerId: params.providerId,
+                    modelId: params.modelId,
+                    limit: rateLimit,
+                    resetsIn,
+                });
+                throw new RateLimitExceededError({
+                    modelId: params.modelId,
+                    providerId: params.providerId,
+                    current: rateLimit, // At or above limit
+                    limit: rateLimit,
+                    retryAfter: resetsIn,
+                });
+            }
+            // Log other errors but don't block the request
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            this.logger.warn("Failed to record request for rate limiting", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                error: errorMessage,
+            });
+        }
+    }
+    /**
+     * Record spend after a request completes.
+     * This should be called AFTER the request completes successfully.
+     *
+     * @param params - Model identification and spend amount
+     *
+     * @example
+     * ```typescript
+     * // In onFinish callback
+     * await budgetService.recordSpend({
+     *   tenantId: "tenant-123",
+     *   providerId: "provider-uuid",
+     *   modelId: "model-uuid",
+     *   amount: 0.0025, // USD
+     *   tokens: 500,    // Optional
+     * });
+     * ```
+     */
+    async recordSpend(params) {
+        var _a;
+        this.logger.info("recordSpend called", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+            amount: params.amount,
+            tokens: params.tokens,
+        });
+        const budgetConfig = await this.getCachedBudgetConfig(params);
+        // Only record if budget config exists (track even if budget amount is 0/unlimited)
+        if (!budgetConfig) {
+            this.logger.info("No budget config found, skipping spend tracking", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+            });
+            return;
+        }
+        this.logger.info("Recording spend for budget tracking", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+            amount: params.amount,
+            budgetAmount: budgetConfig.budgetAmount,
+            isEnabled: budgetConfig.isEnabled,
+        });
+        const PK = modelUsageKeys.pk(params.tenantId, params.providerId, params.modelId);
+        const periodStart = getCurrentPeriodStart((_a = budgetConfig.budgetPeriodType) !== null && _a !== void 0 ? _a : "monthly", budgetConfig.budgetStartDate, budgetConfig.budgetPeriodDays);
+        const SK = modelUsageKeys.periodSk(periodStart);
+        try {
+            // Build update expression based on what we're tracking
+            let updateExpression = "SET currentSpend = if_not_exists(currentSpend, :zero) + :amount, " +
+                "requestCount = if_not_exists(requestCount, :zeroInt) + :one, " +
+                "updatedAt = :now";
+            const expressionValues = {
+                ":zero": 0,
+                ":zeroInt": 0,
+                ":one": 1,
+                ":amount": params.amount,
+                ":now": Date.now(),
+            };
+            // Add token tracking if provided
+            if (params.tokens !== undefined) {
+                updateExpression += ", totalTokens = if_not_exists(totalTokens, :zeroTokens) + :tokens";
+                expressionValues[":zeroTokens"] = 0;
+                expressionValues[":tokens"] = params.tokens;
+            }
+            await this.db.updateItem({ PK, SK }, updateExpression, undefined, expressionValues, {
+                userCredentials: params.userCredentials,
+            });
+            this.logger.debug("Recorded spend", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                amount: params.amount,
+                periodStart,
+            });
+        }
+        catch (error) {
+            // Log but don't throw - spend tracking failure shouldn't block requests
+            this.logger.warn("Failed to record spend", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                error: error instanceof Error ? error.message : String(error),
+            });
+        }
+    }
+    /**
+     * Get current usage statistics for a model.
+     *
+     * @param params - Model identification params
+     * @returns ModelUsageResponse with budget and rate limit info
+     *
+     * @example
+     * ```typescript
+     * const usage = await budgetService.getUsage({
+     *   tenantId: "tenant-123",
+     *   providerId: "provider-uuid",
+     *   modelId: "model-uuid",
+     * });
+     * console.log(`Spent: $${usage.budget?.spent}`);
+     * ```
+     */
+    async getUsage(params) {
+        var _a, _b, _c, _d, _e, _f;
+        this.logger.debug("Getting usage", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+        });
+        const budgetConfig = await this.getCachedBudgetConfig(params);
+        const response = {
+            modelId: params.modelId,
+            providerId: params.providerId,
+            tenantId: params.tenantId,
+            hasLimitsConfigured: false,
+        };
+        if (!budgetConfig) {
+            return response;
+        }
+        response.hasLimitsConfigured = true;
+        const PK = modelUsageKeys.pk(params.tenantId, params.providerId, params.modelId);
+        // Get budget usage if configured (0 is a valid limit, only skip if undefined/null)
+        if (budgetConfig.budgetAmount != null) {
+            const periodType = (_a = budgetConfig.budgetPeriodType) !== null && _a !== void 0 ? _a : "monthly";
+            const periodStart = getCurrentPeriodStart(periodType, budgetConfig.budgetStartDate, budgetConfig.budgetPeriodDays);
+            const periodEnd = getCurrentPeriodEnd(periodType, budgetConfig.budgetStartDate, budgetConfig.budgetPeriodDays);
+            const SK = modelUsageKeys.periodSk(periodStart);
+            const { Item } = await this.db.getItem({ PK, SK }, { userCredentials: params.userCredentials });
+            const usageRecord = Item;
+            const spent = (_b = usageRecord === null || usageRecord === void 0 ? void 0 : usageRecord.currentSpend) !== null && _b !== void 0 ? _b : 0;
+            const remaining = Math.max(0, budgetConfig.budgetAmount - spent);
+            response.budget = {
+                limit: budgetConfig.budgetAmount,
+                spent,
+                remaining,
+                periodStart: periodStart,
+                periodEnd: periodEnd,
+                requestCount: (_c = usageRecord === null || usageRecord === void 0 ? void 0 : usageRecord.requestCount) !== null && _c !== void 0 ? _c : 0,
+            };
+            // Add token usage (always include if we have usage data)
+            const tokensUsed = (_d = usageRecord === null || usageRecord === void 0 ? void 0 : usageRecord.totalTokens) !== null && _d !== void 0 ? _d : 0;
+            if (budgetConfig.tokenLimit || tokensUsed > 0) {
+                response.tokenUsage = {
+                    limit: (_e = budgetConfig.tokenLimit) !== null && _e !== void 0 ? _e : 0,
+                    used: tokensUsed,
+                    remaining: budgetConfig.tokenLimit
+                        ? Math.max(0, budgetConfig.tokenLimit - tokensUsed)
+                        : 0,
+                };
+            }
+        }
+        // Get rate limit usage if configured
+        if (budgetConfig.requestsPerMinute) {
+            const currentMinute = getCurrentMinuteTimestamp();
+            const SK = modelUsageKeys.rateLimitSk(currentMinute);
+            const { Item } = await this.db.getItem({ PK, SK }, { userCredentials: params.userCredentials });
+            const rateLimitRecord = Item;
+            response.rateLimit = {
+                limit: budgetConfig.requestsPerMinute,
+                currentMinuteCount: (_f = rateLimitRecord === null || rateLimitRecord === void 0 ? void 0 : rateLimitRecord.requestCount) !== null && _f !== void 0 ? _f : 0,
+                resetsIn: 60 - Math.floor((Date.now() % 60000) / 1000),
+            };
+        }
+        return response;
+    }
+    /**
+     * Get usage for multiple models in a single call.
+     * More efficient than calling getUsage multiple times.
+     *
+     * @param params - Array of model identifiers
+     * @returns Array of usage responses (in same order as input)
+     *
+     * @example
+     * ```typescript
+     * const usages = await budgetService.getBatchUsage([
+     *   { tenantId: "t1", providerId: "p1", modelId: "uuid1" },
+     *   { tenantId: "t1", providerId: "p1", modelId: "uuid2" },
+     * ]);
+     * ```
+     */
+    async getBatchUsage(params) {
+        // Process all in parallel
+        const results = await Promise.all(params.map((p) => this.getUsage(p)));
+        return results;
+    }
+    /**
+     * Clear the config cache for a specific model.
+     * Call this when budget config is updated to ensure immediate effect.
+     */
+    clearConfigCache(params) {
+        const cacheKey = this.buildCacheKey(params);
+        this.configCache.delete(cacheKey);
+        this.logger.debug("Cleared config cache", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+        });
+    }
+    /**
+     * Clear all cached configs.
+     * Useful for testing or after bulk updates.
+     */
+    clearAllConfigCache() {
+        this.configCache.clear();
+        this.logger.debug("Cleared all config cache");
+    }
+    /**
+     * Check budget limit for a model.
+     */
+    async checkBudget(params, budgetConfig) {
+        var _a, _b, _c;
+        // No budget limit configured (only undefined/null means unlimited, 0 = no spend allowed)
+        if ((budgetConfig === null || budgetConfig === void 0 ? void 0 : budgetConfig.budgetAmount) == null) {
+            return { allowed: true };
+        }
+        const PK = modelUsageKeys.pk(params.tenantId, params.providerId, params.modelId);
+        const periodType = (_a = budgetConfig.budgetPeriodType) !== null && _a !== void 0 ? _a : "monthly";
+        const periodStart = getCurrentPeriodStart(periodType, budgetConfig.budgetStartDate, budgetConfig.budgetPeriodDays);
+        const periodEnd = getCurrentPeriodEnd(periodType, budgetConfig.budgetStartDate, budgetConfig.budgetPeriodDays);
+        const SK = modelUsageKeys.periodSk(periodStart);
+        const { Item } = await this.db.getItem({ PK, SK }, { userCredentials: params.userCredentials });
+        const usageRecord = Item;
+        const currentSpend = (_b = usageRecord === null || usageRecord === void 0 ? void 0 : usageRecord.currentSpend) !== null && _b !== void 0 ? _b : 0;
+        const remaining = Math.max(0, budgetConfig.budgetAmount - currentSpend);
+        // Check budget limit
+        if (currentSpend >= budgetConfig.budgetAmount) {
+            return {
+                allowed: false,
+                reason: "budget_exceeded",
+                spent: currentSpend,
+                limit: budgetConfig.budgetAmount,
+                remaining: 0,
+                periodStart: periodStart,
+                periodEnd: periodEnd,
+            };
+        }
+        // Check token limit if configured
+        if (budgetConfig.tokenLimit) {
+            const tokenSpent = (_c = usageRecord === null || usageRecord === void 0 ? void 0 : usageRecord.totalTokens) !== null && _c !== void 0 ? _c : 0;
+            if (tokenSpent >= budgetConfig.tokenLimit) {
+                return {
+                    allowed: false,
+                    reason: "token_limit_exceeded",
+                    spent: currentSpend,
+                    limit: budgetConfig.budgetAmount,
+                    remaining,
+                    tokenSpent,
+                    tokenLimit: budgetConfig.tokenLimit,
+                    tokenRemaining: 0,
+                    periodStart: periodStart,
+                    periodEnd: periodEnd,
+                };
+            }
+        }
+        return {
+            allowed: true,
+            spent: currentSpend,
+            limit: budgetConfig.budgetAmount,
+            remaining,
+            periodStart: periodStart,
+            periodEnd: periodEnd,
+        };
+    }
+    /**
+     * Get budget config from cache or fetch from LLM service.
+     */
+    async getCachedBudgetConfig(params) {
+        var _a, _b;
+        const cacheKey = this.buildCacheKey(params);
+        const cached = this.configCache.get(cacheKey);
+        // Return cached if still valid
+        if (cached) {
+            const cacheAge = Date.now() - cached.cachedAt;
+            // Handle future cachedAt (clock skew or bug) - treat as expired
+            if (cacheAge < 0) {
+                this.logger.warn("Cache entry has future timestamp, treating as expired", {
+                    tenantId: params.tenantId,
+                    providerId: params.providerId,
+                    modelId: params.modelId,
+                    cachedAt: cached.cachedAt,
+                    now: Date.now(),
+                });
+            }
+            else if (cacheAge < BUDGET_DEFAULTS.CONFIG_CACHE_TTL_MS) {
+                this.logger.info("Returning cached budget config", {
+                    tenantId: params.tenantId,
+                    providerId: params.providerId,
+                    modelId: params.modelId,
+                    hasBudgetConfig: !!cached.budgetConfig,
+                });
+                return cached.budgetConfig;
+            }
+        }
+        // Fetch model from LLM service
+        try {
+            const model = await getLLMService().getModel(params.tenantId, params.providerId, params.modelId, { userCredentials: params.userCredentials });
+            if (!model) {
+                this.logger.info("Model not found, no budget config", {
+                    tenantId: params.tenantId,
+                    providerId: params.providerId,
+                    modelId: params.modelId,
+                });
+                return undefined;
+            }
+            this.logger.info("Fetched model for budget config", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                hasBudgetConfig: !!model.budgetConfig,
+                budgetAmount: (_a = model.budgetConfig) === null || _a === void 0 ? void 0 : _a.budgetAmount,
+                isEnabled: (_b = model.budgetConfig) === null || _b === void 0 ? void 0 : _b.isEnabled,
+            });
+            // Cache the result
+            if (model.budgetConfig) {
+                this.configCache.set(cacheKey, {
+                    budgetConfig: model.budgetConfig,
+                    cachedAt: Date.now(),
+                });
+            }
+            return model.budgetConfig;
+        }
+        catch (error) {
+            this.logger.warn("Failed to fetch model config", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                error: error instanceof Error ? error.message : String(error),
+            });
+            // Return cached even if expired, better than failing
+            return cached === null || cached === void 0 ? void 0 : cached.budgetConfig;
+        }
+    }
+    /**
+     * Build cache key for model config.
+     */
+    buildCacheKey(params) {
+        return `${params.tenantId}:${params.providerId}:${params.modelId}`;
+    }
+}
+// ============================================================================
+// Singleton Factory
+// ============================================================================
+/**
+ * Lazy-initialized singleton instance of ModelBudgetService.
+ */
+let modelBudgetServiceInstance = null;
+/**
+ * Get the singleton ModelBudgetService instance.
+ *
+ * @returns ModelBudgetService instance
+ *
+ * @example
+ * ```typescript
+ * const budgetService = getModelBudgetService();
+ * const result = await budgetService.checkLimits({ tenantId, providerId, modelId });
+ * ```
+ */
+export function getModelBudgetService() {
+    if (!modelBudgetServiceInstance) {
+        modelBudgetServiceInstance = new ModelBudgetService({
+            logger: LoggerFactory.createLogger({
+                serviceName: "model-budget",
+                userContext: USER_CONTEXT_WITH_METADATA,
+            }),
+        });
+    }
+    return modelBudgetServiceInstance;
+}
+//# sourceMappingURL=model-budget.service.js.map

package/dist/model-budget/services/model-budget.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-budget.service.js","sourceRoot":"","sources":["../../../lib/model-budget/services/model-budget.service.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AACxE,OAAO,EAEL,iBAAiB,EACjB,oBAAoB,GACrB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,aAAa,EAAe,MAAM,8BAA8B,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAE7D,OAAO,EAAE,sBAAsB,EAAE,MAAM,wCAAwC,CAAC;AAMhF,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,MAAM,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAEnF,OAAO,EACL,eAAe,EACf,yBAAyB,EACzB,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,2CAA2C,CAAC;AAoBnD,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E,MAAM,OAAO,kBAAmB,SAAQ,cAAc;IASpD,YAAY,OAA+B,EAAE;;QAC3C,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC,CAAC;QAP9D;;;WAGG;QACc,gBAAW,GAAmC,IAAI,GAAG,EAAE,CAAC;QAIvE,IAAI,CAAC,EAAE,GAAG,MAAA,IAAI,CAAC,EAAE,mCAAI,iBAAiB,CAAC,WAAW,CAAwB,MAAM,CAAC,UAAU,CAAC,CAAC;IAC/F,CAAC;IAED,6EAA6E;IAC7E,iBAAiB;IACjB,6EAA6E;IAE7E;;;;;;;;;;;;;;;;;;;OAmBG;IACH,KAAK,CAAC,WAAW,CAAC,MAAyB;QACzC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;YAC1C,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAE9D,gDAAgD;QAChD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,oEAAoE;QACpE,IAAI,YAAY,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;YACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;gBAC7D,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YACH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,mEAAmE;QACnE,8CAA8C;QAE9C,qBAAqB;QACrB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC1B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,YAAY,CAAC,MAAM;gBAC3B,MAAM,EAAE,YAAY;aACrB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,YAAY;SACrB,CAAC;IACJ,CAAC;IAED;;;;;;;;;;;;;;;;OAgBG;IACH,KAAK,CAAC,aAAa,CAAC,MAAyB;QAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gDAAgD,EAAE;YAClE,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAE9D,uCAAuC;QACvC,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,4CAA4C;QAC5C,IAAI,YAAY,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;YACrC,OAAO;QACT,CAAC;QAED,yDAAyD;QACzD,MAAM,SAAS,GAAG,YAAY,CAAC,iBAAiB,CAAC;QACjD,IAAI,CAAC,SAAS,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,aAAa,GAAG,yBAAyB,EAAE,CAAC;QAClD,MAAM,EAAE,GAAG,cAAc,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;QAErD,oDAAoD;QACpD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,eAAe,CAAC,sBAAsB,CAAC;QAEnF,oDAAoD;QACpD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,aAAa,GAAG,KAAK,CAAC;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QAEtD,IAAI,CAAC;YACH,6DAA6D;YAC7D,MAAM,IAAI,CAAC,EAAE,CAAC,UAAU,CACtB,EAAE,EAAE,EAAE,EAAE,EAAE,EACV,2EAA2E,EAC3E,EAAE,MAAM,EAAE,KAAK,EAAE,EACjB,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAC3D;gBACE,mBAAmB,EAAE,6DAA6D;gBAClF,eAAe,EAAE,MAAM,CAAC,eAAe;aACxC,CACF,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE;gBACjD,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,MAAM,EAAE,aAAa;gBACrB,SAAS;aACV,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,qEAAqE;YACrE,MAAM,wBAAwB,GAC5B,CAAC,KAAK,YAAY,YAAY;gBAC5B,KAAK,CAAC,IAAI,KAAK,oBAAoB,CAAC,wBAAwB,CAAC;gBAC/D,CAAC,KAAK,YAAY,KAAK;oBACrB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC,CAAC;YAE3E,IAAI,wBAAwB,EAAE,CAAC;gBAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE;oBACrD,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,KAAK,EAAE,SAAS;oBAChB,QAAQ;iBACT,CAAC,CAAC;gBAEH,MAAM,IAAI,sBAAsB,CAAC;oBAC/B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,OAAO,EAAE,SAAS,EAAE,oBAAoB;oBACxC,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,QAAQ;iBACrB,CAAC,CAAC;YACL,CAAC;YAED,+CAA+C;YAC/C,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4CAA4C,EAAE;gBAC7D,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;;;;OAiBG;IACH,KAAK,CAAC,WAAW,CAAC,MAAyB;;QACzC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE;YACrC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAE9D,mFAAmF;QACnF,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iDAAiD,EAAE;gBAClE,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qCAAqC,EAAE;YACtD,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,YAAY,EAAE,YAAY,CAAC,YAAY;YACvC,SAAS,EAAE,YAAY,CAAC,SAAS;SAClC,CAAC,CAAC;QAEH,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,WAAW,GAAG,qBAAqB,CACvC,MAAA,YAAY,CAAC,gBAAgB,mCAAI,SAAS,EAC1C,YAAY,CAAC,eAAe,EAC5B,YAAY,CAAC,gBAAgB,CAC9B,CAAC;QACF,MAAM,EAAE,GAAG,cAAc,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAEhD,IAAI,CAAC;YACH,uDAAuD;YACvD,IAAI,gBAAgB,GAClB,mEAAmE;gBACnE,+DAA+D;gBAC/D,kBAAkB,CAAC;YAErB,MAAM,gBAAgB,GAA4B;gBAChD,OAAO,EAAE,CAAC;gBACV,UAAU,EAAE,CAAC;gBACb,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE;aACnB,CAAC;YAEF,iCAAiC;YACjC,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,gBAAgB,IAAI,mEAAmE,CAAC;gBACxF,gBAAgB,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;gBACpC,gBAAgB,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9C,CAAC;YAED,MAAM,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,gBAAgB,EAAE,SAAS,EAAE,gBAAgB,EAAE;gBAClF,eAAe,EAAE,MAAM,CAAC,eAAe;aACxC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE;gBAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,WAAW;aACZ,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,wEAAwE;YACxE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;gBACzC,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;;OAeG;IACH,KAAK,CAAC,QAAQ,CAAC,MAAyB;;QACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;YACjC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAE9D,MAAM,QAAQ,GAAuB;YACnC,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,mBAAmB,EAAE,KAAK;SAC3B,CAAC;QAEF,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,QAAQ,CAAC,mBAAmB,GAAG,IAAI,CAAC;QAEpC,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAEjF,mFAAmF;QACnF,IAAI,YAAY,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;YACtC,MAAM,UAAU,GAAG,MAAA,YAAY,CAAC,gBAAgB,mCAAI,SAAS,CAAC;YAC9D,MAAM,WAAW,GAAG,qBAAqB,CACvC,UAAU,EACV,YAAY,CAAC,eAAe,EAC5B,YAAY,CAAC,gBAAgB,CAC9B,CAAC;YACF,MAAM,SAAS,GAAG,mBAAmB,CACnC,UAAU,EACV,YAAY,CAAC,eAAe,EAC5B,YAAY,CAAC,gBAAgB,CAC9B,CAAC;YACF,MAAM,EAAE,GAAG,cAAc,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAEhD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CACpC,EAAE,EAAE,EAAE,EAAE,EAAE,EACV,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAC5C,CAAC;YACF,MAAM,WAAW,GAAG,IAAoC,CAAC;YAEzD,MAAM,KAAK,GAAG,MAAA,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,YAAY,mCAAI,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,YAAY,GAAG,KAAK,CAAC,CAAC;YAEjE,QAAQ,CAAC,MAAM,GAAG;gBAChB,KAAK,EAAE,YAAY,CAAC,YAAY;gBAChC,KAAK;gBACL,SAAS;gBACT,WAAW,EAAE,WAAW;gBACxB,SAAS,EAAE,SAAS;gBACpB,YAAY,EAAE,MAAA,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,YAAY,mCAAI,CAAC;aAC7C,CAAC;YAEF,yDAAyD;YACzD,MAAM,UAAU,GAAG,MAAA,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,WAAW,mCAAI,CAAC,CAAC;YACjD,IAAI,YAAY,CAAC,UAAU,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;gBAC9C,QAAQ,CAAC,UAAU,GAAG;oBACpB,KAAK,EAAE,MAAA,YAAY,CAAC,UAAU,mCAAI,CAAC;oBACnC,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,YAAY,CAAC,UAAU;wBAChC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,UAAU,GAAG,UAAU,CAAC;wBACnD,CAAC,CAAC,CAAC;iBACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,qCAAqC;QACrC,IAAI,YAAY,CAAC,iBAAiB,EAAE,CAAC;YACnC,MAAM,aAAa,GAAG,yBAAyB,EAAE,CAAC;YAClD,MAAM,EAAE,GAAG,cAAc,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CACpC,EAAE,EAAE,EAAE,EAAE,EAAE,EACV,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAC5C,CAAC;YACF,MAAM,eAAe,GAAG,IAAmC,CAAC;YAE5D,QAAQ,CAAC,SAAS,GAAG;gBACnB,KAAK,EAAE,YAAY,CAAC,iBAAiB;gBACrC,kBAAkB,EAAE,MAAA,eAAe,aAAf,eAAe,uBAAf,eAAe,CAAE,YAAY,mCAAI,CAAC;gBACtD,QAAQ,EAAE,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,GAAG,IAAI,CAAC;aACvD,CAAC;QACJ,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,aAAa,CAAC,MAA2B;QAC7C,0BAA0B;QAC1B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvE,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,MAAyB;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE;YACxC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACjB,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CACvB,MAAyB,EACzB,YAA0B;;QAE1B,yFAAyF;QACzF,IAAI,CAAA,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,YAAY,KAAI,IAAI,EAAE,CAAC;YACvC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,UAAU,GAAG,MAAA,YAAY,CAAC,gBAAgB,mCAAI,SAAS,CAAC;QAC9D,MAAM,WAAW,GAAG,qBAAqB,CACvC,UAAU,EACV,YAAY,CAAC,eAAe,EAC5B,YAAY,CAAC,gBAAgB,CAC9B,CAAC;QACF,MAAM,SAAS,GAAG,mBAAmB,CACnC,UAAU,EACV,YAAY,CAAC,eAAe,EAC5B,YAAY,CAAC,gBAAgB,CAC9B,CAAC;QACF,MAAM,EAAE,GAAG,cAAc,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAEhD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC;QAChG,MAAM,WAAW,GAAG,IAAoC,CAAC;QAEzD,MAAM,YAAY,GAAG,MAAA,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,YAAY,mCAAI,CAAC,CAAC;QACpD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,YAAY,GAAG,YAAY,CAAC,CAAC;QAExE,qBAAqB;QACrB,IAAI,YAAY,IAAI,YAAY,CAAC,YAAY,EAAE,CAAC;YAC9C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,iBAAiB;gBACzB,KAAK,EAAE,YAAY;gBACnB,KAAK,EAAE,YAAY,CAAC,YAAY;gBAChC,SAAS,EAAE,CAAC;gBACZ,WAAW,EAAE,WAAW;gBACxB,SAAS,EAAE,SAAS;aACrB,CAAC;QACJ,CAAC;QAED,kCAAkC;QAClC,IAAI,YAAY,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,MAAA,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,WAAW,mCAAI,CAAC,CAAC;YACjD,IAAI,UAAU,IAAI,YAAY,CAAC,UAAU,EAAE,CAAC;gBAC1C,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,sBAAsB;oBAC9B,KAAK,EAAE,YAAY;oBACnB,KAAK,EAAE,YAAY,CAAC,YAAY;oBAChC,SAAS;oBACT,UAAU;oBACV,UAAU,EAAE,YAAY,CAAC,UAAU;oBACnC,cAAc,EAAE,CAAC;oBACjB,WAAW,EAAE,WAAW;oBACxB,SAAS,EAAE,SAAS;iBACrB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,YAAY;YACnB,KAAK,EAAE,YAAY,CAAC,YAAY;YAChC,SAAS;YACT,WAAW,EAAE,WAAW;YACxB,SAAS,EAAE,SAAS;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB,CACjC,MAAyB;;QAEzB,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAE9C,+BAA+B;QAC/B,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC;YAE9C,gEAAgE;YAChE,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;gBACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uDAAuD,EAAE;oBACxE,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE;iBAChB,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,QAAQ,GAAG,eAAe,CAAC,mBAAmB,EAAE,CAAC;gBAC1D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gCAAgC,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,eAAe,EAAE,CAAC,CAAC,MAAM,CAAC,YAAY;iBACvC,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC,YAAY,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,+BAA+B;QAC/B,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,aAAa,EAAE,CAAC,QAAQ,CAC1C,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,OAAO,EACd,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAC5C,CAAC;YAEF,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,EAAE;oBACpD,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;iBACxB,CAAC,CAAC;gBACH,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;gBAClD,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY;gBACrC,YAAY,EAAE,MAAA,KAAK,CAAC,YAAY,0CAAE,YAAY;gBAC9C,SAAS,EAAE,MAAA,KAAK,CAAC,YAAY,0CAAE,SAAS;aACzC,CAAC,CAAC;YAEH,mBAAmB;YACnB,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;gBACvB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE;oBAC7B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;iBACrB,CAAC,CAAC;YACL,CAAC;YAED,OAAO,KAAK,CAAC,YAAY,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE;gBAC/C,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;YAEH,qDAAqD;YACrD,OAAO,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,YAAY,CAAC;QAC9B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,MAAyB;QAC7C,OAAO,GAAG,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;IACrE,CAAC;CACF;AAED,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E;;GAEG;AACH,IAAI,0BAA0B,GAA8B,IAAI,CAAC;AAEjE;;;;;;;;;;GAUG;AACH,MAAM,UAAU,qBAAqB;IACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAChC,0BAA0B,GAAG,IAAI,kBAAkB,CAAC;YAClD,MAAM,EAAE,aAAa,CAAC,YAAY,CAAC;gBACjC,WAAW,EAAE,cAAc;gBAC3B,WAAW,EAAE,0BAA0B;aACxC,CAAC;SACH,CAAC,CAAC;IACL,CAAC;IACD,OAAO,0BAA0B,CAAC;AACpC,CAAC"}

package/dist/model-budget/services/usage-tracking.service.d.ts

@@ -0,0 +1,85 @@
+/**
+ * Usage Tracking Service
+ *
+ * Centralized service for tracking LLM usage across all operations
+ * (chat, agents, config generation, etc.). Provides a unified interface
+ * for checking limits before requests and recording usage after completion.
+ */
+import type { LanguageModelUsage } from "ai";
+import type { Model } from "../../llm-providers/llm-providers.schemas";
+import type { UsageTrackingParams, UsageData, UsageRecordResult, UsageTrackingServiceDeps } from "../model-budget.interfaces";
+/**
+ * Service for tracking LLM usage across all operations.
+ *
+ * Provides methods to:
+ * - Check limits before making LLM calls
+ * - Record usage after LLM calls complete
+ * - Handle streaming responses with async usage tracking
+ *
+ * @example
+ * ```typescript
+ * const tracker = getUsageTrackingService();
+ *
+ * // Before making an LLM call
+ * await tracker.checkLimitsOrThrow({ tenantId, providerId, modelId, userCredentials });
+ *
+ * // After LLM call completes
+ * await tracker.recordUsage(
+ *   { tenantId, providerId, modelId, userCredentials },
+ *   model,
+ *   result.usage,
+ * );
+ * ```
+ */
+export declare class UsageTrackingService {
+    private logger;
+    private budgetService;
+    constructor(deps?: UsageTrackingServiceDeps);
+    /**
+     * Check budget and rate limits before making an LLM call.
+     * Uses atomic rate limit check to prevent race conditions.
+     * Throws appropriate error if limits are exceeded.
+     *
+     * @param params - Tracking parameters
+     * @throws BudgetExceededError if budget limit is exceeded
+     * @throws RateLimitExceededError if rate limit is exceeded
+     * @throws TokenLimitExceededError if token limit is exceeded
+     */
+    checkLimitsOrThrow(params: UsageTrackingParams): Promise<void>;
+    /**
+     * Record usage after an LLM call completes.
+     * Only records the spend (request count is recorded in checkLimitsOrThrow).
+     *
+     * @param params - Tracking parameters
+     * @param model - Model configuration (for pricing)
+     * @param usage - Usage data from AI SDK
+     * @returns Result with tokens and cost recorded
+     */
+    recordUsage(params: UsageTrackingParams, model: Model, usage?: UsageData | LanguageModelUsage | null): Promise<UsageRecordResult>;
+    /**
+     * Record usage from a streaming response.
+     * This is a fire-and-forget operation that handles the async usage promise.
+     *
+     * @param params - Tracking parameters
+     * @param model - Model configuration (for pricing)
+     * @param usagePromise - Promise that resolves to usage data when stream completes
+     */
+    recordStreamUsage(params: UsageTrackingParams, model: Model, usagePromise: Promise<UsageData | LanguageModelUsage | undefined>): void;
+    /**
+     * Wrapper for executing an LLM operation with automatic usage tracking.
+     * Checks limits before, executes the operation, and records usage after.
+     *
+     * @param params - Tracking parameters
+     * @param model - Model configuration (for pricing)
+     * @param operation - The LLM operation to execute
+     * @returns The result of the operation
+     */
+    withTracking<T extends {
+        usage?: UsageData | LanguageModelUsage;
+    }>(params: UsageTrackingParams, model: Model, operation: () => Promise<T>): Promise<T>;
+}
+/**
+ * Get or create the singleton UsageTrackingService instance.
+ */
+export declare function getUsageTrackingService(deps?: UsageTrackingServiceDeps): UsageTrackingService;
+//# sourceMappingURL=usage-tracking.service.d.ts.map

package/dist/model-budget/services/usage-tracking.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracking.service.d.ts","sourceRoot":"","sources":["../../../lib/model-budget/services/usage-tracking.service.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AAC7C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,2CAA2C,CAAC;AAWvE,OAAO,KAAK,EACV,mBAAmB,EACnB,SAAS,EACT,iBAAiB,EACjB,wBAAwB,EACzB,MAAM,4CAA4C,CAAC;AAOpD;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,aAAa,CAAqB;gBAE9B,IAAI,CAAC,EAAE,wBAAwB;IAU3C;;;;;;;;;OASG;IACG,kBAAkB,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAwCpE;;;;;;;;OAQG;IACG,WAAW,CACf,MAAM,EAAE,mBAAmB,EAC3B,KAAK,EAAE,KAAK,EACZ,KAAK,CAAC,EAAE,SAAS,GAAG,kBAAkB,GAAG,IAAI,GAC5C,OAAO,CAAC,iBAAiB,CAAC;IA2C7B;;;;;;;OAOG;IACH,iBAAiB,CACf,MAAM,EAAE,mBAAmB,EAC3B,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,OAAO,CAAC,SAAS,GAAG,kBAAkB,GAAG,SAAS,CAAC,GAChE,IAAI;IAiBP;;;;;;;;OAQG;IACG,YAAY,CAAC,CAAC,SAAS;QAAE,KAAK,CAAC,EAAE,SAAS,GAAG,kBAAkB,CAAA;KAAE,EACrE,MAAM,EAAE,mBAAmB,EAC3B,KAAK,EAAE,KAAK,EACZ,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAC1B,OAAO,CAAC,CAAC,CAAC;CAYd;AAQD;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,IAAI,CAAC,EAAE,wBAAwB,GAAG,oBAAoB,CAU7F"}