@compass-ai/nova 1.0.73 → 1.0.74

package/dist/services/action-logger.js

@@ -0,0 +1,544 @@
+/**
+ * Action logging service for recording LLM decisions and execution status.
+ *
+ * This service provides comprehensive logging of all actions decided and executed
+ * by the LLM, including the reasoning behind decisions, execution results, and
+ * tracking of action chains.
+ */
+import path from 'path';
+import { v4 as uuidv4 } from 'uuid';
+import { formatTimestamp } from '../utils/format.js';
+import { logger } from '../utils/logger.js';
+import { getEventEmitter } from '../core/event-emitter.js';
+import { getSessionManager } from './session-manager.js';
+import { getProjectCompassDir } from '../utils/path.js';
+import { safeWriteJson, safeReadJson, pathExists, ensureDir, } from '../utils/file-system.js';
+/**
+ * Type of action the LLM can perform
+ */
+export var ActionType;
+(function (ActionType) {
+    ActionType["FILE_READ"] = "file_read";
+    ActionType["FILE_WRITE"] = "file_write";
+    ActionType["FILE_DELETE"] = "file_delete";
+    ActionType["FILE_RENAME"] = "file_rename";
+    ActionType["COMMAND_EXEC"] = "command_exec";
+    ActionType["CODE_ANALYSIS"] = "code_analysis";
+    ActionType["CODE_GENERATION"] = "code_generation";
+    ActionType["REFACTORING"] = "refactoring";
+    ActionType["TESTING"] = "testing";
+    ActionType["DEBUGGING"] = "debugging";
+    ActionType["DOCUMENTATION"] = "documentation";
+    ActionType["SEARCH"] = "search";
+    ActionType["PLANNING"] = "planning";
+    ActionType["APPROVAL_REQUEST"] = "approval_request";
+    ActionType["CUSTOM"] = "custom";
+})(ActionType || (ActionType = {}));
+/**
+ * Status of an action
+ */
+export var ActionStatus;
+(function (ActionStatus) {
+    ActionStatus["PLANNED"] = "planned";
+    ActionStatus["PENDING_APPROVAL"] = "pending_approval";
+    ActionStatus["APPROVED"] = "approved";
+    ActionStatus["REJECTED"] = "rejected";
+    ActionStatus["IN_PROGRESS"] = "in_progress";
+    ActionStatus["COMPLETED"] = "completed";
+    ActionStatus["FAILED"] = "failed";
+    ActionStatus["CANCELLED"] = "cancelled";
+})(ActionStatus || (ActionStatus = {}));
+/**
+ * ActionLogger class provides comprehensive logging of LLM actions.
+ *
+ * Features:
+ * - Action recording with reasoning and parameters
+ * - Status tracking through action lifecycle
+ * - Action chain management for related operations
+ * - Approval workflow integration
+ * - Persistent logging to disk
+ * - Statistics and analysis
+ */
+class ActionLogger {
+    actions = new Map();
+    chains = new Map();
+    currentChainId = null;
+    projectRoot = process.cwd();
+    /**
+     * Records a new action from the LLM.
+     *
+     * @param type - Type of action
+     * @param description - Human-readable description
+     * @param parameters - Action parameters
+     * @param options - Additional action options
+     * @returns The created Action object
+     *
+     * @example
+     * ```typescript
+     * const logger = getActionLogger();
+     * const action = logger.recordAction(
+     *   ActionType.FILE_WRITE,
+     *   'Update user authentication logic',
+     *   { filePath: 'src/auth.ts', content: '...' },
+     *   { reasoning: 'Need to add JWT token validation' }
+     * );
+     * ```
+     */
+    recordAction(type, description, parameters, options) {
+        const action = {
+            id: uuidv4(),
+            type,
+            status: options?.requiresApproval
+                ? ActionStatus.PENDING_APPROVAL
+                : ActionStatus.PLANNED,
+            description,
+            reasoning: options?.reasoning,
+            parameters,
+            expectedOutcome: options?.expectedOutcome,
+            parentActionId: options?.parentActionId,
+            relatedActions: [],
+            createdAt: formatTimestamp(),
+            metadata: options?.metadata,
+        };
+        if (options?.requiresApproval) {
+            action.approval = {
+                required: true,
+            };
+        }
+        this.actions.set(action.id, action);
+        // Add to current chain if one exists
+        if (this.currentChainId) {
+            const chain = this.chains.get(this.currentChainId);
+            if (chain) {
+                chain.actions.push(action);
+            }
+        }
+        logger.debug(`Action recorded: ${description} (${action.id})`);
+        // Emit event for UI
+        const eventEmitter = getEventEmitter();
+        eventEmitter.emit('action-recorded', { action });
+        // Record in session
+        const sessionManager = getSessionManager();
+        sessionManager.recordOperation({
+            type: 'action_recorded',
+            description: `LLM action: ${description}`,
+            status: 'completed',
+            details: {
+                actionId: action.id,
+                actionType: type,
+                parameters: parameters,
+            },
+        });
+        return action;
+    }
+    /**
+     * Starts an action chain for grouping related actions.
+     *
+     * @param name - Chain name
+     * @param description - Chain description
+     * @param metadata - Optional metadata
+     * @returns The created ActionChain object
+     *
+     * @example
+     * ```typescript
+     * const chain = logger.startActionChain(
+     *   'Implement Login Feature',
+     *   'All actions related to adding user authentication'
+     * );
+     * ```
+     */
+    startActionChain(name, description, metadata) {
+        const chain = {
+            id: uuidv4(),
+            name,
+            description,
+            actions: [],
+            createdAt: formatTimestamp(),
+            status: 'active',
+            metadata,
+        };
+        this.chains.set(chain.id, chain);
+        this.currentChainId = chain.id;
+        logger.info(`Action chain started: ${name} (${chain.id})`);
+        // Emit event for UI
+        const eventEmitter = getEventEmitter();
+        eventEmitter.emit('action-chain-started', { chain });
+        return chain;
+    }
+    /**
+     * Ends the current action chain.
+     *
+     * @param chainId - ID of the chain to end
+     * @param status - Final status of the chain
+     */
+    endActionChain(chainId, status = 'completed') {
+        const chain = this.chains.get(chainId);
+        if (!chain) {
+            throw new Error(`Action chain not found: ${chainId}`);
+        }
+        chain.status = status;
+        if (this.currentChainId === chainId) {
+            this.currentChainId = null;
+        }
+        logger.info(`Action chain ended: ${chain.name} (${chainId}) - ${status}`);
+        // Emit event for UI
+        const eventEmitter = getEventEmitter();
+        eventEmitter.emit('action-chain-ended', { chain, status });
+    }
+    /**
+     * Updates the status of an action.
+     *
+     * @param actionId - ID of the action to update
+     * @param status - New status
+     */
+    updateActionStatus(actionId, status) {
+        const action = this.actions.get(actionId);
+        if (!action) {
+            throw new Error(`Action not found: ${actionId}`);
+        }
+        const oldStatus = action.status;
+        action.status = status;
+        if (status === ActionStatus.IN_PROGRESS && !action.startedAt) {
+            action.startedAt = formatTimestamp();
+        }
+        if ((status === ActionStatus.COMPLETED ||
+            status === ActionStatus.FAILED ||
+            status === ActionStatus.CANCELLED) &&
+            !action.completedAt) {
+            action.completedAt = formatTimestamp();
+            if (action.startedAt) {
+                action.duration =
+                    new Date(action.completedAt).getTime() -
+                        new Date(action.startedAt).getTime();
+            }
+        }
+        logger.debug(`Action status updated: ${actionId} - ${oldStatus} -> ${status}`);
+        // Emit event for UI
+        const eventEmitter = getEventEmitter();
+        eventEmitter.emit('action-status-changed', { action, oldStatus, status });
+    }
+    /**
+     * Records approval decision for an action.
+     *
+     * @param actionId - ID of the action
+     * @param granted - Whether approval was granted
+     * @param decidedBy - User who made the decision
+     * @param rejectionReason - Reason if rejected
+     */
+    recordApproval(actionId, granted, decidedBy, rejectionReason) {
+        const action = this.actions.get(actionId);
+        if (!action) {
+            throw new Error(`Action not found: ${actionId}`);
+        }
+        if (!action.approval) {
+            action.approval = { required: true };
+        }
+        action.approval.granted = granted;
+        action.approval.decidedAt = formatTimestamp();
+        action.approval.decidedBy = decidedBy;
+        action.approval.rejectionReason = rejectionReason;
+        action.status = granted ? ActionStatus.APPROVED : ActionStatus.REJECTED;
+        logger.info(`Action approval recorded: ${actionId} - ${granted ? 'Approved' : 'Rejected'}`);
+        // Emit event for UI
+        const eventEmitter = getEventEmitter();
+        eventEmitter.emit('action-approval', { action, granted });
+    }
+    /**
+     * Records the result of action execution.
+     *
+     * @param actionId - ID of the action
+     * @param result - Action result
+     */
+    recordResult(actionId, result) {
+        const action = this.actions.get(actionId);
+        if (!action) {
+            throw new Error(`Action not found: ${actionId}`);
+        }
+        action.result = result;
+        action.status = result.success
+            ? ActionStatus.COMPLETED
+            : ActionStatus.FAILED;
+        if (!action.completedAt) {
+            action.completedAt = formatTimestamp();
+            if (action.startedAt) {
+                action.duration =
+                    new Date(action.completedAt).getTime() -
+                        new Date(action.startedAt).getTime();
+            }
+        }
+        logger.info(`Action result recorded: ${actionId} - ${result.success ? 'Success' : 'Failed'}`);
+        // Emit event for UI
+        const eventEmitter = getEventEmitter();
+        eventEmitter.emit('action-result', { action, result });
+        // Record in session
+        const sessionManager = getSessionManager();
+        sessionManager.recordOperation({
+            type: result.success ? 'action_completed' : 'action_failed',
+            description: `${action.description}`,
+            status: result.success ? 'completed' : 'failed',
+            details: {
+                actionId: action.id,
+                actionType: action.type,
+                duration: action.duration,
+                result: result,
+            },
+        });
+    }
+    /**
+     * Links two actions as related.
+     *
+     * @param actionId - ID of the first action
+     * @param relatedActionId - ID of the related action
+     */
+    linkActions(actionId, relatedActionId) {
+        const action = this.actions.get(actionId);
+        const relatedAction = this.actions.get(relatedActionId);
+        if (!action || !relatedAction) {
+            throw new Error('One or both actions not found');
+        }
+        if (!action.relatedActions.includes(relatedActionId)) {
+            action.relatedActions.push(relatedActionId);
+        }
+        if (!relatedAction.relatedActions.includes(actionId)) {
+            relatedAction.relatedActions.push(actionId);
+        }
+        logger.debug(`Actions linked: ${actionId} <-> ${relatedActionId}`);
+    }
+    /**
+     * Gets an action by ID.
+     *
+     * @param actionId - ID of the action
+     * @returns The Action object or undefined
+     */
+    getAction(actionId) {
+        return this.actions.get(actionId);
+    }
+    /**
+     * Gets an action chain by ID.
+     *
+     * @param chainId - ID of the chain
+     * @returns The ActionChain object or undefined
+     */
+    getActionChain(chainId) {
+        return this.chains.get(chainId);
+    }
+    /**
+     * Gets all actions.
+     *
+     * @returns Array of all Action objects
+     */
+    getAllActions() {
+        return Array.from(this.actions.values());
+    }
+    /**
+     * Gets all action chains.
+     *
+     * @returns Array of all ActionChain objects
+     */
+    getAllChains() {
+        return Array.from(this.chains.values());
+    }
+    /**
+     * Gets actions by status.
+     *
+     * @param status - Status to filter by
+     * @returns Array of matching Action objects
+     */
+    getActionsByStatus(status) {
+        return Array.from(this.actions.values()).filter((action) => action.status === status);
+    }
+    /**
+     * Gets actions by type.
+     *
+     * @param type - Type to filter by
+     * @returns Array of matching Action objects
+     */
+    getActionsByType(type) {
+        return Array.from(this.actions.values()).filter((action) => action.type === type);
+    }
+    /**
+     * Gets actions pending approval.
+     *
+     * @returns Array of Action objects awaiting approval
+     */
+    getPendingApprovals() {
+        return this.getActionsByStatus(ActionStatus.PENDING_APPROVAL);
+    }
+    /**
+     * Gets the current active action chain.
+     *
+     * @returns The current ActionChain or null
+     */
+    getCurrentChain() {
+        if (!this.currentChainId) {
+            return null;
+        }
+        return this.chains.get(this.currentChainId) || null;
+    }
+    /**
+     * Generates statistics about logged actions.
+     *
+     * @returns ActionStatistics object
+     */
+    getStatistics() {
+        const actions = Array.from(this.actions.values());
+        const byStatus = {
+            [ActionStatus.PLANNED]: 0,
+            [ActionStatus.PENDING_APPROVAL]: 0,
+            [ActionStatus.APPROVED]: 0,
+            [ActionStatus.REJECTED]: 0,
+            [ActionStatus.IN_PROGRESS]: 0,
+            [ActionStatus.COMPLETED]: 0,
+            [ActionStatus.FAILED]: 0,
+            [ActionStatus.CANCELLED]: 0,
+        };
+        const byType = {};
+        let totalDuration = 0;
+        let completedCount = 0;
+        actions.forEach((action) => {
+            byStatus[action.status]++;
+            if (!byType[action.type]) {
+                byType[action.type] = 0;
+            }
+            byType[action.type]++;
+            if (action.status === ActionStatus.COMPLETED && action.duration) {
+                totalDuration += action.duration;
+                completedCount++;
+            }
+        });
+        const successfulActions = byStatus[ActionStatus.COMPLETED];
+        const failedActions = byStatus[ActionStatus.FAILED];
+        const totalFinished = successfulActions + failedActions;
+        const successRate = totalFinished > 0 ? (successfulActions / totalFinished) * 100 : 0;
+        const averageDuration = completedCount > 0 ? totalDuration / completedCount : 0;
+        // Find most common type
+        let mostCommonType = null;
+        let maxCount = 0;
+        Object.entries(byType).forEach(([type, count]) => {
+            if (count > maxCount) {
+                maxCount = count;
+                mostCommonType = type;
+            }
+        });
+        return {
+            totalActions: actions.length,
+            byStatus,
+            byType,
+            successRate,
+            averageDuration,
+            mostCommonType,
+        };
+    }
+    /**
+     * Persists action log to disk.
+     *
+     * @param projectRoot - Optional project root path
+     */
+    async persist(projectRoot) {
+        const root = projectRoot || this.projectRoot;
+        const compassDir = getProjectCompassDir(root);
+        await ensureDir(compassDir);
+        const logPath = path.join(compassDir, 'action-log.json');
+        const data = {
+            actions: Array.from(this.actions.values()),
+            chains: Array.from(this.chains.values()),
+            timestamp: formatTimestamp(),
+        };
+        await safeWriteJson(logPath, data);
+        logger.debug('Action log persisted to disk');
+    }
+    /**
+     * Loads action log from disk.
+     *
+     * @param projectRoot - Optional project root path
+     * @returns True if log was loaded successfully
+     */
+    async load(projectRoot) {
+        const root = projectRoot || this.projectRoot;
+        const logPath = path.join(getProjectCompassDir(root), 'action-log.json');
+        if (!(await pathExists(logPath))) {
+            return false;
+        }
+        try {
+            const data = await safeReadJson(logPath);
+            if (data) {
+                this.actions.clear();
+                this.chains.clear();
+                data.actions.forEach((action) => {
+                    this.actions.set(action.id, action);
+                });
+                data.chains.forEach((chain) => {
+                    this.chains.set(chain.id, chain);
+                });
+                logger.info('Action log loaded from disk');
+                return true;
+            }
+            return false;
+        }
+        catch (error) {
+            logger.error('Failed to load action log', error);
+            return false;
+        }
+    }
+    /**
+     * Clears all actions and chains.
+     */
+    clear() {
+        this.actions.clear();
+        this.chains.clear();
+        this.currentChainId = null;
+        logger.info('Action log cleared');
+    }
+    /**
+     * Exports action log data.
+     *
+     * @returns Object containing all actions and chains
+     */
+    export() {
+        return {
+            actions: Array.from(this.actions.values()),
+            chains: Array.from(this.chains.values()),
+        };
+    }
+    /**
+     * Sets the project root path.
+     *
+     * @param projectRoot - New project root path
+     */
+    setProjectRoot(projectRoot) {
+        this.projectRoot = projectRoot;
+    }
+}
+// Singleton instance
+let actionLoggerInstance = null;
+/**
+ * Gets the singleton ActionLogger instance.
+ *
+ * @returns The ActionLogger instance
+ *
+ * @example
+ * ```typescript
+ * const logger = getActionLogger();
+ * const action = logger.recordAction(
+ *   ActionType.FILE_WRITE,
+ *   'Create new component',
+ *   { filePath: 'src/components/Button.tsx' }
+ * );
+ * ```
+ */
+export function getActionLogger() {
+    if (!actionLoggerInstance) {
+        actionLoggerInstance = new ActionLogger();
+    }
+    return actionLoggerInstance;
+}
+/**
+ * Creates a new ActionLogger instance.
+ *
+ * @returns A new ActionLogger instance
+ */
+export function createActionLogger() {
+    return new ActionLogger();
+}
+export { ActionLogger };
+//# sourceMappingURL=action-logger.js.map

