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); + } + }