ruff:
This commit is contained in:
@@ -7,12 +7,13 @@ from src.plugins.chat.message import MessageSending
|
||||
|
||||
logger = get_module_logger("message_sender")
|
||||
|
||||
|
||||
class DirectMessageSender:
|
||||
"""直接消息发送器"""
|
||||
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
|
||||
async def send_message(
|
||||
self,
|
||||
chat_stream: ChatStream,
|
||||
@@ -20,7 +21,7 @@ class DirectMessageSender:
|
||||
reply_to_message: Optional[Message] = None,
|
||||
) -> None:
|
||||
"""发送消息到聊天流
|
||||
|
||||
|
||||
Args:
|
||||
chat_stream: 聊天流
|
||||
content: 消息内容
|
||||
@@ -29,21 +30,18 @@ class DirectMessageSender:
|
||||
try:
|
||||
# 创建消息内容
|
||||
segments = [Seg(type="text", data={"text": content})]
|
||||
|
||||
|
||||
# 检查是否需要引用回复
|
||||
if reply_to_message:
|
||||
reply_id = reply_to_message.message_id
|
||||
message_sending = MessageSending(
|
||||
segments=segments,
|
||||
reply_to_id=reply_id
|
||||
)
|
||||
message_sending = MessageSending(segments=segments, reply_to_id=reply_id)
|
||||
else:
|
||||
message_sending = MessageSending(segments=segments)
|
||||
|
||||
|
||||
# 发送消息
|
||||
await chat_stream.send_message(message_sending)
|
||||
logger.info(f"消息已发送: {content}")
|
||||
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"发送消息失败: {str(e)}")
|
||||
raise
|
||||
raise
|
||||
|
||||
Reference in New Issue
Block a user