wave-agent-sdk 0.7.2 → 0.8.1

package/dist/agent.js

@@ -1,11 +1,8 @@
 import { LspManager } from "./managers/lspManager.js";
 import { configValidator } from "./utils/configValidator.js";
-import { loadSessionFromJsonl, handleSessionRestoration, } from "./services/session.js";
-import { setGlobalLogger } from "./utils/globalLogger.js";
+import { InitializationService } from "./services/initializationService.js";
+import { InteractionService } from "./services/interactionService.js";
 import { ConfigurationService } from "./services/configurationService.js";
-import * as fs from "fs/promises";
-import path from "path";
-import os from "os";
 import { setupAgentContainer } from "./utils/containerSetup.js";
 export class Agent {
     // Dynamic configuration getter methods
@@ -13,7 +10,7 @@ export class Agent {
         return this.configurationService.resolveGatewayConfig(this.options.apiKey, this.options.baseURL, this.options.defaultHeaders, this.options.fetchOptions, this.options.fetch);
     }
     getModelConfig() {
-        return this.configurationService.resolveModelConfig(this.options.agentModel, this.options.fastModel, this.options.maxTokens, this.getPermissionMode());
+        return this.configurationService.resolveModelConfig(this.options.model, this.options.fastModel, this.options.maxTokens, this.getPermissionMode());
     }
     getMaxInputTokens() {
         return this.configurationService.resolveMaxInputTokens(this.options.maxInputTokens);
@@ -31,8 +28,7 @@ export class Agent {
      * @param options - Configuration options for the Agent instance
      */
     constructor(options) {
-        this.bashManager = null;
-        this._usages = []; // Usage tracking array
+        this.bangManager = null;
         // Memory content storage
         this._projectMemoryContent = "";
         this._userMemoryContent = "";
@@ -52,26 +48,23 @@ export class Agent {
             configurationService: this.configurationService,
             systemPrompt: this.systemPrompt,
             stream: this.stream,
-            onSessionIdChange: () => {
-                this.taskManager.setTaskListId(this.messageManager.getRootSessionId());
-            },
-            onSessionTasksChange: (tasks) => {
-                this.options.callbacks?.onSessionTasksChange?.(tasks);
-            },
             onTasksChange: (tasks) => {
                 this.options.callbacks?.onTasksChange?.(tasks);
             },
+            onBackgroundTasksChange: (tasks) => {
+                this.options.callbacks?.onBackgroundTasksChange?.(tasks);
+            },
             onPermissionModeChange: (mode) => {
                 this.options.callbacks?.onPermissionModeChange?.(mode);
             },
             handlePlanModeTransition: (mode) => {
-                this.handlePlanModeTransition(mode);
+                this.planManager.handlePlanModeTransition(mode);
             },
             setPermissionMode: (mode) => {
                 this.setPermissionMode(mode);
             },
             addPermissionRule: (rule) => this.addPermissionRule(rule),
-            addUsage: (usage) => this.addUsage(usage),
+            addUsage: (usage) => this.messageManager.addUsage(usage),
             getGatewayConfig: () => this.getGatewayConfig(),
             getModelConfig: () => this.getModelConfig(),
             getMaxInputTokens: () => this.getMaxInputTokens(),
@@ -96,7 +89,7 @@ export class Agent {
         this.aiManager = this.container.get("AIManager");
         this.slashCommandManager = this.container.get("SlashCommandManager");
         this.pluginManager = this.container.get("PluginManager");
-        this.bashManager = this.container.get("BashManager");
+        this.bangManager = this.container.get("BangManager");
         // Set initial permission mode if provided
         if (options.permissionMode) {
             this.setPermissionMode(options.permissionMode);
@@ -110,33 +103,11 @@ export class Agent {
         return this.messageManager.getMessages();
     }
     get usages() {
-        return [...this._usages]; // Return copy to prevent external modification
+        return this.messageManager.getUsages();
     }
     get sessionFilePath() {
         return this.messageManager.getTranscriptPath();
     }
-    /**
-     * Rebuild usage array from messages containing usage metadata
-     * Called during session restoration to reconstruct usage tracking
-     */
-    rebuildUsageFromMessages(messages) {
-        this._usages = [];
-        messages.forEach((message) => {
-            if (message.role === "assistant" && message.usage) {
-                this._usages.push(message.usage);
-            }
-        });
-        // Trigger callback after rebuilding usage array
-        this.messageManager.triggerUsageChange();
-    }
-    /**
-     * Add usage data to the tracking array and trigger callbacks
-     * @param usage Usage data from AI operations
-     */
-    addUsage(usage) {
-        this._usages.push(usage);
-        this.messageManager.triggerUsageChange();
-    }
     get latestTotalTokens() {
         return this.messageManager.getlatestTotalTokens();
     }
@@ -166,7 +137,7 @@ export class Agent {
     }
     /** Get bash command execution status */
     get isCommandRunning() {
-        return this.bashManager?.isCommandRunning ?? false;
+        return this.bangManager?.isCommandRunning ?? false;
     }
     /** Get background bash shell output */
     getBackgroundShellOutput(id, filter) {
@@ -247,171 +218,62 @@ export class Agent {
         // Validate resolved configuration
         configValidator.validateGatewayConfig(gatewayConfig);
         configValidator.validateMaxInputTokens(maxInputTokens);
-        configValidator.validateModelConfig(modelConfig.agentModel, modelConfig.fastModel);
+        configValidator.validateModelConfig(modelConfig.model, modelConfig.fastModel);
     }
     /** Private initialization method, handles async initialization logic */
     async initialize(options) {
-        // Initialize managers first
-        try {
-            // Initialize SkillManager
-            await this.skillManager.initialize();
-            // Initialize SubagentManager (load and cache configurations)
-            await this.subagentManager.initialize();
-            // Register managers in container for tool access
-            this.container.register("SubagentManager", this.subagentManager);
-            this.container.register("SkillManager", this.skillManager);
-            // Initialize built-in tools
-            this.toolManager.initializeBuiltInTools();
-            // Initialize plugins
-            await this.pluginManager.loadPlugins(this.options.plugins || []);
-        }
-        catch (error) {
-            this.logger?.error("Failed to initialize managers and tools:", error);
-            // Don't throw error to prevent app startup failure
-        }
-        // Initialize MCP servers with auto-connect
-        try {
-            await this.mcpManager.initialize(this.workdir, true);
-            if (this.lspManager instanceof LspManager) {
-                await this.lspManager.initialize(this.workdir);
-            }
-        }
-        catch (error) {
-            this.logger?.error("Failed to initialize MCP servers:", error);
-            // Don't throw error to prevent app startup failure
-        }
-        // Initialize hooks configuration
-        try {
-            // Load hooks configuration using ConfigurationService
-            this.logger?.debug("Loading hooks configuration...");
-            const configResult = await this.configurationService.loadMergedConfiguration(this.workdir);
-            this.hookManager.loadConfigurationFromWaveConfig(configResult.configuration);
-            // Update plugin manager with enabled plugins configuration
-            if (configResult.configuration?.enabledPlugins) {
-                this.pluginManager.updateEnabledPlugins(configResult.configuration.enabledPlugins);
-            }
-            this.logger?.debug("Hooks system initialized successfully");
-        }
-        catch (error) {
-            this.logger?.error("Failed to initialize hooks system:", error);
-            // Don't throw error to prevent app startup failure
-        }
-        // Resolve and validate configuration after loading settings.json
-        this.resolveAndValidateConfig();
-        // Set global logger for SDK-wide access before discovering rules
-        setGlobalLogger(this.logger || null);
-        // Discover modular memory rules
-        try {
-            await this.memoryRuleManager.discoverRules();
-        }
-        catch (error) {
-            this.logger?.error("Failed to discover memory rules:", error);
-        }
-        // Initialize live configuration reload
-        try {
-            this.logger?.debug("Initializing live configuration reload...");
-            await this.liveConfigManager.initialize();
-            this.logger?.debug("Live configuration reload initialized successfully");
-        }
-        catch (error) {
-            this.logger?.error("Failed to initialize live configuration reload:", error);
-            // Don't throw error to prevent app startup failure - continue without live reload
-        }
-        // Load memory files during initialization
-        try {
-            this.logger?.debug("Loading memory files...");
-            // Load project memory from AGENTS.md (bypass memory store for direct file access)
-            try {
-                const projectMemoryPath = path.join(this.workdir, "AGENTS.md");
-                this._projectMemoryContent = await fs.readFile(projectMemoryPath, "utf-8");
-                this.logger?.debug("Project memory loaded successfully");
-            }
-            catch (error) {
-                this._projectMemoryContent = "";
-                this.logger?.debug("Project memory file not found or unreadable, using empty content:", error instanceof Error ? error.message : String(error));
-            }
-            // Load user memory (bypass memory store for direct file access)
-            try {
-                const userMemoryPath = path.join(os.homedir(), ".wave", "AGENTS.md");
-                this._userMemoryContent = await fs.readFile(userMemoryPath, "utf-8");
-                this.logger?.debug("User memory loaded successfully");
-            }
-            catch (error) {
-                this._userMemoryContent = "";
-                this.logger?.debug("User memory file not found or unreadable, using empty content:", error instanceof Error ? error.message : String(error));
-            }
-            this.logger?.debug("Memory initialization completed");
-        }
-        catch (error) {
-            // Ensure memory is always initialized even if loading fails
-            this._projectMemoryContent = "";
-            this._userMemoryContent = "";
-            this.logger?.error("Failed to load memory files:", error);
-            // Don't throw error to prevent app startup failure
-        }
-        // Handle session restoration or set provided messages
-        if (options?.messages) {
-            // If messages are provided, use them directly (useful for testing)
-            this.messageManager.setMessages(options.messages);
-            // Rebuild usage array from restored messages
-            this.rebuildUsageFromMessages(options.messages);
-        }
-        else {
-            // Otherwise, handle session restoration
-            const sessionToRestore = await handleSessionRestoration(options?.restoreSessionId, options?.continueLastSession, this.messageManager.getWorkdir());
-            // Rebuild usage array from restored messages
-            this.rebuildUsageFromMessages(sessionToRestore?.messages || []);
-            if (sessionToRestore) {
-                this.messageManager.initializeFromSession(sessionToRestore);
-                // Update task manager with the root session ID to ensure continuity across compressions
-                this.taskManager.setTaskListId(sessionToRestore.rootSessionId || sessionToRestore.id);
-                // After session is initialized, load tasks for the session
-                const tasks = await this.taskManager.listTasks();
-                this.options.callbacks?.onSessionTasksChange?.(tasks);
-            }
-        }
+        await InitializationService.initialize({
+            skillManager: this.skillManager,
+            subagentManager: this.subagentManager,
+            container: this.container,
+            toolManager: this.toolManager,
+            pluginManager: this.pluginManager,
+            options: this.options,
+            slashCommandManager: this.slashCommandManager,
+            logger: this.logger,
+            mcpManager: this.mcpManager,
+            workdir: this.workdir,
+            lspManager: this.lspManager,
+            configurationService: this.configurationService,
+            hookManager: this.hookManager,
+            messageManager: this.messageManager,
+            memoryRuleManager: this.memoryRuleManager,
+            liveConfigManager: this.liveConfigManager,
+            taskManager: this.taskManager,
+            setProjectMemory: (content) => {
+                this._projectMemoryContent = content;
+            },
+            setUserMemory: (content) => {
+                this._userMemoryContent = content;
+            },
+            resolveAndValidateConfig: () => this.resolveAndValidateConfig(),
+        }, options);
     }
     /**
      * Restore a session by ID, switching to the target session without destroying the Agent instance
      * @param sessionId - The ID of the session to restore
      */
     async restoreSession(sessionId) {
-        // 1. Validation
-        if (!sessionId || sessionId === this.sessionId) {
-            return; // No-op if session ID is invalid or already current
-        }
-        // 2. Auto-save current session
-        try {
-            await this.messageManager.saveSession();
-        }
-        catch (error) {
-            this.logger?.warn("Failed to save current session before restore:", error);
-            // Continue with restoration even if save fails
-        }
-        // 3. Load target session
-        const sessionData = await loadSessionFromJsonl(sessionId, this.messageManager.getWorkdir());
-        if (!sessionData) {
-            throw new Error(`Session not found: ${sessionId}`);
-        }
-        // 4. Clean current state
-        this.abortMessage(); // Abort any running operations
-        this.subagentManager.cleanup(); // Clean up active subagents
-        // 5. Rebuild usage (in correct order)
-        this.rebuildUsageFromMessages(sessionData.messages);
-        // 6. Initialize session state last
-        this.messageManager.initializeFromSession(sessionData);
-        // Update task manager with the root session ID to ensure continuity across compressions
-        this.taskManager.setTaskListId(sessionData.rootSessionId || sessionData.id);
-        // 7. Load tasks for the restored session
-        const tasks = await this.taskManager.listTasks();
-        this.options.callbacks?.onSessionTasksChange?.(tasks);
+        await InteractionService.restoreSession({
+            messageManager: this.messageManager,
+            slashCommandManager: this.slashCommandManager,
+            hookManager: this.hookManager,
+            workdir: this.workdir,
+            configurationService: this.configurationService,
+            logger: this.logger,
+            aiManager: this.aiManager,
+            subagentManager: this.subagentManager,
+            taskManager: this.taskManager,
+            options: this.options,
+            abortMessage: () => this.abortMessage(),
+        }, sessionId);
     }
     abortAIMessage() {
         this.aiManager.abortAIMessage();
     }
     /** Execute bash command */
     async executeBashCommand(command) {
-        await this.bashManager?.executeCommand(command);
+        await this.bangManager?.executeCommand(command);
     }
     clearMessages() {
         this.messageManager.clearMessages();
@@ -424,7 +286,7 @@ export class Agent {
     }
     /** Interrupt bash command execution */
     abortBashCommand() {
-        this.bashManager?.abortCommand();
+        this.bangManager?.abortCommand();
     }
     /** Interrupt slash command execution */
     abortSlashCommand() {
@@ -514,65 +376,19 @@ export class Agent {
      * ```
      */
     async sendMessage(content, images) {
-        try {
-            // Handle slash command - check if it's a slash command (starts with /)
-            if (content.startsWith("/")) {
-                const command = content.trim();
-                if (!command || command === "/")
-                    return;
-                // Parse and validate slash command
-                const { isValid, commandId, args } = this.slashCommandManager.parseAndValidateSlashCommand(command);
-                if (isValid && commandId !== undefined) {
-                    // Execute valid slash command
-                    await this.slashCommandManager.executeCommand(commandId, args);
-                    return;
-                }
-                // If command doesn't exist, continue as normal message processing
-                // Don't add to history, let normal message processing logic below handle it
-            }
-            // Handle normal AI message
-            // Add user message first, will automatically sync to UI
-            this.messageManager.addUserMessage({
-                content,
-                images: images?.map((img) => ({
-                    path: img.path,
-                    mimeType: img.mimeType,
-                })),
-            });
-            // Execute UserPromptSubmit hooks after adding the user message
-            if (this.hookManager) {
-                try {
-                    const hookResults = await this.hookManager.executeHooks("UserPromptSubmit", {
-                        event: "UserPromptSubmit",
-                        projectDir: this.workdir,
-                        timestamp: new Date(),
-                        // UserPromptSubmit doesn't need toolName
-                        sessionId: this.sessionId,
-                        transcriptPath: this.messageManager.getTranscriptPath(),
-                        cwd: this.workdir,
-                        userPrompt: content,
-                        env: this.configurationService.getEnvironmentVars(), // Include configuration environment variables
-                    });
-                    // Process hook results and determine if we should continue
-                    const processResult = this.hookManager.processHookResults("UserPromptSubmit", hookResults, this.messageManager);
-                    // If hook processing indicates we should block (exit code 2), stop here
-                    if (processResult.shouldBlock) {
-                        this.logger?.info("UserPromptSubmit hook blocked prompt processing with error:", processResult.errorMessage);
-                        return; // Don't send to AI
-                    }
-                }
-                catch (error) {
-                    this.logger?.warn("UserPromptSubmit hooks execution failed:", error);
-                    // Continue processing even if hooks fail
-                }
-            }
-            // Send AI message
-            await this.aiManager.sendAIMessage();
-        }
-        catch (error) {
-            console.error("Failed to add user message:", error);
-            // Loading state will be automatically updated by the useEffect that watches messages
-        }
+        await InteractionService.sendMessage({
+            messageManager: this.messageManager,
+            slashCommandManager: this.slashCommandManager,
+            hookManager: this.hookManager,
+            workdir: this.workdir,
+            configurationService: this.configurationService,
+            logger: this.logger,
+            aiManager: this.aiManager,
+            subagentManager: this.subagentManager,
+            taskManager: this.taskManager,
+            options: this.options,
+            abortMessage: () => this.abortMessage(),
+        }, content, images);
     }
     // ========== MCP Management Methods ==========
     /** Get all MCP server status */
@@ -597,8 +413,10 @@ export class Agent {
         return this.slashCommandManager.hasCommand(commandId);
     }
     /** Reload custom commands */
-    reloadCustomCommands() {
+    async reloadCustomCommands() {
+        await this.skillManager.initialize();
         this.slashCommandManager.reloadCustomCommands();
+        this.slashCommandManager.registerSkillCommands(this.skillManager.getAvailableSkills());
     }
     /** Get custom command details */
     getCustomCommand(commandId) {
@@ -627,7 +445,7 @@ export class Agent {
     setPermissionMode(mode) {
         this.logger?.debug("Setting permission mode", { mode });
         this.toolManager.setPermissionMode(mode);
-        this.handlePlanModeTransition(mode);
+        this.planManager.handlePlanModeTransition(mode);
         this.options.callbacks?.onPermissionModeChange?.(mode);
     }
     /**
@@ -637,7 +455,9 @@ export class Agent {
     async truncateHistory(index) {
         await this.messageManager.truncateHistory(index, this.reversionManager);
         // After truncating history, the task list might have changed, so refresh it.
-        await this.taskManager.refreshTasks();
+        // We explicitly load tasks and trigger the callback to ensure UI updates immediately and in order.
+        const tasks = await this.taskManager.listTasks();
+        this.options.callbacks?.onTasksChange?.(tasks);
     }
     /**
      * Get the full message thread including parent sessions
@@ -652,9 +472,18 @@ export class Agent {
         return this.permissionManager.getPlanFilePath();
     }
     /**
-     * Get all currently allowed rules (for testing and UI)
+     * Get all currently allowed rules (user-defined and default)
      */
     getAllowedRules() {
+        return [
+            ...this.permissionManager.getAllowedRules(),
+            ...this.permissionManager.getDefaultAllowedRules(),
+        ];
+    }
+    /**
+     * Get only user-defined allowed rules
+     */
+    getUserAllowedRules() {
         return this.permissionManager.getAllowedRules();
     }
     /**
@@ -668,53 +497,7 @@ export class Agent {
      * @param rule - The rule to add (e.g., "Bash(ls)")
      */
     async addPermissionRule(rule) {
-        // 1. Expand rule if it's a Bash command
-        let rulesToAdd = [rule];
-        const bashMatch = rule.match(/^Bash\((.*)\)$/);
-        if (bashMatch) {
-            const command = bashMatch[1];
-            rulesToAdd = this.permissionManager.expandBashRule(command, this.workdir);
-        }
-        for (const ruleToAdd of rulesToAdd) {
-            // 2. Update PermissionManager state
-            const currentRules = this.permissionManager.getAllowedRules();
-            if (!currentRules.includes(ruleToAdd)) {
-                this.permissionManager.updateAllowedRules([...currentRules, ruleToAdd]);
-                // 3. Persist to settings.local.json
-                try {
-                    await this.configurationService.addAllowedRule(this.workdir, ruleToAdd);
-                    this.logger?.debug("Persistent permission rule added", {
-                        rule: ruleToAdd,
-                    });
-                }
-                catch (error) {
-                    this.logger?.error("Failed to persist permission rule", {
-                        rule: ruleToAdd,
-                        error: error instanceof Error ? error.message : String(error),
-                    });
-                }
-            }
-        }
-    }
-    /**
-     * Handle plan mode transition, generating or clearing plan file path
-     * @param mode - The current effective permission mode
-     */
-    handlePlanModeTransition(mode) {
-        if (mode === "plan") {
-            this.planManager
-                .getOrGeneratePlanFilePath(this.messageManager.getRootSessionId())
-                .then(({ path }) => {
-                this.logger?.debug("Plan file path generated", { path });
-                this.permissionManager.setPlanFilePath(path);
-            })
-                .catch((error) => {
-                this.logger?.error("Failed to generate plan file path", error);
-            });
-        }
-        else {
-            this.permissionManager.setPlanFilePath(undefined);
-        }
+        await this.permissionManager.addPermissionRule(rule);
     }
     /**
      * Get the current task list ID

package/dist/index.d.ts

@@ -15,5 +15,7 @@ export * from "./utils/stringUtils.js";
 export * from "./utils/customCommands.js";
 export * from "./utils/hookMatcher.js";
 export * from "./utils/tokenCalculation.js";
+export * from "./utils/gitUtils.js";
+export * from "./utils/nameGenerator.js";
 export * from "./types/index.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,mBAAmB,CAAC;AAGlC,cAAc,sBAAsB,CAAC;AAGrC,cAAc,YAAY,CAAC;AAC3B,cAAc,kBAAkB,CAAC;AAGjC,cAAc,uBAAuB,CAAC;AACtC,cAAc,kCAAkC,CAAC;AACjD,cAAc,uBAAuB,CAAC;AACtC,cAAc,uBAAuB,CAAC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,qBAAqB,CAAC;AACpC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,iBAAiB,CAAC;AAChC,cAAc,0BAA0B,CAAC;AACzC,cAAc,wBAAwB,CAAC;AACvC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,wBAAwB,CAAC;AACvC,cAAc,6BAA6B,CAAC;AAG5C,cAAc,kBAAkB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,mBAAmB,CAAC;AAGlC,cAAc,sBAAsB,CAAC;AAGrC,cAAc,YAAY,CAAC;AAC3B,cAAc,kBAAkB,CAAC;AAGjC,cAAc,uBAAuB,CAAC;AACtC,cAAc,kCAAkC,CAAC;AACjD,cAAc,uBAAuB,CAAC;AACtC,cAAc,uBAAuB,CAAC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,qBAAqB,CAAC;AACpC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,iBAAiB,CAAC;AAChC,cAAc,0BAA0B,CAAC;AACzC,cAAc,wBAAwB,CAAC;AACvC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,wBAAwB,CAAC;AACvC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,qBAAqB,CAAC;AACpC,cAAc,0BAA0B,CAAC;AACzC,cAAc,kBAAkB,CAAC"}

package/dist/index.js

@@ -19,5 +19,6 @@ export * from "./utils/stringUtils.js";
 export * from "./utils/customCommands.js";
 export * from "./utils/hookMatcher.js";
 export * from "./utils/tokenCalculation.js";
-// Export types
+export * from "./utils/gitUtils.js";
+export * from "./utils/nameGenerator.js";
 export * from "./types/index.js";

package/dist/managers/aiManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aiManager.d.ts","sourceRoot":"","sources":["../../src/managers/aiManager.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAY3E,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAIlD,MAAM,WAAW,kBAAkB;IACjC,wBAAwB,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;IAC5D,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACvC;AAED,MAAM,WAAW,gBAAgB;IAC/B,SAAS,CAAC,EAAE,kBAAkB,CAAC;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,uEAAuE;IACvE,MAAM,CAAC,EAAE,OAAO,CAAC;IAEjB,gBAAgB,EAAE,MAAM,aAAa,CAAC;IACtC,cAAc,EAAE,MAAM,WAAW,CAAC;IAClC,iBAAiB,EAAE,MAAM,MAAM,CAAC;IAChC,WAAW,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;IACtC,kBAAkB,CAAC,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACnD;AAED,qBAAa,SAAS;IAkBlB,OAAO,CAAC,SAAS;IAjBZ,SAAS,EAAE,OAAO,CAAS;IAClC,OAAO,CAAC,eAAe,CAAgC;IACvD,OAAO,CAAC,mBAAmB,CAAgC;IAC3D,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,MAAM,CAAU;IAGxB,OAAO,CAAC,kBAAkB,CAAsB;IAChD,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,mBAAmB,CAAe;IAC1C,OAAO,CAAC,aAAa,CAA2B;IAChD,OAAO,CAAC,oBAAoB,CAAC,CAA+B;gBAIlD,SAAS,EAAE,SAAS,EAC5B,OAAO,EAAE,gBAAgB;IAgB3B,OAAO,KAAK,WAAW,GAEtB;IAED,OAAO,KAAK,cAAc,GAEzB;IAED,OAAO,KAAK,WAAW,GAItB;IAED,OAAO,KAAK,qBAAqB,GAEhC;IAED,OAAO,KAAK,WAAW,GAEtB;IAED,OAAO,KAAK,gBAAgB,GAM3B;IAED,OAAO,KAAK,iBAAiB,GAE5B;IAGM,gBAAgB,IAAI,aAAa;IAIjC,cAAc,IAAI,WAAW;IAI7B,iBAAiB,IAAI,MAAM;IAI3B,WAAW,IAAI,MAAM,GAAG,SAAS;IAIxC,OAAO,CAAC,aAAa,CAAkB;IACvC,OAAO,CAAC,SAAS,CAAqB;IAEtC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAYvB,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,IAAI;IAItC,cAAc,IAAI,IAAI;IAuB7B,OAAO,CAAC,qBAAqB;YAsBf,8BAA8B;IA6ErC,gBAAgB,IAAI,OAAO;IAI3B,gBAAgB,CAAC,aAAa,EAAE,OAAO,GAAG,IAAI;IAOrD,OAAO,KAAK,eAAe,GAE1B;IAED,OAAO,KAAK,YAAY,GAEvB;IAEY,aAAa,CACxB,OAAO,GAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,oEAAoE;QACpE,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,iEAAiE;QACjE,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;KACf,GACL,OAAO,CAAC,IAAI,CAAC;IAugBhB;;;;OAIG;YACW,gBAAgB;IAiE9B;;;OAGG;YACW,sBAAsB;IA6DpC;;OAEG;YACW,uBAAuB;CAwDtC"}
+{"version":3,"file":"aiManager.d.ts","sourceRoot":"","sources":["../../src/managers/aiManager.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAY3E,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAIlD,MAAM,WAAW,kBAAkB;IACjC,wBAAwB,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;IAC5D,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACvC;AAED,MAAM,WAAW,gBAAgB;IAC/B,SAAS,CAAC,EAAE,kBAAkB,CAAC;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,uEAAuE;IACvE,MAAM,CAAC,EAAE,OAAO,CAAC;IAEjB,gBAAgB,EAAE,MAAM,aAAa,CAAC;IACtC,cAAc,EAAE,MAAM,WAAW,CAAC;IAClC,iBAAiB,EAAE,MAAM,MAAM,CAAC;IAChC,WAAW,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;IACtC,kBAAkB,CAAC,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACnD;AAED,qBAAa,SAAS;IAkBlB,OAAO,CAAC,SAAS;IAjBZ,SAAS,EAAE,OAAO,CAAS;IAClC,OAAO,CAAC,eAAe,CAAgC;IACvD,OAAO,CAAC,mBAAmB,CAAgC;IAC3D,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,MAAM,CAAU;IAGxB,OAAO,CAAC,kBAAkB,CAAsB;IAChD,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,mBAAmB,CAAe;IAC1C,OAAO,CAAC,aAAa,CAA2B;IAChD,OAAO,CAAC,oBAAoB,CAAC,CAA+B;gBAIlD,SAAS,EAAE,SAAS,EAC5B,OAAO,EAAE,gBAAgB;IAgB3B,OAAO,KAAK,WAAW,GAEtB;IAED,OAAO,KAAK,cAAc,GAEzB;IAED,OAAO,KAAK,WAAW,GAItB;IAED,OAAO,KAAK,qBAAqB,GAEhC;IAED,OAAO,KAAK,WAAW,GAEtB;IAED,OAAO,KAAK,gBAAgB,GAM3B;IAED,OAAO,KAAK,iBAAiB,GAE5B;IAGM,gBAAgB,IAAI,aAAa;IAIjC,cAAc,IAAI,WAAW;IAI7B,iBAAiB,IAAI,MAAM;IAI3B,WAAW,IAAI,MAAM,GAAG,SAAS;IAIxC,OAAO,CAAC,aAAa,CAAkB;IACvC,OAAO,CAAC,SAAS,CAAqB;IAEtC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAsBvB,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,IAAI;IAItC,cAAc,IAAI,IAAI;IAuB7B,OAAO,CAAC,qBAAqB;YAsBf,8BAA8B;IA6ErC,gBAAgB,IAAI,OAAO;IAI3B,gBAAgB,CAAC,aAAa,EAAE,OAAO,GAAG,IAAI;IAOrD,OAAO,KAAK,eAAe,GAE1B;IAED,OAAO,KAAK,YAAY,GAEvB;IAEY,aAAa,CACxB,OAAO,GAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,oEAAoE;QACpE,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,iEAAiE;QACjE,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;KACf,GACL,OAAO,CAAC,IAAI,CAAC;IAmgBhB;;;;OAIG;YACW,gBAAgB;IAiE9B;;;OAGG;YACW,sBAAsB;IA6DpC;;OAEG;YACW,uBAAuB;CAwDtC"}

package/dist/managers/aiManager.js

@@ -62,7 +62,16 @@ export class AIManager {
      * Get filtered tool configuration based on tools list
      */
     getFilteredToolsConfig(tools) {
-        const allTools = this.toolManager.getToolsConfig();
+        // Get available subagents and skills for dynamic prompts
+        const availableSubagents = this.subagentManager?.getConfigurations();
+        const availableSkills = this.skillManager
+            ?.getAvailableSkills()
+            .filter((skill) => !skill.disableModelInvocation);
+        const allTools = this.toolManager.getToolsConfig({
+            availableSubagents,
+            availableSkills,
+            workdir: this.workdir,
+        });
         // If no tools specified, return all tools
         if (!tools || tools.length === 0) {
             return allTools;
@@ -149,7 +158,7 @@ export class AIManager {
                             prompt_tokens: compressionResult.usage.prompt_tokens,
                             completion_tokens: compressionResult.usage.completion_tokens,
                             total_tokens: compressionResult.usage.total_tokens,
-                            model: model || this.getModelConfig().agentModel,
+                            model: model || this.getModelConfig().model,
                             operation_type: "compress",
                         };
                     }
@@ -246,9 +255,6 @@ export class AIManager {
                     planModeOptions = { planFilePath, planExists };
                 }
             }
-            // Get available subagents and skills for dynamic prompts
-            const availableSubagents = this.subagentManager?.getConfigurations();
-            const availableSkills = this.skillManager?.getAvailableSkills();
             // Call AI service with streaming callbacks if enabled
             const callAgentOptions = {
                 gatewayConfig: this.getGatewayConfig(),
@@ -265,8 +271,6 @@ export class AIManager {
                     language: this.getLanguage(),
                     isSubagent: !!this.subagentType,
                     planMode: planModeOptions,
-                    availableSubagents,
-                    availableSkills,
                 }), // Pass custom system prompt
                 maxTokens: maxTokens, // Pass max tokens override
             };
@@ -347,7 +351,7 @@ export class AIManager {
                     prompt_tokens: result.usage.prompt_tokens,
                     completion_tokens: result.usage.completion_tokens,
                     total_tokens: result.usage.total_tokens,
-                    model: model || this.getModelConfig().agentModel,
+                    model: model || this.getModelConfig().model,
                     operation_type: "agent",
                     // Preserve cache fields if present
                     ...(result.usage.cache_read_input_tokens !== undefined && {
@@ -383,9 +387,6 @@ export class AIManager {
                     }
                 }
             }
-            if (result.finish_reason === "length" && toolCalls.length === 0) {
-                this.messageManager.addErrorBlock("AI response was truncated due to length limit. Please try to reduce the complexity of your request or split it into smaller parts.");
-            }
             if (toolCalls.length > 0) {
                 // Execute all tools in parallel using Promise.all
                 const toolExecutionPromises = toolCalls.map(async (functionToolCall) => {
@@ -503,8 +504,8 @@ export class AIManager {
             }
             // Handle token statistics and message compression
             await this.handleTokenUsageAndCompression(result.usage, abortController, model);
-            // Check if there are tool operations, if so automatically initiate next AI service call
-            if (toolCalls.length > 0) {
+            // Check if there are tool operations or response was truncated, if so automatically initiate next AI service call
+            if (toolCalls.length > 0 || result.finish_reason === "length") {
                 // Record committed snapshots to message history
                 if (this.reversionManager) {
                     const snapshots = this.reversionManager.getAndClearCommittedSnapshots();
@@ -523,6 +524,12 @@ export class AIManager {
                     logger?.info("Some tools were manually backgrounded, stopping recursion.");
                 }
                 else if (!isCurrentlyAborted) {
+                    // If response was truncated and no tools were called, add a continuation message
+                    if (result.finish_reason === "length" && toolCalls.length === 0) {
+                        this.messageManager.addUserMessage({
+                            content: "Your response was cut off because it exceeded the output token limit. Please break your work into smaller pieces. Continue from where you left off.",
+                        });
+                    }
                     // Recursively call AI service, increment recursion depth, and pass same configuration
                     await this.sendAIMessage({
                         recursionDepth: recursionDepth + 1,

package/dist/managers/backgroundTaskManager.d.ts

@@ -2,7 +2,7 @@ import { type ChildProcess } from "child_process";
 import { BackgroundTask } from "../types/processes.js";
 import { Container } from "../utils/container.js";
 export interface BackgroundTaskManagerCallbacks {
-    onTasksChange?: (tasks: BackgroundTask[]) => void;
+    onBackgroundTasksChange?: (tasks: BackgroundTask[]) => void;
 }
 export interface BackgroundTaskManagerOptions {
     callbacks?: BackgroundTaskManagerCallbacks;