Fix projectId in generative palm, add text2vec octo configure method, update CI images

Author: tsmith023

.github/workflows/main.yaml

@@ -7,8 +7,8 @@ on:
   pull_request:
 
 env:
-  WEAVIATE_124: 1.24.19
-  WEAVIATE_125: 1.25.5
+  WEAVIATE_124: 1.24.20
+  WEAVIATE_125: 1.25.7
 
 jobs:
   checks:

src/collections/config/types/generative.ts

@@ -59,7 +59,7 @@ export type GenerativePaLMConfig = {
   apiEndpoint?: string;
   maxOutputTokens?: number;
   modelId?: string;
-  projectId: string;
+  projectId?: string;
   temperature?: number;
   topK?: number;
   topP?: number;

src/collections/config/types/vectorizer.ts

@@ -24,6 +24,7 @@ export type Vectorizer =
   | 'text2vec-gpt4all'
   | 'text2vec-huggingface'
   | 'text2vec-jina'
+  | 'text2vec-octoai'
   | 'text2vec-ollama'
   | 'text2vec-openai'
   | 'text2vec-palm'
@@ -254,6 +255,20 @@ export type Text2VecJinaConfig = {
   vectorizeCollectionName?: boolean;
 };
 
+/**
+ * The configuration for text vectorization using OctoAI.
+ *
+ * See the [documentation](https://weaviate.io/developers/weaviate/modules/retriever-vectorizer-modules/text2vec-octoai) for detailed usage.
+ */
+export type Text2VecOctoAIConfig = {
+  /** The base URL to use where API requests should go. */
+  baseURL?: string;
+  /** The model to use. */
+  model?: string;
+  /** Whether to vectorize the collection name. */
+  vectorizeCollectionName?: boolean;
+};
+
 /**
  * The configuration for text vectorization using Ollama.
  *
@@ -378,6 +393,8 @@ export type VectorizerConfigType<V> = V extends 'img2vec-neural'
   ? Text2VecHuggingFaceConfig | undefined
   : V extends 'text2vec-jina'
   ? Text2VecJinaConfig | undefined
+  : V extends 'text2vec-octoai'
+  ? Text2VecOctoAIConfig | undefined
   : V extends 'text2vec-ollama'
   ? Text2VecOllamaConfig | undefined
   : V extends 'text2vec-openai'

src/collections/configure/generative.ts

@@ -182,10 +182,12 @@ export default {
    *
    * See the [documentation](https://weaviate.io/developers/weaviate/modules/reader-generator-modules/generative-palm) for detailed usage.
    *
-   * @param {GenerativePaLMConfigCreate} config The configuration for the `generative-palm` module.
+   * @param {GenerativePaLMConfigCreate} [config] The configuration for the `generative-palm` module.
    * @returns {ModuleConfig<'generative-palm', GenerativePaLMConfig>} The configuration object.
    */
-  palm: (config: GenerativePaLMConfigCreate): ModuleConfig<'generative-palm', GenerativePaLMConfig> => {
+  palm: (
+    config?: GenerativePaLMConfigCreate
+  ): ModuleConfig<'generative-palm', GenerativePaLMConfig | undefined> => {
     return {
       name: 'generative-palm',
       config,

src/collections/configure/types/vectorizer.ts

@@ -10,6 +10,7 @@ import {
   Text2VecGPT4AllConfig,
   Text2VecHuggingFaceConfig,
   Text2VecJinaConfig,
+  Text2VecOctoAIConfig,
   Text2VecOllamaConfig,
   Text2VecOpenAIConfig,
   Text2VecPalmConfig,
@@ -144,6 +145,8 @@ export type Text2VecHuggingFaceConfigCreate = Text2VecHuggingFaceConfig;
 
 export type Text2VecJinaConfigCreate = Text2VecJinaConfig;
 
+export type Text2VecOctoAIConfigCreate = Text2VecOctoAIConfig;
+
 export type Text2VecOllamaConfigCreate = Text2VecOllamaConfig;
 
 export type Text2VecOpenAIConfigCreate = Text2VecOpenAIConfig;
@@ -176,6 +179,8 @@ export type VectorizerConfigCreateType<V> = V extends 'img2vec-neural'
   ? Text2VecHuggingFaceConfigCreate | undefined
   : V extends 'text2vec-jina'
   ? Text2VecJinaConfigCreate | undefined
+  : V extends 'text2vec-octoai'
+  ? Text2VecOctoAIConfigCreate | undefined
   : V extends 'text2vec-ollama'
   ? Text2VecOllamaConfigCreate | undefined
   : V extends 'text2vec-openai'

src/collections/configure/unit.test.ts

@@ -801,6 +801,45 @@ describe('Unit testing of the vectorizer factory class', () => {
     });
   });
 
+  it('should create the correct Text2VecOctoAIConfig type with defaults', () => {
+    const config = configure.vectorizer.text2VecOctoAI();
+    expect(config).toEqual<VectorConfigCreate<never, undefined, 'hnsw', 'text2vec-octoai'>>({
+      name: undefined,
+      vectorIndex: {
+        name: 'hnsw',
+        config: undefined,
+      },
+      vectorizer: {
+        name: 'text2vec-octoai',
+        config: undefined,
+      },
+    });
+  });
+
+  it('should create the correct Text2VecOctoAIConfig type with all values', () => {
+    const config = configure.vectorizer.text2VecOctoAI({
+      name: 'test',
+      baseURL: 'base-url',
+      model: 'model',
+      vectorizeCollectionName: true,
+    });
+    expect(config).toEqual<VectorConfigCreate<never, 'test', 'hnsw', 'text2vec-octoai'>>({
+      name: 'test',
+      vectorIndex: {
+        name: 'hnsw',
+        config: undefined,
+      },
+      vectorizer: {
+        name: 'text2vec-octoai',
+        config: {
+          baseURL: 'base-url',
+          model: 'model',
+          vectorizeCollectionName: true,
+        },
+      },
+    });
+  });
+
   it('should create the correct Text2VecOllamaConfig type with defaults', () => {
     const config = configure.vectorizer.text2VecOllama();
     expect(config).toEqual<VectorConfigCreate<never, undefined, 'hnsw', 'text2vec-ollama'>>({
@@ -1235,14 +1274,10 @@ describe('Unit testing of the generative factory class', () => {
   });
 
   it('should create the correct GenerativePaLMConfig type with required & default values', () => {
-    const config = configure.generative.palm({
-      projectId: 'project-id',
-    });
-    expect(config).toEqual<ModuleConfig<'generative-palm', GenerativePaLMConfig>>({
+    const config = configure.generative.palm();
+    expect(config).toEqual<ModuleConfig<'generative-palm', undefined>>({
       name: 'generative-palm',
-      config: {
-        projectId: 'project-id',
-      },
+      config: undefined,
     });
   });
 

src/collections/configure/vectorizer.ts

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