@salimassili/ai-costguard 1.1.3

package/dist/core/PolicyEngine.d.ts

@@ -0,0 +1,148 @@
+/**
+ * PolicyEngine.ts - Hierarchical Policy Engine
+ *
+ * Enterprise-grade policy system:
+ * - Tenant-level policies
+ * - Team-level policies
+ * - User-level policies
+ * - Workflow-level policies
+ *
+ * Rule types:
+ * - max cost per workflow
+ * - max tool-call depth
+ * - semantic repetition threshold
+ * - warn → throttle → block modes
+ */
+import { EventEmitter } from 'events';
+export type PolicyLevel = 'tenant' | 'team' | 'user' | 'workflow';
+export type EnforcementMode = 'monitor' | 'warn' | 'throttle' | 'block';
+export type PolicyAction = 'allow' | 'warn' | 'throttle' | 'block';
+export interface PolicyRule {
+    id: string;
+    name: string;
+    description: string;
+    enabled: boolean;
+    priority: number;
+    condition: PolicyCondition;
+    action: PolicyAction;
+    throttleParams?: {
+        delayMs: number;
+        rateLimitPerMinute?: number;
+    };
+}
+export interface PolicyCondition {
+    type: 'cost_threshold' | 'depth_limit' | 'repetition_score' | 'loop_detected' | 'token_limit' | 'time_window' | 'custom';
+    operator: 'gt' | 'gte' | 'lt' | 'lte' | 'eq' | 'contains' | 'regex';
+    threshold: number | string | boolean;
+    windowMs?: number;
+    and?: PolicyCondition[];
+    or?: PolicyCondition[];
+}
+export interface PolicySet {
+    id: string;
+    tenantId: string;
+    level: PolicyLevel;
+    targetId: string;
+    rules: PolicyRule[];
+    defaultAction: PolicyAction;
+    maxCostPerWorkflow: number;
+    maxDepth: number;
+    maxTokensPerRequest: number;
+    repetitionThreshold: number;
+    createdAt: number;
+    updatedAt: number;
+}
+export interface PolicyEvaluationContext {
+    tenantId: string;
+    teamId?: string;
+    userId?: string;
+    workflowId?: string;
+    sessionId: string;
+    requestId: string;
+    currentCost: number;
+    currentDepth: number;
+    currentTokens: number;
+    repetitionScore: number;
+    loopDetected: boolean;
+    actionType: string;
+    timestamp: number;
+}
+export interface PolicyEvaluationResult {
+    action: PolicyAction;
+    reason: string;
+    confidence: number;
+    rulesTriggered: PolicyRule[];
+    policyChain: Array<{
+        level: PolicyLevel;
+        policyId: string;
+        action: PolicyAction;
+    }>;
+    throttleParams?: {
+        delayMs: number;
+        rateLimitPerMinute?: number;
+    };
+}
+export interface PolicyViolation {
+    rule: PolicyRule;
+    context: PolicyEvaluationContext;
+    actualValue: any;
+    expectedValue: any;
+    timestamp: number;
+}
+/**
+ * Policy Engine
+ *
+ * Evaluates hierarchical policies in order:
+ * 1. Tenant-level (most restrictive applies)
+ * 2. Team-level
+ * 3. User-level
+ * 4. Workflow-level (most specific)
+ *
+ * Action precedence: block > throttle > warn > allow
+ */
+export declare class PolicyEngine extends EventEmitter {
+    private policies;
+    private tenantPolicies;
+    private violationHistory;
+    private maxViolationsPerTenant;
+    constructor();
+    /**
+     * Register a policy set
+     */
+    registerPolicy(policySet: PolicySet): void;
+    /**
+     * Evaluate policies for a request
+     * Returns the most restrictive action across all policy levels
+     */
+    evaluate(context: PolicyEvaluationContext): PolicyEvaluationResult;
+    /**
+     * Evaluate a single condition
+     */
+    private evaluateCondition;
+    /**
+     * Get policies relevant to a context
+     */
+    private getPoliciesForContext;
+    /**
+     * Record a policy violation
+     */
+    private recordViolation;
+    /**
+     * Get actual value for a condition from context
+     */
+    private getActualValue;
+    /**
+     * Action precedence for determining most restrictive
+     */
+    private actionPrecedence;
+    /**
+     * Get violations for a tenant
+     */
+    getViolations(tenantId: string, limit?: number): PolicyViolation[];
+    /**
+     * Create default tenant policy
+     */
+    createDefaultPolicy(tenantId: string): PolicySet;
+}
+export declare const policyEngine: PolicyEngine;
+//# sourceMappingURL=PolicyEngine.d.ts.map

