@agenr/agenr-plugin 1.8.0 → 1.8.1

package/dist/{chunk-6CEKKEFZ.js → chunk-O45JQ6O3.js}

@@ -2737,7 +2737,7 @@ async function extractClaimKeyDecision(entry, llm, config, options = {}) {
 async function getEntityHints(db) {
   return db.getDistinctClaimKeyPrefixes();
 }
-async function runBatchClaimExtraction(results, ports, config, _concurrency = 10, onWarning, onDiagnostic) {
+async function runBatchClaimExtraction(results, ports, config, concurrency = 10, onWarning, onDiagnostic) {
   if (!config.enabled) {
     return /* @__PURE__ */ new Map();
   }
@@ -2746,8 +2746,12 @@ async function runBatchClaimExtraction(results, ports, config, _concurrency = 10
   const extractedEntries = /* @__PURE__ */ new Map();
   const diagnostics = /* @__PURE__ */ new Map();
   const retryEntries = [];
-  for (const result of results) {
-    for (const entry of result.entries) {
+  const stageSize = normalizeClaimExtractionConcurrency(concurrency);
+  const orderedEntries = results.flatMap((result) => result.entries);
+  for (let stageStart = 0; stageStart < orderedEntries.length; stageStart += stageSize) {
+    const stageEntries = orderedEntries.slice(stageStart, stageStart + stageSize);
+    const stageRequests = [];
+    for (const entry of stageEntries) {
       if (entry.claim_key) {
         recordClaimKeyHint(hintState, entry.claim_key);
         continue;
@@ -2765,7 +2769,18 @@ async function runBatchClaimExtraction(results, ports, config, _concurrency = 10
         });
         continue;
       }
-      const decision = await extractBatchClaimKeyDecision(entry, llm, config, hintState, onWarning);
+      stageRequests.push({
+        entry,
+        hintSnapshot: buildClaimExtractionHintSnapshot(hintState, entry)
+      });
+    }
+    const stageDecisions = await Promise.all(
+      stageRequests.map(async ({ entry, hintSnapshot }) => ({
+        entry,
+        decision: await extractBatchClaimKeyDecision(entry, llm, config, hintSnapshot, onWarning)
+      }))
+    );
+    for (const { entry, decision } of stageDecisions) {
       diagnostics.set(entry, decision.diagnostic);
       if (decision.result?.claimKey) {
         applyClaimExtractionResultToEntry(entry, decision.result);
@@ -2777,18 +2792,26 @@ async function runBatchClaimExtraction(results, ports, config, _concurrency = 10
     }
   }
   if (retryEntries.length > 0 && extractedEntries.size > 0) {
-    for (const entry of retryEntries) {
-      if (entry.claim_key) {
-        continue;
-      }
-      const decision = await extractBatchClaimKeyDecision(entry, llm, config, hintState, onWarning);
-      diagnostics.set(entry, decision.diagnostic);
-      if (!decision.result?.claimKey) {
-        continue;
+    for (let stageStart = 0; stageStart < retryEntries.length; stageStart += stageSize) {
+      const stageRequests = retryEntries.slice(stageStart, stageStart + stageSize).filter((entry) => !entry.claim_key).map((entry) => ({
+        entry,
+        hintSnapshot: buildClaimExtractionHintSnapshot(hintState, entry)
+      }));
+      const stageDecisions = await Promise.all(
+        stageRequests.map(async ({ entry, hintSnapshot }) => ({
+          entry,
+          decision: await extractBatchClaimKeyDecision(entry, llm, config, hintSnapshot, onWarning)
+        }))
+      );
+      for (const { entry, decision } of stageDecisions) {
+        diagnostics.set(entry, decision.diagnostic);
+        if (!decision.result?.claimKey) {
+          continue;
+        }
+        applyClaimExtractionResultToEntry(entry, decision.result);
+        recordClaimKeyHint(hintState, decision.result.claimKey);
+        extractedEntries.set(entry, decision.result);
       }
-      applyClaimExtractionResultToEntry(entry, decision.result);
-      recordClaimKeyHint(hintState, decision.result.claimKey);
-      extractedEntries.set(entry, decision.result);
     }
   }
   for (const result of results) {
@@ -2801,7 +2824,20 @@ async function runBatchClaimExtraction(results, ports, config, _concurrency = 10
   }
   return extractedEntries;
 }
-async function extractBatchClaimKeyDecision(entry, llm, config, hintState, onWarning) {
+function normalizeClaimExtractionConcurrency(value) {
+  if (!Number.isInteger(value) || value <= 0) {
+    return 10;
+  }
+  return value;
+}
+function buildClaimExtractionHintSnapshot(hintState, entry) {
+  return {
+    hints: buildEntryHints(hintState, entry),
+    supportClaimKeys: [...hintState.supportClaimKeys],
+    entityPrefixStats: hintState.entityPrefixStats
+  };
+}
+async function extractBatchClaimKeyDecision(entry, llm, config, hintSnapshot, onWarning) {
   try {
     return await extractClaimKeyDecision(
       {
@@ -2814,10 +2850,10 @@ async function extractBatchClaimKeyDecision(entry, llm, config, hintState, onWar
       llm,
       config,
       {
-        hints: buildEntryHints(hintState, entry),
+        hints: hintSnapshot.hints,
         onWarning,
-        supportClaimKeys: [...hintState.supportClaimKeys],
-        entityPrefixStats: hintState.entityPrefixStats
+        supportClaimKeys: hintSnapshot.supportClaimKeys,
+        entityPrefixStats: hintSnapshot.entityPrefixStats
       }
     );
   } catch {
@@ -3735,7 +3771,7 @@ async function maybeExtractClaimKeys(preparedEntries, options) {
         db: claimExtraction.db
       },
       claimExtraction.config,
-      1,
+      claimExtraction.config.concurrency ?? 10,
       options.onWarning,
       (entry, diagnostic) => {
         const preparedEntry = preparedEntries.find((candidate) => candidate.input === entry);

package/dist/index.js

@@ -7,7 +7,7 @@ import {
   parseTuiSessionKey,
   readOpenClawSessionsStore,
   storeEntriesDetailed
-} from "./chunk-6CEKKEFZ.js";
+} from "./chunk-O45JQ6O3.js";
 import {
   EMBEDDING_DIMENSIONS,
   ENTRY_TYPES,
@@ -1055,7 +1055,7 @@ function registerAgenrOpenClawTools(api, servicesPromise, logger) {
 var openclaw_plugin_default = {
   id: "agenr",
   name: "agenr",
-  version: "1.8.0",
+  version: "1.8.1",
   description: "agenr memory plugin for OpenClaw",
   kind: "memory",
   contracts: {

package/openclaw.plugin.json

@@ -1,7 +1,7 @@
 {
   "id": "agenr",
   "name": "agenr",
-  "version": "1.8.0",
+  "version": "1.8.1",
   "description": "agenr memory plugin for OpenClaw",
   "kind": "memory",
   "contracts": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agenr/agenr-plugin",
-  "version": "1.8.0",
+  "version": "1.8.1",
   "description": "agenr memory plugin for OpenClaw",
   "type": "module",
   "exports": {