Skip to content

Commit 1e68862

Browse files
committed
Fix SimpleVectorStore' deprecated constructor use
- Use SimpleVectorStoreBuilder for SimpleVectorStore instance creation
1 parent 2804604 commit 1e68862

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/acme/AcmeIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ void acmeChain() {
8282
// Step 2 - Create embeddings and save to vector store
8383

8484
logger.info("Creating Embeddings...");
85-
VectorStore vectorStore = new SimpleVectorStore(this.embeddingModel);
85+
VectorStore vectorStore = SimpleVectorStore.builder(this.embeddingModel).build();
8686

8787
vectorStore.accept(textSplitter.apply(jsonReader.get()));
8888

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/vectorstore/SimplePersistentVectorStoreIT.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,14 +56,14 @@ void persist() {
5656
JsonReader jsonReader = new JsonReader(this.bikesJsonResource, new ProductMetadataGenerator(), "price", "name",
5757
"shortDescription", "description", "tags");
5858
List<Document> documents = jsonReader.get();
59-
SimpleVectorStore vectorStore = new SimpleVectorStore(this.embeddingModel);
59+
SimpleVectorStore vectorStore = SimpleVectorStore.builder(this.embeddingModel).build();
6060
vectorStore.add(documents);
6161

6262
File tempFile = new File(this.workingDir.toFile(), "temp.txt");
6363
vectorStore.save(tempFile);
6464
assertThat(tempFile).isNotEmpty();
6565
assertThat(tempFile).content().contains("Velo 99 XR1 AXS");
66-
SimpleVectorStore vectorStore2 = new SimpleVectorStore(this.embeddingModel);
66+
SimpleVectorStore vectorStore2 = SimpleVectorStore.builder(this.embeddingModel).build();
6767

6868
vectorStore2.load(tempFile);
6969
List<Document> similaritySearch = vectorStore2.similaritySearch("Velo 99 XR1 AXS");

spring-ai-core/src/test/java/org/springframework/ai/vectorstore/SimpleVectorStoreTests.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ void shouldSaveAndLoadVectorStore() throws IOException {
144144
File saveFile = this.tempDir.resolve("vector-store.json").toFile();
145145
this.vectorStore.save(saveFile);
146146

147-
SimpleVectorStore loadedStore = new SimpleVectorStore(this.mockEmbeddingModel);
147+
SimpleVectorStore loadedStore = SimpleVectorStore.builder(this.mockEmbeddingModel).build();
148148
loadedStore.load(saveFile);
149149

150150
List<Document> results = loadedStore.similaritySearch("test content");

spring-ai-integration-tests/src/test/java/org/springframework/ai/integration/tests/TestApplication.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ public class TestApplication {
3333

3434
@Bean
3535
SimpleVectorStore simpleVectorStore(EmbeddingModel embeddingModel) {
36-
return new SimpleVectorStore(embeddingModel);
36+
return SimpleVectorStore.builder(embeddingModel).build();
3737
}
3838

3939
}

0 commit comments

Comments
 (0)