@agentskit/memory 0.8.3 → 0.9.0

package/dist/chunk-G5S2A3MJ.js

@@ -0,0 +1,48 @@
+var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
+  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
+}) : x)(function(x) {
+  if (typeof require !== "undefined") return require.apply(this, arguments);
+  throw Error('Dynamic require of "' + x + '" is not supported');
+});
+
+// src/personalization.ts
+function createInMemoryPersonalization() {
+  const profiles = /* @__PURE__ */ new Map();
+  return {
+    async get(subjectId) {
+      const hit = profiles.get(subjectId);
+      return hit ? { ...hit, traits: { ...hit.traits } } : null;
+    },
+    async set(profile) {
+      profiles.set(profile.subjectId, {
+        ...profile,
+        traits: { ...profile.traits },
+        updatedAt: profile.updatedAt || (/* @__PURE__ */ new Date()).toISOString()
+      });
+    },
+    async merge(subjectId, traits) {
+      const existing = profiles.get(subjectId);
+      const next = {
+        subjectId,
+        traits: { ...existing?.traits ?? {}, ...traits },
+        updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      profiles.set(subjectId, next);
+      return { ...next, traits: { ...next.traits } };
+    },
+    async delete(subjectId) {
+      profiles.delete(subjectId);
+    }
+  };
+}
+function renderProfileContext(profile) {
+  if (!profile || Object.keys(profile.traits).length === 0) return "";
+  const lines = Object.entries(profile.traits).filter(([, value]) => value !== void 0 && value !== null && value !== "").map(([key, value]) => `- ${key}: ${typeof value === "string" ? value : JSON.stringify(value)}`);
+  if (lines.length === 0) return "";
+  return `## User profile
+${lines.join("\n")}`;
+}
+
+export { __require, createInMemoryPersonalization, renderProfileContext };
+//# sourceMappingURL=chunk-G5S2A3MJ.js.map
+//# sourceMappingURL=chunk-G5S2A3MJ.js.map

package/dist/chunk-G5S2A3MJ.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/personalization.ts"],"names":[],"mappings":";;;;;;;;AA0BO,SAAS,6BAAA,GAAsD;AACpE,EAAA,MAAM,QAAA,uBAAe,GAAA,EAAoC;AAEzD,EAAA,OAAO;AAAA,IACL,MAAM,IAAI,SAAA,EAAW;AACnB,MAAA,MAAM,GAAA,GAAM,QAAA,CAAS,GAAA,CAAI,SAAS,CAAA;AAClC,MAAA,OAAO,GAAA,GAAM,EAAE,GAAG,GAAA,EAAK,MAAA,EAAQ,EAAE,GAAG,GAAA,CAAI,MAAA,EAAO,EAAE,GAAI,IAAA;AAAA,IACvD,CAAA;AAAA,IACA,MAAM,IAAI,OAAA,EAAS;AACjB,MAAA,QAAA,CAAS,GAAA,CAAI,QAAQ,SAAA,EAAW;AAAA,QAC9B,GAAG,OAAA;AAAA,QACH,MAAA,EAAQ,EAAE,GAAG,OAAA,CAAQ,MAAA,EAAO;AAAA,QAC5B,WAAW,OAAA,CAAQ,SAAA,IAAA,iBAAa,IAAI,IAAA,IAAO,WAAA;AAAY,OACxD,CAAA;AAAA,IACH,CAAA;AAAA,IACA,MAAM,KAAA,CAAM,SAAA,EAAW,MAAA,EAAQ;AAC7B,MAAA,MAAM,QAAA,GAAW,QAAA,CAAS,GAAA,CAAI,SAAS,CAAA;AACvC,MAAA,MAAM,IAAA,GAA+B;AAAA,QACnC,SAAA;AAAA,QACA,MAAA,EAAQ,EAAE,GAAI,QAAA,EAAU,UAAU,EAAC,EAAI,GAAG,MAAA,EAAO;AAAA,QACjD,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,OACpC;AACA,MAAA,QAAA,CAAS,GAAA,CAAI,WAAW,IAAI,CAAA;AAC5B,MAAA,OAAO,EAAE,GAAG,IAAA,EAAM,MAAA,EAAQ,EAAE,GAAG,IAAA,CAAK,QAAO,EAAE;AAAA,IAC/C,CAAA;AAAA,IACA,MAAM,OAAO,SAAA,EAAW;AACtB,MAAA,QAAA,CAAS,OAAO,SAAS,CAAA;AAAA,IAC3B;AAAA,GACF;AACF;AAOO,SAAS,qBAAqB,OAAA,EAAgD;AACnF,EAAA,IAAI,CAAC,WAAW,MAAA,CAAO,IAAA,CAAK,QAAQ,MAAM,CAAA,CAAE,MAAA,KAAW,CAAA,EAAG,OAAO,EAAA;AACjE,EAAA,MAAM,QAAQ,MAAA,CAAO,OAAA,CAAQ,OAAA,CAAQ,MAAM,EACxC,MAAA,CAAO,CAAC,GAAG,KAAK,CAAA,KAAM,KAAA,KAAU,MAAA,IAAa,KAAA,KAAU,QAAQ,KAAA,KAAU,EAAE,CAAA,CAC3E,GAAA,CAAI,CAAC,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM,KAAK,GAAG,CAAA,EAAA,EAAK,OAAO,KAAA,KAAU,WAAW,KAAA,GAAQ,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC,CAAA,CAAE,CAAA;AACjG,EAAA,IAAI,KAAA,CAAM,MAAA,KAAW,CAAA,EAAG,OAAO,EAAA;AAC/B,EAAA,OAAO,CAAA;AAAA,EAAoB,KAAA,CAAM,IAAA,CAAK,IAAI,CAAC,CAAA,CAAA;AAC7C","file":"chunk-G5S2A3MJ.js","sourcesContent":["/**\n * Personalization — a persisted profile per subject (user id,\n * account id, device). The agent reads it on every run to\n * condition responses; the runtime updates it when new facts\n * appear.\n */\n\nexport interface PersonalizationProfile {\n  subjectId: string\n  /** Human-editable notes, facts, preferences. */\n  traits: Record<string, unknown>\n  /** ISO timestamp of the latest update. */\n  updatedAt: string\n}\n\nexport interface PersonalizationStore {\n  get: (subjectId: string) => Promise<PersonalizationProfile | null>\n  set: (profile: PersonalizationProfile) => Promise<void>\n  merge: (subjectId: string, traits: Record<string, unknown>) => Promise<PersonalizationProfile>\n  delete?: (subjectId: string) => Promise<void>\n}\n\n/**\n * In-memory personalization store — tests, single-process demos.\n * Bring your own for production (Postgres, Redis, DynamoDB).\n */\nexport function createInMemoryPersonalization(): PersonalizationStore {\n  const profiles = new Map<string, PersonalizationProfile>()\n\n  return {\n    async get(subjectId) {\n      const hit = profiles.get(subjectId)\n      return hit ? { ...hit, traits: { ...hit.traits } } : null\n    },\n    async set(profile) {\n      profiles.set(profile.subjectId, {\n        ...profile,\n        traits: { ...profile.traits },\n        updatedAt: profile.updatedAt || new Date().toISOString(),\n      })\n    },\n    async merge(subjectId, traits) {\n      const existing = profiles.get(subjectId)\n      const next: PersonalizationProfile = {\n        subjectId,\n        traits: { ...(existing?.traits ?? {}), ...traits },\n        updatedAt: new Date().toISOString(),\n      }\n      profiles.set(subjectId, next)\n      return { ...next, traits: { ...next.traits } }\n    },\n    async delete(subjectId) {\n      profiles.delete(subjectId)\n    },\n  }\n}\n\n/**\n * Render a profile into a system-prompt fragment the runtime can\n * prepend. Kept intentionally short — full profile dumps bloat\n * context and leak unnecessary detail to the model.\n */\nexport function renderProfileContext(profile: PersonalizationProfile | null): string {\n  if (!profile || Object.keys(profile.traits).length === 0) return ''\n  const lines = Object.entries(profile.traits)\n    .filter(([, value]) => value !== undefined && value !== null && value !== '')\n    .map(([key, value]) => `- ${key}: ${typeof value === 'string' ? value : JSON.stringify(value)}`)\n  if (lines.length === 0) return ''\n  return `## User profile\\n${lines.join('\\n')}`\n}\n"]}

package/dist/index.cjs

@@ -1,6 +1,7 @@
 'use strict';
 
 var core = require('@agentskit/core');
+var security = require('@agentskit/core/security');
 
 var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
   get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
@@ -21,7 +22,11 @@ function fileChatMemory(path) {
     },
     async save(messages) {
       const fs = await import('fs/promises');
-      await fs.writeFile(path, JSON.stringify(core.serializeMessages(messages), null, 2), "utf8");
+      await fs.writeFile(
+        path,
+        JSON.stringify(core.serializeMessages(messages), null, 2),
+        { encoding: "utf8", mode: 384 }
+      );
     },
     async clear() {
       try {
@@ -197,7 +202,7 @@ async function createRedisClientAdapter(url) {
       return await client.keys(pattern);
     },
     async disconnect() {
-      await client.disconnect();
+      await client.close();
     },
     async call(command, ...args) {
       return await client.sendCommand([command, ...args.map(String)]);
@@ -425,16 +430,23 @@ function createVectraStore(dirPath) {
   return {
     async upsert(docs) {
       const idx = await getIndex();
-      for (const doc of docs) {
-        await idx.insertItem({
-          vector: doc.vector,
-          metadata: { _id: doc.id, ...doc.metadata }
-        });
+      await idx.beginUpdate();
+      try {
+        for (const doc of docs) {
+          await idx.insertItem({
+            vector: doc.vector,
+            metadata: { _id: doc.id, ...doc.metadata }
+          });
+        }
+        await idx.endUpdate();
+      } catch (err) {
+        idx.cancelUpdate();
+        throw err;
       }
     },
     async query(vector, topK) {
       const idx = await getIndex();
-      const results = await idx.queryItems(vector, topK);
+      const results = await idx.queryItems(vector, "", topK);
       return results.map((r) => ({
         id: String(r.item.metadata._id ?? ""),
         score: r.score,
@@ -1259,6 +1271,102 @@ function createHierarchicalMemory(options) {
   };
 }
 
+// src/forget.ts
+function isForgettable(value) {
+  return !!value && typeof value === "object" && "forgetSubject" in value && typeof value.forgetSubject === "function";
+}
+async function hash(input) {
+  const encoded = new TextEncoder().encode(input);
+  const digest = await crypto.subtle.digest("SHA-256", encoded);
+  return Array.from(new Uint8Array(digest)).map((b) => b.toString(16).padStart(2, "0")).join("");
+}
+async function forgetSubject(memories, subjectId) {
+  const reports = [];
+  for (const memory of memories) {
+    if (!isForgettable(memory)) continue;
+    reports.push(await memory.forgetSubject(subjectId));
+  }
+  const totalDeleted = reports.reduce((sum, r) => sum + r.deletedCount, 0);
+  const evidenceHash = await hash(
+    JSON.stringify({ subjectId, reports: reports.map((r) => ({ b: r.backend, n: r.deletedCount, at: r.at })) })
+  );
+  return { subjectId, reports, totalDeleted, evidenceHash };
+}
+function makeForgettable(memory, options) {
+  return Object.assign(memory, {
+    __agentskitBackend: options.backend,
+    forgetSubject: async (subjectId) => {
+      const ids = await options.listIds(subjectId);
+      const failures = [];
+      try {
+        await options.deleteIds(ids);
+      } catch (err) {
+        for (const id of ids) failures.push({ id, reason: err.message });
+      }
+      return {
+        backend: options.backend,
+        deletedCount: ids.length - failures.length,
+        at: (/* @__PURE__ */ new Date()).toISOString(),
+        failures: failures.length ? failures : void 0
+      };
+    }
+  });
+}
+async function transform(input, opts) {
+  if (opts.mode === "tokenize") {
+    if (!opts.vault) {
+      throw new core.ConfigError({
+        code: core.ErrorCodes.AK_CONFIG_INVALID,
+        message: 'wrapMemoryWithRedaction: vault is required when mode is "tokenize"'
+      });
+    }
+    if (!opts.allowedRoles) {
+      throw new core.ConfigError({
+        code: core.ErrorCodes.AK_CONFIG_INVALID,
+        message: 'wrapMemoryWithRedaction: allowedRoles is required when mode is "tokenize"'
+      });
+    }
+    const result = await security.tokenize(input, {
+      rules: opts.rules,
+      vault: opts.vault,
+      allowedRoles: opts.allowedRoles,
+      audit: opts.audit
+    });
+    return result.value;
+  }
+  return security.createPIIRedactor({ rules: opts.rules }).redact(input).value;
+}
+function wrapChatMemoryWithRedaction(inner, options) {
+  return {
+    load: () => inner.load(),
+    save: async (messages) => {
+      const redacted = await Promise.all(
+        messages.map(async (m) => ({
+          ...m,
+          content: await transform(m.content ?? "", options)
+        }))
+      );
+      await inner.save(redacted);
+    },
+    clear: inner.clear ? () => inner.clear() : void 0
+  };
+}
+function wrapVectorMemoryWithRedaction(inner, options) {
+  return {
+    store: async (docs) => {
+      const redacted = await Promise.all(
+        docs.map(async (d) => ({
+          ...d,
+          content: await transform(d.content, options)
+        }))
+      );
+      await inner.store(redacted);
+    },
+    search: (embedding, opts) => inner.search(embedding, opts),
+    delete: inner.delete ? (ids) => inner.delete(ids) : void 0
+  };
+}
+
 exports.chroma = chroma;
 exports.createEncryptedMemory = createEncryptedMemory;
 exports.createHierarchicalMemory = createHierarchicalMemory;
@@ -1266,6 +1374,8 @@ exports.createInMemoryGraph = createInMemoryGraph;
 exports.createInMemoryPersonalization = createInMemoryPersonalization;
 exports.fileChatMemory = fileChatMemory;
 exports.fileVectorMemory = fileVectorMemory;
+exports.forgetSubject = forgetSubject;
+exports.makeForgettable = makeForgettable;
 exports.matchesFilter = matchesFilter;
 exports.milvusVectorStore = milvusVectorStore;
 exports.mongoAtlasVectorStore = mongoAtlasVectorStore;
@@ -1280,5 +1390,7 @@ exports.supabaseVectorStore = supabaseVectorStore;
 exports.tursoChatMemory = tursoChatMemory;
 exports.upstashVector = upstashVector;
 exports.weaviateVectorStore = weaviateVectorStore;
+exports.wrapChatMemoryWithRedaction = wrapChatMemoryWithRedaction;
+exports.wrapVectorMemoryWithRedaction = wrapVectorMemoryWithRedaction;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map