kimiflare 0.13.4 → 0.13.6

package/dist/index.js

@@ -87,8 +87,6 @@ async function* readSSE(stream, signal) {
   const reader = stream.getReader();
   const decoder = new TextDecoder("utf-8");
   let buffer = "";
-  const onAbort = () => reader.cancel(new DOMException("aborted", "AbortError"));
-  signal?.addEventListener("abort", onAbort, { once: true });
   try {
     while (true) {
       if (signal?.aborted) throw new DOMException("aborted", "AbortError");
@@ -108,7 +106,6 @@ async function* readSSE(stream, signal) {
     const tail = extractData(buffer.trim());
     if (tail !== null) yield tail;
   } finally {
-    signal?.removeEventListener("abort", onAbort);
     reader.releaseLock();
   }
 }
@@ -233,87 +230,57 @@ async function* parseStream(body, signal) {
   const toolCalls = /* @__PURE__ */ new Map();
   let lastUsage = null;
   let finishReason = null;
-  let timeoutId = null;
-  let timedOut = false;
-  const resetTimeout = () => {
-    if (timeoutId) clearTimeout(timeoutId);
-    timeoutId = setTimeout(() => {
-      timedOut = true;
-    }, STREAM_TIMEOUT_MS);
-  };
-  try {
-    for await (const dataStr of readSSE(body, signal)) {
-      if (timedOut) {
-        throw new KimiApiError(
-          `kimiflare: stream timed out (no data for ${STREAM_TIMEOUT_MS / 1e3}s)`,
-          void 0,
-          void 0
-        );
-      }
-      resetTimeout();
-      if (dataStr === "[DONE]") break;
-      let chunk = null;
-      try {
-        chunk = JSON.parse(dataStr);
-      } catch {
-        continue;
+  for await (const dataStr of readSSE(body, signal)) {
+    if (dataStr === "[DONE]") break;
+    let chunk = null;
+    try {
+      chunk = JSON.parse(dataStr);
+    } catch {
+      continue;
+    }
+    if (!chunk) continue;
+    if (chunk.usage) {
+      lastUsage = chunk.usage;
+      yield { type: "usage", usage: chunk.usage };
+    }
+    const choice = chunk.choices?.[0];
+    if (!choice) continue;
+    const d = choice.delta;
+    if (d) {
+      if (typeof d.reasoning_content === "string" && d.reasoning_content.length) {
+        yield { type: "reasoning", delta: d.reasoning_content };
       }
-      if (!chunk) continue;
-      if (chunk.usage) {
-        lastUsage = chunk.usage;
-        yield { type: "usage", usage: chunk.usage };
+      if (typeof d.content === "string" && d.content.length) {
+        yield { type: "text", delta: d.content };
       }
-      const choice = chunk.choices?.[0];
-      if (!choice) continue;
-      const d = choice.delta;
-      if (d) {
-        if (typeof d.reasoning_content === "string" && d.reasoning_content.length) {
-          yield { type: "reasoning", delta: d.reasoning_content };
-        }
-        if (typeof d.content === "string" && d.content.length) {
-          yield { type: "text", delta: d.content };
-        }
-        if (Array.isArray(d.tool_calls)) {
-          for (const tc of d.tool_calls) {
-            const idx = typeof tc.index === "number" ? tc.index : 0;
-            let buf = toolCalls.get(idx);
-            const incomingName = tc.function?.name ?? null;
-            const incomingId = tc.id ?? null;
-            if (!buf) {
-              buf = { id: incomingId ?? `tc_${idx}`, name: incomingName ?? "", args: "" };
-              toolCalls.set(idx, buf);
-              if (buf.name) {
-                yield { type: "tool_call_start", index: idx, id: buf.id, name: buf.name };
-              }
-            } else {
-              if (!buf.name && incomingName) {
-                buf.name = incomingName;
-                yield { type: "tool_call_start", index: idx, id: buf.id, name: buf.name };
-              }
-              if (buf.id.startsWith("tc_") && incomingId) buf.id = incomingId;
+      if (Array.isArray(d.tool_calls)) {
+        for (const tc of d.tool_calls) {
+          const idx = typeof tc.index === "number" ? tc.index : 0;
+          let buf = toolCalls.get(idx);
+          const incomingName = tc.function?.name ?? null;
+          const incomingId = tc.id ?? null;
+          if (!buf) {
+            buf = { id: incomingId ?? `tc_${idx}`, name: incomingName ?? "", args: "" };
+            toolCalls.set(idx, buf);
+            if (buf.name) {
+              yield { type: "tool_call_start", index: idx, id: buf.id, name: buf.name };
             }
-            const argDelta = tc.function?.arguments;
-            if (typeof argDelta === "string" && argDelta.length) {
-              buf.args += argDelta;
-              yield { type: "tool_call_args", index: idx, argsDelta: argDelta };
+          } else {
+            if (!buf.name && incomingName) {
+              buf.name = incomingName;
+              yield { type: "tool_call_start", index: idx, id: buf.id, name: buf.name };
             }
+            if (buf.id.startsWith("tc_") && incomingId) buf.id = incomingId;
+          }
+          const argDelta = tc.function?.arguments;
+          if (typeof argDelta === "string" && argDelta.length) {
+            buf.args += argDelta;
+            yield { type: "tool_call_args", index: idx, argsDelta: argDelta };
           }
         }
       }
-      if (choice.finish_reason) finishReason = choice.finish_reason;
     }
-    for (const [idx, buf] of [...toolCalls.entries()].sort((a, b) => a[0] - b[0])) {
-      if (!buf.name) continue;
-      yield {
-        type: "tool_call_complete",
-        index: idx,
-        id: buf.id,
-        name: buf.name,
-        arguments: buf.args
-      };
-    }
-  } finally {
-    if (timeoutId) clearTimeout(timeoutId);
+    if (choice.finish_reason) finishReason = choice.finish_reason;
   }
   for (const [idx, buf] of [...toolCalls.entries()].sort((a, b) => a[0] - b[0])) {
     if (!buf.name) continue;
@@ -387,7 +354,7 @@ function sleep(ms, signal) {
     signal?.addEventListener("abort", onAbort, { once: true });
   });
 }
-var RETRYABLE_CODES, MAX_ATTEMPTS, STREAM_TIMEOUT_MS;
+var RETRYABLE_CODES, MAX_ATTEMPTS;
 var init_client = __esm({
   "src/agent/client.ts"() {
     "use strict";
@@ -396,7 +363,6 @@ var init_client = __esm({
     init_messages();
     RETRYABLE_CODES = /* @__PURE__ */ new Set([3040]);
     MAX_ATTEMPTS = 5;
-    STREAM_TIMEOUT_MS = 6e4;
   }
 });
 
@@ -3626,60 +3592,6 @@ function findImagePaths(text) {
   }
   return [...new Set(paths)];
 }
-function stripImagesFromHistory(messages) {
-  for (const m of messages) {
-    if (!Array.isArray(m.content)) continue;
-    let changed = false;
-    const next = [];
-    for (const part of m.content) {
-      if (part.type === "image_url") {
-        changed = true;
-        next.push({ type: "text", text: "[image]" });
-      } else {
-        next.push(part);
-      }
-    }
-    if (changed) {
-      m.content = next;
-    }
-  }
-}
-function truncateOldToolResults(messages, keepRecent) {
-  const toolIndices = [];
-  for (let i = 0; i < messages.length; i++) {
-    if (messages[i].role === "tool") toolIndices.push(i);
-  }
-  const cutoff = toolIndices.length - keepRecent;
-  for (let i = 0; i < cutoff; i++) {
-    const idx = toolIndices[i];
-    const m = messages[idx];
-    const text = typeof m.content === "string" ? m.content : "";
-    if (text.length > 500) {
-      m.content = text.slice(0, 500) + "\n\u2026 [truncated]";
-    }
-  }
-}
-async function maybeAutoCompact(messages, cfg, onInfo) {
-  const heapUsed = process.memoryUsage().heapUsed;
-  if (heapUsed < HEAP_AUTO_COMPACT_THRESHOLD || messages.length <= MSG_AUTO_COMPACT_THRESHOLD) {
-    return false;
-  }
-  onInfo("auto-compacting to reduce memory usage");
-  const result = await compactMessages({
-    accountId: cfg.accountId,
-    apiToken: cfg.apiToken,
-    model: cfg.model,
-    messages,
-    keepLastTurns: 2
-  });
-  if (result.replacedCount > 0) {
-    messages.length = 0;
-    messages.push(...result.newMessages);
-    onInfo(`compacted ${result.replacedCount} messages into a summary`);
-    return true;
-  }
-  return false;
-}
 function App({ initialCfg, initialUpdateResult }) {
   const { exit } = useApp();
   const [cfg, setCfg] = useState6(initialCfg);
@@ -3731,7 +3643,6 @@ function App({ initialCfg, initialUpdateResult }) {
   const executorRef = useRef3(new ToolExecutor(ALL_TOOLS));
   const activeAsstIdRef = useRef3(null);
   const activeControllerRef = useRef3(null);
-  const permResolveRef = useRef3(null);
   const sessionIdRef = useRef3(null);
   const modeRef = useRef3(mode);
   const effortRef = useRef3(effort);
@@ -3740,7 +3651,6 @@ function App({ initialCfg, initialUpdateResult }) {
   const updateCheckedRef = useRef3(false);
   const updateNudgedRef = useRef3(false);
   const compactSuggestedRef = useRef3(false);
-  const interruptRequestedRef = useRef3(false);
   useEffect4(() => {
     if (!cfg || updateCheckedRef.current) return;
     updateCheckedRef.current = true;
@@ -3874,15 +3784,8 @@ function App({ initialCfg, initialUpdateResult }) {
   }, [cfg]);
   useInput2((inputChar, key) => {
     if (key.ctrl && inputChar === "c") {
-      if (busy) {
-        if (interruptRequestedRef.current) {
-          process.exit(0);
-        }
-        interruptRequestedRef.current = true;
-        activeControllerRef.current?.abort();
-        permResolveRef.current?.("deny");
-        permResolveRef.current = null;
-        setPerm(null);
+      if (busy && activeControllerRef.current) {
+        activeControllerRef.current.abort();
         setQueue([]);
         setEvents((e) => [...e, { kind: "info", key: mkKey(), text: "(interrupted)" }]);
       } else {
@@ -3974,7 +3877,6 @@ function App({ initialCfg, initialUpdateResult }) {
       setBusy(false);
       setTurnStartedAt(null);
       activeControllerRef.current = null;
-      interruptRequestedRef.current = false;
     }
   }, [cfg, busy, saveSessionSafe]);
   const openResumePicker = useCallback(async () => {
@@ -4105,21 +4007,10 @@ function App({ initialCfg, initialUpdateResult }) {
               resolve2("deny");
               return;
             }
-            permResolveRef.current = resolve2;
-            setPerm({ tool: req.tool, args: req.args, resolve: (d) => {
-              permResolveRef.current = null;
-              resolve2(d);
-            } });
+            setPerm({ tool: req.tool, args: req.args, resolve: resolve2 });
           })
         }
       });
-      stripImagesFromHistory(messagesRef.current);
-      truncateOldToolResults(messagesRef.current, 8);
-      await maybeAutoCompact(
-        messagesRef.current,
-        cfg,
-        (text) => setEvents((es) => [...es, { kind: "info", key: mkKey(), text }])
-      );
       if (existsSync(join6(cwd, "KIMI.md"))) {
         messagesRef.current[0] = {
           role: "system",
@@ -4136,26 +4027,17 @@ function App({ initialCfg, initialUpdateResult }) {
         ]);
       }
     } catch (e) {
-      if (e.name === "AbortError") {
-        setEvents((es) => [...es, { kind: "info", key: mkKey(), text: "(interrupted)" }]);
-        setEvents(
-          (evts) => evts.map((e2) => e2.kind === "tool" && e2.status === "running" ? { ...e2, status: "error", result: "(interrupted)" } : e2)
-        );
-      } else {
+      if (e.name !== "AbortError") {
         setEvents((es) => [
           ...es,
           { kind: "error", key: mkKey(), text: `init failed: ${e.message}` }
         ]);
       }
     } finally {
-      const asstId = activeAsstIdRef.current;
-      if (asstId !== null) updateAssistant(asstId, () => ({ streaming: false }));
       setBusy(false);
       setTurnStartedAt(null);
       activeAsstIdRef.current = null;
       activeControllerRef.current = null;
-      permResolveRef.current = null;
-      interruptRequestedRef.current = false;
     }
   }, [cfg, busy, updateAssistant, updateTool]);
   const handleResumePick = useCallback(
@@ -4567,28 +4449,14 @@ use: /thinking low | medium | high`
                 resolve2("deny");
                 return;
               }
-              permResolveRef.current = resolve2;
-              setPerm({ tool: req.tool, args: req.args, resolve: (d) => {
-                permResolveRef.current = null;
-                resolve2(d);
-              } });
+              setPerm({ tool: req.tool, args: req.args, resolve: resolve2 });
             })
           }
         });
-        stripImagesFromHistory(messagesRef.current);
-        truncateOldToolResults(messagesRef.current, 8);
-        await maybeAutoCompact(
-          messagesRef.current,
-          cfg,
-          (text2) => setEvents((es) => [...es, { kind: "info", key: mkKey(), text: text2 }])
-        );
-        void saveSessionSafe();
+        await saveSessionSafe();
       } catch (e) {
         if (e.name === "AbortError") {
-          setEvents((es) => [...es, { kind: "info", key: mkKey(), text: "(interrupted)" }]);
-          setEvents(
-            (evts) => evts.map((e2) => e2.kind === "tool" && e2.status === "running" ? { ...e2, status: "error", result: "(interrupted)" } : e2)
-          );
+          setEvents((es) => [...es, { kind: "info", key: mkKey(), text: "(aborted)" }]);
         } else {
           const isInvalidJson400 = e instanceof KimiApiError && e.httpStatus === 400 && e.message.includes("invalid escaped character");
           if (isInvalidJson400) {
@@ -4609,14 +4477,10 @@ use: /thinking low | medium | high`
           }
         }
       } finally {
-        const asstId = activeAsstIdRef.current;
-        if (asstId !== null) updateAssistant(asstId, () => ({ streaming: false }));
         setBusy(false);
         setTurnStartedAt(null);
         activeAsstIdRef.current = null;
         activeControllerRef.current = null;
-        permResolveRef.current = null;
-        interruptRequestedRef.current = false;
       }
     },
     [cfg, handleSlash, updateAssistant, updateTool, saveSessionSafe]
@@ -4779,7 +4643,7 @@ async function renderApp(cfg, updateResult) {
   const instance = render(/* @__PURE__ */ jsx13(App, { initialCfg: cfg, initialUpdateResult: updateResult }));
   await instance.waitUntilExit();
 }
-var CONTEXT_LIMIT, AUTO_COMPACT_SUGGEST_PCT, MAX_EVENTS, nextAssistantId, nextKey, mkKey, MAX_IMAGES_PER_MESSAGE, EFFORT_DESCRIPTIONS, HEAP_AUTO_COMPACT_THRESHOLD, MSG_AUTO_COMPACT_THRESHOLD;
+var CONTEXT_LIMIT, AUTO_COMPACT_SUGGEST_PCT, MAX_EVENTS, nextAssistantId, nextKey, mkKey, MAX_IMAGES_PER_MESSAGE, EFFORT_DESCRIPTIONS;
 var init_app = __esm({
   "src/app.tsx"() {
     "use strict";
@@ -4816,8 +4680,6 @@ var init_app = __esm({
       medium: "medium \u2014 balanced (default). Solid quality on most edits, fast on trivial prompts.",
       high: "high \u2014 deepest reasoning; slowest. Best for complex debugging, architecture, multi-file refactors."
     };
-    HEAP_AUTO_COMPACT_THRESHOLD = 2e9;
-    MSG_AUTO_COMPACT_THRESHOLD = 50;
   }
 });
 
@@ -4895,10 +4757,7 @@ async function runPrintMode(opts2) {
     { role: "user", content: opts2.prompt }
   ];
   const controller = new AbortController();
-  process.once("SIGINT", () => {
-    controller.abort();
-    setTimeout(() => process.exit(1), 500);
-  });
+  process.on("SIGINT", () => controller.abort());
   let printedReasoningHeader = false;
   let printedAnswerHeader = false;
   await runAgentTurn({