Converted vector operations to search operations.

See #4706
Original pull request: #4882

Author: marcingrzejszczak

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/DefaultIndexOperations.java

@@ -24,11 +24,9 @@
 import org.springframework.data.mongodb.MongoDatabaseFactory;
 import org.springframework.data.mongodb.UncategorizedMongoDbException;
 import org.springframework.data.mongodb.core.convert.QueryMapper;
-import org.springframework.data.mongodb.core.index.DefaultVectorIndexOperations;
 import org.springframework.data.mongodb.core.index.IndexDefinition;
 import org.springframework.data.mongodb.core.index.IndexInfo;
 import org.springframework.data.mongodb.core.index.IndexOperations;
-import org.springframework.data.mongodb.core.index.VectorIndexOperations;
 import org.springframework.data.mongodb.core.mapping.MongoPersistentEntity;
 import org.springframework.lang.Nullable;
 import org.springframework.util.Assert;
@@ -210,11 +208,6 @@ private List<IndexInfo> getIndexData(MongoCursor<Document> cursor) {
 		});
 	}
 
-	@Override
-	public VectorIndexOperations vectorIndex() {
-		return new DefaultVectorIndexOperations(mongoOperations, collectionName, type);
-	}
-
 	@Nullable
 	public <T> T execute(CollectionCallback<T> callback) {
 

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/MongoTemplate.java

@@ -85,10 +85,13 @@
 import org.springframework.data.mongodb.core.convert.MongoWriter;
 import org.springframework.data.mongodb.core.convert.QueryMapper;
 import org.springframework.data.mongodb.core.convert.UpdateMapper;
+import org.springframework.data.mongodb.core.index.DefaultSearchIndexOperations;
 import org.springframework.data.mongodb.core.index.IndexOperations;
 import org.springframework.data.mongodb.core.index.IndexOperationsProvider;
 import org.springframework.data.mongodb.core.index.MongoMappingEventPublisher;
 import org.springframework.data.mongodb.core.index.MongoPersistentEntityIndexCreator;
+import org.springframework.data.mongodb.core.index.SearchIndexOperations;
+import org.springframework.data.mongodb.core.index.SearchIndexOperationsProvider;
 import org.springframework.data.mongodb.core.mapping.MongoMappingContext;
 import org.springframework.data.mongodb.core.mapping.MongoPersistentEntity;
 import org.springframework.data.mongodb.core.mapping.MongoPersistentProperty;
@@ -183,7 +186,7 @@
  * @author Jakub Zurawa
  */
 public class MongoTemplate
-		implements MongoOperations, ApplicationContextAware, IndexOperationsProvider, ReadPreferenceAware {
+		implements MongoOperations, ApplicationContextAware, IndexOperationsProvider, SearchIndexOperationsProvider, ReadPreferenceAware {
 
 	private static final Log LOGGER = LogFactory.getLog(MongoTemplate.class);
 	private static final WriteResultChecking DEFAULT_WRITE_RESULT_CHECKING = WriteResultChecking.NONE;
@@ -3010,6 +3013,21 @@ static RuntimeException potentiallyConvertRuntimeException(RuntimeException ex,
 		return resolved == null ? ex : resolved;
 	}
 
+	@Override
+	public SearchIndexOperations searchIndexOps(String collectionName) {
+		return searchIndexOps(null, collectionName);
+	}
+
+	@Override
+	public SearchIndexOperations searchIndexOps(Class<?> type) {
+		return new DefaultSearchIndexOperations(this, type);
+	}
+
+	@Override
+	public SearchIndexOperations searchIndexOps(Class<?> type, String collectionName) {
+		return new DefaultSearchIndexOperations(this, collectionName, type);
+	}
+
 	// Callback implementations
 
 	/**

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/DefaultVectorIndexOperations.java renamed to spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/DefaultSearchIndexOperations.java

@@ -26,23 +26,23 @@
 import org.springframework.data.mongodb.core.aggregation.Aggregation;
 import org.springframework.data.mongodb.core.aggregation.AggregationResults;
 import org.springframework.data.mongodb.core.convert.QueryMapper;
-import org.springframework.data.mongodb.core.index.VectorIndex.Filter;
+import org.springframework.data.mongodb.core.index.SearchIndex.Filter;
 import org.springframework.data.mongodb.core.mapping.MongoPersistentEntity;
 import org.springframework.lang.NonNull;
 import org.springframework.lang.Nullable;
 
 /**
  * @author Christoph Strobl
  */
-public class DefaultVectorIndexOperations extends DefaultIndexOperations implements VectorIndexOperations {
+public class DefaultSearchIndexOperations extends DefaultIndexOperations implements SearchIndexOperations {
 
-	private static final Log LOGGER = LogFactory.getLog(VectorIndexOperations.class);
+	private static final Log LOGGER = LogFactory.getLog(SearchIndexOperations.class);
 
-	public DefaultVectorIndexOperations(MongoOperations mongoOperations, Class<?> type) {
+	public DefaultSearchIndexOperations(MongoOperations mongoOperations, Class<?> type) {
 		this(mongoOperations, mongoOperations.getCollectionName(type), type);
 	}
 
-	public DefaultVectorIndexOperations(MongoOperations mongoOperations, String collectionName, @Nullable Class<?> type) {
+	public DefaultSearchIndexOperations(MongoOperations mongoOperations, String collectionName, @Nullable Class<?> type) {
 		super(mongoOperations, collectionName, type);
 	}
 
@@ -62,7 +62,7 @@ public boolean exists(String indexName) {
 	}
 
 	@Override
-	public void updateIndex(VectorIndex index) {
+	public void updateIndex(SearchIndex index) {
 
 		MongoPersistentEntity<?> entity = lookupPersistentEntity(type, collectionName);
 
@@ -106,10 +106,10 @@ public List<IndexInfo> getIndexInfo() {
 	}
 
 	@Override
-	public String ensureIndex(IndexDefinition indexDefinition) {
+	public String ensureIndex(SearchIndexDefinition indexDefinition) {
 
-		if (!(indexDefinition instanceof VectorIndex vsi)) {
-			return super.ensureIndex(indexDefinition);
+		if (!(indexDefinition instanceof SearchIndex vsi)) {
+			throw new IllegalStateException("Index definitions must be of type VectorIndex");
 		}
 
 		MongoPersistentEntity<?> entity = lookupPersistentEntity(type, collectionName);
@@ -129,7 +129,7 @@ public String ensureIndex(IndexDefinition indexDefinition) {
 	}
 
 	@NonNull
-	private Document createIndexDocument(VectorIndex vsi, MongoPersistentEntity<?> entity) {
+	private Document createIndexDocument(SearchIndex vsi, MongoPersistentEntity<?> entity) {
 
 		Document index = new Document(vsi.getIndexOptions());
 		Document definition = new Document();

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/IndexOperations.java

@@ -25,7 +25,7 @@
  * @author Christoph Strobl
  * @author Jens Schauder
  */
-public interface IndexOperations extends VectorIndexOperationsProvider {
+public interface IndexOperations {
 
 	/**
 	 * Ensure that an index for the provided {@link IndexDefinition} exists for the collection indicated by the entity

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/IndexOperationsAdapter.java

@@ -40,11 +40,6 @@ static IndexOperationsAdapter blocking(ReactiveIndexOperations reactiveIndexOper
 
 		return new IndexOperationsAdapter() {
 
-			@Override
-			public VectorIndexOperations vectorIndex() {
-				throw new IllegalStateException("currently not supported");
-			}
-
 			@Override
 			public String ensureIndex(IndexDefinition indexDefinition) {
 				return reactiveIndexOperations.ensureIndex(indexDefinition).block();

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/VectorIndex.java renamed to spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/SearchIndex.java

@@ -44,57 +44,58 @@
  *
  * @author Christoph Strobl
  */
-public class VectorIndex implements IndexDefinition {
+public class SearchIndex implements SearchIndexDefinition {
 
     private final String name;
     private String path;
     private int dimensions;
     private String similarity;
     private List<Filter> filters;
+    private String quantization = Quantization.NONE.name();
 
     /**
-     * Create a new {@link VectorIndex} instance.
+     * Create a new {@link SearchIndex} instance.
      *
      * @param name The name of the index.
      */
-    public VectorIndex(String name) {
+    public SearchIndex(String name) {
         this.name = name;
     }
 
     /**
-     * Create a new {@link VectorIndex} instance using similarity based on the angle between vectors.
+     * Create a new {@link SearchIndex} instance using similarity based on the angle between vectors.
      *
      * @param name The name of the index.
-     * @return new instance of {@link VectorIndex}.
+     * @return new instance of {@link SearchIndex}.
      */
-    public static VectorIndex cosine(String name) {
+    public static SearchIndex cosine(String name) {
 
-        VectorIndex idx = new VectorIndex(name);
+        SearchIndex idx = new SearchIndex(name);
         return idx.similarity(SimilarityFunction.COSINE);
     }
 
     /**
-     * Create a new {@link VectorIndex} instance using similarity based the distance between vector ends.
+     * Create a new {@link SearchIndex} instance using similarity based the distance between vector ends.
      *
      * @param name The name of the index.
-     * @return new instance of {@link VectorIndex}.
+     * @return new instance of {@link SearchIndex}.
      */
-    public static VectorIndex euclidean(String name) {
+    public static SearchIndex euclidean(String name) {
 
-        VectorIndex idx = new VectorIndex(name);
+        SearchIndex idx = new SearchIndex(name);
         return idx.similarity(SimilarityFunction.EUCLIDEAN);
     }
 
     /**
-     * Create a new {@link VectorIndex} instance using similarity based on based on both angle and magnitude of the
+     * Create a new {@link SearchIndex} instance using similarity based on based on both angle and magnitude of the
      * vectors.
      *
      * @param name The name of the index.
-     * @return new instance of {@link VectorIndex}.
+     * @return new instance of {@link SearchIndex}.
      */
-    public static VectorIndex dotProduct(String name) {
+    public static SearchIndex dotProduct(String name) {
 
-        VectorIndex idx = new VectorIndex(name);
+        SearchIndex idx = new SearchIndex(name);
         return idx.similarity(SimilarityFunction.DOT_PRODUCT);
     }
 
@@ -104,7 +105,7 @@ public static VectorIndex dotProduct(String name) {
      * @param path The path using dot notation.
      * @return this.
      */
-    public VectorIndex path(String path) {
+    public SearchIndex path(String path) {
 
         this.path = path;
         return this;
@@ -116,7 +117,7 @@ public VectorIndex path(String path) {
      * @param dimensions value between {@code 0} and {@code 4096}.
      * @return this.
      */
-    public VectorIndex dimensions(int dimensions) {
+    public SearchIndex dimensions(int dimensions) {
         this.dimensions = dimensions;
         return this;
     }
@@ -129,7 +130,7 @@ public VectorIndex dimensions(int dimensions) {
      * @see SimilarityFunction
      * @see #similarity(SimilarityFunction)
      */
-    public VectorIndex similarity(String similarity) {
+    public SearchIndex similarity(String similarity) {
         this.similarity = similarity;
         return this;
     }
@@ -140,17 +141,41 @@ public VectorIndex similarity(String similarity) {
      * @param similarity must not be {@literal null}.
      * @return this.
      */
-    public VectorIndex similarity(SimilarityFunction similarity) {
+    public SearchIndex similarity(SimilarityFunction similarity) {
         return similarity(similarity.getFunctionName());
     }
 
+
+    /**
+     * Quantization used.
+     *
+     * @param quantization should be one of {@literal none | scalar | binary}.
+     * @return this.
+     * @see Quantization
+     * @see #quantization(Quantization)
+     */
+    public SearchIndex quantization(String quantization) {
+        this.quantization = quantization;
+        return this;
+    }
+
+    /**
+     * Quntization used.
+     *
+     * @param quantization must not be {@literal null}.
+     * @return this.
+     */
+    public SearchIndex quantization(Quantization quantization) {
+        return similarity(quantization.getQuantizationName());
+    }
+
     /**
      * Add a {@link Filter} that can be used to narrow search scope.
      *
      * @param filter must not be {@literal null}.
      * @return this.
      */
-    public VectorIndex filter(Filter filter) {
+    public SearchIndex filter(Filter filter) {
 
         if (this.filters == null) {
             this.filters = new ArrayList<>(3);
@@ -167,21 +192,10 @@ public VectorIndex filter(Filter filter) {
      * @return this.
      * @see #filter(Filter)
      */
-    public VectorIndex filter(String path) {
+    public SearchIndex filter(String path) {
         return filter(new Filter(path));
     }
 
-    @Override
-    public Document getIndexKeys() {
-
-        // List<Document> fields = new ArrayList<>(filters.size()+1);
-        // fields.
-
-        // needs to be wrapped in new Document("definition", before sending to server
-        // return new Document("fields", fields);
-        return new Document();
-    }
-
     @Override
     public Document getIndexOptions() {
         return new Document("name", name).append("type", "vectorSearch");
@@ -224,4 +238,18 @@ public String getFunctionName() {
             return functionName;
         }
     }
+
+    public enum Quantization {
+        NONE("none"), SCALAR("scalar"), BINARY("binary");
+
+        String quantizationName;
+
+        Quantization(String quantizationName) {
+            this.quantizationName = quantizationName;
+        }
+
+        public String getQuantizationName() {
+            return quantizationName;
+        }
+    }
 }

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/SearchIndexDefinition.java

@@ -0,0 +1,31 @@
+/*
+ * Copyright 2011-2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.mongodb.core.index;
+
+import org.bson.Document;
+
+/**
+ * @author Marcin Grzejszczak
+ */
+public interface SearchIndexDefinition {
+
+	/**
+	 * Get the index properties such as {@literal unique},...
+	 *
+	 * @return never {@literal null}.
+	 */
+	Document getIndexOptions();
+}

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/VectorIndexOperations.java renamed to spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/SearchIndexOperations.java

@@ -20,11 +20,11 @@
 /**
  * @author Christoph Strobl
  */
-public interface VectorIndexOperations {
+public interface SearchIndexOperations {
 
-	String ensureIndex(IndexDefinition indexDefinition);
+	String ensureIndex(SearchIndexDefinition indexDefinition);
 
-	void updateIndex(VectorIndex index);
+	void updateIndex(SearchIndex index);
 
 	boolean exists(String indexName);
 

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/VectorIndexOperationsProvider.java renamed to spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/SearchIndexOperationsProvider.java

@@ -18,8 +18,11 @@
 /**
  * @author Christoph Strobl
  */
-public interface VectorIndexOperationsProvider {
+public interface SearchIndexOperationsProvider {
 
-    VectorIndexOperations vectorIndex();
+    SearchIndexOperations searchIndexOps(String collectionName);
 
+    SearchIndexOperations searchIndexOps(Class<?> type);
+
+    SearchIndexOperations searchIndexOps(Class<?> type, String collectionName);
 }