Add support for generative-anthropic

Author: tsmith023

src/collections/config/types/generative.ts

@@ -14,6 +14,15 @@ export type GenerativeAWSConfig = {
   endpoint?: string;
 };
 
+export type GenerativeAnthropicConfig = {
+  maxTokens?: number;
+  model?: string;
+  stopSequences?: string[];
+  temperature?: number;
+  topK?: number;
+  topP?: number;
+};
+
 export type GenerativeAnyscaleConfig = {
   model?: string;
   temperature?: number;
@@ -83,6 +92,7 @@ export type GenerativeConfigType<G> = G extends 'generative-openai'
   : Record<string, any> | undefined;
 
 export type GenerativeSearch =
+  | 'generative-anthropic'
   | 'generative-anyscale'
   | 'generative-aws'
   | 'generative-mistral'

src/collections/configure/generative.ts

@@ -1,5 +1,6 @@
 import {
   GenerativeAWSConfig,
+  GenerativeAnthropicConfig,
   GenerativeAnyscaleConfig,
   GenerativeAzureOpenAIConfig,
   GenerativeCohereConfig,
@@ -12,6 +13,7 @@ import {
 } from '../config/types/index.js';
 import {
   GenerativeAWSConfigCreate,
+  GenerativeAnthropicConfigCreate,
   GenerativeAnyscaleConfigCreate,
   GenerativeAzureOpenAIConfigCreate,
   GenerativeCohereConfigCreate,
@@ -23,12 +25,28 @@ import {
 } from '../index.js';
 
 export default {
+  /**
+   * Create a `ModuleConfig<'generative-anthropic', GenerativeAnthropicConfig | undefined>` object for use when performing AI generation using the `generative-anthropic` module.
+   *
+   * See the [documentation](https://weaviate.io/developers/weaviate/modules/reader-generator-modules/generative-anthropic) for detailed usage.
+   *
+   * @param {GenerativeAnthropicConfigCreate} [config] The configuration for the `generative-anthropic` module.
+   * @returns {ModuleConfig<'generative-anthropic', GenerativeAnthropicConfig | undefined>} The configuration object.
+   */
+  anthropic(
+    config?: GenerativeAnthropicConfigCreate
+  ): ModuleConfig<'generative-anthropic', GenerativeAnthropicConfig | undefined> {
+    return {
+      name: 'generative-anthropic',
+      config,
+    };
+  },
   /**
    * Create a `ModuleConfig<'generative-anyscale', GenerativeAnyscaleConfig | undefined>` object for use when performing AI generation using the `generative-anyscale` module.
    *
    * See the [documentation](https://weaviate.io/developers/weaviate/modules/reader-generator-modules/generative-anyscale) for detailed usage.
    *
-   * @param {GenerativeAnyscaleConfigCreate} config The configuration for the `generative-aws` module.
+   * @param {GenerativeAnyscaleConfigCreate} [config] The configuration for the `generative-aws` module.
    * @returns {ModuleConfig<'generative-anyscale', GenerativeAnyscaleConfig | undefined>} The configuration object.
    */
   anyscale(

src/collections/configure/types/generative.ts

@@ -1,5 +1,6 @@
 import {
   GenerativeAWSConfig,
+  GenerativeAnthropicConfig,
   GenerativeAnyscaleConfig,
   GenerativeMistralConfig,
   GenerativeOctoAIConfig,
@@ -20,6 +21,8 @@ export type GenerativeOpenAIConfigCreate = GenerativeOpenAIConfigBaseCreate & {
   model?: string;
 };
 
+export type GenerativeAnthropicConfigCreate = GenerativeAnthropicConfig;
+
 export type GenerativeAzureOpenAIConfigCreate = GenerativeOpenAIConfigBaseCreate & {
   resourceName: string;
   deploymentId: string;

src/collections/configure/unit.test.ts

@@ -1,5 +1,6 @@
 import {
   GenerativeAWSConfig,
+  GenerativeAnthropicConfig,
   GenerativeAnyscaleConfig,
   GenerativeAzureOpenAIConfig,
   GenerativeCohereConfig,
@@ -1066,6 +1067,36 @@ describe('Unit testing of the vectorizer factory class', () => {
 });
 
 describe('Unit testing of the generative factory class', () => {
+  it('should create the correct GenerativeAnthropicConfig type with required & default values', () => {
+    const config = configure.generative.anthropic();
+    expect(config).toEqual<ModuleConfig<'generative-anthropic', GenerativeAnthropicConfig | undefined>>({
+      name: 'generative-anthropic',
+      config: undefined,
+    });
+  });
+
+  it('should create the correct GenerativeAnthropicConfig type with all values', () => {
+    const config = configure.generative.anthropic({
+      maxTokens: 100,
+      model: 'model',
+      stopSequences: ['stop1', 'stop2'],
+      temperature: 0.5,
+      topK: 10,
+      topP: 0.8,
+    });
+    expect(config).toEqual<ModuleConfig<'generative-anthropic', GenerativeAnthropicConfig>>({
+      name: 'generative-anthropic',
+      config: {
+        maxTokens: 100,
+        model: 'model',
+        stopSequences: ['stop1', 'stop2'],
+        temperature: 0.5,
+        topK: 10,
+        topP: 0.8,
+      },
+    });
+  });
+
   it('should create the correct GenerativeAnyscaleConfig type with required & default values', () => {
     const config = configure.generative.anyscale();
     expect(config).toEqual<ModuleConfig<'generative-anyscale', GenerativeAnyscaleConfig | undefined>>({