kimiflare 0.9.0 → 0.9.2

package/dist/index.js

@@ -172,7 +172,7 @@ function isRetryable(err, attempt) {
 async function* runKimi(opts2) {
   const url = `https://api.cloudflare.com/client/v4/accounts/${opts2.accountId}/ai/run/${opts2.model}`;
   const body = {
-    messages: opts2.messages,
+    messages: sanitizeMessagesForApi(opts2.messages),
     ...opts2.tools && opts2.tools.length ? { tools: opts2.tools, tool_choice: "auto", parallel_tool_calls: true } : {},
     stream: true,
     temperature: opts2.temperature ?? 0.2,
@@ -294,6 +294,30 @@ async function* parseStream(body, signal) {
   }
   yield { type: "done", finishReason, usage: lastUsage };
 }
+function sanitizeMessagesForApi(messages) {
+  return messages.map((m) => {
+    if (!m.tool_calls || m.tool_calls.length === 0) return m;
+    return {
+      ...m,
+      tool_calls: m.tool_calls.map((tc) => ({
+        ...tc,
+        function: {
+          name: tc.function.name,
+          arguments: validateJsonArguments(tc.function.arguments)
+        }
+      }))
+    };
+  });
+}
+function validateJsonArguments(raw) {
+  if (!raw || !raw.trim()) return "{}";
+  try {
+    JSON.parse(raw);
+    return raw;
+  } catch {
+    return "{}";
+  }
+}
 function extractCloudflareError(parsed) {
   if (!parsed || typeof parsed !== "object") return null;
   const cf = parsed;
@@ -387,10 +411,11 @@ async function runAgentTurn(opts2) {
           opts2.callbacks.onToolCallArgs?.(ev.index, ev.argsDelta);
           break;
         case "tool_call_complete": {
+          const safeArgs = validateToolArguments(ev.arguments);
           const call = {
             id: ev.id,
             type: "function",
-            function: { name: ev.name, arguments: ev.arguments }
+            function: { name: ev.name, arguments: safeArgs }
           };
           toolCalls.push(call);
           opts2.callbacks.onToolCallFinalized?.(call);
@@ -438,6 +463,15 @@ async function runAgentTurn(opts2) {
   }
   throw new Error(`kimiflare: tool iteration limit reached (${opts2.maxToolIterations ?? 50})`);
 }
+function validateToolArguments(raw) {
+  if (!raw || !raw.trim()) return "{}";
+  try {
+    JSON.parse(raw);
+    return raw;
+  } catch {
+    return "{}";
+  }
+}
 var init_loop = __esm({
   "src/agent/loop.ts"() {
     "use strict";
@@ -1474,10 +1508,11 @@ var init_tool_view = __esm({
 });
 
 // src/ui/markdown.tsx
+import { useMemo } from "react";
 import { Box as Box3, Text as Text3 } from "ink";
 import { jsx as jsx3, jsxs as jsxs3 } from "react/jsx-runtime";
 function MD({ text, theme }) {
-  const blocks = parseBlocks(text);
+  const blocks = useMemo(() => parseBlocks(text), [text]);
   return /* @__PURE__ */ jsx3(Box3, { flexDirection: "column", children: blocks.map((b, i) => /* @__PURE__ */ jsx3(Block, { block: b, theme }, i)) });
 }
 function parseBlocks(src) {
@@ -1633,19 +1668,10 @@ var init_markdown = __esm({
 });
 
 // src/ui/chat.tsx
+import React2 from "react";
 import { Box as Box4, Text as Text4 } from "ink";
 import Spinner2 from "ink-spinner";
 import { jsx as jsx4, jsxs as jsxs4 } from "react/jsx-runtime";
-function ChatView({ events, showReasoning, theme, verbose }) {
-  return /* @__PURE__ */ jsx4(Box4, { flexDirection: "column", children: events.map((e, i) => {
-    const prev = events[i - 1];
-    const showSeparator = e.kind === "user" && prev && (prev.kind === "assistant" || prev.kind === "tool");
-    return /* @__PURE__ */ jsxs4(Box4, { flexDirection: "column", children: [
-      showSeparator && /* @__PURE__ */ jsx4(Box4, { marginY: 1, children: /* @__PURE__ */ jsx4(Text4, { color: theme.info.color, dimColor: theme.info.dim, children: "\u2500".repeat(40) }) }),
-      /* @__PURE__ */ jsx4(EventView, { evt: e, showReasoning, theme, verbose })
-    ] }, e.key);
-  }) });
-}
 function EventView({
   evt,
   showReasoning,
@@ -1686,11 +1712,22 @@ function EventView({
     evt.text
   ] });
 }
+var ChatView;
 var init_chat = __esm({
   "src/ui/chat.tsx"() {
     "use strict";
     init_tool_view();
     init_markdown();
+    ChatView = React2.memo(function ChatView2({ events, showReasoning, theme, verbose }) {
+      return /* @__PURE__ */ jsx4(Box4, { flexDirection: "column", children: events.map((e, i) => {
+        const prev = events[i - 1];
+        const showSeparator = e.kind === "user" && prev && (prev.kind === "assistant" || prev.kind === "tool");
+        return /* @__PURE__ */ jsxs4(Box4, { flexDirection: "column", children: [
+          showSeparator && /* @__PURE__ */ jsx4(Box4, { marginY: 1, children: /* @__PURE__ */ jsx4(Text4, { color: theme.info.color, dimColor: theme.info.dim, children: "\u2500".repeat(40) }) }),
+          /* @__PURE__ */ jsx4(EventView, { evt: e, showReasoning, theme, verbose })
+        ] }, e.key);
+      }) });
+    });
   }
 });
 
@@ -1943,19 +1980,25 @@ var init_theme_picker = __esm({
 });
 
 // src/ui/task-list.tsx
-import { useEffect as useEffect2, useState as useState3 } from "react";
+import { useEffect as useEffect2, useRef, useState as useState3 } from "react";
 import { Box as Box9, Text as Text9 } from "ink";
 import Spinner4 from "ink-spinner";
 import { jsx as jsx9, jsxs as jsxs9 } from "react/jsx-runtime";
 function TaskList({ tasks, theme, startedAt, tokensDelta }) {
   const [now, setNow] = useState3(Date.now());
+  const tasksRef = useRef(tasks);
+  tasksRef.current = tasks;
   useEffect2(() => {
     if (startedAt === null) return;
-    const allDone2 = tasks.length > 0 && tasks.every((t) => t.status === "completed");
-    if (allDone2) return;
-    const id = setInterval(() => setNow(Date.now()), 1e3);
+    const id = setInterval(() => {
+      setNow(Date.now());
+      const current = tasksRef.current;
+      if (current.length > 0 && current.every((t) => t.status === "completed")) {
+        clearInterval(id);
+      }
+    }, 1e3);
     return () => clearInterval(id);
-  }, [startedAt, tasks]);
+  }, [startedAt]);
   if (tasks.length === 0) return null;
   const active = tasks.find((t) => t.status === "in_progress");
   const done = tasks.filter((t) => t.status === "completed").length;
@@ -2547,7 +2590,7 @@ var init_source = __esm({
 });
 
 // src/ui/text-input.tsx
-import { useState as useState4, useEffect as useEffect3, useRef } from "react";
+import { useState as useState4, useEffect as useEffect3, useRef as useRef2 } from "react";
 import { Text as Text10, useInput } from "ink";
 import { jsx as jsx10 } from "react/jsx-runtime";
 function shouldTreatAsPaste(input) {
@@ -2583,7 +2626,7 @@ function CustomTextInput({
   enablePaste = false
 }) {
   const [cursorOffset, setCursorOffset] = useState4(value.length);
-  const pastesRef = useRef(/* @__PURE__ */ new Map());
+  const pastesRef = useRef2(/* @__PURE__ */ new Map());
   useEffect3(() => {
     if (!focus) return;
     setCursorOffset((prev) => prev > value.length ? value.length : prev);
@@ -3269,16 +3312,29 @@ var app_exports = {};
 __export(app_exports, {
   renderApp: () => renderApp
 });
-import { useState as useState6, useRef as useRef2, useEffect as useEffect4, useCallback } from "react";
+import { useState as useState6, useRef as useRef3, useEffect as useEffect4, useCallback } from "react";
 import { Box as Box12, Text as Text13, useApp, useInput as useInput2, render } from "ink";
 import { existsSync } from "fs";
 import { join as join5 } from "path";
 import { unlink } from "fs/promises";
 import { jsx as jsx13, jsxs as jsxs12 } from "react/jsx-runtime";
+function capEvents(prev) {
+  if (prev.length <= MAX_EVENTS) return prev;
+  return prev.slice(prev.length - MAX_EVENTS);
+}
 function App({ initialCfg, initialUpdateResult }) {
   const { exit } = useApp();
   const [cfg, setCfg] = useState6(initialCfg);
-  const [events, setEvents] = useState6([]);
+  const [events, setRawEvents] = useState6([]);
+  const setEvents = useCallback(
+    (updater) => {
+      setRawEvents((prev) => {
+        const next = typeof updater === "function" ? updater(prev) : updater;
+        return capEvents(next);
+      });
+    },
+    []
+  );
   const [input, setInput] = useState6("");
   const [busy, setBusy] = useState6(false);
   const [usage, setUsage] = useState6(null);
@@ -3303,7 +3359,7 @@ function App({ initialCfg, initialUpdateResult }) {
   const [verbose, setVerbose] = useState6(false);
   const [hasUpdate, setHasUpdate] = useState6(initialUpdateResult?.hasUpdate ?? false);
   const [latestVersion, setLatestVersion] = useState6(initialUpdateResult?.latestVersion ?? null);
-  const messagesRef = useRef2([
+  const messagesRef = useRef3([
     {
       role: "system",
       content: buildSystemPrompt({
@@ -3314,17 +3370,17 @@ function App({ initialCfg, initialUpdateResult }) {
       })
     }
   ]);
-  const executorRef = useRef2(new ToolExecutor(ALL_TOOLS));
-  const activeAsstIdRef = useRef2(null);
-  const activeControllerRef = useRef2(null);
-  const sessionIdRef = useRef2(null);
-  const modeRef = useRef2(mode);
-  const effortRef = useRef2(effort);
-  const tasksRef = useRef2([]);
-  const usageRef = useRef2(null);
-  const updateCheckedRef = useRef2(false);
-  const updateNudgedRef = useRef2(false);
-  const compactSuggestedRef = useRef2(false);
+  const executorRef = useRef3(new ToolExecutor(ALL_TOOLS));
+  const activeAsstIdRef = useRef3(null);
+  const activeControllerRef = useRef3(null);
+  const sessionIdRef = useRef3(null);
+  const modeRef = useRef3(mode);
+  const effortRef = useRef3(effort);
+  const tasksRef = useRef3([]);
+  const usageRef = useRef3(null);
+  const updateCheckedRef = useRef3(false);
+  const updateNudgedRef = useRef3(false);
+  const compactSuggestedRef = useRef3(false);
   useEffect4(() => {
     if (!cfg || updateCheckedRef.current) return;
     updateCheckedRef.current = true;
@@ -4070,10 +4126,23 @@ use: /thinking low | medium | high`
         if (e.name === "AbortError") {
           setEvents((es) => [...es, { kind: "info", key: mkKey(), text: "(aborted)" }]);
         } else {
-          setEvents((es) => [
-            ...es,
-            { kind: "error", key: mkKey(), text: e.message ?? String(e) }
-          ]);
+          const isInvalidJson400 = e instanceof KimiApiError && e.httpStatus === 400 && e.message.includes("invalid escaped character");
+          if (isInvalidJson400) {
+            messagesRef.current.pop();
+            setEvents((es) => [
+              ...es,
+              {
+                kind: "error",
+                key: mkKey(),
+                text: "API rejected request (invalid JSON in conversation history). Retrying may work; run /clear to reset if it persists."
+              }
+            ]);
+          } else {
+            setEvents((es) => [
+              ...es,
+              { kind: "error", key: mkKey(), text: e.message ?? String(e) }
+            ]);
+          }
         }
       } finally {
         setBusy(false);
@@ -4242,7 +4311,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, nextAssistantId, nextKey, mkKey, EFFORT_DESCRIPTIONS;
+var CONTEXT_LIMIT, AUTO_COMPACT_SUGGEST_PCT, MAX_EVENTS, nextAssistantId, nextKey, mkKey, EFFORT_DESCRIPTIONS;
 var init_app = __esm({
   "src/app.tsx"() {
     "use strict";
@@ -4251,6 +4320,7 @@ var init_app = __esm({
     init_compact();
     init_executor();
     init_messages();
+    init_errors();
     init_chat();
     init_status();
     init_permission();
@@ -4267,6 +4337,7 @@ var init_app = __esm({
     init_sessions();
     CONTEXT_LIMIT = 262e3;
     AUTO_COMPACT_SUGGEST_PCT = 0.8;
+    MAX_EVENTS = 500;
     nextAssistantId = 1;
     nextKey = 1;
     mkKey = () => `evt_${nextKey++}`;