Update Default Models Anthropic/Bedrock (#1770)

* feat(configs-drivers-amazon-bedrock): update default models for prompt driver, embedding driver, image generation driver

* feat(configs-drivers-anthropic): update default models

Author: collindutter

docs/examples/src/talk_to_a_document.py

@@ -28,7 +28,7 @@ def on_after_run(_: BaseTask) -> None:
 agent = Agent(
     tasks=[
         PromptTask(
-            prompt_driver=AnthropicPromptDriver(model="claude-3-5-sonnet-20240620", max_attempts=0),
+            prompt_driver=AnthropicPromptDriver(model="claude-3-7-sonnet-latest", max_attempts=0),
             on_before_run=on_before_run,
             on_after_run=on_after_run,
             input=[

docs/griptape-framework/drivers/src/image_generation_drivers_3.py

@@ -7,7 +7,7 @@
 
 driver = AmazonBedrockImageGenerationDriver(
     image_generation_model_driver=model_driver,
-    model="amazon.titan-image-generator-v1",
+    model="amazon.titan-image-generator-v2:0",
 )
 
 

docs/griptape-framework/structures/src/config_defaults.py

@@ -8,5 +8,5 @@
 
 Defaults.drivers_config = AnthropicDriversConfig()
 anthropic_agent = Agent(
-    prompt_driver=AnthropicPromptDriver(model="claude-3-5-sonnet-20240620"),  # Override the default prompt driver
+    prompt_driver=AnthropicPromptDriver(model="claude-3-7-sonnet-latest"),  # Override the default prompt driver
 )

griptape/configs/drivers/amazon_bedrock_drivers_config.py

@@ -27,22 +27,24 @@ class AmazonBedrockDriversConfig(DriversConfig):
 
     @lazy_property()
     def prompt_driver(self) -> AmazonBedrockPromptDriver:
-        return AmazonBedrockPromptDriver(session=self.session, model="anthropic.claude-3-5-sonnet-20240620-v1:0")
+        return AmazonBedrockPromptDriver(session=self.session, model="anthropic.claude-3-7-sonnet-20250219-v1:0")
 
     @lazy_property()
     def embedding_driver(self) -> AmazonBedrockTitanEmbeddingDriver:
-        return AmazonBedrockTitanEmbeddingDriver(session=self.session, model="amazon.titan-embed-text-v1")
+        return AmazonBedrockTitanEmbeddingDriver(session=self.session, model="amazon.titan-embed-text-v2:0")
 
     @lazy_property()
     def image_generation_driver(self) -> AmazonBedrockImageGenerationDriver:
         return AmazonBedrockImageGenerationDriver(
             session=self.session,
-            model="amazon.titan-image-generator-v1",
+            model="amazon.titan-image-generator-v2:0",
             image_generation_model_driver=BedrockTitanImageGenerationModelDriver(),
         )
 
     @lazy_property()
     def vector_store_driver(self) -> LocalVectorStoreDriver:
         return LocalVectorStoreDriver(
-            embedding_driver=AmazonBedrockTitanEmbeddingDriver(session=self.session, model="amazon.titan-embed-text-v1")
+            embedding_driver=AmazonBedrockTitanEmbeddingDriver(
+                session=self.session, model="amazon.titan-embed-text-v2:0"
+            )
         )

griptape/configs/drivers/anthropic_drivers_config.py

@@ -9,4 +9,4 @@
 class AnthropicDriversConfig(DriversConfig):
     @lazy_property()
     def prompt_driver(self) -> AnthropicPromptDriver:
-        return AnthropicPromptDriver(model="claude-3-5-sonnet-20240620")
+        return AnthropicPromptDriver(model="claude-3-7-sonnet-latest")

griptape/tokenizers/amazon_bedrock_tokenizer.py

@@ -33,6 +33,7 @@ class AmazonBedrockTokenizer(BaseTokenizer):
         "amazon.titan-text-premier-v1": 32000,
     }
     MODEL_PREFIXES_TO_MAX_OUTPUT_TOKENS = {
+        "anthropic.claude-3-7": 8192,
         "anthropic.claude-3-5": 8192,
         "anthropic.claude": 4096,
         "cohere": 4096,

tests/unit/configs/drivers/test_amazon_bedrock_drivers_config.py

@@ -29,7 +29,7 @@ def test_to_dict(self, config):
                 "type": "LocalConversationMemoryDriver",
                 "persist_file": None,
             },
-            "embedding_driver": {"model": "amazon.titan-embed-text-v1", "type": "AmazonBedrockTitanEmbeddingDriver"},
+            "embedding_driver": {"model": "amazon.titan-embed-text-v2:0", "type": "AmazonBedrockTitanEmbeddingDriver"},
             "image_generation_driver": {
                 "image_generation_model_driver": {
                     "cfg_scale": 7,
@@ -39,13 +39,13 @@ def test_to_dict(self, config):
                 },
                 "image_height": 512,
                 "image_width": 512,
-                "model": "amazon.titan-image-generator-v1",
+                "model": "amazon.titan-image-generator-v2:0",
                 "seed": None,
                 "type": "AmazonBedrockImageGenerationDriver",
             },
             "prompt_driver": {
                 "max_tokens": None,
-                "model": "anthropic.claude-3-5-sonnet-20240620-v1:0",
+                "model": "anthropic.claude-3-7-sonnet-20250219-v1:0",
                 "stream": False,
                 "temperature": 0.1,
                 "type": "AmazonBedrockPromptDriver",
@@ -56,7 +56,7 @@ def test_to_dict(self, config):
             },
             "vector_store_driver": {
                 "embedding_driver": {
-                    "model": "amazon.titan-embed-text-v1",
+                    "model": "amazon.titan-embed-text-v2:0",
                     "type": "AmazonBedrockTitanEmbeddingDriver",
                 },
                 "type": "LocalVectorStoreDriver",
@@ -85,7 +85,7 @@ def test_to_dict_with_values(self, config_with_values):
                 "type": "LocalConversationMemoryDriver",
                 "persist_file": None,
             },
-            "embedding_driver": {"model": "amazon.titan-embed-text-v1", "type": "AmazonBedrockTitanEmbeddingDriver"},
+            "embedding_driver": {"model": "amazon.titan-embed-text-v2:0", "type": "AmazonBedrockTitanEmbeddingDriver"},
             "image_generation_driver": {
                 "image_generation_model_driver": {
                     "cfg_scale": 7,
@@ -95,13 +95,13 @@ def test_to_dict_with_values(self, config_with_values):
                 },
                 "image_height": 512,
                 "image_width": 512,
-                "model": "amazon.titan-image-generator-v1",
+                "model": "amazon.titan-image-generator-v2:0",
                 "seed": None,
                 "type": "AmazonBedrockImageGenerationDriver",
             },
             "prompt_driver": {
                 "max_tokens": None,
-                "model": "anthropic.claude-3-5-sonnet-20240620-v1:0",
+                "model": "anthropic.claude-3-7-sonnet-20250219-v1:0",
                 "stream": False,
                 "temperature": 0.1,
                 "type": "AmazonBedrockPromptDriver",
@@ -112,7 +112,7 @@ def test_to_dict_with_values(self, config_with_values):
             },
             "vector_store_driver": {
                 "embedding_driver": {
-                    "model": "amazon.titan-embed-text-v1",
+                    "model": "amazon.titan-embed-text-v2:0",
                     "type": "AmazonBedrockTitanEmbeddingDriver",
                 },
                 "type": "LocalVectorStoreDriver",

tests/unit/configs/drivers/test_anthropic_drivers_config.py

@@ -21,7 +21,7 @@ def test_to_dict(self, config):
                 "temperature": 0.1,
                 "max_tokens": 1000,
                 "stream": False,
-                "model": "claude-3-5-sonnet-20240620",
+                "model": "claude-3-7-sonnet-latest",
                 "top_p": 0.999,
                 "top_k": 250,
                 "use_native_tools": True,