@agentmemory/agentmemory 0.9.26 → 0.9.28

package/dist/{src-u7kAEUC0.mjs → src-CzgoepGU.mjs}

@@ -1,9 +1,8 @@
-import { a as jaccardSimilarity, i as generateId, n as STREAM, r as fingerprintId, t as KV } from "./schema-Dsr_V2Wp.mjs";
-import { n as logger, t as bootLog } from "./logger-yHTcEBAI.mjs";
-import { t as VERSION } from "./version-BMFYFFut.mjs";
-import { a as isManagedImagePath, n as deleteImage, r as getMaxBytes, t as IMAGES_DIR } from "./image-store-Gpo2mgM9.mjs";
-import { r as withKeyedLock, t as getImageRefCount } from "./image-refs-C7h9L5wx.mjs";
-import { _ as loadConfig, a as getAgentId, b as loadSnapshotConfig, c as getFollowupWindowSeconds, d as isAutoCompressEnabled, f as isConsolidationEnabled, g as loadClaudeBridgeConfig, h as isGraphExtractionEnabled, i as detectLlmProviderKind, m as isDropStaleIndexEnabled, n as getVisibleTools, o as getConsolidationDecayDays, p as isContextInjectionEnabled, r as detectEmbeddingProvider, s as getEnvVar, t as getAllTools, u as isAgentScopeIsolated, v as loadEmbeddingConfig, x as loadTeamConfig, y as loadFallbackConfig } from "./tools-registry-Dzxv9iUu.mjs";
+import { a as jaccardSimilarity, i as generateId, n as STREAM, r as fingerprintId, t as KV } from "./schema-CNiCWzV9.mjs";
+import { n as logger, t as bootLog } from "./logger-BEB7pCI9.mjs";
+import { i as VERSION, n as getAllTools, r as getVisibleTools } from "./tools-registry-D3NEoe5m.mjs";
+import { a as isManagedImagePath, i as getMaxBytes, n as IMAGES_DIR, o as saveImageToDisk, r as deleteImage, t as withKeyedLock } from "./keyed-mutex-DTHtcZrt.mjs";
+import { _ as loadFallbackConfig, a as getEnvVar, c as isAgentScopeIsolated, d as isContextInjectionEnabled, f as isDropStaleIndexEnabled, g as loadEmbeddingConfig, h as loadConfig, i as getConsolidationDecayDays, l as isAutoCompressEnabled, m as loadClaudeBridgeConfig, n as detectLlmProviderKind, o as getFollowupWindowSeconds, p as isGraphExtractionEnabled, r as getAgentId, t as detectEmbeddingProvider, u as isConsolidationEnabled, v as loadSnapshotConfig, y as loadTeamConfig } from "./config-ChC9EtgV.mjs";
 import { createRequire } from "node:module";
 import { execFile } from "node:child_process";
 import { constants, existsSync, mkdirSync, readFileSync, unlinkSync, writeFileSync } from "node:fs";
@@ -3109,6 +3108,12 @@ function registerSearchFunction(sdk, kv) {
 		}
 		const projectFilter = typeof data.project === "string" && data.project.trim().length > 0 ? data.project.trim() : void 0;
 		const cwdFilter = typeof data.cwd === "string" && data.cwd.trim().length > 0 ? data.cwd.trim() : void 0;
