Add AGENT_37_SONNET support and update agent configurations

Author: jardinetsouffleton

main.py

@@ -14,13 +14,17 @@
     RANDOM_SEARCH_AGENT,
     AGENT_4o,
     AGENT_4o_MINI,
+    AGENT_o3_MINI,
+    AGENT_o1_MINI,
+    AGENT_37_SONNET,
+    AGENT_CLAUDE_SONNET_35,
 )
 from agentlab.experiments.study import Study
 
 logging.getLogger().setLevel(logging.INFO)
 
 # choose your agent or provide a new agent
-agent_args = [AGENT_4o_MINI]
+agent_args = [AGENT_o1_MINI]
 # agent_args = [AGENT_4o]
 
 
@@ -35,14 +39,14 @@
 # Set reproducibility_mode = True for reproducibility
 # this will "ask" agents to be deterministic. Also, it will prevent you from launching if you have
 # local changes. For your custom agents you need to implement set_reproducibility_mode
-reproducibility_mode = False
+reproducibility_mode = True
 
 # Set relaunch = True to relaunch an existing study, this will continue incomplete
 # experiments and relaunch errored experiments
 relaunch = False
 
 ## Number of parallel jobs
-n_jobs = 4  # Make sure to use 1 job when debugging in VSCode
+n_jobs = 5  # Make sure to use 1 job when debugging in VSCode
 # n_jobs = -1  # to use all available cores
 
 

src/agentlab/agents/generic_agent/__init__.py

@@ -16,6 +16,7 @@
     AGENT_4o,
     AGENT_4o_MINI,
     AGENT_CLAUDE_SONNET_35,
+    AGENT_37_SONNET,
     AGENT_4o_VISION,
     AGENT_o3_MINI,
     AGENT_o1_MINI,
@@ -34,6 +35,7 @@
     "RANDOM_SEARCH_AGENT",
     "AGENT_CUSTOM",
     "AGENT_CLAUDE_SONNET_35",
+    "AGENT_37_SONNET",
     "AGENT_4o_VISION",
     "AGENT_4o_MINI_VISION",
     "AGENT_CLAUDE_SONNET_35_VISION",

src/agentlab/agents/generic_agent/agent_configs.py

@@ -264,7 +264,10 @@
     chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/anthropic/claude-3.5-sonnet:beta"],
     flags=FLAGS_GPT_4o,
 )
-
+AGENT_37_SONNET = GenericAgentArgs(
+    chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/anthropic/claude-3.7-sonnet"],
+    flags=FLAGS_GPT_4o,
+)
 AGENT_o3_MINI = GenericAgentArgs(
     chat_model_args=CHAT_MODEL_ARGS_DICT["openai/o3-mini-2025-01-31"],
     flags=FLAGS_GPT_4o,

src/agentlab/llm/llm_configs.py

@@ -184,6 +184,14 @@
         max_new_tokens=2_000,
         temperature=1e-1,
     ),
+    "openrouter/anthropic/claude-3.7-sonnet": OpenRouterModelArgs(
+        model_name="anthropic/claude-3.7-sonnet",
+        max_total_tokens=200_000,
+        max_input_tokens=200_000,
+        max_new_tokens=8_192,
+        temperature=1e-1,
+        vision_support=True,
+    ),
     "openrouter/openai/o1-mini-2024-09-12": OpenRouterModelArgs(
         model_name="openai/o1-mini-2024-09-12",
         max_total_tokens=128_000,