From bf961e62d693ae1a8ec10aa4d36c2fedb84bfa23 Mon Sep 17 00:00:00 2001 From: chengma Date: Mon, 29 Jun 2026 08:45:04 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AE=8C=E6=88=90T-501=E8=AE=BE?= =?UTF-8?q?=E7=BD=AE=E9=A1=B5AI=E6=A8=A1=E5=9E=8B=E7=AE=A1=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增SettingsTab，接入AI模型下拉、新增、删除、详情编辑和保存 - 新增AIModelTestWorker，后台调用appconfig.test_ai_model测试连接 - 密钥输入使用密码框打码，删除保护保持text/image模型类别约束 - 补充GUI测试覆盖模型加载、保存删除和测试连接worker - 同步任务看板、current-state、routes、api和progress --- app/gui.py | 415 +++++++++++++++++++++++++++++++++++++++++- docs/06-tasks.md | 2 +- docs/api.md | 15 +- docs/current-state.md | 15 +- docs/routes.md | 14 +- progress.md | 10 + tests/test_gui.py | 155 +++++++++++++++- 7 files changed, 603 insertions(+), 23 deletions(-) diff --git a/app/gui.py b/app/gui.py index 7197b55..5fb89bd 100644 --- a/app/gui.py +++ b/app/gui.py @@ -11,6 +11,7 @@ try: from PySide6.QtWidgets import ( QAbstractItemView, QApplication, + QCheckBox, QComboBox, QDialog, QDialogButtonBox, @@ -2416,6 +2417,402 @@ if QT_IMPORT_ERROR is None: return {"alias": self.account.alias, "status": status} + class AIModelTestWorker(BaseWorker): + """Test one AI model connection without blocking the GUI thread.""" + + def __init__(self, model_name, ai_models_path=None): + super().__init__() + self.model_name = model_name + self.ai_models_path = ai_models_path or appconfig.AI_MODELS_PATH + + def execute(self): + result = appconfig.test_ai_model(self.model_name, path=self.ai_models_path) + payload = dict(result or {}) + payload["name"] = self.model_name + return payload + + + class SettingsTab(QWidget): + """Tab 5: AI model definitions stored in config/ai_models.json.""" + + CATEGORY_ITEMS = [("文本", "text"), ("图像", "image")] + API_TYPE_ITEMS = [("chat", "chat"), ("images_edits", "images_edits"), ("auto", "auto")] + + def __init__( + self, + parent=None, + config=None, + ai_models_path=None, + status_callback=None, + ): + super().__init__(parent) + self.config = appconfig.load_config() if config is None else config + self.ai_models_path = ( + ai_models_path + or self.config.get("ai_models_path") + or appconfig.AI_MODELS_PATH + ) + self.status_callback = status_callback + self.models = [] + self.current_model_name = None + self.test_worker = None + self.test_thread = None + + self.model_combo = QComboBox() + self.model_combo.setObjectName("aiModelCombo") + self.add_model_button = QPushButton("新增") + self.delete_model_button = QPushButton("删除") + + left_panel = QWidget() + left_layout = QVBoxLayout(left_panel) + left_layout.setContentsMargins(0, 0, 12, 0) + left_layout.addWidget(QLabel("AI 模型")) + left_layout.addWidget(self.model_combo) + left_toolbar = QHBoxLayout() + left_toolbar.addWidget(self.add_model_button) + left_toolbar.addWidget(self.delete_model_button) + left_toolbar.addStretch(1) + left_layout.addLayout(left_toolbar) + left_layout.addStretch(1) + + self.enabled_checkbox = QCheckBox("启用") + self.name_edit = QLineEdit() + self.name_edit.setObjectName("modelNameEdit") + self.category_combo = QComboBox() + self.category_combo.setObjectName("modelCategoryCombo") + for label, value in self.CATEGORY_ITEMS: + self.category_combo.addItem(label, value) + self.api_type_combo = QComboBox() + self.api_type_combo.setObjectName("modelApiTypeCombo") + for label, value in self.API_TYPE_ITEMS: + self.api_type_combo.addItem(label, value) + self.model_id_edit = QLineEdit() + self.model_id_edit.setObjectName("modelIdEdit") + self.url_edit = QLineEdit() + self.url_edit.setObjectName("modelUrlEdit") + self.api_key_edit = QLineEdit() + self.api_key_edit.setObjectName("modelApiKeyEdit") + self.api_key_edit.setEchoMode(QLineEdit.Password) + self.connect_timeout_spin = QSpinBox() + self.connect_timeout_spin.setObjectName("connectTimeoutSpin") + self.connect_timeout_spin.setRange(1, 3600) + self.connect_timeout_spin.setValue(30) + self.save_model_button = QPushButton("保存") + self.test_connection_button = QPushButton("测试连接") + self.test_result_label = QLabel("") + self.test_result_label.setWordWrap(True) + + form = QFormLayout() + form.addRow("", self.enabled_checkbox) + form.addRow("服务商名", self.name_edit) + form.addRow("类别", self.category_combo) + form.addRow("api_type", self.api_type_combo) + form.addRow("模型ID", self.model_id_edit) + form.addRow("网址", self.url_edit) + form.addRow("密钥", self.api_key_edit) + form.addRow("连接超时(秒)", self.connect_timeout_spin) + + action_layout = QHBoxLayout() + action_layout.addWidget(self.save_model_button) + action_layout.addWidget(self.test_connection_button) + action_layout.addStretch(1) + + right_panel = QWidget() + right_layout = QVBoxLayout(right_panel) + right_layout.setContentsMargins(12, 0, 0, 0) + right_layout.addWidget(QLabel("模型详情")) + right_layout.addLayout(form) + right_layout.addLayout(action_layout) + right_layout.addWidget(self.test_result_label) + right_layout.addStretch(1) + + self.splitter = QSplitter(Qt.Horizontal) + self.splitter.addWidget(left_panel) + self.splitter.addWidget(right_panel) + self.splitter.setStretchFactor(0, 1) + self.splitter.setStretchFactor(1, 3) + self.splitter.setSizes([280, 860]) + + layout = QVBoxLayout(self) + layout.setContentsMargins(18, 18, 18, 18) + layout.addWidget(self.splitter, 1) + + self.model_combo.currentIndexChanged.connect(self.load_selected_model) + self.add_model_button.clicked.connect(self.add_model) + self.delete_model_button.clicked.connect(self.delete_model) + self.save_model_button.clicked.connect(self.save_model) + self.test_connection_button.clicked.connect(self.test_connection) + + self.refresh_models() + + def _set_status(self, message): + if self.status_callback is not None: + self.status_callback(message) + + def refresh_models(self, selected=None): + try: + self.models = appconfig.list_ai_models( + path=self.ai_models_path, + reveal_api_key=True, + ) + except Exception as exc: + self.models = [] + self.current_model_name = None + self._show_error(exc) + + current = selected or self.current_model_name + self.model_combo.blockSignals(True) + self.model_combo.clear() + for model in self.models: + label = f"{model['name']} · {self._category_label(model['category'])}" + if not model.get("enabled", True): + label += " · 已停用" + self.model_combo.addItem(label, model["name"]) + index = self.model_combo.findData(current) + self.model_combo.setCurrentIndex(index if index >= 0 else (0 if self.models else -1)) + self.model_combo.blockSignals(False) + self.load_selected_model() + + def load_selected_model(self, index=None): + name = self.model_combo.currentData() + model = self._model_by_name(name) + self.current_model_name = model["name"] if model else None + self._populate_form(model) + self._update_button_state() + + def add_model(self, checked=False): + name = self._unique_model_name("新文本模型") + model = { + "name": name, + "category": "text", + "enabled": True, + "url": "", + "model": "", + "api_key": "", + "api_type": "chat", + "connect_timeout_seconds": 30, + "timeout_seconds": 0, + "extra_body": {}, + } + try: + appconfig.add_ai_model(model, path=self.ai_models_path) + except Exception as exc: + self._show_error(exc) + return + self.refresh_models(selected=name) + self._set_status(f"AI 模型已新增：{name}") + + def save_model(self, checked=False): + model = self._form_values() + if model is None: + return + try: + if self.current_model_name is None: + appconfig.add_ai_model(model, path=self.ai_models_path) + else: + appconfig.update_ai_model( + self.current_model_name, + path=self.ai_models_path, + **model, + ) + except Exception as exc: + self._show_error(exc) + return + self.refresh_models(selected=model["name"]) + self._set_status(f"AI 模型已保存：{model['name']}") + + def delete_model(self, checked=False): + model = self._current_model() + if model is None: + return + if not self._can_delete_model(model): + self._set_status("每个类别至少保留一个模型，当前模型不能删除") + return + answer = QMessageBox.question( + self, + "删除 AI 模型", + f"确认删除模型「{model['name']}」？", + QMessageBox.Yes | QMessageBox.No, + QMessageBox.No, + ) + if answer != QMessageBox.Yes: + return + try: + appconfig.delete_ai_model(model["name"], path=self.ai_models_path) + except Exception as exc: + self._show_error(exc) + return + self.refresh_models() + self._set_status(f"AI 模型已删除：{model['name']}") + + def test_connection(self, checked=False): + if self.test_thread is not None: + self._set_status("模型连接测试正在进行...") + return + model = self._current_model() + if model is None: + return + if self.name_edit.text().strip() != model["name"]: + self._set_status("请先保存模型名称变更后再测试连接") + return + worker = AIModelTestWorker(model["name"], ai_models_path=self.ai_models_path) + worker.finished.connect(self._on_test_finished) + worker.failed.connect(self._on_test_failed) + thread = run_worker(worker, thread_name="AIModelTestWorker", start=False) + thread.finished.connect(lambda: self._forget_test_thread(thread)) + self.test_worker = worker + self.test_thread = thread + self._set_test_running(True) + self.test_result_label.setText("正在测试连接...") + self._set_status(f"正在测试 AI 模型连接：{model['name']}") + thread.start() + + def _form_values(self): + current = self._current_model() or {} + name = self.name_edit.text().strip() + if not name: + self._show_error("AI 模型服务商名不能为空") + return None + extra_body = current.get("extra_body", {}) + if not isinstance(extra_body, dict): + extra_body = {} + return { + "name": name, + "category": self.category_combo.currentData() or "text", + "enabled": self.enabled_checkbox.isChecked(), + "url": self.url_edit.text().strip(), + "model": self.model_id_edit.text().strip(), + "api_key": self.api_key_edit.text(), + "api_type": self.api_type_combo.currentData() or "auto", + "connect_timeout_seconds": self.connect_timeout_spin.value(), + "timeout_seconds": int(current.get("timeout_seconds", 0) or 0), + "extra_body": dict(extra_body), + } + + def _populate_form(self, model): + widgets = [ + self.enabled_checkbox, + self.name_edit, + self.category_combo, + self.api_type_combo, + self.model_id_edit, + self.url_edit, + self.api_key_edit, + self.connect_timeout_spin, + ] + for widget in widgets: + widget.blockSignals(True) + if model is None: + self.enabled_checkbox.setChecked(False) + self.name_edit.clear() + self.category_combo.setCurrentIndex(0) + self.api_type_combo.setCurrentIndex(0) + self.model_id_edit.clear() + self.url_edit.clear() + self.api_key_edit.clear() + self.connect_timeout_spin.setValue(30) + else: + self.enabled_checkbox.setChecked(bool(model.get("enabled", True))) + self.name_edit.setText(model.get("name", "")) + self._set_combo_by_data(self.category_combo, model.get("category", "text")) + self._set_combo_by_data(self.api_type_combo, model.get("api_type", "auto")) + self.model_id_edit.setText(model.get("model", "")) + self.url_edit.setText(model.get("url", "")) + self.api_key_edit.setText(model.get("api_key", "")) + self.connect_timeout_spin.setValue( + int(model.get("connect_timeout_seconds", 30) or 30) + ) + for widget in widgets: + widget.blockSignals(False) + + def _set_combo_by_data(self, combo, value): + index = combo.findData(value) + combo.setCurrentIndex(index if index >= 0 else 0) + + def _update_button_state(self): + has_model = self._current_model() is not None + testing = self.test_thread is not None + for widget in ( + self.enabled_checkbox, + self.name_edit, + self.category_combo, + self.api_type_combo, + self.model_id_edit, + self.url_edit, + self.api_key_edit, + self.connect_timeout_spin, + self.save_model_button, + ): + widget.setEnabled(has_model and not testing) + self.add_model_button.setEnabled(not testing) + self.delete_model_button.setEnabled( + has_model and not testing and self._can_delete_model(self._current_model()) + ) + self.test_connection_button.setEnabled(has_model and not testing) + + def _set_test_running(self, running): + self._update_button_state() + self.test_connection_button.setEnabled( + not running and self._current_model() is not None + ) + + def _forget_test_thread(self, thread): + if self.test_thread is thread: + self.test_thread = None + self.test_worker = None + self._set_test_running(False) + + def _on_test_finished(self, payload): + if payload.get("ok"): + status = payload.get("status") + suffix = f"（HTTP {status}）" if status else "" + message = f"测试连接成功：{payload.get('name')}{suffix}" + else: + error = payload.get("error") or "连接失败" + status = payload.get("status") + status_text = f"HTTP {status}，" if status else "" + message = f"测试连接失败：{status_text}{error}" + self.test_result_label.setText(message) + self._set_status(message) + + def _on_test_failed(self, _task_id, error): + message = f"测试连接失败：{error}" + self.test_result_label.setText(message) + self._set_status(message) + + def _show_error(self, error): + message = str(error) + QMessageBox.warning(self, "设置", message) + self._set_status(message) + + def _current_model(self): + return self._model_by_name(self.current_model_name) + + def _model_by_name(self, name): + for model in self.models: + if model.get("name") == name: + return model + return None + + def _unique_model_name(self, base): + names = {model.get("name") for model in self.models} + if base not in names: + return base + counter = 2 + while f"{base} {counter}" in names: + counter += 1 + return f"{base} {counter}" + + def _can_delete_model(self, model): + if model is None: + return False + category = model.get("category") + return sum(1 for item in self.models if item.get("category") == category) > 1 + + def _category_label(self, category): + return {"text": "文本", "image": "图像"}.get(category, category) + + class AccountsTab(QWidget): COLUMNS = ["账号名", "别名", "地区", "端口", "登录状态", "备注"] @@ -2672,10 +3069,15 @@ if QT_IMPORT_ERROR is None: class MainWindow(QMainWindow): """Main application window with the fixed five-tab workflow.""" - def __init__(self, db_path=None, config=None): + def __init__(self, db_path=None, config=None, ai_models_path=None): super().__init__() self.config = appconfig.load_config() if config is None else config self.db_path = _database_path(db_path, self.config) + self.ai_models_path = ( + ai_models_path + or self.config.get("ai_models_path") + or appconfig.AI_MODELS_PATH + ) self.setWindowTitle("cmshopee") self.resize(1180, 760) self.tabs = QTabWidget() @@ -2714,12 +3116,11 @@ if QT_IMPORT_ERROR is None: config=self.config, status_callback=self.statusBar().showMessage, ) - widget = QWidget() - widget.setObjectName(title) - layout = QVBoxLayout(widget) - layout.setContentsMargins(18, 18, 18, 18) - layout.addStretch(1) - return widget + return SettingsTab( + config=self.config, + ai_models_path=self.ai_models_path, + status_callback=self.statusBar().showMessage, + ) def _on_tab_changed(self, index): self.statusBar().showMessage(f"当前：{self.tabs.tabText(index)}") diff --git a/docs/06-tasks.md b/docs/06-tasks.md index 33fcc13..9647440 100644 --- a/docs/06-tasks.md +++ b/docs/06-tasks.md @@ -78,7 +78,7 @@ | ID | 任务 | 依赖 | 验收要点 | 状态 | | --- | --- | --- | --- | --- | -| T-501 | Tab⑤ 设置 · AI 模型管理 UI（master-detail）：模型下拉 + 新增/删除 + 详情(服务商名/类别/api_type/模型ID/网址/密钥打码/连接超时) + 测试连接 | T-005, T-104 | 复用 T-005 后端；name 唯一；类别 text/image；至少各一个；密钥本地明文保存、UI 打码；测试连接可用 | TODO | +| T-501 | Tab⑤ 设置 · AI 模型管理 UI（master-detail）：模型下拉 + 新增/删除 + 详情(服务商名/类别/api_type/模型ID/网址/密钥打码/连接超时) + 测试连接 | T-005, T-104 | 复用 T-005 后端；name 唯一；类别 text/image；至少各一个；密钥本地明文保存、UI 打码；测试连接可用 | DONE | | T-501b | Tab⑤ 设置 · 角色与生成参数：标题大模型/图片大模型下拉、分辨率(512/1k/2k/4k)、返回超时随分辨率、并发/重试/jpg质量、路径/端口 | T-501 | 角色下拉按类别过滤；分辨率选后返回超时按 resolution_timeouts 自动；持久化 config.json | TODO | | T-501c | Tab⑤ 设置 · Shopee 更新安全开关 | T-501b | 配置测试商品 ID、是否允许真实提交、是否允许更新封面、单次最大更新条数、成功后是否自动关闭本次新开编辑 tab；默认关闭真实提交和封面更新；Tab③ 执行前读取并拦截不符合安全配置的更新 | TODO | | T-502 | 满 9 张封面：删第一张再上传 | T-001 | 实测删除确认框并处理；满 9 张流程正确 | TODO | diff --git a/docs/api.md b/docs/api.md index b84d571..891143d 100644 --- a/docs/api.md +++ b/docs/api.md @@ -293,14 +293,16 @@ render_prompt(template_text, task) -> str ```python # GUI 入口 main() -> int # 创建 QApplication + MainWindow -class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤；支持注入 db_path/config 便于测试 +class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤；支持注入 db_path/config/ai_models_path 便于测试 class CollectTab(QWidget) # ① 导入采集：导入 Excel + 汇总栏 + QTableView 任务列表 + 未匹配略过标记 class GenerateTab(QWidget) # ② AI生成：提示词管理 + 筛选任务 + 开始/停止生成 + 新旧封面预览 class ApplyTab(QWidget) # ③ 更新shopee：筛选已生成任务 + 确认后串行执行更新 +class SettingsTab(QWidget) # ⑤ 设置：AI 模型 master-detail 管理 class CollectWorker(BaseWorker) # ① 后台采集：账号就绪预检 -> editor.collect -> db.set_collected/mark_skipped/mark_failed class GenerateWorker(BaseWorker) # ② 后台生成：ai.generate_batch -> db.set_generated/mark_failed + 进度 class ApplyWorker(BaseWorker) # ③ 后台更新：账号就绪预检 -> editor.apply_task -> db.set_applied/mark_skipped class WriteBackWorker(BaseWorker) # ①/③ 后台回写：旧字段或更新结果写回原 Excel +class AIModelTestWorker(BaseWorker) # ⑤ 后台测试 AI 模型连接：appconfig.test_ai_model class TaskTableModel(QAbstractTableModel) # 任务表格模型：账号/别名/商品ID/阶段；未匹配别名显示“略过” class GenerateTaskTableModel(QAbstractTableModel) # ② 任务表格模型：店铺/商品ID/旧标题/新标题/状态 class ApplyTaskTableModel(QAbstractTableModel) # ③ 任务表格模型：店铺/商品ID/新标题/新封面/阶段/结果 @@ -310,7 +312,7 @@ TAB_TITLES: list[str] # 固定 Tab 顺序 TAB_STYLE: str # 顶层 Tab 栏防误点样式：最小宽度/padding/间距/当前态 ``` -`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局/提示词/开始生成/停止/封面对照预览、③ 更新shopee筛选列表与确认后串行更新、④ 账号管理；⑤ 仍为空壳，业务控件按后续 Tab 任务逐步填充。缺 PySide6 时 `main()` 返回 1 并输出明确提示。 +`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局/提示词/开始生成/停止/封面对照预览、③ 更新shopee筛选列表与确认后串行更新、④ 账号管理、⑤ AI 模型管理。缺 PySide6 时 `main()` 返回 1 并输出明确提示。 主 Tab 栏必须在 `MainWindow` 初始化时应用 `TAB_STYLE`：5 个 Tab 不使用 Qt 默认紧凑宽度，需保证点击区域稳定、间距清晰、当前 Tab 高亮明显。该样式属于全局导航基础，不归后续业务 Tab 任务重复实现。 @@ -320,6 +322,15 @@ TAB_STYLE: str # 顶层 Tab 栏防误点样式： - 弹窗字段：账号名、别名、地区、调试端口、密码、备注、slug、数据目录；密码框使用打码显示。 - 「启动登录」只启动 Chrome，人工登录；「检测登录」用 worker 跑 `accounts.detect_login()` 并刷新状态列；「快捷方式」调用 `accounts.create_shortcut()` 生成默认桌面 `.lnk`。 +⑤ 设置当前要点（T-501）： + +- `SettingsTab` 左侧为 AI 模型下拉与新增/删除按钮，右侧为模型详情表单。 +- 详情字段：启用、服务商名、类别（text/image）、api_type、模型 ID、网址、密钥、连接超时。 +- 模型数据读写复用 `appconfig.list_ai_models(..., reveal_api_key=True)`、`add_ai_model()`、`update_ai_model()`、`delete_ai_model()`；保存时保留现有 `extra_body` 与 `timeout_seconds`，不接管 T-501b 的生成参数。 +- 密钥字段使用 `QLineEdit.Password` 打码显示；明文只写入已 gitignore 的 `config/ai_models.json`，不得进入日志/导出。 +- 删除按钮在当前类别只剩 1 个模型时禁用；后端仍以“至少启用一个 text 和 image 模型”为硬约束。 +- 「测试连接」创建 `AIModelTestWorker` 后台调用 `appconfig.test_ai_model()`，GUI 主线程不直接发网络请求。 + ① 导入采集当前要点（T-202/T-202b）： - 「导入 Excel...」按钮调用 `excel.import_tasks(file_paths, path=db_path)`，导入成功后刷新当前批次任务。 diff --git a/docs/current-state.md b/docs/current-state.md index 8395027..435217a 100644 --- a/docs/current-state.md +++ b/docs/current-state.md @@ -6,10 +6,10 @@ ## 当前快照 - 日期：2026-06-27 -- 阶段：V0 单账号 CDP 流程已验证；V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-205b 采集后关闭自动新建商品页 tab、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度、T-401 Tab③ 更新列表筛选与开始更新确认、T-402 Tab③ 确认后串行更新、T-403 Tab③ 结果回写与结束汇总。 +- 阶段：V0 单账号 CDP 流程已验证；V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-205b 采集后关闭自动新建商品页 tab、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度、T-401 Tab③ 更新列表筛选与开始更新确认、T-402 Tab③ 确认后串行更新、T-403 Tab③ 结果回写与结束汇总、T-501 Tab⑤ AI 模型管理 UI。 - 技术栈：Python 3.10+，自研 CDP（websocket-client + requests），SQLite（sqlite3）+ `config.json` + openpyxl + AI（`config/ai_models.json` 通用 HTTP，chat JSON / images_edits），GUI PySide6 5 Tab（已定）。 -- 生产代码：已建立 `app/` 包 + 根入口 `main.py`；`app/cdp.py` 为已验证 CDP 底座，已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target；`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力，并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab；`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取，以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接；`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`，按默认文本/图片模型发起通用 HTTP 调用，支持重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed` 与停止取消未开始项；`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换；`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数；`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel、更新结果回写原 Excel 与另存副本；`app/config.py` 已实现账号 slug 与 user-data-dir 创建；`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录、检测登录、生成快捷方式；`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成；`app/gui.py` 已实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、① 导入采集的 Excel 导入按钮/导入汇总栏/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker、② AI生成左右布局/标题与封面提示词管理/批次店铺状态筛选/任务列表/变量预览/开始生成/停止/进度/双击新旧封面预览与 `GenerateWorker`、③ 更新shopee筛选栏/任务列表/开始更新确认弹窗/确认后 `ApplyWorker` 串行更新/账号就绪预检与④引导/逐条 `set_applied`/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏；`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。 -- 测试：`tests/` 已建立；T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`，当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测与商品 tab 生命周期/excel 导入/旧字段与更新结果回写/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集/gui ② AI生成布局筛选、提示词管理、生成 worker 与双击预览/gui ③ 更新shopee筛选列表、确认弹窗、`ApplyWorker` 串行更新与账号预检、结果回写与汇总/gui ④ 账号管理/worker signal 与线程包装；CDP/Shopee 改动仍需测试商品手动验证。 +- 生产代码：已建立 `app/` 包 + 根入口 `main.py`；`app/cdp.py` 为已验证 CDP 底座，已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target；`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力，并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab；`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取，以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接；`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`，按默认文本/图片模型发起通用 HTTP 调用，支持重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed` 与停止取消未开始项；`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换；`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数；`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel、更新结果回写原 Excel 与另存副本；`app/config.py` 已实现账号 slug 与 user-data-dir 创建；`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录、检测登录、生成快捷方式；`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成；`app/gui.py` 已实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、① 导入采集的 Excel 导入按钮/导入汇总栏/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker、② AI生成左右布局/标题与封面提示词管理/批次店铺状态筛选/任务列表/变量预览/开始生成/停止/进度/双击新旧封面预览与 `GenerateWorker`、③ 更新shopee筛选栏/任务列表/开始更新确认弹窗/确认后 `ApplyWorker` 串行更新/账号就绪预检与④引导/逐条 `set_applied`/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏、⑤ 设置 AI 模型 master-detail 管理与 `AIModelTestWorker` 后台测试连接；`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。 +- 测试：`tests/` 已建立；T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`，当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测与商品 tab 生命周期/excel 导入/旧字段与更新结果回写/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集/gui ② AI生成布局筛选、提示词管理、生成 worker 与双击预览/gui ③ 更新shopee筛选列表、确认弹窗、`ApplyWorker` 串行更新与账号预检、结果回写与汇总/gui ④ 账号管理/gui ⑤ AI 模型管理/worker signal 与线程包装；CDP/Shopee 改动仍需测试商品手动验证。 - 数据：`config.json`、`config/ai_models.json`、`cmshopee.db`、`chrome_user_data_dir/`、`images/` 已由 `.gitignore` 排除；运营填写后的 Excel 业务文件默认忽略，标准空模板 `shopee待处理任务模板.xlsx` 可提交；`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置，`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。 ## 既定设计要点（文档已定） @@ -32,7 +32,7 @@ | `prototypes/` | 已有 | 已验证原型/探查脚本（demo/set_title/set_cover/get_title/cookies/inspect_images/grab/1.py），保留作人工回归与探查参考；见 `prototypes/README.md` | | `chrome-remote-debug-lan.md` | 已有 | WSL→Windows CDP 转发排查记录 | | `app/__init__.py` / `app/__main__.py` / `main.py` | 已有 | 正式包与启动入口；`python main.py` / `python -m app` 可运行占位入口 | -| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-302/T-302p/T-303/T-401/T-402/T-403 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试；② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`；③ 更新shopee筛选栏、任务列表、开始更新确认弹窗、确认后 `ApplyWorker` 串行更新、账号预检与④引导、逐条 `set_applied`、自动回写结果到 Excel、结束汇总弹窗与手动回写重试；④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 | +| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-302/T-302p/T-303/T-401/T-402/T-403/T-501 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试；② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`；③ 更新shopee筛选栏、任务列表、开始更新确认弹窗、确认后 `ApplyWorker` 串行更新、账号预检与④引导、逐条 `set_applied`、自动回写结果到 Excel、结束汇总弹窗与手动回写重试；④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式；⑤ 设置 AI 模型下拉、新增/删除、详情编辑、密钥打码、测试连接 worker | | `app/workers.py` | 已有 | T-104b 产出：`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 | | `app/accounts.py` | 已有 | T-105/T-106 产出：账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 | | `app/editor.py` | 已有 | T-001/T-103/T-205b 产出：登录状态检测、打开商品页、读/写标题、读/下载封面、采集后关闭自动新建商品页 tab、上传拖封面、更新按钮、apply_task | @@ -42,7 +42,7 @@ | `app/db.py` | 已有 | T-003 产出：batches/accounts/tasks schema；WAL/busy_timeout/foreign_keys；账号/批次/任务与 set_* 阶段写库 | | `app/config.py` | 已有 | T-101 产出：别名→稳定 slug；创建并返回绝对 user-data-dir | | `app/chrome.py` | 已有 | T-102/T-106 产出：Chrome 启动参数、`subprocess.Popen` 启动、`/json/version` 端口探测、PowerShell `.lnk` 快捷方式 | -| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-301/T-302/T-302p/T-303/T-401/T-402/T-403 产出：stdlib unittest 基座；覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/prompts/gui/workers | +| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-301/T-302/T-302p/T-303/T-401/T-402/T-403/T-501 产出：stdlib unittest 基座；覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/prompts/gui/workers | | `app/excel.py` | 已有 | T-201/T-204/T-403 产出：多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计；按源文件/工作表/行号回写旧标题与旧封面路径；按源文件/工作表/行号回写新标题、新封面路径、更新状态；支持原文件被占用时另存副本 | | `shopee待处理任务模板.xlsx` | 已有，已提交 | 标准空 Excel 模板；单工作表 `待处理任务`，表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`；运营复制后填写，填写副本不提交 | | `config.json` / `config/ai_models.json` / `cmshopee.db` / `chrome_user_data_dir/` / `images/` | 本地存在或按需生成，已忽略 | 含配置、密钥、业务、登录态、图片，不提交版本库 | @@ -60,13 +60,14 @@ 任务状态以 [`06-tasks.md`](06-tasks.md) 为准，历史记录见 [`../progress.md`](../progress.md)。 -- 已完成：T-000（正式代码包结构）、T-001（`app/editor.py` 模块化）、T-002（`app/appconfig.py` + `config.json`）、T-003（SQLite 建表）、T-004（本地数据 gitignore）、T-005（AI 模型清单后端）、T-006（单元测试基座）、T-101（账号 slug/user-data-dir）、T-102（Chrome 启动器）、T-103（登录保活与检测）、T-104（PySide6 五 Tab 主窗口骨架）、T-104b（PySide6 worker 基座）、T-105（Tab④ 账号管理）、T-106（账号快捷方式）、T-201（Excel 导入：解析多文件输入列入库）、T-202（Tab① 任务列表 + 导入按钮 + 别名匹配标记）、T-202b（Tab① 导入汇总栏）、T-203（采集旧标题+旧封面）、T-204（回写旧字段到原 Excel）、T-204b（采集完成自动回写旧字段）、T-205（首次未配账号 / Chrome 未启动 / 未登录引导保护）、T-205b（采集后关闭自动新建商品页 tab）、T-301（AI 生成接口）、T-302（Tab② 左右布局与任务列表）、T-302p（提示词管理）、T-303（Tab② 开始生成 + 停止 + 进度）、T-401（Tab③ 更新列表筛选 + 开始更新确认弹窗）、T-402（Tab③ 确认后串行更新）、T-403（Tab③ 结果回写与结束汇总）。 +- 已完成：T-000（正式代码包结构）、T-001（`app/editor.py` 模块化）、T-002（`app/appconfig.py` + `config.json`）、T-003（SQLite 建表）、T-004（本地数据 gitignore）、T-005（AI 模型清单后端）、T-006（单元测试基座）、T-101（账号 slug/user-data-dir）、T-102（Chrome 启动器）、T-103（登录保活与检测）、T-104（PySide6 五 Tab 主窗口骨架）、T-104b（PySide6 worker 基座）、T-105（Tab④ 账号管理）、T-106（账号快捷方式）、T-201（Excel 导入：解析多文件输入列入库）、T-202（Tab① 任务列表 + 导入按钮 + 别名匹配标记）、T-202b（Tab① 导入汇总栏）、T-203（采集旧标题+旧封面）、T-204（回写旧字段到原 Excel）、T-204b（采集完成自动回写旧字段）、T-205（首次未配账号 / Chrome 未启动 / 未登录引导保护）、T-205b（采集后关闭自动新建商品页 tab）、T-301（AI 生成接口）、T-302（Tab② 左右布局与任务列表）、T-302p（提示词管理）、T-303（Tab② 开始生成 + 停止 + 进度）、T-401（Tab③ 更新列表筛选 + 开始更新确认弹窗）、T-402（Tab③ 确认后串行更新）、T-403（Tab③ 结果回写与结束汇总）、T-501（Tab⑤ AI 模型管理 UI）。 - 正在进行：无。 -- 下一个可领取任务：**T-501（Tab⑤ 设置 · AI 模型管理 UI）**。T-404 真实 Shopee 单条更新冒烟已登记，但依赖 T-501c 设置安全开关完成后再执行。 +- 下一个可领取任务：**T-501b（Tab⑤ 设置 · 角色与生成参数）**。T-404 真实 Shopee 单条更新冒烟已登记，但依赖 T-501c 设置安全开关完成后再执行。 ## 当前已知限制 - ① 采集依赖对应账号 Chrome 已用专属 user-data-dir 和 CDP 端口启动并登录；T-205 已在采集前拦截未配置账号、Chrome 未启动、未登录，并引导去④账号管理，但不会无提示批量启动所有账号 Chrome。 +- T-501 已完成 `config/ai_models.json` 模型清单 UI；标题/图片默认模型角色、并发、重试、分辨率、jpg 质量、路径/端口等仍留给 T-501b。 - T-301/T-303 已完成通用 HTTP AI 接口、批量生成编排和 GUI 接入 mock 单测；真实 AI 生成还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次成本可控的小样本实测。 - T-403 已完成③更新结果回写与结束汇总；真实 Shopee 更新冒烟暂不执行，延后到 T-501c 设置安全开关完成后做 T-404 单条验收。T-404 默认先只测标题更新，封面更新作为可选子项。 diff --git a/docs/routes.md b/docs/routes.md index 8a911d5..b9a0ef5 100644 --- a/docs/routes.md +++ b/docs/routes.md @@ -123,10 +123,13 @@ ## ⑤ 设置 -- AI：服务商 / 文本模型 / 图像模型 / API Key（本地明文存，UI 打码，不入日志）。 -- AI 生成参数：**标题并发数、图片并发数、失败重试次数、分辨率、jpg 质量**。 -- 本地图片目录（旧封面下载、新封面生成的存放根目录）。 -- Chrome 路径、默认调试端口 / 端口范围、超时、DB 路径。 +- AI 模型管理（T-501 已接入）：左侧模型下拉 + 新增/删除，右侧详情编辑。 + - 字段：启用、服务商名、类别（文本/图像）、api_type、模型ID、网址、密钥、连接超时。 + - 密钥存在 `config/ai_models.json`，本地明文保存；UI 使用密码框打码显示，不进入日志。 + - 保存复用 `appconfig` 模型清单后端，保持 name 唯一、至少 text/image 各一个模型。 + - 「测试连接」通过后台 worker 调 `appconfig.test_ai_model()`，不阻塞主线程。 +- AI 生成参数（T-501b）：标题/图片模型角色下拉、标题并发数、图片并发数、失败重试次数、分辨率、jpg 质量。 +- 路径与端口（T-501b）：本地图片目录、Chrome 路径、默认调试端口 / 端口范围、超时、DB 路径。 ## 流程导航 @@ -153,10 +156,11 @@ | `GenerateTab(QWidget)` | ② | 左提示词管理 + 右筛选/任务列表；双击看新旧封面；开始生成/停止/进度已接入 `GenerateWorker` | | `ApplyTab(QWidget)` | ③ | 已生成任务筛选 + 开始更新确认 + 确认后串行更新 + 结果回写与结束汇总 | | `AccountsTab(QWidget)` | ④ | 账号增删改、启动登录、检测登录、生成快捷方式 | -| `SettingsTab(QWidget)` | ⑤ | AI/目录/Chrome 配置 | +| `SettingsTab(QWidget)` | ⑤ | AI 模型 master-detail 管理；后续接入生成参数、目录、Chrome 配置 | | `TaskTableModel(QAbstractTableModel)` | ①②③ | 任务表格数据模型，供 `QTableView` 使用 | | `BaseWorker(QObject)` | 后台 | 定义 `progress/log/row_updated/failed/finished/cancelled` signals | | `ApplyWorker(BaseWorker)` | ③ | 账号就绪预检、串行调用 `editor.apply_task()`、逐条 `set_applied()`，失败继续 | +| `AIModelTestWorker(BaseWorker)` | ⑤ | 后台调用 `appconfig.test_ai_model()` 测试模型连接 | | `WriteBackWorker(BaseWorker)` | ①③ | ①回写旧字段；③回写新标题/新封面/更新状态 | > 采集、生成、更新都是耗时操作，使用 `QObject` worker + `QThread`。Worker 不直接操作 QWidget，只通过 signal 通知主线程刷新 UI。 diff --git a/progress.md b/progress.md index 777d59f..dff31e4 100644 --- a/progress.md +++ b/progress.md @@ -575,3 +575,13 @@ - 细化：取消语义明确为协作式取消，只阻止后续未开始任务，不中断当前正在执行的商品；Tab① 旧标题/旧封面采集已有真机冒烟，未真机闭环的是真实 AI 生成与 Tab③ 在线提交更新。 - 变更：更新 `docs/06-tasks.md`、`docs/current-state.md`、`docs/reviews/2026-06-27-T401-T403-acceptance.md`。 - 验证：仅文档更新，未运行测试。 + +## 【2026-06-27】T-501 Tab⑤ AI 模型管理 UI + +- 状态：DONE +- 变更：`app/gui.py` 新增 `SettingsTab`，将⑤设置从空壳替换为 AI 模型 master-detail 管理界面。左侧为模型下拉、新增、删除；右侧为启用、服务商名、类别、api_type、模型ID、网址、密钥、连接超时、保存、测试连接。 +- 安全：密钥字段使用 `QLineEdit.Password` 打码；模型清单仍写入 gitignore 的 `config/ai_models.json`；删除按钮在当前类别只剩一个模型时禁用，后端继续保证至少有 text/image 各一个启用模型。 +- Worker：新增 `AIModelTestWorker`，通过后台线程调用 `appconfig.test_ai_model()`，避免在 GUI 主线程发网络请求。 +- 测试：`tests/test_gui.py` 覆盖 MainWindow 挂载⑤、模型加载与密钥打码、新增/保存/删除、测试连接 worker、`AIModelTestWorker.execute()`。 +- 文档：`docs/06-tasks.md` 将 T-501 标为 DONE；同步 `docs/routes.md`、`docs/api.md`、`docs/current-state.md`，下一个可领取任务更新为 T-501b。 +- 验证：`python -m compileall app main.py tests` 通过；`python -m unittest discover -s tests -p test_gui.py` 通过（36 tests）；`python -m unittest discover -s tests` 通过（92 tests）。 diff --git a/tests/test_gui.py b/tests/test_gui.py index b0f049a..db4f335 100644 --- a/tests/test_gui.py +++ b/tests/test_gui.py @@ -9,7 +9,7 @@ sys.path.insert(0, os.path.dirname(__file__)) from _helpers import TempDirMixin from app import gui -from app import accounts, db, prompts +from app import accounts, appconfig, db, prompts if gui.QT_IMPORT_ERROR is not None: raise unittest.SkipTest("PySide6 未安装") @@ -20,6 +20,7 @@ from PySide6.QtWidgets import QApplication, QLineEdit, QPlainTextEdit, QTableVie from app.gui import ( AccountDialog, AccountsTab, + AIModelTestWorker, ApplyTab, ApplyWorker, CollectWorker, @@ -27,6 +28,7 @@ from app.gui import ( GenerateWorker, GenerateTab, MainWindow, + SettingsTab, TAB_STYLE, TAB_TITLES, WriteBackWorker, @@ -51,6 +53,7 @@ class GuiTests(TempDirMixin, unittest.TestCase): "image_dir": os.path.join(temp_dir, "images"), "db_path": os.path.join(temp_dir, "cmshopee.db"), "debug_port_range": [9222, 9260], + "ai_models_path": os.path.join(temp_dir, "ai_models.json"), } def test_main_window_has_five_tabs_in_workflow_order(self): @@ -71,6 +74,7 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assertIsInstance(window.tabs.widget(0), CollectTab) self.assertIsInstance(window.tabs.widget(1), GenerateTab) self.assertIsInstance(window.tabs.widget(2), ApplyTab) + self.assertIsInstance(window.tabs.widget(4), SettingsTab) self.assertEqual( "回写旧数据到 Excel", window.tabs.widget(0).write_back_button.text(), @@ -78,6 +82,155 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assert_removed(temp_dir) + def test_settings_tab_loads_ai_models_and_masks_key_field(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + models_path = cfg["ai_models_path"] + appconfig.save_ai_models_config( + { + "models": [ + { + "name": "Text A", + "category": "text", + "enabled": True, + "url": "https://example.invalid/text", + "model": "text-model", + "api_key": "sk-text-secret", + "api_type": "chat", + "connect_timeout_seconds": 11, + "timeout_seconds": 0, + "extra_body": {"temperature": 0}, + }, + { + "name": "Image A", + "category": "image", + "enabled": True, + "url": "https://example.invalid/image", + "model": "image-model", + "api_key": "sk-image-secret", + "api_type": "auto", + "connect_timeout_seconds": 22, + "timeout_seconds": 0, + "extra_body": {}, + }, + ] + }, + path=models_path, + ) + + tab = SettingsTab(config=cfg, ai_models_path=models_path) + self.addCleanup(tab.close) + + self.assertEqual(2, tab.model_combo.count()) + self.assertEqual("Text A", tab.name_edit.text()) + self.assertEqual("text", tab.category_combo.currentData()) + self.assertEqual("chat", tab.api_type_combo.currentData()) + self.assertEqual("text-model", tab.model_id_edit.text()) + self.assertEqual("https://example.invalid/text", tab.url_edit.text()) + self.assertEqual("sk-text-secret", tab.api_key_edit.text()) + self.assertEqual(QLineEdit.Password, tab.api_key_edit.echoMode()) + self.assertEqual(11, tab.connect_timeout_spin.value()) + self.assertFalse(tab.delete_model_button.isEnabled()) + + self.assert_removed(temp_dir) + + def test_settings_tab_adds_saves_and_deletes_model(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + models_path = cfg["ai_models_path"] + statuses = [] + tab = SettingsTab( + config=cfg, + ai_models_path=models_path, + status_callback=statuses.append, + ) + self.addCleanup(tab.close) + + tab.add_model() + self.assertEqual("新文本模型", tab.current_model_name) + self.assertEqual(3, tab.model_combo.count()) + + tab.name_edit.setText("Text Custom") + tab.url_edit.setText("https://example.invalid/v1/chat/completions") + tab.model_id_edit.setText("demo-text") + tab.api_key_edit.setText("sk-custom-secret") + tab.api_type_combo.setCurrentIndex(tab.api_type_combo.findData("chat")) + tab.connect_timeout_spin.setValue(12) + tab.save_model() + + saved = appconfig.get_model("Text Custom", path=models_path) + self.assertEqual("text", saved["category"]) + self.assertEqual("demo-text", saved["model"]) + self.assertEqual("sk-custom-secret", saved["api_key"]) + self.assertEqual(12, saved["connect_timeout_seconds"]) + self.assertIn("AI 模型已保存：Text Custom", statuses[-1]) + self.assertTrue(tab.delete_model_button.isEnabled()) + + with mock.patch("app.gui.QMessageBox.question", return_value=gui.QMessageBox.Yes): + tab.delete_model() + + names = [model["name"] for model in appconfig.list_ai_models(path=models_path)] + self.assertNotIn("Text Custom", names) + self.assertIn("AI 模型已删除：Text Custom", statuses[-1]) + + self.assert_removed(temp_dir) + + def test_settings_tab_starts_connection_test_worker(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + models_path = cfg["ai_models_path"] + statuses = [] + tab = SettingsTab( + config=cfg, + ai_models_path=models_path, + status_callback=statuses.append, + ) + self.addCleanup(tab.close) + + class FakeSignal: + def __init__(self): + self.callbacks = [] + + def connect(self, callback): + self.callbacks.append(callback) + + class FakeThread: + def __init__(self): + self.finished = FakeSignal() + self.started = False + + def start(self): + self.started = True + + fake_thread = FakeThread() + with mock.patch("app.gui.run_worker", return_value=fake_thread) as run_worker: + tab.test_connection() + + run_worker.assert_called_once() + self.assertIsInstance(tab.test_worker, AIModelTestWorker) + self.assertIs(tab.test_thread, fake_thread) + self.assertTrue(fake_thread.started) + self.assertFalse(tab.test_connection_button.isEnabled()) + self.assertIn("正在测试 AI 模型连接", statuses[-1]) + + self.assert_removed(temp_dir) + + def test_ai_model_test_worker_calls_appconfig(self): + with self.make_temp_dir() as temp_dir: + models_path = os.path.join(temp_dir, "ai_models.json") + worker = AIModelTestWorker("Text A", ai_models_path=models_path) + + with mock.patch( + "app.gui.appconfig.test_ai_model", + return_value={"ok": True, "status": 200}, + ) as test_ai_model: + result = worker.execute() + + test_ai_model.assert_called_once_with("Text A", path=models_path) + self.assertEqual({"ok": True, "status": 200, "name": "Text A"}, result) + + self.assert_removed(temp_dir) + def test_generate_tab_has_prompt_editors_and_task_table(self): with self.make_temp_dir() as temp_dir: title_prompt_path = os.path.join(temp_dir, "title_prompt.txt")