@zhin.js/agent 0.2.1 → 0.3.0

package/lib/security/budget-limiter.d.ts

@@ -1,161 +0,0 @@
-/**
- * Token/Cost 预算限制系统
- *
- * 控制 AI Agent 的资源使用，防止：
- * - Token 滥用
- * - 成本超支
- * - 资源耗尽
- *
- */
-export interface BudgetConfig {
-    /** 是否启用预算限制 */
-    enabled: boolean;
-    /** 每会话最大 Token 数 */
-    maxTokensPerSession?: number;
-    /** 每用户每天最大 Token 数 */
-    maxTokensPerUserPerDay?: number;
-    /** 每会话最大成本（美元） */
-    maxCostPerSession?: number;
-    /** 每用户每天最大成本（美元） */
-    maxCostPerUserPerDay?: number;
-    /** 每会话最大工具调用次数 */
-    maxToolCallsPerSession?: number;
-    /** 每会话最大迭代次数 */
-    maxIterationsPerSession?: number;
-    /** 每会话最大持续时间（毫秒） */
-    maxSessionDuration?: number;
-    /** 警告阈值（百分比） */
-    warningThreshold?: number;
-    /** 是否在达到限制时自动结束会话 */
-    autoTerminate?: boolean;
-}
-export interface TokenUsage {
-    /** 输入 Token 数 */
-    inputTokens: number;
-    /** 输出 Token 数 */
-    outputTokens: number;
-    /** 总 Token 数 */
-    totalTokens: number;
-    /** 估算成本（美元） */
-    estimatedCost: number;
-}
-export interface SessionBudget {
-    /** 会话 ID */
-    sessionId: string;
-    /** 用户 ID */
-    userId?: string;
-    /** 开始时间 */
-    startTime: number;
-    /** Token 使用情况 */
-    tokenUsage: TokenUsage;
-    /** 工具调用次数 */
-    toolCallCount: number;
-    /** 迭代次数 */
-    iterationCount: number;
-    /** 是否已达到限制 */
-    limitReached: boolean;
-    /** 达到的限制类型 */
-    limitType?: string;
-    /** 警告已发送 */
-    warningsSent: Set<string>;
-}
-export interface BudgetCheckResult {
-    allowed: boolean;
-    reason?: string;
-    /** 当前使用情况 */
-    current?: {
-        tokens: number;
-        cost: number;
-        toolCalls: number;
-        iterations: number;
-        duration: number;
-    };
-    /** 限制情况 */
-    limits?: {
-        maxTokens?: number;
-        maxCost?: number;
-        maxToolCalls?: number;
-        maxIterations?: number;
-        maxDuration?: number;
-    };
-    /** 是否为警告 */
-    isWarning?: boolean;
-}
-export declare class BudgetLimiter {
-    private config;
-    private sessions;
-    private dailyUsage;
-    private cleanupTimer?;
-    private static readonly MAX_DAILY_ENTRIES;
-    constructor(config?: Partial<BudgetConfig>);
-    /**
-     * 创建新会话预算
-     */
-    createSession(sessionId: string, userId?: string): SessionBudget;
-    /**
-     * 获取会话预算
-     */
-    getSession(sessionId: string): SessionBudget | undefined;
-    /**
-     * 记录 Token 使用
-     */
-    recordTokenUsage(sessionId: string, usage: TokenUsage): BudgetCheckResult;
-    /**
-     * 记录工具调用
-     */
-    recordToolCall(sessionId: string): BudgetCheckResult;
-    /**
-     * 记录迭代
-     */
-    recordIteration(sessionId: string): BudgetCheckResult;
-    /**
-     * 检查预算限制
-     */
-    private checkBudget;
-    /**
-     * 更新每日使用情况
-     */
-    private updateDailyUsage;
-    /**
-     * 清理过期会话和每日记录
-     */
-    private cleanup;
-    /**
-     * 结束会话
-     */
-    endSession(sessionId: string): SessionBudget | undefined;
-    /**
-     * 获取会话统计
-     */
-    getSessionStats(sessionId: string): {
-        tokenUsage: TokenUsage;
-        toolCallCount: number;
-        iterationCount: number;
-        duration: number;
-        limits: Record<string, number | undefined>;
-    } | undefined;
-    /**
-     * 更新配置
-     */
-    updateConfig(config: Partial<BudgetConfig>): void;
-    /**
-     * 获取配置
-     */
-    getConfig(): BudgetConfig;
-    dispose(): void;
-}
-/**
- * 获取全局预算限制器实例
- */
-export declare function getBudgetLimiter(): BudgetLimiter;
-/**
- * 初始化预算限制器
- */
-export declare function initBudgetLimiter(config: Partial<BudgetConfig>): BudgetLimiter;
-/** 重置全局预算限制器（用于测试隔离） */
-export declare function resetBudgetLimiter(): void;
-/**
- * 检查预算限制
- */
-export declare function checkBudgetLimit(sessionId: string, type: 'tokens' | 'toolCall' | 'iteration', usage?: TokenUsage): BudgetCheckResult;
-//# sourceMappingURL=budget-limiter.d.ts.map

