@defai.digital/ax-cli 2.7.0 → 2.8.0

package/dist/agent/subagent.js

@@ -0,0 +1,402 @@
+/**
+ * Subagent Base Class
+ *
+ * Base class for all specialized subagents. Subagents are lightweight AI agents
+ * that focus on specific tasks (testing, documentation, refactoring, etc.) with
+ * limited tool access and shallow conversation history.
+ */
+import { EventEmitter } from 'events';
+import { LLMClient } from '../llm/client.js';
+import { SubagentRole, SubagentState } from './subagent-types.js';
+// Import tools
+import { BashTool } from '../tools/bash.js';
+import { TextEditorTool } from '../tools/text-editor.js';
+import { SearchTool } from '../tools/search.js';
+/**
+ * Base Subagent class
+ */
+export class Subagent extends EventEmitter {
+    id;
+    role;
+    config;
+    llmClient;
+    chatHistory;
+    messages;
+    tools;
+    isActive;
+    currentTaskId;
+    status;
+    constructor(role, configOverrides) {
+        super();
+        this.id = `${role}-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
+        this.role = role;
+        // Merge with default config for this role
+        const defaultConfig = this.getDefaultConfig(role);
+        this.config = { ...defaultConfig, ...configOverrides, role };
+        this.chatHistory = [];
+        this.messages = [];
+        this.tools = new Map();
+        this.isActive = false;
+        this.currentTaskId = null;
+        // Initialize status
+        this.status = {
+            id: this.id,
+            taskId: '',
+            role: this.role,
+            state: SubagentState.PENDING,
+            progress: 0,
+            startTime: new Date(),
+        };
+        // Initialize LLM client with same settings as main agent
+        this.llmClient = new LLMClient(process.env.YOUR_API_KEY || '', process.env.GROK_MODEL || 'glm-4.6', process.env.GROK_BASE_URL);
+        // Initialize allowed tools
+        this.initializeTools();
+        // Set system prompt
+        this.messages.push({
+            role: 'system',
+            content: this.buildSystemPrompt(),
+        });
+    }
+    /**
+     * Get default configuration for a role
+     */
+    getDefaultConfig(role) {
+        // Default tools and settings based on role
+        const roleDefaults = {
+            [SubagentRole.GENERAL]: {
+                allowedTools: ['bash', 'text_editor', 'search'],
+                maxToolRounds: 30,
+                contextDepth: 20,
+                priority: 1,
+            },
+            [SubagentRole.TESTING]: {
+                allowedTools: ['bash', 'text_editor', 'search'],
+                maxToolRounds: 20,
+                contextDepth: 10,
+                priority: 2,
+            },
+            [SubagentRole.DOCUMENTATION]: {
+                allowedTools: ['text_editor', 'search'],
+                maxToolRounds: 15,
+                contextDepth: 20,
+                priority: 1,
+            },
+            [SubagentRole.REFACTORING]: {
+                allowedTools: ['bash', 'text_editor', 'search'],
+                maxToolRounds: 25,
+                contextDepth: 20,
+                priority: 2,
+            },
+            [SubagentRole.ANALYSIS]: {
+                allowedTools: ['bash', 'text_editor', 'search'],
+                maxToolRounds: 15,
+                contextDepth: 15,
+                priority: 3,
+            },
+            [SubagentRole.DEBUG]: {
+                allowedTools: ['bash', 'text_editor', 'search'],
+                maxToolRounds: 30,
+                contextDepth: 15,
+                priority: 3,
+            },
+            [SubagentRole.PERFORMANCE]: {
+                allowedTools: ['bash', 'search'],
+                maxToolRounds: 25,
+                contextDepth: 10,
+                priority: 2,
+            },
+        };
+        return {
+            role,
+            allowedTools: [],
+            maxToolRounds: 20,
+            contextDepth: 10,
+            priority: 1,
+            ...roleDefaults[role],
+        };
+    }
+    /**
+     * Initialize tools based on allowed tools in config
+     */
+    initializeTools() {
+        const allowedTools = this.config.allowedTools;
+        if (allowedTools.includes('bash')) {
+            this.tools.set('bash', new BashTool());
+        }
+        if (allowedTools.includes('text_editor')) {
+            // TextEditorTool doesn't need parameters
+            this.tools.set('text_editor', new TextEditorTool());
+        }
+        if (allowedTools.includes('search')) {
+            this.tools.set('search', new SearchTool());
+        }
+    }
+    /**
+     * Build system prompt for this subagent role
+     * Override in specialized subagent classes
+     */
+    buildSystemPrompt() {
+        return this.config.customSystemPrompt || `You are a specialized ${this.role} agent.`;
+    }
+    /**
+     * Execute a task
+     */
+    async executeTask(task) {
+        this.isActive = true;
+        this.currentTaskId = task.id;
+        const startTime = Date.now();
+        // Update status
+        this.status = {
+            id: this.id,
+            taskId: task.id,
+            role: this.role,
+            state: SubagentState.RUNNING,
+            progress: 0,
+            startTime: new Date(),
+            toolsUsed: [],
+            toolRoundsUsed: 0,
+        };
+        const result = {
+            id: this.id,
+            taskId: task.id,
+            role: this.role,
+            success: false,
+            output: '',
+            executionTime: 0,
+            status: this.status,
+            filesModified: [],
+            filesCreated: [],
+            toolCalls: [],
+        };
+        try {
+            // Emit both 'start' and 'task-started' for compatibility
+            this.emit('start', { taskId: task.id, role: this.role });
+            this.emit('task-started', { taskId: task.id, role: this.role });
+            // Add task context to messages
+            const contextPrompt = this.buildContextPrompt(task);
+            this.messages.push({
+                role: 'user',
+                content: contextPrompt,
+            });
+            // Add to chat history
+            this.chatHistory.push({
+                type: 'user',
+                content: contextPrompt,
+                timestamp: new Date(),
+            });
+            // Execute with tool rounds limit
+            let toolRounds = 0;
+            let isComplete = false;
+            while (toolRounds < this.config.maxToolRounds && !isComplete) {
+                const response = await this.llmClient.chat(this.messages, this.getToolDefinitions(), { stream: false });
+                if (!response.choices || response.choices.length === 0) {
+                    throw new Error('No response from LLM');
+                }
+                const choice = response.choices[0];
+                const message = choice.message;
+                // Handle content
+                if (message.content) {
+                    this.chatHistory.push({
+                        type: 'assistant',
+                        content: message.content,
+                        timestamp: new Date(),
+                    });
+                    result.output = message.content;
+                    this.emit('progress', {
+                        taskId: task.id,
+                        content: message.content,
+                    });
+                }
+                // Handle tool calls
+                if (message.tool_calls && message.tool_calls.length > 0) {
+                    this.messages.push({
+                        role: 'assistant',
+                        content: message.content || '',
+                        tool_calls: message.tool_calls,
+                    });
+                    for (const toolCall of message.tool_calls) {
+                        const toolResult = await this.executeToolCall(toolCall);
+                        // Track tool usage in status
+                        if (!this.status.toolsUsed.includes(toolCall.function.name)) {
+                            this.status.toolsUsed.push(toolCall.function.name);
+                        }
+                        // Track tool calls
+                        if (!result.toolCalls) {
+                            result.toolCalls = [];
+                        }
+                        result.toolCalls.push(toolCall);
+                        // Track files created/modified
+                        if (toolCall.function.name === 'text_editor' && toolResult.success) {
+                            const args = JSON.parse(toolCall.function.arguments);
+                            if (args.path) {
+                                if (args.command === 'create') {
+                                    if (!result.filesCreated.includes(args.path)) {
+                                        result.filesCreated.push(args.path);
+                                    }
+                                }
+                                else {
+                                    if (!result.filesModified.includes(args.path)) {
+                                        result.filesModified.push(args.path);
+                                    }
+                                }
+                            }
+                        }
+                        // Add tool result to messages
+                        this.messages.push({
+                            role: 'tool',
+                            tool_call_id: toolCall.id,
+                            content: toolResult.output || toolResult.error || '',
+                        });
+                        this.chatHistory.push({
+                            type: 'tool_result',
+                            content: toolResult.output || toolResult.error || '',
+                            timestamp: new Date(),
+                            toolCall: toolCall,
+                            toolResult: toolResult,
+                        });
+                        this.emit('tool-executed', {
+                            taskId: task.id,
+                            toolName: toolCall.function.name,
+                            result: toolResult,
+                        });
+                    }
+                    toolRounds++;
+                    this.status.toolRoundsUsed = toolRounds;
+                }
+                else {
+                    // No more tool calls, task complete
+                    isComplete = true;
+                }
+                // Check timeout
+                const elapsed = Date.now() - startTime;
+                if (this.config.timeout && elapsed > this.config.timeout) {
+                    throw new Error(`Task timeout after ${elapsed}ms`);
+                }
+            }
+            result.success = true;
+            result.executionTime = Date.now() - startTime;
+            this.status.state = SubagentState.COMPLETED;
+            this.status.endTime = new Date();
+            this.status.progress = 100;
+            result.status = { ...this.status };
+            this.emit('task-completed', { taskId: task.id, result });
+        }
+        catch (error) {
+            result.success = false;
+            result.error = error.message;
+            result.executionTime = Date.now() - startTime;
+            this.status.state = SubagentState.FAILED;
+            this.status.endTime = new Date();
+            this.status.error = error.message;
+            result.status = { ...this.status };
+            this.emit('task-failed', { taskId: task.id, error: error.message });
+        }
+        finally {
+            this.isActive = false;
+            this.currentTaskId = null;
+        }
+        return result;
+    }
+    /**
+     * Build context prompt from task
+     */
+    buildContextPrompt(task) {
+        let prompt = `Task: ${task.description}\n\n`;
+        if (task.context) {
+            prompt += `Context:\n`;
+            if (task.context.metadata?.workingDirectory) {
+                prompt += `- Working Directory: ${task.context.metadata.workingDirectory}\n`;
+            }
+            if (task.context.conversationHistory && task.context.conversationHistory.length > 0) {
+                prompt += `\nRecent conversation:\n`;
+                task.context.conversationHistory.slice(-5).forEach((entry) => {
+                    prompt += `${entry.type}: ${entry.content.substring(0, 200)}\n`;
+                });
+            }
+        }
+        return prompt;
+    }
+    /**
+     * Execute a tool call
+     */
+    async executeToolCall(toolCall) {
+        const toolName = toolCall.function.name;
+        const tool = this.tools.get(toolName);
+        if (!tool) {
+            return {
+                success: false,
+                error: `Tool '${toolName}' not available for ${this.role} agent`,
+            };
+        }
+        try {
+            const args = JSON.parse(toolCall.function.arguments);
+            const result = await tool.execute(args);
+            return result;
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Tool execution error: ${error.message}`,
+            };
+        }
+    }
+    /**
+     * Get tool definitions for LLM
+     */
+    getToolDefinitions() {
+        const definitions = [];
+        for (const [, tool] of this.tools) {
+            if (typeof tool.getToolDefinition === 'function') {
+                definitions.push(tool.getToolDefinition());
+            }
+        }
+        return definitions;
+    }
+    /**
+     * Get current status
+     */
+    getStatus() {
+        return { ...this.status };
+    }
+    /**
+     * Abort execution
+     */
+    abort() {
+        this.isActive = false;
+        this.status.state = SubagentState.CANCELLED;
+        this.status.endTime = new Date();
+        this.emit('cancel', { role: this.role });
+    }
+    /**
+     * Terminate subagent
+     */
+    async terminate() {
+        this.isActive = false;
+        this.currentTaskId = null;
+        this.removeAllListeners();
+        this.emit('terminated', { role: this.role });
+    }
+    /**
+     * Get chat history (for debugging/monitoring)
+     */
+    getChatHistory() {
+        return [...this.chatHistory];
+    }
+    /**
+     * Get logs (alias for getChatHistory for test compatibility)
+     */
+    getLogs() {
+        return [...this.chatHistory];
+    }
+    /**
+     * Receive a message from parent/orchestrator
+     */
+    async receiveMessage(message) {
+        this.emit('message', message);
+        // Handle cancellation messages
+        if (message.type === 'cancellation') {
+            this.abort();
+        }
+    }
+}
+//# sourceMappingURL=subagent.js.map

