@corbat-tech/coco 2.8.1 → 2.9.0

package/dist/index.d.ts

@@ -1032,6 +1032,7 @@ interface StreamChunk {
     type: "text" | "tool_use_start" | "tool_use_delta" | "tool_use_end" | "done";
     text?: string;
     toolCall?: Partial<ToolCall>;
+    stopReason?: "end_turn" | "max_tokens" | "stop_sequence" | "tool_use";
 }
 /**
  * LLM Provider interface

package/dist/index.js

@@ -25,6 +25,7 @@ import JSON5 from 'json5';
 import 'events';
 import 'minimatch';
 import { simpleGit } from 'simple-git';
+import { diffWords } from 'diff';
 import hljs from 'highlight.js/lib/core';
 import bash from 'highlight.js/lib/languages/bash';
 import css from 'highlight.js/lib/languages/css';
@@ -11803,6 +11804,7 @@ var AnthropicProvider = class {
       };
       const timeoutInterval = setInterval(checkTimeout, 5e3);
       try {
+        let streamStopReason;
         for await (const event of stream) {
           lastActivityTime = Date.now();
           if (event.type === "content_block_delta") {
@@ -11810,9 +11812,14 @@ var AnthropicProvider = class {
             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);
       }
@@ -11849,9 +11856,15 @@ var AnthropicProvider = class {
       };
       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) {
@@ -11922,7 +11935,7 @@ var AnthropicProvider = class {
             }
           }
         }
-        yield { type: "done" };
+        yield { type: "done", stopReason: streamStopReason };
       } finally {
         clearInterval(timeoutInterval);
       }
@@ -12418,13 +12431,18 @@ var OpenAIProvider = class {
         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);
     }
@@ -12489,6 +12507,7 @@ var OpenAIProvider = class {
         return input;
       };
       try {
+        let streamStopReason;
         for await (const chunk of stream) {
           const delta = chunk.choices[0]?.delta;
           if (delta?.content || delta?.tool_calls) {
@@ -12535,6 +12554,9 @@ var OpenAIProvider = class {
             }
           }
           const finishReason = chunk.choices[0]?.finish_reason;
+          if (finishReason) {
+            streamStopReason = this.mapFinishReason(finishReason);
+          }
           if (finishReason && toolCallBuilders.size > 0) {
             for (const [, builder] of toolCallBuilders) {
               yield {
@@ -12559,7 +12581,7 @@ var OpenAIProvider = class {
             }
           };
         }
-        yield { type: "done" };
+        yield { type: "done", stopReason: streamStopReason };
       } finally {
         clearInterval(timeoutInterval);
       }
@@ -13322,7 +13344,7 @@ var CodexProvider = class {
         }
       }
     }
-    yield { type: "done" };
+    yield { type: "done", stopReason: response.stopReason };
   }
   /**
    * Stream a chat response with tool use
@@ -13475,13 +13497,18 @@ var GeminiProvider = class {
       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 text = chunk.text();
         if (text) {
           yield { type: "text", text };
         }
+        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);
     }
@@ -13516,11 +13543,16 @@ var GeminiProvider = class {
       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 text = chunk.text();
         if (text) {
           yield { type: "text", text };
         }
+        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) {
@@ -13554,7 +13586,7 @@ var GeminiProvider = class {
           }
         }
       }
-      yield { type: "done" };
+      yield { type: "done", stopReason: streamStopReason };
     } catch (error) {
       throw this.handleError(error);
     }
@@ -18884,6 +18916,10 @@ function highlightLine(line, lang) {
 }
 
 // src/cli/repl/output/diff-renderer.ts
+var bgDeleteLine = chalk3.bgRgb(80, 20, 20);
+var bgAddLine = chalk3.bgRgb(20, 60, 20);
+var bgDeleteWord = chalk3.bgRgb(160, 40, 40);
+var bgAddWord = chalk3.bgRgb(40, 120, 40);
 function parseDiff(raw) {
   const files = [];
   const lines = raw.split("\n");
@@ -19008,6 +19044,46 @@ function parseHunk(lines, start) {
     nextIndex: i
   };
 }
+function pairAdjacentLines(lines) {
+  const pairs = [];
+  let i = 0;
+  while (i < lines.length) {
+    const deleteStart = i;
+    while (i < lines.length && lines[i].type === "delete") i++;
+    const deleteEnd = i;
+    const addStart = i;
+    while (i < lines.length && lines[i].type === "add") i++;
+    const addEnd = i;
+    const deleteCount = deleteEnd - deleteStart;
+    const addCount = addEnd - addStart;
+    if (deleteCount > 0 && addCount > 0) {
+      const pairCount = Math.min(deleteCount, addCount);
+      for (let j = 0; j < pairCount; j++) {
+        pairs.push({ deleteIdx: deleteStart + j, addIdx: addStart + j });
+      }
+    }
+    if (i === deleteEnd && i === addEnd) {
+      i++;
+    }
+  }
+  return pairs;
+}
+function highlightWordChanges(deletedContent, addedContent) {
+  const changes = diffWords(deletedContent, addedContent);
+  let styledDelete = "";
+  let styledAdd = "";
+  for (const change of changes) {
+    if (change.added) {
+      styledAdd += bgAddWord(change.value);
+    } else if (change.removed) {
+      styledDelete += bgDeleteWord(change.value);
+    } else {
+      styledDelete += bgDeleteLine(change.value);
+      styledAdd += bgAddLine(change.value);
+    }
+  }
+  return { styledDelete, styledAdd };
+}
 var getTerminalWidth = () => process.stdout.columns || 80;
 function stripAnsi(str) {
   return str.replace(/\x1b\[[0-9;]*m/g, "");
@@ -19076,32 +19152,64 @@ function renderFileBlock(file, opts) {
         ) + hunkLabel
       );
     }
-    for (const line of hunk.lines) {
+    const pairs = pairAdjacentLines(hunk.lines);
+    const pairedDeleteIndices = new Set(pairs.map((p4) => p4.deleteIdx));
+    const pairedAddIndices = new Set(pairs.map((p4) => p4.addIdx));
+    const pairByAdd = new Map(pairs.map((p4) => [p4.addIdx, p4.deleteIdx]));
+    const wordHighlights = /* @__PURE__ */ new Map();
+    for (const pair of pairs) {
+      const delLine = hunk.lines[pair.deleteIdx];
+      const addLine = hunk.lines[pair.addIdx];
+      wordHighlights.set(pair.deleteIdx, highlightWordChanges(delLine.content, addLine.content));
+    }
+    for (let li = 0; li < hunk.lines.length; li++) {
+      const line = hunk.lines[li];
       const lineNo = formatLineNo(line, showLineNumbers);
       const prefix = line.type === "add" ? "+" : line.type === "delete" ? "-" : " ";
-      let content = line.content;
-      if (line.type !== "delete" && lang) {
-        content = highlightLine(content, lang);
-      }
-      const lineStr = `${lineNo}${prefix} ${content}`;
-      const plainLen = stripAnsi(lineStr).length;
-      const pad = Math.max(0, contentWidth - plainLen);
       if (line.type === "add") {
+        const isPaired = pairedAddIndices.has(li);
+        let content;
+        if (isPaired) {
+          const delIdx = pairByAdd.get(li);
+          content = wordHighlights.get(delIdx).styledAdd;
+        } else {
+          content = line.content;
+        }
+        const innerText = `${lineNo}${prefix} ${content}`;
+        const plainLen = stripAnsi(innerText).length + 1;
+        const pad = Math.max(0, contentWidth - plainLen);
         console.log(
-          chalk3.magenta("\u2502") + chalk3.green(` ${lineStr}`) + " ".repeat(pad) + " " + chalk3.magenta("\u2502")
+          chalk3.magenta("\u2502") + bgAddLine(` ${innerText}` + " ".repeat(pad + 2)) + chalk3.magenta("\u2502")
         );
       } else if (line.type === "delete") {
+        const isPaired = pairedDeleteIndices.has(li);
+        let content;
+        if (isPaired) {
+          content = wordHighlights.get(li).styledDelete;
+        } else {
+          content = line.content;
+        }
+        const innerText = `${lineNo}${prefix} ${content}`;
+        const plainLen = stripAnsi(innerText).length + 1;
+        const pad = Math.max(0, contentWidth - plainLen);
         console.log(
-          chalk3.magenta("\u2502") + chalk3.red(` ${lineStr}`) + " ".repeat(pad) + " " + chalk3.magenta("\u2502")
+          chalk3.magenta("\u2502") + bgDeleteLine(` ${innerText}` + " ".repeat(pad + 2)) + chalk3.magenta("\u2502")
         );
       } else {
+        let content = line.content;
+        if (lang) {
+          content = highlightLine(content, lang);
+        }
+        const lineStr = `${lineNo}${prefix} ${content}`;
+        const plainLen = stripAnsi(lineStr).length;
+        const pad = Math.max(0, contentWidth - plainLen);
         console.log(
           chalk3.magenta("\u2502") + chalk3.dim(` ${lineStr}`) + " ".repeat(pad) + " " + chalk3.magenta("\u2502")
         );
       }
     }
   }
-  console.log(chalk3.magenta("\u2570" + "\u2500".repeat(maxWidth - 2) + "\u256F"));
+  console.log(chalk3.magenta("\u2570" + "\u2500".repeat(Math.max(0, maxWidth - 2)) + "\u256F"));
 }
 function formatLineNo(line, show) {
   if (!show) return "";