package/lib/security/budget-limiter.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"budget-limiter.d.ts","sourceRoot":"","sources":["../../src/security/budget-limiter.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAMH,MAAM,WAAW,YAAY;IAC3B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,oBAAoB;IACpB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,sBAAsB;IACtB,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,kBAAkB;IAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,oBAAoB;IACpB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,kBAAkB;IAClB,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,gBAAgB;IAChB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,oBAAoB;IACpB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,gBAAgB;IAChB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,qBAAqB;IACrB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAiBD,MAAM,WAAW,UAAU;IACzB,iBAAiB;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe;IACf,aAAa,EAAE,MAAM,CAAC;CACvB;AAID,MAAM,WAAW,aAAa;IAC5B,YAAY;IACZ,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,iBAAiB;IACjB,UAAU,EAAE,UAAU,CAAC;IACvB,aAAa;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW;IACX,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc;IACd,YAAY,EAAE,OAAO,CAAC;IACtB,cAAc;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY;IACZ,YAAY,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;CAC3B;AAID,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,aAAa;IACb,OAAO,CAAC,EAAE;QACR,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,WAAW;IACX,MAAM,CAAC,EAAE;QACP,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,YAAY;IACZ,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAID,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,QAAQ,CAAyC;IACzD,OAAO,CAAC,UAAU,CAAsC;IACxD,OAAO,CAAC,YAAY,CAAC,CAAiC;IACtD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAS;gBAEtC,MAAM,GAAE,OAAO,CAAC,YAAY,CAAM;IAU9C;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,aAAa;IAgBhE;;OAEG;IACH,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAIxD;;OAEG;IACH,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,iBAAiB;IAyBzE;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,iBAAiB;IAcpD;;OAEG;IACH,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,iBAAiB;IAcrD;;OAEG;IACH,OAAO,CAAC,WAAW;IAiJnB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAgBxB;;OAEG;IACH,OAAO,CAAC,OAAO;IA+Bf;;OAEG;IACH,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAQxD;;OAEG;IACH,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG;QAClC,UAAU,EAAE,UAAU,CAAC;QACvB,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;KAC5C,GAAG,SAAS;IAmBb;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,IAAI;IAIjD;;OAEG;IACH,SAAS,IAAI,YAAY;IAIzB,OAAO,IAAI,IAAI;CAQhB;AAMD;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,aAAa,CAYhD;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,aAAa,CAG9E;AAED,wBAAwB;AACxB,wBAAgB,kBAAkB,IAAI,IAAI,CAEzC;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,GAAG,UAAU,GAAG,WAAW,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,iBAAiB,CAcpI"}

package/lib/security/budget-limiter.js

