From ff87a566c4c4b7d25ed01463f4ced76a5cdfe091 Mon Sep 17 00:00:00 2001 From: teknium1 Date: Sat, 14 Mar 2026 22:28:15 -0700 Subject: [PATCH] fix(test): make Nous setup prompt selection robust to optional vision step --- tests/hermes_cli/test_setup.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/hermes_cli/test_setup.py b/tests/hermes_cli/test_setup.py index 4d0ab8873..e0abc639c 100644 --- a/tests/hermes_cli/test_setup.py +++ b/tests/hermes_cli/test_setup.py @@ -25,7 +25,11 @@ def test_nous_oauth_setup_keeps_current_model_when_syncing_disk_provider( config = load_config() - prompt_choices = iter([0, 2]) + # Provider selection always comes first. Depending on available vision + # backends, setup may either skip the optional vision step or prompt for + # it before the default-model choice. Provide enough selections for both + # paths while still ending on "keep current model". + prompt_choices = iter([0, 2, 2]) monkeypatch.setattr( "hermes_cli.setup.prompt_choice", lambda *args, **kwargs: next(prompt_choices),