From 2ef6647feadba4eecfb850d561d953843dfe6be8 Mon Sep 17 00:00:00 2001 From: Fu Cheng Date: Sat, 28 Sep 2024 14:32:25 +1200 Subject: [PATCH] Extend VectorStore from DocumentRetriever --- .../springframework/ai/vectorstore/VectorStore.java | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/spring-ai-core/src/main/java/org/springframework/ai/vectorstore/VectorStore.java b/spring-ai-core/src/main/java/org/springframework/ai/vectorstore/VectorStore.java index dadcdda3b18..0e75a228872 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/vectorstore/VectorStore.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/vectorstore/VectorStore.java @@ -19,6 +19,7 @@ import java.util.Optional; import org.springframework.ai.document.Document; +import org.springframework.ai.document.DocumentRetriever; import org.springframework.ai.document.DocumentWriter; /** @@ -29,7 +30,7 @@ * This interface allows for adding, deleting, and searching documents based on their * similarity to a given query. */ -public interface VectorStore extends DocumentWriter { +public interface VectorStore extends DocumentWriter, DocumentRetriever { default String getName() { return this.getClass().getSimpleName(); @@ -74,4 +75,14 @@ default List similaritySearch(String query) { return this.similaritySearch(SearchRequest.query(query)); } + /** + * Retrieves relevant documents + * @param query query string + * @return relevant documents + */ + @Override + default List retrieve(String query) { + return similaritySearch(query); + } + }