package/dist/core/PolicyEngine.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PolicyEngine.d.ts","sourceRoot":"","sources":["../../src/core/PolicyEngine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,GAAG,UAAU,CAAC;AAClE,MAAM,MAAM,eAAe,GAAG,SAAS,GAAG,MAAM,GAAG,UAAU,GAAG,OAAO,CAAC;AACxE,MAAM,MAAM,YAAY,GAAG,OAAO,GAAG,MAAM,GAAG,UAAU,GAAG,OAAO,CAAC;AAEnE,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,eAAe,CAAC;IAC3B,MAAM,EAAE,YAAY,CAAC;IACrB,cAAc,CAAC,EAAE;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,gBAAgB,GAAG,aAAa,GAAG,kBAAkB,GAAG,eAAe,GACvE,aAAa,GAAG,aAAa,GAAG,QAAQ,CAAC;IAC/C,QAAQ,EAAE,IAAI,GAAG,KAAK,GAAG,IAAI,GAAG,KAAK,GAAG,IAAI,GAAG,UAAU,GAAG,OAAO,CAAC;IACpE,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;IAErC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,GAAG,CAAC,EAAE,eAAe,EAAE,CAAC;IACxB,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,WAAW,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,UAAU,EAAE,CAAC;IACpB,aAAa,EAAE,YAAY,CAAC;IAC5B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,YAAY,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,sBAAsB;IACrC,MAAM,EAAE,YAAY,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,UAAU,EAAE,CAAC;IAC7B,WAAW,EAAE,KAAK,CAAC;QACjB,KAAK,EAAE,WAAW,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,YAAY,CAAC;KACtB,CAAC,CAAC;IACH,cAAc,CAAC,EAAE;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,UAAU,CAAC;IACjB,OAAO,EAAE,uBAAuB,CAAC;IACjC,WAAW,EAAE,GAAG,CAAC;IACjB,aAAa,EAAE,GAAG,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;;;;;;;;GAUG;AACH,qBAAa,YAAa,SAAQ,YAAY;IAC5C,OAAO,CAAC,QAAQ,CAAyB;IACzC,OAAO,CAAC,cAAc,CAA2B;IACjD,OAAO,CAAC,gBAAgB,CAAiC;IACzD,OAAO,CAAC,sBAAsB,CAAgB;;IAS9C;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IAqB1C;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,uBAAuB,GAAG,sBAAsB;IA2ElE;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAkDzB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA+B7B;;OAEG;IACH,OAAO,CAAC,eAAe;IAoBvB;;OAEG;IACH,OAAO,CAAC,cAAc;IAiBtB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAUxB;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,GAAE,MAAY,GAAG,eAAe,EAAE;IAKvE;;OAEG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS;CAqEjD;AAGD,eAAO,MAAM,YAAY,cAAqB,CAAC"}

package/dist/core/PolicyEngine.js

