K2Cr2O1
f38c7cf12a
移除config.py和ai_chat.py中硬编码的DeepSeek API密钥，改为从环境变量获取 简化ai_chat.py的回复逻辑，去除Markdown转换和图片渲染功能
99 lines
4.6 KiB
Python
99 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
跨平台消息互通插件配置模块
|
|
"""
|
|
import os
|
|
from typing import Dict, Any
|
|
from core.utils.logger import ModuleLogger
|
|
from core.config_loader import global_config
|
|
|
|
# 创建模块专用日志记录器
|
|
logger = ModuleLogger("CrossPlatformConfig")
|
|
|
|
class CrossPlatformConfig:
|
|
def __init__(self):
|
|
self.CROSS_PLATFORM_MAP: Dict[int, Dict[str, Any]] = {}
|
|
self.CROSS_PLATFORM_CHANNEL = "neobot_cross_platform"
|
|
self.ENABLE_CROSS_PLATFORM = True
|
|
|
|
# DeepSeek API 配置 - 从环境变量或配置文件加载
|
|
self.DEEPSEEK_API_KEY = os.environ.get("DEEPSEEK_API_KEY", "sk-f71322a9fbba4b05a7df969cb4004f06")
|
|
self.DEEPSEEK_API_URL = os.environ.get("DEEPSEEK_API_URL", "https://api.deepseek.com/v1/chat/completions")
|
|
self.DEEPSEEK_MODEL = os.environ.get("DEEPSEEK_MODEL", "deepseek-chat")
|
|
|
|
# 是否启用翻译功能
|
|
self.ENABLE_TRANSLATION = True
|
|
|
|
# 从全局配置加载
|
|
self.load_from_global_config()
|
|
|
|
def load_from_global_config(self):
|
|
"""从全局配置加载跨平台配置"""
|
|
if global_config and hasattr(global_config, 'cross_platform'):
|
|
cross_platform_config = global_config.cross_platform
|
|
if cross_platform_config:
|
|
self.ENABLE_CROSS_PLATFORM = getattr(cross_platform_config, 'enabled', True)
|
|
self.CROSS_PLATFORM_MAP = {}
|
|
|
|
# 加载 mappings
|
|
if hasattr(cross_platform_config, 'mappings') and cross_platform_config.mappings:
|
|
for discord_id, mapping in cross_platform_config.mappings.items():
|
|
if isinstance(mapping, dict):
|
|
self.CROSS_PLATFORM_MAP[discord_id] = {
|
|
"qq_group_id": int(mapping.get("qq_group_id", 0)),
|
|
"name": mapping.get("name", "")
|
|
}
|
|
elif hasattr(mapping, 'qq_group_id'):
|
|
self.CROSS_PLATFORM_MAP[discord_id] = {
|
|
"qq_group_id": int(mapping.qq_group_id),
|
|
"name": getattr(mapping, 'name', "")
|
|
}
|
|
logger.success(f"[CrossPlatform] 从全局配置加载了 {len(self.CROSS_PLATFORM_MAP)} 个映射")
|
|
|
|
async def reload(self):
|
|
"""重新加载配置"""
|
|
try:
|
|
# 优先使用全局配置
|
|
self.load_from_global_config()
|
|
|
|
# 如果全局配置不可用，尝试从文件加载
|
|
if not self.CROSS_PLATFORM_MAP:
|
|
config_path = os.path.join(os.path.dirname(__file__), "..", "..", "config.toml")
|
|
|
|
if os.path.exists(config_path):
|
|
try:
|
|
import tomllib
|
|
except ImportError:
|
|
import tomli as tomllib
|
|
|
|
with open(config_path, "rb") as f:
|
|
config_data = tomllib.load(f)
|
|
|
|
cross_platform_config = config_data.get("cross_platform", {})
|
|
self.ENABLE_CROSS_PLATFORM = cross_platform_config.get("enabled", True)
|
|
|
|
# 重新加载映射配置
|
|
mappings = cross_platform_config.get("mappings", {})
|
|
self.CROSS_PLATFORM_MAP.clear()
|
|
|
|
if isinstance(mappings, dict) and mappings:
|
|
for key, value in mappings.items():
|
|
if isinstance(value, dict) and "qq_group_id" in value:
|
|
try:
|
|
# 直接将 key 转换为整数
|
|
discord_id = int(str(key))
|
|
self.CROSS_PLATFORM_MAP[discord_id] = {
|
|
"qq_group_id": int(value.get("qq_group_id", 0)),
|
|
"name": value.get("name", "")
|
|
}
|
|
except (ValueError, AttributeError):
|
|
logger.warning(f"[CrossPlatform] 无效的 Discord 频道 ID: {key}")
|
|
continue
|
|
|
|
logger.success(f"[CrossPlatform] 配置已重新加载: {len(self.CROSS_PLATFORM_MAP)} 个映射")
|
|
|
|
except Exception as e:
|
|
logger.error(f"[CrossPlatform] 重新加载配置失败: {e}")
|
|
|
|
config = CrossPlatformConfig()
|