From 08ded8baee1fee75de53b0315619003bf24190aa Mon Sep 17 00:00:00 2001 From: liuwei Date: Mon, 20 Apr 2026 15:15:38 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=A7=BB=E9=99=A4t=5Fllm=5Fconfig?= =?UTF-8?q?=E7=9B=B8=E5=85=B3=E4=BB=A3=E7=A0=81=E4=BE=9D=E8=B5=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 变更项： 1. 删除废弃数据层文件 db/llm_config_db.py。 2. Robot 启动流程移除 llm_config_db 初始化与导入逻辑。 3. DashboardServer 移除 llm_config_db 挂载引用。 4. 复查确认代码中已无 t_llm_config/llm_config_db/LLMConfigDBOperator 运行时依赖。 --- admin/dashboard/server.py | 1 - db/llm_config_db.py | 125 -------------------------------------- robot.py | 8 --- 3 files changed, 134 deletions(-) delete mode 100644 db/llm_config_db.py diff --git a/admin/dashboard/server.py b/admin/dashboard/server.py index 0d0d782..a8061f0 100644 --- a/admin/dashboard/server.py +++ b/admin/dashboard/server.py @@ -52,7 +52,6 @@ class DashboardServer: self.plugin_schedule_manager = robot_instance.plugin_schedule_manager self.group_plugin_config_db = robot_instance.group_plugin_config_db self.llm_catalog_db = robot_instance.llm_catalog_db - self.llm_config_db = robot_instance.llm_config_db self.group_plugin_config_service = robot_instance.group_plugin_config_service # 获取联系人管理器实例 self.contact_manager = robot_instance.contact_manager diff --git a/db/llm_config_db.py b/db/llm_config_db.py deleted file mode 100644 index 8d65b58..0000000 --- a/db/llm_config_db.py +++ /dev/null @@ -1,125 +0,0 @@ -# -*- coding: utf-8 -*- -import json -from typing import Any, Dict - -from loguru import logger - -from db.base import BaseDBOperator -from db.connection import DBConnectionManager - - -class LLMConfigDBOperator(BaseDBOperator): - """LLM 配置数据库操作器。 - - 设计目标： - 1. 把原先存放在 config.yaml 的 llm 配置迁移到 MySQL，便于后台实时维护； - 2. 采用“单行配置”模型，降低维护复杂度：一行记录保存 default_backend/backends/scenes； - 3. 支持“首次启动自动导入 YAML 配置”，保证迁移过程对线上透明。 - """ - - def __init__(self, db_manager: DBConnectionManager): - super().__init__(db_manager) - - def init_tables(self) -> bool: - """初始化 LLM 配置表。 - - 字段说明： - - id: 固定主键，当前仅使用 id=1 作为全局配置； - - default_backend: 全局默认后端； - - backends_json: 后端配置大对象（JSON 字符串）； - - scenes_json: 场景路由对象（JSON 字符串）； - - source: 记录当前配置来源，便于后续排障。 - """ - try: - return self.execute_update( - """ - CREATE TABLE IF NOT EXISTS t_llm_config ( - id TINYINT PRIMARY KEY, - default_backend VARCHAR(128) NOT NULL DEFAULT '', - backends_json JSON NOT NULL, - scenes_json JSON NOT NULL, - source VARCHAR(32) NOT NULL DEFAULT 'mysql', - created_at DATETIME DEFAULT CURRENT_TIMESTAMP, - updated_at DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP - ) - """ - ) - except Exception as e: - logger.error(f"初始化 t_llm_config 失败: {e}") - return False - - @staticmethod - def _loads_json(value: Any) -> Dict[str, Any]: - """将数据库 JSON 字段统一解析为 dict。""" - if isinstance(value, dict): - return value - if isinstance(value, str): - try: - obj = json.loads(value) - return obj if isinstance(obj, dict) else {} - except json.JSONDecodeError: - return {} - return {} - - def get_config(self) -> Dict[str, Any]: - """读取数据库中的 LLM 配置。""" - row = self.execute_query( - """ - SELECT id, default_backend, backends_json, scenes_json, source, updated_at - FROM t_llm_config - WHERE id = 1 - LIMIT 1 - """, - fetch_one=True, - ) or {} - if not row: - return {} - - return { - "default_backend": str(row.get("default_backend") or "").strip(), - "backends": self._loads_json(row.get("backends_json")), - "scenes": self._loads_json(row.get("scenes_json")), - "source": str(row.get("source") or "mysql").strip(), - "updated_at": row.get("updated_at"), - } - - def save_config(self, llm_config: Dict[str, Any], source: str = "mysql") -> bool: - """保存（覆盖）LLM 配置到数据库。""" - data = llm_config or {} - default_backend = str(data.get("default_backend") or "").strip() - backends = data.get("backends", {}) or {} - scenes = data.get("scenes", {}) or {} - if not isinstance(backends, dict): - backends = {} - if not isinstance(scenes, dict): - scenes = {} - - sql = """ - INSERT INTO t_llm_config (id, default_backend, backends_json, scenes_json, source) - VALUES (1, %s, %s, %s, %s) - ON DUPLICATE KEY UPDATE - default_backend = VALUES(default_backend), - backends_json = VALUES(backends_json), - scenes_json = VALUES(scenes_json), - source = VALUES(source) - """ - params = ( - default_backend, - json.dumps(backends, ensure_ascii=False), - json.dumps(scenes, ensure_ascii=False), - str(source or "mysql"), - ) - return self.execute_update(sql, params) - - def bootstrap_from_yaml_if_empty(self, yaml_llm_config: Dict[str, Any]) -> bool: - """当数据库为空时，把 YAML 里的 llm 配置导入到数据库。 - - 迁移策略： - 1. 只在“表中无 id=1 配置”时执行，避免覆盖后台已维护的数据； - 2. 导入后标记 source=yaml_bootstrap，便于识别初始数据来源； - 3. 返回 True 表示“已有配置或导入成功”，False 表示导入失败。 - """ - existed = self.get_config() - if existed: - return True - return self.save_config(yaml_llm_config or {}, source="yaml_bootstrap") diff --git a/robot.py b/robot.py index ed4a4c4..fbfcbb3 100644 --- a/robot.py +++ b/robot.py @@ -19,7 +19,6 @@ from db.connection import DBConnectionManager from db.contacts_db import ContactsDBOperator from db.group_plugin_config_db import GroupPluginConfigDBOperator from db.llm_catalog_db import LLMCatalogDBOperator -from db.llm_config_db import LLMConfigDBOperator from db.plugin_schedule_db import PluginScheduleDBOperator from db.system_job_db import SystemJobDBOperator from utils.system_jobs import SystemJobLoader @@ -74,19 +73,12 @@ class Robot: self.contacts_db = ContactsDBOperator(self.db_manager) self.group_plugin_config_db = GroupPluginConfigDBOperator(self.db_manager) self.llm_catalog_db = LLMCatalogDBOperator(self.db_manager) - self.llm_config_db = LLMConfigDBOperator(self.db_manager) self.plugin_schedule_db = PluginScheduleDBOperator(self.db_manager) self.system_job_db = SystemJobDBOperator(self.db_manager) self.group_plugin_config_db.init_tables() # 新版 LLM 目录模型（Provider 模板 / Dify 应用 / Scene）初始化。 self.llm_catalog_db.init_tables() self.llm_catalog_db.bootstrap_from_legacy_llm(self.config.llm) - # LLM 配置迁移到 MySQL： - # 1. 先确保表存在； - # 2. 若库里没有配置，则从 config.yaml 的 llm 节点导入一次； - # 3. 后续运行时以数据库为准，YAML 仅作为初始导入来源与兜底。 - self.llm_config_db.init_tables() - self.llm_config_db.bootstrap_from_yaml_if_empty(self.config.llm) self.group_plugin_config_service = GroupPluginConfigService( db_operator=self.group_plugin_config_db, redis_client=self.db_manager.get_redis_connection(),