@@ -0,0 +1,336 @@
+"use strict";
+/**
+ * PolicyEngine.ts - Hierarchical Policy Engine
+ *
+ * Enterprise-grade policy system:
+ * - Tenant-level policies
+ * - Team-level policies
+ * - User-level policies
+ * - Workflow-level policies
+ *
+ * Rule types:
+ * - max cost per workflow
+ * - max tool-call depth
+ * - semantic repetition threshold
+ * - warn → throttle → block modes
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.policyEngine = exports.PolicyEngine = void 0;
+const events_1 = require("events");
+/**
+ * Policy Engine
+ *
+ * Evaluates hierarchical policies in order:
+ * 1. Tenant-level (most restrictive applies)
+ * 2. Team-level
+ * 3. User-level
+ * 4. Workflow-level (most specific)
+ *
+ * Action precedence: block > throttle > warn > allow
+ */
+class PolicyEngine extends events_1.EventEmitter {
+    constructor() {
+        super();
+        this.maxViolationsPerTenant = 1000;
+        this.policies = new Map();
+        this.tenantPolicies = new Map();
+        this.violationHistory = new Map();
+    }
+    /**
+     * Register a policy set
+     */
+    registerPolicy(policySet) {
+        const key = `${policySet.tenantId}:${policySet.level}:${policySet.targetId}`;
+        this.policies.set(key, policySet);
+        // Update tenant policies list
+        const tenantPols = this.tenantPolicies.get(policySet.tenantId) || [];
+        const existingIndex = tenantPols.findIndex(p => p.level === policySet.level && p.targetId === policySet.targetId);
+        if (existingIndex >= 0) {
+            tenantPols[existingIndex] = policySet;
+        }
+        else {
+            tenantPols.push(policySet);
+        }
+        this.tenantPolicies.set(policySet.tenantId, tenantPols);
+        this.emit('policy:registered', policySet);
+    }
+    /**
+     * Evaluate policies for a request
+     * Returns the most restrictive action across all policy levels
+     */
+    evaluate(context) {
+        const policyChain = [];
+        const rulesTriggered = [];
+        let finalAction = 'allow';
+        let finalReason = 'No policies violated';
+        let finalThrottleParams;
+        // Get policies for all levels
+        const policies = this.getPoliciesForContext(context);
+        // Evaluate from tenant (least specific) to workflow (most specific)
+        const levels = ['tenant', 'team', 'user', 'workflow'];
+        for (const level of levels) {
+            const levelPolicies = policies.filter(p => p.level === level);
+            for (const policy of levelPolicies) {
+                // Evaluate each rule in priority order
+                const sortedRules = [...policy.rules].sort((a, b) => b.priority - a.priority);
+                for (const rule of sortedRules) {
+                    if (!rule.enabled)
+                        continue;
+                    const triggered = this.evaluateCondition(rule.condition, context);
+                    if (triggered) {
+                        rulesTriggered.push(rule);
+                        policyChain.push({
+                            level,
+                            policyId: policy.id,
+                            action: rule.action,
+                        });
+                        // Record violation
+                        this.recordViolation(rule, context);
+                        // Apply most restrictive action
+                        if (this.actionPrecedence(rule.action) > this.actionPrecedence(finalAction)) {
+                            finalAction = rule.action;
+                            finalReason = `Policy rule "${rule.name}" triggered at ${level} level`;
+                            finalThrottleParams = rule.throttleParams;
+                        }
+                        // If block, no need to continue evaluating
+                        if (finalAction === 'block') {
+                            break;
+                        }
+                    }
+                }
+                if (finalAction === 'block')
+                    break;
+            }
+            if (finalAction === 'block')
+                break;
+        }
+        // Apply default tenant policy if no rules triggered
+        if (finalAction === 'allow' && policies.length > 0) {
+            const tenantPolicy = policies.find(p => p.level === 'tenant');
+            if (tenantPolicy && tenantPolicy.defaultAction !== 'allow') {
+                finalAction = tenantPolicy.defaultAction;
+                finalReason = 'Default tenant policy applied';
+            }
+        }
+        return {
+            action: finalAction,
+            reason: finalReason,
+            confidence: rulesTriggered.length > 0 ? 0.9 : 1.0,
+            rulesTriggered,
+            policyChain,
+            throttleParams: finalThrottleParams,
+        };
+    }
+    /**
+     * Evaluate a single condition
+     */
+    evaluateCondition(condition, context) {
+        // Handle composite conditions
+        if (condition.and) {
+            return condition.and.every(c => this.evaluateCondition(c, context));
+        }
+        if (condition.or) {
+            return condition.or.some(c => this.evaluateCondition(c, context));
+        }
+        // Get the value to compare based on condition type
+        let value;
+        switch (condition.type) {
+            case 'cost_threshold':
+                value = context.currentCost;
+                break;
+            case 'depth_limit':
+                value = context.currentDepth;
+                break;
+            case 'token_limit':
+                value = context.currentTokens;
+                break;
+            case 'repetition_score':
+                value = context.repetitionScore;
+                break;
+            case 'loop_detected':
+                value = context.loopDetected;
+                break;
+            default:
+                value = null;
+        }
+        // Compare based on operator
+        switch (condition.operator) {
+            case 'gt':
+                return value > condition.threshold;
+            case 'gte':
+                return value >= condition.threshold;
+            case 'lt':
+                return value < condition.threshold;
+            case 'lte':
+                return value <= condition.threshold;
+            case 'eq':
+                return value === condition.threshold;
+            case 'contains':
+                return String(value).includes(String(condition.threshold));
+            default:
+                return false;
+        }
+    }
+    /**
+     * Get policies relevant to a context
+     */
+    getPoliciesForContext(context) {
+        const policies = [];
+        const tenantPols = this.tenantPolicies.get(context.tenantId) || [];
+        for (const policy of tenantPols) {
+            // Check if policy applies to this context
+            let applies = false;
+            switch (policy.level) {
+                case 'tenant':
+                    applies = true;
+                    break;
+                case 'team':
+                    applies = context.teamId === policy.targetId;
+                    break;
+                case 'user':
+                    applies = context.userId === policy.targetId;
+                    break;
+                case 'workflow':
+                    applies = context.workflowId === policy.targetId;
+                    break;
+            }
+            if (applies) {
+                policies.push(policy);
+            }
+        }
+        return policies;
+    }
+    /**
+     * Record a policy violation
+     */
+    recordViolation(rule, context) {
+        const violations = this.violationHistory.get(context.tenantId) || [];
+        violations.push({
+            rule,
+            context,
+            actualValue: this.getActualValue(rule.condition, context),
+            expectedValue: rule.condition.threshold,
+            timestamp: Date.now(),
+        });
+        // Keep only recent violations
+        if (violations.length > this.maxViolationsPerTenant) {
+            violations.splice(0, violations.length - this.maxViolationsPerTenant);
+        }
+        this.violationHistory.set(context.tenantId, violations);
+        this.emit('policy:violation', { rule, context });
+    }
+    /**
+     * Get actual value for a condition from context
+     */
+    getActualValue(condition, context) {
+        switch (condition.type) {
+            case 'cost_threshold':
+                return context.currentCost;
+            case 'depth_limit':
+                return context.currentDepth;
+            case 'token_limit':
+                return context.currentTokens;
+            case 'repetition_score':
+                return context.repetitionScore;
+            case 'loop_detected':
+                return context.loopDetected;
+            default:
+                return null;
+        }
+    }
+    /**
+     * Action precedence for determining most restrictive
+     */
+    actionPrecedence(action) {
+        const precedence = {
+            allow: 0,
+            warn: 1,
+            throttle: 2,
+            block: 3,
+        };
+        return precedence[action];
+    }
+    /**
+     * Get violations for a tenant
+     */
+    getViolations(tenantId, limit = 100) {
+        const violations = this.violationHistory.get(tenantId) || [];
+        return violations.slice(-limit);
+    }
+    /**
+     * Create default tenant policy
+     */
+    createDefaultPolicy(tenantId) {
+        return {
+            id: `default-${tenantId}`,
+            tenantId,
+            level: 'tenant',
+            targetId: tenantId,
+            rules: [
+                {
+                    id: 'max-cost',
+                    name: 'Maximum Cost Per Workflow',
+                    description: 'Block workflows exceeding $10',
+                    enabled: true,
+                    priority: 100,
+                    condition: {
+                        type: 'cost_threshold',
+                        operator: 'gt',
+                        threshold: 10,
+                    },
+                    action: 'block',
+                },
+                {
+                    id: 'max-depth',
+                    name: 'Maximum Tool Call Depth',
+                    description: 'Block workflows exceeding 10 levels deep',
+                    enabled: true,
+                    priority: 90,
+                    condition: {
+                        type: 'depth_limit',
+                        operator: 'gt',
+                        threshold: 10,
+                    },
+                    action: 'block',
+                },
+                {
+                    id: 'loop-detection',
+                    name: 'Loop Detection',
+                    description: 'Block when loops are detected',
+                    enabled: true,
+                    priority: 95,
+                    condition: {
+                        type: 'loop_detected',
+                        operator: 'eq',
+                        threshold: true,
+                    },
+                    action: 'block',
+                },
+                {
+                    id: 'high-cost-warning',
+                    name: 'High Cost Warning',
+                    description: 'Warn when cost exceeds $5',
+                    enabled: true,
+                    priority: 50,
+                    condition: {
+                        type: 'cost_threshold',
+                        operator: 'gt',
+                        threshold: 5,
+                    },
+                    action: 'warn',
+                },
+            ],
+            defaultAction: 'allow',
+            maxCostPerWorkflow: 10,
+            maxDepth: 10,
+            maxTokensPerRequest: 100000,
+            repetitionThreshold: 0.8,
+            createdAt: Date.now(),
+            updatedAt: Date.now(),
+        };
+    }
+}
+exports.PolicyEngine = PolicyEngine;
+// Export singleton
+exports.policyEngine = new PolicyEngine();
+//# sourceMappingURL=PolicyEngine.js.map

