Add sonnet v2

AutoCodeRoverSG · Feb 6, 2025 · 2c4c543 · 2c4c543
1 parent 3536ac2
commit 2c4c543
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -300,7 +300,8 @@ The current list of supported models:
 |                | gpt-3.5-turbo-16k-0613 | --model gpt-3.5-turbo-16k-0613 |
 |                | gpt-3.5-turbo-0613     | --model gpt-3.5-turbo-0613 |
 |                | gpt-4-0613             | --model gpt-4-0613 |
-| Anthropic      | Claude 3.5 Sonnet      | --model claude-3-5-sonnet-20240620 |
+| Anthropic      | Claude 3.5 Sonnet v2   | --model claude-3-5-sonnet-20241022 |
+|                | Claude 3.5 Sonnet      | --model claude-3-5-sonnet-20240620 |
 |                | Claude 3 Opus          | --model claude-3-opus-20240229 |
 |                | Claude 3 Sonnet        | --model claude-3-sonnet-20240229 |
 |                | Claude 3 Haiku         | --model claude-3-haiku-20240307 |

diff --git a/app/model/claude.py b/app/model/claude.py
@@ -154,3 +154,11 @@ def __init__(self):
             "claude-3-5-sonnet-20240620", 0.000003, 0.000015, parallel_tool_call=True
         )
         self.note = "Most intelligent model from Antropic"
+
+
+class Claude3_5SonnetNew(AnthropicModel):
+    def __init__(self):
+        super().__init__(
+            "claude-3-5-sonnet-20241022", 0.000003, 0.000015, parallel_tool_call=True
+        )
+        self.note = "Most intelligent model from Antropic"
diff --git a/app/model/register.py b/app/model/register.py
@@ -32,6 +32,7 @@ def register_all_models() -> None:
     common.register_model(claude.Claude3Sonnet())
     common.register_model(claude.Claude3Haiku())
     common.register_model(claude.Claude3_5Sonnet())
+    common.register_model(claude.Claude3_5SonnetNew())
 
     common.register_model(bedrock.AnthropicClaude3Opus())
     common.register_model(bedrock.AnthropicClaude3Sonnet())