@absolutejs/rag 0.6.0 → 0.6.2

package/dist/index.js

@@ -36126,6 +36126,10 @@ var planRAGCorpus = async (store, owner, desired) => {
     unchanged
   };
 };
+var filterToWritten = (planned, written) => {
+  const ids = new Set(written);
+  return planned.filter((doc) => ids.has(doc.chunkId));
+};
 var reconcileRAGCorpus = async (store, owner, desired, apply) => {
   const plan = await planRAGCorpus(store, owner, desired);
   if (plan.remove.length > 0) {
@@ -36134,11 +36138,15 @@ var reconcileRAGCorpus = async (store, owner, desired, apply) => {
   }
   let embedded = 0;
   if (plan.embed.length > 0) {
-    embedded = await apply.embed(plan.embed, owner);
-    await store.remember(owner, await Promise.all(plan.embed.map(async (doc) => ({
-      chunkId: doc.chunkId,
-      textHash: await corpusTextHash(doc.text)
-    }))));
+    const outcome = await apply.embed(plan.embed, owner);
+    const written = typeof outcome === "number" ? plan.embed.slice(0, Math.max(0, Math.min(outcome, plan.embed.length))) : filterToWritten(plan.embed, outcome);
+    embedded = written.length;
+    if (written.length > 0) {
+      await store.remember(owner, await Promise.all(written.map(async (doc) => ({
+        chunkId: doc.chunkId,
+        textHash: await corpusTextHash(doc.text)
+      }))));
+    }
   }
   return { embedded, removed: plan.remove.length, unchanged: plan.unchanged };
 };
@@ -36442,5 +36450,5 @@ export {
   addRAGEvaluationSuiteCase
 };
 
-//# debugId=8567BF91160369B364756E2164756E21
+//# debugId=9A8B228F28FF5ED164756E2164756E21
 //# sourceMappingURL=index.js.map