Add missing modules and module params for

- `baseURL` in `generative-anthropic`
- `videoIntervalSeconds` in `multi2vec-google`
- `outputEncoding` in `Multi2VecVoyageAIConfig`
- the entire `text2vec-nvidia` module

Author: tsmith023

src/collections/config/types/generative.ts

@@ -15,6 +15,7 @@ export type GenerativeAWSConfig = {
 };
 
 export type GenerativeAnthropicConfig = {
+  baseURL?: string;
   maxTokens?: number;
   model?: string;
   stopSequences?: string[];

src/collections/config/types/vectorizer.ts

@@ -35,6 +35,7 @@ export type Vectorizer =
   | 'text2vec-gpt4all'
   | 'text2vec-huggingface'
   | 'text2vec-jinaai'
+  | 'text2vec-nvidia'
   | 'text2vec-mistral'
   | 'text2vec-ollama'
   | 'text2vec-openai'
@@ -169,6 +170,8 @@ export type Multi2VecGoogleConfig = {
   textFields?: string[];
   /** The video fields used when vectorizing. */
   videoFields?: string[];
+  /** Length of a video interval in seconds. */
+  videoIntervalSeconds?: number;
   /** The model ID in use. */
   modelId?: string;
   /** The dimensionality of the vector once embedded. */
@@ -223,6 +226,8 @@ export type Multi2VecVoyageAIConfig = {
   imageFields?: string[];
   /** The model to use. */
   model?: string;
+  /** How the output from the model should be encoded on return. */
+  outputEncoding?: string;
   /** The text fields used when vectorizing. */
   textFields?: string[];
   /** Whether the input should be truncated to fit in the context window. */
@@ -363,6 +368,22 @@ export type Text2VecJinaAIConfig = {
 /** @deprecated Use `Text2VecJinaAIConfig` instead. */
 export type Text2VecJinaConfig = Text2VecJinaAIConfig;
 
+/**
+ * The configuration for text vectorization using the Nvidia module.
+ *
+ * See the [documentation](https://weaviate.io/developers/weaviate/model-providers/nvidia/embeddings) for detailed usage.
+ */
+export type Text2VecNvidiaConfig = {
+  /** The base URL to use where API requests should go. */
+  baseURL?: string;
+  /** The model to use. */
+  model?: string;
+  /** Whether to truncate when vectorising. */
+  truncate?: boolean;
+  /** Whether to vectorize the collection name. */
+  vectorizeCollectionName?: boolean;
+};
+
 /**
  * The configuration for text vectorization using the Mistral module.
  *
@@ -541,6 +562,8 @@ export type VectorizerConfigType<V> = V extends 'img2vec-neural'
   ? Text2VecHuggingFaceConfig | undefined
   : V extends 'text2vec-jinaai'
   ? Text2VecJinaAIConfig | undefined
+  : V extends 'text2vec-nvidia'
+  ? Text2VecNvidiaConfig | undefined
   : V extends 'text2vec-mistral'
   ? Text2VecMistralConfig | undefined
   : V extends 'text2vec-ollama'

src/collections/configure/types/vectorizer.ts

@@ -13,6 +13,7 @@ import {
   Text2VecHuggingFaceConfig,
   Text2VecJinaAIConfig,
   Text2VecMistralConfig,
+  Text2VecNvidiaConfig,
   Text2VecOllamaConfig,
   Text2VecOpenAIConfig,
   Text2VecTransformersConfig,
@@ -198,6 +199,8 @@ export type Text2VecHuggingFaceConfigCreate = Text2VecHuggingFaceConfig;
 
 export type Text2VecJinaAIConfigCreate = Text2VecJinaAIConfig;
 
+export type Text2VecNvidiaConfigCreate = Text2VecNvidiaConfig;
+
 export type Text2VecMistralConfigCreate = Text2VecMistralConfig;
 
 export type Text2VecOllamaConfigCreate = Text2VecOllamaConfig;
@@ -247,6 +250,8 @@ export type VectorizerConfigCreateType<V> = V extends 'img2vec-neural'
   ? Text2VecHuggingFaceConfigCreate | undefined
   : V extends 'text2vec-jinaai'
   ? Text2VecJinaAIConfigCreate | undefined
+  : V extends 'text2vec-nvidia'
+  ? Text2VecNvidiaConfigCreate | undefined
   : V extends 'text2vec-mistral'
   ? Text2VecMistralConfigCreate | undefined
   : V extends 'text2vec-ollama'

src/collections/configure/unit.test.ts

@@ -1161,6 +1161,47 @@ describe('Unit testing of the vectorizer factory class', () => {
     });
   });
 
+  it('should create the correct Text2VecNvidiaConfig type with defaults', () => {
+    const config = configure.vectorizer.text2VecNvidia();
+    expect(config).toEqual<VectorConfigCreate<never, undefined, 'hnsw', 'text2vec-nvidia'>>({
+      name: undefined,
+      vectorIndex: {
+        name: 'hnsw',
+        config: undefined,
+      },
+      vectorizer: {
+        name: 'text2vec-nvidia',
+        config: undefined,
+      },
+    });
+  });
+
+  it('should create the correct Text2VecNvidiaConfig type with all values', () => {
+    const config = configure.vectorizer.text2VecNvidia({
+      name: 'test',
+      baseURL: 'base-url',
+      model: 'model',
+      truncate: true,
+      vectorizeCollectionName: true,
+    });
+    expect(config).toEqual<VectorConfigCreate<never, 'test', 'hnsw', 'text2vec-nvidia'>>({
+      name: 'test',
+      vectorIndex: {
+        name: 'hnsw',
+        config: undefined,
+      },
+      vectorizer: {
+        name: 'text2vec-nvidia',
+        config: {
+          baseURL: 'base-url',
+          model: 'model',
+          truncate: true,
+          vectorizeCollectionName: true,
+        },
+      },
+    });
+  });
+
   it('should create the correct Text2VecMistralConfig type with defaults', () => {
     const config = configure.vectorizer.text2VecMistral();
     expect(config).toEqual<VectorConfigCreate<never, undefined, 'hnsw', 'text2vec-mistral'>>({

src/collections/configure/vectorizer.ts

@@ -518,6 +518,19 @@ export const vectorizer = {
       },
     });
   },
+  text2VecNvidia: <T, N extends string | undefined = undefined, I extends VectorIndexType = 'hnsw'>(
+    opts?: ConfigureTextVectorizerOptions<T, N, I, 'text2vec-nvidia'>
+  ): VectorConfigCreate<PrimitiveKeys<T>, N, I, 'text2vec-nvidia'> => {
+    const { name, sourceProperties, vectorIndexConfig, ...config } = opts || {};
+    return makeVectorizer(name, {
+      sourceProperties,
+      vectorIndexConfig,
+      vectorizerConfig: {
+        name: 'text2vec-nvidia',
+        config: Object.keys(config).length === 0 ? undefined : config,
+      },
+    });
+  },
   /**
    * Create a `VectorConfigCreate` object with the vectorizer set to `'text2vec-mistral'`.
    *