Skip to content

Commit 02347f8

Browse files
add flag to select llm config
1 parent 92f9a74 commit 02347f8

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

src/agentlab/agents/hitl_agent/launch_hint_ui.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,7 @@
1515

1616
import bgym
1717

18-
from agentlab.agents.hitl_agent.generic_human_guided_agent import (
19-
HUMAN_GUIDED_GENERIC_AGENT,
20-
)
18+
from agentlab.agents.hitl_agent.generic_human_guided_agent import get_base_agent
2119
from agentlab.experiments.exp_utils import RESULTS_DIR
2220
from agentlab.experiments.study import Study
2321

@@ -120,6 +118,12 @@ def parse_args():
120118
required=False,
121119
help="Task seed to use for the selected task.",
122120
)
121+
p.add_argument(
122+
"--llm-config",
123+
dest="llm_config",
124+
default="openai/gpt-5-mini-2025-08-07",
125+
help="LLM configuration to use for the agent (e.g., 'azure/gpt-5-mini-2025-08-07').",
126+
)
123127
p.add_argument(
124128
"--headless",
125129
action=argparse.BooleanOptionalAction,
@@ -157,7 +161,7 @@ def main():
157161
"--benchmark, --task-name, and --seed are required unless using --download-hints"
158162
)
159163
benchmark = build_benchmark(args.benchmark, args.task_name, args.seed, args.headless)
160-
agent_configs = [HUMAN_GUIDED_GENERIC_AGENT]
164+
agent_configs = [get_base_agent(args.llm_config)]
161165
# study is needed to run the 'set_benchmark' method which sets appropriate agent parameters.
162166
study = Study(agent_args=agent_configs, benchmark=benchmark, logging_level=logging.WARNING)
163167
study.run(

0 commit comments

Comments
 (0)