@@ -1,392 +0,0 @@
-/**
- * Token/Cost 预算限制系统
- *
- * 控制 AI Agent 的资源使用，防止：
- * - Token 滥用
- * - 成本超支
- * - 资源耗尽
- *
- */
-import { getPlugin } from '@zhin.js/core';
-const DEFAULT_CONFIG = {
-    enabled: true,
-    maxTokensPerSession: 1000000, // 1M tokens
-    maxTokensPerUserPerDay: 5000000, // 5M tokens
-    maxCostPerSession: 10.0, // $10
-    maxCostPerUserPerDay: 50.0, // $50
-    maxToolCallsPerSession: 100,
-    maxIterationsPerSession: 20,
-    maxSessionDuration: 3600000, // 1 小时
-    warningThreshold: 80, // 80%
-    autoTerminate: false,
-};
-// ── 预算限制器类 ──────────────────────────────────────────────────────
-export class BudgetLimiter {
-    config;
-    sessions = new Map();
-    dailyUsage = new Map(); // userId -> usage
-    cleanupTimer;
-    static MAX_DAILY_ENTRIES = 10000;
-    constructor(config = {}) {
-        this.config = { ...DEFAULT_CONFIG, ...config };
-        // 定期清理过期会话
-        this.cleanupTimer = setInterval(() => this.cleanup(), 300000);
-        if (this.cleanupTimer && typeof this.cleanupTimer === 'object' && 'unref' in this.cleanupTimer) {
-            this.cleanupTimer.unref();
-        }
-    }
-    /**
-     * 创建新会话预算
-     */
-    createSession(sessionId, userId) {
-        const budget = {
-            sessionId,
-            userId,
-            startTime: Date.now(),
-            tokenUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0, estimatedCost: 0 },
-            toolCallCount: 0,
-            iterationCount: 0,
-            limitReached: false,
-            warningsSent: new Set(),
-        };
-        this.sessions.set(sessionId, budget);
-        return budget;
-    }
-    /**
-     * 获取会话预算
-     */
-    getSession(sessionId) {
-        return this.sessions.get(sessionId);
-    }
-    /**
-     * 记录 Token 使用
-     */
-    recordTokenUsage(sessionId, usage) {
-        if (!this.config.enabled) {
-            return { allowed: true };
-        }
-        const session = this.sessions.get(sessionId);
-        if (!session) {
-            return { allowed: true };
-        }
-        // 更新会话使用情况
-        session.tokenUsage.inputTokens += usage.inputTokens;
-        session.tokenUsage.outputTokens += usage.outputTokens;
-        session.tokenUsage.totalTokens += usage.totalTokens;
-        session.tokenUsage.estimatedCost += usage.estimatedCost;
-        // 更新每日使用情况
-        if (session.userId) {
-            this.updateDailyUsage(session.userId, usage);
-        }
-        // 检查限制
-        return this.checkBudget(session);
-    }
-    /**
-     * 记录工具调用
-     */
-    recordToolCall(sessionId) {
-        if (!this.config.enabled) {
-            return { allowed: true };
-        }
-        const session = this.sessions.get(sessionId);
-        if (!session) {
-            return { allowed: true };
-        }
-        session.toolCallCount++;
-        return this.checkBudget(session);
-    }
-    /**
-     * 记录迭代
-     */
-    recordIteration(sessionId) {
-        if (!this.config.enabled) {
-            return { allowed: true };
-        }
-        const session = this.sessions.get(sessionId);
-        if (!session) {
-            return { allowed: true };
-        }
-        session.iterationCount++;
-        return this.checkBudget(session);
-    }
-    /**
-     * 检查预算限制
-     */
-    checkBudget(session) {
-        const now = Date.now();
-        const duration = now - session.startTime;
-        // 构建当前使用情况
-        const current = {
-            tokens: session.tokenUsage.totalTokens,
-            cost: session.tokenUsage.estimatedCost,
-            toolCalls: session.toolCallCount,
-            iterations: session.iterationCount,
-            duration,
-        };
-        // 构建限制情况
-        const limits = {
-            maxTokens: this.config.maxTokensPerSession,
-            maxCost: this.config.maxCostPerSession,
-            maxToolCalls: this.config.maxToolCallsPerSession,
-            maxIterations: this.config.maxIterationsPerSession,
-            maxDuration: this.config.maxSessionDuration,
-        };
-        // 检查每日成本限制（优先检查成本，因为成本更敏感）
-        if (session.userId && this.config.maxCostPerUserPerDay) {
-            const dailyUsage = this.dailyUsage.get(session.userId);
-            if (dailyUsage && dailyUsage.estimatedCost >= this.config.maxCostPerUserPerDay) {
-                session.limitReached = true;
-                session.limitType = 'dailyCost';
-                return {
-                    allowed: false,
-                    reason: `已达到每日成本限制 ($${this.config.maxCostPerUserPerDay})`,
-                    current,
-                    limits,
-                };
-            }
-        }
-        // 检查会话成本限制
-        if (this.config.maxCostPerSession && session.tokenUsage.estimatedCost >= this.config.maxCostPerSession) {
-            session.limitReached = true;
-            session.limitType = 'sessionCost';
-            return {
-                allowed: false,
-                reason: `已达到会话成本限制 ($${this.config.maxCostPerSession})`,
-                current,
-                limits,
-            };
-        }
-        // 检查每日 Token 限制
-        if (session.userId && this.config.maxTokensPerUserPerDay) {
-            const dailyUsage = this.dailyUsage.get(session.userId);
-            if (dailyUsage && dailyUsage.totalTokens >= this.config.maxTokensPerUserPerDay) {
-                session.limitReached = true;
-                session.limitType = 'dailyTokens';
-                return {
-                    allowed: false,
-                    reason: `已达到每日 Token 使用限制 (${this.config.maxTokensPerUserPerDay})`,
-                    current,
-                    limits,
-                };
-            }
-        }
-        // 检查会话 Token 限制
-        if (this.config.maxTokensPerSession && session.tokenUsage.totalTokens >= this.config.maxTokensPerSession) {
-            session.limitReached = true;
-            session.limitType = 'sessionTokens';
-            return {
-                allowed: false,
-                reason: `已达到会话 Token 使用限制 (${this.config.maxTokensPerSession})`,
-                current,
-                limits,
-            };
-        }
-        // 检查工具调用限制
-        if (this.config.maxToolCallsPerSession && session.toolCallCount >= this.config.maxToolCallsPerSession) {
-            session.limitReached = true;
-            session.limitType = 'toolCalls';
-            return {
-                allowed: false,
-                reason: `已达到工具调用次数限制 (${this.config.maxToolCallsPerSession})`,
-                current,
-                limits,
-            };
-        }
-        // 检查迭代限制
-        if (this.config.maxIterationsPerSession && session.iterationCount >= this.config.maxIterationsPerSession) {
-            session.limitReached = true;
-            session.limitType = 'iterations';
-            return {
-                allowed: false,
-                reason: `已达到迭代次数限制 (${this.config.maxIterationsPerSession})`,
-                current,
-                limits,
-            };
-        }
-        // 检查会话时长限制
-        if (this.config.maxSessionDuration && duration >= this.config.maxSessionDuration) {
-            session.limitReached = true;
-            session.limitType = 'duration';
-            return {
-                allowed: false,
-                reason: `已达到会话时长限制 (${this.config.maxSessionDuration / 60000} 分钟)`,
-                current,
-                limits,
-            };
-        }
-        // 检查警告阈值
-        const warnings = [];
-        const threshold = (this.config.warningThreshold || 80) / 100;
-        if (this.config.maxTokensPerSession) {
-            const usageRatio = session.tokenUsage.totalTokens / this.config.maxTokensPerSession;
-            if (usageRatio >= threshold && !session.warningsSent.has('tokens')) {
-                warnings.push(`Token 使用已达 ${Math.round(usageRatio * 100)}%`);
-                session.warningsSent.add('tokens');
-            }
-        }
-        if (this.config.maxCostPerSession) {
-            const usageRatio = session.tokenUsage.estimatedCost / this.config.maxCostPerSession;
-            if (usageRatio >= threshold && !session.warningsSent.has('cost')) {
-                warnings.push(`成本已达 ${Math.round(usageRatio * 100)}%`);
-                session.warningsSent.add('cost');
-            }
-        }
-        if (warnings.length > 0) {
-            return {
-                allowed: true,
-                reason: warnings.join('; '),
-                current,
-                limits,
-                isWarning: true,
-            };
-        }
-        return { allowed: true, current, limits };
-    }
-    /**
-     * 更新每日使用情况
-     */
-    updateDailyUsage(userId, usage) {
-        const today = new Date().toISOString().split('T')[0]; // YYYY-MM-DD
-        const key = `${userId}:${today}`;
-        let daily = this.dailyUsage.get(key);
-        if (!daily) {
-            daily = { inputTokens: 0, outputTokens: 0, totalTokens: 0, estimatedCost: 0 };
-            this.dailyUsage.set(key, daily);
-        }
-        daily.inputTokens += usage.inputTokens;
-        daily.outputTokens += usage.outputTokens;
-        daily.totalTokens += usage.totalTokens;
-        daily.estimatedCost += usage.estimatedCost;
-    }
-    /**
-     * 清理过期会话和每日记录
-     */
-    cleanup() {
-        const now = Date.now();
-        const maxAge = 24 * 60 * 60 * 1000; // 24 小时
-        // 清理过期会话
-        for (const [sessionId, session] of this.sessions.entries()) {
-            if (now - session.startTime > maxAge) {
-                this.sessions.delete(sessionId);
-            }
-        }
-        // 清理过期的每日记录
-        const today = new Date().toISOString().split('T')[0];
-        for (const key of this.dailyUsage.keys()) {
-            if (!key.endsWith(today)) {
-                this.dailyUsage.delete(key);
-            }
-        }
-        // 防止当日条目无限增长
-        if (this.dailyUsage.size > BudgetLimiter.MAX_DAILY_ENTRIES) {
-            const excess = this.dailyUsage.size - BudgetLimiter.MAX_DAILY_ENTRIES;
-            let removed = 0;
-            for (const key of this.dailyUsage.keys()) {
-                if (removed >= excess)
-                    break;
-                this.dailyUsage.delete(key);
-                removed++;
-            }
-        }
-    }
-    /**
-     * 结束会话
-     */
-    endSession(sessionId) {
-        const session = this.sessions.get(sessionId);
-        if (session) {
-            this.sessions.delete(sessionId);
-        }
-        return session;
-    }
-    /**
-     * 获取会话统计
-     */
-    getSessionStats(sessionId) {
-        const session = this.sessions.get(sessionId);
-        if (!session)
-            return undefined;
-        return {
-            tokenUsage: { ...session.tokenUsage },
-            toolCallCount: session.toolCallCount,
-            iterationCount: session.iterationCount,
-            duration: Date.now() - session.startTime,
-            limits: {
-                maxTokens: this.config.maxTokensPerSession,
-                maxCost: this.config.maxCostPerSession,
-                maxToolCalls: this.config.maxToolCallsPerSession,
-                maxIterations: this.config.maxIterationsPerSession,
-                maxDuration: this.config.maxSessionDuration,
-            },
-        };
-    }
-    /**
-     * 更新配置
-     */
-    updateConfig(config) {
-        this.config = { ...this.config, ...config };
-    }
-    /**
-     * 获取配置
-     */
-    getConfig() {
-        return { ...this.config };
-    }
-    dispose() {
-        if (this.cleanupTimer) {
-            clearInterval(this.cleanupTimer);
-            this.cleanupTimer = undefined;
-        }
-        this.sessions.clear();
-        this.dailyUsage.clear();
-    }
-}
-// ── 全局预算限制器实例 ────────────────────────────────────────────────
-let globalBudgetLimiter = null;
-/**
- * 获取全局预算限制器实例
- */
-export function getBudgetLimiter() {
-    if (!globalBudgetLimiter) {
-        try {
-            const plugin = getPlugin();
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            const config = plugin.root?.inject('config')?.ai?.agent?.budget;
-            globalBudgetLimiter = new BudgetLimiter(config);
-        }
-        catch {
-            globalBudgetLimiter = new BudgetLimiter();
-        }
-    }
-    return globalBudgetLimiter;
-}
-/**
- * 初始化预算限制器
- */
-export function initBudgetLimiter(config) {
-    globalBudgetLimiter = new BudgetLimiter(config);
-    return globalBudgetLimiter;
-}
-/** 重置全局预算限制器（用于测试隔离） */
-export function resetBudgetLimiter() {
-    globalBudgetLimiter = null;
-}
-/**
- * 检查预算限制
- */
-export function checkBudgetLimit(sessionId, type, usage) {
-    const limiter = getBudgetLimiter();
-    switch (type) {
-        case 'tokens':
-            if (!usage)
-                return { allowed: true };
-            return limiter.recordTokenUsage(sessionId, usage);
-        case 'toolCall':
-            return limiter.recordToolCall(sessionId);
-        case 'iteration':
-            return limiter.recordIteration(sessionId);
-        default:
-            return { allowed: true };
-    }
-}
-//# sourceMappingURL=budget-limiter.js.map

