kimiflare 0.32.0 → 0.33.0

package/dist/index.js

@@ -945,6 +945,87 @@ var init_cost_debug = __esm({
   }
 });
 
+// src/memory/extractors.ts
+function safeJsonParse(text) {
+  try {
+    return JSON.parse(text);
+  } catch {
+    return null;
+  }
+}
+var EXTRACTORS;
+var init_extractors = __esm({
+  "src/memory/extractors.ts"() {
+    "use strict";
+    EXTRACTORS = [
+      {
+        id: "package_json",
+        match: (tool, file) => tool === "read" && /package\.json$/.test(file || ""),
+        extract: (content, file) => {
+          const pkg = safeJsonParse(content);
+          if (!pkg) return null;
+          const deps = Object.keys(pkg.dependencies || {}).slice(0, 10);
+          const devDeps = Object.keys(pkg.devDependencies || {}).slice(0, 5);
+          const scripts = Object.keys(pkg.scripts || {}).slice(0, 5);
+          return {
+            content: `Project dependencies: ${deps.join(", ") || "none"}. Dev dependencies: ${devDeps.join(", ") || "none"}. Scripts: ${scripts.join(", ") || "none"}. Type: ${pkg.type || "commonjs"}.`,
+            category: "fact",
+            importance: 4,
+            topicKey: "project_dependencies",
+            relatedFiles: file ? [file] : void 0
+          };
+        }
+      },
+      {
+        id: "tsconfig",
+        match: (tool, file) => tool === "read" && /tsconfig.*\.json$/.test(file || ""),
+        extract: (content, file) => {
+          const ts = safeJsonParse(content);
+          if (!ts) return null;
+          const opts2 = ts.compilerOptions || {};
+          return {
+            content: `TypeScript config: target=${opts2.target || "default"}, module=${opts2.module || "default"}, strict=${opts2.strict || false}, jsx=${opts2.jsx || "none"}.`,
+            category: "fact",
+            importance: 4,
+            topicKey: "project_tsconfig",
+            relatedFiles: file ? [file] : void 0
+          };
+        }
+      },
+      {
+        id: "entry_point",
+        match: (tool, file) => tool === "read" && /src\/(index|main)\.(ts|tsx|js|jsx)$/.test(file || ""),
+        extract: (content, file) => {
+          const exports = content.match(/export\s+(?:default\s+)?(?:function|class|const|interface|type)\s+(\w+)/g);
+          const exportNames = exports ? exports.map((e) => e.split(/\s+/).pop()).filter((n) => !!n).slice(0, 5) : [];
+          return {
+            content: `Entry point ${file} exports: ${exportNames.join(", ") || "default export or side effects"}.`,
+            category: "fact",
+            importance: 3,
+            topicKey: "project_entry_point",
+            relatedFiles: file ? [file] : void 0
+          };
+        }
+      },
+      {
+        id: "edit_event",
+        match: (tool, file) => (tool === "edit" || tool === "write") && !!file,
+        extract: (_content, file) => {
+          if (!file) return null;
+          const safeKey = file.replace(/[^a-zA-Z0-9]/g, "_");
+          return {
+            content: `File modified: ${file}.`,
+            category: "event",
+            importance: 2,
+            topicKey: `event_edit_${safeKey}`,
+            relatedFiles: [file]
+          };
+        }
+      }
+    ];
+  }
+});
+
 // src/agent/strip-reasoning.ts
 function stripHistoricalReasoning(messages, opts2 = {}) {
   const keepLast = opts2.keepLast ?? DEFAULT_KEEP_LAST;
@@ -1328,7 +1409,28 @@ Use console.log() to return results. Only console.log output will be sent back t
   const webFetchHistory = [];
   const MAX_WEB_FETCH_PER_TURN = 5;
   const WEB_FETCH_DOMAIN_THRESHOLD = 2;
-  for (let iter = 0; iter < max; iter++) {
+  let cumulativePromptTokens = 0;
+  let iter = 0;
+  let budgetExhausted = false;
+  while (true) {
+    if (budgetExhausted) {
+      opts2.messages.push({
+        role: "system",
+        content: "You have reached the cumulative input token budget for this session. Please synthesize your findings and provide a final summary of what was accomplished."
+      });
+    }
+    if (iter >= max) {
+      if (opts2.continueOnLimit) {
+        opts2.messages.push({
+          role: "system",
+          content: "You have reached the tool-call limit for this session. The counter has been reset so you can continue working. Please proceed with your task."
+        });
+        iter = 0;
+      } else {
+        throw new Error(`kimiflare: tool iteration limit reached (${max})`);
+      }
+    }
+    iter++;
     turn++;
     const previousMessages = opts2.messages.slice();
     const toolCalls = [];
@@ -1426,7 +1528,13 @@ Use console.log() to return results. Only console.log output will be sent back t
       }
     }
     if (opts2.signal.aborted) throw new DOMException("aborted", "AbortError");
-    if (lastUsage) opts2.callbacks.onUsageFinal?.(lastUsage, gatewayMeta);
+    if (lastUsage) {
+      opts2.callbacks.onUsageFinal?.(lastUsage, gatewayMeta);
+      cumulativePromptTokens += lastUsage.prompt_tokens;
+      if (!budgetExhausted && opts2.maxInputTokens !== void 0 && opts2.maxInputTokens > 0 && cumulativePromptTokens >= opts2.maxInputTokens && toolCalls.length > 0) {
+        budgetExhausted = true;
+      }
+    }
     const assistantMsg = {
       role: "assistant",
       content: content ? sanitizeString(content) : null,
@@ -1455,6 +1563,9 @@ Use console.log() to return results. Only console.log output will be sent back t
           shadowStrip: shadowStripMetrics
         });
       }
