Skip to content

Commit a32ba5d

Browse files
update agent model list in benchmark command
Signed-off-by: Rahul Shetty <[email protected]>
1 parent 49b30cb commit a32ba5d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/agentready/cli/benchmark.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
import click
99

10-
from ..services.eval_harness.harbor_config import HarborConfig
10+
from ..services.eval_harness.harbor_config import ALLOWED_MODELS, HarborConfig
1111
from ..services.eval_harness.tbench_runner import _real_tbench_result
1212
from ..services.harbor.agent_toggler import AssessorStateToggler
1313
from ..services.harbor.comparer import compare_assessor_impact
@@ -35,7 +35,7 @@
3535
)
3636
@click.option(
3737
"--model",
38-
type=click.Choice(["claude-haiku-4-5", "claude-sonnet-4-5"]),
38+
type=click.Choice(list(ALLOWED_MODELS)),
3939
default="claude-haiku-4-5",
4040
help="Model for evaluation",
4141
)

0 commit comments

Comments
 (0)