spike: overload vectorIndex method to partially override the default vector config

bevzzz · bevzzz · commit b721a52ca21a · 2025-09-09T19:22:41.000+02:00
diff --git a/src/main/java/io/weaviate/client6/v1/api/collections/VectorIndex.java b/src/main/java/io/weaviate/client6/v1/api/collections/VectorIndex.java
@@ -3,6 +3,7 @@
 import java.io.IOException;
 import java.util.EnumMap;
 import java.util.Map;
+import java.util.function.Function;
 
 import com.google.gson.Gson;
 import com.google.gson.JsonParser;
@@ -15,11 +16,17 @@
 
 import io.weaviate.client6.v1.api.collections.vectorindex.Flat;
 import io.weaviate.client6.v1.api.collections.vectorindex.Hnsw;
+import io.weaviate.client6.v1.internal.ObjectBuilder;
 import io.weaviate.client6.v1.internal.json.JsonEnum;
 
 public interface VectorIndex {
   static final String DEFAULT_VECTOR_NAME = "default";
   static final VectorIndex DEFAULT_VECTOR_INDEX = Hnsw.of();
+  static final Function<Function<Hnsw.Builder, ObjectBuilder<Hnsw>>, VectorIndex> DEFAULT_VECTOR_INDEX_FUNC = Hnsw::of;
+
+  static VectorIndex createDefault(Function<Hnsw.Builder, ObjectBuilder<Hnsw>> fn) {
+    return DEFAULT_VECTOR_INDEX_FUNC.apply(fn);
+  }
 
   public enum Kind implements JsonEnum<Kind> {
     HNSW("hnsw"),
diff --git a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Img2VecNeuralVectorizer.java b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Img2VecNeuralVectorizer.java
@@ -9,6 +9,7 @@
 
 import io.weaviate.client6.v1.api.collections.VectorIndex;
 import io.weaviate.client6.v1.api.collections.Vectorizer;
+import io.weaviate.client6.v1.api.collections.vectorindex.Hnsw;
 import io.weaviate.client6.v1.internal.ObjectBuilder;
 
 public record Img2VecNeuralVectorizer(
@@ -54,6 +55,17 @@ public Builder imageFields(String... fields) {
       return imageFields(Arrays.asList(fields));
     }
 
+    /**
+     * Override default vector index configuration.
+     *
+     * <a href=
+     * "https://docs.weaviate.io/weaviate/config-refs/indexing/vector-index#hnsw-index-parameters">HNSW</a>
+     * is the default vector index.
+     */
+    public Builder vectorIndex(Function<Hnsw.Builder, ObjectBuilder<Hnsw>> fn) {
+      return vectorIndex(VectorIndex.createDefault(fn));
+    }
+
     /**
      * Override default vector index configuration.
      *
