Added configuration options for database, collection names, embedding dimensions, index and metric types, and field names in the MilvusVectorStoreAutoConfiguration.

- Configuration options like `databaseName`, `collectionName`, `embeddingDimension`, `indexType`, and `metricType` were added.
- Additional parameters including `indexParameters`, `iDFieldName`, `contentFieldName`, `metadataFieldName`, and `embeddingFieldName` were incorporated into the builder method.

Signed-off-by: waileong <[email protected]>

Author: waileong

spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/milvus/MilvusVectorStoreAutoConfiguration.java

@@ -22,6 +22,8 @@
 import io.milvus.client.MilvusServiceClient;
 import io.milvus.param.ConnectParam;
 
+import io.milvus.param.IndexType;
+import io.milvus.param.MetricType;
 import org.springframework.ai.embedding.BatchingStrategy;
 import org.springframework.ai.embedding.EmbeddingModel;
 import org.springframework.ai.embedding.TokenCountBatchingStrategy;
@@ -69,11 +71,22 @@ public MilvusVectorStore vectorStore(MilvusServiceClient milvusClient, Embedding
 			ObjectProvider<VectorStoreObservationConvention> customObservationConvention) {
 
 		return MilvusVectorStore.builder(milvusClient, embeddingModel)
-			.initializeSchema(properties.isInitializeSchema())
-			.batchingStrategy(batchingStrategy)
-			.observationRegistry(observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP))
-			.customObservationConvention(customObservationConvention.getIfAvailable(() -> null))
-			.build();
+				.initializeSchema(properties.isInitializeSchema())
+				.databaseName(properties.getDatabaseName())
+				.collectionName(properties.getCollectionName())
+				.embeddingDimension(properties.getEmbeddingDimension())
+				.indexType(IndexType.valueOf(properties.getIndexType().name()))
+				.metricType(MetricType.valueOf(properties.getMetricType().name()))
+				.indexParameters(properties.getIndexParameters())
+				.iDFieldName(properties.getIdFieldName())
+				.autoId(properties.isAutoId())
+				.contentFieldName(properties.getContentFieldName())
+				.metadataFieldName(properties.getMetadataFieldName())
+				.embeddingFieldName(properties.getEmbeddingFieldName())
+				.batchingStrategy(batchingStrategy)
+				.observationRegistry(observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP))
+				.customObservationConvention(customObservationConvention.getIfAvailable(() -> null))
+				.build();
 	}
 
 	@Bean