@agentmemory/agentmemory 0.9.10 → 0.9.12

package/dist/{src-d1vwmuPU.mjs → src-Cqsy23f_.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 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-C04WM1dS.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-NUso4hxA.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";
@@ -2054,7 +2054,7 @@ var SearchIndex = class SearchIndex {
 		return this.tokenize(parts.join(" ").toLowerCase());
 	}
 	tokenize(text) {
-		return text.replace(/[^\w\s/.\-_]/g, " ").split(/\s+/).filter((t) => t.length > 1).map((t) => stem(t));
+		return text.replace(/[^\p{L}\p{N}\s/.\\-_]/gu, " ").split(/\s+/).filter((t) => t.length > 1).map((t) => stem(t));
 	}
 	getSortedTerms() {
 		if (!this.sortedTerms) this.sortedTerms = Array.from(this.invertedIndex.keys()).sort();
@@ -2340,13 +2340,55 @@ async function deleteAccessLog(kv, memoryId) {
 //#endregion
 //#region src/functions/search.ts
 let index = null;
+let vectorIndex = null;
+let currentEmbeddingProvider = null;
 function getSearchIndex() {
 	if (!index) index = new SearchIndex();
 	return index;
 }
+function setVectorIndex(idx) {
+	vectorIndex = idx;
+}
+function setEmbeddingProvider(provider) {
+	currentEmbeddingProvider = provider;
+}
+const EMBED_MAX_CHARS = 16e3;
+function clipEmbedInput(text) {
+	if (text.length <= EMBED_MAX_CHARS) return text;
+	return text.slice(0, EMBED_MAX_CHARS);
+}
+async function vectorIndexAddGuarded(id, sessionId, text, context) {
+	const vi = vectorIndex;
+	const ep = currentEmbeddingProvider;
+	if (!vi || !ep) return false;
+	try {
+		const embedding = await ep.embed(clipEmbedInput(text));
+		if (embedding.length !== ep.dimensions) {
+			logger.warn("vector-index add: dimension mismatch — skipping", {
+				kind: context.kind,
+				id: context.logId,
+				provider: ep.name,
+				expected: ep.dimensions,
+				received: embedding.length
+			});
+			return false;
+		}
+		vi.add(id, sessionId, embedding);
+		return true;
+	} catch (err) {
+		logger.warn("vector-index add: embed failed — skipping", {
+			kind: context.kind,
+			id: context.logId,
+			provider: ep.name,
+			error: err instanceof Error ? err.message : String(err)
+		});
+		return false;
+	}
+}
 async function rebuildIndex(kv) {
 	const idx = getSearchIndex();
 	idx.clear();
+	vectorIndex?.clear();
 	let count = 0;
 	try {
 		const memories = await kv.list(KV.memories);
@@ -2354,6 +2396,10 @@ async function rebuildIndex(kv) {
 			if (memory.isLatest === false) continue;
 			if (!memory.title || !memory.content) continue;
 			idx.add(memoryToObservation(memory));
+			await vectorIndexAddGuarded(memory.id, memory.sessionIds[0] ?? "memory", memory.title + " " + memory.content, {
+				kind: "memory",
+				logId: memory.id
+			});
 			count++;
 		}
 	} catch (err) {
@@ -2378,6 +2424,10 @@ async function rebuildIndex(kv) {
 	if (failedSessions.length > 0) logger.warn("rebuildIndex: failed to load observations for sessions", { failedSessions });
 	for (const observations of obsPerSession) for (const obs of observations) if (obs.title && obs.narrative) {
 		idx.add(obs);
+		await vectorIndexAddGuarded(obs.id, obs.sessionId, obs.title + " " + obs.narrative, {
+			kind: "observation",
+			logId: obs.id
+		});
 		count++;
 	}
 	return count;
@@ -2600,10 +2650,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-BmdmQ4tI.mjs");
+				const { saveImageToDisk } = await import("./image-store-RY_BkYWK.mjs");
 				const { filePath, bytesWritten } = await saveImageToDisk(pendingImageData);
 				raw.imageData = filePath;
-				const { incrementImageRef } = await import("./image-refs-BIABW8uB.mjs");
+				const { incrementImageRef } = await import("./image-refs-BkMGNmEs.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", {
@@ -2616,7 +2666,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-BmdmQ4tI.mjs");
+					const { deleteImage } = await import("./image-store-RY_BkYWK.mjs");
 					const { deletedBytes } = await deleteImage(raw.imageData);
 					if (deletedBytes > 0) sdk.triggerVoid("mem::disk-size-delta", { deltaBytes: -deletedBytes });
 				}
@@ -2684,6 +2734,10 @@ function registerObserveFunction(sdk, kv, dedupMap, maxObservationsPerSession) {
 				const synthetic = buildSyntheticCompression(raw);
 				await kv.set(KV.observations(payload.sessionId), obsId, synthetic);
 				getSearchIndex().add(synthetic);
+				await vectorIndexAddGuarded(synthetic.id, synthetic.sessionId, synthetic.title + " " + (synthetic.narrative || ""), {
+					kind: "synthetic",
+					logId: synthetic.id
+				});
 				await sdk.trigger({
 					function_id: "stream::set",
 					payload: {
@@ -3821,7 +3875,20 @@ function registerCompressFunction(sdk, kv, provider, metricsStore) {
 				...data.raw.imageData ? { imageRef: data.raw.imageData } : {}
 			};
 			await kv.set(KV.observations(data.sessionId), data.observationId, compressed);
-			getSearchIndex().add(compressed);
+			try {
+				getSearchIndex().add(compressed);
+			} catch (err) {
+				logger.warn("Failed to index compressed observation into BM25", {
+					obsId: compressed.id,
+					sessionId: compressed.sessionId,
+					title: compressed.title,
+					error: err instanceof Error ? err.message : String(err)
+				});
+			}
+			await vectorIndexAddGuarded(compressed.id, compressed.sessionId, compressed.title + " " + (compressed.narrative || ""), {
+				kind: "observation",
+				logId: compressed.id
+			});
 			const streamResults = await Promise.allSettled([sdk.trigger({
 				function_id: "stream::set",
 				payload: {
@@ -4661,6 +4728,10 @@ function registerRememberFunction(sdk, kv) {
 					error: err instanceof Error ? err.message : String(err)
 				});
 			}
+			await vectorIndexAddGuarded(memory.id, memory.sessionIds[0] ?? "memory", memory.title + " " + memory.content, {
+				kind: "memory",
+				logId: memory.id
+			});
 			if (supersededId) await sdk.trigger({
 				function_id: "mem::cascade-update",
 				payload: { supersededMemoryId: supersededId },
@@ -4681,7 +4752,7 @@ function registerRememberFunction(sdk, kv) {
 		const deletedMemoryIds = [];
 		const deletedObservationIds = [];
 		let deletedSession = false;
-		const { decrementImageRef } = await import("./image-refs-BIABW8uB.mjs");
+		const { decrementImageRef } = await import("./image-refs-BkMGNmEs.mjs");
 		if (data.memoryId) {
 			const mem = await kv.get(KV.memories, data.memoryId);
 			await kv.delete(KV.memories, data.memoryId);
@@ -4740,7 +4811,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-BIABW8uB.mjs");
+		const { decrementImageRef } = await import("./image-refs-BkMGNmEs.mjs");
 		const configOverride = await kv.get(KV.config, "eviction").catch(() => null);
 		const cfg = {
 			...DEFAULTS$1,
@@ -5343,7 +5414,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-BIABW8uB.mjs");
+		const { decrementImageRef } = await import("./image-refs-BkMGNmEs.mjs");
 		const result = {
 			ttlExpired: [],
 			contradictions: [],
@@ -5582,7 +5653,9 @@ function registerExportImportFunction(sdk, kv) {
 			"0.9.7",
 			"0.9.8",
 			"0.9.9",
-			"0.9.10"
+			"0.9.10",
+			"0.9.11",
+			"0.9.12"
 		]).has(importData.version)) return {
 			success: false,
 			error: `Unsupported export version: ${importData.version}`
@@ -11939,7 +12012,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-BIABW8uB.mjs");
+		const { decrementImageRef } = await import("./image-refs-BkMGNmEs.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,
@@ -18076,6 +18149,8 @@ async function main() {
 	const metricsStore = new MetricsStore(kv);
 	const dedupMap = new DedupMap();
 	const vectorIndex = embeddingProvider ? new VectorIndex() : null;
+	setVectorIndex(vectorIndex);
+	setEmbeddingProvider(embeddingProvider);
 	initMetrics(hasGetMeter(sdk) ? sdk.getMeter.bind(sdk) : void 0);
 	registerPrivacyFunction(sdk);
 	registerObserveFunction(sdk, kv, dedupMap, config.maxObservationsPerSession);
@@ -18293,4 +18368,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-d1vwmuPU.mjs.map
+//# sourceMappingURL=src-Cqsy23f_.mjs.map