@moonshot-ai/kimi-code 2.0.0 → 2.0.1

package/dist/search-worker.mjs

@@ -3,7 +3,7 @@ import fs from "node:fs";
 import path, { join, relative } from "node:path";
 import { fileURLToPath } from "node:url";
 import { createHash, randomUUID } from "node:crypto";
-import fs$1, { open, readFile, readdir, stat } from "node:fs/promises";
+import fs$1, { open, readFile, readdir, rm, stat } from "node:fs/promises";
 import os from "node:os";
 import { AsyncLocalStorage } from "node:async_hooks";
 
@@ -12401,6 +12401,7 @@ function analyzeWireLine(line) {
 const TEXT_INDEX_NAME = "body";
 const TRI_INDEX_NAME = "tri";
 const WIRE_FILENAME = "wire.jsonl";
+const TEXT_BUILD_TMP_SUFFIX = ".tmpbuild";
 const FILE_META_PREFIX = "\0meta\\file\\";
 const SESSION_META_PREFIX = "\0meta\\session\\";
 const STATS_KEY = "\0meta\\stats";
@@ -12423,6 +12424,7 @@ const SYNC_ROUND_TIME_BUDGET_MS = 3e4;
 const SYNC_FAILURE_ESCALATION_LIMIT = 5;
 const SESSION_SYNC_FAILURE_SKIP_LIMIT = 5;
 const SESSION_SYNC_SKIP_COOLDOWN_MS = 3e5;
+const SYNC_SESSION_CAP_DEFAULT = 500;
 const EMPTY_BUFFER = Buffer.alloc(0);
 function syncBudgetExhausted(budget) {
 	return budget.bytesLeft <= 0 || Date.now() >= budget.deadline;
@@ -12441,6 +12443,44 @@ async function sessionDirectoryIdentity(dir) {
 		throw error;
 	}
 }
+async function statMtimeMs(path) {
+	try {
+		return (await stat(path)).mtimeMs;
+	} catch {
+		return;
+	}
+}
+async function probeSessionSource(dir) {
+	let dirStat;
+	try {
+		dirStat = await stat(dir);
+	} catch {
+		return;
+	}
+	if (!dirStat.isDirectory()) return void 0;
+	const agentsMtimeMs = await statMtimeMs(join(dir, "agents"));
+	const stateMtimeMs = await statMtimeMs(join(dir, "state.json")) ?? await statMtimeMs(join(dir, "session-meta", "state.json"));
+	return {
+		dirMtimeMs: dirStat.mtimeMs,
+		agentsMtimeMs,
+		stateMtimeMs
+	};
+}
+function formatSessionSourceProbe(probe) {
+	return `${probe.dirMtimeMs}|${probe.agentsMtimeMs ?? "x"}|${probe.stateMtimeMs ?? "x"}`;
+}
+async function wireFileUnchanged(meta) {
+	if (meta.mtimeMs === void 0) return false;
+	let st;
+	try {
+		st = await stat(meta.path);
+	} catch {
+		return false;
+	}
+	if (st.size !== meta.size || st.mtimeMs !== meta.mtimeMs) return false;
+	if (meta.ino !== void 0 && st.ino !== meta.ino) return false;
+	return true;
+}
 const querySourceChecks = {
 	running: false,
 	waiters: /* @__PURE__ */ new Set()
@@ -12581,6 +12621,7 @@ var SearchIndexCore = class {
 	syncRoundBytes = SYNC_ROUND_BYTE_BUDGET;
 	syncRoundMs = SYNC_ROUND_TIME_BUDGET_MS;
 	syncSkipCooldownMs = SESSION_SYNC_SKIP_COOLDOWN_MS;
+	syncSessionCap = SYNC_SESSION_CAP_DEFAULT;
 	constructor(options) {
 		this.options = options;
 	}
@@ -12610,6 +12651,27 @@ var SearchIndexCore = class {
 			throw new GlobalSearchError("index_unavailable", "search service is disposed");
 		}
 		await this.publishDb(db, null);
+		if (!db.readOnly) await this.cleanInterruptedTextBuilds();
+	}
+	async cleanInterruptedTextBuilds() {
+		const entries = await readdir(this.indexDir, { withFileTypes: true }).catch(() => void 0);
+		if (entries === void 0) return;
+		for (const entry of entries) {
+			if (!entry.isDirectory() || !entry.name.endsWith(TEXT_BUILD_TMP_SUFFIX)) continue;
+			const path = join(this.indexDir, entry.name);
+			try {
+				await rm(path, {
+					recursive: true,
+					force: true
+				});
+				this.log.info("global search: removed an interrupted text-build leftover", { dir: path });
+			} catch (error) {
+				this.log.warn("global search: cannot remove an interrupted text-build leftover", {
+					dir: path,
+					error: errorMessage(error)
+				});
+			}
+		}
 	}
 	tokenGeneration() {
 		return `${this.options.bootSalt}:${this.generation}`;
@@ -12822,6 +12884,7 @@ var SearchIndexCore = class {
 			deadline: Date.now() + this.syncRoundMs
 		};
 		let indexed = 0;
+		let syncedFresh = 0;
 		let truncated = false;
 		let failures = 0;
 		for (const summary of sessions) {
@@ -12833,8 +12896,13 @@ var SearchIndexCore = class {
 				truncated = true;
 				break;
 			}
+			if (syncedFresh >= this.syncSessionCap) {
+				truncated = true;
+				break;
+			}
 			const result = await this.syncSession(db, summary, budget);
 			if (result === void 0) continue;
+			if (result.skipped !== true) syncedFresh++;
 			if (result.truncated) truncated = true;
 			if (result.failed) {
 				const count = (this.sessionSyncFailures.get(summary.id) ?? 0) + 1;
@@ -12917,6 +12985,13 @@ var SearchIndexCore = class {
 		await db.del(SESSION_META_PREFIX + sessionId);
 	}
 	async syncSession(db, summary, budget) {
+		const metaKey = SESSION_META_PREFIX + summary.id;
+		const previous = db.get(metaKey);
+		if (await this.sessionSourceUnchanged(db, summary, previous)) return {
+			truncated: false,
+			failed: false,
+			skipped: true
+		};
 		let identity;
 		try {
 			identity = await sessionDirectoryIdentity(summary.dir);
@@ -12932,8 +13007,6 @@ var SearchIndexCore = class {
 			return;
 		}
 		const title = await sessionDirectoryTitle(summary.dir, this.log);
-		const metaKey = SESSION_META_PREFIX + summary.id;
-		const previous = db.get(metaKey);
 		if (previous?.kind !== "sessionMeta" || previous.identity !== identity || previous.dir !== summary.dir) {
 			await this.deleteSessionDocs(db, summary.id);
 			if (previous !== void 0) this.syncReplaced = true;
@@ -12988,12 +13061,38 @@ var SearchIndexCore = class {
 				if (existing !== void 0) this.syncReplaced = true;
 			}
 		} else if (existing !== void 0) await db.del(titleKey);
+		if (!truncated && !failed) await this.stampSessionSource(db, metaKey, meta, summary);
 		return {
 			truncated,
 			failed,
 			error
 		};
 	}
+	async sessionSourceUnchanged(db, summary, previous) {
+		if (previous?.kind !== "sessionMeta") return false;
+		if (previous.dir !== summary.dir) return false;
+		if (previous.sourceUpdatedAt !== summary.updatedAt) return false;
+		if ((previous.title ?? "") !== (summary.title ?? "")) return false;
+		const syncSource = previous.syncSource;
+		if (syncSource === void 0) return false;
+		const probe = await probeSessionSource(summary.dir);
+		if (probe === void 0 || formatSessionSourceProbe(probe) !== syncSource) return false;
+		for (const row of db.query({ key: { prefix: fileMetaPrefixFor(summary.id) } })) {
+			const meta = row.value;
+			if (meta.kind !== "fileMeta") continue;
+			if (!await wireFileUnchanged(meta)) return false;
+		}
+		return true;
+	}
+	async stampSessionSource(db, metaKey, meta, summary) {
+		const probe = await probeSessionSource(summary.dir);
+		if (probe === void 0) return;
+		await db.set(metaKey, {
+			...meta,
+			sourceUpdatedAt: summary.updatedAt,
+			syncSource: formatSessionSourceProbe(probe)
+		});
+	}
 	async deleteFileDocs(db, meta) {
 		const prefix = `${meta.sessionId}/${meta.agentId}/${meta.source}:`;
 		for (const row of db.query({
@@ -13520,6 +13619,7 @@ const core = new SearchIndexCore({
 		});
 	}
 });
+if (typeof data.syncSessionCap === "number") core.syncSessionCap = data.syncSessionCap;
 function toErrorPayload(error) {
 	if (error instanceof GlobalSearchError) return {
 		message: error.message,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@moonshot-ai/kimi-code",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "The Starting Point for Next-Gen Agents",
   "license": "MIT",
   "author": "Moonshot AI",
@@ -72,16 +72,16 @@
     "zod": "^4.3.6",
     "@moonshot-ai/acp-server": "^0.0.1",
     "@moonshot-ai/agent-core-v2": "^0.4.3",
+    "@moonshot-ai/kap-server": "^0.2.3",
     "@moonshot-ai/kimi-code-sdk": "^0.20.0",
     "@moonshot-ai/kimi-code-oauth": "^0.4.0",
-    "@moonshot-ai/kap-server": "^0.2.3",
-    "@moonshot-ai/kimi-telemetry": "^0.1.1",
-    "@moonshot-ai/migration-legacy": "^0.1.16",
+    "@moonshot-ai/kimi-telemetry": "^0.2.0",
     "@moonshot-ai/minidb": "^0.2.0",
-    "@moonshot-ai/vis-server": "^0.1.1",
+    "@moonshot-ai/migration-legacy": "^0.1.16",
+    "@moonshot-ai/pi-tui": "^0.84.5",
     "@moonshot-ai/remote-control": "^0.0.0",
-    "@moonshot-ai/vis-web": "0.1.1",
-    "@moonshot-ai/pi-tui": "^0.84.5"
+    "@moonshot-ai/vis-server": "^0.1.1",
+    "@moonshot-ai/vis-web": "0.1.1"
   },
   "engines": {
     "node": ">=22.19.0"