ready for experiment

Author: TLSDC

main.py

@@ -7,12 +7,13 @@
 """
 
 import logging
+
 from agentlab.agents.generic_agent import (
+    AGENT_LLAMA3_70B,
+    AGENT_LLAMA31_70B,
     RANDOM_SEARCH_AGENT,
     AGENT_4o,
     AGENT_4o_MINI,
-    AGENT_LLAMA3_70B,
-    AGENT_LLAMA31_70B,
 )
 from agentlab.experiments.study import Study
 

src/agentlab/agents/webarena_basic_agent/webarena_basic_agent.py

@@ -131,9 +131,19 @@ def parser(response: str) -> tuple[dict, bool, str]:
 chat_model_args = CHAT_MODEL_ARGS_DICT["openai/gpt-4o-mini-2024-07-18"]
 
 
-AGENT_CONFIG = WebArenaBasicAgentArgs(
+AGENT_4o_MINI = WebArenaBasicAgentArgs(
     temperature=0.1,
-    chat_model_args=chat_model_args,
+    chat_model_args=CHAT_MODEL_ARGS_DICT["openai/gpt-4o-mini-2024-07-18"],
+)
+
+AGENT_4o = WebArenaBasicAgentArgs(
+    temperature=0.1,
+    chat_model_args=CHAT_MODEL_ARGS_DICT["azure/gpt-4o-2024-08-06"],
+)
+
+AGENT_SONNET = WebArenaBasicAgentArgs(
+    temperature=0.1,
+    chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/anthropic/claude-3.5-sonnet:beta"],
 )