Fix SimpleVectorStore' deprecated constructor use

- Use SimpleVectorStoreBuilder for SimpleVectorStore instance creation
This commit is contained in:
Ilayaperumal Gopinathan
2024-12-20 20:29:40 +00:00
parent 2804604933
commit 1e68862142
4 changed files with 5 additions and 5 deletions

View File

@@ -82,7 +82,7 @@ public class AcmeIT extends AbstractIT {
// Step 2 - Create embeddings and save to vector store
logger.info("Creating Embeddings...");
VectorStore vectorStore = new SimpleVectorStore(this.embeddingModel);
VectorStore vectorStore = SimpleVectorStore.builder(this.embeddingModel).build();
vectorStore.accept(textSplitter.apply(jsonReader.get()));

View File

@@ -56,14 +56,14 @@ public class SimplePersistentVectorStoreIT {
JsonReader jsonReader = new JsonReader(this.bikesJsonResource, new ProductMetadataGenerator(), "price", "name",
"shortDescription", "description", "tags");
List<Document> documents = jsonReader.get();
SimpleVectorStore vectorStore = new SimpleVectorStore(this.embeddingModel);
SimpleVectorStore vectorStore = SimpleVectorStore.builder(this.embeddingModel).build();
vectorStore.add(documents);
File tempFile = new File(this.workingDir.toFile(), "temp.txt");
vectorStore.save(tempFile);
assertThat(tempFile).isNotEmpty();
assertThat(tempFile).content().contains("Velo 99 XR1 AXS");
SimpleVectorStore vectorStore2 = new SimpleVectorStore(this.embeddingModel);
SimpleVectorStore vectorStore2 = SimpleVectorStore.builder(this.embeddingModel).build();
vectorStore2.load(tempFile);
List<Document> similaritySearch = vectorStore2.similaritySearch("Velo 99 XR1 AXS");

View File

@@ -144,7 +144,7 @@ class SimpleVectorStoreTests {
File saveFile = this.tempDir.resolve("vector-store.json").toFile();
this.vectorStore.save(saveFile);
SimpleVectorStore loadedStore = new SimpleVectorStore(this.mockEmbeddingModel);
SimpleVectorStore loadedStore = SimpleVectorStore.builder(this.mockEmbeddingModel).build();
loadedStore.load(saveFile);
List<Document> results = loadedStore.similaritySearch("test content");

View File

@@ -33,7 +33,7 @@ public class TestApplication {
@Bean
SimpleVectorStore simpleVectorStore(EmbeddingModel embeddingModel) {
return new SimpleVectorStore(embeddingModel);
return SimpleVectorStore.builder(embeddingModel).build();
}
}