@eleboucher/pi-memini 0.6.9 → 0.6.11

package/dist/index.js

@@ -905,10 +905,28 @@ function meminiExtension(pi) {
   } catch (error) {
     warn(`command registration skipped: ${String(error)}`);
   }
+  const MAX_TRACKED_SESSIONS = 200;
   const pendingUser = /* @__PURE__ */ new Map();
+  const rememberPendingUser = (session, prompt) => {
+    pendingUser.set(session, prompt);
+    while (pendingUser.size > MAX_TRACKED_SESSIONS) {
+      const oldest = pendingUser.keys().next().value;
+      if (oldest === void 0) break;
+      pendingUser.delete(oldest);
+    }
+  };
   const captured = /* @__PURE__ */ new Set();
+  const MAX_CAPTURED = 200;
+  const rememberCaptured = (id) => {
+    captured.add(id);
+    while (captured.size > MAX_CAPTURED) {
+      const oldest = captured.values().next().value;
+      if (oldest === void 0) break;
+      captured.delete(oldest);
+    }
+  };
   const injectedBySession = /* @__PURE__ */ new Map();
-  const MAX_TRACKED_SESSIONS = 200;
+  const MAX_INJECTED_PER_SESSION = 200;
   const rememberInjected = (session, ids) => {
     let seen = injectedBySession.get(session);
     if (!seen) {
@@ -921,16 +939,39 @@ function meminiExtension(pi) {
       }
     }
     for (const id of ids) if (id) seen.add(id);
+    while (seen.size > MAX_INJECTED_PER_SESSION) {
+      const oldest = seen.values().next().value;
+      if (oldest === void 0) break;
+      seen.delete(oldest);
+    }
+  };
+  let serverExcludeIds = true;
+  const searchExcluding = async (body, excludeIds) => {
+    if (!serverExcludeIds || excludeIds.length === 0) {
+      return client.postJson("/v1/search", body);
+    }
+    try {
+      const result = await client.postJson("/v1/search", { ...body, exclude_ids: excludeIds });
+      if (result !== null) return result;
+    } catch {
+    }
+    const retry = await client.postJson("/v1/search", body);
+    if (retry !== null) {
+      serverExcludeIds = false;
+      warn("memini: server does not accept exclude_ids; using client-side dedupe only");
+    }
+    return retry;
   };
   pi.on("before_agent_start", async (event, ctx) => {
     const sid = sessionIdOf(ctx);
     const query = String(event?.prompt || "").trim();
-    if (query && sid) pendingUser.set(sid, query);
+    if (query && sid) rememberPendingUser(sid, query);
     if (!cfg.recall || !query) return;
     const body = { query, limit: cfg.recall_limit };
     if (sid) body.exclude_metadata = { session_id: sid };
     if (cfg.recall_min_score > 0) body.min_score = cfg.recall_min_score;
-    const result = await client.postJson("/v1/search", body);
+    const excludeIds = sid ? [...injectedBySession.get(sid) ?? []] : [];
+    const result = await searchExcluding(body, excludeIds);
     const floor = cfg.recall_min_score > 0 ? cfg.recall_min_score : 0;
     let rawHits = Array.isArray(result?.results) ? result.results : [];
     if (sid) {
@@ -979,7 +1020,7 @@ function meminiExtension(pi) {
       metadata
     });
     if (stored !== null) {
-      if (dedupKey) captured.add(dedupKey);
+      if (dedupKey) rememberCaptured(dedupKey);
       if (sid) pendingUser.delete(sid);
     }
   });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eleboucher/pi-memini",
-  "version": "0.6.9",
+  "version": "0.6.11",
   "description": "Shared cross-session memory for the Pi coding agent, backed by a memini service.",
   "files": [
     "dist"