diff --git a/embedding-stores/infinispan/deployment/src/test/java/io/quarkiverse/langchain4j/infinispan/deployment/InfinispanEmbeddingStoreTest.java b/embedding-stores/infinispan/deployment/src/test/java/io/quarkiverse/langchain4j/infinispan/deployment/InfinispanEmbeddingStoreTest.java index 23a4b146f..3e1456d97 100644 --- a/embedding-stores/infinispan/deployment/src/test/java/io/quarkiverse/langchain4j/infinispan/deployment/InfinispanEmbeddingStoreTest.java +++ b/embedding-stores/infinispan/deployment/src/test/java/io/quarkiverse/langchain4j/infinispan/deployment/InfinispanEmbeddingStoreTest.java @@ -42,7 +42,7 @@ public static void initEmbeddingModel() { @Override protected void clearStore() { - embeddingStore.deleteAll(); + embeddingStore.removeAll(); } @Override diff --git a/embedding-stores/infinispan/runtime/src/main/java/io/quarkiverse/langchain4j/infinispan/InfinispanEmbeddingStore.java b/embedding-stores/infinispan/runtime/src/main/java/io/quarkiverse/langchain4j/infinispan/InfinispanEmbeddingStore.java index e777e02b6..312b47140 100644 --- a/embedding-stores/infinispan/runtime/src/main/java/io/quarkiverse/langchain4j/infinispan/InfinispanEmbeddingStore.java +++ b/embedding-stores/infinispan/runtime/src/main/java/io/quarkiverse/langchain4j/infinispan/InfinispanEmbeddingStore.java @@ -7,6 +7,7 @@ import java.util.ArrayList; import java.util.Arrays; +import java.util.Collection; import java.util.HashMap; import java.util.List; import java.util.Map; @@ -158,8 +159,14 @@ public EmbeddingSearchResult search(EmbeddingSearchRequest request) }).filter(Objects::nonNull).collect(Collectors.toList())); } - public void deleteAll() { - remoteCache.clearAsync(); + @Override + public void removeAll(Collection ids) { + ids.forEach(id -> remoteCache.remove(id)); + } + + @Override + public void removeAll() { + remoteCache.clear(); } public static class Builder {