50 lines
2.6 KiB
Python
50 lines
2.6 KiB
Python
from astrbot.api.event import filter, AstrMessageEvent, MessageEventResult
|
|
|
|
from astrbot.api.star import Context, Star, register
|
|
from astrbot.api import logger
|
|
from astrbot.api.provider import (ProviderRequest,LLMResponse )
|
|
from astrbot.core.agent.message import TextPart
|
|
from .service.timer import timer
|
|
|
|
|
|
@register("wait_msg", "Wbucket", "一个简单的 llm等待回复插件", "0.0.1")
|
|
class MyPlugin(Star):
|
|
def __init__(self, context: Context):
|
|
super().__init__(context)
|
|
|
|
async def initialize(self):
|
|
"""可选择实现异步的插件初始化方法，当实例化该插件类之后会自动调用该方法。"""
|
|
|
|
# 注册指令的装饰器
|
|
@filter.on_llm_request()
|
|
async def content_inject(self, event: AstrMessageEvent,req : ProviderRequest):
|
|
"""这是一个 hello world 指令""" # 这是 handler 的描述，将会被解析方便用户了解插件内容。建议填写。
|
|
# logger.info("系统提示词："+req.system_prompt)
|
|
req.extra_user_content_parts.append(TextPart(
|
|
text=(
|
|
"-----wt使用----"
|
|
"在回复内容中加入<wt>n</wt>模板输入想要等待的时间,n=等待秒数\n"
|
|
"最短10s最长30min(1800s),不添加则正常回复(只有对方回复你才会被唤醒)\n"
|
|
"例如'<wt>15</wt>'-->等待15秒后再次唤醒.\n"
|
|
"使用场景:希望对方在多久之后能回你，或者希望这个话题能在短期内继续而不是因为对方的不回复就夭折\n"
|
|
"例如如下场景:\n"
|
|
"--\n"
|
|
"assistant:'你觉得咸粽子好吃韩式甜粽子好吃?<wt>40</wt>'\n"
|
|
"(系统时间：四十秒后,系统定时唤醒)\n"
|
|
"user:'<wt>无回应</wt>'\n"
|
|
"assistant:'你怎么不理我<wt>40</wt>'\n"
|
|
"(系统时间：二十秒后,用户回复，定时任务取消)\n"
|
|
"user:'抱歉啊刚看到，我觉得咸的好吃awa'\n"
|
|
"每主动发出消息建议带上wt标签，标签必须闭合且只包裹你要等待的秒数。\n"
|
|
"如果话题完结了，不想再无意义被唤醒，或者回复完毕别人的问题还没被追问的时候则不必再加。自己看着决定即可。\n"
|
|
"-------"
|
|
)
|
|
))
|
|
|
|
@filter.on_llm_response()
|
|
async def pattern_extract(self,event:AstrMessageEvent,req:LLMResponse):
|
|
await timer(event,req)
|
|
|
|
async def terminate(self):
|
|
"""可选择实现异步的插件销毁方法，当插件被卸载/停用时会调用。"""
|