+      if (budgetExhausted) {
+        throw new BudgetExhaustedError();
+      }
       return;
     }
     for (const tc of toolCalls) {
@@ -1589,6 +1700,30 @@ ${sandboxResult.output}` : sandboxResult.output;
           name: result.name
         });
         opts2.callbacks.onToolResult?.(result);
+        if (opts2.memoryManager) {
+          let filePath;
+          try {
+            const args = JSON.parse(tc.function.arguments || "{}");
+            filePath = args.path;
+          } catch {
+          }
+          for (const extractor of EXTRACTORS) {
+            if (extractor.match(tc.function.name, filePath)) {
+              const memory = extractor.extract(result.content, filePath);
+              if (memory) {
+                void opts2.memoryManager.remember(
+                  memory.content,
+                  memory.category,
+                  memory.importance,
+                  opts2.cwd,
+                  opts2.sessionId ?? "unknown",
+                  opts2.signal
+                ).catch(() => {
+                });
+              }
+            }
+          }
+        }
         recentToolCalls.push(loopSignature);
         if (recentToolCalls.length > LOOP_WINDOW) recentToolCalls.shift();
       }
@@ -1604,8 +1739,10 @@ ${sandboxResult.output}` : sandboxResult.output;
         shadowStrip: shadowStripMetrics
       });
     }
+    if (budgetExhausted) {
+      throw new BudgetExhaustedError();
+    }
   }
-  throw new Error(`kimiflare: tool iteration limit reached (${opts2.maxToolIterations ?? 50})`);
 }
 function validateToolArguments(raw) {
   if (!raw || !raw.trim()) return "{}";
@@ -1616,7 +1753,7 @@ function validateToolArguments(raw) {
     return "{}";
   }
 }
