Skip to content

Commit 2a94f69

Browse files
authored
Merge pull request #2015 from karesti/removeAndRemoveAll
Implement removeIds and removeAll
2 parents dc40f6f + dbd3cba commit 2a94f69

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

embedding-stores/infinispan/deployment/src/test/java/io/quarkiverse/langchain4j/infinispan/deployment/InfinispanEmbeddingStoreTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ public static void initEmbeddingModel() {
4242

4343
@Override
4444
protected void clearStore() {
45-
embeddingStore.deleteAll();
45+
embeddingStore.removeAll();
4646
}
4747

4848
@Override

embedding-stores/infinispan/runtime/src/main/java/io/quarkiverse/langchain4j/infinispan/InfinispanEmbeddingStore.java

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
import java.util.ArrayList;
99
import java.util.Arrays;
10+
import java.util.Collection;
1011
import java.util.HashMap;
1112
import java.util.List;
1213
import java.util.Map;
@@ -158,8 +159,14 @@ public EmbeddingSearchResult<TextSegment> search(EmbeddingSearchRequest request)
158159
}).filter(Objects::nonNull).collect(Collectors.toList()));
159160
}
160161

161-
public void deleteAll() {
162-
remoteCache.clearAsync();
162+
@Override
163+
public void removeAll(Collection<String> ids) {
164+
ids.forEach(id -> remoteCache.remove(id));
165+
}
166+
167+
@Override
168+
public void removeAll() {
169+
remoteCache.clear();
163170
}
164171

165172
public static class Builder {

0 commit comments

Comments
 (0)