From e64c9153f70ac74898ca0fb158057607cd05bfd9 Mon Sep 17 00:00:00 2001
From: Anthony <brnaba@amazon.fr>
Date: Tue, 23 Jul 2024 15:41:44 +0200
Subject: [PATCH] added sonnet 3.5 as a default classifier model for bedrock

---
 src/classifiers/bedrockClassifier.ts | 4 ++--
 src/types/index.ts                   | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

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";