Soulter
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
from typing import List, Union, Optional
|
|
from dataclasses import dataclass
|
|
from type.register import RegisteredPlatform
|
|
from type.types import Context
|
|
from type.astrbot_message import AstrBotMessage, MessageType
|
|
|
|
@dataclass
|
|
class MessageResult():
|
|
result_message: Union[str, list]
|
|
is_command_call: Optional[bool] = False
|
|
use_t2i: Optional[bool] = None # None 为跟随用户设置
|
|
callback: Optional[callable] = None
|
|
|
|
class AstrMessageEvent():
|
|
|
|
def __init__(self,
|
|
message_str: str,
|
|
message_obj: AstrBotMessage,
|
|
platform: RegisteredPlatform,
|
|
role: str,
|
|
context: Context,
|
|
session_id: str = None,
|
|
unified_msg_origin: str = None,
|
|
only_command: bool = False):
|
|
'''
|
|
AstrBot 消息事件。
|
|
|
|
`message_str`: 纯消息字符串
|
|
`message_obj`: AstrBotMessage 对象
|
|
`platform`: 平台对象
|
|
`role`: 角色，`admin` or `member`
|
|
`context`: 全局对象
|
|
`session_id`: 会话id
|
|
`unified_msg_origin`: 统一消息来源
|
|
`only_command`: 是否只处理指令，而不使用 LLM 回复
|
|
'''
|
|
self.context = context
|
|
self.message_str = message_str
|
|
self.message_obj = message_obj
|
|
self.platform = platform
|
|
self.role = role
|
|
self.session_id = session_id
|
|
self.unified_msg_origin = unified_msg_origin
|
|
self.only_command = only_command
|
|
|
|
def from_astrbot_message(message: AstrBotMessage,
|
|
context: Context,
|
|
platform_name: str,
|
|
session_id: str,
|
|
|
|
unified_msg_origin: str = None,
|
|
only_command: bool = False):
|
|
|
|
# 解析 role
|
|
sender_id = str(message.sender.user_id)
|
|
if sender_id in context.config_helper.admins_id:
|
|
role = 'admin'
|
|
else:
|
|
role = 'member'
|
|
|
|
ame = AstrMessageEvent(message.message_str,
|
|
message,
|
|
context.find_platform(platform_name),
|
|
role,
|
|
context,
|
|
session_id,
|
|
unified_msg_origin,
|
|
only_command=only_command)
|
|
return ame
|
|
|