diff --git a/main/xiaozhi-server/core/connection.py b/main/xiaozhi-server/core/connection.py index 8a5f1ffd..9a59d388 100644 --- a/main/xiaozhi-server/core/connection.py +++ b/main/xiaozhi-server/core/connection.py @@ -216,26 +216,67 @@ class ConnectionHandler: async def handle_config_update(self, message): """处理配置更新请求""" - config_model = message.get("model") - new_content = message.get("content") + content = message.get("content", {}) + secret = content.pop("secret", None) + new_config = content - # 打印更新前的配置 - old_value = self.config.get(config_model) - self.logger.bind(tag=TAG).info(f"配置更新: {config_model} 从 {old_value} 更新为 {new_content}") + # 密钥验证 + if secret != "43c716ae-37c4-49ba-84ab-2f0fbfcd7115": + self.logger.bind(tag=TAG).warning("配置更新请求的密钥不正确") + await self.websocket.send(json.dumps({ + "type": "config_update_response", + "status": "error", + "message": "密钥不正确" + })) + return - # 更新配置 - self.config[config_model] = new_content + # 遍历所有支持的配置模块 + updated_modules = [] + for config_model in ["tts", "llm", "vad", "asr", "memory", "intent"]: + if config_model not in new_config: + continue - # 如果是模型相关配置,重新初始化模型 - if config_model in ["LLM", "TTS", "ASR", "VAD", "Intent", "Memory"]: - self._initialize_components(self.config) - self.logger.bind(tag=TAG).info(f"已使用新配置重新初始化 {config_model} 模块") + new_content = new_config[config_model] + old_content = self.config.get(config_model, {}) - # 返回更新确认 + # 记录配置变更 + self.logger.bind(tag=TAG).info( + f"配置更新: {config_model} 旧值: {json.dumps(old_content, ensure_ascii=False)} " + f"新值: {json.dumps(new_content, ensure_ascii=False)}" + ) + + # 深度合并配置 + if isinstance(old_content, dict) and isinstance(new_content, dict): + merged = {**old_content, **new_content} + self.config[config_model] = merged + else: + self.config[config_model] = new_content + + # 标记需要重新初始化的模块 + if config_model in ["llm", "tts", "asr", "vad", "intent", "memory"]: + updated_modules.append(config_model) + + # 批量初始化模块 + if updated_modules: + try: + self._initialize_components(self.config) + self.logger.bind(tag=TAG).info( + f"已重新初始化模块: {', '.join(updated_modules)}" + ) + except Exception as e: + self.logger.bind(tag=TAG).error(f"模块初始化失败: {str(e)}") + await self.websocket.send(json.dumps({ + "type": "config_update_response", + "status": "error", + "message": f"模块初始化失败: {str(e)}" + })) + return + + # 返回成功响应 await self.websocket.send(json.dumps({ "type": "config_update_response", "status": "success", - "message": f"{config_model} 配置已更新" + "message": f"已更新配置: {', '.join(updated_modules)}" })) def _initialize_components(self, private_config): diff --git a/main/xiaozhi-server/core/handle/textHandle.py b/main/xiaozhi-server/core/handle/textHandle.py index f65d436a..f0683d59 100644 --- a/main/xiaozhi-server/core/handle/textHandle.py +++ b/main/xiaozhi-server/core/handle/textHandle.py @@ -62,7 +62,7 @@ async def handleTextMessage(conn, message): if "states" in msg_json: asyncio.create_task(handleIotStatus(conn, msg_json["states"])) elif msg_json["type"] == "server": - if "model" in msg_json: + if msg_json["action"] == "update_config": await conn.handle_config_update(msg_json) except json.JSONDecodeError: await conn.websocket.send(message)