114 lines
4.9 KiB
Python
114 lines
4.9 KiB
Python
import astrbot.api.message_components as Comp
|
|
import copy
|
|
from astrbot.api import logger
|
|
from astrbot.api.event import AstrMessageEvent, filter
|
|
from astrbot.api.star import Context, Star, register
|
|
from astrbot.core.utils.session_waiter import (
|
|
SessionWaiter,
|
|
USER_SESSIONS,
|
|
FILTERS,
|
|
session_waiter,
|
|
SessionController,
|
|
)
|
|
from sys import maxsize
|
|
|
|
|
|
@register(
|
|
"session_controller",
|
|
"Cvandia & Soulter",
|
|
"为插件支持会话控制",
|
|
"v1.0.1",
|
|
"https://astrbot.app",
|
|
)
|
|
class Waiter(Star):
|
|
"""会话控制"""
|
|
|
|
def __init__(self, context: Context):
|
|
super().__init__(context)
|
|
|
|
self.empty_mention_waiting = self.context.get_config()["platform_settings"][
|
|
"empty_mention_waiting"
|
|
]
|
|
self.wake_prefix = self.context.get_config()["wake_prefix"]
|
|
|
|
@filter.event_message_type(filter.EventMessageType.ALL, priority=maxsize)
|
|
async def handle_session_control_agent(self, event: AstrMessageEvent):
|
|
"""会话控制代理"""
|
|
for session_filter in FILTERS:
|
|
session_id = session_filter.filter(event)
|
|
if session_id in USER_SESSIONS:
|
|
await SessionWaiter.trigger(session_id, event)
|
|
event.stop_event()
|
|
|
|
@filter.event_message_type(filter.EventMessageType.ALL, priority=maxsize - 1)
|
|
async def handle_empty_mention(self, event: AstrMessageEvent):
|
|
"""实现了对只有一个 @ 的消息内容的处理"""
|
|
try:
|
|
messages = event.get_messages()
|
|
if len(messages) == 1:
|
|
if (
|
|
isinstance(messages[0], Comp.At)
|
|
and str(messages[0].qq) == str(event.get_self_id())
|
|
and self.empty_mention_waiting
|
|
) or (
|
|
isinstance(messages[0], Comp.Plain)
|
|
and messages[0].text.strip() in self.wake_prefix
|
|
):
|
|
try:
|
|
# 尝试使用 LLM 生成更生动的回复
|
|
func_tools_mgr = self.context.get_llm_tool_manager()
|
|
|
|
# 获取用户当前的对话信息
|
|
curr_cid = await self.context.conversation_manager.get_curr_conversation_id(
|
|
event.unified_msg_origin
|
|
)
|
|
conversation = None
|
|
|
|
if curr_cid:
|
|
conversation = await self.context.conversation_manager.get_conversation(
|
|
event.unified_msg_origin, curr_cid
|
|
)
|
|
else:
|
|
# 创建新对话
|
|
curr_cid = await self.context.conversation_manager.new_conversation(
|
|
event.unified_msg_origin
|
|
)
|
|
|
|
# 使用 LLM 生成回复
|
|
yield event.request_llm(
|
|
prompt="注意，你正在社交媒体上中与用户进行聊天，用户只是通过@来唤醒你，但并未在这条消息中输入内容，他可能会在接下来一条发送他想发送的内容。请你友好地询问用户想要聊些什么或者需要什么帮助，回复要符合人设，不要太过机械化。注意，你仅需要输出要回复用户的内容，不要输出其他任何东西",
|
|
func_tool_manager=func_tools_mgr,
|
|
session_id=curr_cid,
|
|
contexts=[],
|
|
system_prompt="",
|
|
conversation=conversation,
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"LLM response failed: {str(e)}")
|
|
# LLM 回复失败，使用原始预设回复
|
|
yield event.plain_result("想要问什么呢？😄")
|
|
|
|
@session_waiter(60)
|
|
async def empty_mention_waiter(
|
|
controller: SessionController, event: AstrMessageEvent
|
|
):
|
|
event.message_obj.message.insert(
|
|
0, Comp.At(qq=event.get_self_id(), name=event.get_self_id())
|
|
)
|
|
new_event = copy.copy(event)
|
|
# 重新推入事件队列
|
|
self.context.get_event_queue().put_nowait(new_event)
|
|
event.stop_event()
|
|
controller.stop()
|
|
|
|
try:
|
|
await empty_mention_waiter(event)
|
|
except TimeoutError as _:
|
|
pass
|
|
except Exception as e:
|
|
yield event.plain_result("发生错误，请联系管理员: " + str(e))
|
|
finally:
|
|
event.stop_event()
|
|
except Exception as e:
|
|
logger.error("handle_empty_mention error: " + str(e))
|