package/lib/security/budget-limiter.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"budget-limiter.js","sourceRoot":"","sources":["../../src/security/budget-limiter.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AA2B1C,MAAM,cAAc,GAAiB;IACnC,OAAO,EAAE,IAAI;IACb,mBAAmB,EAAE,OAAO,EAAS,YAAY;IACjD,sBAAsB,EAAE,OAAO,EAAM,YAAY;IACjD,iBAAiB,EAAE,IAAI,EAAc,MAAM;IAC3C,oBAAoB,EAAE,IAAI,EAAW,MAAM;IAC3C,sBAAsB,EAAE,GAAG;IAC3B,uBAAuB,EAAE,EAAE;IAC3B,kBAAkB,EAAE,OAAO,EAAU,OAAO;IAC5C,gBAAgB,EAAE,EAAE,EAAiB,MAAM;IAC3C,aAAa,EAAE,KAAK;CACrB,CAAC;AA+DF,mEAAmE;AAEnE,MAAM,OAAO,aAAa;IAChB,MAAM,CAAe;IACrB,QAAQ,GAA+B,IAAI,GAAG,EAAE,CAAC;IACjD,UAAU,GAA4B,IAAI,GAAG,EAAE,CAAC,CAAC,kBAAkB;IACnE,YAAY,CAAkC;IAC9C,MAAM,CAAU,iBAAiB,GAAG,KAAK,CAAC;IAElD,YAAY,SAAgC,EAAE;QAC5C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAE/C,WAAW;QACX,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,YAAY,IAAI,OAAO,IAAI,CAAC,YAAY,KAAK,QAAQ,IAAI,OAAO,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC/F,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB,EAAE,MAAe;QAC9C,MAAM,MAAM,GAAkB;YAC5B,SAAS;YACT,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,UAAU,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;YACjF,aAAa,EAAE,CAAC;YAChB,cAAc,EAAE,CAAC;YACjB,YAAY,EAAE,KAAK;YACnB,YAAY,EAAE,IAAI,GAAG,EAAE;SACxB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,SAAiB;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAiB,EAAE,KAAiB;QACnD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,WAAW;QACX,OAAO,CAAC,UAAU,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;QACpD,OAAO,CAAC,UAAU,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY,CAAC;QACtD,OAAO,CAAC,UAAU,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;QACpD,OAAO,CAAC,UAAU,CAAC,aAAa,IAAI,KAAK,CAAC,aAAa,CAAC;QAExD,WAAW;QACX,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO;QACP,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,SAAiB;QAC9B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,OAAO,CAAC,aAAa,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,SAAiB;QAC/B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,OAAO,CAAC,cAAc,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,OAAsB;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,GAAG,GAAG,OAAO,CAAC,SAAS,CAAC;QAEzC,WAAW;QACX,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,WAAW;YACtC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC,aAAa;YACtC,SAAS,EAAE,OAAO,CAAC,aAAa;YAChC,UAAU,EAAE,OAAO,CAAC,cAAc;YAClC,QAAQ;SACT,CAAC;QAEF,SAAS;QACT,MAAM,MAAM,GAAG;YACb,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB;YAC1C,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB;YACtC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,sBAAsB;YAChD,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,uBAAuB;YAClD,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB;SAC5C,CAAC;QAEF,2BAA2B;QAC3B,IAAI,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACvD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvD,IAAI,UAAU,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;gBAC/E,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC5B,OAAO,CAAC,SAAS,GAAG,WAAW,CAAC;gBAChC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,eAAe,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG;oBAC1D,OAAO;oBACP,MAAM;iBACP,CAAC;YACJ,CAAC;QACH,CAAC;QAED,WAAW;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,OAAO,CAAC,UAAU,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YACvG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,OAAO,CAAC,SAAS,GAAG,aAAa,CAAC;YAClC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,eAAe,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG;gBACvD,OAAO;gBACP,MAAM;aACP,CAAC;QACJ,CAAC;QAED,gBAAgB;QAChB,IAAI,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAC;YACzD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvD,IAAI,UAAU,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAC;gBAC/E,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC5B,OAAO,CAAC,SAAS,GAAG,aAAa,CAAC;gBAClC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,qBAAqB,IAAI,CAAC,MAAM,CAAC,sBAAsB,GAAG;oBAClE,OAAO;oBACP,MAAM;iBACP,CAAC;YACJ,CAAC;QACH,CAAC;QAED,gBAAgB;QAChB,IAAI,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;YACzG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,OAAO,CAAC,SAAS,GAAG,eAAe,CAAC;YACpC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,qBAAqB,IAAI,CAAC,MAAM,CAAC,mBAAmB,GAAG;gBAC/D,OAAO;gBACP,MAAM;aACP,CAAC;QACJ,CAAC;QAED,WAAW;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,sBAAsB,IAAI,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,sBAAsB,EAAE,CAAC;YACtG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,OAAO,CAAC,SAAS,GAAG,WAAW,CAAC;YAChC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,gBAAgB,IAAI,CAAC,MAAM,CAAC,sBAAsB,GAAG;gBAC7D,OAAO;gBACP,MAAM;aACP,CAAC;QACJ,CAAC;QAED,SAAS;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,uBAAuB,IAAI,OAAO,CAAC,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC;YACzG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,OAAO,CAAC,SAAS,GAAG,YAAY,CAAC;YACjC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,cAAc,IAAI,CAAC,MAAM,CAAC,uBAAuB,GAAG;gBAC5D,OAAO;gBACP,MAAM;aACP,CAAC;QACJ,CAAC;QAED,WAAW;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACjF,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,OAAO,CAAC,SAAS,GAAG,UAAU,CAAC;YAC/B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,cAAc,IAAI,CAAC,MAAM,CAAC,kBAAkB,GAAG,KAAK,MAAM;gBAClE,OAAO;gBACP,MAAM;aACP,CAAC;QACJ,CAAC;QAED,SAAS;QACT,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC,GAAG,GAAG,CAAC;QAE7D,IAAI,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;YACpC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC;YACpF,IAAI,UAAU,IAAI,SAAS,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnE,QAAQ,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;gBAC7D,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;YACpF,IAAI,UAAU,IAAI,SAAS,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBACjE,QAAQ,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;gBACvD,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC3B,OAAO;gBACP,MAAM;gBACN,SAAS,EAAE,IAAI;aAChB,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;IAC5C,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,MAAc,EAAE,KAAiB;QACxD,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;QACnE,MAAM,GAAG,GAAG,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC;QAEjC,IAAI,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;YAC9E,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAClC,CAAC;QAED,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;QACvC,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY,CAAC;QACzC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;QACvC,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,aAAa,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,OAAO;QACb,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,QAAQ;QAE5C,SAAS;QACT,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;YAC3D,IAAI,GAAG,GAAG,OAAO,CAAC,SAAS,GAAG,MAAM,EAAE,CAAC;gBACrC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,YAAY;QACZ,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,aAAa;QACb,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,aAAa,CAAC,iBAAiB,EAAE,CAAC;YAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,aAAa,CAAC,iBAAiB,CAAC;YACtE,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;gBACzC,IAAI,OAAO,IAAI,MAAM;oBAAE,MAAM;gBAC7B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,SAAiB;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,SAAiB;QAO/B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAE/B,OAAO;YACL,UAAU,EAAE,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE;YACrC,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,SAAS;YACxC,MAAM,EAAE;gBACN,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB;gBAC1C,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB;gBACtC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,sBAAsB;gBAChD,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,uBAAuB;gBAClD,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB;aAC5C;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAA6B;QACxC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;;AAGH,gEAAgE;AAEhE,IAAI,mBAAmB,GAAyB,IAAI,CAAC;AAErD;;GAEG;AACH,MAAM,UAAU,gBAAgB;IAC9B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,8DAA8D;YAC9D,MAAM,MAAM,GAAI,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAS,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC;YACzE,mBAAmB,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;QAClD,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB,GAAG,IAAI,aAAa,EAAE,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAA6B;IAC7D,mBAAmB,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IAChD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED,wBAAwB;AACxB,MAAM,UAAU,kBAAkB;IAChC,mBAAmB,GAAG,IAAI,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,SAAiB,EAAE,IAAyC,EAAE,KAAkB;IAC/G,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC;IAEnC,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,IAAI,CAAC,KAAK;gBAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACrC,OAAO,OAAO,CAAC,gBAAgB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACpD,KAAK,UAAU;YACb,OAAO,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC3C,KAAK,WAAW;YACd,OAAO,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAC5C;YACE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC7B,CAAC;AACH,CAAC"}

