feat: SSHClient v1.0.0 - PyQt5 + paramiko 跨平台 SSH 客户端

功能: - 多主机管理 (增删改查, 密码/私钥双认证, 导入导出) - 远程终端 (命令执行 + 常用命令快捷栏 + 超时控制) - SFTP 文件浏览 (上传/下载带进度, 新建/删除/重命名) - 实时监控 (CPU/内存/磁盘/网络, 1-10秒可调刷新) - AI Agent (OpenAI 兼容 API, 5 工具自动调用: 命令/指标/列文件/读文件/上传) 技术栈: PyQt5 + paramiko + psutil + requests + PyInstaller 打包: build_windows.bat / build.sh 一键产出 ~57MB 单文件 exe 测试: core 6/6 + UI 3/3 + E2E 6/6 全部通过
2026-07-28 21:01:31 +08:00
commit a57dbc0252
22 changed files with 3294 additions and 0 deletions
@@ -0,0 +1,135 @@
+"""
+AI Agent 配置对话框
+支持自定义 OpenAI 兼容 API（OpenAI / DeepSeek / Moonshot / Ollama 等）
+"""
+from PyQt5.QtCore import Qt
+from PyQt5.QtWidgets import (
+    QDialog, QVBoxLayout, QHBoxLayout, QFormLayout, QLineEdit, QPushButton,
+    QComboBox, QTextEdit, QLabel, QDialogButtonBox, QMessageBox, QGroupBox,
+)
+
+from core.ai_agent import AIAgent
+from core.manager import load_ai_config, save_ai_config
+
+
+PRESETS = [
+    ("OpenAI 官方", "https://api.openai.com/v1", "gpt-4o-mini"),
+    ("DeepSeek", "https://api.deepseek.com/v1", "deepseek-chat"),
+    ("Moonshot Kimi", "https://api.moonshot.cn/v1", "moonshot-v1-8k"),
+    ("通义千问 (DashScope 兼容)", "https://dashscope.aliyuncs.com/compatible-mode/v1", "qwen-turbo"),
+    ("智谱 GLM (BigModel)", "https://open.bigmodel.cn/api/paas/v4", "glm-4-flash"),
+    ("Ollama (本地)", "http://127.0.0.1:11434/v1", "qwen2.5:7b"),
+    ("自定义", "", ""),
+]
+
+
+class AIConfigDialog(QDialog):
+    def __init__(self, agent: AIAgent, parent=None):
+        super().__init__(parent)
+        self.agent = agent
+        self.setWindowTitle("AI Agent 设置")
+        self.resize(560, 480)
+        self._build()
+
+        # 用持久化的配置优先，再用 agent 当前值填充
+        cfg = load_ai_config()
+        self.api_key_edit.setText(cfg.get("api_key", "") or agent.api_key)
+        self.base_url_edit.setText(cfg.get("base_url", "") or agent.base_url)
+        self.model_edit.setText(cfg.get("model", "") or agent.model)
+        self.system_prompt_edit.setPlainText(
+            cfg.get("system_prompt", "") or agent.system_prompt
+        )
+        # 匹配预设
+        for i, (_, url, model) in enumerate(PRESETS):
+            if url and url == self.base_url_edit.text() and model == self.model_edit.text():
+                self.preset_combo.setCurrentIndex(i)
+                break
+
+    def _build(self):
+        layout = QVBoxLayout(self)
+
+        # 预设
+        preset_box = QGroupBox("服务商预设")
+        pv = QVBoxLayout(preset_box)
+        row = QHBoxLayout()
+        row.addWidget(QLabel("快速选择:"))
+        self.preset_combo = QComboBox()
+        for name, _, _ in PRESETS:
+            self.preset_combo.addItem(name)
+        self.preset_combo.currentIndexChanged.connect(self._on_preset_changed)
+        row.addWidget(self.preset_combo, 1)
+        pv.addLayout(row)
+        layout.addWidget(preset_box)
+
+        # 表单
+        form_box = QGroupBox("API 配置")
+        form = QFormLayout(form_box)
+        self.api_key_edit = QLineEdit()
+        self.api_key_edit.setEchoMode(QLineEdit.Password)
+        self.api_key_edit.setPlaceholderText("sk-...")
+        form.addRow("API Key:", self.api_key_edit)
+        self.base_url_edit = QLineEdit()
+        self.base_url_edit.setPlaceholderText("https://api.openai.com/v1")
+        form.addRow("Base URL:", self.base_url_edit)
+        self.model_edit = QLineEdit()
+        self.model_edit.setPlaceholderText("gpt-4o-mini")
+        form.addRow("Model:", self.model_edit)
+        layout.addWidget(form_box)
+
+        # 系统提示
+        sp_box = QGroupBox("系统提示词（可自定义 AI 行为）")
+        sv = QVBoxLayout(sp_box)
+        self.system_prompt_edit = QTextEdit()
+        self.system_prompt_edit.setMaximumHeight(120)
+        sv.addWidget(self.system_prompt_edit)
+        layout.addWidget(sp_box)
+
+        # 按钮
+        bb = QDialogButtonBox()
+        self.btn_test = bb.addButton("测试连接", QDialogButtonBox.ActionRole)
+        self.btn_test.clicked.connect(self._test)
+        bb.addButton(QDialogButtonBox.Save).clicked.connect(self._save)
+        bb.addButton(QDialogButtonBox.Cancel).clicked.connect(self.reject)
+        layout.addWidget(bb)
+
+    def _on_preset_changed(self, idx: int):
+        name, url, model = PRESETS[idx]
+        if url:
+            self.base_url_edit.setText(url)
+            self.model_edit.setText(model)
+
+    def _save(self):
+        api_key = self.api_key_edit.text().strip()
+        base_url = self.base_url_edit.text().strip()
+        model = self.model_edit.text().strip()
+        system_prompt = self.system_prompt_edit.toPlainText().strip()
+        if not base_url or not model:
+            QMessageBox.warning(self, "保存失败", "Base URL 和 Model 必填")
+            return
+        self.agent.update_config(api_key, base_url, model, system_prompt)
+        save_ai_config({
+            "api_key": api_key, "base_url": base_url,
+            "model": model, "system_prompt": system_prompt,
+        })
+        self.accept()
+
+    def _test(self):
+        # 保存到 agent 临时测一下
+        api_key = self.api_key_edit.text().strip()
+        base_url = self.base_url_edit.text().strip()
+        model = self.model_edit.text().strip()
+        if not api_key or not base_url or not model:
+            QMessageBox.warning(self, "提示", "请先填好 API Key / Base URL / Model")
+            return
+        # 保存一份用于测试的临时 agent
+        tmp = AIAgent(api_key=api_key, base_url=base_url, model=model)
+        try:
+            from PyQt5.QtWidgets import QApplication
+            QApplication.setOverrideCursor(Qt.WaitCursor)
+            try:
+                tmp.chat("ping", lambda n, a: "ok")
+            finally:
+                QApplication.restoreOverrideCursor()
+            QMessageBox.information(self, "成功", "连接成功！")
+        except Exception as e:
+            QMessageBox.critical(self, "失败", f"连接失败：\n{e}")