improve schema initialization logic and update deprecated code for Milvus vectorstore

Signed-off-by: jonghoon park <[email protected]>

Author: dev-jonghoonpark

vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/milvus/MilvusVectorStore.java

@@ -38,6 +38,7 @@
 import io.milvus.param.R;
 import io.milvus.param.R.Status;
 import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.CollectionSchemaParam;
 import io.milvus.param.collection.CreateCollectionParam;
 import io.milvus.param.collection.DropCollectionParam;
 import io.milvus.param.collection.FieldType;
@@ -443,6 +444,8 @@ void createCollection() {
 		if (!isDatabaseCollectionExists()) {
 			createCollection(this.databaseName, this.collectionName, this.idFieldName, this.isAutoId,
 					this.contentFieldName, this.metadataFieldName, this.embeddingFieldName);
+			createIndex(this.databaseName, this.collectionName, this.embeddingFieldName, this.indexType,
+					this.metricType, this.indexParameters);
 		}
 
 		R<DescribeIndexResponse> indexDescriptionResponse = this.milvusClient
@@ -452,19 +455,8 @@ void createCollection() {
 				.build());
 
 		if (indexDescriptionResponse.getData() == null) {
-			R<RpcStatus> indexStatus = this.milvusClient.createIndex(CreateIndexParam.newBuilder()
-				.withDatabaseName(this.databaseName)
-				.withCollectionName(this.collectionName)
-				.withFieldName(this.embeddingFieldName)
-				.withIndexType(this.indexType)
-				.withMetricType(this.metricType)
-				.withExtraParam(this.indexParameters)
-				.withSyncMode(Boolean.FALSE)
-				.build());
-
-			if (indexStatus.getException() != null) {
-				throw new RuntimeException("Failed to create Index", indexStatus.getException());
-			}
+			createIndex(this.databaseName, this.collectionName, this.embeddingFieldName, this.indexType,
+					this.metricType, this.indexParameters);
 		}
 
 		R<RpcStatus> loadCollectionStatus = this.milvusClient.loadCollection(LoadCollectionParam.newBuilder()
@@ -507,10 +499,12 @@ void createCollection(String databaseName, String collectionName, String idField
 			.withDescription("Spring AI Vector Store")
 			.withConsistencyLevel(ConsistencyLevelEnum.STRONG)
 			.withShardsNum(2)
-			.addFieldType(docIdFieldType)
-			.addFieldType(contentFieldType)
-			.addFieldType(metadataFieldType)
-			.addFieldType(embeddingFieldType)
+			.withSchema(CollectionSchemaParam.newBuilder()
+				.addFieldType(docIdFieldType)
+				.addFieldType(contentFieldType)
+				.addFieldType(metadataFieldType)
+				.addFieldType(embeddingFieldType)
+				.build())
 			.build();
 
 		R<RpcStatus> collectionStatus = this.milvusClient.createCollection(createCollectionReq);
@@ -520,6 +514,23 @@ void createCollection(String databaseName, String collectionName, String idField
 
 	}
 
+	void createIndex(String databaseName, String collectionName, String embeddingFieldName, IndexType indexType,
+			MetricType metricType, String indexParameters) {
+		R<RpcStatus> indexStatus = this.milvusClient.createIndex(CreateIndexParam.newBuilder()
+			.withDatabaseName(databaseName)
+			.withCollectionName(collectionName)
+			.withFieldName(embeddingFieldName)
+			.withIndexType(indexType)
+			.withMetricType(metricType)
+			.withExtraParam(indexParameters)
+			.withSyncMode(Boolean.FALSE)
+			.build());
+
+		if (indexStatus.getException() != null) {
+			throw new RuntimeException("Failed to create Index", indexStatus.getException());
+		}
+	}
+
 	int embeddingDimensions() {
 		if (this.embeddingDimension != INVALID_EMBEDDING_DIMENSION) {
 			return this.embeddingDimension;