added sonnet 3.5 as a default classifier model for bedrock

awslabs · Jul 23, 2024 · e64c915 · e64c915
1 parent 2ebc7c6
commit e64c915
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/classifiers/bedrockClassifier.ts b/src/classifiers/bedrockClassifier.ts
@@ -1,5 +1,5 @@
 import {
-  BEDROCK_MODEL_ID_CLAUDE_3_SONNET,
+  BEDROCK_MODEL_ID_CLAUDE_3_5_SONNET,
   ClassifierResult,
   ConversationMessage,
   ParticipantRole,
@@ -79,7 +79,7 @@ export class BedrockClassifier extends Classifier{
 
     // Initialize default values or use provided options
     this.client = new BedrockRuntimeClient();
-    this.modelId = options.modelId || BEDROCK_MODEL_ID_CLAUDE_3_SONNET;
+    this.modelId = options.modelId || BEDROCK_MODEL_ID_CLAUDE_3_5_SONNET;
     // Initialize inferenceConfig only if it's provided in options
     this.inferenceConfig = {
       maxTokens: options.inferenceConfig?.maxTokens,

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -5,6 +5,7 @@ import { AccumulatorTransform } from "../utils/helpers";
 
 export const BEDROCK_MODEL_ID_CLAUDE_3_HAIKU = "anthropic.claude-3-haiku-20240307-v1:0";
 export const BEDROCK_MODEL_ID_CLAUDE_3_SONNET = "anthropic.claude-3-sonnet-20240229-v1:0";
+export const BEDROCK_MODEL_ID_CLAUDE_3_5_SONNET = "anthropic.claude-3-5-sonnet-20240620-v1:0";
 export const BEDROCK_MODEL_ID_LLAMA_3_70B = "meta.llama3-70b-instruct-v1:0";
 export const OPENAI_MODEL_ID_GPT_O_MINI = "gpt-4o-mini";