kimiflare 0.40.0 → 0.41.0

package/dist/index.js

@@ -1861,6 +1861,9 @@ var init_code_mode = __esm({
 });
 
 // src/agent/loop.ts
+function isHighSignalMemory(memory) {
+  return memory.topicKey === "project_dependencies" || memory.topicKey === "project_tsconfig" || memory.topicKey === "project_entry_point" || memory.category === "instruction" || memory.category === "preference" || memory.category === "event" && memory.importance >= 3;
+}
 async function runAgentTurn(opts2) {
   const turnStart = performance.now();
   const max = opts2.maxToolIterations ?? 50;
@@ -2260,6 +2263,15 @@ ${sandboxResult.output}` : `${warningPrefix}${sandboxResult.output}`;
                       void 0,
                       memory.topicKey
                     );
+                    if (isHighSignalMemory(memory)) {
+                      const sid = opts2.sessionId ?? "default";
+                      const current = (driftAccumulator.get(sid) ?? 0) + 1;
+                      driftAccumulator.set(sid, current);
+                      if (current >= DRIFT_THRESHOLD) {
+                        opts2.callbacks.onKimiMdStale?.();
+                        driftAccumulator.set(sid, 0);
+                      }
+                    }
                   }
                 } catch {
                 }
@@ -2271,6 +2283,12 @@ ${sandboxResult.output}` : `${warningPrefix}${sandboxResult.output}`;
         if (recentToolCalls.length > LOOP_WINDOW) recentToolCalls.shift();
       }
     }
+    if (opts2.sessionId) {
+      const current = driftAccumulator.get(opts2.sessionId) ?? 0;
+      if (current > 0) {
+        driftAccumulator.set(opts2.sessionId, Math.max(0, current - 1));
+      }
+    }
     if (opts2.onIterationEnd) {
       opts2.messages = await opts2.onIterationEnd(opts2.messages, opts2.signal);
       if (opts2.signal.aborted) throw new DOMException("aborted", "AbortError");
@@ -2303,7 +2321,7 @@ function validateToolArguments(raw) {
     return "{}";
   }
 }
-var BudgetExhaustedError, codeModeApiCache;
+var BudgetExhaustedError, codeModeApiCache, driftAccumulator, DRIFT_THRESHOLD;
 var init_loop = __esm({
   "src/agent/loop.ts"() {
     "use strict";
@@ -2321,6 +2339,8 @@ var init_loop = __esm({
       }
     };
     codeModeApiCache = /* @__PURE__ */ new Map();
+    driftAccumulator = /* @__PURE__ */ new Map();
+    DRIFT_THRESHOLD = 5;
   }
 });
 
@@ -7253,7 +7273,7 @@ import { useEffect, useState } from "react";
 import { Box as Box5, Text as Text5 } from "ink";
 import Spinner3 from "ink-spinner";
 import { jsx as jsx6, jsxs as jsxs5 } from "react/jsx-runtime";
-function StatusBar({ model, usage, sessionUsage, thinking, turnStartedAt, mode, effort, contextLimit, hasUpdate, latestVersion, gatewayMeta, codeMode, cloudMode, cloudBudget }) {
+function StatusBar({ model, usage, sessionUsage, thinking, turnStartedAt, mode, effort, contextLimit, hasUpdate, latestVersion, gatewayMeta, codeMode, cloudMode, cloudBudget, kimiMdStale }) {
   const theme = useTheme();
   const [now2, setNow] = useState(Date.now());
   const modeColor = mode === "plan" ? theme.modeBadge.plan : mode === "auto" ? theme.modeBadge.auto : theme.modeBadge.edit;
@@ -7296,6 +7316,10 @@ function StatusBar({ model, usage, sessionUsage, thinking, turnStartedAt, mode,
         "update available",
         latestVersion ? ` \u2192 ${latestVersion}` : "",
         " \xB7 run /update"
+      ] }) : null,
+      kimiMdStale ? /* @__PURE__ */ jsxs5(Text5, { color: theme.warn, bold: true, children: [
+        "  \xB7  ",
+        "\u26A0 KIMI.md stale \xB7 run /init"
       ] }) : null
     ] })
   ] });
