49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
|
"""Regression test for #11884: _make_agent must resolve runtime provider.
|
||
|
|
|
||
|
|
Without resolve_runtime_provider(), bare-slug models in config
|
||
|
|
(e.g. ``claude-opus-4-6`` with ``model.provider: anthropic``) leave
|
||
|
|
provider/base_url/api_key empty in AIAgent, causing HTTP 404.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
|
||
|
|
def test_make_agent_passes_resolved_provider():
|
||
|
|
"""_make_agent forwards provider/base_url/api_key/api_mode from
|
||
|
|
resolve_runtime_provider to AIAgent."""
|
||
|
|
|
||
|
|
fake_runtime = {
|
||
|
|
"provider": "anthropic",
|
||
|
|
"base_url": "https://api.anthropic.com",
|
||
|
|
"api_key": "sk-test-key",
|
||
|
|
"api_mode": "anthropic_messages",
|
||
|
|
"command": None,
|
||
|
|
"args": None,
|
||
|
|
"credential_pool": None,
|
||
|
|
}
|
||
|
|
|
||
|
|
fake_cfg = {
|
||
|
|
"model": {"default": "claude-opus-4-6", "provider": "anthropic"},
|
||
|
|
"agent": {"system_prompt": "test"},
|
||
|
|
}
|
||
|
|
|
||
|
|
with patch("tui_gateway.server._load_cfg", return_value=fake_cfg), \
|
||
|
|
patch("tui_gateway.server._get_db", return_value=MagicMock()), \
|
||
|
|
patch("tui_gateway.server._load_tool_progress_mode", return_value="compact"), \
|
||
|
|
patch("tui_gateway.server._load_reasoning_config", return_value=None), \
|
||
|
|
patch("tui_gateway.server._load_service_tier", return_value=None), \
|
||
|
|
patch("tui_gateway.server._load_enabled_toolsets", return_value=None), \
|
||
|
|
patch("hermes_cli.runtime_provider.resolve_runtime_provider", return_value=fake_runtime) as mock_resolve, \
|
||
|
|
patch("run_agent.AIAgent") as mock_agent:
|
||
|
|
|
||
|
|
from tui_gateway.server import _make_agent
|
||
|
|
_make_agent("sid-1", "key-1")
|
||
|
|
|
||
|
|
mock_resolve.assert_called_once_with(requested=None)
|
||
|
|
|
||
|
|
call_kwargs = mock_agent.call_args
|
||
|
|
assert call_kwargs.kwargs["provider"] == "anthropic"
|
||
|
|
assert call_kwargs.kwargs["base_url"] == "https://api.anthropic.com"
|
||
|
|
assert call_kwargs.kwargs["api_key"] == "sk-test-key"
|
||
|
|
assert call_kwargs.kwargs["api_mode"] == "anthropic_messages"
|