package/dist/agent/subagent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"subagent.js","sourceRoot":"","sources":["../../src/agent/subagent.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAS7C,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAElE,eAAe;AACf,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD;;GAEG;AACH,MAAM,OAAO,QAAS,SAAQ,YAAY;IACxB,EAAE,CAAS;IACX,IAAI,CAAe;IACnB,MAAM,CAAiB;IAC7B,SAAS,CAAY;IACrB,WAAW,CAAc;IACzB,QAAQ,CAAe;IACvB,KAAK,CAAmB;IACxB,QAAQ,CAAU;IAClB,aAAa,CAAgB;IAC7B,MAAM,CAAiB;IAEjC,YAAY,IAAkB,EAAE,eAAyC;QACvE,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,EAAE,GAAG,GAAG,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAC7E,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QAEjB,0CAA0C;QAC1C,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,aAAa,EAAE,GAAG,eAAe,EAAE,IAAI,EAAE,CAAC;QAE7D,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1B,oBAAoB;QACpB,IAAI,CAAC,MAAM,GAAG;YACZ,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,aAAa,CAAC,OAAO;YAC5B,QAAQ,EAAE,CAAC;YACX,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,yDAAyD;QACzD,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,CAC5B,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,EAAE,EAC9B,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,SAAS,EACnC,OAAO,CAAC,GAAG,CAAC,aAAa,CAC1B,CAAC;QAEF,2BAA2B;QAC3B,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,oBAAoB;QACpB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE;SAClC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,IAAkB;QACzC,2CAA2C;QAC3C,MAAM,YAAY,GAAkD;YAClE,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE;gBACtB,YAAY,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,QAAQ,CAAC;gBAC/C,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;YACD,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE;gBACtB,YAAY,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,QAAQ,CAAC;gBAC/C,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;YACD,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE;gBAC5B,YAAY,EAAE,CAAC,aAAa,EAAE,QAAQ,CAAC;gBACvC,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;YACD,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE;gBAC1B,YAAY,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,QAAQ,CAAC;gBAC/C,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;YACD,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE;gBACvB,YAAY,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,QAAQ,CAAC;gBAC/C,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;YACD,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE;gBACpB,YAAY,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,QAAQ,CAAC;gBAC/C,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;YACD,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE;gBAC1B,YAAY,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC;gBAChC,aAAa,EAAE,EAAE;gBACjB,YAAY,EAAE,EAAE;gBAChB,QAAQ,EAAE,CAAC;aACZ;SACF,CAAC;QAEF,OAAO;YACL,IAAI;YACJ,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,EAAE;YACjB,YAAY,EAAE,EAAE;YAChB,QAAQ,EAAE,CAAC;YACX,GAAG,YAAY,CAAC,IAAI,CAAC;SACJ,CAAC;IACtB,CAAC;IAED;;OAEG;IACK,eAAe;QACrB,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAE9C,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACzC,yCAAyC;YACzC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,EAAE,IAAI,cAAc,EAAE,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,UAAU,EAAE,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED;;;OAGG;IACO,iBAAiB;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,yBAAyB,IAAI,CAAC,IAAI,SAAS,CAAC;IACvF,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,IAAkB;QAClC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,CAAC;QAE7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,gBAAgB;QAChB,IAAI,CAAC,MAAM,GAAG;YACZ,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,aAAa,CAAC,OAAO;YAC5B,QAAQ,EAAE,CAAC;YACX,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,EAAE;YACb,cAAc,EAAE,CAAC;SAClB,CAAC;QAEF,MAAM,MAAM,GAAmB;YAC7B,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,EAAE;YACV,aAAa,EAAE,CAAC;YAChB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,aAAa,EAAE,EAAE;YACjB,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,EAAE;SACd,CAAC;QAEF,IAAI,CAAC;YACH,yDAAyD;YACzD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAEhE,+BAA+B;YAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACpD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,aAAa;aACvB,CAAC,CAAC;YAEH,sBAAsB;YACtB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,aAAa;gBACtB,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC,CAAC;YAEH,iCAAiC;YACjC,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,OAAO,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC7D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CACxC,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,kBAAkB,EAAE,EACzB,EAAE,MAAM,EAAE,KAAK,EAAE,CAClB,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvD,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;gBAC1C,CAAC;gBAED,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACnC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;gBAE/B,iBAAiB;gBACjB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBACpB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,SAAS,EAAE,IAAI,IAAI,EAAE;qBACtB,CAAC,CAAC;oBAEH,MAAM,CAAC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;oBAChC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;wBACpB,MAAM,EAAE,IAAI,CAAC,EAAE;wBACf,OAAO,EAAE,OAAO,CAAC,OAAO;qBACzB,CAAC,CAAC;gBACL,CAAC;gBAED,oBAAoB;gBACpB,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;wBAC9B,UAAU,EAAE,OAAO,CAAC,UAAU;qBAC/B,CAAC,CAAC;oBAEH,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;wBAC1C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;wBAExD,6BAA6B;wBAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;4BAC7D,IAAI,CAAC,MAAM,CAAC,SAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;wBACtD,CAAC;wBAED,mBAAmB;wBACnB,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;4BACtB,MAAM,CAAC,SAAS,GAAG,EAAE,CAAC;wBACxB,CAAC;wBACD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAEhC,+BAA+B;wBAC/B,IAAI,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,aAAa,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;4BACnE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;4BACrD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gCACd,IAAI,IAAI,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oCAC9B,IAAI,CAAC,MAAM,CAAC,YAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;wCAC9C,MAAM,CAAC,YAAa,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oCACvC,CAAC;gCACH,CAAC;qCAAM,CAAC;oCACN,IAAI,CAAC,MAAM,CAAC,aAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;wCAC/C,MAAM,CAAC,aAAc,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oCACxC,CAAC;gCACH,CAAC;4BACH,CAAC;wBACH,CAAC;wBAED,8BAA8B;wBAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;4BACjB,IAAI,EAAE,MAAM;4BACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;4BACzB,OAAO,EAAE,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,KAAK,IAAI,EAAE;yBACrD,CAAC,CAAC;wBAEH,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;4BACpB,IAAI,EAAE,aAAa;4BACnB,OAAO,EAAE,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,KAAK,IAAI,EAAE;4BACpD,SAAS,EAAE,IAAI,IAAI,EAAE;4BACrB,QAAQ,EAAE,QAAQ;4BAClB,UAAU,EAAE,UAAU;yBACvB,CAAC,CAAC;wBAEH,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;4BACzB,MAAM,EAAE,IAAI,CAAC,EAAE;4BACf,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;4BAChC,MAAM,EAAE,UAAU;yBACnB,CAAC,CAAC;oBACL,CAAC;oBAED,UAAU,EAAE,CAAC;oBACb,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,UAAU,CAAC;gBAC1C,CAAC;qBAAM,CAAC;oBACN,oCAAoC;oBACpC,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;gBAED,gBAAgB;gBAChB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBACvC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACzD,MAAM,IAAI,KAAK,CAAC,sBAAsB,OAAO,IAAI,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YAED,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC;YACtB,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC9C,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,SAAS,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC;YAC3B,MAAM,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAE3D,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;YACvB,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC;YAC7B,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC9C,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC;YAClC,MAAM,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACtE,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;YACtB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC5B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,IAAkB;QAC3C,IAAI,MAAM,GAAG,SAAS,IAAI,CAAC,WAAW,MAAM,CAAC;QAE7C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,YAAY,CAAC;YACvB,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,IAAI,wBAAwB,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,IAAI,CAAC;YAC/E,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,CAAC,mBAAmB,IAAI,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpF,MAAM,IAAI,0BAA0B,CAAC;gBACrC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,KAAgB,EAAE,EAAE;oBACtE,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC;gBAClE,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,eAAe,CAAC,QAAqB;QACjD,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,SAAS,QAAQ,uBAAuB,IAAI,CAAC,IAAI,QAAQ;aACjE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACxC,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,yBAAyB,KAAK,CAAC,OAAO,EAAE;aAChD,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,kBAAkB;QACxB,MAAM,WAAW,GAAU,EAAE,CAAC;QAE9B,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAClC,IAAI,OAAO,IAAI,CAAC,iBAAiB,KAAK,UAAU,EAAE,CAAC;gBACjD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,SAAS,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS;QACb,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QACtB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,OAMpB;QACC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAE9B,+BAA+B;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YACpC,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,CAAC;IACH,CAAC;CACF"}

