feat: add static factories for vectorizer modules

Drop Contextionary vectorizer

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 Contextionary CONTEXTIONARY = Contextionary.createDefault();
+  public static final Transformers TRANSFORMERS = Transformers.createDefault();
   public static final Img2VecNeural IMG2VEC_NEURAL = Img2VecNeural.createDefault();
   public static final MinIo MINIO = MinIo.createDefault();
 

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

@@ -0,0 +1,42 @@
+package io.weaviate.containers;
+
+import org.testcontainers.containers.GenericContainer;
+
+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 static final String HOST_NAME = "transformers";
+  public static final String URL = HOST_NAME + ":8080";
+
+  static Transformers createDefault() {
+    return new Builder().build();
+  }
+
+  static Transformers.Builder custom() {
+    return new Builder();
+  }
+
+  public static class Builder {
+    private String versionTag;
+
+    public Builder() {
+      this.versionTag = VERSION;
+    }
+
+    public Transformers build() {
+      var container = new Transformers(DOCKER_IMAGE + ":" + versionTag);
+      container
+          .withEnv("ENABLE_CUDA", "0");
+      container.withCreateContainerCmdModifier(cmd -> cmd.withHostName(HOST_NAME));
+      return container;
+    }
+  }
+
+  public Transformers(String image) {
+    super(image);
+  }
+}

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

@@ -112,9 +112,9 @@ public Builder withDefaultVectorizer(String module) {
       return this;
     }
 
-    public Builder withContextionaryUrl(String url) {
-      addModules(Contextionary.MODULE);
-      environment.put("CONTEXTIONARY_URL", url);
+    public Builder withTransformersUrl(String url) {
+      addModules(Transformers.MODULE);
+      environment.put("TRANSFORMERS_INFERENCE_API", url);
       return this;
     }
 
@@ -138,6 +138,7 @@ public Builder withFilesystemBackup(String fsPath) {
       environment.put("BACKUP_FILESYSTEM_PATH", fsPath);
       return this;
     }
+
     public Builder withAdminUsers(String... admins) {
       adminUsers.addAll(Arrays.asList(admins));
       return this;

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

@@ -41,19 +41,19 @@
 import io.weaviate.client6.v1.api.collections.vectorindex.MultiVector;
 import io.weaviate.containers.Container;
 import io.weaviate.containers.Container.ContainerGroup;
-import io.weaviate.containers.Contextionary;
 import io.weaviate.containers.Img2VecNeural;
+import io.weaviate.containers.Transformers;
 import io.weaviate.containers.Weaviate;
 
 public class SearchITest extends ConcurrentTest {
   private static final ContainerGroup compose = Container.compose(
       Weaviate.custom()
-          .withContextionaryUrl(Contextionary.URL)
+          .withTransformersUrl(Transformers.URL)
           .withImageInference(Img2VecNeural.URL, Img2VecNeural.MODULE)
           .addModules("generative-dummy")
           .build(),
       Container.IMG2VEC_NEURAL,
-      Container.CONTEXTIONARY);
+      Container.TRANSFORMERS);
   @ClassRule // Bind containers to the lifetime of the test
   public static final TestRule _rule = compose.asTestRule();
   private static final WeaviateClient client = compose.getClient();