wave-agent-sdk 0.14.3 → 0.14.4

package/builtin/skills/settings/SKILLS.md

@@ -42,6 +42,9 @@ When this skill is invoked, follow these steps:
 - `allowed-tools`: (Optional) List of tools the skill can use.
 - `context: fork`: (Optional) Run the skill in a separate subagent.
 - `agent`: (Optional) Specify the subagent type (default: `general-purpose`).
+- `disable-model-invocation`: (Optional, default: `false`) Set to `true` to hide the skill from the AI's available skills list. The skill can still be invoked by users via slash commands.
+- `user-invocable`: (Optional, default: `true`) Set to `false` to hide the skill from the `/` slash command menu. The AI can still invoke it unless `disable-model-invocation` is also set.
+- `model`: (Optional) Override the AI model used for skill execution (e.g., `"gpt-4o"`, `"o3-mini"`).
 
 ## Skill Locations
 

package/dist/managers/aiManager.d.ts

@@ -14,6 +14,8 @@ export interface AIManagerOptions {
     stream?: boolean;
     /**Optional model override (e.g. for subagents) */
     modelOverride?: string;
+    /**Optional max turns limit to prevent runaway recursion (e.g. for auto-memory extraction) */
+    maxTurns?: number;
 }
 export declare class AIManager {
     private container;
@@ -29,6 +31,7 @@ export declare class AIManager {
     private modelOverride?;
     private _onCwdChange?;
     private consecutiveCompactionFailures;
+    private readonly maxTurns?;
     constructor(container: Container, options: AIManagerOptions);
     private get toolManager();
     private get messageManager();

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

@@ -1 +1 @@
-{"version":3,"file":"aiManager.d.ts","sourceRoot":"","sources":["../../src/managers/aiManager.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,KAAK,EAGN,MAAM,mBAAmB,CAAC;AAY3B,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAMlD,MAAM,WAAW,kBAAkB;IACjC,uBAAuB,CAAC,EAAE,CAAC,YAAY,EAAE,OAAO,KAAK,IAAI,CAAC;IAC1D,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACtC,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;CACxC;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;IACjB,kDAAkD;IAClD,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,qBAAa,SAAS;IAgBlB,OAAO,CAAC,SAAS;IAfZ,SAAS,EAAE,OAAO,CAAS;IAClC,OAAO,CAAC,eAAe,CAAgC;IACvD,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC7C,OAAO,CAAC,mBAAmB,CAAgC;IAC3D,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,MAAM,CAAU;IACxB,OAAO,CAAC,aAAa,CAAC,CAAS;IAC/B,OAAO,CAAC,YAAY,CAAC,CAA2B;IAChD,OAAO,CAAC,6BAA6B,CAAa;gBAIxC,SAAS,EAAE,SAAS,EAC5B,OAAO,EAAE,gBAAgB;IAY3B,OAAO,KAAK,WAAW,GAEtB;IAED,OAAO,KAAK,cAAc,GAEzB;IAED,OAAO,KAAK,aAAa,GAIxB;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;IAED,OAAO,KAAK,oBAAoB,GAE/B;IAGM,gBAAgB,IAAI,aAAa;IAIjC,cAAc,IAAI,WAAW;IA6B7B,iBAAiB,IAAI,MAAM;IAI3B,WAAW,IAAI,MAAM,GAAG,SAAS;IAIjC,oBAAoB,IAAI,OAAO;IAI/B,UAAU,IAAI,MAAM;IAIpB,kBAAkB,IAAI,MAAM;IAI5B,cAAc,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAI/D,OAAO,CAAC,YAAY,CAAkB;IACtC,OAAO,CAAC,SAAS,CAAqB;IAEtC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAevB,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,IAAI;IAUtC,cAAc,IAAI,IAAI;IAuB7B,OAAO,CAAC,qBAAqB;YAuBf,6BAA6B;IAwPpC,eAAe,IAAI,OAAO;IAI1B,eAAe,CAAC,YAAY,EAAE,OAAO,GAAG,IAAI;IAOnD,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,SAAS,CAAC,EAAE,MAAM,CAAC;KACf,GACL,OAAO,CAAC,IAAI,CAAC;IAopBhB;;;;OAIG;YACW,gBAAgB;IAkF9B;;;OAGG;YACW,sBAAsB;IA+DpC;;OAEG;YACW,uBAAuB;CA0DtC"}
+{"version":3,"file":"aiManager.d.ts","sourceRoot":"","sources":["../../src/managers/aiManager.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,KAAK,EAGN,MAAM,mBAAmB,CAAC;AAY3B,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAOlD,MAAM,WAAW,kBAAkB;IACjC,uBAAuB,CAAC,EAAE,CAAC,YAAY,EAAE,OAAO,KAAK,IAAI,CAAC;IAC1D,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACtC,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;CACxC;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;IACjB,kDAAkD;IAClD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,6FAA6F;IAC7F,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,SAAS;IAiBlB,OAAO,CAAC,SAAS;IAhBZ,SAAS,EAAE,OAAO,CAAS;IAClC,OAAO,CAAC,eAAe,CAAgC;IACvD,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC7C,OAAO,CAAC,mBAAmB,CAAgC;IAC3D,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,MAAM,CAAU;IACxB,OAAO,CAAC,aAAa,CAAC,CAAS;IAC/B,OAAO,CAAC,YAAY,CAAC,CAA2B;IAChD,OAAO,CAAC,6BAA6B,CAAa;IAClD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAS;gBAIzB,SAAS,EAAE,SAAS,EAC5B,OAAO,EAAE,gBAAgB;IAa3B,OAAO,KAAK,WAAW,GAEtB;IAED,OAAO,KAAK,cAAc,GAEzB;IAED,OAAO,KAAK,aAAa,GAIxB;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;IAED,OAAO,KAAK,oBAAoB,GAE/B;IAGM,gBAAgB,IAAI,aAAa;IAIjC,cAAc,IAAI,WAAW;IA6B7B,iBAAiB,IAAI,MAAM;IAI3B,WAAW,IAAI,MAAM,GAAG,SAAS;IAIjC,oBAAoB,IAAI,OAAO;IAI/B,UAAU,IAAI,MAAM;IAIpB,kBAAkB,IAAI,MAAM;IAI5B,cAAc,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAI/D,OAAO,CAAC,YAAY,CAAkB;IACtC,OAAO,CAAC,SAAS,CAAqB;IAEtC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAevB,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,IAAI;IAUtC,cAAc,IAAI,IAAI;IAuB7B,OAAO,CAAC,qBAAqB;YAuBf,6BAA6B;IAwPpC,eAAe,IAAI,OAAO;IAI1B,eAAe,CAAC,YAAY,EAAE,OAAO,GAAG,IAAI;IAOnD,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,SAAS,CAAC,EAAE,MAAM,CAAC;KACf,GACL,OAAO,CAAC,IAAI,CAAC;IA8qBhB;;;;OAIG;YACW,gBAAgB;IAkF9B;;;OAGG;YACW,sBAAsB;IA+DpC;;OAEG;YACW,uBAAuB;CA0DtC"}

package/dist/managers/aiManager.js

@@ -5,6 +5,7 @@ import { parseTaskNotificationXml } from "../utils/notificationXml.js";
 import { calculateComprehensiveTotalTokens } from "../utils/tokenCalculation.js";
 import * as fs from "node:fs/promises";
 import { buildSystemPrompt } from "../prompts/index.js";
+import { recoverTruncatedJson } from "../utils/stringUtils.js";
 import { logger } from "../utils/globalLogger.js";
 export class AIManager {
     // Service overrides
@@ -23,6 +24,7 @@ export class AIManager {
         this.callbacks = options.callbacks ?? {};
         this.modelOverride = options.modelOverride;
         this._onCwdChange = options.callbacks?.onCwdChange; // Initialize onCwdChange
+        this.maxTurns = options.maxTurns;
     }
     get toolManager() {
         return this.container.get("ToolManager");
@@ -584,35 +586,45 @@ export class AIManager {
                     const toolName = functionToolCall.function?.name || "";
                     // Safely parse tool parameters, handle tools without parameters
                     let toolArgs = {};
+                    let jsonRecovered = false;
                     const argsString = functionToolCall.function?.arguments?.trim();
                     if (!argsString || argsString === "") {
                         // Tool without parameters, use empty object
                         toolArgs = {};
                     }
                     else {
+                        let recoveredArgs = argsString;
                         try {
                             toolArgs = JSON.parse(argsString);
                         }
-                        catch (parseError) {
-                            // For non-empty but malformed JSON, still throw exception
-                            let errorMessage = `Failed to parse tool arguments`;
-                            if (result.finish_reason === "length") {
-                                errorMessage +=
-                                    " (output truncated, please reduce your output)";
+                        catch {
+                            // Attempt to recover truncated JSON (e.g., missing closing braces)
+                            recoveredArgs = recoverTruncatedJson(argsString);
+                            try {
+                                toolArgs = JSON.parse(recoveredArgs);
+                                jsonRecovered = true;
+                                logger.warn(`Recovered truncated JSON for tool "${toolName}"`);
+                            }
+                            catch (parseError) {
+                                let errorMessage = `Failed to parse tool arguments`;
+                                if (result.finish_reason === "length") {
+                                    errorMessage +=
+                                        " (output truncated, please reduce your output)";
+                                }
+                                logger?.error(errorMessage, parseError);
+                                this.messageManager.updateToolBlock({
+                                    id: toolId,
+                                    parameters: argsString,
+                                    result: errorMessage,
+                                    success: false,
+                                    error: errorMessage,
+                                    stage: "end",
+                                    name: toolName,
+                                    compactParams: "",
+                                    timestamp: Date.now(),
+                                });
+                                return;
                             }
-                            logger?.error(errorMessage, parseError);
-                            this.messageManager.updateToolBlock({
-                                id: toolId,
-                                parameters: argsString,
-                                result: errorMessage,
-                                success: false,
-                                error: errorMessage,
-                                stage: "end",
-                                name: toolName,
-                                compactParams: "",
-                                timestamp: Date.now(),
-                            });
-                            return;
                         }
                     }
                     const compactParams = this.generateCompactParams(toolName, toolArgs);
@@ -681,12 +693,18 @@ export class AIManager {
                         };
                         // Execute tool
                         const toolResult = await this.toolManager.execute(functionToolCall.function?.name || "", toolArgs, context);
+                        // Build result content, adding truncation warning if JSON was recovered
+                        let toolResultContent = toolResult.content ||
+                            (toolResult.error ? `Error: ${toolResult.error}` : "");
+                        if (jsonRecovered) {
+                            toolResultContent +=
+                                "\n\n⚠️ Tool arguments were truncated (likely exceeded max output tokens). Please reduce your output or split into multiple tool calls.";
+                        }
                         // Update message state - tool execution completed
                         this.messageManager.updateToolBlock({
                             id: toolId,
                             parameters: argsString,
-                            result: toolResult.content ||
-                                (toolResult.error ? `Error: ${toolResult.error}` : ""),
+                            result: toolResultContent,
                             success: toolResult.success,
                             error: toolResult.error,
                             stage: "end",
@@ -727,75 +745,82 @@ export class AIManager {
             this.messageManager.finalizeStreamingBlocks();
             // 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();
-                    if (snapshots.length > 0) {
-                        this.messageManager.addFileHistoryBlock(snapshots);
-                    }
+                // Check maxTurns limit before recursing
+                if (this.maxTurns && recursionDepth + 1 >= this.maxTurns) {
+                    logger?.debug(`Max turns (${this.maxTurns}) reached, stopping recursion.`);
                 }
-                // Check interruption status
-                const isCurrentlyAborted = abortController.signal.aborted || toolAbortController.signal.aborted;
-                // Check if all tools were manually backgrounded
-                const lastMessage = this.messageManager.getMessages()[this.messageManager.getMessages().length - 1];
-                const toolBlocks = lastMessage?.blocks.filter((block) => block.type === "tool") || [];
-                const hasBackgrounded = toolBlocks.length > 0 &&
-                    toolBlocks.some((block) => block.isManuallyBackgrounded);
-                if (hasBackgrounded) {
-                    logger?.info("Some tools were manually backgrounded, stopping recursion.");
-                }
-                else if (!isCurrentlyAborted) {
-                    // If response was truncated, add a hidden continuation message
-                    if (result.finish_reason === "length") {
-                        this.messageManager.addUserMessage({
-                            content: "Output token limit hit. Resume directly — no apology, no recap of what you were doing. Pick up mid-thought if that is where the cut happened. Break remaining work into smaller pieces.",
-                            isMeta: true,
-                        });
+                else {
+                    // Record committed snapshots to message history
+                    if (this.reversionManager) {
+                        const snapshots = this.reversionManager.getAndClearCommittedSnapshots();
+                        if (snapshots.length > 0) {
+                            this.messageManager.addFileHistoryBlock(snapshots);
+                        }
                     }
-                    // Duplicate Tool Call Detection
-                    if (toolCalls.length > 0) {
-                        const messages = this.messageManager.getMessages();
-                        // Find the most recent assistant message BEFORE the current one that has tool blocks
-                        // The current assistant message is messages[messages.length - 1]
-                        let previousAssistantWithTools;
-                        for (let i = messages.length - 2; i >= 0; i--) {
-                            const msg = messages[i];
-                            if (msg.role === "assistant" &&
-                                msg.blocks.some((b) => b.type === "tool")) {
-                                previousAssistantWithTools = msg;
-                                break;
-                            }
+                    // Check interruption status
+                    const isCurrentlyAborted = abortController.signal.aborted ||
+                        toolAbortController.signal.aborted;
+                    // Check if all tools were manually backgrounded
+                    const lastMessage = this.messageManager.getMessages()[this.messageManager.getMessages().length - 1];
+                    const toolBlocks = lastMessage?.blocks.filter((block) => block.type === "tool") || [];
+                    const hasBackgrounded = toolBlocks.length > 0 &&
+                        toolBlocks.some((block) => block.isManuallyBackgrounded);
+                    if (hasBackgrounded) {
+                        logger?.info("Some tools were manually backgrounded, stopping recursion.");
+                    }
+                    else if (!isCurrentlyAborted) {
+                        // If response was truncated, add a hidden continuation message
+                        if (result.finish_reason === "length") {
+                            this.messageManager.addUserMessage({
+                                content: "Output token limit hit. Resume directly — no apology, no recap of what you were doing. Pick up mid-thought if that is where the cut happened. Break remaining work into smaller pieces.",
+                                isMeta: true,
+                            });
                         }
-                        if (previousAssistantWithTools) {
-                            const previousToolBlocks = previousAssistantWithTools.blocks.filter((b) => b.type === "tool");
-                            for (const currentToolCall of toolCalls) {
-                                const currentName = currentToolCall.function?.name;
-                                const currentArgs = currentToolCall.function?.arguments;
-                                const isDuplicate = previousToolBlocks.some((prevBlock) => prevBlock.name === currentName &&
-                                    prevBlock.parameters === currentArgs);
-                                if (isDuplicate && currentName) {
-                                    const toolId = currentToolCall.id;
-                                    const lastMessage = messages[messages.length - 1];
-                                    const toolBlock = lastMessage.blocks.find((b) => b.type === "tool" && b.id === toolId);
-                                    if (toolBlock) {
-                                        const warning = `\n\nNote: You just called this tool with the same arguments in the previous turn. Please ensure you are not in a loop and consider if you need to change your approach.`;
-                                        this.messageManager.updateToolBlock({
-                                            id: toolId,
-                                            result: (toolBlock.result || "") + warning,
-                                            stage: "end",
-                                        });
+                        // Duplicate Tool Call Detection
+                        if (toolCalls.length > 0) {
+                            const messages = this.messageManager.getMessages();
+                            // Find the most recent assistant message BEFORE the current one that has tool blocks
+                            // The current assistant message is messages[messages.length - 1]
+                            let previousAssistantWithTools;
+                            for (let i = messages.length - 2; i >= 0; i--) {
+                                const msg = messages[i];
+                                if (msg.role === "assistant" &&
+                                    msg.blocks.some((b) => b.type === "tool")) {
+                                    previousAssistantWithTools = msg;
+                                    break;
+                                }
+                            }
+                            if (previousAssistantWithTools) {
+                                const previousToolBlocks = previousAssistantWithTools.blocks.filter((b) => b.type === "tool");
+                                for (const currentToolCall of toolCalls) {
+                                    const currentName = currentToolCall.function?.name;
+                                    const currentArgs = currentToolCall.function?.arguments;
+                                    const isDuplicate = previousToolBlocks.some((prevBlock) => prevBlock.name === currentName &&
+                                        prevBlock.parameters === currentArgs);
+                                    if (isDuplicate && currentName) {
+                                        const toolId = currentToolCall.id;
+                                        const lastMessage = messages[messages.length - 1];
+                                        const toolBlock = lastMessage.blocks.find((b) => b.type === "tool" && b.id === toolId);
+                                        if (toolBlock) {
+                                            const warning = `\n\nNote: You just called this tool with the same arguments in the previous turn. Please ensure you are not in a loop and consider if you need to change your approach.`;
+                                            this.messageManager.updateToolBlock({
+                                                id: toolId,
+                                                result: (toolBlock.result || "") + warning,
+                                                stage: "end",
+                                            });
+                                        }
                                     }
                                 }
                             }
                         }
+                        // Recursively call AI service, increment recursion depth, and pass same configuration
+                        await this.sendAIMessage({
+                            recursionDepth: recursionDepth + 1,
+                            model,
+                            allowedRules,
+                            maxTokens,
+                        });
                     }
-                    // Recursively call AI service, increment recursion depth, and pass same configuration
-                    await this.sendAIMessage({
-                        recursionDepth: recursionDepth + 1,
-                        model,
-                        allowedRules,
-                        maxTokens,
-                    });
                 }
             }
         }

package/dist/managers/forkedAgentManager.d.ts

@@ -30,6 +30,7 @@ export declare class ForkedAgentManager {
         allowedTools?: string[];
         model?: string;
         permissionModeOverride?: PermissionMode;
+        maxTurns?: number;
     }, prompt: string): Promise<string>;
     private executeFork;
     /**

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

@@ -1 +1 @@
-{"version":3,"file":"forkedAgentManager.d.ts","sourceRoot":"","sources":["../../src/managers/forkedAgentManager.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEjD,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAmB,KAAK,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE9D,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC;IAC3B,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;CAC5C;AAED,MAAM,WAAW,2BAA2B;IAC1C,yBAAyB,CAAC,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,KAAK,IAAI,CAAC;CACnE;AAED,qBAAa,kBAAkB;IAK3B,OAAO,CAAC,SAAS;IAJnB,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,SAAS,CAA8B;gBAGrC,SAAS,EAAE,SAAS,EAC5B,OAAO,GAAE;QAAE,SAAS,CAAC,EAAE,2BAA2B,CAAA;KAAO;IAK3D,OAAO,KAAK,eAAe,GAE1B;IAED;;;OAGG;IACG,cAAc,CAClB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,OAAO,EAAE,EACnB,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,sBAAsB,CAAC,EAAE,cAAc,CAAC;KACzC,EACD,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,CAAC;YA0BJ,WAAW;IA2EzB;;OAEG;IACH,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAazB;;OAEG;IACH,OAAO,IAAI,IAAI;IASf;;OAEG;IACH,cAAc,IAAI,gBAAgB,EAAE;IAIpC,OAAO,CAAC,YAAY;CAKrB"}
+{"version":3,"file":"forkedAgentManager.d.ts","sourceRoot":"","sources":["../../src/managers/forkedAgentManager.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEjD,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAmB,KAAK,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE9D,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC;IAC3B,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;CAC5C;AAED,MAAM,WAAW,2BAA2B;IAC1C,yBAAyB,CAAC,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,KAAK,IAAI,CAAC;CACnE;AAED,qBAAa,kBAAkB;IAK3B,OAAO,CAAC,SAAS;IAJnB,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,SAAS,CAA8B;gBAGrC,SAAS,EAAE,SAAS,EAC5B,OAAO,GAAE;QAAE,SAAS,CAAC,EAAE,2BAA2B,CAAA;KAAO;IAK3D,OAAO,KAAK,eAAe,GAE1B;IAED;;;OAGG;IACG,cAAc,CAClB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,OAAO,EAAE,EACnB,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,sBAAsB,CAAC,EAAE,cAAc,CAAC;QACxC,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,EACD,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,CAAC;YA0BJ,WAAW;IA6EzB;;OAEG;IACH,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAazB;;OAEG;IACH,OAAO,IAAI,IAAI;IASf;;OAEG;IACH,cAAc,IAAI,gBAAgB,EAAE;IAIpC,OAAO,CAAC,YAAY;CAKrB"}

package/dist/managers/forkedAgentManager.js

@@ -48,6 +48,7 @@ export class ForkedAgentManager {
                 allowedTools: parameters.allowedTools,
                 model: parameters.model,
                 permissionModeOverride: parameters.permissionModeOverride,
+                maxTurns: parameters.maxTurns,
             }, false);
             // Pre-load the message manager with conversation history
             instance.messageManager.setMessages(messages);

package/dist/managers/subagentManager.d.ts

@@ -93,6 +93,7 @@ export declare class SubagentManager {
         model?: string;
         stream?: boolean;
         permissionModeOverride?: PermissionMode;
+        maxTurns?: number;
     }, runInBackground?: boolean, onUpdate?: () => void): Promise<SubagentInstance>;
     /**
      * Execute agent using subagent instance

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

@@ -1 +1 @@
-{"version":3,"file":"subagentManager.d.ts","sourceRoot":"","sources":["../../src/managers/subagentManager.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAS/C,OAAO,EACL,iBAAiB,EACjB,KAAK,0BAA0B,EAChC,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAElD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAG9D,MAAM,WAAW,wBAAwB;IAEvC,gDAAgD;IAChD,0BAA0B,CAAC,EAAE,CAC3B,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,iBAAiB,KACtB,IAAI,CAAC;IACV,wDAAwD;IACxD,+BAA+B,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/D,0DAA0D;IAC1D,iCAAiC,CAAC,EAAE,CAClC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,KAChB,IAAI,CAAC;IACV,4DAA4D;IAC5D,mCAAmC,CAAC,EAAE,CACpC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,KAChB,IAAI,CAAC;IACV,oDAAoD;IACpD,0BAA0B,CAAC,EAAE,CAC3B,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,0BAA0B,KAC/B,IAAI,CAAC;IACV,8CAA8C;IAC9C,wBAAwB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;IAC7E,yDAAyD;IACzD,iCAAiC,CAAC,EAAE,CAClC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,KACX,IAAI,CAAC;CACX;AAED,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,qBAAqB,CAAC;IACrC,SAAS,EAAE,SAAS,CAAC;IACrB,cAAc,EAAE,cAAc,CAAC;IAC/B,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,EAAE,cAAc,GAAG,QAAQ,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,CAAC;IACtE,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,IAAI,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC;CAC5B;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,wBAAwB,CAAC;IACrC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACtC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,SAAS,CAAuC;IACxD,OAAO,CAAC,0BAA0B,CAAwC;IAC1E,OAAO,CAAC,oBAAoB,CAAwC;IAEpE,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,SAAS,CAAC,CAA2B;IAC7C,OAAO,CAAC,YAAY,CAAC,CAAyB;IAC9C,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,MAAM,CAAU;gBAEZ,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,sBAAsB;IAQjE,OAAO,KAAK,oBAAoB,GAE/B;IAED;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA+BjC;;OAEG;IACH,OAAO,CAAC,8BAA8B;IAWtC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAY5D;;OAEG;IACH,iBAAiB,IAAI,qBAAqB,EAAE;IAS5C;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM;IAa/B;;;OAGG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,qBAAqB,EAAE,GAC9B,IAAI;IAgCP;;OAEG;IACG,cAAc,CAClB,aAAa,EAAE,qBAAqB,EACpC,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,MAAM,EAAE,MAAM,CAAC;QACf,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,sBAAsB,CAAC,EAAE,cAAc,CAAC;KACzC,EACD,eAAe,CAAC,EAAE,OAAO,EACzB,QAAQ,CAAC,EAAE,MAAM,IAAI,GACpB,OAAO,CAAC,gBAAgB,CAAC;IAiJ5B;;;;;OAKG;IACG,YAAY,CAChB,QAAQ,EAAE,gBAAgB,EAC1B,MAAM,EAAE,MAAM,EACd,WAAW,CAAC,EAAE,WAAW,EACzB,eAAe,CAAC,EAAE,OAAO,GACxB,OAAO,CAAC,MAAM,CAAC;IAiFZ,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YA6C/C,eAAe;IAmJ7B;;OAEG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI;IAIxD;;OAEG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,GACjC,IAAI;IAOP;;OAEG;IACH,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAOhE;;OAEG;IACH,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAazC;;OAEG;IACH,kBAAkB,IAAI,gBAAgB,EAAE;IAOxC;;OAEG;IACH,OAAO,IAAI,IAAI;IAIf;;;OAGG;IACH,OAAO,CAAC,uBAAuB;CA+FhC"}
+{"version":3,"file":"subagentManager.d.ts","sourceRoot":"","sources":["../../src/managers/subagentManager.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAEzB,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAS/C,OAAO,EACL,iBAAiB,EACjB,KAAK,0BAA0B,EAChC,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAElD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAG9D,MAAM,WAAW,wBAAwB;IAEvC,gDAAgD;IAChD,0BAA0B,CAAC,EAAE,CAC3B,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,iBAAiB,KACtB,IAAI,CAAC;IACV,wDAAwD;IACxD,+BAA+B,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/D,0DAA0D;IAC1D,iCAAiC,CAAC,EAAE,CAClC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,KAChB,IAAI,CAAC;IACV,4DAA4D;IAC5D,mCAAmC,CAAC,EAAE,CACpC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,KAChB,IAAI,CAAC;IACV,oDAAoD;IACpD,0BAA0B,CAAC,EAAE,CAC3B,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,0BAA0B,KAC/B,IAAI,CAAC;IACV,8CAA8C;IAC9C,wBAAwB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;IAC7E,yDAAyD;IACzD,iCAAiC,CAAC,EAAE,CAClC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,KACX,IAAI,CAAC;CACX;AAED,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,qBAAqB,CAAC;IACrC,SAAS,EAAE,SAAS,CAAC;IACrB,cAAc,EAAE,cAAc,CAAC;IAC/B,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,EAAE,cAAc,GAAG,QAAQ,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,CAAC;IACtE,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,IAAI,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC;CAC5B;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,wBAAwB,CAAC;IACrC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACtC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,SAAS,CAAuC;IACxD,OAAO,CAAC,0BAA0B,CAAwC;IAC1E,OAAO,CAAC,oBAAoB,CAAwC;IAEpE,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,SAAS,CAAC,CAA2B;IAC7C,OAAO,CAAC,YAAY,CAAC,CAAyB;IAC9C,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,MAAM,CAAU;gBAEZ,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,sBAAsB;IAQjE,OAAO,KAAK,oBAAoB,GAE/B;IAED;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA+BjC;;OAEG;IACH,OAAO,CAAC,8BAA8B;IAWtC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAY5D;;OAEG;IACH,iBAAiB,IAAI,qBAAqB,EAAE;IAS5C;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM;IAa/B;;;OAGG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,qBAAqB,EAAE,GAC9B,IAAI;IAgCP;;OAEG;IACG,cAAc,CAClB,aAAa,EAAE,qBAAqB,EACpC,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,MAAM,EAAE,MAAM,CAAC;QACf,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,sBAAsB,CAAC,EAAE,cAAc,CAAC;QACxC,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,EACD,eAAe,CAAC,EAAE,OAAO,EACzB,QAAQ,CAAC,EAAE,MAAM,IAAI,GACpB,OAAO,CAAC,gBAAgB,CAAC;IAkJ5B;;;;;OAKG;IACG,YAAY,CAChB,QAAQ,EAAE,gBAAgB,EAC1B,MAAM,EAAE,MAAM,EACd,WAAW,CAAC,EAAE,WAAW,EACzB,eAAe,CAAC,EAAE,OAAO,GACxB,OAAO,CAAC,MAAM,CAAC;IAiFZ,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YA6C/C,eAAe;IAmJ7B;;OAEG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI;IAIxD;;OAEG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,GACjC,IAAI;IAOP;;OAEG;IACH,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAOhE;;OAEG;IACH,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAazC;;OAEG;IACH,kBAAkB,IAAI,gBAAgB,EAAE;IAOxC;;OAEG;IACH,OAAO,IAAI,IAAI;IAIf;;;OAGG;IACH,OAAO,CAAC,uBAAuB;CA+FhC"}

package/dist/managers/subagentManager.js

@@ -203,6 +203,7 @@ export class SubagentManager {
             subagentType: parameters.subagent_type, // Pass subagent type for hook context
             modelOverride: parameters.model || configuration.model, // Pass model override
             stream: parameters.stream ?? this.stream, // Pass streaming mode flag
+            maxTurns: parameters.maxTurns, // Pass maxTurns limit
             callbacks: {
                 onUsageAdded: this.onUsageAdded,
             },

package/dist/services/autoMemoryService.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"autoMemoryService.d.ts","sourceRoot":"","sources":["../../src/services/autoMemoryService.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAUlD;;;GAGG;AACH,qBAAa,iBAAiB;IAIhB,OAAO,CAAC,SAAS;IAH7B,OAAO,CAAC,mBAAmB,CAAuB;IAClD,OAAO,CAAC,wBAAwB,CAAa;gBAEzB,SAAS,EAAE,SAAS;IAExC,OAAO,KAAK,cAAc,GAEzB;IAED,OAAO,KAAK,kBAAkB,GAE7B;IAED,OAAO,KAAK,aAAa,GAExB;IAED,OAAO,KAAK,oBAAoB,GAE/B;IAED;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqE/C;;OAEG;YACW,aAAa;CA2D5B"}
+{"version":3,"file":"autoMemoryService.d.ts","sourceRoot":"","sources":["../../src/services/autoMemoryService.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAUlD;;;GAGG;AACH,qBAAa,iBAAiB;IAIhB,OAAO,CAAC,SAAS;IAH7B,OAAO,CAAC,mBAAmB,CAAuB;IAClD,OAAO,CAAC,wBAAwB,CAAa;gBAEzB,SAAS,EAAE,SAAS;IAExC,OAAO,KAAK,cAAc,GAEzB;IAED,OAAO,KAAK,kBAAkB,GAE7B;IAED,OAAO,KAAK,aAAa,GAExB;IAED,OAAO,KAAK,oBAAoB,GAE/B;IAED;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqE/C;;OAEG;YACW,aAAa;CA4D5B"}

package/dist/services/autoMemoryService.js

@@ -128,6 +128,7 @@ export class AutoMemoryService {
             ],
             model: "fastModel", // Use fast model for background tasks to reduce latency and cost
             permissionModeOverride: "dontAsk", // Auto-deny out-of-scope writes without prompting user
+            maxTurns: 5, // Limit turns to prevent verification rabbit-holes
         }, `${prompt}\n\nThe memory directory for this project is: ${memoryDir}`);
         logger.debug("Auto-memory extraction started in background.");
     }

package/dist/utils/constants.d.ts

@@ -21,6 +21,6 @@ export declare const USER_MEMORY_FILE: string;
 /**
  * AI related constants
  */
-export declare const DEFAULT_WAVE_MAX_INPUT_TOKENS = 96000;
-export declare const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 8192;
+export declare const DEFAULT_WAVE_MAX_INPUT_TOKENS = 128000;
+export declare const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 16384;
 //# sourceMappingURL=constants.d.ts.map

package/dist/utils/constants.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/utils/constants.ts"],"names":[],"mappings":"AAAA;;GAEG;AAKH;;;GAGG;AACH,eAAO,MAAM,cAAc,QAAmC,CAAC;AAE/D;;GAEG;AACH,eAAO,MAAM,mBAAmB,QAA6C,CAAC;AAE9E;;GAEG;AACH,eAAO,MAAM,mBAAmB,QAA0C,CAAC;AAE3E;;GAEG;AACH,eAAO,MAAM,gBAAgB,QAAyC,CAAC;AAEvE;;GAEG;AACH,eAAO,MAAM,6BAA6B,QAAQ,CAAC;AACnD,eAAO,MAAM,8BAA8B,OAAO,CAAC"}
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/utils/constants.ts"],"names":[],"mappings":"AAAA;;GAEG;AAKH;;;GAGG;AACH,eAAO,MAAM,cAAc,QAAmC,CAAC;AAE/D;;GAEG;AACH,eAAO,MAAM,mBAAmB,QAA6C,CAAC;AAE9E;;GAEG;AACH,eAAO,MAAM,mBAAmB,QAA0C,CAAC;AAE3E;;GAEG;AACH,eAAO,MAAM,gBAAgB,QAAyC,CAAC;AAEvE;;GAEG;AACH,eAAO,MAAM,6BAA6B,SAAS,CAAC;AACpD,eAAO,MAAM,8BAA8B,QAAQ,CAAC"}

package/dist/utils/constants.js

@@ -23,5 +23,5 @@ export const USER_MEMORY_FILE = path.join(DATA_DIRECTORY, "AGENTS.md");
 /**
  * AI related constants
  */
-export const DEFAULT_WAVE_MAX_INPUT_TOKENS = 96000; // Default token limit
-export const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 8192; // Default output token limit
+export const DEFAULT_WAVE_MAX_INPUT_TOKENS = 128000; // Default token limit
+export const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 16384; // Default output token limit

package/dist/utils/convertMessagesForAPI.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessagesForAPI.d.ts","sourceRoot":"","sources":["../../src/utils/convertMessagesForAPI.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAKjD,OAAO,EAEL,0BAA0B,EAC3B,MAAM,qBAAqB,CAAC;AA0B7B;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,OAAO,EAAE,GAClB,0BAA0B,EAAE,CAsP9B"}
+{"version":3,"file":"convertMessagesForAPI.d.ts","sourceRoot":"","sources":["../../src/utils/convertMessagesForAPI.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAKjD,OAAO,EAEL,0BAA0B,EAC3B,MAAM,qBAAqB,CAAC;AAiC7B;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,OAAO,EAAE,GAClB,0BAA0B,EAAE,CAsP9B"}

package/dist/utils/convertMessagesForAPI.js

@@ -1,9 +1,10 @@
 import { convertImageToBase64 } from "./messageOperations.js";
 import { taskNotificationToXml } from "./notificationXml.js";
-import { stripAnsiColors } from "./stringUtils.js";
+import { recoverTruncatedJson, stripAnsiColors } from "./stringUtils.js";
 import { logger } from "./globalLogger.js";
 /**
- * Safely handle tool call parameters, ensuring a legal JSON string is returned
+ * Safely handle tool call parameters, ensuring a legal JSON string is returned.
+ * Attempts to recover truncated JSON (e.g., missing closing braces).
  * @param args Tool call parameters
  * @returns Legal JSON string
  */
@@ -16,12 +17,19 @@ function safeToolArguments(args) {
         JSON.parse(args);
         return args;
     }
-    catch (error) {
-        logger.error(`Invalid tool arguments: ${args}`, error);
-        // If not valid JSON, return a fallback empty object with the original string as a comment or property
-        return JSON.stringify({
-            invalid_arguments: args,
-        });
+    catch {
+        // Attempt to recover truncated JSON
+        const recovered = recoverTruncatedJson(args);
+        try {
+            JSON.parse(recovered);
+            return recovered;
+        }
+        catch {
+            // Truly malformed JSON — return sanitized fallback
+            return JSON.stringify({
+                invalid_arguments: args,
+            });
+        }
     }
 }
 /**

package/dist/utils/stringUtils.d.ts

@@ -23,6 +23,14 @@ export declare const stripAnsiColors: (text: string) => string;
  * @returns Formatted line number prefix
  */
 export declare function formatLineNumberPrefix(lineNumber: number): string;
+/**
+ * Attempt to recover truncated JSON (e.g., missing closing braces due to max tokens).
+ * Tracks brace depth and only recovers if there are unclosed `{` braces.
+ * Will NOT recover if there are unclosed `[` brackets (can't guess the content).
+ * @param jsonStr Potentially truncated JSON string
+ * @returns Recovered JSON string, or the original if unrecoverable
+ */
+export declare function recoverTruncatedJson(jsonStr: string): string;
 /**
  * Efficiently get the last N lines of a string without splitting the whole string.
  */

package/dist/utils/stringUtils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stringUtils.d.ts","sourceRoot":"","sources":["../../src/utils/stringUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAgC/D;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAChC,aAAa,EAAE,MAAM,GACpB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAwBxB;AAED;;;;GAIG;AACH,eAAO,MAAM,eAAe,GAAI,MAAM,MAAM,KAAG,MAK9C,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAEjE;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAehE"}
+{"version":3,"file":"stringUtils.d.ts","sourceRoot":"","sources":["../../src/utils/stringUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAgC/D;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAChC,aAAa,EAAE,MAAM,GACpB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAwBxB;AAED;;;;GAIG;AACH,eAAO,MAAM,eAAe,GAAI,MAAM,MAAM,KAAG,MAK9C,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAEjE;AAED;;;;;;GAMG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAkC5D;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAehE"}

package/dist/utils/stringUtils.js

@@ -77,6 +77,51 @@ export const stripAnsiColors = (text) => {
 export function formatLineNumberPrefix(lineNumber) {
     return `${lineNumber.toString().padStart(6)}\t`;
 }
+/**
+ * Attempt to recover truncated JSON (e.g., missing closing braces due to max tokens).
+ * Tracks brace depth and only recovers if there are unclosed `{` braces.
+ * Will NOT recover if there are unclosed `[` brackets (can't guess the content).
+ * @param jsonStr Potentially truncated JSON string
+ * @returns Recovered JSON string, or the original if unrecoverable
+ */
+export function recoverTruncatedJson(jsonStr) {
+    let braceDepth = 0;
+    let bracketDepth = 0;
+    let inString = false;
+    let escaped = false;
+    for (const ch of jsonStr) {
+        if (escaped) {
+            escaped = false;
+            continue;
+        }
+        if (ch === "\\" && inString) {
+            escaped = true;
+            continue;
+        }
+        if (ch === '"') {
+            inString = !inString;
+            continue;
+        }
+        if (!inString) {
+            if (ch === "{")
+                braceDepth++;
+            if (ch === "}")
+                braceDepth--;
+            if (ch === "[")
+                bracketDepth++;
+            if (ch === "]")
+                bracketDepth--;
+        }
+    }
+    // Build recovery suffix
+    let suffix = "";
+    if (inString)
+        suffix += '"'; // Close unclosed string
+    if (braceDepth > 0 && bracketDepth === 0) {
+        suffix += "}".repeat(braceDepth);
+    }
+    return suffix ? jsonStr + suffix : jsonStr;
+}
 /**
  * Efficiently get the last N lines of a string without splitting the whole string.
  */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "0.14.3",
+  "version": "0.14.4",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",

package/src/managers/aiManager.ts

@@ -24,6 +24,7 @@ import type { SubagentManager } from "./subagentManager.js";
 import type { SkillManager } from "./skillManager.js";
 import { buildSystemPrompt } from "../prompts/index.js";
 import { Container } from "../utils/container.js";
+import { recoverTruncatedJson } from "../utils/stringUtils.js";
 import { ConfigurationService } from "../services/configurationService.js";
 import type { NotificationQueue } from "./notificationQueue.js";
 
@@ -44,6 +45,8 @@ export interface AIManagerOptions {
   stream?: boolean;
   /**Optional model override (e.g. for subagents) */
   modelOverride?: string;
+  /**Optional max turns limit to prevent runaway recursion (e.g. for auto-memory extraction) */
+  maxTurns?: number;
 }
 
 export class AIManager {
@@ -59,6 +62,7 @@ export class AIManager {
   private modelOverride?: string;
   private _onCwdChange?: (newCwd: string) => void; // Store callback for CWD changes
   private consecutiveCompactionFailures: number = 0;
+  private readonly maxTurns?: number;
 
   // Service overrides
   constructor(
@@ -73,6 +77,7 @@ export class AIManager {
     this.callbacks = options.callbacks ?? {};
     this.modelOverride = options.modelOverride;
     this._onCwdChange = options.callbacks?.onCwdChange; // Initialize onCwdChange
+    this.maxTurns = options.maxTurns;
   }
 
   private get toolManager(): ToolManager {
@@ -814,34 +819,45 @@ export class AIManager {
             const toolName = functionToolCall.function?.name || "";
             // Safely parse tool parameters, handle tools without parameters
             let toolArgs: Record<string, unknown> = {};
+            let jsonRecovered = false;
             const argsString = functionToolCall.function?.arguments?.trim();
 
             if (!argsString || argsString === "") {
               // Tool without parameters, use empty object
               toolArgs = {};
             } else {
+              let recoveredArgs = argsString;
               try {
                 toolArgs = JSON.parse(argsString);
-              } catch (parseError) {
-                // For non-empty but malformed JSON, still throw exception
-                let errorMessage = `Failed to parse tool arguments`;
-                if (result.finish_reason === "length") {
-                  errorMessage +=
-                    " (output truncated, please reduce your output)";
+              } catch {
+                // Attempt to recover truncated JSON (e.g., missing closing braces)
+                recoveredArgs = recoverTruncatedJson(argsString);
+                try {
+                  toolArgs = JSON.parse(recoveredArgs);
+                  jsonRecovered = true;
+                  logger.warn(
+                    `Recovered truncated JSON for tool "${toolName}"`,
+                  );
+                } catch (parseError) {
+                  let errorMessage = `Failed to parse tool arguments`;
+                  if (result.finish_reason === "length") {
+                    errorMessage +=
+                      " (output truncated, please reduce your output)";
+                  }
+                  logger?.error(errorMessage, parseError);
+                  this.messageManager.updateToolBlock({
+                    id: toolId,
+                    parameters: argsString,
+                    result: errorMessage,
+                    success: false,
+                    error: errorMessage,
+                    stage: "end",
+                    name: toolName,
+                    compactParams: "",
+                    timestamp: Date.now(),
+                  });
+                  return;
                 }
-                logger?.error(errorMessage, parseError);
-                this.messageManager.updateToolBlock({
-                  id: toolId,
-                  parameters: argsString,
-                  result: errorMessage,
-                  success: false,
-                  error: errorMessage,
-                  stage: "end",
-                  name: toolName,
-                  compactParams: "",
-                  timestamp: Date.now(),
-                });
-                return;
               }
             }
 
@@ -942,13 +958,20 @@ export class AIManager {
                 context,
               );
 
+              // Build result content, adding truncation warning if JSON was recovered
+              let toolResultContent =
+                toolResult.content ||
+                (toolResult.error ? `Error: ${toolResult.error}` : "");
+              if (jsonRecovered) {
+                toolResultContent +=
+                  "\n\n⚠️ Tool arguments were truncated (likely exceeded max output tokens). Please reduce your output or split into multiple tool calls.";
+              }
+
               // Update message state - tool execution completed
               this.messageManager.updateToolBlock({
                 id: toolId,
                 parameters: argsString,
-                result:
-                  toolResult.content ||
-                  (toolResult.error ? `Error: ${toolResult.error}` : ""),
+                result: toolResultContent,
                 success: toolResult.success,
                 error: toolResult.error,
                 stage: "end",
@@ -1001,108 +1024,116 @@ export class AIManager {
 
       // 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();
-          if (snapshots.length > 0) {
-            this.messageManager.addFileHistoryBlock(snapshots);
-          }
-        }
-
-        // Check interruption status
-        const isCurrentlyAborted =
-          abortController.signal.aborted || toolAbortController.signal.aborted;
-
-        // Check if all tools were manually backgrounded
-        const lastMessage =
-          this.messageManager.getMessages()[
-            this.messageManager.getMessages().length - 1
-          ];
-        const toolBlocks =
-          lastMessage?.blocks.filter(
-            (block): block is import("../types/messaging.js").ToolBlock =>
-              block.type === "tool",
-          ) || [];
-        const hasBackgrounded =
-          toolBlocks.length > 0 &&
-          toolBlocks.some((block) => block.isManuallyBackgrounded);
-
-        if (hasBackgrounded) {
-          logger?.info(
-            "Some tools were manually backgrounded, stopping recursion.",
+        // Check maxTurns limit before recursing
+        if (this.maxTurns && recursionDepth + 1 >= this.maxTurns) {
+          logger?.debug(
+            `Max turns (${this.maxTurns}) reached, stopping recursion.`,
           );
-        } else if (!isCurrentlyAborted) {
-          // If response was truncated, add a hidden continuation message
-          if (result.finish_reason === "length") {
-            this.messageManager.addUserMessage({
-              content:
-                "Output token limit hit. Resume directly — no apology, no recap of what you were doing. Pick up mid-thought if that is where the cut happened. Break remaining work into smaller pieces.",
-              isMeta: true,
-            });
+        } else {
+          // Record committed snapshots to message history
+          if (this.reversionManager) {
+            const snapshots =
+              this.reversionManager.getAndClearCommittedSnapshots();
+            if (snapshots.length > 0) {
+              this.messageManager.addFileHistoryBlock(snapshots);
+            }
           }
 
-          // Duplicate Tool Call Detection
-          if (toolCalls.length > 0) {
-            const messages = this.messageManager.getMessages();
-            // Find the most recent assistant message BEFORE the current one that has tool blocks
-            // The current assistant message is messages[messages.length - 1]
-            let previousAssistantWithTools: Message | undefined;
-            for (let i = messages.length - 2; i >= 0; i--) {
-              const msg = messages[i];
-              if (
-                msg.role === "assistant" &&
-                msg.blocks.some((b) => b.type === "tool")
-              ) {
-                previousAssistantWithTools = msg;
-                break;
-              }
+          // Check interruption status
+          const isCurrentlyAborted =
+            abortController.signal.aborted ||
+            toolAbortController.signal.aborted;
+
+          // Check if all tools were manually backgrounded
+          const lastMessage =
+            this.messageManager.getMessages()[
+              this.messageManager.getMessages().length - 1
+            ];
+          const toolBlocks =
+            lastMessage?.blocks.filter(
+              (block): block is import("../types/messaging.js").ToolBlock =>
+                block.type === "tool",
+            ) || [];
+          const hasBackgrounded =
+            toolBlocks.length > 0 &&
+            toolBlocks.some((block) => block.isManuallyBackgrounded);
+
+          if (hasBackgrounded) {
+            logger?.info(
+              "Some tools were manually backgrounded, stopping recursion.",
+            );
+          } else if (!isCurrentlyAborted) {
+            // If response was truncated, add a hidden continuation message
+            if (result.finish_reason === "length") {
+              this.messageManager.addUserMessage({
+                content:
+                  "Output token limit hit. Resume directly — no apology, no recap of what you were doing. Pick up mid-thought if that is where the cut happened. Break remaining work into smaller pieces.",
+                isMeta: true,
+              });
             }
 
-            if (previousAssistantWithTools) {
-              const previousToolBlocks =
-                previousAssistantWithTools.blocks.filter(
-                  (b): b is import("../types/messaging.js").ToolBlock =>
-                    b.type === "tool",
-                );
+            // Duplicate Tool Call Detection
+            if (toolCalls.length > 0) {
+              const messages = this.messageManager.getMessages();
+              // Find the most recent assistant message BEFORE the current one that has tool blocks
+              // The current assistant message is messages[messages.length - 1]
+              let previousAssistantWithTools: Message | undefined;
+              for (let i = messages.length - 2; i >= 0; i--) {
+                const msg = messages[i];
+                if (
+                  msg.role === "assistant" &&
+                  msg.blocks.some((b) => b.type === "tool")
+                ) {
+                  previousAssistantWithTools = msg;
+                  break;
+                }
+              }
 
-              for (const currentToolCall of toolCalls) {
-                const currentName = currentToolCall.function?.name;
-                const currentArgs = currentToolCall.function?.arguments;
+              if (previousAssistantWithTools) {
+                const previousToolBlocks =
+                  previousAssistantWithTools.blocks.filter(
+                    (b): b is import("../types/messaging.js").ToolBlock =>
+                      b.type === "tool",
+                  );
 
-                const isDuplicate = previousToolBlocks.some(
-                  (prevBlock) =>
-                    prevBlock.name === currentName &&
-                    prevBlock.parameters === currentArgs,
-                );
+                for (const currentToolCall of toolCalls) {
+                  const currentName = currentToolCall.function?.name;
+                  const currentArgs = currentToolCall.function?.arguments;
 
-                if (isDuplicate && currentName) {
-                  const toolId = currentToolCall.id;
-                  const lastMessage = messages[messages.length - 1];
-                  const toolBlock = lastMessage.blocks.find(
-                    (b): b is import("../types/messaging.js").ToolBlock =>
-                      b.type === "tool" && b.id === toolId,
+                  const isDuplicate = previousToolBlocks.some(
+                    (prevBlock) =>
+                      prevBlock.name === currentName &&
+                      prevBlock.parameters === currentArgs,
                   );
-                  if (toolBlock) {
-                    const warning = `\n\nNote: You just called this tool with the same arguments in the previous turn. Please ensure you are not in a loop and consider if you need to change your approach.`;
-                    this.messageManager.updateToolBlock({
-                      id: toolId,
-                      result: (toolBlock.result || "") + warning,
-                      stage: "end",
-                    });
+
+                  if (isDuplicate && currentName) {
+                    const toolId = currentToolCall.id;
+                    const lastMessage = messages[messages.length - 1];
+                    const toolBlock = lastMessage.blocks.find(
+                      (b): b is import("../types/messaging.js").ToolBlock =>
+                        b.type === "tool" && b.id === toolId,
+                    );
+                    if (toolBlock) {
+                      const warning = `\n\nNote: You just called this tool with the same arguments in the previous turn. Please ensure you are not in a loop and consider if you need to change your approach.`;
+                      this.messageManager.updateToolBlock({
+                        id: toolId,
+                        result: (toolBlock.result || "") + warning,
+                        stage: "end",
+                      });
+                    }
                   }
                 }
               }
             }
-          }
 
-          // Recursively call AI service, increment recursion depth, and pass same configuration
-          await this.sendAIMessage({
-            recursionDepth: recursionDepth + 1,
-            model,
-            allowedRules,
-            maxTokens,
-          });
+            // Recursively call AI service, increment recursion depth, and pass same configuration
+            await this.sendAIMessage({
+              recursionDepth: recursionDepth + 1,
+              model,
+              allowedRules,
+              maxTokens,
+            });
+          }
         }
       }
     } catch (error) {

package/src/managers/forkedAgentManager.ts

@@ -47,6 +47,7 @@ export class ForkedAgentManager {
       allowedTools?: string[];
       model?: string;
       permissionModeOverride?: PermissionMode;
+      maxTurns?: number;
     },
     prompt: string,
   ): Promise<string> {
@@ -84,6 +85,7 @@ export class ForkedAgentManager {
       allowedTools?: string[];
       model?: string;
       permissionModeOverride?: PermissionMode;
+      maxTurns?: number;
     },
     prompt: string,
   ): Promise<void> {
@@ -103,6 +105,7 @@ export class ForkedAgentManager {
           allowedTools: parameters.allowedTools,
           model: parameters.model,
           permissionModeOverride: parameters.permissionModeOverride,
+          maxTurns: parameters.maxTurns,
         },
         false,
       );

package/src/managers/subagentManager.ts

@@ -251,6 +251,7 @@ export class SubagentManager {
       model?: string;
       stream?: boolean;
       permissionModeOverride?: PermissionMode;
+      maxTurns?: number;
     },
     runInBackground?: boolean,
     onUpdate?: () => void,
@@ -356,6 +357,7 @@ export class SubagentManager {
       subagentType: parameters.subagent_type, // Pass subagent type for hook context
       modelOverride: parameters.model || configuration.model, // Pass model override
       stream: parameters.stream ?? this.stream, // Pass streaming mode flag
+      maxTurns: parameters.maxTurns, // Pass maxTurns limit
       callbacks: {
         onUsageAdded: this.onUsageAdded,
       },

package/src/services/autoMemoryService.ts

@@ -164,6 +164,7 @@ export class AutoMemoryService {
         ],
         model: "fastModel", // Use fast model for background tasks to reduce latency and cost
         permissionModeOverride: "dontAsk", // Auto-deny out-of-scope writes without prompting user
+        maxTurns: 5, // Limit turns to prevent verification rabbit-holes
       },
       `${prompt}\n\nThe memory directory for this project is: ${memoryDir}`,
     );

package/src/utils/constants.ts

@@ -29,5 +29,5 @@ export const USER_MEMORY_FILE = path.join(DATA_DIRECTORY, "AGENTS.md");
 /**
  * AI related constants
  */
-export const DEFAULT_WAVE_MAX_INPUT_TOKENS = 96000; // Default token limit
-export const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 8192; // Default output token limit
+export const DEFAULT_WAVE_MAX_INPUT_TOKENS = 128000; // Default token limit
+export const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 16384; // Default output token limit

package/src/utils/convertMessagesForAPI.ts

@@ -2,7 +2,7 @@ import type { Message } from "../types/index.js";
 import { convertImageToBase64 } from "./messageOperations.js";
 import { taskNotificationToXml } from "./notificationXml.js";
 import { ChatCompletionMessageToolCall } from "openai/resources";
-import { stripAnsiColors } from "./stringUtils.js";
+import { recoverTruncatedJson, stripAnsiColors } from "./stringUtils.js";
 import {
   ChatCompletionContentPart,
   ChatCompletionMessageParam,
@@ -10,7 +10,8 @@ import {
 import { logger } from "./globalLogger.js";
 
 /**
- * Safely handle tool call parameters, ensuring a legal JSON string is returned
+ * Safely handle tool call parameters, ensuring a legal JSON string is returned.
+ * Attempts to recover truncated JSON (e.g., missing closing braces).
  * @param args Tool call parameters
  * @returns Legal JSON string
  */
@@ -23,12 +24,18 @@ function safeToolArguments(args: string): string {
     // Try to parse as JSON to validate format
     JSON.parse(args);
     return args;
-  } catch (error) {
-    logger.error(`Invalid tool arguments: ${args}`, error);
-    // If not valid JSON, return a fallback empty object with the original string as a comment or property
-    return JSON.stringify({
-      invalid_arguments: args,
-    });
+  } catch {
+    // Attempt to recover truncated JSON
+    const recovered = recoverTruncatedJson(args);
+    try {
+      JSON.parse(recovered);
+      return recovered;
+    } catch {
+      // Truly malformed JSON — return sanitized fallback
+      return JSON.stringify({
+        invalid_arguments: args,
+      });
+    }
   }
 }
 

package/src/utils/stringUtils.ts

@@ -92,6 +92,49 @@ export function formatLineNumberPrefix(lineNumber: number): string {
   return `${lineNumber.toString().padStart(6)}\t`;
 }
 
+/**
+ * Attempt to recover truncated JSON (e.g., missing closing braces due to max tokens).
+ * Tracks brace depth and only recovers if there are unclosed `{` braces.
+ * Will NOT recover if there are unclosed `[` brackets (can't guess the content).
+ * @param jsonStr Potentially truncated JSON string
+ * @returns Recovered JSON string, or the original if unrecoverable
+ */
+export function recoverTruncatedJson(jsonStr: string): string {
+  let braceDepth = 0;
+  let bracketDepth = 0;
+  let inString = false;
+  let escaped = false;
+
+  for (const ch of jsonStr) {
+    if (escaped) {
+      escaped = false;
+      continue;
+    }
+    if (ch === "\\" && inString) {
+      escaped = true;
+      continue;
+    }
+    if (ch === '"') {
+      inString = !inString;
+      continue;
+    }
+    if (!inString) {
+      if (ch === "{") braceDepth++;
+      if (ch === "}") braceDepth--;
+      if (ch === "[") bracketDepth++;
+      if (ch === "]") bracketDepth--;
+    }
+  }
+
+  // Build recovery suffix
+  let suffix = "";
+  if (inString) suffix += '"'; // Close unclosed string
+  if (braceDepth > 0 && bracketDepth === 0) {
+    suffix += "}".repeat(braceDepth);
+  }
+  return suffix ? jsonStr + suffix : jsonStr;
+}
+
 /**
  * Efficiently get the last N lines of a string without splitting the whole string.
  */