@corbat-tech/coco 2.8.1 → 2.8.2

package/dist/cli/index.js

@@ -1031,6 +1031,7 @@ var init_anthropic = __esm({
           };
           const timeoutInterval = setInterval(checkTimeout, 5e3);
           try {
+            let streamStopReason;
             for await (const event of stream) {
               lastActivityTime = Date.now();
               if (event.type === "content_block_delta") {
@@ -1038,9 +1039,14 @@ var init_anthropic = __esm({
                 if (delta.type === "text_delta" && delta.text) {
                   yield { type: "text", text: delta.text };
                 }
+              } else if (event.type === "message_delta") {
+                const delta = event.delta;
+                if (delta.stop_reason) {
+                  streamStopReason = this.mapStopReason(delta.stop_reason);
+                }
               }
             }
-            yield { type: "done" };
+            yield { type: "done", stopReason: streamStopReason };
           } finally {
             clearInterval(timeoutInterval);
           }
@@ -1077,9 +1083,15 @@ var init_anthropic = __esm({
           };
           const timeoutInterval = setInterval(checkTimeout, 5e3);
           try {
+            let streamStopReason;
             for await (const event of stream) {
               lastActivityTime = Date.now();
-              if (event.type === "content_block_start") {
+              if (event.type === "message_delta") {
+                const delta = event.delta;
+                if (delta.stop_reason) {
+                  streamStopReason = this.mapStopReason(delta.stop_reason);
+                }
+              } else if (event.type === "content_block_start") {
                 const contentBlock = event.content_block;
                 if (contentBlock.type === "tool_use") {
                   if (currentToolCall) {
@@ -1150,7 +1162,7 @@ var init_anthropic = __esm({
                 }
               }
             }
-            yield { type: "done" };
+            yield { type: "done", stopReason: streamStopReason };
           } finally {
             clearInterval(timeoutInterval);
           }
@@ -1653,13 +1665,18 @@ var init_openai = __esm({
             stream: true,
             ...supportsTemp && { temperature: options?.temperature ?? this.config.temperature ?? 0 }
           });
+          let streamStopReason;
           for await (const chunk of stream) {
             const delta = chunk.choices[0]?.delta;
             if (delta?.content) {
               yield { type: "text", text: delta.content };
             }
+            const finishReason = chunk.choices[0]?.finish_reason;
+            if (finishReason) {
+              streamStopReason = this.mapFinishReason(finishReason);
+            }
           }
-          yield { type: "done" };
+          yield { type: "done", stopReason: streamStopReason };
         } catch (error) {
           throw this.handleError(error);
         }
@@ -1724,6 +1741,7 @@ var init_openai = __esm({
             return input;
           };
           try {
+            let streamStopReason;
             for await (const chunk of stream) {
               const delta = chunk.choices[0]?.delta;
               if (delta?.content || delta?.tool_calls) {
@@ -1770,6 +1788,9 @@ var init_openai = __esm({
                 }
               }
               const finishReason = chunk.choices[0]?.finish_reason;
+              if (finishReason) {
+                streamStopReason = this.mapFinishReason(finishReason);
+              }
               if (finishReason && toolCallBuilders.size > 0) {
                 for (const [, builder] of toolCallBuilders) {
                   yield {
@@ -1794,7 +1815,7 @@ var init_openai = __esm({
                 }
               };
             }
-            yield { type: "done" };
+            yield { type: "done", stopReason: streamStopReason };
           } finally {
             clearInterval(timeoutInterval);
           }
@@ -3601,7 +3622,7 @@ var init_codex = __esm({
             }
           }
         }
-        yield { type: "done" };
+        yield { type: "done", stopReason: response.stopReason };
       }
       /**
        * Stream a chat response with tool use
@@ -3769,13 +3790,18 @@ var init_gemini = __esm({
           const { history, lastMessage } = this.convertMessages(messages);
           const chat = model.startChat({ history });
           const result = await chat.sendMessageStream(lastMessage);
+          let streamStopReason;
           for await (const chunk of result.stream) {
             const text13 = chunk.text();
             if (text13) {
               yield { type: "text", text: text13 };
             }
+            const finishReason = chunk.candidates?.[0]?.finishReason;
+            if (finishReason) {
+              streamStopReason = this.mapFinishReason(finishReason);
+            }
           }
-          yield { type: "done" };
+          yield { type: "done", stopReason: streamStopReason };
         } catch (error) {
           throw this.handleError(error);
         }
@@ -3810,11 +3836,16 @@ var init_gemini = __esm({
           const chat = model.startChat({ history });
           const result = await chat.sendMessageStream(lastMessage);
           const emittedToolCalls = /* @__PURE__ */ new Set();
+          let streamStopReason;
           for await (const chunk of result.stream) {
             const text13 = chunk.text();
             if (text13) {
               yield { type: "text", text: text13 };
             }
+            const finishReason = chunk.candidates?.[0]?.finishReason;
+            if (finishReason) {
+              streamStopReason = this.mapFinishReason(finishReason);
+            }
             const candidate = chunk.candidates?.[0];
             if (candidate?.content?.parts) {
               for (const part of candidate.content.parts) {
@@ -3848,7 +3879,7 @@ var init_gemini = __esm({
               }
             }
           }
-          yield { type: "done" };
+          yield { type: "done", stopReason: streamStopReason };
         } catch (error) {
           throw this.handleError(error);
         }
@@ -44640,6 +44671,7 @@ async function executeAgentTurn(session, userMessage, provider, toolRegistry, op
     let responseContent = "";
     const collectedToolCalls = [];
     let thinkingEnded = false;
+    let lastStopReason;
     const toolCallBuilders = /* @__PURE__ */ new Map();
     for await (const chunk of provider.streamWithTools(messages, {
       tools,
@@ -44694,6 +44726,9 @@ async function executeAgentTurn(session, userMessage, provider, toolRegistry, op
         }
       }
       if (chunk.type === "done") {
+        if (chunk.stopReason) {
+          lastStopReason = chunk.stopReason;
+        }
         if (!thinkingEnded) {
           options.onThinkingEnd?.();
           thinkingEnded = true;
@@ -44712,6 +44747,14 @@ async function executeAgentTurn(session, userMessage, provider, toolRegistry, op
       if (options.signal?.aborted) {
         return abortReturn();
       }
+      if (lastStopReason === "max_tokens" && responseContent) {
+        addMessage(session, { role: "assistant", content: responseContent });
+        addMessage(session, {
+          role: "user",
+          content: "[System: Your previous response was cut off due to the output token limit. Continue exactly where you left off.]"
+        });
+        continue;
+      }
       addMessage(session, { role: "assistant", content: responseContent });
       break;
     }
@@ -44933,6 +44976,14 @@ async function executeAgentTurn(session, userMessage, provider, toolRegistry, op
       break;
     }
   }
+  if (iteration >= maxIterations) {
+    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 });
+  }
   options.onStream?.({ type: "done" });
   return {
     content: finalContent,