add llama4-support

Author: amanjaiswal73892

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,