@corbat-tech/coco 2.22.2 → 2.23.1

package/dist/cli/index.js

@@ -1792,6 +1792,7 @@ var init_openai = __esm({
             requestParams
           );
           const toolCallBuilders = /* @__PURE__ */ new Map();
+          let lastToolCallKey = null;
           const streamTimeout = this.config.timeout ?? 12e4;
           let lastActivityTime = Date.now();
           const timeoutController = new AbortController();
@@ -1841,9 +1842,9 @@ var init_openai = __esm({
               }
               if (delta?.tool_calls) {
                 for (const toolCallDelta of delta.tool_calls) {
-                  const index = toolCallDelta.index ?? toolCallBuilders.size;
-                  if (!toolCallBuilders.has(index)) {
-                    toolCallBuilders.set(index, {
+                  const key = typeof toolCallDelta.index === "number" ? `index:${toolCallDelta.index}` : typeof toolCallDelta.id === "string" && toolCallDelta.id.length > 0 ? `id:${toolCallDelta.id}` : toolCallBuilders.size === 1 ? Array.from(toolCallBuilders.keys())[0] ?? `fallback:${toolCallBuilders.size}` : lastToolCallKey ?? `fallback:${toolCallBuilders.size}`;
+                  if (!toolCallBuilders.has(key)) {
+                    toolCallBuilders.set(key, {
                       id: toolCallDelta.id ?? "",
                       name: toolCallDelta.function?.name ?? "",
                       arguments: ""
@@ -1856,7 +1857,8 @@ var init_openai = __esm({
                       }
                     };
                   }
-                  const builder = toolCallBuilders.get(index);
+                  const builder = toolCallBuilders.get(key);
+                  lastToolCallKey = key;
                   if (toolCallDelta.id) {
                     builder.id = toolCallDelta.id;
                   }
@@ -2444,6 +2446,7 @@ var init_openai = __esm({
             requestParams
           );
           const fnCallBuilders = /* @__PURE__ */ new Map();
+          const outputIndexToBuilderKey = /* @__PURE__ */ new Map();
           const streamTimeout = this.config.timeout ?? 12e4;
           let lastActivityTime = Date.now();
           const timeoutController = new AbortController();
@@ -2473,8 +2476,11 @@ var init_openai = __esm({
                     fnCallBuilders.set(itemKey, {
                       callId: fc.call_id,
                       name: fc.name,
-                      arguments: ""
+                      arguments: fc.arguments ?? ""
                     });
+                    if (typeof event.output_index === "number") {
+                      outputIndexToBuilderKey.set(event.output_index, itemKey);
+                    }
                     yield {
                       type: "tool_use_start",
                       toolCall: { id: fc.call_id, name: fc.name }
@@ -2483,7 +2489,9 @@ var init_openai = __esm({
                   break;
                 case "response.function_call_arguments.delta":
                   {
-                    const builder = fnCallBuilders.get(event.item_id);
+                    const builderKey = (event.item_id && fnCallBuilders.has(event.item_id) ? event.item_id : null) ?? (typeof event.output_index === "number" ? outputIndexToBuilderKey.get(event.output_index) ?? null : null) ?? (fnCallBuilders.size === 1 ? Array.from(fnCallBuilders.keys())[0] : null);
+                    if (!builderKey) break;
+                    const builder = fnCallBuilders.get(builderKey);
                     if (builder) {
                       builder.arguments += event.delta;
                     }
@@ -2491,17 +2499,22 @@ var init_openai = __esm({
                   break;
                 case "response.function_call_arguments.done":
                   {
-                    const builder = fnCallBuilders.get(event.item_id);
+                    const builderKey = (event.item_id && fnCallBuilders.has(event.item_id) ? event.item_id : null) ?? (typeof event.output_index === "number" ? outputIndexToBuilderKey.get(event.output_index) ?? null : null) ?? (fnCallBuilders.size === 1 ? Array.from(fnCallBuilders.keys())[0] : null);
+                    if (!builderKey) break;
+                    const builder = fnCallBuilders.get(builderKey);
                     if (builder) {
                       yield {
                         type: "tool_use_end",
                         toolCall: {
                           id: builder.callId,
                           name: builder.name,
-                          input: this.parseResponsesArguments(event.arguments)
+                          input: this.parseResponsesArguments(event.arguments ?? builder.arguments)
                         }
                       };
-                      fnCallBuilders.delete(event.item_id);
+                      fnCallBuilders.delete(builderKey);
+                      for (const [idx, key] of outputIndexToBuilderKey.entries()) {
+                        if (key === builderKey) outputIndexToBuilderKey.delete(idx);
+                      }
                     }
                   }
                   break;
@@ -4640,6 +4653,7 @@ var init_codex = __esm({
           let outputTokens = 0;
           const toolCalls = [];
           const fnCallBuilders = /* @__PURE__ */ new Map();
+          const outputIndexToBuilderKey = /* @__PURE__ */ new Map();
           await this.readSSEStream(response, (event) => {
             if (event.id) responseId = event.id;
             switch (event.type) {
@@ -4656,25 +4670,35 @@ var init_codex = __esm({
                   fnCallBuilders.set(itemKey, {
                     callId: item.call_id,
                     name: item.name,
-                    arguments: ""
+                    arguments: item.arguments ?? ""
                   });
+                  if (typeof event.output_index === "number") {
+                    outputIndexToBuilderKey.set(event.output_index, itemKey);
+                  }
                 }
                 break;
               }
               case "response.function_call_arguments.delta": {
-                const builder = fnCallBuilders.get(event.item_id);
+                const builderKey = (event.item_id && fnCallBuilders.has(event.item_id) ? event.item_id : null) ?? (typeof event.output_index === "number" ? outputIndexToBuilderKey.get(event.output_index) ?? null : null) ?? (fnCallBuilders.size === 1 ? Array.from(fnCallBuilders.keys())[0] : null);
+                if (!builderKey) break;
+                const builder = fnCallBuilders.get(builderKey);
                 if (builder) builder.arguments += event.delta ?? "";
                 break;
               }
               case "response.function_call_arguments.done": {
-                const builder = fnCallBuilders.get(event.item_id);
+                const builderKey = (event.item_id && fnCallBuilders.has(event.item_id) ? event.item_id : null) ?? (typeof event.output_index === "number" ? outputIndexToBuilderKey.get(event.output_index) ?? null : null) ?? (fnCallBuilders.size === 1 ? Array.from(fnCallBuilders.keys())[0] : null);
+                if (!builderKey) break;
+                const builder = fnCallBuilders.get(builderKey);
                 if (builder) {
                   toolCalls.push({
                     id: builder.callId,
                     name: builder.name,
-                    input: parseArguments(event.arguments)
+                    input: parseArguments(event.arguments ?? builder.arguments)
                   });
-                  fnCallBuilders.delete(event.item_id);
+                  fnCallBuilders.delete(builderKey);
+                  for (const [idx, key] of outputIndexToBuilderKey.entries()) {
+                    if (key === builderKey) outputIndexToBuilderKey.delete(idx);
+                  }
                 }
                 break;
               }
@@ -4787,6 +4811,7 @@ var init_codex = __esm({
         const decoder = new TextDecoder();
         let buffer = "";
         const fnCallBuilders = /* @__PURE__ */ new Map();
+        const outputIndexToBuilderKey = /* @__PURE__ */ new Map();
         let lastActivityTime = Date.now();
         const timeoutController = new AbortController();
         const timeoutInterval = setInterval(() => {
@@ -4825,8 +4850,11 @@ var init_codex = __esm({
                     fnCallBuilders.set(itemKey, {
                       callId: item.call_id,
                       name: item.name,
-                      arguments: ""
+                      arguments: item.arguments ?? ""
                     });
+                    if (typeof event.output_index === "number") {
+                      outputIndexToBuilderKey.set(event.output_index, itemKey);
+                    }
                     yield {
                       type: "tool_use_start",
                       toolCall: { id: item.call_id, name: item.name }
@@ -4835,14 +4863,18 @@ var init_codex = __esm({
                   break;
                 }
                 case "response.function_call_arguments.delta": {
-                  const builder = fnCallBuilders.get(event.item_id);
+                  const builderKey = (event.item_id && fnCallBuilders.has(event.item_id) ? event.item_id : null) ?? (typeof event.output_index === "number" ? outputIndexToBuilderKey.get(event.output_index) ?? null : null) ?? (fnCallBuilders.size === 1 ? Array.from(fnCallBuilders.keys())[0] : null);
+                  if (!builderKey) break;
+                  const builder = fnCallBuilders.get(builderKey);
                   if (builder) {
                     builder.arguments += event.delta ?? "";
                   }
                   break;
                 }
                 case "response.function_call_arguments.done": {
-                  const builder = fnCallBuilders.get(event.item_id);
+                  const builderKey = (event.item_id && fnCallBuilders.has(event.item_id) ? event.item_id : null) ?? (typeof event.output_index === "number" ? outputIndexToBuilderKey.get(event.output_index) ?? null : null) ?? (fnCallBuilders.size === 1 ? Array.from(fnCallBuilders.keys())[0] : null);
+                  if (!builderKey) break;
+                  const builder = fnCallBuilders.get(builderKey);
                   if (builder) {
                     yield {
                       type: "tool_use_end",
@@ -4852,7 +4884,10 @@ var init_codex = __esm({
                         input: parseArguments(event.arguments ?? builder.arguments)
                       }
                     };
-                    fnCallBuilders.delete(event.item_id);
+                    fnCallBuilders.delete(builderKey);
+                    for (const [idx, key] of outputIndexToBuilderKey.entries()) {
+                      if (key === builderKey) outputIndexToBuilderKey.delete(idx);
+                    }
                   }
                   break;
                 }
@@ -5243,8 +5278,8 @@ var init_gemini = __esm({
           const { history, lastMessage } = this.convertMessages(messages);
           const chat = model.startChat({ history });
           const result = await chat.sendMessageStream(lastMessage);
-          const emittedToolCalls = /* @__PURE__ */ new Set();
           let streamStopReason;
+          let streamToolCallCounter = 0;
           for await (const chunk of result.stream) {
             const text13 = chunk.text();
             if (text13) {
@@ -5259,30 +5294,23 @@ var init_gemini = __esm({
               for (const part of candidate.content.parts) {
                 if ("functionCall" in part && part.functionCall) {
                   const funcCall = part.functionCall;
-                  const sortedArgs = funcCall.args ? Object.keys(funcCall.args).sort().map(
-                    (k) => `${k}:${JSON.stringify(funcCall.args[k])}`
-                  ).join(",") : "";
-                  const callKey = `${funcCall.name}-${sortedArgs}`;
-                  if (!emittedToolCalls.has(callKey)) {
-                    emittedToolCalls.add(callKey);
-                    const toolCall = {
-                      id: funcCall.name,
-                      // Gemini uses name as ID
-                      name: funcCall.name,
-                      input: funcCall.args ?? {}
-                    };
-                    yield {
-                      type: "tool_use_start",
-                      toolCall: {
-                        id: toolCall.id,
-                        name: toolCall.name
-                      }
-                    };
-                    yield {
-                      type: "tool_use_end",
-                      toolCall
-                    };
-                  }
+                  streamToolCallCounter++;
+                  const toolCall = {
+                    id: `gemini_call_${streamToolCallCounter}`,
+                    name: funcCall.name,
+                    input: funcCall.args ?? {}
+                  };
+                  yield {
+                    type: "tool_use_start",
+                    toolCall: {
+                      id: toolCall.id,
+                      name: toolCall.name
+                    }
+                  };
+                  yield {
+                    type: "tool_use_end",
+                    toolCall
+                  };
                 }
               }
             }
@@ -5357,13 +5385,13 @@ var init_gemini = __esm({
        * Convert messages to Gemini format
        */
       convertMessages(messages) {
+        const toolNameByUseId = this.buildToolUseNameMap(messages);
+        const conversation = messages.filter((m) => m.role !== "system");
         const history = [];
         let lastUserMessage = "";
-        for (const msg of messages) {
-          if (msg.role === "system") {
-            continue;
-          }
-          const parts = this.convertContent(msg.content);
+        for (let i = 0; i < conversation.length; i++) {
+          const msg = conversation[i];
+          const isLastMessage = i === conversation.length - 1;
           if (msg.role === "user") {
             if (Array.isArray(msg.content) && msg.content[0]?.type === "tool_result") {
               const functionResponses = [];
@@ -5372,23 +5400,49 @@ var init_gemini = __esm({
                   const toolResult = block;
                   functionResponses.push({
                     functionResponse: {
-                      name: toolResult.tool_use_id,
-                      // Gemini uses name, we store it in tool_use_id
+                      // Gemini expects the function name in functionResponse.name.
+                      // Recover it from prior assistant tool_use blocks when possible.
+                      name: toolNameByUseId.get(toolResult.tool_use_id) ?? toolResult.tool_use_id,
                       response: { result: toolResult.content }
                     }
                   });
                 }
               }
-              history.push({ role: "user", parts: functionResponses });
+              if (isLastMessage) {
+                lastUserMessage = functionResponses;
+              } else {
+                history.push({ role: "user", parts: functionResponses });
+              }
             } else {
-              lastUserMessage = parts;
+              const parts = this.convertContent(msg.content);
+              if (isLastMessage) {
+                lastUserMessage = parts;
+              } else {
+                history.push({ role: "user", parts });
+              }
             }
           } else if (msg.role === "assistant") {
+            const parts = this.convertContent(msg.content);
             history.push({ role: "model", parts });
           }
         }
         return { history, lastMessage: lastUserMessage };
       }
+      /**
+       * Build a map from tool_use IDs to function names from assistant history.
+       */
+      buildToolUseNameMap(messages) {
+        const map = /* @__PURE__ */ new Map();
+        for (const msg of messages) {
+          if (msg.role !== "assistant" || !Array.isArray(msg.content)) continue;
+          for (const block of msg.content) {
+            if (block.type === "tool_use") {
+              map.set(block.id, block.name);
+            }
+          }
+        }
+        return map;
+      }
       /**
        * Convert content to Gemini parts
        */
@@ -5465,14 +5519,15 @@ var init_gemini = __esm({
         let textContent = "";
         const toolCalls = [];
         if (candidate?.content?.parts) {
+          let toolIndex = 0;
           for (const part of candidate.content.parts) {
             if ("text" in part && part.text) {
               textContent += part.text;
             }
             if ("functionCall" in part && part.functionCall) {
+              toolIndex++;
               toolCalls.push({
-                id: part.functionCall.name,
-                // Use name as ID for Gemini
+                id: `gemini_call_${toolIndex}`,
                 name: part.functionCall.name,
                 input: part.functionCall.args ?? {}
               });
@@ -9617,7 +9672,15 @@ Responses are short and direct by default. Lead with the answer or action, not r
 
 ## File Changes
 
-**Never output raw diff or unified diff format in your responses.** Use edit_file and write_file to make changes \u2014 Coco renders a visual diff automatically. Do not quote file contents back to the user after reading them unless explicitly asked.`;
+**Never output raw diff or unified diff format in your responses.** Use edit_file and write_file to make changes \u2014 Coco renders a visual diff automatically.
+
+## Output Discipline
+
+**NEVER echo file contents in your responses.** When you read a file, extract only the specific information needed \u2014 do NOT reprint whole files, functions, or large excerpts. The terminal shows which files you read and their line counts \u2014 the user does not need to see the content again.
+
+**During tool-calling iterations, keep text minimal.** A single short orienting line before tool calls is acceptable. Do NOT explain every step, narrate what you are about to do, or produce paragraphs between tool calls. Reserve explanatory text for your final response after all tools have completed.
+
+**Code blocks in responses are expensive.** Only include a code block when the user explicitly asks to see code, or when the code IS the deliverable (e.g., a script to paste in a terminal). Never include a code block to "show your work" when you can write the file directly instead.`;
     SHELL_METACHARACTERS = /[;|&`$(){}<>!\n\\'"]/;
     SAFE_COMMAND_VALIDATORS = {
       git: (args) => {
@@ -31538,6 +31601,15 @@ async function selectModelInteractively(models, currentModelId) {
     renderMenu();
   });
 }
+async function persistModelPreference(provider, model) {
+  try {
+    await saveProviderPreference(provider, model);
+  } catch (error) {
+    const reason = error instanceof Error ? error.message : String(error);
+    console.log(chalk2.yellow(`\u26A0 Could not persist model preference: ${reason}`));
+    console.log(chalk2.dim("  Model changed for this session only.\n"));
+  }
+}
 var modelCommand = {
   name: "model",
   aliases: ["m"],
@@ -31586,7 +31658,7 @@ var modelCommand = {
         return false;
       }
       session.config.provider.model = selectedModel;
-      await saveProviderPreference(currentProvider, selectedModel);
+      await persistModelPreference(currentProvider, selectedModel);
       const modelInfo2 = providerDef.models.find((m) => m.id === selectedModel);
       console.log(chalk2.green(`\u2713 Switched to ${modelInfo2?.name ?? selectedModel}
 `));
@@ -31608,7 +31680,7 @@ var modelCommand = {
     if (!foundInProvider) {
       console.log(chalk2.yellow(`Model "${newModel}" not in known list, setting anyway...`));
       session.config.provider.model = newModel;
-      await saveProviderPreference(currentProvider, newModel);
+      await persistModelPreference(currentProvider, newModel);
       console.log(chalk2.green(`\u2713 Model set to: ${newModel}
 `));
       return false;
@@ -31623,7 +31695,7 @@ var modelCommand = {
       return false;
     }
     session.config.provider.model = newModel;
-    await saveProviderPreference(currentProvider, newModel);
+    await persistModelPreference(currentProvider, newModel);
     const modelInfo = providerDef.models.find((m) => m.id === newModel);
     console.log(chalk2.green(`\u2713 Switched to ${modelInfo?.name ?? newModel}
 `));
@@ -47052,6 +47124,8 @@ var streamingIndicatorInterval = null;
 var streamingIndicatorFrame = 0;
 var STREAMING_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
 var getTerminalWidth2 = () => process.stdout.columns || 80;
+var MAX_STREAMING_CODE_LINES = 50;
+var STREAMING_CODE_HEAD_LINES = 40;
 function startStreamingIndicator() {
   if (streamingIndicatorActive) return;
   streamingIndicatorActive = true;
@@ -47375,11 +47449,17 @@ function renderSimpleCodeBlock(lang, lines, blockId) {
   const isDiff = lang === "diff" || !lang && looksLikeDiff(lines);
   const title = lang || "Code";
   console.log(chalk2.dim(`${title} \xB7 #${blockId}`));
+  let displayLines = lines;
+  let truncatedCount = 0;
+  if (!isDiff && lines.length > MAX_STREAMING_CODE_LINES) {
+    displayLines = lines.slice(0, STREAMING_CODE_HEAD_LINES);
+    truncatedCount = lines.length - STREAMING_CODE_HEAD_LINES;
+  }
   const bgDel = chalk2.bgRgb(80, 20, 20);
   const bgAdd = chalk2.bgRgb(20, 60, 20);
   let oldLineNo = 0;
   let newLineNo = 0;
-  for (const line of lines) {
+  for (const line of displayLines) {
     if (isDiff) {
       const hunkMatch = line.match(/^@@ -(\d+)(?:,\d+)? \+(\d+)(?:,\d+)? @@/);
       if (hunkMatch) {
@@ -47407,6 +47487,9 @@ function renderSimpleCodeBlock(lang, lines, blockId) {
       }
     }
   }
+  if (truncatedCount > 0) {
+    console.log(chalk2.dim(`  \u2026 ${truncatedCount} more lines \xB7 /copy ${blockId} for full content`));
+  }
   console.log(chalk2.dim(`  #${blockId} \xB7 /copy ${blockId}`));
 }
 function formatDiffLineNo(line, oldLineNo, newLineNo) {
@@ -50058,6 +50141,8 @@ async function executeAgentTurn(session, userMessage, provider, toolRegistry, op
   const maxIterations = session.config.agent.maxToolIterations;
   const toolErrorCounts = /* @__PURE__ */ new Map();
   const MAX_CONSECUTIVE_TOOL_ERRORS = 3;
+  const ITERATION_LIMIT_WARNING_RATIO = 0.75;
+  const ITERATION_LIMIT_SUMMARY_PROMPT = `[System: You have now used all allowed iterations and tool calls are no longer available. Write your final response immediately: (1) briefly state what was completed, (2) describe what still needs to be done, (3) give specific next steps so the user can continue. Be concise and direct.]`;
   const INLINE_RESULT_MAX_CHARS = 8e3;
   const INLINE_RESULT_HEAD_CHARS = 6500;
   const INLINE_RESULT_TAIL_CHARS = 1e3;
@@ -50072,6 +50157,8 @@ ${tail}`;
   }
   while (iteration < maxIterations) {
     iteration++;
+    const isLastIteration = iteration === maxIterations;
+    const iterationTextChunks = [];
     if (options.signal?.aborted) {
       return abortReturn();
     }
@@ -50099,7 +50186,7 @@ ${tail}`;
             }
             responseContent += chunk.text;
             finalContent += chunk.text;
-            options.onStream?.(chunk);
+            iterationTextChunks.push(chunk);
           }
           if (chunk.type === "tool_use_start" && chunk.toolCall) {
             flushLineBuffer();
@@ -50176,6 +50263,11 @@ ${tail}`;
         error: errorMsg
       };
     }
+    if (collectedToolCalls.length === 0) {
+      for (const bufferedChunk of iterationTextChunks) {
+        options.onStream?.(bufferedChunk);
+      }
+    }
     const inputText = messages.map((m) => {
       if (typeof m.content === "string") return m.content;
       try {
@@ -50351,10 +50443,20 @@ ${tail}`;
       });
       const declineReason = declinedTools.get(toolCall.id);
       if (declineReason) {
+        let skipContent;
+        if (declineReason === "User declined") {
+          skipContent = "The user explicitly declined this tool call. You MUST find a different approach \u2014 do not retry the same action or parameters.";
+        } else if (declineReason.toLowerCase().includes("timeout") || declineReason.toLowerCase().includes("timed out")) {
+          skipContent = `Tool execution timed out: ${declineReason}. Try a simpler or faster alternative, or break the task into smaller steps.`;
+        } else if (declineReason.toLowerCase().includes("abort")) {
+          skipContent = "Tool execution was cancelled.";
+        } else {
+          skipContent = `Tool execution was skipped: ${declineReason}`;
+        }
         toolResults.push({
           type: "tool_result",
           tool_use_id: toolCall.id,
-          content: `Tool execution was declined: ${declineReason}`,
+          content: skipContent,
           is_error: true
         });
         continue;
@@ -50408,6 +50510,28 @@ ${tail}`;
         }
       }
     }
+    if (toolResults.length > 0) {
+      const warningThreshold = Math.ceil(maxIterations * ITERATION_LIMIT_WARNING_RATIO);
+      const lastIdx = toolResults.length - 1;
+      const last = toolResults[lastIdx];
+      if (isLastIteration && !stuckInErrorLoop) {
+        toolResults[lastIdx] = {
+          ...last,
+          content: typeof last.content === "string" ? last.content + `
+
+${ITERATION_LIMIT_SUMMARY_PROMPT}` : ITERATION_LIMIT_SUMMARY_PROMPT
+        };
+      } else if (iteration === warningThreshold) {
+        const remaining = maxIterations - iteration;
+        const warning = `
+
+[System: Iteration budget warning \u2014 ${iteration} of ${maxIterations} iterations used, ${remaining} remaining. Begin wrapping up your task. Prioritize the most critical remaining steps.]`;
+        toolResults[lastIdx] = {
+          ...last,
+          content: typeof last.content === "string" ? last.content + warning : warning
+        };
+      }
+    }
     const assistantContent = response.content ? [{ type: "text", text: response.content }, ...toolUses] : toolUses;
     addMessage(session, {
       role: "assistant",
@@ -50452,14 +50576,38 @@ ${tail}`;
       }
       break;
     }
-  }
-  if (iteration >= maxIterations) {
-    const notice = `
+    if (isLastIteration && toolResults.length > 0) {
+      let summaryThinkingEnded = false;
+      options.onThinkingStart?.();
+      try {
+        const finalMessages = getConversationContext(session, toolRegistry);
+        for await (const chunk of provider.streamWithTools(finalMessages, {
+          tools: [],
+          maxTokens: session.config.provider.maxTokens,
+          signal: options.signal
+        })) {
+          if (options.signal?.aborted) break;
+          if (chunk.type === "text" && chunk.text) {
+            if (!summaryThinkingEnded) {
+              options.onThinkingEnd?.();
+              summaryThinkingEnded = true;
+            }
+            finalContent += chunk.text;
+            options.onStream?.(chunk);
+          }
+          if (chunk.type === "done") break;
+        }
+      } catch {
+        const notice = `
 
----
-_Reached the iteration limit (${maxIterations}). The task may be incomplete. You can say "continue" to resume._`;
-    finalContent += notice;
-    options.onStream?.({ type: "text", text: notice });
+I have reached the maximum iteration limit (${maxIterations}). The task may be incomplete. Type "continue" to give me more iterations, or describe what you'd like me to focus on next.`;
+        finalContent += notice;
+        options.onStream?.({ type: "text", text: notice });
+      } finally {
+        if (!summaryThinkingEnded) options.onThinkingEnd?.();
+      }
+      break;
+    }
   }
   options.onStream?.({ type: "done" });
   return {