@@ -9712,11 +9736,21 @@ var init_usage_tracker = __esm({
   }
 });
 
+// src/memory/schema.ts
+var DEFAULT_EMBEDDING_DIM;
+var init_schema = __esm({
+  "src/memory/schema.ts"() {
+    "use strict";
+    DEFAULT_EMBEDDING_DIM = 768;
+  }
+});
+
 // src/memory/db.ts
 var db_exports = {};
 __export(db_exports, {
   clearMemoriesForRepo: () => clearMemoriesForRepo,
   closeMemoryDb: () => closeMemoryDb,
+  countHighSignalMemoriesSince: () => countHighSignalMemoriesSince,
   deleteExcessMemories: () => deleteExcessMemories,
   deleteMemoriesByIds: () => deleteMemoriesByIds,
   deleteOldMemories: () => deleteOldMemories,
@@ -10103,6 +10137,19 @@ function getMemoryById(db, id) {
   const row = db.prepare("SELECT * FROM memories WHERE id = ?").get(id);
   return row ? rowToMemory(row) : null;
 }
+function countHighSignalMemoriesSince(db, repoPath, since) {
+  const row = db.prepare(
+    `SELECT COUNT(*) as count FROM memories
+       WHERE repo_path = ? AND created_at > ?
+       AND forgotten = 0 AND superseded_by IS NULL
+       AND (
+         topic_key IN ('project_dependencies', 'project_tsconfig', 'project_entry_point')
+         OR category IN ('instruction', 'preference')
+         OR (category = 'event' AND importance >= 3)
+       )`
+  ).get(repoPath, since);
+  return row?.count ?? 0;
+}
 var dbInstance, dbPathInstance;
 var init_db = __esm({
   "src/memory/db.ts"() {
@@ -10464,6 +10511,7 @@ var init_manager3 = __esm({
   "src/memory/manager.ts"() {
     "use strict";
     init_client();
+    init_schema();
     init_db();
     init_embeddings();
     init_retrieval();
@@ -10593,6 +10641,48 @@ Return a JSON array of strings. Example:
         }
         return { id: memory.id, superseded: supersededIds.length > 0 ? supersededIds : void 0 };
       }
+      /**
+       * Count high-signal memories created since the given timestamp.
+       * Used for KIMI.md drift detection (Trigger A: session-start check).
+       */
+      countHighSignalMemoriesSince(repoPath, since) {
+        if (!this.db) return 0;
+        return countHighSignalMemoriesSince(this.db, repoPath, since);
+      }
+      /**
+       * Get the timestamp of the most recent KIMI.md refresh memory.
+       * Returns 0 if none exists.
+       */
+      getLastKimiMdRefreshTime(repoPath) {
+        if (!this.db) return 0;
+        const rows = this.db.prepare(
+          `SELECT created_at FROM memories
+         WHERE repo_path = ? AND topic_key = 'kimi_md_refresh'
+         AND forgotten = 0 AND superseded_by IS NULL
+         ORDER BY created_at DESC LIMIT 1`
+        ).all(repoPath);
+        return rows[0]?.created_at ?? 0;
+      }
+      /**
+       * Record that KIMI.md was refreshed. Creates a lightweight memory
+       * so drift detection knows when the snapshot was last updated.
+       */
+      async recordKimiMdRefresh(repoPath, sessionId) {
+        if (!this.db) return;
+        const embedding = new Float32Array(DEFAULT_EMBEDDING_DIM);
+        insertMemory(
+          this.db,
+          {
+            content: `KIMI.md refreshed for ${repoPath}`,
+            category: "event",
+            sourceSessionId: sessionId,
+            repoPath,
+            importance: 2,
+            topicKey: "kimi_md_refresh"
+          },
+          embedding
+        );
+      }
       /**
        * Recall memories using the full hybrid retrieval pipeline.
        */
@@ -13476,6 +13566,7 @@ function App({
   const [latestVersion, setLatestVersion] = useState10(initialUpdateResult?.latestVersion ?? null);
   const [theme, setTheme] = useState10(resolveTheme(initialCfg?.theme));
   const [showThemePicker, setShowThemePicker] = useState10(false);
+  const [kimiMdStale, setKimiMdStale] = useState10(false);
   useEffect6(() => {
     if (!cfg?.cloudMode || !initialCloudToken) return;
     let cancelled = false;
@@ -13531,6 +13622,8 @@ function App({
   const busyRef = useRef3(busy);
   const memoryManagerRef = useRef3(null);
   const sessionStartRecallRef = useRef3(null);
+  const kimiMdStaleNudgedRef = useRef3(false);
+  const turnCounterRef = useRef3(0);
   const pendingTextRef = useRef3(/* @__PURE__ */ new Map());
   const flushTimeoutRef = useRef3(null);
   const customCommandsRef = useRef3([]);
@@ -13760,6 +13853,13 @@ function App({
         } catch {
         }
       })();
+      if (existsSync3(join22(cwd, "KIMI.md"))) {
+        const lastRefresh = manager.getLastKimiMdRefreshTime(cwd);
+        const driftCount = manager.countHighSignalMemoriesSince(cwd, lastRefresh);
+        if (driftCount >= 5) {
+          setKimiMdStale(true);
+        }
+      }
     } else {
       memoryManagerRef.current?.close();
       memoryManagerRef.current = null;
@@ -14531,7 +14631,17 @@ function App({
             }
             permResolveRef.current = resolve2;
             setPerm({ tool: req.tool, args: req.args, resolve: resolve2 });
-          })
+          }),
+          onKimiMdStale: () => {
+            if (!kimiMdStaleNudgedRef.current) {
+              kimiMdStaleNudgedRef.current = true;
+              setKimiMdStale(true);
+              setEvents((e) => [
+                ...e,
+                { kind: "info", key: mkKey(), text: "Project context may be stale. Run /init to refresh KIMI.md based on recent changes." }
+              ]);
+            }
+          }
         }
       });
       if (existsSync3(join22(cwd, "KIMI.md"))) {
@@ -14560,6 +14670,9 @@ function App({
           ...e,
           { kind: "info", key: mkKey(), text: "KIMI.md generated; context loaded for future turns" }
         ]);
+        void memoryManagerRef.current?.recordKimiMdRefresh(cwd, ensureSessionId());
+        setKimiMdStale(false);
+        kimiMdStaleNudgedRef.current = false;
       }
     } catch (e) {
       if (e.name === "AbortError") {
@@ -15506,6 +15619,13 @@ ${lines.join("\n")}` }]);
           };
         }
       }
+      turnCounterRef.current += 1;
+      if (turnCounterRef.current % 15 === 0 && existsSync3(join22(process.cwd(), "KIMI.md")) && !kimiMdStale) {
+        setEvents((e) => [
+          ...e,
+          { kind: "info", key: mkKey(), text: "Tip: Rerunning /init occasionally helps KimiFlare stay accurate as your project evolves." }
+        ]);
+      }
       setBusy(true);
       gatewayMetaRef.current = null;
       setGatewayMeta(null);
@@ -15643,7 +15763,17 @@ ${lines.join("\n")}` }]);
         onToolLimitReached: () => new Promise((resolve2) => {
           limitResolveRef.current = resolve2;
           setLimitModal({ limit: 50, resolve: resolve2 });
-        })
+        }),
+        onKimiMdStale: () => {
+          if (!kimiMdStaleNudgedRef.current) {
+            kimiMdStaleNudgedRef.current = true;
+            setKimiMdStale(true);
+            setEvents((e) => [
+              ...e,
+              { kind: "info", key: mkKey(), text: "Project context may be stale. Run /init to refresh KIMI.md based on recent changes." }
+            ]);
+          }
+        }
       };
       try {
         await runAgentTurn({
@@ -16100,7 +16230,8 @@ ${lines.join("\n")}` }]);
           gatewayMeta,
           codeMode,
           cloudMode: cfg.cloudMode,
-          cloudBudget
+          cloudBudget,
+          kimiMdStale
         }
       ),
       activePicker?.kind === "file" && /* @__PURE__ */ jsx23(