package/dist/core/PolicyEngine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PolicyEngine.js","sourceRoot":"","sources":["../../src/core/PolicyEngine.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;AAEH,mCAAsC;AAuFtC;;;;;;;;;;GAUG;AACH,MAAa,YAAa,SAAQ,qBAAY;IAM5C;QACE,KAAK,EAAE,CAAC;QAHF,2BAAsB,GAAW,IAAI,CAAC;QAI5C,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAE,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,SAAoB;QACjC,MAAM,GAAG,GAAG,GAAG,SAAS,CAAC,QAAQ,IAAI,SAAS,CAAC,KAAK,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7E,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAElC,8BAA8B;QAC9B,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACrE,MAAM,aAAa,GAAG,UAAU,CAAC,SAAS,CACxC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,KAAK,IAAI,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC,QAAQ,CACtE,CAAC;QAEF,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;YACvB,UAAU,CAAC,aAAa,CAAC,GAAG,SAAS,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,OAAgC;QACvC,MAAM,WAAW,GAA0E,EAAE,CAAC;QAC9F,MAAM,cAAc,GAAiB,EAAE,CAAC;QACxC,IAAI,WAAW,GAAiB,OAAO,CAAC;QACxC,IAAI,WAAW,GAAG,sBAAsB,CAAC;QACzC,IAAI,mBAAiF,CAAC;QAEtF,8BAA8B;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAErD,oEAAoE;QACpE,MAAM,MAAM,GAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QAErE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;YAE9D,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;gBACnC,uCAAuC;gBACvC,MAAM,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAE9E,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;oBAC/B,IAAI,CAAC,IAAI,CAAC,OAAO;wBAAE,SAAS;oBAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;oBAElE,IAAI,SAAS,EAAE,CAAC;wBACd,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC1B,WAAW,CAAC,IAAI,CAAC;4BACf,KAAK;4BACL,QAAQ,EAAE,MAAM,CAAC,EAAE;4BACnB,MAAM,EAAE,IAAI,CAAC,MAAM;yBACpB,CAAC,CAAC;wBAEH,mBAAmB;wBACnB,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;wBAEpC,gCAAgC;wBAChC,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;4BAC5E,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC;4BAC1B,WAAW,GAAG,gBAAgB,IAAI,CAAC,IAAI,kBAAkB,KAAK,QAAQ,CAAC;4BACvE,mBAAmB,GAAG,IAAI,CAAC,cAAc,CAAC;wBAC5C,CAAC;wBAED,2CAA2C;wBAC3C,IAAI,WAAW,KAAK,OAAO,EAAE,CAAC;4BAC5B,MAAM;wBACR,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,IAAI,WAAW,KAAK,OAAO;oBAAE,MAAM;YACrC,CAAC;YAED,IAAI,WAAW,KAAK,OAAO;gBAAE,MAAM;QACrC,CAAC;QAED,oDAAoD;QACpD,IAAI,WAAW,KAAK,OAAO,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC;YAC9D,IAAI,YAAY,IAAI,YAAY,CAAC,aAAa,KAAK,OAAO,EAAE,CAAC;gBAC3D,WAAW,GAAG,YAAY,CAAC,aAAa,CAAC;gBACzC,WAAW,GAAG,+BAA+B,CAAC;YAChD,CAAC;QACH,CAAC;QAED,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,WAAW;YACnB,UAAU,EAAE,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG;YACjD,cAAc;YACd,WAAW;YACX,cAAc,EAAE,mBAAmB;SACpC,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,SAA0B,EAAE,OAAgC;QACpF,8BAA8B;QAC9B,IAAI,SAAS,CAAC,GAAG,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,SAAS,CAAC,EAAE,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;QACpE,CAAC;QAED,mDAAmD;QACnD,IAAI,KAAU,CAAC;QACf,QAAQ,SAAS,CAAC,IAAI,EAAE,CAAC;YACvB,KAAK,gBAAgB;gBACnB,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC;gBAC5B,MAAM;YACR,KAAK,aAAa;gBAChB,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC;gBAC7B,MAAM;YACR,KAAK,aAAa;gBAChB,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC;gBAC9B,MAAM;YACR,KAAK,kBAAkB;gBACrB,KAAK,GAAG,OAAO,CAAC,eAAe,CAAC;gBAChC,MAAM;YACR,KAAK,eAAe;gBAClB,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC;gBAC7B,MAAM;YACR;gBACE,KAAK,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,4BAA4B;QAC5B,QAAQ,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC3B,KAAK,IAAI;gBACP,OAAO,KAAK,GAAG,SAAS,CAAC,SAAS,CAAC;YACrC,KAAK,KAAK;gBACR,OAAO,KAAK,IAAI,SAAS,CAAC,SAAS,CAAC;YACtC,KAAK,IAAI;gBACP,OAAO,KAAK,GAAG,SAAS,CAAC,SAAS,CAAC;YACrC,KAAK,KAAK;gBACR,OAAO,KAAK,IAAI,SAAS,CAAC,SAAS,CAAC;YACtC,KAAK,IAAI;gBACP,OAAO,KAAK,KAAK,SAAS,CAAC,SAAS,CAAC;YACvC,KAAK,UAAU;gBACb,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7D;gBACE,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,qBAAqB,CAAC,OAAgC;QAC5D,MAAM,QAAQ,GAAgB,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAEnE,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;YAChC,0CAA0C;YAC1C,IAAI,OAAO,GAAG,KAAK,CAAC;YAEpB,QAAQ,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrB,KAAK,QAAQ;oBACX,OAAO,GAAG,IAAI,CAAC;oBACf,MAAM;gBACR,KAAK,MAAM;oBACT,OAAO,GAAG,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,QAAQ,CAAC;oBAC7C,MAAM;gBACR,KAAK,MAAM;oBACT,OAAO,GAAG,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,QAAQ,CAAC;oBAC7C,MAAM;gBACR,KAAK,UAAU;oBACb,OAAO,GAAG,OAAO,CAAC,UAAU,KAAK,MAAM,CAAC,QAAQ,CAAC;oBACjD,MAAM;YACV,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACZ,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,IAAgB,EAAE,OAAgC;QACxE,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAErE,UAAU,CAAC,IAAI,CAAC;YACd,IAAI;YACJ,OAAO;YACP,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC;YACzD,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS;YACvC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,8BAA8B;QAC9B,IAAI,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpD,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,SAA0B,EAAE,OAAgC;QACjF,QAAQ,SAAS,CAAC,IAAI,EAAE,CAAC;YACvB,KAAK,gBAAgB;gBACnB,OAAO,OAAO,CAAC,WAAW,CAAC;YAC7B,KAAK,aAAa;gBAChB,OAAO,OAAO,CAAC,YAAY,CAAC;YAC9B,KAAK,aAAa;gBAChB,OAAO,OAAO,CAAC,aAAa,CAAC;YAC/B,KAAK,kBAAkB;gBACrB,OAAO,OAAO,CAAC,eAAe,CAAC;YACjC,KAAK,eAAe;gBAClB,OAAO,OAAO,CAAC,YAAY,CAAC;YAC9B;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,MAAoB;QAC3C,MAAM,UAAU,GAAiC;YAC/C,KAAK,EAAE,CAAC;YACR,IAAI,EAAE,CAAC;YACP,QAAQ,EAAE,CAAC;YACX,KAAK,EAAE,CAAC;SACT,CAAC;QACF,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB,EAAE,QAAgB,GAAG;QACjD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC7D,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,QAAgB;QAClC,OAAO;YACL,EAAE,EAAE,WAAW,QAAQ,EAAE;YACzB,QAAQ;YACR,KAAK,EAAE,QAAQ;YACf,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE;gBACL;oBACE,EAAE,EAAE,UAAU;oBACd,IAAI,EAAE,2BAA2B;oBACjC,WAAW,EAAE,+BAA+B;oBAC5C,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,GAAG;oBACb,SAAS,EAAE;wBACT,IAAI,EAAE,gBAAgB;wBACtB,QAAQ,EAAE,IAAI;wBACd,SAAS,EAAE,EAAE;qBACd;oBACD,MAAM,EAAE,OAAO;iBAChB;gBACD;oBACE,EAAE,EAAE,WAAW;oBACf,IAAI,EAAE,yBAAyB;oBAC/B,WAAW,EAAE,0CAA0C;oBACvD,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,EAAE;oBACZ,SAAS,EAAE;wBACT,IAAI,EAAE,aAAa;wBACnB,QAAQ,EAAE,IAAI;wBACd,SAAS,EAAE,EAAE;qBACd;oBACD,MAAM,EAAE,OAAO;iBAChB;gBACD;oBACE,EAAE,EAAE,gBAAgB;oBACpB,IAAI,EAAE,gBAAgB;oBACtB,WAAW,EAAE,+BAA+B;oBAC5C,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,EAAE;oBACZ,SAAS,EAAE;wBACT,IAAI,EAAE,eAAe;wBACrB,QAAQ,EAAE,IAAI;wBACd,SAAS,EAAE,IAAI;qBAChB;oBACD,MAAM,EAAE,OAAO;iBAChB;gBACD;oBACE,EAAE,EAAE,mBAAmB;oBACvB,IAAI,EAAE,mBAAmB;oBACzB,WAAW,EAAE,2BAA2B;oBACxC,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,EAAE;oBACZ,SAAS,EAAE;wBACT,IAAI,EAAE,gBAAgB;wBACtB,QAAQ,EAAE,IAAI;wBACd,SAAS,EAAE,CAAC;qBACb;oBACD,MAAM,EAAE,MAAM;iBACf;aACF;YACD,aAAa,EAAE,OAAO;YACtB,kBAAkB,EAAE,EAAE;YACtB,QAAQ,EAAE,EAAE;YACZ,mBAAmB,EAAE,MAAM;YAC3B,mBAAmB,EAAE,GAAG;YACxB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;CACF;AAnVD,oCAmVC;AAED,mBAAmB;AACN,QAAA,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/core/PricingConfig.d.ts

