Fix SimpleVectorStore' deprecated constructor use
- Use SimpleVectorStoreBuilder for SimpleVectorStore instance creation
This commit is contained in:
@@ -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()));
|
||||
|
||||
|
||||
@@ -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");
|
||||
|
||||
@@ -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");
|
||||
|
||||
@@ -33,7 +33,7 @@ public class TestApplication {
|
||||
|
||||
@Bean
|
||||
SimpleVectorStore simpleVectorStore(EmbeddingModel embeddingModel) {
|
||||
return new SimpleVectorStore(embeddingModel);
|
||||
return SimpleVectorStore.builder(embeddingModel).build();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user