test: migrate to model2vec container for tests

Author: bevzzz

src/it/java/io/weaviate/containers/Container.java

@@ -14,7 +14,7 @@
 
 public class Container {
   public static final Weaviate WEAVIATE = Weaviate.createDefault();
-  public static final Transformers TRANSFORMERS = Transformers.createDefault();
+  public static final Model2Vec MODEL2VEC = Model2Vec.createDefault();
   public static final Img2VecNeural IMG2VEC_NEURAL = Img2VecNeural.createDefault();
   public static final MinIo MINIO = MinIo.createDefault();
 

…io/weaviate/containers/Transformers.java …va/io/weaviate/containers/Model2Vec.javasrc/it/java/io/weaviate/containers/Transformers.java renamed to src/it/java/io/weaviate/containers/Model2Vec.java src/it/java/io/weaviate/containers/Transformers.java renamed to src/it/java/io/weaviate/containers/Model2Vec.java

@@ -4,19 +4,19 @@
 
 import io.weaviate.client6.v1.api.collections.VectorConfig;
 
-public class Transformers extends GenericContainer<Transformers> {
-  public static final String VERSION = "sentence-transformers-all-MiniLM-L6-v2";
-  public static final String DOCKER_IMAGE = "cr.weaviate.io/semitechnologies/transformers-inference";
-  public static final String MODULE = VectorConfig.Kind.TEXT2VEC_TRANSFORMERS.jsonValue();
+public class Model2Vec extends GenericContainer<Model2Vec> {
+  public static final String VERSION = "minishlab-potion-retrieval-32M";
+  public static final String DOCKER_IMAGE = "cr.weaviate.io/semitechnologies/model2vec-inference";
+  public static final String MODULE = VectorConfig.Kind.TEXT2VEC_MODEL2VEC.jsonValue();
 
-  public static final String HOST_NAME = "transformers";
+  public static final String HOST_NAME = "model2vec";
   public static final String URL = HOST_NAME + ":8080";
 
-  static Transformers createDefault() {
+  static Model2Vec createDefault() {
     return new Builder().build();
   }
 
-  static Transformers.Builder custom() {
+  static Model2Vec.Builder custom() {
     return new Builder();
   }
 
@@ -27,16 +27,14 @@ public Builder() {
       this.versionTag = VERSION;
     }
 
-    public Transformers build() {
-      var container = new Transformers(DOCKER_IMAGE + ":" + versionTag);
-      container
-          .withEnv("ENABLE_CUDA", "0");
+    public Model2Vec build() {
+      var container = new Model2Vec(DOCKER_IMAGE + ":" + versionTag);
       container.withCreateContainerCmdModifier(cmd -> cmd.withHostName(HOST_NAME));
       return container;
     }
   }
 
-  public Transformers(String image) {
+  public Model2Vec(String image) {
     super(image);
   }
 }

src/it/java/io/weaviate/containers/Weaviate.java

@@ -112,9 +112,9 @@ public Builder withDefaultVectorizer(String module) {
       return this;
     }
 
-    public Builder withTransformersUrl(String url) {
-      addModules(Transformers.MODULE);
-      environment.put("TRANSFORMERS_INFERENCE_API", "http://" + url);
+    public Builder withModel2VecUrl(String url) {
+      addModules(Model2Vec.MODULE);
+      environment.put("MODEL2VEC_INFERENCE_API", "http://" + url);
       return this;
     }
 

src/it/java/io/weaviate/integration/SearchITest.java

@@ -42,18 +42,18 @@
 import io.weaviate.containers.Container;
 import io.weaviate.containers.Container.ContainerGroup;
 import io.weaviate.containers.Img2VecNeural;
-import io.weaviate.containers.Transformers;
+import io.weaviate.containers.Model2Vec;
 import io.weaviate.containers.Weaviate;
 
 public class SearchITest extends ConcurrentTest {
   private static final ContainerGroup compose = Container.compose(
       Weaviate.custom()
-          .withTransformersUrl(Transformers.URL)
+          .withModel2VecUrl(Model2Vec.URL)
           .withImageInference(Img2VecNeural.URL, Img2VecNeural.MODULE)
           .addModules("generative-dummy")
           .build(),
       Container.IMG2VEC_NEURAL,
-      Container.TRANSFORMERS);
+      Container.MODEL2VEC);
   @ClassRule // Bind containers to the lifetime of the test
   public static final TestRule _rule = compose.asTestRule();
   private static final WeaviateClient client = compose.getClient();
@@ -151,7 +151,7 @@ public void testNearText() throws IOException {
     client.collections.create(nsSongs,
         col -> col
             .properties(Property.text("title"))
-            .vectorConfig(VectorConfig.text2VecTransformers()));
+            .vectorConfig(VectorConfig.text2VecModel2Vec()));
 
     var songs = client.collections.use(nsSongs);
     var submarine = songs.data.insert(Map.of("title", "Yellow Submarine"));
@@ -173,7 +173,7 @@ public void testNearText() throws IOException {
 
   @Test
   public void testNearText_groupBy() throws IOException {
-    var vectorizer = VectorConfig.text2VecTransformers();
+    var vectorizer = VectorConfig.text2VecModel2Vec();
 
     var nsArtists = ns("Artists");
     client.collections.create(nsArtists,
@@ -370,7 +370,7 @@ public void testNearObject() throws IOException {
     client.collections.create(nsAnimals,
         collection -> collection
             .properties(Property.text("kind"))
-            .vectorConfig(VectorConfig.text2VecTransformers()));
+            .vectorConfig(VectorConfig.text2VecModel2Vec()));
 
     var animals = client.collections.use(nsAnimals);
 
@@ -399,7 +399,7 @@ public void testHybrid() throws IOException {
     client.collections.create(nsHobbies,
         collection -> collection
             .properties(Property.text("name"), Property.text("description"))
-            .vectorConfig(VectorConfig.text2VecTransformers()));
+            .vectorConfig(VectorConfig.text2VecModel2Vec()));
 
     var hobbies = client.collections.use(nsHobbies);
 
@@ -432,7 +432,7 @@ public void testBadRequest() throws IOException {
     client.collections.create(nsThings,
         collection -> collection
             .properties(Property.text("name"))
-            .vectorConfig(VectorConfig.text2VecTransformers()));
+            .vectorConfig(VectorConfig.text2VecModel2Vec()));
 
     var things = client.collections.use(nsThings);
     var balloon = things.data.insert(Map.of("name", "balloon"));
@@ -449,7 +449,7 @@ public void testBadRequest_async() throws Throwable {
       async.collections.create(nsThings,
           collection -> collection
               .properties(Property.text("name"))
-              .vectorConfig(VectorConfig.text2VecTransformers()))
+              .vectorConfig(VectorConfig.text2VecModel2Vec()))
           .join();
 
       var things = async.collections.use(nsThings);
@@ -470,7 +470,7 @@ public void testMetadataAll() throws IOException {
     client.collections.create(nsThings,
         c -> c
             .properties(Property.text("name"))
-            .vectorConfig(VectorConfig.text2VecTransformers(
+            .vectorConfig(VectorConfig.text2VecModel2Vec(
                 t2v -> t2v.sourceProperties("name"))));
 
     var things = client.collections.use(nsThings);
@@ -563,7 +563,7 @@ public void testGenerative_bm25() throws IOException {
         c -> c
             .properties(Property.text("title"))
             .generativeModule(new DummyGenerative())
-            .vectorConfig(VectorConfig.text2VecTransformers(
+            .vectorConfig(VectorConfig.text2VecModel2Vec(
                 t2v -> t2v.sourceProperties("title"))));
 
     var things = client.collections.use(nsThings);
@@ -604,7 +604,7 @@ public void testGenerative_bm25_groupBy() throws IOException {
         c -> c
             .properties(Property.text("title"))
             .generativeModule(new DummyGenerative())
-            .vectorConfig(VectorConfig.text2VecTransformers(
+            .vectorConfig(VectorConfig.text2VecModel2Vec(
                 t2v -> t2v.sourceProperties("title"))));
 
     var things = client.collections.use(nsThings);

src/main/java/io/weaviate/client6/v1/api/collections/VectorConfig.java

@@ -58,7 +58,7 @@ public enum Kind implements JsonEnum<Kind> {
     TEXT2VEC_GOOGLE("text2vec-google"),
     TEXT2VEC_GOOGLEAISTUDIO("text2vec-google"),
     TEXT2VEC_HUGGINGFACE("text2vec-huggingface"),
-    REF2VEC_CENTROID("text2vec-huggingface"),
+    REF2VEC_CENTROID("ref2vec-centroid"),
     TEXT2VEC_JINAAI("text2vec-jinaai"),
     TEXT2VEC_MISTRAL("text2vec-mistral"),
     TEXT2VEC_MORPH("text2vec-morph"),

src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Multi2VecJinaAiVectorizer.java

@@ -51,7 +51,7 @@ private static record Weights(
 
   @Override
   public VectorConfig.Kind _kind() {
-    return VectorConfig.Kind.MULTI2VEC_COHERE;
+    return VectorConfig.Kind.MULTI2VEC_JINAAI;
   }
 
   @Override

src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecDatabricksVectorizer.java

@@ -31,7 +31,7 @@ public record Text2VecDatabricksVectorizer(
 
   @Override
   public VectorConfig.Kind _kind() {
-    return VectorConfig.Kind.TEXT2VEC_MISTRAL;
+    return VectorConfig.Kind.TEXT2VEC_DATABRICKS;
   }
 
   @Override

src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecModel2VecVectorizer.java

@@ -30,7 +30,7 @@ public record Text2VecModel2VecVectorizer(
 
   @Override
   public VectorConfig.Kind _kind() {
-    return VectorConfig.Kind.TEXT2VEC_COHERE;
+    return VectorConfig.Kind.TEXT2VEC_MODEL2VEC;
   }
 
   @Override

src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecMorphVectorizer.java

@@ -25,7 +25,7 @@ public record Text2VecMorphVectorizer(
 
   @Override
   public VectorConfig.Kind _kind() {
-    return VectorConfig.Kind.TEXT2VEC_HUGGINGFACE;
+    return VectorConfig.Kind.TEXT2VEC_MORPH;
   }
 
   @Override