Skip to content

Commit 49acedd

Browse files
committed
Oracle vector store builder refactoring
1 parent b25d6e8 commit 49acedd

File tree

9 files changed

+324
-78
lines changed

9 files changed

+324
-78
lines changed

spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/oracle/OracleVectorStoreAutoConfiguration.java

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
import org.springframework.ai.embedding.BatchingStrategy;
2424
import org.springframework.ai.embedding.EmbeddingModel;
2525
import org.springframework.ai.embedding.TokenCountBatchingStrategy;
26-
import org.springframework.ai.vectorstore.OracleVectorStore;
26+
import org.springframework.ai.vectorstore.oracle.OracleVectorStore;
2727
import org.springframework.ai.vectorstore.observation.VectorStoreObservationConvention;
2828
import org.springframework.beans.factory.ObjectProvider;
2929
import org.springframework.boot.autoconfigure.AutoConfiguration;
@@ -59,11 +59,22 @@ public OracleVectorStore vectorStore(JdbcTemplate jdbcTemplate, EmbeddingModel e
5959
OracleVectorStoreProperties properties, ObjectProvider<ObservationRegistry> observationRegistry,
6060
ObjectProvider<VectorStoreObservationConvention> customObservationConvention,
6161
BatchingStrategy batchingStrategy) {
62-
return new OracleVectorStore(jdbcTemplate, embeddingModel, properties.getTableName(), properties.getIndexType(),
63-
properties.getDistanceType(), properties.getDimensions(), properties.getSearchAccuracy(),
64-
properties.isInitializeSchema(), properties.isRemoveExistingVectorStoreTable(),
65-
properties.isForcedNormalization(), observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP),
66-
customObservationConvention.getIfAvailable(() -> null), batchingStrategy);
62+
63+
return OracleVectorStore.builder()
64+
.jdbcTemplate(jdbcTemplate)
65+
.embeddingModel(embeddingModel)
66+
.tableName(properties.getTableName())
67+
.indexType(properties.getIndexType())
68+
.distanceType(properties.getDistanceType())
69+
.dimensions(properties.getDimensions())
70+
.searchAccuracy(properties.getSearchAccuracy())
71+
.initializeSchema(properties.isInitializeSchema())
72+
.removeExistingVectorStoreTable(properties.isRemoveExistingVectorStoreTable())
73+
.forcedNormalization(properties.isForcedNormalization())
74+
.observationRegistry(observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP))
75+
.customObservationConvention(customObservationConvention.getIfAvailable(() -> null))
76+
.batchingStrategy(batchingStrategy)
77+
.build();
6778
}
6879

6980
}

spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/oracle/OracleVectorStoreProperties.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
package org.springframework.ai.autoconfigure.vectorstore.oracle;
1818

1919
import org.springframework.ai.autoconfigure.vectorstore.CommonVectorStoreProperties;
20-
import org.springframework.ai.vectorstore.OracleVectorStore;
20+
import org.springframework.ai.vectorstore.oracle.OracleVectorStore;
2121
import org.springframework.boot.context.properties.ConfigurationProperties;
2222

2323
/**
@@ -42,7 +42,7 @@ public class OracleVectorStoreProperties extends CommonVectorStoreProperties {
4242

4343
private boolean forcedNormalization;
4444

45-
private int searchAccuracy = org.springframework.ai.vectorstore.OracleVectorStore.DEFAULT_SEARCH_ACCURACY;
45+
private int searchAccuracy = OracleVectorStore.DEFAULT_SEARCH_ACCURACY;
4646

4747
public String getTableName() {
4848
return this.tableName;

spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/vectorstore/oracle/OracleVectorStorePropertiesTests.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,9 @@
1818

1919
import org.junit.jupiter.api.Test;
2020

21-
import org.springframework.ai.vectorstore.OracleVectorStore;
22-
import org.springframework.ai.vectorstore.OracleVectorStore.OracleVectorStoreDistanceType;
23-
import org.springframework.ai.vectorstore.OracleVectorStore.OracleVectorStoreIndexType;
21+
import org.springframework.ai.vectorstore.oracle.OracleVectorStore;
22+
import org.springframework.ai.vectorstore.oracle.OracleVectorStore.OracleVectorStoreDistanceType;
23+
import org.springframework.ai.vectorstore.oracle.OracleVectorStore.OracleVectorStoreIndexType;
2424

2525
import static org.assertj.core.api.Assertions.assertThat;
2626

0 commit comments

Comments
 (0)