-var codeModeApiCache;
+var BudgetExhaustedError, codeModeApiCache;
 var init_loop = __esm({
   "src/agent/loop.ts"() {
     "use strict";
@@ -1624,8 +1761,15 @@ var init_loop = __esm({
     init_registry();
     init_messages();
     init_cost_debug();
+    init_extractors();
     init_strip_reasoning();
     init_code_mode();
+    BudgetExhaustedError = class extends Error {
+      constructor(message2 = "Cumulative input token budget exhausted") {
+        super(message2);
+        this.name = "BudgetExhaustedError";
+      }
+    };
     codeModeApiCache = /* @__PURE__ */ new Map();
   }
 });
@@ -7413,7 +7557,7 @@ var init_help_menu = __esm({
         commands: [
           { command: "/init", description: "scan this repo and write a KIMI.md" },
           { command: "/logout", description: "clear credentials" },
-          { command: "filePicker", description: "enable with KIMIFLARE_FILE_PICKER=1 or filePicker: true in config", selectable: false }
+          { command: "filePicker", description: "enabled by default; disable with KIMIFLARE_FILE_PICKER=0 or filePicker: false in config", selectable: false }
         ]
       }
     ];
@@ -11026,7 +11170,7 @@ function App({
   const [draftInput, setDraftInput] = useState9("");
   const [mode, setMode] = useState9("edit");
   const [codeMode, setCodeMode] = useState9(initialCfg?.codeMode ?? false);
-  const filePickerEnabled = initialCfg?.filePicker ?? false;
+  const filePickerEnabled = initialCfg?.filePicker ?? true;
   const [effort, setEffort] = useState9(
     initialCfg?.reasoningEffort ?? DEFAULT_REASONING_EFFORT
   );
@@ -11833,23 +11977,13 @@ function App({
       return;
     }
     const cwd = process.cwd();
-    for (const name of ["KIMI.md", "KIMIFLARE.md", "AGENT.md"]) {
-      if (existsSync2(join17(cwd, name))) {
-        setEvents((e) => [
-          ...e,
-          {
-            kind: "info",
-            key: mkKey(),
-            text: `${name} already exists at ${join17(cwd, name)} \u2014 delete it first if you want to regenerate`
-          }
-        ]);
-        return;
-      }
-    }
-    const prompt = [
-      "Generate a KIMI.md at the repository root so future agents have project context.",
+    const existingName = ["KIMI.md", "KIMIFLARE.md", "AGENT.md"].find((n) => existsSync2(join17(cwd, n)));
+    const isRefresh = existingName !== void 0;
+    const promptParts = [
+      isRefresh ? `Regenerate ${existingName} at the repository root to refresh project context. If the file already exists, read it first and preserve anything still accurate, updating only what has changed.` : "Generate a KIMI.md at the repository root so future agents have project context.",
       "",
       "First, use the `glob`, `read`, and `grep` tools to understand the project: read `package.json`, the top-level `README.md` if present, the tsconfig / build config, and skim the top-level source directory structure.",
+      isRefresh ? `Also read the existing ${existingName} so you know what to keep vs. update.` : null,
       "",
       "Then call the `write` tool to create `KIMI.md` at the repo root with these sections, terse (aim \u2264 100 lines total):",
       "",
@@ -11860,8 +11994,9 @@ function App({
       "- **Do / Don't** \u2014 quirks or rules future agents should know.",
       "",
       "Do not call `tasks_set` for this. Just read what you need, then write the file."
-    ].join("\n");
-    setEvents((e) => [...e, { kind: "user", key: mkKey(), text: "/init" }]);
+    ];
+    const prompt = promptParts.filter((p) => p !== null).join("\n");
+    setEvents((e) => [...e, { kind: "user", key: mkKey(), text: isRefresh ? `/init (refreshing ${existingName})` : "/init" }]);
     messagesRef.current.push({ role: "user", content: sanitizeString(prompt) });
     setBusy(true);
     setTurnStartedAt(Date.now());
@@ -13433,7 +13568,7 @@ init_update_check();
 init_version();
 import { Command } from "commander";
 var program = new Command();
-program.name("kimiflare").description("Terminal coding agent powered by Kimi-K2.6 on Cloudflare Workers AI.").version(getAppVersion()).option("-p, --print <prompt>", "one-shot mode: send prompt, stream reply to stdout, exit").option("-m, --model <id>", "model id (defaults to @cf/moonshotai/kimi-k2.6)").option("--dangerously-allow-all", "auto-approve every permission prompt (print mode only)").option("--reasoning", "include reasoning in stdout (print mode only)");
+program.name("kimiflare").description("Terminal coding agent powered by Kimi-K2.6 on Cloudflare Workers AI.").version(getAppVersion()).option("-p, --print <prompt>", "one-shot mode: send prompt, stream reply to stdout, exit").option("-m, --model <id>", "model id (defaults to @cf/moonshotai/kimi-k2.6)").option("--dangerously-allow-all", "auto-approve every permission prompt (print mode only)").option("--reasoning", "include reasoning in stdout (print mode only)").option("--continue-on-limit", "reset tool-call counter and continue when the 50-call limit is hit (print mode only)").option("--max-input-tokens <n>", "cumulative prompt token budget; exits 42 when exhausted (print mode only)", (v) => parseInt(v, 10));
 program.command("cost").description("Show cost attribution by task type (requires costAttribution enabled)").option("-w, --week", "last 7 days (default)").option("-m, --month", "last 30 days").option("-d, --day", "today only").option("-s, --session <id>", "single session detail").option("-c, --category <name>", "filter by category").option("--json", "machine-readable output").option("--reclassify", "re-run classification on all sessions").option("--local-only", "skip Cloudflare reconciliation").action(async (cmdOpts) => {
   const cfg = await loadConfig();
   const enabled = cfg?.costAttribution ?? false;
@@ -13482,6 +13617,8 @@ async function main() {
       allowAll: !!opts.dangerouslyAllowAll,
       showReasoning: !!opts.reasoning,
       codeMode: cfg.codeMode,
+      continueOnLimit: !!opts.continueOnLimit,
+      maxInputTokens: opts.maxInputTokens,
       updateResult
     });
     return;
@@ -13529,52 +13666,63 @@ async function runPrintMode(opts2) {
   process.on("SIGINT", () => controller.abort());
   let printedReasoningHeader = false;
   let printedAnswerHeader = false;
-  await runAgentTurn({
-    accountId: opts2.accountId,
-    apiToken: opts2.apiToken,
-    model: opts2.model,
-    gateway: gatewayFromPrintOpts(opts2),
-    messages,
-    tools: ALL_TOOLS,
-    executor,
-    cwd,
-    signal: controller.signal,
-    codeMode: opts2.codeMode,
-    coauthor: opts2.coauthor !== false ? { name: opts2.coauthorName || "kimiflare", email: opts2.coauthorEmail || "kimiflare@proton.me" } : void 0,
-    callbacks: {
-      onReasoningDelta: opts2.showReasoning ? (delta) => {
-        if (!printedReasoningHeader) {
-          process.stderr.write("\x1B[2m--- reasoning ---\n");
-          printedReasoningHeader = true;
-        }
-        process.stderr.write(delta);
-      } : void 0,
-      onTextDelta: (delta) => {
-        if (opts2.showReasoning && printedReasoningHeader && !printedAnswerHeader) {
-          process.stderr.write("\n--- answer ---\x1B[0m\n");
-          printedAnswerHeader = true;
-        }
-        process.stdout.write(delta);
-      },
-      onToolCallFinalized: (call) => {
-        process.stderr.write(`\x1B[2m[tool ${call.function.name}(${call.function.arguments})]\x1B[0m
+  try {
+    await runAgentTurn({
+      accountId: opts2.accountId,
+      apiToken: opts2.apiToken,
+      model: opts2.model,
+      gateway: gatewayFromPrintOpts(opts2),
+      messages,
+      tools: ALL_TOOLS,
+      executor,
+      cwd,
+      signal: controller.signal,
+      codeMode: opts2.codeMode,
+      continueOnLimit: opts2.continueOnLimit,
+      maxInputTokens: opts2.maxInputTokens,
+      coauthor: opts2.coauthor !== false ? { name: opts2.coauthorName || "kimiflare", email: opts2.coauthorEmail || "kimiflare@proton.me" } : void 0,
+      callbacks: {
+        onReasoningDelta: opts2.showReasoning ? (delta) => {
+          if (!printedReasoningHeader) {
+            process.stderr.write("\x1B[2m--- reasoning ---\n");
+            printedReasoningHeader = true;
+          }
+          process.stderr.write(delta);
+        } : void 0,
+        onTextDelta: (delta) => {
+          if (opts2.showReasoning && printedReasoningHeader && !printedAnswerHeader) {
+            process.stderr.write("\n--- answer ---\x1B[0m\n");
+            printedAnswerHeader = true;
+          }
+          process.stdout.write(delta);
+        },
+        onToolCallFinalized: (call) => {
+          process.stderr.write(`\x1B[2m[tool ${call.function.name}(${call.function.arguments})]\x1B[0m
 `);
-      },
-      onToolResult: (result) => {
-        const snippet = result.content.length > 400 ? result.content.slice(0, 400) + "..." : result.content;
-        process.stderr.write(`\x1B[2m[result: ${snippet.replace(/\n/g, " \u23CE ")}]\x1B[0m
+        },
+        onToolResult: (result) => {
+          const snippet = result.content.length > 400 ? result.content.slice(0, 400) + "..." : result.content;
+          process.stderr.write(`\x1B[2m[result: ${snippet.replace(/\n/g, " \u23CE ")}]\x1B[0m
 `);
-      },
-      askPermission: async ({ tool, args }) => {
-        if (opts2.allowAll) return "allow";
-        process.stderr.write(
-          `\x1B[31m[permission denied: ${tool.name}(${JSON.stringify(args)}) \u2014 pass --dangerously-allow-all to approve in print mode]\x1B[0m
+        },
+        askPermission: async ({ tool, args }) => {
+          if (opts2.allowAll) return "allow";
+          process.stderr.write(
+            `\x1B[31m[permission denied: ${tool.name}(${JSON.stringify(args)}) \u2014 pass --dangerously-allow-all to approve in print mode]\x1B[0m
 `
-        );
-        return "deny";
+          );
+          return "deny";
+        }
       }
+    });
+  } catch (err) {
+    if (err instanceof BudgetExhaustedError) {
+      process.stderr.write("\n\x1B[33m[Budget exhausted \u2014 exiting with code 42]\x1B[0m\n");
+      process.exitCode = 42;
+      return;
     }
-  });
+    throw err;
+  }
   process.stdout.write("\n");
 }
 //# sourceMappingURL=index.js.map