Added missing models to perform_writeup and launch_oe_scientist

angeloluidens · Aug 14, 2024 · 08be9a8 · 08be9a8
1 parent 1505441
commit 08be9a8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/ai_scientist/perform_writeup.py b/ai_scientist/perform_writeup.py
@@ -528,6 +528,11 @@ def perform_writeup(
             "gpt-4o-2024-05-13",
             "deepseek-coder-v2-0724",
             "llama3.1-405b",
+            # Anthropic Claude models via Amazon Bedrock
+            "bedrock/anthropic.claude-3-sonnet-20240229-v1:0",
+            "bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0",
+            "bedrock/anthropic.claude-3-haiku-20240307-v1:0",
+            "bedrock/anthropic.claude-3-opus-20240229-v1:0"
         ],
         help="Model to use for AI Scientist.",
     )

diff --git a/experimental/launch_oe_scientist.py b/experimental/launch_oe_scientist.py
@@ -42,6 +42,11 @@ def parse_arguments():
             "gpt-4o-2024-05-13",
             "deepseek-coder-v2-0724",
             "llama3.1-405b",
+            # Anthropic Claude models via Amazon Bedrock
+            "bedrock/anthropic.claude-3-sonnet-20240229-v1:0",
+            "bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0",
+            "bedrock/anthropic.claude-3-haiku-20240307-v1:0",
+            "bedrock/anthropic.claude-3-opus-20240229-v1:0"
         ],
         help="Model to use for AI Scientist.",
     )