feat: Add support for claude sonnet 3.5 model through Amazon bedrock (#174)

shams858 · Shamsul Arefin · web-flow · commit c26b20a5d896 · 2025-02-24T08:36:33.000-08:00
Co-authored-by: Shamsul Arefin &lt;shamsul.arefin@iqvia.com&gt;
diff --git a/src/core/llm.py b/src/core/llm.py
@@ -39,6 +39,7 @@
     GroqModelName.LLAMA_33_70B: "llama-3.3-70b-versatile",
     GroqModelName.LLAMA_GUARD_3_8B: "llama-guard-3-8b",
     AWSModelName.BEDROCK_HAIKU: "anthropic.claude-3-5-haiku-20241022-v1:0",
+    AWSModelName.BEDROCK_SONNET: "anthropic.claude-3-5-sonnet-20240620-v1:0",
     OllamaModelName.OLLAMA_GENERIC: "ollama",
     FakeModelName.FAKE: "fake",
 }
diff --git a/src/schema/models.py b/src/schema/models.py
@@ -61,6 +61,7 @@ class AWSModelName(StrEnum):
     """https://docs.aws.amazon.com/bedrock/latest/userguide/models-supported.html"""
 
     BEDROCK_HAIKU = "bedrock-3.5-haiku"
+    BEDROCK_SONNET = "bedrock-3.5-sonnet"
 
 
 class OllamaModelName(StrEnum):

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,7 @@`
`39`	`39`	`GroqModelName.LLAMA_33_70B: "llama-3.3-70b-versatile",`
`40`	`40`	`GroqModelName.LLAMA_GUARD_3_8B: "llama-guard-3-8b",`
`41`	`41`	`AWSModelName.BEDROCK_HAIKU: "anthropic.claude-3-5-haiku-20241022-v1:0",`
	`42`	`+ AWSModelName.BEDROCK_SONNET: "anthropic.claude-3-5-sonnet-20240620-v1:0",`
`42`	`43`	`OllamaModelName.OLLAMA_GENERIC: "ollama",`
`43`	`44`	`FakeModelName.FAKE: "fake",`
`44`	`45`	`}`