feat: Updates to vectorizer & generative module config options

Author: databyjp

src/Weaviate.Client/Configure/Vectorizer.cs

@@ -356,6 +356,16 @@ public static VectorConfigBuilder Text2VecDatabricks(
                 }
             );
 
+        public static VectorConfigBuilder Text2VecGpt4All(
+            bool? vectorizeCollectionName = null
+        ) =>
+            new(
+                new Vectorizer.Text2VecGpt4All
+                {
+                    VectorizeCollectionName = vectorizeCollectionName,
+                }
+            );        
+        
         public static VectorConfigBuilder Text2VecHuggingFace(
             string? endpointURL = null,
             string? model = null,
@@ -426,6 +436,16 @@ public static VectorConfigBuilder Text2VecMistral(
                 }
             );
 
+        public static VectorConfigBuilder Text2VecModel2Vec(
+            bool? vectorizeCollectionName = null
+        ) =>
+            new(
+                new Vectorizer.Text2VecModel2Vec
+                {
+                    VectorizeCollectionName = vectorizeCollectionName,
+                }
+            );        
+        
         public static VectorConfigBuilder Text2VecOllama(
             string? apiEndpoint = null,
             string? model = null,
@@ -516,12 +536,14 @@ public static VectorConfigBuilder Text2VecVoyageAI(
 
         public static VectorConfigBuilder Text2VecJinaAI(
             string? model = null,
+            int? dimensions = null,
             bool? vectorizeCollectionName = null
         ) =>
             new(
                 new Vectorizer.Text2VecJinaAI
                 {
                     Model = model,
+                    Dimensions = dimensions,
                     VectorizeCollectionName = vectorizeCollectionName,
                 }
             );

src/Weaviate.Client/Models/GenerativeConfig.cs

@@ -20,11 +20,11 @@ public record Custom : IGenerativeConfig
     public abstract record OpenAIBase : IGenerativeConfig
     {
         public string? BaseURL { get; set; }
-        public int? FrequencyPenaltyProperty { get; set; }
-        public int? MaxTokensProperty { get; set; }
-        public int? PresencePenaltyProperty { get; set; }
-        public double? TemperatureProperty { get; set; }
-        public double? TopPProperty { get; set; }
+        public int? FrequencyPenalty { get; set; }
+        public int? MaxTokens { get; set; }
+        public int? PresencePenalty { get; set; }
+        public double? Temperature { get; set; }
+        public double? TopP { get; set; }
         public abstract string Type { get; }
     }
 
@@ -68,12 +68,12 @@ public record Cohere : IGenerativeConfig
         public const string TypeValue = "generative-cohere";
         public string Type => TypeValue;
 
-        public int? KProperty { get; set; }
+        public int? K { get; set; }
         public string? Model { get; set; }
-        public int? MaxTokensProperty { get; set; }
-        public string? ReturnLikelihoodsProperty { get; set; }
-        public string[]? StopSequencesProperty { get; set; }
-        public double? TemperatureProperty { get; set; }
+        public int? MaxTokens { get; set; }
+        public string? ReturnLikelihoods { get; set; }
+        public string[]? StopSequences { get; set; }
+        public double? Temperature { get; set; }
     }
 
     public record Databricks : IGenerativeConfig

src/Weaviate.Client/Models/Vectorizer.Declarations.cs

@@ -221,6 +221,18 @@ public Text2VecDatabricks()
             : base(IdentifierValue) { }
     }
 
+    /// <summary>
+    /// The configuration for text vectorization using the GPT4All module.
+    /// See the documentation for detailed usage.
+    /// </summary>
+    public partial record Text2VecGpt4All : VectorizerConfig
+    {
+        public const string IdentifierValue = "text2vec-gpt4all";
+
+        public Text2VecGpt4All()
+            : base(IdentifierValue) { }
+    }    
+    
     /// <summary>
     /// The configuration for text vectorization using the HuggingFace module.
     /// See the documentation for detailed usage.
@@ -285,6 +297,18 @@ public Text2VecMistral()
             : base(IdentifierValue) { }
     }
 
+    /// <summary>
+    /// The configuration for text vectorization using the Model2Vec module.
+    /// See the documentation for detailed usage.
+    /// </summary>
+    public partial record Text2VecModel2Vec : VectorizerConfig
+    {
+        public const string IdentifierValue = "text2vec-model2vec";
+
+        public Text2VecModel2Vec()
+            : base(IdentifierValue) { }
+    }
+
     /// <summary>
     /// The configuration for text vectorization using the Ollama module.
     /// See the documentation for detailed usage.

src/Weaviate.Client/Models/Vectorizer.cs

@@ -190,7 +190,7 @@ public partial record Text2VecDatabricks
         public bool? VectorizeCollectionName { get; set; } = null;
     }
 
-    public partial record Text2VecGPT4All
+    public partial record Text2VecGpt4All
     {
         public bool? VectorizeCollectionName { get; set; } = null;
     }
@@ -210,6 +210,7 @@ public partial record Text2VecHuggingFace
     public partial record Text2VecJinaAI
     {
         public string? Model { get; set; } = null;
+        public int? Dimensions { get; set; } = null;
         public bool? VectorizeCollectionName { get; set; } = null;
     }
 
@@ -237,7 +238,12 @@ public partial record Multi2VecNvidia
     public partial record Text2VecMistral
     {
         public string? BaseURL { get; set; } = null;
-        public string? Model { get; set; } = null;
+        public string? Model { get; set; } = null;        
+        public bool? VectorizeCollectionName { get; set; } = null;
+    }
+    
+    public partial record Text2VecModel2Vec
+    {
         public bool? VectorizeCollectionName { get; set; } = null;
     }
 

src/Weaviate.Client/Models/Vectorizers/Factory.cs

@@ -23,10 +23,12 @@ internal static class VectorizerConfigFactory
         { Vectorizer.Text2VecAzureOpenAI.IdentifierValue, typeof(Vectorizer.Text2VecAzureOpenAI) },
         { Vectorizer.Text2VecCohere.IdentifierValue, typeof(Vectorizer.Text2VecCohere) },
         { Vectorizer.Text2VecDatabricks.IdentifierValue, typeof(Vectorizer.Text2VecDatabricks) },
+        { Vectorizer.Text2VecGpt4All.IdentifierValue, typeof(Vectorizer.Text2VecGpt4All) },        
         { Vectorizer.Text2VecHuggingFace.IdentifierValue, typeof(Vectorizer.Text2VecHuggingFace) },
         { Vectorizer.Text2VecJinaAI.IdentifierValue, typeof(Vectorizer.Text2VecJinaAI) },
         { Vectorizer.Text2VecNvidia.IdentifierValue, typeof(Vectorizer.Text2VecNvidia) },
         { Vectorizer.Text2VecMistral.IdentifierValue, typeof(Vectorizer.Text2VecMistral) },
+        { Vectorizer.Text2VecModel2Vec.IdentifierValue, typeof(Vectorizer.Text2VecModel2Vec) },
         { Vectorizer.Text2VecOllama.IdentifierValue, typeof(Vectorizer.Text2VecOllama) },
         { Vectorizer.Text2VecOpenAI.IdentifierValue, typeof(Vectorizer.Text2VecOpenAI) },
         { Vectorizer.Text2VecGoogle.IdentifierValue, typeof(Vectorizer.Text2VecGoogle) },