feat: OpenAI 多模型注册表 + /model 命令，告别手改配置文件
痛点：很多模型都是 OpenAI 格式，但换模型只能手改 config.json 的单槽位，
且只能存一个。新增模型注册表与 /model 命令族：
- 新增 xcli_core/model_registry.py：在 api.engines.openai 下维护
models 列表（name + base_url + api_key + model）与 current 指针，全部落盘。
向后兼容：无 models 时把扁平 api_key/base_url/model 当作单条隐式模型。
- OpenaiAI 接入：__init__ 应用注册表 current 模型；新增
set_registry_model / list_registry_models / add_model / remove_model /
current_model_name。注意保留引擎原有的轻量 set_model（只改 model 名）
与 list_models（查 API /models 端点），新方法与它们不重名、互不干扰。
- 新增 /model 命令族（仅 openai 引擎）：
/model 列出已配置模型（* 标记当前）
/model <名称> 切换当前模型并持久化
/model add 交互式录入（名称/base_url/api_key/model），回车即存并切换
/model rm <名称> 删除（删当前则回退到列表第一个）
- 启动横幅引擎名后显示当前模型；紧凑提示与 /help 速查表均补 /model 入口。
- 测试：新增 tests/test_model_registry.py（13 项，覆盖兼容/增删/切换/落盘/
引擎接入/CLI 路由，全程临时配置不污染真实 config.json）。
全量 230 passed（原 217 passed）。
交互实测：/model add → 已添加并切换到 gpt4o；/model gpt4o → 已切换；
/model rm gpt4o → 回退到 xophunyuan7bmt，配置结构与内容完好。
This commit is contained in:
shuiyu67 committed
1 parent
3a86da7444
commit
1383d32f8b
7 files changed
+493
-2
No files matched your search
@@ -81,6 +81,13 @@ class BaseAICLI:
|
||||
self.current_engine = None
|
||||
if self.current_engine:
|
||||
self.current_engine.cli = self # 设置引用以便访问插件
|
||||
# 记录当前模型（openai 引擎注册表名，其它引擎取 model 值）
|
||||
try:
|
||||
self.current_model = self.current_engine.current_model_name()
|
||||
except Exception:
|
||||
self.current_model = getattr(self.current_engine, 'model', None)
|
||||
else:
|
||||
self.current_model = None
|
||||
|
||||
# 初始化安全层
|
||||
from .safety import get_safety
|
||||
@@ -306,6 +313,83 @@ class BaseAICLI:
|
||||
help_lines.append(f" /engine {cmd:<10} - 执行{cmd}命令")
|
||||
return '\n'.join(help_lines) if help_lines else " 无可用引擎管理命令"
|
||||
|
||||
# ── 模型管理（OpenAI 引擎多模型切换）──
|
||||
|
||||
def handle_model_command(self, args):
|
||||
"""处理 /model 命令族。仅对 openai 引擎有效。"""
|
||||
eng = self.current_engine
|
||||
if getattr(eng, 'name', None) != 'openai':
|
||||
cur = getattr(eng, 'name', '?')
|
||||
print(f"{Fore.RED}模型管理仅适用于 openai 引擎（当前: {cur}）。"
|
||||
f"先 /engine switch openai{Style.RESET_ALL}")
|
||||
return
|
||||
|
||||
if not args or args == 'list':
|
||||
self._model_list(eng)
|
||||
return
|
||||
|
||||
parts = args.split(' ', 1)
|
||||
cmd = parts[0]
|
||||
rest = parts[1].strip() if len(parts) > 1 else ''
|
||||
if cmd == 'add':
|
||||
self._model_add_interactive(eng)
|
||||
elif cmd in ('rm', 'remove', 'del'):
|
||||
if not rest:
|
||||
print(f"{Fore.RED}用法: /model rm <名称>{Style.RESET_ALL}")
|
||||
return
|
||||
if eng.remove_model(rest):
|
||||
print(f"{Fore.GREEN}已删除模型: {rest}{Style.RESET_ALL}")
|
||||
else:
|
||||
print(f"{Fore.RED}未找到模型: {rest}{Style.RESET_ALL}")
|
||||
elif cmd in ('set', 'switch'):
|
||||
self._model_switch(eng, rest)
|
||||
else:
|
||||
# 无子命令：/model <名称> 视为切换
|
||||
self._model_switch(eng, cmd)
|
||||
|
||||
def _model_list(self, eng):
|
||||
models, current = eng.list_registry_models()
|
||||
if not models:
|
||||
print(f"{Fore.YELLOW}尚未配置任何模型，用 /model add 添加{Style.RESET_ALL}")
|
||||
return
|
||||
print(f"{Fore.GREEN}已配置的 OpenAI 模型:{Style.RESET_ALL}")
|
||||
for m in models:
|
||||
mark = f"{Fore.CYAN} *{Style.RESET_ALL}" if m.get('name') == current else ""
|
||||
print(f" {m.get('name')}{mark} ({m.get('model')} @ {m.get('base_url') or '无 base_url'})")
|
||||
print(f"{Fore.CYAN}当前: {current} ｜ /model <名称> 切换 ｜ /model add 新增 ｜ "
|
||||
f"/model rm <名称> 删除{Style.RESET_ALL}")
|
||||
|
||||
def _model_switch(self, eng, name):
|
||||
if not name:
|
||||
print(f"{Fore.RED}用法: /model <名称> 或 /model 查看列表{Style.RESET_ALL}")
|
||||
return
|
||||
if eng.set_registry_model(name):
|
||||
self.current_model = name
|
||||
print(f"{Fore.GREEN}已切换到模型: {name}{Style.RESET_ALL}")
|
||||
else:
|
||||
print(f"{Fore.RED}未找到模型: {name}（/model 查看可用）{Style.RESET_ALL}")
|
||||
|
||||
def _model_add_interactive(self, eng):
|
||||
print(f"{Fore.CYAN}添加 OpenAI 模型（回车留空则用默认/为空）{Style.RESET_ALL}")
|
||||
try:
|
||||
name = input(f"{Fore.CYAN}名称(用于切换, 如 gpt4o): {Style.RESET_ALL}").strip()
|
||||
if not name:
|
||||
print(f"{Fore.RED}名称不能为空{Style.RESET_ALL}")
|
||||
return
|
||||
base_url = input(f"{Fore.CYAN}base_url(如 https://api.openai.com/v1): {Style.RESET_ALL}").strip()
|
||||
api_key = input(f"{Fore.CYAN}api_key: {Style.RESET_ALL}").strip()
|
||||
model = input(f"{Fore.CYAN}model(如 gpt-4o): {Style.RESET_ALL}").strip()
|
||||
if not model:
|
||||
print(f"{Fore.RED}model 不能为空{Style.RESET_ALL}")
|
||||
return
|
||||
except (EOFError, KeyboardInterrupt):
|
||||
print(f"{Fore.YELLOW}\n已取消添加{Style.RESET_ALL}")
|
||||
return
|
||||
eng.add_model({"name": name, "base_url": base_url, "api_key": api_key, "model": model})
|
||||
eng.set_registry_model(name)
|
||||
self.current_model = name
|
||||
print(f"{Fore.GREEN}已添加并切换到模型: {name}{Style.RESET_ALL}")
|
||||
|
||||
# ── 引擎切换 ──
|
||||
|
||||
def get_available_engines(self):
|
||||
|
||||
@@ -1040,7 +1040,7 @@ multi 操作支持一次修改多处：
|
||||
|
||||
# 一行速览，完整命令清单收进 /help
|
||||
print(f"{Fore.CYAN}/help 全部命令 · /quit 退出 · /tui 图形界面 · "
|
||||
f"@文件路径 读文件给 AI{Style.RESET_ALL}")
|
||||
f"/model 换模型 · @文件路径 读文件给 AI{Style.RESET_ALL}")
|
||||
|
||||
extras = []
|
||||
loaded_engines = list(self.engines.keys())
|
||||
@@ -1141,6 +1141,11 @@ multi 操作支持一次修改多处：
|
||||
print(f"{Fore.RED}当前引擎不支持该命令或命令执行失败{Style.RESET_ALL}")
|
||||
continue
|
||||
|
||||
# /model 命令族 — OpenAI 引擎多模型在线增删切换
|
||||
if user_input == '/model' or user_input.startswith('/model '):
|
||||
self.handle_model_command(user_input[7:].strip())
|
||||
continue
|
||||
|
||||
# /manual 快捷命令 — 切换到 manual 引擎或执行 manual 子命令
|
||||
if user_input == '/manual' or user_input.startswith('/manual '):
|
||||
args = user_input[8:].strip() if user_input.startswith('/manual ') else ''
|
||||
|
||||
@@ -384,6 +384,10 @@ class DisplayMixin:
|
||||
('/tui', '切换到 TUI 图形模式'),
|
||||
('/engine list', '查看可用 AI 引擎'),
|
||||
('/engine switch <名>', '切换 AI 引擎'),
|
||||
('/model', '列出已配置的 OpenAI 模型'),
|
||||
('/model <名称>', '切换当前 OpenAI 模型'),
|
||||
('/model add', '交互式新增 OpenAI 模型'),
|
||||
('/model rm <名称>', '删除已配置的模型'),
|
||||
('/fc [reset]', '查看/清除 FC 能力探测缓存'),
|
||||
('/safe', '切换安全模式 (普通→人工→无限制)'),
|
||||
('/diff', '切换 diff 显示模式 (弹窗/主终端)'),
|
||||
|
||||
@@ -0,0 +1,145 @@
|
||||
"""OpenAI 格式模型注册表。
|
||||
|
||||
把「多个 OpenAI 格式模型」从 config.json 的单槽位升级为可管理的列表，
|
||||
支持在线增删、切换、持久化，免去每次手改配置文件。
|
||||
|
||||
配置结构（api.engines.openai）:
|
||||
{
|
||||
"models": [
|
||||
{"name": "讯飞xop", "base_url": "...", "api_key": "...", "model": "xophunyuan7bmt"},
|
||||
{"name": "gpt4o", "base_url": "https://api.openai.com/v1", "api_key": "...", "model": "gpt-4o"}
|
||||
],
|
||||
"current": "讯飞xop"
|
||||
}
|
||||
|
||||
向后兼容：若没有 models 列表，把扁平的 api_key/base_url/model 当作单条隐式模型，
|
||||
其 name 取 model 值（空则 "default"）。
|
||||
"""
|
||||
|
||||
from xcli_core.config import load_config, save_config
|
||||
|
||||
# 测试可临时覆盖配置路径
|
||||
_CONFIG_PATH = None
|
||||
|
||||
|
||||
def _set_config_path(path):
|
||||
"""仅供测试：指向临时 config.json"""
|
||||
global _CONFIG_PATH
|
||||
_CONFIG_PATH = path
|
||||
|
||||
|
||||
def _load():
|
||||
if _CONFIG_PATH:
|
||||
try:
|
||||
with open(_CONFIG_PATH, 'r', encoding='utf-8') as f:
|
||||
return __import__('json').load(f)
|
||||
except (FileNotFoundError, __import__('json').JSONDecodeError):
|
||||
return {"api": {"engines": {}}, "system": {}}
|
||||
return load_config()
|
||||
|
||||
|
||||
def _persist(cfg):
|
||||
if _CONFIG_PATH:
|
||||
with open(_CONFIG_PATH, 'w', encoding='utf-8') as f:
|
||||
__import__('json').dump(cfg, f, ensure_ascii=False, indent=2)
|
||||
return
|
||||
save_config(cfg)
|
||||
|
||||
|
||||
def _openai_block(cfg):
|
||||
return cfg.setdefault("api", {}).setdefault("engines", {}).setdefault("openai", {})
|
||||
|
||||
|
||||
def list_models(cfg=None):
|
||||
"""返回 (models:list[dict], current:str|None)。
|
||||
|
||||
无 models 列表时返回单条隐式模型（由扁平 api_key/base_url/model 构成）。
|
||||
"""
|
||||
cfg = cfg or _load()
|
||||
block = _openai_block(cfg)
|
||||
models = block.get("models")
|
||||
if not models:
|
||||
m = block.get("model") or "default"
|
||||
implicit = {
|
||||
"name": m,
|
||||
"base_url": block.get("base_url", ""),
|
||||
"api_key": block.get("api_key", ""),
|
||||
"model": block.get("model", ""),
|
||||
}
|
||||
return [implicit], (block.get("current") or m)
|
||||
current = block.get("current") or (models[0]["name"] if models else None)
|
||||
return models, current
|
||||
|
||||
|
||||
def get_model(name=None, cfg=None):
|
||||
"""取某模型配置；name=None 取 current。找不到返回 None。"""
|
||||
models, current = list_models(cfg)
|
||||
target = name or current
|
||||
if not target:
|
||||
return None
|
||||
for m in models:
|
||||
if m.get("name") == target:
|
||||
return dict(m)
|
||||
return None
|
||||
|
||||
|
||||
def set_current(name):
|
||||
"""把 current 指针指向 name 并落盘。name 必须存在，否则返回 False。"""
|
||||
cfg = _load()
|
||||
block = _openai_block(cfg)
|
||||
models = block.get("models") or []
|
||||
if not any(m.get("name") == name for m in models):
|
||||
return False
|
||||
block["current"] = name
|
||||
_persist(cfg)
|
||||
return True
|
||||
|
||||
|
||||
def add_model(entry):
|
||||
"""新增或更新模型，落盘并返回其 name。
|
||||
|
||||
entry: {"name", "base_url", "api_key", "model"}。同名则更新。
|
||||
首次添加会从扁平配置迁移出第一条隐式模型，且保持 current 不跳变。
|
||||
"""
|
||||
name = (entry.get("name") or "").strip()
|
||||
if not name:
|
||||
raise ValueError("模型 name 不能为空")
|
||||
cfg = _load()
|
||||
block = _openai_block(cfg)
|
||||
models = block.get("models")
|
||||
if not models:
|
||||
old_name = block.get("model") or "default"
|
||||
models = [{
|
||||
"name": old_name,
|
||||
"base_url": block.get("base_url", ""),
|
||||
"api_key": block.get("api_key", ""),
|
||||
"model": block.get("model", ""),
|
||||
}]
|
||||
block["models"] = models
|
||||
if not block.get("current"):
|
||||
block["current"] = old_name
|
||||
for m in models:
|
||||
if m.get("name") == name:
|
||||
m.update(entry)
|
||||
_persist(cfg)
|
||||
return name
|
||||
models.append(dict(entry))
|
||||
_persist(cfg)
|
||||
return name
|
||||
|
||||
|
||||
def remove_model(name):
|
||||
"""删除模型，返回是否成功（找不到返回 False）。
|
||||
若删除的是 current，则 current 回退到列表第一个。
|
||||
"""
|
||||
cfg = _load()
|
||||
block = _openai_block(cfg)
|
||||
models = block.get("models") or []
|
||||
new = [m for m in models if m.get("name") != name]
|
||||
if len(new) == len(models):
|
||||
return False
|
||||
block["models"] = new
|
||||
if block.get("current") == name:
|
||||
block["current"] = new[0]["name"] if new else None
|
||||
_persist(cfg)
|
||||
return True
|
||||
Reference in new issue
Block a user