@@ -0,0 +1,55 @@
+/**
+ * PricingConfig.ts - Real API Pricing Configuration
+ *
+ * Maps token counts to actual costs for supported models
+ * Used for real cost extraction from API responses
+ */
+export interface ModelPricing {
+    inputPer1K: number;
+    outputPer1K: number;
+    provider: 'openai' | 'anthropic' | 'google' | 'cohere';
+}
+export interface TokenUsage {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+}
+export interface CostCalculation {
+    inputCost: number;
+    outputCost: number;
+    totalCost: number;
+    model: string;
+    pricing: ModelPricing;
+}
+/**
+ * Pricing Configuration - Singleton
+ * Stores real pricing data for AI providers
+ */
+export declare class PricingConfig {
+    private static instance;
+    private pricing;
+    private constructor();
+    static getInstance(): PricingConfig;
+    /**
+     * Get pricing for a model
+     */
+    getPricing(model: string): ModelPricing;
+    /**
+     * Calculate real cost from token usage
+     */
+    calculateCost(model: string, usage: TokenUsage): CostCalculation;
+    /**
+     * Extract token usage from various API response formats
+     */
+    extractUsage(response: any): TokenUsage | null;
+    /**
+     * Update pricing for a model (for custom models)
+     */
+    updatePricing(model: string, pricing: ModelPricing): void;
+    /**
+     * Get all supported models
+     */
+    getSupportedModels(): string[];
+}
+export declare const pricingConfig: PricingConfig;
+//# sourceMappingURL=PricingConfig.d.ts.map

package/dist/core/PricingConfig.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PricingConfig.d.ts","sourceRoot":"","sources":["../../src/core/PricingConfig.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,QAAQ,CAAC;CACxD;AAED,MAAM,WAAW,UAAU;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,YAAY,CAAC;CACvB;AAED;;;GAGG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAgB;IACvC,OAAO,CAAC,OAAO,CAA+B;IAE9C,OAAO;IAoCP,MAAM,CAAC,WAAW,IAAI,aAAa;IAOnC;;OAEG;IACH,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY;IAevC;;OAEG;IACH,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,eAAe;IAgBhE;;OAEG;IACH,YAAY,CAAC,QAAQ,EAAE,GAAG,GAAG,UAAU,GAAG,IAAI;IAmC9C;;OAEG;IACH,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,GAAG,IAAI;IAIzD;;OAEG;IACH,kBAAkB,IAAI,MAAM,EAAE;CAG/B;AAGD,eAAO,MAAM,aAAa,eAA8B,CAAC"}