+		const isolated = isAgentScopeIsolated();
+		const explicitAgentId = typeof data.agentId === "string" && data.agentId.trim().length > 0 ? data.agentId.trim() : void 0;
+		const wildcardAgent = explicitAgentId === "*";
+		const envAgentId = isolated ? getAgentId() : void 0;
+		const filterAgentId = wildcardAgent ? void 0 : explicitAgentId ?? envAgentId;
+		if (isolated && !wildcardAgent && !explicitAgentId && !envAgentId) throw new Error("mem::search: AGENTMEMORY_AGENT_SCOPE=isolated is set but no agent id is available (env AGENT_ID unset and no explicit agentId in the call). Refusing to read cross-agent rows. Pass agentId: \"*\" to opt in to a wildcard read.");
 		const format = typeof data.format === "string" ? data.format : "full";
 		if (![
 			"full",
@@ -3124,7 +3129,7 @@ function registerSearchFunction(sdk, kv) {
 			const count = await rebuildIndex(kv);
 			logger.info("Search index rebuilt", { entries: count });
 		}
-		const filtering = !!(projectFilter || cwdFilter);
+		const filtering = !!(projectFilter || cwdFilter || filterAgentId);
 		const fetchLimit = filtering ? Math.max(effectiveLimit * 10, 100) : effectiveLimit;
 		const results = idx.search(query, fetchLimit);
 		const sessionCache = /* @__PURE__ */ new Map();
@@ -3141,9 +3146,10 @@ function registerSearchFunction(sdk, kv) {
 			memoryProjectCache.set(obsId, proj);
 			return proj;
 		};
+		const earlyCap = filterAgentId ? fetchLimit : effectiveLimit;
 		const candidates = [];
 		for (const r of results) {
-			if (candidates.length >= effectiveLimit) break;
+			if (candidates.length >= earlyCap) break;
 			if (filtering) {
 				const s = await loadSession(r.sessionId);
 				if (s) {
@@ -3165,7 +3171,10 @@ function registerSearchFunction(sdk, kv) {
 		const enriched = [];
 		for (let i = 0; i < candidates.length; i++) {
 			const obs = obsResults[i];
-			if (obs) enriched.push({
+			if (!obs) continue;
+			if (filterAgentId !== void 0 && obs.agentId !== filterAgentId) continue;
+			if (enriched.length >= effectiveLimit) break;
+			enriched.push({
 				observation: obs,
 				score: candidates[i].score,
 				sessionId: candidates[i].sessionId
@@ -3327,10 +3336,9 @@ function registerObserveFunction(sdk, kv, dedupMap, maxObservationsPerSession) {
 			const inheritedAgentId = existingSession ? existingSession.agentId : getAgentId();
 			if (inheritedAgentId) raw.agentId = inheritedAgentId;
 			if (pendingImageData && (pendingImageData.startsWith("data:image/") || pendingImageData.startsWith("iVBORw0KGgo") || pendingImageData.startsWith("/9j/"))) {
-				const { saveImageToDisk } = await import("./image-store-Gpo2mgM9.mjs").then((n) => n.i);
 				const { filePath, bytesWritten } = await saveImageToDisk(pendingImageData);
 				raw.imageData = filePath;
-				const { incrementImageRef } = await import("./image-refs-C7h9L5wx.mjs").then((n) => n.n);
+				const { incrementImageRef } = await import("./image-refs-C2YU7BIm.mjs");
 				await incrementImageRef(kv, filePath);
 				sdk.trigger({
 					function_id: "mem::disk-size-delta",
@@ -3350,13 +3358,13 @@ function registerObserveFunction(sdk, kv, dedupMap, maxObservationsPerSession) {
 			try {
 				await kv.set(KV.observations(payload.sessionId), obsId, raw);
 			} catch (error) {
-				if (raw.imageData) {
-					const { deleteImage } = await import("./image-store-Gpo2mgM9.mjs").then((n) => n.i);
-					const { deletedBytes } = await deleteImage(raw.imageData);
-					if (deletedBytes > 0) sdk.trigger({
-						function_id: "mem::disk-size-delta",
-						payload: { deltaBytes: -deletedBytes },
-						action: TriggerAction.Void()
+				if (raw.imageData) try {
+					const { decrementImageRef } = await import("./image-refs-C2YU7BIm.mjs");
+					await decrementImageRef(kv, sdk, raw.imageData);
+				} catch (rollbackError) {
+					logger.error("Failed to roll back image ref after observation write failure", {
+						imageRef: raw.imageData,
+						error: rollbackError instanceof Error ? rollbackError.message : String(rollbackError)
 					});
 				}
 				throw error;
@@ -3525,6 +3533,7 @@ function registerImageQuotaCleanup(sdk, kv) {
 				await withKeyedLock(`imgRef:${f.filePath}`, async () => {
 					let refCount;
 					try {
+						const { getImageRefCount } = await import("./image-refs-C2YU7BIm.mjs");
 						refCount = await getImageRefCount(kv, f.filePath);
 					} catch (err) {
 						logger.error("Failed to read refCount; skipping eviction", {
@@ -4615,6 +4624,12 @@ function registerContextFunction(sdk, kv, tokenBudget) {
 	sdk.registerFunction("mem::context", async (data) => {
 		const budget = data.budget || tokenBudget;
 		const blocks = [];
+		const isolated = isAgentScopeIsolated();
+		const explicitAgentId = typeof data.agentId === "string" && data.agentId.trim().length > 0 ? data.agentId.trim() : void 0;
+		const wildcardAgent = explicitAgentId === "*";
+		const envAgentId = isolated ? getAgentId() : void 0;
+		const filterAgentId = wildcardAgent ? void 0 : explicitAgentId ?? envAgentId;
+		if (isolated && !wildcardAgent && !explicitAgentId && !envAgentId) throw new Error("mem::context: AGENTMEMORY_AGENT_SCOPE=isolated is set but no agent id is available (env AGENT_ID unset and no explicit agentId in the call). Refusing to read cross-agent rows. Pass agentId: \"*\" to opt in to a wildcard read.");
 		const [pinnedSlots, profile, lessons] = await Promise.all([
 			isSlotsEnabled() ? listPinnedSlots(kv).catch(() => []) : Promise.resolve([]),
 			kv.get(KV.profiles, data.project).catch(() => null),
@@ -4661,7 +4676,7 @@ function registerContextFunction(sdk, kv, tokenBudget) {
 				sourceIds: relevantLessons.map((l) => l.id)
 			});
 		}
-		const sessions = (await kv.list(KV.sessions)).filter((s) => s.project === data.project && s.id !== data.sessionId).sort((a, b) => new Date(b.startedAt).getTime() - new Date(a.startedAt).getTime()).slice(0, 10);
+		const sessions = (await kv.list(KV.sessions)).filter((s) => s.project === data.project && s.id !== data.sessionId && (filterAgentId === void 0 || s.agentId === filterAgentId)).sort((a, b) => new Date(b.startedAt).getTime() - new Date(a.startedAt).getTime()).slice(0, 10);
 		const summariesPerSession = await Promise.all(sessions.map((s) => kv.get(KV.summaries, s.id).catch(() => null)));
 		const sessionsNeedingObs = [];
 		for (let i = 0; i < sessions.length; i++) {
@@ -5651,7 +5666,7 @@ function registerRememberFunction(sdk, kv) {
 		const deletedMemoryIds = [];
 		const deletedObservationIds = [];
 		let deletedSession = false;
-		const { decrementImageRef } = await import("./image-refs-C7h9L5wx.mjs").then((n) => n.n);
+		const { decrementImageRef } = await import("./image-refs-C2YU7BIm.mjs");
 		if (data.memoryId) {
 			const mem = await kv.get(KV.memories, data.memoryId);
 			await kv.delete(KV.memories, data.memoryId);
@@ -5758,7 +5773,7 @@ async function runRecoveredSessionConsolidation(sdk) {
 function registerEvictFunction(sdk, kv) {
 	sdk.registerFunction("mem::evict", async (data) => {
 		const dryRun = data?.dryRun ?? false;
-		const { decrementImageRef } = await import("./image-refs-C7h9L5wx.mjs").then((n) => n.n);
+		const { decrementImageRef } = await import("./image-refs-C2YU7BIm.mjs");
 		const configOverride = await kv.get(KV.config, "eviction").catch(() => null);
 		const cfg = {
 			...DEFAULTS$1,
@@ -6278,7 +6293,10 @@ function registerSmartSearchFunction(sdk, kv, searchFn) {
 	sdk.registerFunction("mem::smart-search", async (data) => {
 		const isolated = isAgentScopeIsolated();
 		const explicitAgentId = typeof data.agentId === "string" && data.agentId.trim().length > 0 ? data.agentId.trim() : void 0;
-		const filterAgentId = explicitAgentId === "*" ? void 0 : explicitAgentId ?? (isolated ? getAgentId() : void 0);
+		const wildcardAgent = explicitAgentId === "*";
+		const envAgentId = isolated ? getAgentId() : void 0;
+		const filterAgentId = wildcardAgent ? void 0 : explicitAgentId ?? envAgentId;
+		if (isolated && !wildcardAgent && !explicitAgentId && !envAgentId) throw new Error("mem::smart-search: AGENTMEMORY_AGENT_SCOPE=isolated is set but no agent id is available (env AGENT_ID unset and no explicit agentId in the call). Refusing to read cross-agent rows. Pass agentId: \"*\" to opt in to a wildcard read.");
 		if (data.expandIds && data.expandIds.length > 0) {
 			const raw = data.expandIds.slice(0, 20);
 			const items = raw.map((entry) => {
@@ -6570,7 +6588,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-C7h9L5wx.mjs").then((n) => n.n);
+		const { decrementImageRef } = await import("./image-refs-C2YU7BIm.mjs");
 		const result = {
 			ttlExpired: [],
 			contradictions: [],
@@ -6829,7 +6847,9 @@ function registerExportImportFunction(sdk, kv) {
 			"0.9.23",
 			"0.9.24",
 			"0.9.25",
-			"0.9.26"
+			"0.9.26",
+			"0.9.27",
+			"0.9.28"
 		]).has(importData.version)) return {
 			success: false,
 			error: `Unsupported export version: ${importData.version}`
@@ -7353,19 +7373,167 @@ function buildGraphExtractionPrompt(observations) {
 //#region src/functions/graph.ts
 const DEFAULT_GRAPH_QUERY_LIMIT = 500;
 const MAX_GRAPH_QUERY_LIMIT = 5e3;
-function resolvePagination(rawLimit, rawOffset) {
+const SNAPSHOT_TOP_NODES = DEFAULT_GRAPH_QUERY_LIMIT;
+const SNAPSHOT_KEY = "current";
+const LIVE_ENUMERATION_BUDGET_MS = 6e3;
+function withTimeout(p, ms, label) {
+	return new Promise((resolve, reject) => {
+		const t = setTimeout(() => reject(/* @__PURE__ */ new Error(`${label}: exceeded ${ms}ms budget`)), ms);
+		p.then((v) => {
+			clearTimeout(t);
+			resolve(v);
+		}, (err) => {
+			clearTimeout(t);
+			reject(err);
+		});
+	});
+}
+function emptySnapshot() {
 	return {
-		limit: Math.max(1, Math.min(typeof rawLimit === "number" && Number.isFinite(rawLimit) ? Math.floor(rawLimit) : DEFAULT_GRAPH_QUERY_LIMIT, MAX_GRAPH_QUERY_LIMIT)),
-		offset: Math.max(0, typeof rawOffset === "number" && Number.isFinite(rawOffset) ? Math.floor(rawOffset) : 0)
+		version: 1,
+		topNodes: [],
+		topEdges: [],
+		topDegrees: {},
+		stats: {
+			totalNodes: 0,
+			totalEdges: 0,
+			nodesByType: {},
+			edgesByType: {}
+		},
+		updatedAt: (/* @__PURE__ */ new Date(0)).toISOString(),
+		dirty: true
 	};
 }
-function rankByDegree(nodes, edges) {
+async function readSnapshot(kv) {
+	try {
+		const snap = await kv.get(KV.graphSnapshot, SNAPSHOT_KEY);
+		if (snap && typeof snap === "object" && snap.version === 1) return snap;
+		return null;
+	} catch (err) {
+		logger.warn("Graph snapshot read failed", { error: err instanceof Error ? err.message : String(err) });
+		return null;
+	}
+}
+function buildSnapshotFromArrays(nodes, edges) {
+	const liveNodes = nodes.filter((n) => !n.stale);
+	const liveEdges = edges.filter((e) => !e.stale);
 	const degree = /* @__PURE__ */ new Map();
-	for (const edge of edges) {
-		degree.set(edge.sourceNodeId, (degree.get(edge.sourceNodeId) ?? 0) + 1);
-		degree.set(edge.targetNodeId, (degree.get(edge.targetNodeId) ?? 0) + 1);
+	for (const e of liveEdges) {
+		degree.set(e.sourceNodeId, (degree.get(e.sourceNodeId) ?? 0) + 1);
+		degree.set(e.targetNodeId, (degree.get(e.targetNodeId) ?? 0) + 1);
+	}
+	const ranked = [...liveNodes].sort((a, b) => (degree.get(b.id) ?? 0) - (degree.get(a.id) ?? 0)).slice(0, SNAPSHOT_TOP_NODES);
+	const rankedIds = new Set(ranked.map((n) => n.id));
+	const topEdges = liveEdges.filter((e) => rankedIds.has(e.sourceNodeId) && rankedIds.has(e.targetNodeId));
+	const topDegrees = {};
+	for (const n of ranked) topDegrees[n.id] = degree.get(n.id) ?? 0;
+	const nodesByType = {};
+	for (const n of liveNodes) nodesByType[n.type] = (nodesByType[n.type] || 0) + 1;
+	const edgesByType = {};
+	for (const e of liveEdges) edgesByType[e.type] = (edgesByType[e.type] || 0) + 1;
+	return {
+		version: 1,
+		topNodes: ranked,
+		topEdges,
+		topDegrees,
+		stats: {
+			totalNodes: liveNodes.length,
+			totalEdges: liveEdges.length,
+			nodesByType,
+			edgesByType
+		},
+		updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+		dirty: false
+	};
+}
+function paginateFromSnapshot(snap, filterType, limit, offset) {
+	const filteredNodes = filterType ? snap.topNodes.filter((n) => n.type === filterType) : snap.topNodes;
+	const total = filterType ? snap.stats.nodesByType[filterType] ?? 0 : snap.stats.totalNodes;
+	const pageNodes = filteredNodes.slice(offset, offset + limit);
+	const pageIds = new Set(pageNodes.map((n) => n.id));
+	return {
+		nodes: pageNodes,
+		edges: snap.topEdges.filter((e) => pageIds.has(e.sourceNodeId) && pageIds.has(e.targetNodeId)),
+		depth: 0,
+		totalNodes: total,
+		totalEdges: snap.stats.totalEdges,
+		truncated: total > pageNodes.length,
+		limit,
+		offset,
+		fromSnapshot: true
+	};
+}
+const REBUILD_SAFE_NODE_CEILING = 25e3;
+function nameIndexKey(type, name) {
+	return `${type}|${name}`;
+}
+function edgeIndexKey(sourceNodeId, targetNodeId, type) {
+	return `${sourceNodeId}|${targetNodeId}|${type}`;
+}
+async function applyDegreeDelta(kv, snap, nodeId, delta) {
+	const prev = await kv.get(KV.graphNodeDegree, nodeId) ?? 0;
+	const next = Math.max(0, prev + delta);
+	await kv.set(KV.graphNodeDegree, nodeId, next);
+	if (snap.topNodes.findIndex((n) => n.id === nodeId) !== -1) {
+		snap.topDegrees[nodeId] = next;
+		snap.topNodes.sort((a, b) => (snap.topDegrees[b.id] ?? 0) - (snap.topDegrees[a.id] ?? 0));
+		return next;
+	}
+	if (snap.topNodes.length < SNAPSHOT_TOP_NODES) {
+		const node = await kv.get(KV.graphNodes, nodeId);
+		if (node && !node.stale) {
+			snap.topNodes.push(node);
+			snap.topDegrees[node.id] = next;
+			snap.topNodes.sort((a, b) => (snap.topDegrees[b.id] ?? 0) - (snap.topDegrees[a.id] ?? 0));
+		}
+		return next;
 	}
-	return [...nodes].sort((a, b) => (degree.get(b.id) ?? 0) - (degree.get(a.id) ?? 0));
+	const tailEntry = snap.topNodes[snap.topNodes.length - 1];
+	if (!tailEntry) return next;
+	if (next > (snap.topDegrees[tailEntry.id] ?? 0)) {
+		const node = await kv.get(KV.graphNodes, nodeId);
+		if (node && !node.stale) {
+			const evicted = snap.topNodes.pop();
+			if (evicted) delete snap.topDegrees[evicted.id];
+			snap.topNodes.push(node);
+			snap.topDegrees[node.id] = next;
+			snap.topNodes.sort((a, b) => (snap.topDegrees[b.id] ?? 0) - (snap.topDegrees[a.id] ?? 0));
+		}
+	}
+	return next;
+}
+function snapshotPushEdgeIfBothInTop(snap, edge) {
+	const topIds = new Set(snap.topNodes.map((n) => n.id));
+	if (topIds.has(edge.sourceNodeId) && topIds.has(edge.targetNodeId)) {
+		if (!snap.topEdges.find((e) => e.id === edge.id)) snap.topEdges.push(edge);
+	}
+}
+function mergeNode(existing, incoming, obsIds, capturedAt) {
+	return {
+		...existing,
+		sourceObservationIds: [...new Set([
+			...existing.sourceObservationIds,
+			...incoming.sourceObservationIds,
+			...obsIds
+		])],
+		properties: {
+			...existing.properties,
+			...incoming.properties
+		},
+		updatedAt: capturedAt
+	};
+}
+function mergeEdge(existing, obsIds) {
+	return {
+		...existing,
+		sourceObservationIds: [...new Set([...existing.sourceObservationIds, ...obsIds])]
+	};
+}
+function resolvePagination(rawLimit, rawOffset) {
+	return {
+		limit: Math.max(1, Math.min(typeof rawLimit === "number" && Number.isFinite(rawLimit) ? Math.floor(rawLimit) : DEFAULT_GRAPH_QUERY_LIMIT, MAX_GRAPH_QUERY_LIMIT)),
+		offset: Math.max(0, typeof rawOffset === "number" && Number.isFinite(rawOffset) ? Math.floor(rawOffset) : 0)
+	};
 }
 function paginate(nodes, allEdges, depth, limit, offset) {
 	const totalNodes = nodes.length;
@@ -7462,45 +7630,76 @@ function registerGraphFunction(sdk, kv, provider) {
 			const response = await provider.compress(GRAPH_EXTRACTION_SYSTEM, prompt);
 			const obsIds = data.observations.map((o) => o.id);
 			const { nodes, edges } = parseGraphXml(response, obsIds);
-			const existingNodes = await kv.list(KV.graphNodes);
-			const existingEdges = await kv.list(KV.graphEdges);
+			const snap = await readSnapshot(kv) ?? emptySnapshot();
+			const capturedAt = (/* @__PURE__ */ new Date()).toISOString();
+			let newNodeCount = 0;
+			let newEdgeCount = 0;
+			const newEdgesForTopCheck = [];
 			for (const node of nodes) {
-				const existing = existingNodes.find((n) => n.name === node.name && n.type === node.type);
+				const indexKey = nameIndexKey(node.type, node.name);
+				const existingId = await kv.get(KV.graphNameIndex, indexKey);
+				let existing = null;
+				if (existingId) {
+					existing = await kv.get(KV.graphNodes, existingId);
+					if (existing && snap.resetAt && typeof existing.createdAt === "string" && existing.createdAt < snap.resetAt) existing = null;
+				}
 				if (existing) {
-					const merged = {
-						...existing,
-						sourceObservationIds: [...new Set([...existing.sourceObservationIds, ...obsIds])],
-						properties: {
-							...existing.properties,
-							...node.properties
-						}
-					};
+					const merged = mergeNode(existing, node, obsIds, capturedAt);
 					await kv.set(KV.graphNodes, existing.id, merged);
-					const idx = existingNodes.findIndex((n) => n.id === existing.id);
-					if (idx !== -1) existingNodes[idx] = merged;
+					const topIdx = snap.topNodes.findIndex((n) => n.id === existing.id);
+					if (topIdx !== -1) snap.topNodes[topIdx] = merged;
 				} else {
 					await kv.set(KV.graphNodes, node.id, node);
-					existingNodes.push(node);
+					await kv.set(KV.graphNameIndex, indexKey, node.id);
+					await kv.set(KV.graphNodeDegree, node.id, 0);
+					snap.stats.totalNodes += 1;
+					snap.stats.nodesByType[node.type] = (snap.stats.nodesByType[node.type] ?? 0) + 1;
+					newNodeCount += 1;
+					if (snap.topNodes.length < SNAPSHOT_TOP_NODES) {
+						snap.topNodes.push(node);
+						snap.topDegrees[node.id] = 0;
+					}
 				}
 			}
 			for (const edge of edges) {
-				const edgeKey = `${edge.sourceNodeId}|${edge.targetNodeId}|${edge.type}`;
-				const existingEdge = existingEdges.find((e) => `${e.sourceNodeId}|${e.targetNodeId}|${e.type}` === edgeKey);
-				if (existingEdge) {
-					existingEdge.sourceObservationIds = [...new Set([...existingEdge.sourceObservationIds, ...obsIds])];
-					await kv.set(KV.graphEdges, existingEdge.id, existingEdge);
+				const eKey = edgeIndexKey(edge.sourceNodeId, edge.targetNodeId, edge.type);
+				const existingId = await kv.get(KV.graphEdgeKey, eKey);
+				let existing = null;
+				if (existingId) {
+					existing = await kv.get(KV.graphEdges, existingId);
+					if (existing && snap.resetAt && typeof existing.createdAt === "string" && existing.createdAt < snap.resetAt) existing = null;
+				}
+				if (existing) {
+					const merged = mergeEdge(existing, obsIds);
+					await kv.set(KV.graphEdges, existing.id, merged);
+					const topIdx = snap.topEdges.findIndex((e) => e.id === existing.id);
+					if (topIdx !== -1) snap.topEdges[topIdx] = merged;
 				} else {
 					await kv.set(KV.graphEdges, edge.id, edge);
-					existingEdges.push(edge);
+					await kv.set(KV.graphEdgeKey, eKey, edge.id);
+					snap.stats.totalEdges += 1;
+					snap.stats.edgesByType[edge.type] = (snap.stats.edgesByType[edge.type] ?? 0) + 1;
+					newEdgeCount += 1;
+					await applyDegreeDelta(kv, snap, edge.sourceNodeId, 1);
+					await applyDegreeDelta(kv, snap, edge.targetNodeId, 1);
+					newEdgesForTopCheck.push(edge);
 				}
 			}
+			for (const edge of newEdgesForTopCheck) snapshotPushEdgeIfBothInTop(snap, edge);
+			if (newNodeCount > 0 || newEdgeCount > 0) {
+				snap.updatedAt = capturedAt;
+				snap.dirty = false;
+				await kv.set(KV.graphSnapshot, SNAPSHOT_KEY, snap);
+			}
 			await recordAudit(kv, "observe", "mem::graph-extract", obsIds, {
 				nodesExtracted: nodes.length,
 				edgesExtracted: edges.length
 			});
 			logger.info("Graph extraction complete", {
 				nodes: nodes.length,
-				edges: edges.length
+				edges: edges.length,
+				newNodes: newNodeCount,
+				newEdges: newEdgeCount
 			});
 			return {
 				success: true,
@@ -7517,10 +7716,49 @@ function registerGraphFunction(sdk, kv, provider) {
 		}
 	});
 	sdk.registerFunction("mem::graph-query", async (data) => {
-		const allNodes = (await kv.list(KV.graphNodes)).filter((n) => !n.stale);
-		const allEdges = (await kv.list(KV.graphEdges)).filter((e) => !e.stale);
 		const maxDepth = Math.min(data.maxDepth || 3, 5);
 		const { limit, offset } = resolvePagination(data.limit, data.offset);
+		if (!data.query && !data.startNodeId) {
+			const snap = await readSnapshot(kv);
+			if (snap && snap.stats.totalNodes > 0) return paginateFromSnapshot(snap, data.nodeType, limit, offset);
+			return {
+				nodes: [],
+				edges: [],
+				depth: 0,
+				totalNodes: 0,
+				totalEdges: 0,
+				truncated: false,
+				limit,
+				offset,
+				warning: "No graph snapshot available. Either no graph has been extracted yet, or you are on a legacy corpus from a pre-#814 agentmemory build. Run POST /agentmemory/graph/snapshot-rebuild (safe up to ~25K nodes) or POST /agentmemory/graph/reset to wipe and let future extracts repopulate."
+			};
+		}
+		let allNodes;
+		let allEdges;
+		try {
+			const [rawNodes, rawEdges] = await withTimeout(Promise.all([kv.list(KV.graphNodes), kv.list(KV.graphEdges)]), LIVE_ENUMERATION_BUDGET_MS, "graph-query enumeration");
+			allNodes = rawNodes.filter((n) => !n.stale);
+			allEdges = rawEdges.filter((e) => !e.stale);
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : String(err);
+			logger.warn("Graph query enumeration timed out, using snapshot", { error: msg });
+			const snap = await readSnapshot(kv);
+			if (snap) return {
+				...paginateFromSnapshot(snap, data.nodeType, limit, offset),
+				warning: "Live graph enumeration exceeded budget. Query / startNodeId paths degrade on >25K-node corpora until a per-node edge index lands. Result reflects top-degree snapshot, not the requested walk."
+			};
+			return {
+				nodes: [],
+				edges: [],
+				depth: 0,
+				totalNodes: 0,
+				totalEdges: 0,
+				truncated: false,
+				limit,
+				offset,
+				warning: "Graph enumeration exceeded budget and no snapshot is available."
+			};
+		}
 		if (data.query) {
 			const lower = data.query.toLowerCase();
 			return paginate(allNodes.filter((n) => n.name.toLowerCase().includes(lower) || Object.values(n.properties).some((v) => typeof v === "string" && v.toLowerCase().includes(lower))), allEdges, 0, limit, offset);
@@ -7557,22 +7795,116 @@ function registerGraphFunction(sdk, kv, provider) {
 			}
 			return paginate(resultNodes, resultEdges, maxDepth, limit, offset);
 		}
-		let filtered = allNodes;
-		if (data.nodeType) filtered = allNodes.filter((n) => n.type === data.nodeType);
-		return paginate(rankByDegree(filtered, allEdges), allEdges, 0, limit, offset);
+		return paginate([], [], 0, limit, offset);
 	});
 	sdk.registerFunction("mem::graph-stats", async () => {
-		const nodes = await kv.list(KV.graphNodes);
-		const edges = await kv.list(KV.graphEdges);
-		const nodesByType = {};
-		for (const n of nodes) nodesByType[n.type] = (nodesByType[n.type] || 0) + 1;
-		const edgesByType = {};
-		for (const e of edges) edgesByType[e.type] = (edgesByType[e.type] || 0) + 1;
+		const snap = await readSnapshot(kv);
+		if (snap) return {
+			...snap.stats,
+			fromSnapshot: true,
+			updatedAt: snap.updatedAt,
+			...snap.dirty ? { warning: "Snapshot is marked dirty (write was in-flight when read). Counts are eventually consistent." } : {}
+		};
 		return {
-			totalNodes: nodes.length,
-			totalEdges: edges.length,
-			nodesByType,
-			edgesByType
+			totalNodes: 0,
+			totalEdges: 0,
+			nodesByType: {},
+			edgesByType: {},
+			fromSnapshot: false,
+			warning: "No graph snapshot available. Run POST /agentmemory/graph/snapshot-rebuild (safe up to ~25K nodes) or POST /agentmemory/graph/reset to wipe and let future extracts repopulate."
+		};
+	});
+	sdk.registerFunction("mem::graph-snapshot-rebuild", async (data) => {
+		const started = Date.now();
+		const forceRebuild = data?.force === true;
+		try {
+			if (!await readSnapshot(kv) && !forceRebuild) {
+				logger.warn("Graph snapshot rebuild refused: no prior snapshot", { hint: "legacy corpus or empty store" });
+				return {
+					success: false,
+					legacyCorpus: true,
+					error: "No prior snapshot found. Rebuild would call kv.list on KV.graphNodes/Edges, which heartbeat-crashes the worker on corpora past the iii state response budget (~25K nodes). Either (a) call POST /agentmemory/graph/reset to drop into incremental-only mode and rebuild from new extracts, or (b) re-send with `force: true` if you're certain the corpus is small."
+				};
+			}
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : String(err);
+			logger.warn("Graph snapshot pre-flight read failed", { error: msg });
+		}
+		try {
+			const [nodes, edges] = await withTimeout(Promise.all([kv.list(KV.graphNodes), kv.list(KV.graphEdges)]), LIVE_ENUMERATION_BUDGET_MS, "graph-snapshot-rebuild enumeration");
+			if (nodes.length > REBUILD_SAFE_NODE_CEILING) {
+				logger.warn("Graph snapshot rebuild aborted: corpus too large", {
+					totalNodes: nodes.length,
+					ceiling: REBUILD_SAFE_NODE_CEILING
+				});
+				return {
+					success: false,
+					tooLarge: true,
+					totalNodes: nodes.length,
+					ceiling: REBUILD_SAFE_NODE_CEILING,
+					error: `Corpus has ${nodes.length} graph nodes; safe-rebuild ceiling is ${REBUILD_SAFE_NODE_CEILING}. Run POST /agentmemory/graph/reset to wipe and let future extracts rebuild incrementally.`
+				};
+			}
+			const liveNodes = nodes.filter((n) => !n.stale);
+			const liveEdges = edges.filter((e) => !e.stale);
+			const degree = /* @__PURE__ */ new Map();
+			for (const e of liveEdges) {
+				degree.set(e.sourceNodeId, (degree.get(e.sourceNodeId) ?? 0) + 1);
+				degree.set(e.targetNodeId, (degree.get(e.targetNodeId) ?? 0) + 1);
+			}
+			const BATCH_SIZE = 100;
+			for (let i = 0; i < liveNodes.length; i += BATCH_SIZE) {
+				const batch = liveNodes.slice(i, i + BATCH_SIZE);
+				await Promise.all(batch.flatMap((n) => [kv.set(KV.graphNameIndex, nameIndexKey(n.type, n.name), n.id), kv.set(KV.graphNodeDegree, n.id, degree.get(n.id) ?? 0)]));
+			}
+			for (let i = 0; i < liveEdges.length; i += BATCH_SIZE) {
+				const batch = liveEdges.slice(i, i + BATCH_SIZE);
+				await Promise.all(batch.map((e) => kv.set(KV.graphEdgeKey, edgeIndexKey(e.sourceNodeId, e.targetNodeId, e.type), e.id)));
+			}
+			const snap = buildSnapshotFromArrays(nodes, edges);
+			await kv.set(KV.graphSnapshot, SNAPSHOT_KEY, snap);
+			const tookMs = Date.now() - started;
+			logger.info("Graph snapshot rebuilt", {
+				totalNodes: snap.stats.totalNodes,
+				totalEdges: snap.stats.totalEdges,
+				topNodes: snap.topNodes.length,
+				topEdges: snap.topEdges.length,
+				tookMs
+			});
+			return {
+				success: true,
+				...snap.stats,
+				topNodes: snap.topNodes.length,
+				topEdges: snap.topEdges.length,
+				updatedAt: snap.updatedAt,
+				tookMs
+			};
+		} catch (err) {
+			const msg = err instanceof Error ? err.message : String(err);
+			logger.error("Graph snapshot rebuild failed", { error: msg });
+			return {
+				success: false,
+				error: msg
+			};
+		}
+	});
+	sdk.registerFunction("mem::graph-reset", async () => {
+		const started = Date.now();
+		const resetSnapshot = {
+			...emptySnapshot(),
+			resetAt: (/* @__PURE__ */ new Date()).toISOString()
+		};
+		await kv.set(KV.graphSnapshot, SNAPSHOT_KEY, resetSnapshot);
+		const counts = { [KV.graphSnapshot]: 1 };
+		const tookMs = Date.now() - started;
+		logger.info("Graph state reset", {
+			counts,
+			tookMs
+		});
+		return {
+			success: true,
+			cleared: counts,
+			tookMs
 		};
 	});
 }
@@ -13395,7 +13727,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-C7h9L5wx.mjs").then((n) => n.n);
+		const { decrementImageRef } = await import("./image-refs-C2YU7BIm.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,
@@ -14863,11 +15195,15 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 			status_code: 400,
 			body: { error: "budget must be a positive integer" }
 		};
+		const queryAgentId = typeof req.query_params?.["agentId"] === "string" ? req.query_params["agentId"] : void 0;
+		const bodyAgentId = typeof body.agentId === "string" && body.agentId.trim().length > 0 ? body.agentId.trim() : void 0;
 		const payload = {
 			sessionId,
 			project
 		};
 		if (budget !== void 0) payload.budget = budget;
+		const agentId = bodyAgentId ?? queryAgentId;
+		if (agentId !== void 0) payload.agentId = agentId;
 		return {
 			status_code: 200,
 			body: await sdk.trigger({
@@ -14887,6 +15223,7 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 	});
 	sdk.registerFunction("api::search", async (req) => {
 		const body = req.body ?? {};
+		const queryAgentId = typeof req.query_params?.["agentId"] === "string" ? req.query_params["agentId"] : void 0;
 		if (typeof body.query !== "string" || !body.query.trim()) return {
 			status_code: 400,
 			body: { error: "query is required and must be a non-empty string" }
@@ -14915,13 +15252,15 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 			status_code: 400,
 			body: { error: "token_budget must be a positive integer" }
 		};
+		const bodyAgentId = typeof body.agentId === "string" && body.agentId.trim().length > 0 ? body.agentId.trim() : void 0;
 		const payload = {
 			query: body.query.trim(),
 			limit: body.limit,
 			project: body.project,
 			cwd: body.cwd,
 			format: typeof body.format === "string" ? body.format.trim().toLowerCase() : void 0,
-			token_budget: body.token_budget
+			token_budget: body.token_budget,
+			agentId: bodyAgentId ?? queryAgentId
 		};
 		return {
 			status_code: 200,
@@ -15076,7 +15415,8 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 					function_id: "mem::context",
 					payload: {
 						sessionId,
-						project
+						project,
+						...agentId ? { agentId } : {}
 					}
 				})).context
 			}
@@ -15268,9 +15608,14 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 		const normalizedAgentId = typeof req.query_params?.["agentId"] === "string" ? req.query_params["agentId"].trim() : void 0;
 		const wildcardAgent = normalizedAgentId === "*";
 		const filterAgentId = wildcardAgent ? void 0 : (normalizedAgentId && !wildcardAgent ? normalizedAgentId : void 0) ?? (isAgentScopeIsolated() ? getAgentId() : void 0);
+		const filtered = filterAgentId ? sessions.filter((s) => s.agentId === filterAgentId) : sessions;
+		const summaries = await Promise.all(filtered.map((s) => kv.get(KV.summaries, s.id).catch(() => null)));
 		return {
 			status_code: 200,
-			body: { sessions: filterAgentId ? sessions.filter((s) => s.agentId === filterAgentId) : sessions }
+			body: { sessions: filtered.map((s, i) => summaries[i] ? {
+				...s,
+				summary: summaries[i]
+			} : s) }
 		};
 	});
 	sdk.registerTrigger({
@@ -15857,6 +16202,52 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 			http_method: "GET"
 		}
 	});
+	sdk.registerFunction("api::graph-snapshot-rebuild", async (req) => {
+		const authErr = checkAuth(req, secret);
+		if (authErr) return authErr;
+		try {
+			return {
+				status_code: 200,
+				body: await sdk.trigger({
+					function_id: "mem::graph-snapshot-rebuild",
+					payload: {}
+				})
+			};
+		} catch {
+			return graphDisabledResponse();
+		}
+	});
+	sdk.registerTrigger({
+		type: "http",
+		function_id: "api::graph-snapshot-rebuild",
+		config: {
+			api_path: "/agentmemory/graph/snapshot-rebuild",
+			http_method: "POST"
+		}
+	});
+	sdk.registerFunction("api::graph-reset", async (req) => {
+		const authErr = checkAuth(req, secret);
+		if (authErr) return authErr;
+		try {
+			return {
+				status_code: 200,
+				body: await sdk.trigger({
+					function_id: "mem::graph-reset",
+					payload: {}
+				})
+			};
+		} catch {
+			return graphDisabledResponse();
+		}
+	});
+	sdk.registerTrigger({
+		type: "http",
+		function_id: "api::graph-reset",
+		config: {
+			api_path: "/agentmemory/graph/reset",
+			http_method: "POST"
+		}
+	});
 	sdk.registerFunction("api::graph-extract", async (req) => {
 		const authErr = checkAuth(req, secret);
 		if (authErr) return authErr;
@@ -18104,13 +18495,15 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
 //#region src/triggers/events.ts
 function registerEventTriggers(sdk, kv) {
 	sdk.registerFunction("event::session::started", async (data) => {
+		const agentId = (typeof data.agentId === "string" && data.agentId.trim().length > 0 ? data.agentId.trim().slice(0, 128) : void 0) ?? getAgentId();
 		const session = {
 			id: data.sessionId,
 			project: data.project,
 			cwd: data.cwd,
 			startedAt: (/* @__PURE__ */ new Date()).toISOString(),
 			status: "active",
-			observationCount: 0
+			observationCount: 0,
+			...agentId ? { agentId } : {}
 		};
 		await kv.set(KV.sessions, data.sessionId, session);
 		return {
@@ -18119,7 +18512,8 @@ function registerEventTriggers(sdk, kv) {
 				function_id: "mem::context",
 				payload: {
 					sessionId: data.sessionId,
-					project: data.project
+					project: data.project,
+					...agentId ? { agentId } : {}
 				}
 			})).context
 		};
@@ -18291,13 +18685,15 @@ function registerMcpEndpoints(sdk, kv, secret) {
 						status_code: 400,
 						body: { error: "token_budget must be a positive integer" }
 					};
+					const recallAgentId = typeof args.agentId === "string" && args.agentId.trim().length > 0 ? args.agentId.trim() : void 0;
 					const result = await sdk.trigger({
 						function_id: "mem::search",
 						payload: {
 							query: args.query,
 							limit: typeof args.limit === "number" ? args.limit : 10,
 							format,
-							token_budget: tokenBudget
+							token_budget: tokenBudget,
+							agentId: recallAgentId
 						}
 					});
 					return {
@@ -19785,7 +20181,10 @@ function registerMcpEndpoints(sdk, kv, secret) {
 							limit: 10
 						}
 					}).catch(() => ({ results: [] }));
-					const relevant = (await kv.list(KV.memories)).filter((m) => m.isLatest).slice(0, 5);
+					const memories = await kv.list(KV.memories);
+					const isolated = isAgentScopeIsolated();
+					const activeAgentId = isolated ? getAgentId() : void 0;
+					const relevant = isolated && activeAgentId === void 0 ? [] : memories.filter((m) => m.isLatest).filter((m) => activeAgentId === void 0 || m.agentId === activeAgentId).slice(0, 5);
 					return {
 						status_code: 200,
 						body: { messages: [{
@@ -20039,10 +20438,10 @@ async function main() {
 	}
 	registerConsolidationPipelineFunction(sdk, kv, provider);
 	bootLog(`Consolidation pipeline: registered (CONSOLIDATION_ENABLED=${isConsolidationEnabled() ? "true" : "false"})`);
-	if (isAutoCompressEnabled()) bootLog(`WARNING: AGENTMEMORY_AUTO_COMPRESS=true — every PostToolUse observation will be sent to your LLM provider for compression. This spends API tokens proportional to your session tool-use frequency (see #138). Set AGENTMEMORY_AUTO_COMPRESS=false to disable.`);
-	else bootLog(`Auto-compress: OFF (default, #138) — observations indexed via zero-LLM synthetic compression. Set AGENTMEMORY_AUTO_COMPRESS=true to opt-in to LLM-powered summaries (uses your API key).`);
-	if (isContextInjectionEnabled()) bootLog(`WARNING: AGENTMEMORY_INJECT_CONTEXT=true — the PreToolUse and SessionStart hooks will inject up to ~4000 chars of memory context into every tool turn. On Claude Pro this burns session tokens proportional to your tool-call frequency (see #143). Set AGENTMEMORY_INJECT_CONTEXT=false to disable.`);
-	else bootLog(`Context injection: OFF (default, #143) — hooks capture observations but do not inject context into Claude Code's conversation. Set AGENTMEMORY_INJECT_CONTEXT=true to opt-in (warning: expect your Claude Pro allocation to drain faster).`);
+	if (isAutoCompressEnabled()) bootLog(`WARNING: AGENTMEMORY_AUTO_COMPRESS=true — every PostToolUse observation will be sent to your LLM provider for compression. This spends API tokens proportional to your session tool-use frequency. Set AGENTMEMORY_AUTO_COMPRESS=false to disable.`);
+	else bootLog(`Auto-compress: OFF (default) — observations indexed via zero-LLM synthetic compression. Set AGENTMEMORY_AUTO_COMPRESS=true to opt-in to LLM-powered summaries (uses your API key).`);
+	if (isContextInjectionEnabled()) bootLog(`WARNING: AGENTMEMORY_INJECT_CONTEXT=true — the PreToolUse and SessionStart hooks will inject up to ~4000 chars of memory context into every tool turn. On Claude Pro this burns session tokens proportional to your tool-call frequency. Set AGENTMEMORY_INJECT_CONTEXT=false to disable.`);
+	else bootLog(`Context injection: OFF (default) — hooks capture observations but do not inject context into Claude Code's conversation. Set AGENTMEMORY_INJECT_CONTEXT=true to opt-in (warning: expect your Claude Pro allocation to drain faster).`);
 	const teamConfig = loadTeamConfig();
 	if (teamConfig) {
 		registerTeamFunction(sdk, kv, teamConfig);
@@ -20149,14 +20548,14 @@ async function main() {
 			backfilled++;
 		}
 		if (backfilled > 0) {
-			bootLog(`Backfilled ${backfilled} memories into BM25 (legacy gap before #257)`);
+			bootLog(`Backfilled ${backfilled} memories into BM25 (legacy index gap)`);
 			indexPersistence.scheduleSave();
 		}
 	} catch (err) {
 		console.warn(`[agentmemory] Failed to backfill memories into BM25:`, err);
 	}
 	bootLog(`Ready. ${embeddingProvider ? "Triple-stream (BM25+Vector+Graph)" : "BM25+Graph"} search active.`);
-	bootLog(`REST API: 126 endpoints at http://localhost:${config.restPort}/agentmemory/*`);
+	bootLog(`REST API: 128 endpoints at http://localhost:${config.restPort}/agentmemory/*`);
 	bootLog(`MCP surface (opt-in via \`npx @agentmemory/mcp\`): ${getAllTools().length} tools · 6 resources · 3 prompts`);
 	const viewerServer = startViewerServer(config.restPort + 2, kv, sdk, secret, config.restPort);
 	const autoForgetIntervalMs = parseInt(process.env.AUTO_FORGET_INTERVAL_MS || "3600000", 10);
@@ -20233,4 +20632,4 @@ main().catch((err) => {
 //#endregion
 export {};
 
-//# sourceMappingURL=src-u7kAEUC0.mjs.map
+//# sourceMappingURL=src-CzgoepGU.mjs.map