package/dist/checkpoint/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Checkpoint System
+ *
+ * Exports all checkpoint-related functionality
+ */
+export type { Checkpoint, CheckpointConfig, CheckpointFilter, CheckpointInfo, CheckpointMetadata, CheckpointOptions, CheckpointRestoreResult, CheckpointStats, CheckpointIndex, FileSnapshot, } from './types.js';
+export { DEFAULT_CHECKPOINT_CONFIG } from './types.js';
+export { CheckpointStorage, calculateHash, verifyFileSnapshot } from './storage.js';
+export { CheckpointManager, getCheckpointManager, resetCheckpointManager } from './manager.js';

package/dist/checkpoint/index.js

@@ -0,0 +1,11 @@
+/**
+ * Checkpoint System
+ *
+ * Exports all checkpoint-related functionality
+ */
+export { DEFAULT_CHECKPOINT_CONFIG } from './types.js';
+// Storage
+export { CheckpointStorage, calculateHash, verifyFileSnapshot } from './storage.js';
+// Manager
+export { CheckpointManager, getCheckpointManager, resetCheckpointManager } from './manager.js';
+//# sourceMappingURL=index.js.map

package/dist/checkpoint/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/checkpoint/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgBH,OAAO,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AAEvD,UAAU;AACV,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAEpF,UAAU;AACV,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC"}

