[Cosmos][VectorIndex]Adding changes for vectorIndex and vectorEmbeddingPolicy

sdk/cosmos/azure-cosmos/CHANGELOG.md

@@ -4,10 +4,7 @@
 
 #### Features Added
 * Added `cosmosVectorEmbeddingPolicy` in `cosmosContainerProperties` and `vectorIndexes` in `indexPolicy` to support vector search in CosmosDB - See[39379](https://github.com/Azure/azure-sdk-for-java/pull/39379)
-
-#### Breaking Changes
-
-#### Bugs Fixed
+* Added public APIs `getCustomItemSerializer` and `setCustomItemSerializer` to allow customers to specify custom payload transformations or serialization settings. - See [PR 38997](https://github.com/Azure/azure-sdk-for-java/pull/38997) and [PR 39933](https://github.com/Azure/azure-sdk-for-java/pull/39933) 
 
 #### Other Changes
 * Load Blackbird or Afterburner into the ObjectMapper depending upon Java version and presence of modules in classpath. Make Afterburner and Blackbird optional maven dependencies. See - [PR 39689](https://github.com/Azure/azure-sdk-for-java/pull/39689)

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/DocumentCollection.java

@@ -6,8 +6,8 @@
 import com.azure.cosmos.CosmosItemSerializer;
 import com.azure.cosmos.implementation.apachecommons.lang.StringUtils;
 import com.azure.cosmos.implementation.caches.SerializableWrapper;
-import com.azure.cosmos.models.ClientEncryptionPolicy;
 import com.azure.cosmos.models.ChangeFeedPolicy;
+import com.azure.cosmos.models.ClientEncryptionPolicy;
 import com.azure.cosmos.models.ComputedProperty;
 import com.azure.cosmos.models.ConflictResolutionPolicy;
 import com.azure.cosmos.models.CosmosVectorEmbeddingPolicy;
@@ -25,7 +25,6 @@
 import java.util.Collection;
 import java.util.Collections;
 
-import static com.azure.cosmos.BridgeInternal.setProperty;
 import static com.azure.cosmos.implementation.guava25.base.Preconditions.checkNotNull;
 
 /**
@@ -439,8 +438,7 @@ public CosmosVectorEmbeddingPolicy getVectorEmbeddingPolicy() {
      */
     public void setVectorEmbeddingPolicy(CosmosVectorEmbeddingPolicy value) {
         checkNotNull(value, "cosmosVectorEmbeddingPolicy cannot be null");
-        this.cosmosVectorEmbeddingPolicy = value;
-        setProperty(this, Constants.Properties.VECTOR_EMBEDDING_POLICY, value);
+        this.set(Constants.Properties.VECTOR_EMBEDDING_POLICY, value, CosmosItemSerializer.DEFAULT_SERIALIZER);
     }
 
     public void populatePropertyBag() {

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosVectorEmbedding.java

@@ -74,7 +74,7 @@ public CosmosVectorDataType getDataType() {
      * @return CosmosVectorEmbedding
      */
     public CosmosVectorEmbedding setDataType(CosmosVectorDataType dataType) {
-        checkNotNull(dataType, "cosmosVectorDataType cannot be empty");
+        checkNotNull(dataType, "cosmosVectorDataType cannot be null");
         this.dataType = dataType.toString();
         return this;
     }
@@ -95,9 +95,9 @@ public Long getDimensions() {
      * @return CosmosVectorEmbedding
      */
     public CosmosVectorEmbedding setDimensions(Long dimensions) {
-        checkNotNull(dimensions, "dimensions cannot be empty");
+        checkNotNull(dimensions, "dimensions cannot be null");
         if (dimensions < 1) {
-            throw new IllegalArgumentException("Dimensions for the embedding has to be a long value greater than 1 " +
+            throw new IllegalArgumentException("Dimensions for the embedding has to be a long value greater than 0 " +
                 "for the vector embedding policy");
         }
 

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosVectorIndexSpec.java

@@ -3,9 +3,12 @@
 
 package com.azure.cosmos.models;
 
+import com.azure.cosmos.CosmosItemSerializer;
 import com.azure.cosmos.implementation.Constants;
 import com.azure.cosmos.implementation.JsonSerializable;
 
+import static com.azure.cosmos.implementation.guava25.base.Preconditions.checkNotNull;
+
 /**
  * Vector Indexes spec for Azure CosmosDB service.
  */
@@ -35,7 +38,7 @@ public String getPath() {
      * @return the SpatialSpec.
      */
     public CosmosVectorIndexSpec setPath(String path) {
-        this.jsonSerializable.set(Constants.Properties.PATH, path);
+        this.jsonSerializable.set(Constants.Properties.PATH, path, CosmosItemSerializer.DEFAULT_SERIALIZER);
         return this;
     }
 
@@ -47,10 +50,6 @@ public CosmosVectorIndexSpec setPath(String path) {
     public String getType() {
         if (this.type == null) {
             this.type = this.jsonSerializable.getString(Constants.Properties.VECTOR_INDEX_TYPE);
-
-            if (this.type == null) {
-                throw new IllegalArgumentException("INVALID vectorIndexType of " + this.jsonSerializable.getString(Constants.Properties.VECTOR_INDEX_TYPE));
-            }
         }
         return this.type;
     }
@@ -62,8 +61,9 @@ public String getType() {
      * @return the VectorIndexSpec
      */
     public CosmosVectorIndexSpec setType(String type) {
+        checkNotNull(type, "cosmosVectorIndexType cannot be null");
         this.type = type;
-        this.jsonSerializable.set(Constants.Properties.VECTOR_INDEX_TYPE, this.type);
+        this.jsonSerializable.set(Constants.Properties.VECTOR_INDEX_TYPE, this.type, CosmosItemSerializer.DEFAULT_SERIALIZER);
         return this;
     }
 

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/IndexingPolicy.java

@@ -305,7 +305,7 @@ public List<CosmosVectorIndexSpec> getVectorIndexes() {
      */
     public IndexingPolicy setVectorIndexes(List<CosmosVectorIndexSpec> vectorIndexes) {
         this.vectorIndexes = vectorIndexes;
-        this.jsonSerializable.set(Constants.Properties.VECTOR_INDEXES,this.vectorIndexes);
+        this.jsonSerializable.set(Constants.Properties.VECTOR_INDEXES,this.vectorIndexes, CosmosItemSerializer.DEFAULT_SERIALIZER);
         return this;
     }