package/dist/services/action-logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"action-logger.js","sourceRoot":"","sources":["../../src/services/action-logger.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EACL,aAAa,EACb,YAAY,EACZ,UAAU,EACV,SAAS,GACV,MAAM,yBAAyB,CAAC;AAEjC;;GAEG;AACH,MAAM,CAAN,IAAY,UAgBX;AAhBD,WAAY,UAAU;IACpB,qCAAuB,CAAA;IACvB,uCAAyB,CAAA;IACzB,yCAA2B,CAAA;IAC3B,yCAA2B,CAAA;IAC3B,2CAA6B,CAAA;IAC7B,6CAA+B,CAAA;IAC/B,iDAAmC,CAAA;IACnC,yCAA2B,CAAA;IAC3B,iCAAmB,CAAA;IACnB,qCAAuB,CAAA;IACvB,6CAA+B,CAAA;IAC/B,+BAAiB,CAAA;IACjB,mCAAqB,CAAA;IACrB,mDAAqC,CAAA;IACrC,+BAAiB,CAAA;AACnB,CAAC,EAhBW,UAAU,KAAV,UAAU,QAgBrB;AAED;;GAEG;AACH,MAAM,CAAN,IAAY,YASX;AATD,WAAY,YAAY;IACtB,mCAAmB,CAAA;IACnB,qDAAqC,CAAA;IACrC,qCAAqB,CAAA;IACrB,qCAAqB,CAAA;IACrB,2CAA2B,CAAA;IAC3B,uCAAuB,CAAA;IACvB,iCAAiB,CAAA;IACjB,uCAAuB,CAAA;AACzB,CAAC,EATW,YAAY,KAAZ,YAAY,QASvB;AA4ID;;;;;;;;;;GAUG;AACH,MAAM,YAAY;IACR,OAAO,GAAwB,IAAI,GAAG,EAAE,CAAC;IACzC,MAAM,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC7C,cAAc,GAAkB,IAAI,CAAC;IACrC,WAAW,GAAW,OAAO,CAAC,GAAG,EAAE,CAAC;IAE5C;;;;;;;;;;;;;;;;;;;OAmBG;IACH,YAAY,CACV,IAAgB,EAChB,WAAmB,EACnB,UAA+B,EAC/B,OAMC;QAED,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,MAAM,EAAE;YACZ,IAAI;YACJ,MAAM,EAAE,OAAO,EAAE,gBAAgB;gBAC/B,CAAC,CAAC,YAAY,CAAC,gBAAgB;gBAC/B,CAAC,CAAC,YAAY,CAAC,OAAO;YACxB,WAAW;YACX,SAAS,EAAE,OAAO,EAAE,SAAS;YAC7B,UAAU;YACV,eAAe,EAAE,OAAO,EAAE,eAAe;YACzC,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,cAAc,EAAE,EAAE;YAClB,SAAS,EAAE,eAAe,EAAE;YAC5B,QAAQ,EAAE,OAAO,EAAE,QAAQ;SAC5B,CAAC;QAEF,IAAI,OAAO,EAAE,gBAAgB,EAAE,CAAC;YAC9B,MAAM,CAAC,QAAQ,GAAG;gBAChB,QAAQ,EAAE,IAAI;aACf,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAEpC,qCAAqC;QACrC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnD,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,oBAAoB,WAAW,KAAK,MAAM,CAAC,EAAE,GAAG,CAAC,CAAC;QAE/D,oBAAoB;QACpB,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,oBAAoB;QACpB,MAAM,cAAc,GAAG,iBAAiB,EAAE,CAAC;QAC3C,cAAc,CAAC,eAAe,CAAC;YAC7B,IAAI,EAAE,iBAAiB;YACvB,WAAW,EAAE,eAAe,WAAW,EAAE;YACzC,MAAM,EAAE,WAAW;YACnB,OAAO,EAAE;gBACP,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,UAAU,EAAE,IAAI;gBAChB,UAAU,EAAE,UAAU;aAChB;SACT,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;;;;;;OAeG;IACH,gBAAgB,CACd,IAAY,EACZ,WAAmB,EACnB,QAA8B;QAE9B,MAAM,KAAK,GAAgB;YACzB,EAAE,EAAE,MAAM,EAAE;YACZ,IAAI;YACJ,WAAW;YACX,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,eAAe,EAAE;YAC5B,MAAM,EAAE,QAAQ;YAChB,QAAQ;SACT,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,EAAE,CAAC;QAE/B,MAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,KAAK,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;QAE3D,oBAAoB;QACpB,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAErD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;OAKG;IACH,cAAc,CACZ,OAAe,EACf,SAA+C,WAAW;QAE1D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,2BAA2B,OAAO,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QAEtB,IAAI,IAAI,CAAC,cAAc,KAAK,OAAO,EAAE,CAAC;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,uBAAuB,KAAK,CAAC,IAAI,KAAK,OAAO,OAAO,MAAM,EAAE,CAAC,CAAC;QAE1E,oBAAoB;QACpB,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED;;;;;OAKG;IACH,kBAAkB,CAAC,QAAgB,EAAE,MAAoB;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;QAChC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;QAEvB,IAAI,MAAM,KAAK,YAAY,CAAC,WAAW,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC7D,MAAM,CAAC,SAAS,GAAG,eAAe,EAAE,CAAC;QACvC,CAAC;QAED,IACE,CAAC,MAAM,KAAK,YAAY,CAAC,SAAS;YAChC,MAAM,KAAK,YAAY,CAAC,MAAM;YAC9B,MAAM,KAAK,YAAY,CAAC,SAAS,CAAC;YACpC,CAAC,MAAM,CAAC,WAAW,EACnB,CAAC;YACD,MAAM,CAAC,WAAW,GAAG,eAAe,EAAE,CAAC;YAEvC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,MAAM,CAAC,QAAQ;oBACb,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE;wBACtC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;YACzC,CAAC;QACH,CAAC;QAED,MAAM,CAAC,KAAK,CACV,0BAA0B,QAAQ,MAAM,SAAS,OAAO,MAAM,EAAE,CACjE,CAAC;QAEF,oBAAoB;QACpB,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED;;;;;;;OAOG;IACH,cAAc,CACZ,QAAgB,EAChB,OAAgB,EAChB,SAAkB,EAClB,eAAwB;QAExB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACrB,MAAM,CAAC,QAAQ,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QACvC,CAAC;QAED,MAAM,CAAC,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,SAAS,GAAG,eAAe,EAAE,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,eAAe,GAAG,eAAe,CAAC;QAElD,MAAM,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC;QAExE,MAAM,CAAC,IAAI,CACT,6BAA6B,QAAQ,MAAM,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,EAAE,CAC/E,CAAC;QAEF,oBAAoB;QACpB,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,QAAgB,EAAE,MAAoB;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;QACvB,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO;YAC5B,CAAC,CAAC,YAAY,CAAC,SAAS;YACxB,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC;QAExB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YACxB,MAAM,CAAC,WAAW,GAAG,eAAe,EAAE,CAAC;YAEvC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,MAAM,CAAC,QAAQ;oBACb,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE;wBACtC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;YACzC,CAAC;QACH,CAAC;QAED,MAAM,CAAC,IAAI,CACT,2BAA2B,QAAQ,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CACjF,CAAC;QAEF,oBAAoB;QACpB,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAEvD,oBAAoB;QACpB,MAAM,cAAc,GAAG,iBAAiB,EAAE,CAAC;QAC3C,cAAc,CAAC,eAAe,CAAC;YAC7B,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe;YAC3D,WAAW,EAAE,GAAG,MAAM,CAAC,WAAW,EAAE;YACpC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;YAC/C,OAAO,EAAE;gBACP,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,UAAU,EAAE,MAAM,CAAC,IAAI;gBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,MAAM;aACR;SACT,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,WAAW,CAAC,QAAgB,EAAE,eAAuB;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAExD,IAAI,CAAC,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACrD,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,aAAa,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,mBAAmB,QAAQ,QAAQ,eAAe,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;;;;OAKG;IACH,SAAS,CAAC,QAAgB;QACxB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED;;;;;OAKG;IACH,cAAc,CAAC,OAAe;QAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED;;;;OAIG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;;;OAIG;IACH,YAAY;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;OAKG;IACH,kBAAkB,CAAC,MAAoB;QACrC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAC7C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,MAAM,CACrC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,gBAAgB,CAAC,IAAgB;QAC/B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAC7C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,CACjC,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;IAChE,CAAC;IAED;;;;OAIG;IACH,eAAe;QACb,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC;IACtD,CAAC;IAED;;;;OAIG;IACH,aAAa;QACX,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAElD,MAAM,QAAQ,GAAiC;YAC7C,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;YACzB,CAAC,YAAY,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAClC,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;YAC7B,CAAC,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3B,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC;YACxB,CAAC,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC;SAC5B,CAAC;QAEF,MAAM,MAAM,GAA+B,EAAS,CAAC;QAErD,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,cAAc,GAAG,CAAC,CAAC;QAEvB,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACzB,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAE1B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzB,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAEtB,IAAI,MAAM,CAAC,MAAM,KAAK,YAAY,CAAC,SAAS,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAChE,aAAa,IAAI,MAAM,CAAC,QAAQ,CAAC;gBACjC,cAAc,EAAE,CAAC;YACnB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,aAAa,GAAG,QAAQ,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACpD,MAAM,aAAa,GAAG,iBAAiB,GAAG,aAAa,CAAC;QAExD,MAAM,WAAW,GACf,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,GAAG,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAEpE,MAAM,eAAe,GACnB,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1D,wBAAwB;QACxB,IAAI,cAAc,GAAsB,IAAI,CAAC;QAC7C,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE;YAC/C,IAAI,KAAK,GAAG,QAAQ,EAAE,CAAC;gBACrB,QAAQ,GAAG,KAAK,CAAC;gBACjB,cAAc,GAAG,IAAkB,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,QAAQ;YACR,MAAM;YACN,WAAW;YACX,eAAe;YACf,cAAc;SACf,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO,CAAC,WAAoB;QAChC,MAAM,IAAI,GAAG,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC7C,MAAM,UAAU,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,SAAS,CAAC,UAAU,CAAC,CAAC;QAE5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAEzD,MAAM,IAAI,GAAG;YACX,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACxC,SAAS,EAAE,eAAe,EAAE;SAC7B,CAAC;QAEF,MAAM,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,IAAI,CAAC,WAAoB;QAC7B,MAAM,IAAI,GAAG,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,iBAAiB,CAAC,CAAC;QAEzE,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,YAAY,CAG5B,OAAO,CAAC,CAAC;YAEZ,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACrB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;gBAEpB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;oBAC9B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;gBACtC,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;oBAC5B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;gBACnC,CAAC,CAAC,CAAC;gBAEH,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;gBAC3C,OAAO,IAAI,CAAC;YACd,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;YACjD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAE3B,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACpC,CAAC;IAED;;;;OAIG;IACH,MAAM;QACJ,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;SACzC,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;CACF;AAED,qBAAqB;AACrB,IAAI,oBAAoB,GAAwB,IAAI,CAAC;AAErD;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,eAAe;IAC7B,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG,IAAI,YAAY,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,IAAI,YAAY,EAAE,CAAC;AAC5B,CAAC;AAED,OAAO,EAAE,YAAY,EAAE,CAAC"}

