diff --git a/src/classifiers/bedrockClassifier.ts b/src/classifiers/bedrockClassifier.ts index 57a76a9..c224fc0 100644 --- 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 index d661e0a..2241260 100644 --- 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";