Aj/llama4 support (#238)

* add llama4-support

* llama4 maverick L1

* llama4 maverick L2

---------

Co-authored-by: agentlabtraces <[email protected]>

Author: amanjaiswal73892

reproducibility_journal.csv

@@ -74,3 +74,5 @@ Leo Boisvert,GenericAgent-openai_o1-mini-2024-09-12,workarena_l1,0.4.1,2025-02-0
   M: src/agentlab/analyze/agent_xray.py
   M: src/agentlab/llm/llm_configs.py",0.13.3,1d2d7160e5b7ec9954ecb48988f71eb56288dd29,"
 Leo Boisvert,GenericAgent-anthropic_claude-3.7-sonnet,workarena_l1,0.4.1,2025-02-25_02-32-09,d4f900c2-1de1-4e4b-a3ab-495ff2675fff,0.515,0.028,0,330/330,None,Linux (#68-Ubuntu SMP Mon Oct 7 14:34:20 UTC 2024),3.12.3,1.44.0,v0.4.0,c9d2ef9648435ef1119950ecb1a0734497ccc33b,,0.13.3,1d2d7160e5b7ec9954ecb48988f71eb56288dd29,
+agentlabtraces,GenericAgent-meta-llama_llama-4-maverick,workarena_l1,0.4.1,2025-04-14_17-15-56,a6dc4022-2bb7-4b46-8b37-f62c010defc1,0.27,0.024,0,330/330,None,Linux (#135-Ubuntu SMP Fri Sep 27 13:53:58 UTC 2024),3.12.7,1.39.0,v0.4.0,5eb2ecb5e5b293170230bcbed8b17fe192af214a,,0.13.3,70dac253628c476aff1af6a975f27f8563453ad2,
+agentlabtraces,GenericAgent-meta-llama_llama-4-maverick,workarena_l2_agent_curriculum_eval,0.4.1,2025-04-22_15-38-44,d62fed39-caac-4ef3-92ac-b29897c69f88,0.085,0.018,1,235/235,None,Linux (#68-Ubuntu SMP Mon Oct 7 14:34:20 UTC 2024),3.12.7,1.39.0,v0.4.0,43bafbcfbe398fca39e4ffdc57b2f226d2c6d3e1,,0.13.3,70dac253628c476aff1af6a975f27f8563453ad2,

src/agentlab/agents/generic_agent/__init__.py

@@ -10,6 +10,7 @@
     AGENT_3_5,
     AGENT_8B,
     AGENT_CUSTOM,
+    AGENT_LLAMA4_17B_INSTRUCT,
     AGENT_LLAMA3_70B,
     AGENT_LLAMA31_70B,
     RANDOM_SEARCH_AGENT,
@@ -31,6 +32,7 @@
     "AGENT_4o_VISION",
     "AGENT_o3_MINI",
     "AGENT_o1_MINI",
+    "AGENT_LLAMA4_17B_INSTRUCT",
     "AGENT_LLAMA3_70B",
     "AGENT_LLAMA31_70B",
     "AGENT_8B",

src/agentlab/agents/generic_agent/agent_configs.py

@@ -10,6 +10,7 @@
 
 from .generic_agent import GenericAgentArgs
 from .generic_agent_prompt import GenericPromptFlags
+from .tmlr_config import BASE_FLAGS
 
 FLAGS_CUSTOM = GenericPromptFlags(
     obs=dp.ObsFlags(
@@ -296,7 +297,10 @@
     chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/anthropic/claude-3.5-sonnet:beta"],
     flags=FLAGS_GPT_4o_VISION,
 )
-
+AGENT_LLAMA4_17B_INSTRUCT = GenericAgentArgs(
+    chat_model_args=CHAT_MODEL_ARGS_DICT["openrouter/meta-llama/llama-4-maverick"],
+    flags=BASE_FLAGS,
+)
 
 DEFAULT_RS_FLAGS = GenericPromptFlags(
     flag_group="default_rs",

src/agentlab/llm/llm_configs.py

@@ -155,6 +155,14 @@
         max_new_tokens=28_000,
         temperature=1e-1,
     ),
+    "openrouter/meta-llama/llama-4-maverick": OpenRouterModelArgs(
+        model_name="meta-llama/llama-4-maverick",
+        max_total_tokens=128_000,
+        max_input_tokens=100_000,
+        max_new_tokens=28_000,
+        temperature=1e-1,
+        vision_support=True,
+    ),
     "openrouter/meta-llama/llama-3.1-8b-instruct:free": OpenRouterModelArgs(
         model_name="meta-llama/llama-3.1-8b-instruct:free",
         max_total_tokens=128_000,