Compare commits

...
3 Commits
Author SHA1 Message Date
Wbucket 33dc032d82 fix: 不正确的事件 2026-11-06 21:06:51 +08:00
Wbucket dc8753b87c style: context 2026-11-06 15:30:23 +08:00
Wbucket 6f79cae3a3 dev:style 项目结构和分类 2026-11-05 22:54:18 +08:00
3 changed files with 61 additions and 22 deletions
+16 -19
View File
@@ -1,19 +1,23 @@
from astrbot.api.event import filter, AstrMessageEvent, MessageEventResult from astrbot.api.event import filter, AstrMessageEvent, MessageEventResult
from astrbot.core.message.message_event_result import MessageChain
from astrbot.api.star import Context, Star, register from astrbot.api.star import Context, Star, register
from astrbot.api import logger from astrbot.api import logger
from astrbot.api.provider import (ProviderRequest,LLMResponse ) from astrbot.api.provider import (ProviderRequest,LLMResponse )
from astrbot.core.agent.message import TextPart from astrbot.core.agent.message import TextPart
from .tool.pattern import Pattern from .service.timer import TimerService
@register("wait_msg", "Wbucket", "一个简单的 llm等待回复插件", "0.0.1") @register("wait_msg", "Wbucket", "一个简单的 llm等待回复插件", "0.0.1")
class MyPlugin(Star): class MyPlugin(Star):
def __init__(self, context: Context): def __init__(self, context: Context):
super().__init__(context) super().__init__(context)
self.Timer = TimerService(self)
async def initialize(self): async def initialize(self):
"""可选择实现异步的插件初始化方法,当实例化该插件类之后会自动调用该方法。""" """可选择实现异步的插件初始化方法,当实例化该插件类之后会自动调用该方法。"""
# 注册指令的装饰器。指令名为 helloworld。注册成功后,发送 `/helloworld` 就会触发这个指令,并回复 `你好, {user_name}!` # 注册指令的装饰器
@filter.on_llm_request() @filter.on_llm_request()
async def content_inject(self, event: AstrMessageEvent,req : ProviderRequest): async def content_inject(self, event: AstrMessageEvent,req : ProviderRequest):
"""这是一个 hello world 指令""" # 这是 handler 的描述,将会被解析方便用户了解插件内容。建议填写。 """这是一个 hello world 指令""" # 这是 handler 的描述,将会被解析方便用户了解插件内容。建议填写。
@@ -22,33 +26,26 @@ class MyPlugin(Star):
text=( text=(
"-----wt使用----" "-----wt使用----"
"在回复内容中加入<wt>n</wt>模板输入想要等待的时间,n=等待秒数\n" "在回复内容中加入<wt>n</wt>模板输入想要等待的时间,n=等待秒数\n"
"最短2s最长30min(1800s),不添加则正常回复(只有对方回复你才会被唤醒)\n" "最短10s最长30min(1800s),不添加则正常回复(只有对方回复你才会被唤醒)\n"
"例如'<wt>5</wt>'-->等待5秒后再次唤醒.\n" "例如'<wt>15</wt>'-->等待15秒后再次唤醒.\n"
"使用场景:希望对方在多久之后能回你,或者希望这个话题能在短期内继续而不是因为对方的不回复就夭折\n" "使用场景:希望对方在多久之后能回你,或者希望这个话题能在短期内继续而不是因为对方的不回复就夭折\n"
"例如如下场景:\n" "例如如下场景:\n"
"--\n" "--\n"
"assistant:'你觉得咸粽子好吃韩式甜粽子好吃?<wt>40</wt>'\n" "assistant:'你觉得咸粽子好吃还是甜粽子好吃?<wt>40</wt>'\n"
"(系统时间:四十秒后,系统定时唤醒)\n" "(系统时间:四十秒后,系统定时唤醒)\n"
"user:'<wt>无回应</wt>'\n" "user:'<wt>40s过去了,对方无回应</wt>'\n"
"assistant:'你怎么不理我<wt>40</wt>'\n" "assistant:'你怎么不理我<wt>40</wt>'\n"
"(系统时间:二十秒后,用户回复,定时任务取消)\n" "(系统时间:二十秒后,用户回复,定时任务取消)\n"
"user:'抱歉啊刚看到,我觉得咸的好吃awa'\n" "user:'抱歉啊刚看到,我觉得咸的好吃awa'\n"
"每主动回复消息必须带上wt标签,标签必须闭合且只包裹你要等待的秒数!\n" "每主动发出消息建议带上wt标签,标签必须闭合且只包裹你要等待的秒数。\n"
"如果话题完结了,不想再无意义被唤醒,或者回复完毕别人的问题还没被追问的时候则不必再加。自己看着决定即可。\n"
"-------" "-------"
) )
)) ))
# @filter.event_message_type(filter.EventMessageType.PRIVATE_MESSAGE)
# async def awa(self,event:AstrMessageEvent):
# yield event.plain_result(event.get_message_str())
@filter.on_llm_response() @filter.on_decorating_result()
async def pattern_extract(self,event:AstrMessageEvent,req:LLMResponse): async def pattern_extract(self,event:AstrMessageEvent):
if not req.role == 'assistant': await self.Timer.timer(event)
return
logger.info("[wait_msg]:开始解析模板")
llm_res = req.completion_text
dump = Pattern(llm_res)
await event.send(MessageChain().message("awa定时时间消息:{0}过滤后:{1}".format(dump.count,dump.string)))
async def terminate(self): async def terminate(self):
"""可选择实现异步的插件销毁方法,当插件被卸载/停用时会调用。""" """可选择实现异步的插件销毁方法,当插件被卸载/停用时会调用。"""
+41
View File
@@ -0,0 +1,41 @@
from astrbot.api.event import AstrMessageEvent
from astrbot.api.provider import LLMResponse
from astrbot.api import logger
from astrbot.core.message.message_event_result import MessageChain,Plain
from ..tool.pattern import Pattern
from astrbot.api.star import Star
import asyncio
class TimerService:
def __init__(self,star:Star) -> None:
self.star = star
pass
# 解析模板并进行定时
async def parse(self,event:AstrMessageEvent):
res = event.get_result()
if not res:
return
Second:int|None = None
logger.info("[wait_msg]:开始解析模板")
chain = res.chain
for i in chain :
if not isinstance(i,Plain):
continue
dump = Pattern(i.text)
i.text = dump.string
Second = dump.count
if not Second is None:
self.timer(event,Second)
yield
# 定时:
def timer(self,event:AstrMessageEvent,Sec:int):
asyncio.create_task(self.timer_task())
async def timer_task(self):
await asyncio.sleep(1)
logger.warning("awa现在过去一秒了哦")
+3 -2
View File
@@ -1,5 +1,7 @@
import re import re
from astrbot.api import logger from astrbot.api import logger
# 输入内容,获取过滤后的字符串与定时秒数
class Pattern: class Pattern:
_pattern = r'<wt>((?!wt).)*</wt>' _pattern = r'<wt>((?!wt).)*</wt>'
@@ -22,7 +24,7 @@ class Pattern:
def load(self,pattern_str:str)->int|None: def load(self,pattern_str:str)->int|None:
content = pattern_str.strip().removeprefix("<wt>").removesuffix("</wt>").strip() content = pattern_str.strip().removeprefix("<wt>").removesuffix("</wt>").strip()
print(content)
try: try:
num = int(content) num = int(content)
return self.sec_range(num) return self.sec_range(num)
@@ -50,4 +52,3 @@ class Pattern:
"content":o_str, "content":o_str,
"second":timerSec "second":timerSec
} }
print(Pattern("awdwadawd<wt>ada</wt>"))