Skip to content

Commit ca8a282

Browse files
committed
Typesense vector store builder refactoring
1 parent 218c967 commit ca8a282

File tree

7 files changed

+199
-58
lines changed

7 files changed

+199
-58
lines changed

spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/typesense/TypesenseVectorStoreAutoConfiguration.java

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@
2828
import org.springframework.ai.embedding.BatchingStrategy;
2929
import org.springframework.ai.embedding.EmbeddingModel;
3030
import org.springframework.ai.embedding.TokenCountBatchingStrategy;
31-
import org.springframework.ai.vectorstore.TypesenseVectorStore;
32-
import org.springframework.ai.vectorstore.TypesenseVectorStore.TypesenseVectorStoreConfig;
31+
import org.springframework.ai.vectorstore.typesense.TypesenseVectorStore;
32+
import org.springframework.ai.vectorstore.typesense.TypesenseVectorStore.TypesenseVectorStoreConfig;
3333
import org.springframework.ai.vectorstore.observation.VectorStoreObservationConvention;
3434
import org.springframework.beans.factory.ObjectProvider;
3535
import org.springframework.boot.autoconfigure.AutoConfiguration;
@@ -70,14 +70,16 @@ public TypesenseVectorStore vectorStore(Client typesenseClient, EmbeddingModel e
7070
ObjectProvider<VectorStoreObservationConvention> customObservationConvention,
7171
BatchingStrategy batchingStrategy) {
7272

73-
TypesenseVectorStoreConfig config = TypesenseVectorStoreConfig.builder()
74-
.withCollectionName(properties.getCollectionName())
75-
.withEmbeddingDimension(properties.getEmbeddingDimension())
73+
return TypesenseVectorStore.builder()
74+
.client(typesenseClient)
75+
.embeddingModel(embeddingModel)
76+
.collectionName(properties.getCollectionName())
77+
.embeddingDimension(properties.getEmbeddingDimension())
78+
.initializeSchema(properties.isInitializeSchema())
79+
.observationRegistry(observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP))
80+
.customObservationConvention(customObservationConvention.getIfAvailable(() -> null))
81+
.batchingStrategy(batchingStrategy)
7682
.build();
77-
78-
return new TypesenseVectorStore(typesenseClient, embeddingModel, config, properties.isInitializeSchema(),
79-
observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP),
80-
customObservationConvention.getIfAvailable(() -> null), batchingStrategy);
8183
}
8284

8385
@Bean

spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/typesense/TypesenseVectorStoreProperties.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
package org.springframework.ai.autoconfigure.vectorstore.typesense;
1818

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

2323
/**
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
* limitations under the License.
1515
*/
1616

17-
package org.springframework.ai.vectorstore;
17+
package org.springframework.ai.vectorstore.typesense;
1818

1919
import org.springframework.ai.vectorstore.filter.Filter;
2020
import org.springframework.ai.vectorstore.filter.converter.AbstractFilterExpressionConverter;

0 commit comments

Comments
 (0)