@ai-matrx/agents 0.18.5 → 0.19.2

package/dist/sources/react/index.cjs

@@ -382,9 +382,19 @@ function setCardWaitForClean(state, id, wait) {
 function setSourceTopic(state, topic) {
   return state.topic === topic ? state : { ...state, topic };
 }
-function applyReviewedSourceSet(state, set) {
+function removedInReview(reviewed, answer) {
+  const kept = new Set(answer.sources.map(sourceKey));
+  return new Set(reviewed.sources.map(sourceKey).filter((k) => !kept.has(k)));
+}
+function applyReviewedSourceSet(state, set, reviewed) {
   let next = state;
-  for (const card of state.cards) {
+  if (reviewed) {
+    const removed = removedInReview(reviewed, set);
+    for (const card of state.cards) {
+      if (card.draft.ref && removed.has(sourceKey(card.draft.ref))) next = removeCard(next, card.id);
+    }
+  }
+  for (const card of next.cards) {
     const ref = card.draft.ref;
     if (!ref) continue;
     const match = set.sources.find((r) => sameSourceRef(r, ref));
@@ -600,8 +610,8 @@ function createSourceSetController(adapter) {
     hasRef: (resourceType, resourceId) => hasSourceRef(store.getState(), resourceType, resourceId),
     liveCount: () => store.getState().cards.length,
     toSourceSet: (options = {}) => selectSourceSet(store.getState(), options),
-    applySourceSet(set) {
-      commit(applyReviewedSourceSet(store.getState(), set));
+    applySourceSet(set, options = {}) {
+      commit(applyReviewedSourceSet(store.getState(), set, options.reviewed));
     },
     async manifest() {
       const refs = readySourceRefs(store.getState());