package/dist/services/agent-budget-allocator.d.ts

@@ -0,0 +1,111 @@
+/**
+ * Agent Budget Allocator Service
+ *
+ * Manages budget allocation for agent processes, enforcing token and turn limits
+ * to prevent context bloat. Supports multiple allocation strategies.
+ */
+import type { AgentBudget, BudgetAllocationOptions } from '../types/agent-process.js';
+import { createDefaultBudget, getRemainingBudget, getBudgetUsagePercent, isBudgetExceeded } from '../types/agent-process.js';
+/**
+ * Default budget configurations for different agent types.
+ * These are tuned based on typical task complexity for each agent.
+ */
+export declare const DEFAULT_AGENT_BUDGETS: Record<string, AgentBudget>;
+/**
+ * Warning thresholds for budget consumption (percentage).
+ */
+export declare const BUDGET_WARNING_THRESHOLDS: {
+    /** First warning at 70% consumption */
+    WARNING: number;
+    /** Critical warning at 90% consumption */
+    CRITICAL: number;
+};
+/**
+ * Service for allocating and managing agent budgets.
+ */
+declare class AgentBudgetAllocatorService {
+    private customBudgets;
+    /**
+     * Allocates a budget for an agent based on the specified strategy.
+     *
+     * @param agentName - Name of the agent type
+     * @param options - Budget allocation options
+     * @returns Allocated budget
+     */
+    allocateBudget(agentName: string, options?: BudgetAllocationOptions): AgentBudget;
+    /**
+     * Allocates a fixed budget based on agent type defaults or explicit values.
+     */
+    private allocateFixedBudget;
+    /**
+     * Allocates a budget as a percentage of the parent's remaining budget.
+     */
+    private allocatePercentageBudget;
+    /**
+     * Allocates a budget based on task complexity (heuristic).
+     * Uses task length and agent type to estimate needed budget.
+     */
+    private allocateAdaptiveBudget;
+    /**
+     * Validates that current usage is within budget limits.
+     *
+     * @param budget - Budget to validate
+     * @returns true if within limits, false if exceeded
+     */
+    validateBudget(budget: AgentBudget): boolean;
+    /**
+     * Updates budget usage after an API call.
+     *
+     * @param budget - Budget to update
+     * @param tokensUsed - Tokens consumed in this call
+     * @param incrementTurn - Whether to increment turn count
+     * @returns Updated budget
+     */
+    updateUsage(budget: AgentBudget, tokensUsed: number, incrementTurn?: boolean): AgentBudget;
+    /**
+     * Checks if a budget warning should be issued.
+     *
+     * @param budget - Budget to check
+     * @returns Warning level or null if no warning
+     */
+    checkWarning(budget: AgentBudget): 'warning' | 'critical' | null;
+    /**
+     * Estimates if there's enough budget remaining for another turn.
+     * Uses average tokens per turn from current usage.
+     *
+     * @param budget - Budget to check
+     * @returns true if likely enough budget for another turn
+     */
+    hasRoomForNextTurn(budget: AgentBudget): boolean;
+    /**
+     * Registers a custom budget configuration for an agent type.
+     *
+     * @param agentName - Agent type name
+     * @param budget - Budget configuration
+     */
+    registerCustomBudget(agentName: string, budget: Omit<AgentBudget, 'tokensUsed' | 'turnsUsed'>): void;
+    /**
+     * Gets the default budget for an agent type.
+     *
+     * @param agentName - Agent type name
+     * @returns Default budget configuration
+     */
+    getDefaultBudget(agentName: string): AgentBudget;
+    /**
+     * Formats budget status for display/logging.
+     *
+     * @param budget - Budget to format
+     * @returns Human-readable budget status
+     */
+    formatBudgetStatus(budget: AgentBudget): string;
+}
+/**
+ * Gets the singleton AgentBudgetAllocator instance.
+ */
+export declare function getAgentBudgetAllocator(): AgentBudgetAllocatorService;
+/**
+ * Resets the allocator instance (for testing).
+ */
+export declare function resetAgentBudgetAllocator(): void;
+export { createDefaultBudget, getRemainingBudget, getBudgetUsagePercent, isBudgetExceeded, };
+//# sourceMappingURL=agent-budget-allocator.d.ts.map