package/dist/checkpoint/manager.d.ts

@@ -0,0 +1,99 @@
+/**
+ * Checkpoint Manager
+ *
+ * High-level API for checkpoint operations:
+ * - Creating checkpoints
+ * - Listing checkpoints
+ * - Restoring checkpoints
+ * - Pruning old checkpoints
+ * - Automatic checkpoint scheduling
+ */
+import type { Checkpoint, CheckpointConfig, CheckpointInfo, CheckpointOptions, CheckpointRestoreResult, CheckpointStats, CheckpointFilter } from './types.js';
+import type { ChatEntry } from '../agent/llm-agent.js';
+export declare class CheckpointManager {
+    private storage;
+    private config;
+    constructor(config?: Partial<CheckpointConfig>, baseDir?: string);
+    initialize(): Promise<void>;
+    /**
+     * Create a new checkpoint
+     */
+    createCheckpoint(options: CheckpointOptions): Promise<Checkpoint>;
+    /**
+     * Restore a checkpoint
+     */
+    restoreCheckpoint(checkpointId: string): Promise<CheckpointRestoreResult>;
+    /**
+     * Apply a checkpoint (restore files and return conversation state)
+     * Alias for restoreCheckpoint for backward compatibility
+     */
+    applyCheckpoint(checkpointId: string): Promise<CheckpointRestoreResult>;
+    /**
+     * List checkpoints with optional filtering
+     */
+    listCheckpoints(filter?: CheckpointFilter): Promise<CheckpointInfo[]>;
+    /**
+     * Get checkpoint by ID
+     */
+    getCheckpoint(checkpointId: string): Promise<Checkpoint | null>;
+    /**
+     * Get checkpoint info
+     */
+    getCheckpointInfo(checkpointId: string): Promise<CheckpointInfo | null>;
+    /**
+     * Delete a checkpoint
+     */
+    deleteCheckpoint(checkpointId: string): Promise<void>;
+    /**
+     * Get checkpoint statistics
+     */
+    getStats(): Promise<CheckpointStats>;
+    /**
+     * Prune old checkpoints based on config
+     */
+    pruneOldCheckpoints(): Promise<number>;
+    /**
+     * Compress old checkpoints based on config
+     */
+    compressOldCheckpoints(): Promise<number>;
+    /**
+     * Run maintenance (pruning, compression, limit enforcement)
+     */
+    runMaintenance(): Promise<void>;
+    /**
+     * Enforce maximum number of checkpoints
+     */
+    private enforceCheckpointLimit;
+    /**
+     * Enforce storage size limit (in MB)
+     */
+    private enforceStorageLimit;
+    /**
+     * Get conversation state from a checkpoint
+     */
+    getConversationState(checkpointId: string): Promise<ChatEntry[] | null>;
+    /**
+     * Check if checkpoint should be created before an operation
+     */
+    shouldCreateCheckpoint(operation: string): boolean;
+    /**
+     * Update configuration
+     */
+    updateConfig(config: Partial<CheckpointConfig>): void;
+    /**
+     * Get current configuration
+     */
+    getConfig(): CheckpointConfig;
+}
+/**
+ * Get the singleton checkpoint manager instance
+ */
+export declare function getCheckpointManager(config?: Partial<CheckpointConfig>, baseDir?: string): CheckpointManager;
+/**
+ * Initialize a new checkpoint manager instance (creates new singleton)
+ */
+export declare function initCheckpointManager(config?: Partial<CheckpointConfig>, baseDir?: string): CheckpointManager;
+/**
+ * Reset the singleton instance (for testing)
+ */
+export declare function resetCheckpointManager(): void;