558 lines
25 KiB
Python
558 lines
25 KiB
Python
from addons.dashboard.server import AstrBotDashBoard, DashBoardData
|
||
from pydantic import BaseModel
|
||
from typing import Union, Optional
|
||
import uuid
|
||
from util import general_utils as gu
|
||
from util.cmd_config import CmdConfig
|
||
from dataclasses import dataclass
|
||
import sys
|
||
import os
|
||
import threading
|
||
import time
|
||
import asyncio
|
||
from util.plugin_dev.api.v1.config import update_config
|
||
|
||
@dataclass
|
||
class DashBoardConfig():
|
||
config_type: str
|
||
name: Optional[str] = None
|
||
description: Optional[str] = None
|
||
path: Optional[str] = None # 仅 item 才需要
|
||
body: Optional[list['DashBoardConfig']] = None # 仅 group 才需要
|
||
value: Optional[Union[list, dict, str, int, bool]] = None # 仅 item 才需要
|
||
val_type: Optional[str] = None # 仅 item 才需要
|
||
|
||
class DashBoardHelper():
|
||
def __init__(self, global_object, config: dict):
|
||
self.loop = asyncio.new_event_loop()
|
||
asyncio.set_event_loop(self.loop)
|
||
self.logger = global_object.logger
|
||
dashboard_data = global_object.dashboard_data
|
||
dashboard_data.configs = {
|
||
"data": []
|
||
}
|
||
self.parse_default_config(dashboard_data, config)
|
||
self.dashboard_data: DashBoardData = dashboard_data
|
||
self.dashboard = AstrBotDashBoard(global_object)
|
||
self.key_map = {} # key: uuid, value: config key name
|
||
self.cc = CmdConfig()
|
||
|
||
@self.dashboard.register("post_configs")
|
||
def on_post_configs(post_configs: dict):
|
||
try:
|
||
# self.logger.log(f"收到配置更新请求", gu.LEVEL_INFO, tag="可视化面板")
|
||
if 'base_config' in post_configs:
|
||
self.save_config(post_configs['base_config'], namespace='') # 基础配置
|
||
self.save_config(post_configs['config'], namespace=post_configs['namespace']) # 选定配置
|
||
self.parse_default_config(self.dashboard_data, self.cc.get_all())
|
||
# 重启
|
||
threading.Thread(target=self.dashboard.shutdown_bot, args=(2,), daemon=True).start()
|
||
except Exception as e:
|
||
# self.logger.log(f"在保存配置时发生错误：{e}", gu.LEVEL_ERROR, tag="可视化面板")
|
||
raise e
|
||
|
||
# 将 config.yaml、 中的配置解析到 dashboard_data.configs 中
|
||
def parse_default_config(self, dashboard_data: DashBoardData, config: dict):
|
||
|
||
try:
|
||
qq_official_platform_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="QQ_OFFICIAL 平台配置",
|
||
description="",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="启用 QQ_OFFICIAL 平台",
|
||
description="官方的接口，仅支持 QQ 频道。详见 q.qq.com",
|
||
value=config['qqbot']['enable'],
|
||
path="qqbot.enable",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="QQ机器人APPID",
|
||
description="详见 q.qq.com",
|
||
value=config['qqbot']['appid'],
|
||
path="qqbot.appid",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="QQ机器人令牌",
|
||
description="详见 q.qq.com",
|
||
value=config['qqbot']['token'],
|
||
path="qqbot.token",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="QQ机器人 Secret",
|
||
description="详见 q.qq.com",
|
||
value=config['qqbot_secret'],
|
||
path="qqbot_secret",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="是否允许 QQ 频道私聊",
|
||
description="如果启用，机器人会响应私聊消息。",
|
||
value=config['direct_message_mode'],
|
||
path="direct_message_mode",
|
||
),
|
||
]
|
||
)
|
||
qq_gocq_platform_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="OneBot协议平台配置",
|
||
description="",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="启用",
|
||
description="支持cq-http、shamrock等（目前仅支持QQ平台）",
|
||
value=config['gocqbot']['enable'],
|
||
path="gocqbot.enable",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="HTTP 服务器地址",
|
||
description="",
|
||
value=config['gocq_host'],
|
||
path="gocq_host",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="HTTP 服务器端口",
|
||
description="",
|
||
value=config['gocq_http_port'],
|
||
path="gocq_http_port",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="WebSocket 服务器端口",
|
||
description="目前仅支持正向 WebSocket",
|
||
value=config['gocq_websocket_port'],
|
||
path="gocq_websocket_port",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="是否响应群消息",
|
||
description="",
|
||
value=config['gocq_react_group'],
|
||
path="gocq_react_group",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="是否响应私聊消息",
|
||
description="",
|
||
value=config['gocq_react_friend'],
|
||
path="gocq_react_friend",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="是否响应群成员增加消息",
|
||
description="",
|
||
value=config['gocq_react_group_increase'],
|
||
path="gocq_react_group_increase",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="是否响应频道消息",
|
||
description="",
|
||
value=config['gocq_react_guild'],
|
||
path="gocq_react_guild",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="转发阈值（字符数）",
|
||
description="机器人回复的消息长度超出这个值后，会被折叠成转发卡片发出以减少刷屏。",
|
||
value=config['qq_forward_threshold'],
|
||
path="qq_forward_threshold",
|
||
),
|
||
]
|
||
)
|
||
|
||
general_platform_detail_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="通用平台配置",
|
||
description="",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="启动消息文字转图片",
|
||
description="启动后，机器人会将消息转换为图片发送，以降低风控风险。",
|
||
value=config['qq_pic_mode'],
|
||
path="qq_pic_mode",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="消息限制时间",
|
||
description="在此时间内，机器人不会回复同一个用户的消息。单位：秒",
|
||
value=config['limit']['time'],
|
||
path="limit.time",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="消息限制次数",
|
||
description="在上面的时间内，如果用户发送消息超过此次数，则机器人不会回复。单位：次",
|
||
value=config['limit']['count'],
|
||
path="limit.count",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="回复前缀",
|
||
description="[xxxx] 你好！ 其中xxxx是你可以填写的前缀。如果为空则不显示。",
|
||
value=config['reply_prefix'],
|
||
path="reply_prefix",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="list",
|
||
name="通用管理员用户 ID（支持多个管理员）。通过 !myid 指令获取。",
|
||
description="",
|
||
value=config['other_admins'],
|
||
path="other_admins",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="独立会话",
|
||
description="是否启用独立会话模式，即 1 个用户自然账号 1 个会话。",
|
||
value=config['uniqueSessionMode'],
|
||
path="uniqueSessionMode",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="LLM 唤醒词",
|
||
description="如果不为空, 那么只有当消息以此词开头时，才会调用大语言模型进行回复。如设置为 /chat，那么只有当消息以 /chat 开头时，才会调用大语言模型进行回复。",
|
||
value=config['llm_wake_prefix'],
|
||
path="llm_wake_prefix",
|
||
)
|
||
]
|
||
)
|
||
|
||
openai_official_llm_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="OpenAI 官方接口类设置",
|
||
description="",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="list",
|
||
name="OpenAI API Key",
|
||
description="OpenAI API 的 Key。支持使用非官方但兼容的 API（第三方中转key）。",
|
||
value=config['openai']['key'],
|
||
path="openai.key",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="OpenAI API 节点地址(api base)",
|
||
description="OpenAI API 的节点地址，配合非官方 API 使用。如果不想填写，那么请填写 none",
|
||
value=config['openai']['api_base'],
|
||
path="openai.api_base",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="OpenAI model",
|
||
description="OpenAI LLM 模型。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['chatGPTConfigs']['model'],
|
||
path="openai.chatGPTConfigs.model",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="OpenAI max_tokens",
|
||
description="OpenAI 最大生成长度。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['chatGPTConfigs']['max_tokens'],
|
||
path="openai.chatGPTConfigs.max_tokens",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="float",
|
||
name="OpenAI temperature",
|
||
description="OpenAI 温度。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['chatGPTConfigs']['temperature'],
|
||
path="openai.chatGPTConfigs.temperature",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="float",
|
||
name="OpenAI top_p",
|
||
description="OpenAI top_p。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['chatGPTConfigs']['top_p'],
|
||
path="openai.chatGPTConfigs.top_p",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="float",
|
||
name="OpenAI frequency_penalty",
|
||
description="OpenAI frequency_penalty。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['chatGPTConfigs']['frequency_penalty'],
|
||
path="openai.chatGPTConfigs.frequency_penalty",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="float",
|
||
name="OpenAI presence_penalty",
|
||
description="OpenAI presence_penalty。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['chatGPTConfigs']['presence_penalty'],
|
||
path="openai.chatGPTConfigs.presence_penalty",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="int",
|
||
name="OpenAI 总生成长度限制",
|
||
description="OpenAI 总生成长度限制。详见 https://platform.openai.com/docs/api-reference/chat",
|
||
value=config['openai']['total_tokens_limit'],
|
||
path="openai.total_tokens_limit",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="OpenAI 图像生成模型",
|
||
description="OpenAI 图像生成模型。",
|
||
value=config['openai_image_generate']['model'],
|
||
path="openai_image_generate.model",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="OpenAI 图像生成大小",
|
||
description="OpenAI 图像生成大小。",
|
||
value=config['openai_image_generate']['size'],
|
||
path="openai_image_generate.size",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="OpenAI 图像生成风格",
|
||
description="OpenAI 图像生成风格。修改前请参考 OpenAI 官方文档",
|
||
value=config['openai_image_generate']['style'],
|
||
path="openai_image_generate.style",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="OpenAI 图像生成质量",
|
||
description="OpenAI 图像生成质量。修改前请参考 OpenAI 官方文档",
|
||
value=config['openai_image_generate']['quality'],
|
||
path="openai_image_generate.quality",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="问题题首提示词",
|
||
description="如果填写了此项，在每个对大语言模型的请求中，都会在问题前加上此提示词。",
|
||
value=config['llm_env_prompt'],
|
||
path="llm_env_prompt",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="默认人格文本",
|
||
description="默认人格文本",
|
||
value=config['default_personality_str'],
|
||
path="default_personality_str",
|
||
),
|
||
]
|
||
)
|
||
|
||
rev_chatgpt_accounts = config['rev_ChatGPT']['account']
|
||
new_accs = []
|
||
for i in rev_chatgpt_accounts:
|
||
if isinstance(i, dict) and 'access_token' in i:
|
||
new_accs.append(i['access_token'])
|
||
elif isinstance(i, str):
|
||
new_accs.append(i)
|
||
config['rev_ChatGPT']['account'] = new_accs
|
||
|
||
rev_chatgpt_llm_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="逆向语言模型服务设置",
|
||
description="",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="启用逆向语言模型服务",
|
||
description="",
|
||
value=config['rev_ChatGPT']['enable'],
|
||
path="rev_ChatGPT.enable",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="终结点（Endpoint）地址",
|
||
description="逆向服务的终结点服务器的地址。",
|
||
value=config['CHATGPT_BASE_URL'],
|
||
path="CHATGPT_BASE_URL",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="list",
|
||
name="assess_token",
|
||
description="assess_token",
|
||
value=config['rev_ChatGPT']['account'],
|
||
path="rev_ChatGPT.account",
|
||
),
|
||
]
|
||
)
|
||
|
||
baidu_aip_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="百度内容审核",
|
||
description="需要去申请",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="bool",
|
||
name="启动百度内容审核服务",
|
||
description="",
|
||
value=config['baidu_aip']['enable'],
|
||
path="baidu_aip.enable"
|
||
),
|
||
# "app_id": null,
|
||
# "api_key": null,
|
||
# "secret_key": null
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="APP ID",
|
||
description="",
|
||
value=config['baidu_aip']['app_id'],
|
||
path="baidu_aip.app_id"
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="API KEY",
|
||
description="",
|
||
value=config['baidu_aip']['api_key'],
|
||
path="baidu_aip.api_key"
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="SECRET KEY",
|
||
description="",
|
||
value=config['baidu_aip']['secret_key'],
|
||
path="baidu_aip.secret_key"
|
||
)
|
||
]
|
||
)
|
||
|
||
other_group = DashBoardConfig(
|
||
config_type="group",
|
||
name="其他配置",
|
||
description="其他配置描述",
|
||
body=[
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="HTTP 代理地址",
|
||
description="建议上下一致",
|
||
value=config['http_proxy'],
|
||
path="http_proxy",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="HTTPS 代理地址",
|
||
description="建议上下一致",
|
||
value=config['https_proxy'],
|
||
path="https_proxy",
|
||
),
|
||
DashBoardConfig(
|
||
config_type="item",
|
||
val_type="str",
|
||
name="面板用户名",
|
||
description="是的，就是你理解的这个面板的用户名",
|
||
value=config['dashboard_username'],
|
||
path="dashboard_username",
|
||
),
|
||
]
|
||
)
|
||
|
||
dashboard_data.configs['data'] = [
|
||
qq_official_platform_group,
|
||
qq_gocq_platform_group,
|
||
general_platform_detail_group,
|
||
openai_official_llm_group,
|
||
rev_chatgpt_llm_group,
|
||
other_group,
|
||
baidu_aip_group
|
||
]
|
||
|
||
except Exception as e:
|
||
self.logger.log(f"配置文件解析错误：{e}", gu.LEVEL_ERROR)
|
||
raise e
|
||
|
||
|
||
def save_config(self, post_config: list, namespace: str):
|
||
'''
|
||
根据 path 解析并保存配置
|
||
'''
|
||
|
||
queue = post_config
|
||
while len(queue) > 0:
|
||
config = queue.pop(0)
|
||
if config['config_type'] == "group":
|
||
for item in config['body']:
|
||
queue.append(item)
|
||
elif config['config_type'] == "item":
|
||
if config['path'] is None or config['path'] == "":
|
||
continue
|
||
|
||
path = config['path'].split('.')
|
||
if len(path) == 0:
|
||
continue
|
||
|
||
if config['val_type'] == "bool":
|
||
self._write_config(namespace, config['path'], config['value'])
|
||
elif config['val_type'] == "str":
|
||
self._write_config(namespace, config['path'], config['value'])
|
||
elif config['val_type'] == "int":
|
||
try:
|
||
self._write_config(namespace, config['path'], int(config['value']))
|
||
except:
|
||
raise ValueError(f"配置项 {config['name']} 的值必须是整数")
|
||
elif config['val_type'] == "float":
|
||
try:
|
||
self._write_config(namespace, config['path'], float(config['value']))
|
||
except:
|
||
raise ValueError(f"配置项 {config['name']} 的值必须是浮点数")
|
||
elif config['val_type'] == "list":
|
||
if config['value'] is None:
|
||
self._write_config(namespace, config['path'], [])
|
||
elif not isinstance(config['value'], list):
|
||
raise ValueError(f"配置项 {config['name']} 的值必须是列表")
|
||
self._write_config(namespace, config['path'], config['value'])
|
||
else:
|
||
raise NotImplementedError(f"未知或者未实现的配置项类型：{config['val_type']}")
|
||
|
||
def _write_config(self, namespace: str, key: str, value):
|
||
if namespace == "" or namespace.startswith("internal_"):
|
||
# 机器人自带配置，存到 config.yaml
|
||
self.cc.put_by_dot_str(key, value)
|
||
else:
|
||
update_config(namespace, key, value)
|
||
|
||
def run(self):
|
||
self.dashboard.run() |