package/dist/services/agent-budget-allocator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-budget-allocator.d.ts","sourceRoot":"","sources":["../../src/services/agent-budget-allocator.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EACV,WAAW,EACX,uBAAuB,EACxB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,gBAAgB,EACjB,MAAM,2BAA2B,CAAC;AAMnC;;;GAGG;AACH,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAwC7D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,yBAAyB;IACpC,uCAAuC;;IAEvC,0CAA0C;;CAE3C,CAAC;AAMF;;GAEG;AACH,cAAM,2BAA2B;IAC/B,OAAO,CAAC,aAAa,CAAuC;IAE5D;;;;;;OAMG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,uBAAuB,GAAG,WAAW;IAmBjF;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAmB3B;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAyBhC;;;OAGG;IACH,OAAO,CAAC,sBAAsB;IAmB9B;;;;;OAKG;IACH,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO;IAI5C;;;;;;;OAOG;IACH,WAAW,CAAC,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,GAAE,OAAc,GAAG,WAAW;IAQhG;;;;;OAKG;IACH,YAAY,CAAC,MAAM,EAAE,WAAW,GAAG,SAAS,GAAG,UAAU,GAAG,IAAI;IAahE;;;;;;OAMG;IACH,kBAAkB,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO;IAgBhD;;;;;OAKG;IACH,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,YAAY,GAAG,WAAW,CAAC,GAAG,IAAI;IASpG;;;;;OAKG;IACH,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW;IAQhD;;;;;OAKG;IACH,kBAAkB,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM;CAUhD;AAQD;;GAEG;AACH,wBAAgB,uBAAuB,IAAI,2BAA2B,CAKrE;AAED;;GAEG;AACH,wBAAgB,yBAAyB,IAAI,IAAI,CAEhD;AAGD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,gBAAgB,GACjB,CAAC"}