@zhongqian97-code/ecode 0.5.4 → 0.5.6

package/dist/index.js

@@ -116,6 +116,7 @@ import { Box as Box6, useInput as useInput2, useStdout, useStdin } from "ink";
 // src/providers/openai.ts
 import OpenAI from "openai";
 function createOpenAIProvider(profile) {
+  const THINK_END = "</think>";
   const openai = new OpenAI({
     baseURL: profile.baseUrl,
     apiKey: profile.apiKey
@@ -149,14 +150,21 @@ function createOpenAIProvider(profile) {
       return {
         [Symbol.asyncIterator]: async function* () {
           let thinkPhase = "pre";
-          let afterReasoningDetails = false;
+          let scanningForClose = false;
+          let closeSearchBuffer = "";
+          let visibleTextTail = "";
+          function rememberVisibleText(text) {
+            if (!text) return;
+            visibleTextTail = (visibleTextTail + text).slice(-256);
+          }
+          function hasNearbyVisibleOpenThink(maxDistance = 4) {
+            const openIdx = visibleTextTail.lastIndexOf("<think>");
+            const closeIdx = visibleTextTail.lastIndexOf(THINK_END);
+            if (openIdx === -1 || openIdx < closeIdx) return false;
+            return visibleTextTail.length - (openIdx + "<think>".length) <= maxDistance;
+          }
           function processContent(raw) {
             if (!raw) return { text: "", thinking: "" };
-            if (afterReasoningDetails) {
-              afterReasoningDetails = false;
-              thinkPhase = "post";
-              return { text: raw.startsWith("</think>") ? raw.slice(8) : raw, thinking: "" };
-            }
             if (thinkPhase === "post") return { text: raw, thinking: "" };
             if (thinkPhase === "pre") {
               if (raw.startsWith("<think>")) {
@@ -167,11 +175,11 @@ function createOpenAIProvider(profile) {
                 return { text: raw, thinking: "" };
               }
             }
-            const endIdx = raw.indexOf("</think>");
+            const endIdx = raw.indexOf(THINK_END);
             if (endIdx === -1) return { text: "", thinking: raw };
             const thinking = raw.slice(0, endIdx);
             thinkPhase = "post";
-            return { text: raw.slice(endIdx + 8), thinking };
+            return { text: raw.slice(endIdx + THINK_END.length), thinking };
           }
           const requestParams = {
             model: profile.model,
@@ -197,7 +205,7 @@ function createOpenAIProvider(profile) {
               reasoningAccumulator += delta.reasoning_content;
             }
             if (delta.reasoning_details && delta.reasoning_details.length > 0) {
-              afterReasoningDetails = true;
+              scanningForClose = true;
               for (const rd of delta.reasoning_details) {
                 const id = rd.id ?? "";
                 const text = rd.text ?? "";
@@ -233,8 +241,36 @@ function createOpenAIProvider(profile) {
               }
             }
             const isLast = choice.finish_reason != null;
-            const raw = delta.reasoning_details && delta.reasoning_details.length > 0 ? "" : delta.content ?? "";
-            const { text: filteredText, thinking: thinkContent } = processContent(raw);
+            let rawToProcess;
+            if (delta.reasoning_details && delta.reasoning_details.length > 0) {
+              rawToProcess = "";
+            } else if (scanningForClose) {
+              closeSearchBuffer += delta.content ?? "";
+              const closeIdx = closeSearchBuffer.indexOf(THINK_END);
+              if (closeIdx !== -1) {
+                const openIdx = closeSearchBuffer.lastIndexOf("<think>", closeIdx);
+                const hasNearbyLiteralPair = openIdx !== -1 && closeIdx - (openIdx + "<think>".length) <= 8;
+                const afterClose = closeSearchBuffer.slice(closeIdx + THINK_END.length);
+                rawToProcess = hasNearbyLiteralPair ? closeSearchBuffer : afterClose.length > 0 ? afterClose : closeSearchBuffer.slice(0, closeIdx);
+                scanningForClose = false;
+                closeSearchBuffer = "";
+                thinkPhase = "post";
+              } else if (isLast) {
+                rawToProcess = closeSearchBuffer;
+                scanningForClose = false;
+                closeSearchBuffer = "";
+                thinkPhase = "post";
+              } else {
+                rawToProcess = "";
+              }
+            } else {
+              rawToProcess = delta.content ?? "";
+            }
+            if (thinkPhase === "post" && rawToProcess.endsWith(THINK_END) && !rawToProcess.includes("<think>") && !hasNearbyVisibleOpenThink()) {
+              rawToProcess = rawToProcess.slice(0, -THINK_END.length);
+            }
+            const { text: filteredText, thinking: thinkContent } = processContent(rawToProcess);
+            rememberVisibleText(filteredText);
             if (thinkContent) {
               reasoningAccumulator += thinkContent;
             }
@@ -3522,6 +3558,7 @@ var SkillRegistry = class {
 };
 
 // src/pipe.ts
+var PIPE_SYSTEM_PROMPT = "You are a helpful assistant running in headless pipe mode. You have access to read-only file tools (read, glob, grep). Answer concisely and accurately. Follow the user's requested language, format, and length exactly. Do not reveal chain-of-thought or emit raw <think> / </think> tags in the final answer; if you must refer to them, describe them in plain language instead.";
 var PIPE_TOOLS = [READ_TOOL, GLOB_TOOL, GREP_TOOL];
 function emit(out, event) {
   out.write(JSON.stringify(event) + "\n");
@@ -3541,15 +3578,21 @@ async function executeToolCall(name, args) {
   }
   return `Unknown tool: ${name}`;
 }
-async function runPipe(prompt, llm, out = process.stdout) {
-  const messages = [{ role: "user", content: prompt }];
+async function runPipe(prompt, llm, out = process.stdout, systemPrompt) {
+  const messages = [
+    { role: "system", content: systemPrompt ?? PIPE_SYSTEM_PROMPT },
+    { role: "user", content: prompt }
+  ];
   while (true) {
     let assistantText = "";
+    const pendingTextChunks = [];
     let lastUsage;
+    let lastReasoning;
+    let lastReasoningDetails;
     const toolCalls = [];
     for await (const chunk of llm.stream(messages, PIPE_TOOLS)) {
       if (chunk.text) {
-        emit(out, { type: "chunk", text: chunk.text });
+        pendingTextChunks.push(chunk.text);
         assistantText += chunk.text;
       }
       if (chunk.reasoning) {
@@ -3558,6 +3601,8 @@ async function runPipe(prompt, llm, out = process.stdout) {
       if (chunk.done) {
         if (chunk.toolCalls) toolCalls.push(...chunk.toolCalls);
         if (chunk.usage) lastUsage = chunk.usage;
+        if (chunk.reasoning) lastReasoning = chunk.reasoning;
+        if (chunk.reasoningDetails) lastReasoningDetails = chunk.reasoningDetails;
       }
     }
     if (toolCalls.length > 0) {
@@ -3568,7 +3613,9 @@ async function runPipe(prompt, llm, out = process.stdout) {
           id: tc.id,
           type: "function",
           function: { name: tc.name, arguments: tc.arguments }
-        }))
+        })),
+        ...lastReasoning ? { reasoning_content: lastReasoning } : {},
+        ...lastReasoningDetails ? { reasoning_details: lastReasoningDetails } : {}
       });
       for (const tc of toolCalls) {
         emit(out, { type: "tool_call", id: tc.id, name: tc.name, arguments: tc.arguments });
@@ -3577,6 +3624,9 @@ async function runPipe(prompt, llm, out = process.stdout) {
         messages.push({ role: "tool", tool_call_id: tc.id, content });
       }
     } else {
+      for (const text of pendingTextChunks) {
+        emit(out, { type: "chunk", text });
+      }
       const doneEvent = lastUsage ? { type: "done", usage: lastUsage } : { type: "done" };
       emit(out, doneEvent);
       break;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhongqian97-code/ecode",
-  "version": "0.5.4",
+  "version": "0.5.6",
   "description": "A minimal Claude Code clone with REPL interface and bash tool calling",
   "type": "module",
   "author": "zhongqian97-code",