@agentmemory/agentmemory 0.9.2 → 0.9.4

package/dist/{src-tmuZyobT.mjs → src-uDy2jLO-.mjs}

@@ -1,5 +1,5 @@
 import { a as jaccardSimilarity, i as generateId, n as STREAM, r as fingerprintId, t as KV } from "./cli.mjs";
-import { _ as loadTeamConfig, a as getConsolidationDecayDays, c as isAutoCompressEnabled, d as isGraphExtractionEnabled, f as loadClaudeBridgeConfig, g as loadSnapshotConfig, h as loadFallbackConfig, i as detectEmbeddingProvider, l as isConsolidationEnabled, m as loadEmbeddingConfig, n as getVisibleTools, o as getEnvVar, p as loadConfig, r as VERSION, t as getAllTools, u as isContextInjectionEnabled } from "./tools-registry-CHH84gIQ.mjs";
+import { _ as loadSnapshotConfig, a as detectLlmProviderKind, d as isContextInjectionEnabled, f as isGraphExtractionEnabled, g as loadFallbackConfig, h as loadEmbeddingConfig, i as detectEmbeddingProvider, l as isAutoCompressEnabled, m as loadConfig, n as getVisibleTools, o as getConsolidationDecayDays, p as loadClaudeBridgeConfig, r as VERSION, s as getEnvVar, t as getAllTools, u as isConsolidationEnabled, v as loadTeamConfig } from "./tools-registry-Co8VIL4t.mjs";
 import { execFile } from "node:child_process";
 import { constants, existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { basename, dirname, extname, join, resolve, sep } from "node:path";
@@ -2018,8 +2018,10 @@ var SearchIndex = class SearchIndex {
 //#endregion
 //#region src/state/index-persistence.ts
 const DEBOUNCE_MS = 5e3;
+const FAILURE_LOG_THROTTLE_MS = 6e4;
 var IndexPersistence = class {
 	timer = null;
+	lastFailureLogAt = 0;
 	constructor(kv, bm25, vector) {
 		this.kv = kv;
 		this.bm25 = bm25;
@@ -2027,15 +2029,21 @@ var IndexPersistence = class {
 	}
 	scheduleSave() {
 		if (this.timer) clearTimeout(this.timer);
-		this.timer = setTimeout(() => this.save(), DEBOUNCE_MS);
+		this.timer = setTimeout(() => {
+			this.save().catch((err) => this.logFailure(err));
+		}, DEBOUNCE_MS);
 	}
 	async save() {
 		if (this.timer) {
 			clearTimeout(this.timer);
 			this.timer = null;
 		}
-		await this.kv.set(KV.bm25Index, "data", this.bm25.serialize());
-		if (this.vector && this.vector.size > 0) await this.kv.set(KV.bm25Index, "vectors", this.vector.serialize());
+		try {
+			await this.kv.set(KV.bm25Index, "data", this.bm25.serialize());
+			if (this.vector && this.vector.size > 0) await this.kv.set(KV.bm25Index, "vectors", this.vector.serialize());
+		} catch (err) {
+			this.logFailure(err);
+		}
 	}
 	async load() {
 		let bm25 = null;
@@ -2055,6 +2063,18 @@ var IndexPersistence = class {
 			this.timer = null;
 		}
 	}
+	logFailure(err) {
+		const now = Date.now();
+		if (now - this.lastFailureLogAt < FAILURE_LOG_THROTTLE_MS) return;
+		this.lastFailureLogAt = now;
+		const code = err?.code;
+		const message = err instanceof Error ? err.message : String(err);
+		logger.warn("index persistence: failed to save BM25/vector index", {
+			code,
+			message,
+			hint: code === "TIMEOUT" ? "iii-engine state::set timed out; recent index updates remain in memory and will retry on the next debounce flush" : void 0
+		});
+	}
 };
 
 //#endregion
@@ -2507,10 +2527,10 @@ function registerObserveFunction(sdk, kv, dedupMap, maxObservationsPerSession) {
 				};
 			}
 			if (pendingImageData && (pendingImageData.startsWith("data:image/") || pendingImageData.startsWith("iVBORw0KGgo") || pendingImageData.startsWith("/9j/"))) {
-				const { saveImageToDisk } = await import("./image-store-BLOkD0xV.mjs");
+				const { saveImageToDisk } = await import("./image-store-Cn9eD-7D.mjs");
 				const { filePath, bytesWritten } = await saveImageToDisk(pendingImageData);
 				raw.imageData = filePath;
-				const { incrementImageRef } = await import("./image-refs-Dq5wcV-a.mjs");
+				const { incrementImageRef } = await import("./image-refs-BfT7XAa-.mjs");
 				await incrementImageRef(kv, filePath);
 				sdk.triggerVoid("mem::disk-size-delta", { deltaBytes: bytesWritten });
 				if (process.env["AGENTMEMORY_IMAGE_EMBEDDINGS"] === "true") sdk.triggerVoid("mem::vision-embed", {
@@ -2523,7 +2543,7 @@ function registerObserveFunction(sdk, kv, dedupMap, maxObservationsPerSession) {
 				await kv.set(KV.observations(payload.sessionId), obsId, raw);
 			} catch (error) {
 				if (raw.imageData) {
-					const { deleteImage } = await import("./image-store-BLOkD0xV.mjs");
+					const { deleteImage } = await import("./image-store-Cn9eD-7D.mjs");
 					const { deletedBytes } = await deleteImage(raw.imageData);
 					if (deletedBytes > 0) sdk.triggerVoid("mem::disk-size-delta", { deltaBytes: -deletedBytes });
 				}
@@ -4556,7 +4576,7 @@ function registerRememberFunction(sdk, kv) {
 		const deletedMemoryIds = [];
 		const deletedObservationIds = [];
 		let deletedSession = false;
-		const { decrementImageRef } = await import("./image-refs-Dq5wcV-a.mjs");
+		const { decrementImageRef } = await import("./image-refs-BfT7XAa-.mjs");
 		if (data.memoryId) {
 			const mem = await kv.get(KV.memories, data.memoryId);
 			await kv.delete(KV.memories, data.memoryId);
@@ -4615,7 +4635,7 @@ const DEFAULTS$1 = {
 function registerEvictFunction(sdk, kv) {
 	sdk.registerFunction("mem::evict", async (data) => {
 		const dryRun = data?.dryRun ?? false;
-		const { decrementImageRef } = await import("./image-refs-Dq5wcV-a.mjs");
+		const { decrementImageRef } = await import("./image-refs-BfT7XAa-.mjs");
 		const configOverride = await kv.get(KV.config, "eviction").catch(() => null);
 		const cfg = {
 			...DEFAULTS$1,
@@ -5218,7 +5238,7 @@ function registerAutoForgetFunction(sdk, kv) {
 	sdk.registerFunction("mem::auto-forget", async (data) => {
 		const dryRun = data?.dryRun ?? false;
 		const now = Date.now();
-		const { decrementImageRef } = await import("./image-refs-Dq5wcV-a.mjs");
+		const { decrementImageRef } = await import("./image-refs-BfT7XAa-.mjs");
 		const result = {
 			ttlExpired: [],
 			contradictions: [],
@@ -5449,7 +5469,9 @@ function registerExportImportFunction(sdk, kv) {
 			"0.8.13",
 			"0.9.0",
 			"0.9.1",
-			"0.9.2"
+			"0.9.2",
+			"0.9.3",
+			"0.9.4"
 		]).has(importData.version)) return {
 			success: false,
 			error: `Unsupported export version: ${importData.version}`
@@ -11806,7 +11828,7 @@ function registerRetentionFunctions(sdk, kv) {
 		const threshold = typeof data?.threshold === "number" && Number.isFinite(data.threshold) ? data.threshold : DEFAULT_DECAY.tierThresholds.cold;
 		const maxEvictRaw = typeof data?.maxEvict === "number" && Number.isInteger(data.maxEvict) ? data.maxEvict : 50;
 		const maxEvict = Math.min(1e3, Math.max(0, maxEvictRaw));
-		const { decrementImageRef } = await import("./image-refs-Dq5wcV-a.mjs");
+		const { decrementImageRef } = await import("./image-refs-BfT7XAa-.mjs");
 		const candidates = (await kv.list(KV.retentionScores)).filter((s) => s.score < threshold).sort((a, b) => a.score - b.score).slice(0, maxEvict);
 		if (data?.dryRun) return {
 			success: true,
@@ -12067,7 +12089,7 @@ function parseJsonlText(text, fallbackSessionId) {
 		const parsed = JSON.parse(line);
 		if (parsed && typeof parsed === "object") entries.push(parsed);
 	} catch {}
-	let sessionId = fallbackSessionId || "";
+	let sessionId = "";
 	let cwd = "";
 	let firstTs = "";
 	let lastTs = "";
@@ -12128,7 +12150,7 @@ function parseJsonlText(text, fallbackSessionId) {
 			});
 		} else if (entry.type === "summary" || entry.type === "system") {}
 	}
-	const effectiveSessionId = sessionId || generateId("sess");
+	const effectiveSessionId = sessionId || fallbackSessionId || generateId("sess");
 	for (const obs of observations) if (obs.sessionId === "imported") obs.sessionId = effectiveSessionId;
 	const nowIso = (/* @__PURE__ */ new Date()).toISOString();
 	return {
@@ -12238,6 +12260,8 @@ function projectTimeline(observations) {
 
 //#endregion
 //#region src/functions/replay.ts
+const MAX_FILES_DEFAULT = 200;
+const MAX_FILES_UPPER_BOUND = 1e3;
 const SENSITIVE_PATH_PATTERNS = [
 	/(^|[\\/_.-])secret([\\/_.-]|s?$)/i,
 	/(^|[\\/_.-])credentials?([\\/_.-]|$)/i,
@@ -12372,8 +12396,11 @@ async function loadObservations(kv, sessionId) {
 }
 async function findJsonlFiles(root, limit = 200) {
 	const out = [];
+	let discovered = 0;
+	let walked = 0;
+	const traversalCap = Math.max(limit * 50, 5e4);
 	async function walk(dir) {
-		if (out.length >= limit) return;
+		if (walked >= traversalCap) return;
 		let names;
 		try {
 			names = await readdir(dir);
@@ -12381,7 +12408,8 @@ async function findJsonlFiles(root, limit = 200) {
 			return;
 		}
 		for (const name of names) {
-			if (out.length >= limit) return;
+			if (walked >= traversalCap) return;
+			walked++;
 			const full = join(dir, name);
 			let st;
 			try {
@@ -12391,11 +12419,20 @@ async function findJsonlFiles(root, limit = 200) {
 			}
 			if (st.isSymbolicLink()) continue;
 			if (st.isDirectory()) await walk(full);
-			else if (st.isFile() && name.endsWith(".jsonl")) out.push(full);
+			else if (st.isFile() && name.endsWith(".jsonl")) {
+				discovered++;
+				if (out.length < limit) out.push(full);
+			}
 		}
 	}
 	await walk(root);
-	return out;
+	const traversalCapped = walked >= traversalCap;
+	return {
+		files: out,
+		truncated: discovered > out.length || traversalCapped,
+		discovered,
+		traversalCapped
+	};
 }
 function registerReplayFunctions(sdk, kv) {
 	sdk.registerFunction("mem::replay::load", async (data) => {
@@ -12443,10 +12480,21 @@ function registerReplayFunctions(sdk, kv) {
 				error: "path not found"
 			};
 		}
+		const maxFiles = Number.isInteger(data.maxFiles) && data.maxFiles > 0 ? Math.min(data.maxFiles, MAX_FILES_UPPER_BOUND) : MAX_FILES_DEFAULT;
 		let files = [];
-		if (stat.isDirectory()) files = await findJsonlFiles(abs, data.maxFiles || 200);
-		else if (stat.isFile() && abs.endsWith(".jsonl")) files = [abs];
-		else return {
+		let truncated = false;
+		let discovered = 0;
+		let traversalCapped = false;
+		if (stat.isDirectory()) {
+			const found = await findJsonlFiles(abs, maxFiles);
+			files = found.files;
+			truncated = found.truncated;
+			discovered = found.discovered;
+			traversalCapped = found.traversalCapped;
+		} else if (stat.isFile() && abs.endsWith(".jsonl")) {
+			files = [abs];
+			discovered = 1;
+		} else return {
 			success: false,
 			error: "path must be a .jsonl file or directory"
 		};
@@ -12454,7 +12502,12 @@ function registerReplayFunctions(sdk, kv) {
 			success: true,
 			imported: 0,
 			sessionIds: [],
-			observations: 0
+			observations: 0,
+			discovered,
+			truncated,
+			traversalCapped,
+			maxFiles,
+			maxFilesUpperBound: MAX_FILES_UPPER_BOUND
 		};
 		const sessionIds = [];
 		let observationCount = 0;
@@ -12520,7 +12573,12 @@ function registerReplayFunctions(sdk, kv) {
 			success: true,
 			imported: files.length,
 			sessionIds,
-			observations: observationCount
+			observations: observationCount,
+			discovered,
+			truncated,
+			traversalCapped,
+			maxFiles,
+			maxFilesUpperBound: MAX_FILES_UPPER_BOUND
 		};
 	});
 }
@@ -12747,6 +12805,28 @@ function requireConfiguredSecret(secret, feature) {
 		body: { error: `${feature} requires AGENTMEMORY_SECRET` }
 	};
 }
+function flagDisabledResponse(opts) {
+	return {
+		status_code: 503,
+		body: opts
+	};
+}
+function graphDisabledResponse() {
+	return flagDisabledResponse({
+		error: "Knowledge graph not enabled",
+		flag: "GRAPH_EXTRACTION_ENABLED",
+		enableHow: "Set GRAPH_EXTRACTION_ENABLED=true and restart. Requires an LLM provider key.",
+		docsHref: "https://github.com/rohitg00/agentmemory#knowledge-graph"
+	});
+}
+function consolidationDisabledResponse() {
+	return flagDisabledResponse({
+		error: "Consolidation pipeline not enabled",
+		flag: "CONSOLIDATION_ENABLED",
+		enableHow: "Set CONSOLIDATION_ENABLED=true and restart. Requires an LLM provider key.",
+		docsHref: "https://github.com/rohitg00/agentmemory#consolidation"
+	});
+}
 function asNonEmptyString$1(value) {
 	if (typeof value !== "string") return null;
 	const trimmed = value.trim();
@@ -12798,6 +12878,77 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 			http_method: "GET"
 		}
 	});
+	sdk.registerFunction("api::config-flags", async (req) => {
+		const authErr = checkAuth(req, secret);
+		if (authErr) return authErr;
+		return {
+			status_code: 200,
+			body: {
+				version: VERSION,
+				provider: detectLlmProviderKind(),
+				embeddingProvider: detectEmbeddingProvider() ? "embeddings" : "none",
+				flags: [
+					{
+						key: "GRAPH_EXTRACTION_ENABLED",
+						label: "Knowledge graph extraction",
+						enabled: isGraphExtractionEnabled(),
+						default: false,
+						affects: ["Graph", "Dashboard"],
+						needsLlm: true,
+						description: "Extracts entities and relations from observations into a knowledge graph.",
+						enableHow: "Set GRAPH_EXTRACTION_ENABLED=true and provide an LLM key, then restart.",
+						docsHref: "https://github.com/rohitg00/agentmemory#knowledge-graph"
+					},
+					{
+						key: "CONSOLIDATION_ENABLED",
+						label: "Memory consolidation",
+						enabled: isConsolidationEnabled(),
+						default: false,
+						affects: [
+							"Dashboard",
+							"Memories",
+							"Crystals"
+						],
+						needsLlm: true,
+						description: "Periodically summarizes sessions into semantic facts + procedures.",
+						enableHow: "Set CONSOLIDATION_ENABLED=true and provide an LLM key, then restart.",
+						docsHref: "https://github.com/rohitg00/agentmemory#consolidation"
+					},
+					{
+						key: "AGENTMEMORY_AUTO_COMPRESS",
+						label: "LLM-powered observation compression",
+						enabled: isAutoCompressEnabled(),
+						default: false,
+						affects: ["Memories", "Timeline"],
+						needsLlm: true,
+						description: "Every observation is compressed by the LLM for richer summaries (costs tokens). OFF uses zero-LLM synthetic compression.",
+						enableHow: "Set AGENTMEMORY_AUTO_COMPRESS=true and provide an LLM key.",
+						docsHref: "https://github.com/rohitg00/agentmemory/issues/138"
+					},
+					{
+						key: "AGENTMEMORY_INJECT_CONTEXT",
+						label: "In-conversation context injection",
+						enabled: isContextInjectionEnabled(),
+						default: false,
+						affects: ["Hooks"],
+						needsLlm: false,
+						description: "Hooks write recalled context into Claude Code's conversation. OFF captures in the background without injecting.",
+						enableHow: "Set AGENTMEMORY_INJECT_CONTEXT=true and restart.",
+						docsHref: "https://github.com/rohitg00/agentmemory/issues/143"
+					}
+				]
+			}
+		};
+	});
+	sdk.registerTrigger({
+		type: "http",
+		function_id: "api::config-flags",
+		config: {
+			api_path: "/agentmemory/config/flags",
+			http_method: "GET",
+			middleware_function_ids: ["middleware::api-auth"]
+		}
+	});
 	sdk.registerFunction("api::health", async (req) => {
 		const health = await getLatestHealth(kv);
 		const functionMetrics = metricsStore ? await metricsStore.getAll() : [];
@@ -13032,11 +13183,12 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 			payload.path = body.path.trim();
 		}
 		if (body.maxFiles !== void 0) {
-			if (!Number.isInteger(body.maxFiles) || body.maxFiles < 1) return {
+			const n = body.maxFiles;
+			if (!Number.isInteger(n) || n < 1 || n > MAX_FILES_UPPER_BOUND) return {
 				status_code: 400,
-				body: { error: "maxFiles must be a positive integer" }
+				body: { error: `maxFiles must be an integer between 1 and ${MAX_FILES_UPPER_BOUND}` }
 			};
-			payload.maxFiles = body.maxFiles;
+			payload.maxFiles = n;
 		}
 		return {
 			status_code: 202,
@@ -13618,10 +13770,7 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 				})
 			};
 		} catch {
-			return {
-				status_code: 404,
-				body: { error: "Knowledge graph not enabled" }
-			};
+			return graphDisabledResponse();
 		}
 	});
 	sdk.registerTrigger({
@@ -13644,10 +13793,7 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 				})
 			};
 		} catch {
-			return {
-				status_code: 404,
-				body: { error: "Knowledge graph not enabled" }
-			};
+			return graphDisabledResponse();
 		}
 	});
 	sdk.registerTrigger({
@@ -13674,10 +13820,7 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 				})
 			};
 		} catch {
-			return {
-				status_code: 404,
-				body: { error: "Knowledge graph not enabled" }
-			};
+			return graphDisabledResponse();
 		}
 	});
 	sdk.registerTrigger({
@@ -13700,10 +13843,7 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 				})
 			};
 		} catch {
-			return {
-				status_code: 404,
-				body: { error: "Consolidation pipeline not enabled" }
-			};
+			return consolidationDisabledResponse();
 		}
 	});
 	sdk.registerTrigger({
@@ -13964,6 +14104,32 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 			http_method: "GET"
 		}
 	});
+	sdk.registerFunction("api::memory-by-id", async (req) => {
+		const authErr = checkAuth(req, secret);
+		if (authErr) return authErr;
+		const id = req.path_params?.["id"];
+		if (!id || typeof id !== "string") return {
+			status_code: 400,
+			body: { error: "id path parameter is required" }
+		};
+		const memory = await kv.get(KV.memories, id);
+		if (!memory) return {
+			status_code: 404,
+			body: { error: `memory not found: ${id}` }
+		};
+		return {
+			status_code: 200,
+			body: { memory }
+		};
+	});
+	sdk.registerTrigger({
+		type: "http",
+		function_id: "api::memory-by-id",
+		config: {
+			api_path: "/agentmemory/memories/:id",
+			http_method: "GET"
+		}
+	});
 	sdk.registerFunction("api::semantic-list", async (req) => {
 		const authErr = checkAuth(req, secret);
 		if (authErr) return authErr;
@@ -15842,6 +16008,15 @@ function registerEventTriggers(sdk, kv) {
 				error: err instanceof Error ? err.message : String(err)
 			});
 		}
+		if (isGraphExtractionEnabled()) try {
+			const compressed = (await kv.list(KV.observations(data.sessionId))).filter((o) => o.title);
+			if (compressed.length > 0) sdk.triggerVoid("mem::graph-extract", { observations: compressed });
+		} catch (err) {
+			logger.warn("graph-extract triggerVoid failed", {
+				sessionId: data.sessionId,
+				error: err instanceof Error ? err.message : String(err)
+			});
+		}
 		return summary;
 	});
 	sdk.registerTrigger({
@@ -17753,6 +17928,14 @@ function initMetrics(getMeter) {
 function hasGetMeter(sdk) {
 	return typeof sdk === "object" && sdk !== null && "getMeter" in sdk && typeof sdk.getMeter === "function";
 }
+let lastUnhandledLogAt = 0;
+process.on("unhandledRejection", (reason) => {
+	const now = Date.now();
+	if (now - lastUnhandledLogAt < 6e4) return;
+	lastUnhandledLogAt = now;
+	const r = reason;
+	console.warn(`[agentmemory] unhandledRejection (suppressed):`, r?.code ? `${r.code} ${r.function_id ?? ""} ${r.message ?? ""}`.trim() : reason);
+});
 async function main() {
 	const config = loadConfig();
 	const embeddingConfig = loadEmbeddingConfig();
@@ -17959,4 +18142,4 @@ main().catch((err) => {
 
 //#endregion
 export { deleteImage as a, saveImageToDisk as c, IMAGES_DIR as i, touchImage as l, getImageRefCount as n, getMaxBytes as o, incrementImageRef as r, isManagedImagePath as s, decrementImageRef as t };
-//# sourceMappingURL=src-tmuZyobT.mjs.map
+//# sourceMappingURL=src-uDy2jLO-.mjs.map