Dt8333
7dd95d8a59
* chore: auto fix by ruff * refactor: 统一修正返回类型注解为 None/bool 以匹配实现 * refactor: 将 _get_next_page 改为异步并移除多余的请求错误抛出 * refactor: 将 get_client 的返回类型改为 object * style: 为 LarkMessageEvent 的相关方法添加返回类型注解 None --------- Co-authored-by: Soulter <37870767+Soulter@users.noreply.github.com>
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
import time
|
|
import uuid
|
|
from typing import Any
|
|
|
|
from astrbot.core.message.components import Plain
|
|
from astrbot.core.message.message_event_result import MessageChain
|
|
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
|
from astrbot.core.platform.astrbot_message import AstrBotMessage, MessageMember
|
|
from astrbot.core.platform.message_session import MessageSession
|
|
from astrbot.core.platform.message_type import MessageType
|
|
from astrbot.core.platform.platform_metadata import PlatformMetadata
|
|
|
|
|
|
class CronMessageEvent(AstrMessageEvent):
|
|
"""Synthetic event used when a cron job triggers the main agent loop."""
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
context,
|
|
session: MessageSession,
|
|
message: str,
|
|
sender_id: str = "astrbot",
|
|
sender_name: str = "Scheduler",
|
|
extras: dict[str, Any] | None = None,
|
|
message_type: MessageType = MessageType.FRIEND_MESSAGE,
|
|
) -> None:
|
|
platform_meta = PlatformMetadata(
|
|
name="cron",
|
|
description="CronJob",
|
|
id=session.platform_id,
|
|
)
|
|
|
|
msg_obj = AstrBotMessage()
|
|
msg_obj.type = message_type
|
|
msg_obj.self_id = sender_id
|
|
msg_obj.session_id = session.session_id
|
|
msg_obj.message_id = uuid.uuid4().hex
|
|
msg_obj.sender = MessageMember(user_id=session.session_id, nickname=sender_name)
|
|
msg_obj.message = [Plain(message)]
|
|
msg_obj.message_str = message
|
|
msg_obj.raw_message = message
|
|
msg_obj.timestamp = int(time.time())
|
|
|
|
super().__init__(message, msg_obj, platform_meta, session.session_id)
|
|
|
|
# Ensure we use the original session for sending messages
|
|
self.session = session
|
|
self.context_obj = context
|
|
self.is_at_or_wake_command = True
|
|
self.is_wake = True
|
|
|
|
if extras:
|
|
self._extras.update(extras)
|
|
|
|
async def send(self, message: MessageChain) -> None:
|
|
if message is None:
|
|
return
|
|
await self.context_obj.send_message(self.session, message)
|
|
await super().send(message)
|
|
|
|
async def send_streaming(self, generator, use_fallback: bool = False) -> None:
|
|
async for chain in generator:
|
|
await self.send(chain)
|
|
|
|
|
|
__all__ = ["CronMessageEvent"]
|