qing
d03dbc7f91
- models 配置节：多 OpenAI Chat Completions 兼容端点按档案名切换，默认空 dict 零破坏回退旧 api 节 - --model 双语义：档案名（命中 models 节完整采用）/ 裸模型名（同端点换模型） - resolve_api_key 档案级 env 三级语义：未配置回退 api.api_key_env / 显式名称 / 空串免密（Ollama/vLLM） - 档案级 reasoning_enabled 附加 client 属性，DeepSeekChat._reasoning_enabled 优先读取 - llm_client 增加 openai_compatible/openai provider 别名 - README 更新：aibox/README 六处（特性/配置/多模型档案专节/CLI 参数/更新日志 v3.2/测试计数 264）+ 根 README aibox 小节 - 单测 17 项新增（test_models_profile.py），全套 264 项全过
214 lines
8.3 KiB
Python
214 lines
8.3 KiB
Python
"""多模型档案（models 配置节）单元测试
|
|
|
|
覆盖：
|
|
- _resolve_model_profile 三级优先级：档案名命中 > default_model 档案 > 回退 None
|
|
- 裸模型名语义：用默认档案端点、覆盖档案内模型名
|
|
- create_llm_client 档案构造：url/model/timeout 来自档案
|
|
- resolve_api_key：档案级 env 名、空串免密（allow_missing）、缺 key 报错
|
|
- 档案级 reasoning_enabled 附加到 client 并被 DeepSeekChat._reasoning_enabled 读取
|
|
- 向后兼容：无 models 节时走旧 api 扁平节
|
|
|
|
不依赖真实 API/网络：仅测构造与参数解析，不发请求。
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import types
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
# ---- 环境桩：保证 import aibox.py 不失败 ----
|
|
if "websockets" not in sys.modules:
|
|
sys.modules["websockets"] = types.ModuleType("websockets")
|
|
|
|
from llm_client import ( # noqa: E402
|
|
create_llm_client,
|
|
resolve_api_key,
|
|
_resolve_model_profile,
|
|
)
|
|
|
|
|
|
class FakeConfig:
|
|
"""带 get(key, default) 的最小配置桩。"""
|
|
|
|
def __init__(self, data):
|
|
self._data = data
|
|
|
|
def get(self, key, default=None):
|
|
cur = self._data
|
|
for k in key.split("."):
|
|
if not isinstance(cur, dict) or k not in cur:
|
|
return default
|
|
cur = cur[k]
|
|
return cur
|
|
|
|
|
|
MODELS = {
|
|
"deepseek": {
|
|
"api_url": "https://api.deepseek.com/chat/completions",
|
|
"model": "deepseek-v4-flash",
|
|
"api_key_env": "DEEPSEEK_API_KEY",
|
|
"timeout": 120,
|
|
},
|
|
"kimi": {
|
|
"api_url": "https://api.moonshot.cn/v1/chat/completions",
|
|
"model": "kimi-k2",
|
|
"api_key_env": "MOONSHOT_API_KEY",
|
|
"reasoning_enabled": True,
|
|
},
|
|
"local": {
|
|
"api_url": "http://localhost:11434/v1/chat/completions",
|
|
"model": "qwen3:32b",
|
|
"api_key_env": "", # 免密端点
|
|
"timeout": 300,
|
|
},
|
|
}
|
|
|
|
|
|
def _cfg(models=MODELS, default_model="deepseek", **api_over):
|
|
api = {
|
|
"provider": "openai_compatible",
|
|
"deepseek_api_url": "https://api.deepseek.com/chat/completions",
|
|
"deepseek_model": "deepseek-v4-flash",
|
|
"api_key_env": "DEEPSEEK_API_KEY",
|
|
"reasoning_enabled": False,
|
|
"timeout": 30,
|
|
"default_model": default_model,
|
|
}
|
|
api.update(api_over)
|
|
data = {"api": api}
|
|
if models is not None:
|
|
data["models"] = models
|
|
return FakeConfig(data)
|
|
|
|
|
|
# ==================== _resolve_model_profile 优先级 ====================
|
|
|
|
class TestResolveProfile:
|
|
def test_explicit_profile_name(self):
|
|
"""model=档案名 → 完整采用该档案，模型名取档案内 model"""
|
|
profile, model = _resolve_model_profile(_cfg(), "kimi")
|
|
assert profile is MODELS["kimi"]
|
|
assert model == "kimi-k2"
|
|
|
|
def test_bare_model_name_uses_default_profile(self):
|
|
"""model=裸名 → 默认档案端点 + 裸名作为模型名"""
|
|
profile, model = _resolve_model_profile(_cfg(), "deepseek-v3-legacy")
|
|
assert profile is MODELS["deepseek"]
|
|
assert model == "deepseek-v3-legacy"
|
|
|
|
def test_no_model_uses_default_profile(self):
|
|
"""未指定 model → 默认档案与档案内模型名"""
|
|
profile, model = _resolve_model_profile(_cfg(), None)
|
|
assert profile is MODELS["deepseek"]
|
|
assert model == "deepseek-v4-flash"
|
|
|
|
def test_fallback_when_no_models_section(self):
|
|
"""无 models 节 → (None, model) 回退旧 api 扁平节"""
|
|
profile, model = _resolve_model_profile(_cfg(models=None), "anything")
|
|
assert profile is None
|
|
assert model == "anything"
|
|
|
|
def test_fallback_when_default_model_missing(self):
|
|
"""default_model 指向不存在的档案且 model 未命中 → 回退"""
|
|
profile, model = _resolve_model_profile(_cfg(default_model="nope"), "whatever")
|
|
assert profile is None
|
|
assert model == "whatever"
|
|
|
|
|
|
# ==================== create_llm_client 档案构造 ====================
|
|
|
|
class TestCreateClient:
|
|
def test_profile_selects_url_model_timeout(self):
|
|
"""档案命中 → client 的 url/model/timeout 全部来自档案"""
|
|
with mock.patch.dict(os.environ, {"MOONSHOT_API_KEY": "sk-moon"}):
|
|
c = create_llm_client(_cfg(), model="kimi")
|
|
assert c.base_url == "https://api.moonshot.cn/v1/chat/completions"
|
|
assert c.model == "kimi-k2"
|
|
assert c.api_key == "sk-moon"
|
|
|
|
def test_bare_name_keeps_default_endpoint(self):
|
|
"""裸模型名 → 默认档案端点 + 裸模型名"""
|
|
with mock.patch.dict(os.environ, {"DEEPSEEK_API_KEY": "sk-ds"}):
|
|
c = create_llm_client(_cfg(), model="deepseek-v4-lite")
|
|
assert c.base_url == "https://api.deepseek.com/chat/completions"
|
|
assert c.model == "deepseek-v4-lite"
|
|
|
|
def test_backward_compat_flat_api(self):
|
|
"""无 models 节 → 旧 api 扁平节行为完全不变"""
|
|
with mock.patch.dict(os.environ, {"DEEPSEEK_API_KEY": "sk-ds"}):
|
|
c = create_llm_client(_cfg(models=None))
|
|
assert c.base_url == "https://api.deepseek.com/chat/completions"
|
|
assert c.model == "deepseek-v4-flash"
|
|
|
|
def test_profile_reasoning_flag_attached(self):
|
|
"""档案 reasoning_enabled → client 属性附加"""
|
|
with mock.patch.dict(os.environ, {"MOONSHOT_API_KEY": "sk-moon"}):
|
|
c = create_llm_client(_cfg(), model="kimi")
|
|
assert c.reasoning_enabled is True
|
|
|
|
def test_profile_without_env_falls_back_to_api_section(self):
|
|
"""档案未配置 api_key_env → 回退 api.api_key_env（用户自定义变量名兼容，回归保证）"""
|
|
models = dict(MODELS)
|
|
models["noenv"] = {
|
|
"api_url": "https://ark.example/v1/chat/completions",
|
|
"model": "doubao-pro",
|
|
}
|
|
with mock.patch.dict(os.environ, {"ARK_API_KEY": "sk-ark"}):
|
|
c = create_llm_client(_cfg(models=models, api_key_env="ARK_API_KEY"),
|
|
model="noenv")
|
|
assert c.api_key == "sk-ark"
|
|
assert c.base_url == "https://ark.example/v1/chat/completions"
|
|
|
|
def test_no_profile_no_reasoning_attr(self):
|
|
"""无档案 → client 无 reasoning_enabled 属性（走全局配置）"""
|
|
with mock.patch.dict(os.environ, {"DEEPSEEK_API_KEY": "sk-ds"}):
|
|
c = create_llm_client(_cfg(models=None))
|
|
assert not hasattr(c, "reasoning_enabled")
|
|
|
|
|
|
# ==================== resolve_api_key 档案级解析 ====================
|
|
|
|
class TestResolveApiKey:
|
|
def test_profile_env_name(self):
|
|
"""档案级 api_key_env 指定环境变量"""
|
|
with mock.patch.dict(os.environ, {"MOONSHOT_API_KEY": "sk-moon"}):
|
|
assert resolve_api_key(None, _cfg(), env_name="MOONSHOT_API_KEY") == "sk-moon"
|
|
|
|
def test_empty_env_name_allows_missing(self):
|
|
"""api_key_env 空串 + allow_missing → 返回空串（免密端点）"""
|
|
assert resolve_api_key(None, _cfg(), env_name="", allow_missing=True) == ""
|
|
|
|
def test_missing_key_raises(self):
|
|
"""无 key 且不允许免密 → RuntimeError"""
|
|
env = {k: "" for k in ("DEEPSEEK_API_KEY", "MOONSHOT_API_KEY")}
|
|
with mock.patch.dict(os.environ, env, clear=False), \
|
|
mock.patch("builtins.open", side_effect=OSError):
|
|
with pytest.raises(RuntimeError):
|
|
resolve_api_key(None, _cfg(), env_name="MOONSHOT_API_KEY")
|
|
|
|
def test_explicit_key_wins(self):
|
|
"""显式传入 api_key 优先"""
|
|
assert resolve_api_key("sk-explicit", _cfg()) == "sk-explicit"
|
|
|
|
|
|
# ==================== DeepSeekChat 档案级 reasoning 读取 ====================
|
|
|
|
class TestChatReasoningFlag:
|
|
def test_reasoning_enabled_reads_client_attr(self):
|
|
"""client 带 reasoning_enabled 属性 → DeepSeekChat._reasoning_enabled 优先读它"""
|
|
from aibox import DeepSeekChat
|
|
from test_once_mode import _make_ai # 复用 FakeLLMClient 构造
|
|
ai = _make_ai()
|
|
ai.client.reasoning_enabled = True # 档案级开
|
|
assert ai._reasoning_enabled() is True
|
|
ai.client.reasoning_enabled = False # 档案级关（覆盖全局）
|
|
assert ai._reasoning_enabled() is False
|
|
|
|
def test_reasoning_falls_back_to_global(self):
|
|
"""client 无属性 → 回退 constants.REASONING_ENABLED"""
|
|
from test_once_mode import _make_ai
|
|
ai = _make_ai()
|
|
assert ai._reasoning_enabled() == False # noqa: E712 全局默认 False
|