From daef2eecdd4b26b6a96820b612fb2388195edcae Mon Sep 17 00:00:00 2001
From: Logan Attwood <logan.attwood@gmail.com>
Date: Wed, 6 Nov 2024 12:28:24 -0400
Subject: [PATCH] Add Sonnet & Haiku ModelSettings for Bedrock

---
 aider/models.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/aider/models.py b/aider/models.py
index 73b1d04ff..95a4483a0 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -299,6 +299,23 @@ MODEL_SETTINGS = [
         cache_control=True,
         reminder="user",
     ),
+    ModelSettings(
+        "bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0",
+        "diff",
+        weak_model_name="bedrock/anthropic.claude-3-5-haiku-20241022-v1:0",
+        editor_model_name="bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0",
+        editor_edit_format="editor-diff",
+        use_repo_map=True,
+        examples_as_sys_msg=True,
+        extra_params={
+            "extra_headers": {
+                "anthropic-beta": ANTHROPIC_BETA_HEADER,
+            },
+            "max_tokens": 8192,
+        },
+        cache_control=True,
+        reminder="user",
+    ),
     ModelSettings(
         "anthropic/claude-3-5-sonnet-latest",
         "diff",
@@ -357,6 +374,18 @@ MODEL_SETTINGS = [
         },
         cache_control=True,
     ),
+    ModelSettings(
+        "bedrock/anthropic.claude-3-5-haiku-20241022-v1:0",
+        "diff",
+        weak_model_name="bedrock/anthropic.claude-3-5-haiku-20241022-v1:0",
+        use_repo_map=True,
+        extra_params={
+            "extra_headers": {
+                "anthropic-beta": ANTHROPIC_BETA_HEADER,
+            },
+        },
+        cache_control=True,
+    ),
     ModelSettings(
         "claude-3-5-haiku-20241022",
         "diff",