feat: add configurations for generative modules:

- Mistral
- Anyscale
- Databricks

Author: bevzzz

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

@@ -13,14 +13,20 @@
 import com.google.gson.stream.JsonToken;
 import com.google.gson.stream.JsonWriter;
 
+import io.weaviate.client6.v1.api.collections.generative.AnyscaleGenerative;
 import io.weaviate.client6.v1.api.collections.generative.CohereGenerative;
+import io.weaviate.client6.v1.api.collections.generative.DatabricksGenerative;
 import io.weaviate.client6.v1.api.collections.generative.DummyGenerative;
+import io.weaviate.client6.v1.api.collections.generative.MistralGenerative;
 import io.weaviate.client6.v1.internal.ObjectBuilder;
 import io.weaviate.client6.v1.internal.json.JsonEnum;
 
 public interface Generative {
   public enum Kind implements JsonEnum<Kind> {
+    ANYSCALE("generative-anyscale"),
     COHERE("generative-cohere"),
+    DATABRICKS("generative-databricks"),
+    MISTRAL("generative-mistral"),
     DUMMY("generative-dummy");
 
     private static final Map<String, Kind> jsonValueMap = JsonEnum.collectNames(Kind.values());
@@ -69,7 +75,10 @@ private final void addAdapter(Gson gson, Generative.Kind kind, Class<? extends G
     }
 
     private final void init(Gson gson) {
+      addAdapter(gson, Generative.Kind.ANYSCALE, AnyscaleGenerative.class);
       addAdapter(gson, Generative.Kind.COHERE, CohereGenerative.class);
+      addAdapter(gson, Generative.Kind.DATABRICKS, DatabricksGenerative.class);
+      addAdapter(gson, Generative.Kind.MISTRAL, MistralGenerative.class);
       addAdapter(gson, Generative.Kind.DUMMY, DummyGenerative.class);
     }
 

src/main/java/io/weaviate/client6/v1/api/collections/generative/AnyscaleGenerative.java

@@ -0,0 +1,65 @@
+package io.weaviate.client6.v1.api.collections.generative;
+
+import java.util.function.Function;
+
+import com.google.gson.annotations.SerializedName;
+
+import io.weaviate.client6.v1.api.collections.Generative;
+import io.weaviate.client6.v1.internal.ObjectBuilder;
+
+public record AnyscaleGenerative(
+    @SerializedName("baseURL") String baseUrl,
+    @SerializedName("model") String model,
+    @SerializedName("temperature") Float temperature) implements Generative {
+
+  @Override
+  public Kind _kind() {
+    return Generative.Kind.ANYSCALE;
+  }
+
+  @Override
+  public Object _self() {
+    return this;
+  }
+
+  public static AnyscaleGenerative of() {
+    return of(ObjectBuilder.identity());
+  }
+
+  public static AnyscaleGenerative of(Function<Builder, ObjectBuilder<AnyscaleGenerative>> fn) {
+    return fn.apply(new Builder()).build();
+  }
+
+  public AnyscaleGenerative(Builder builder) {
+    this(
+        builder.baseUrl,
+        builder.model,
+        builder.temperature);
+  }
+
+  public static class Builder implements ObjectBuilder<AnyscaleGenerative> {
+    private String baseUrl;
+    private String model;
+    private Float temperature;
+
+    public Builder baseUrl(String baseUrl) {
+      this.baseUrl = baseUrl;
+      return this;
+    }
+
+    public Builder model(String model) {
+      this.model = model;
+      return this;
+    }
+
+    public Builder temperature(float temperature) {
+      this.temperature = temperature;
+      return this;
+    }
+
+    @Override
+    public AnyscaleGenerative build() {
+      return new AnyscaleGenerative(this);
+    }
+  }
+}

src/main/java/io/weaviate/client6/v1/api/collections/generative/DatabricksGenerative.java

@@ -0,0 +1,81 @@
+package io.weaviate.client6.v1.api.collections.generative;
+
+import java.util.function.Function;
+
+import com.google.gson.annotations.SerializedName;
+
+import io.weaviate.client6.v1.api.collections.Generative;
+import io.weaviate.client6.v1.internal.ObjectBuilder;
+
+public record DatabricksGenerative(
+    @SerializedName("endpoint") String baseUrl,
+    @SerializedName("maxTokens") Integer maxTokens,
+    @SerializedName("topK") Integer topK,
+    @SerializedName("topP") Float topP,
+    @SerializedName("temperature") Float temperature) implements Generative {
+
+  @Override
+  public Kind _kind() {
+    return Generative.Kind.DATABRICKS;
+  }
+
+  @Override
+  public Object _self() {
+    return this;
+  }
+
+  public static DatabricksGenerative of() {
+    return of(ObjectBuilder.identity());
+  }
+
+  public static DatabricksGenerative of(Function<Builder, ObjectBuilder<DatabricksGenerative>> fn) {
+    return fn.apply(new Builder()).build();
+  }
+
+  public DatabricksGenerative(Builder builder) {
+    this(
+        builder.endpoint,
+        builder.maxTokens,
+        builder.topK,
+        builder.topP,
+        builder.temperature);
+  }
+
+  public static class Builder implements ObjectBuilder<DatabricksGenerative> {
+    private String endpoint;
+    private Integer maxTokens;
+    private Integer topK;
+    private Float topP;
+    private Float temperature;
+
+    public Builder endpoint(String endpoint) {
+      this.endpoint = endpoint;
+      return this;
+    }
+
+    public Builder maxTokens(int maxTokens) {
+      this.maxTokens = maxTokens;
+      return this;
+    }
+
+    public Builder topK(int topK) {
+      this.topK = topK;
+      return this;
+    }
+
+    public Builder topP(float topP) {
+      this.topP = topP;
+      return this;
+    }
+
+    public Builder temperature(float temperature) {
+      this.temperature = temperature;
+      return this;
+    }
+
+    @Override
+    public DatabricksGenerative build() {
+      return new DatabricksGenerative(this);
+    }
+  }
+}

src/main/java/io/weaviate/client6/v1/api/collections/generative/MistralGenerative.java

@@ -0,0 +1,73 @@
+package io.weaviate.client6.v1.api.collections.generative;
+
+import java.util.function.Function;
+
+import com.google.gson.annotations.SerializedName;
+
+import io.weaviate.client6.v1.api.collections.Generative;
+import io.weaviate.client6.v1.internal.ObjectBuilder;
+
+public record MistralGenerative(
+    @SerializedName("baseURL") String baseUrl,
+    @SerializedName("model") String model,
+    @SerializedName("maxTokens") Integer maxTokens,
+    @SerializedName("temperature") Float temperature) implements Generative {
+
+  @Override
+  public Kind _kind() {
+    return Generative.Kind.MISTRAL;
+  }
+
+  @Override
+  public Object _self() {
+    return this;
+  }
+
+  public static MistralGenerative of() {
+    return of(ObjectBuilder.identity());
+  }
+
+  public static MistralGenerative of(Function<Builder, ObjectBuilder<MistralGenerative>> fn) {
+    return fn.apply(new Builder()).build();
+  }
+
+  public MistralGenerative(Builder builder) {
+    this(
+        builder.baseUrl,
+        builder.model,
+        builder.maxTokens,
+        builder.temperature);
+  }
+
+  public static class Builder implements ObjectBuilder<MistralGenerative> {
+    private String baseUrl;
+    private String model;
+    private Integer maxTokens;
+    private Float temperature;
+
+    public Builder baseUrl(String baseUrl) {
+      this.baseUrl = baseUrl;
+      return this;
+    }
+
+    public Builder maxTokens(int maxTokens) {
+      this.maxTokens = maxTokens;
+      return this;
+    }
+
+    public Builder model(String model) {
+      this.model = model;
+      return this;
+    }
+
+    public Builder temperature(float temperature) {
+      this.temperature = temperature;
+      return this;
+    }
+
+    @Override
+    public MistralGenerative build() {
+      return new MistralGenerative(this);
+    }
+  }
+}