mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-13 09:01:54 +00:00
af978ecb17
Rebased onto current main and re-ported across the restructured surfaces: model flows now thread confirm_provider/base_url/api_key through hermes_cli/model_setup_flows.py, the Discord picker lives in plugins/platforms/discord/adapter.py, and the web dashboard picker applies chat-mode switches via config.set so the expensive-model confirmation can ride the response. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from types import SimpleNamespace
|
|
|
|
from hermes_cli.model_switch import ModelSwitchResult
|
|
|
|
|
|
def _bound(fn, instance):
|
|
return fn.__get__(instance, type(instance))
|
|
|
|
|
|
def test_prompt_toolkit_model_picker_defers_confirmation_off_key_handler(monkeypatch):
|
|
import cli as cli_mod
|
|
|
|
result = ModelSwitchResult(
|
|
success=True,
|
|
new_model="openai/gpt-5.5-pro",
|
|
target_provider="nous",
|
|
)
|
|
monkeypatch.setattr(
|
|
"hermes_cli.model_switch.switch_model",
|
|
lambda **_kwargs: result,
|
|
)
|
|
|
|
captured = {}
|
|
|
|
class _Thread:
|
|
def __init__(self, *, target, args, daemon):
|
|
captured["target"] = target
|
|
captured["args"] = args
|
|
captured["daemon"] = daemon
|
|
|
|
def start(self):
|
|
captured["started"] = True
|
|
|
|
monkeypatch.setattr(cli_mod.threading, "Thread", _Thread)
|
|
|
|
self_ = SimpleNamespace(
|
|
_app=object(),
|
|
_model_picker_state={
|
|
"stage": "model",
|
|
"provider_data": {"slug": "nous"},
|
|
"model_list": ["openai/gpt-5.5-pro"],
|
|
"selected": 0,
|
|
"user_provs": None,
|
|
"custom_provs": None,
|
|
},
|
|
provider="nous",
|
|
model="openai/gpt-5.5",
|
|
base_url="",
|
|
api_key="",
|
|
_restore_modal_input_snapshot=lambda: None,
|
|
_invalidate=lambda **_kwargs: None,
|
|
)
|
|
self_._close_model_picker = _bound(cli_mod.HermesCLI._close_model_picker, self_)
|
|
self_._confirm_and_apply_model_switch_result = (
|
|
lambda *_args: captured.setdefault("ran_inline", True)
|
|
)
|
|
|
|
_bound(cli_mod.HermesCLI._handle_model_picker_selection, self_)()
|
|
|
|
assert self_._model_picker_state is None
|
|
assert captured["started"] is True
|
|
assert captured["daemon"] is True
|
|
assert captured["args"] == (result, False)
|
|
assert "ran_inline" not in captured
|