package/lib/security/network-policy.d.ts

@@ -1,98 +0,0 @@
-/**
- * 网络访问安全策略
- *
- * 控制 AI Agent 的网络请求，防止：
- * - 访问恶意域名
- * - 数据外泄
- * - SSRF 攻击
- * - 资源滥用
- *
- */
-export interface NetworkPolicyConfig {
-    /** 是否启用网络策略 */
-    enabled: boolean;
-    /** 域名白名单（允许访问） */
-    allowedDomains?: string[];
-    /** 域名黑名单（禁止访问） */
-    blockedDomains?: string[];
-    /** IP 黑名单 */
-    blockedIPs?: string[];
-    /** 禁止私有 IP */
-    blockPrivateIPs?: boolean;
-    /** 最大请求大小（字节） */
-    maxRequestSize?: number;
-    /** 最大响应大小（字节） */
-    maxResponseSize?: number;
-    /** 请求超时（毫秒） */
-    requestTimeout?: number;
-    /** 每分钟最大请求数 */
-    rateLimit?: number;
-    /** 禁止的协议 */
-    blockedProtocols?: string[];
-    /** 禁止的端口 */
-    blockedPorts?: number[];
-}
-export interface NetworkPolicyResult {
-    allowed: boolean;
-    reason?: string;
-    /** 脱敏后的 URL */
-    sanitizedUrl?: string;
-}
-export declare class NetworkPolicy {
-    private config;
-    private rateLimiter;
-    private cleanupTimer?;
-    constructor(config?: Partial<NetworkPolicyConfig>);
-    /**
-     * 检查 URL 是否允许访问
-     */
-    checkUrl(url: string, userId?: string): NetworkPolicyResult;
-    /**
-     * 检查域名是否在黑名单中
-     */
-    private isBlockedDomain;
-    /**
-     * 检查域名是否在白名单中
-     */
-    private isAllowedDomain;
-    /**
-     * 检查是否为可疑域名
-     */
-    private isSuspiciousDomain;
-    /**
-     * 检查是否为私有 IP
-     */
-    private isPrivateIP;
-    /**
-     * 获取默认端口
-     */
-    private getDefaultPort;
-    /**
-     * 脱敏域名
-     */
-    private maskDomain;
-    /**
-     * 更新配置
-     */
-    updateConfig(config: Partial<NetworkPolicyConfig>): void;
-    /**
-     * 获取配置（脱敏）
-     */
-    getConfig(): NetworkPolicyConfig;
-    dispose(): void;
-}
-/**
- * 获取全局网络策略实例
- */
-export declare function getNetworkPolicy(): NetworkPolicy;
-/**
- * 初始化网络策略
- */
-export declare function initNetworkPolicy(config: Partial<NetworkPolicyConfig>): NetworkPolicy;
-/** 重置全局网络策略（用于测试隔离） */
-export declare function resetNetworkPolicy(): void;
-/**
- * 检查 URL 是否允许访问
- */
-export declare function checkNetworkAccess(url: string, userId?: string): NetworkPolicyResult;
-//# sourceMappingURL=network-policy.d.ts.map

package/lib/security/network-policy.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"network-policy.d.ts","sourceRoot":"","sources":["../../src/security/network-policy.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAMH,MAAM,WAAW,mBAAmB;IAClC,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,kBAAkB;IAClB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,kBAAkB;IAClB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa;IACb,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,cAAc;IACd,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,iBAAiB;IACjB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,iBAAiB;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,eAAe;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY;IACZ,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,YAAY;IACZ,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB;AAkGD,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAID,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAsB;IACpC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAC,CAAiC;gBAE1C,MAAM,GAAE,OAAO,CAAC,mBAAmB,CAAM;IAWrD;;OAEG;IACH,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,mBAAmB;IAuF3D;;OAEG;IACH,OAAO,CAAC,eAAe;IAavB;;OAEG;IACH,OAAO,CAAC,eAAe;IAcvB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAI1B;;OAEG;IACH,OAAO,CAAC,WAAW;IAUnB;;OAEG;IACH,OAAO,CAAC,cAAc;IAUtB;;OAEG;IACH,OAAO,CAAC,UAAU;IASlB;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,mBAAmB,CAAC,GAAG,IAAI;IAIxD;;OAEG;IACH,SAAS,IAAI,mBAAmB;IAIhC,OAAO,IAAI,IAAI;CAMhB;AAMD;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,aAAa,CAKhD;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,mBAAmB,CAAC,GAAG,aAAa,CAGrF;AAED,uBAAuB;AACvB,wBAAgB,kBAAkB,IAAI,IAAI,CAEzC;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,mBAAmB,CAEpF"}