From 6f79cae3a31845a184f08042933b7247174621f0 Mon Sep 17 00:00:00 2001 From: Wbucket <2595141012@stu.hqu.edu.cn> Date: Thu, 5 Nov 2026 22:54:18 +0800 Subject: [PATCH] =?UTF-8?q?dev:style=20=E9=A1=B9=E7=9B=AE=E7=BB=93?= =?UTF-8?q?=E6=9E=84=E5=92=8C=E5=88=86=E7=B1=BB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 27 +++++++++++---------------- service/timer.py | 15 +++++++++++++++ tool/pattern.py | 5 +++-- 3 files changed, 29 insertions(+), 18 deletions(-) create mode 100644 service/timer.py diff --git a/main.py b/main.py index 84309ad..777e0ab 100644 --- a/main.py +++ b/main.py @@ -1,10 +1,12 @@ 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 import logger from astrbot.api.provider import (ProviderRequest,LLMResponse ) from astrbot.core.agent.message import TextPart -from .tool.pattern import Pattern +from .service.timer import timer + + @register("wait_msg", "Wbucket", "一个简单的 llm等待回复插件", "0.0.1") class MyPlugin(Star): def __init__(self, context: Context): @@ -13,7 +15,7 @@ class MyPlugin(Star): async def initialize(self): """可选择实现异步的插件初始化方法,当实例化该插件类之后会自动调用该方法。""" - # 注册指令的装饰器。指令名为 helloworld。注册成功后,发送 `/helloworld` 就会触发这个指令,并回复 `你好, {user_name}!` + # 注册指令的装饰器 @filter.on_llm_request() async def content_inject(self, event: AstrMessageEvent,req : ProviderRequest): """这是一个 hello world 指令""" # 这是 handler 的描述,将会被解析方便用户了解插件内容。建议填写。 @@ -22,8 +24,8 @@ class MyPlugin(Star): text=( "-----wt使用----" "在回复内容中加入n模板输入想要等待的时间,n=等待秒数\n" - "最短2s最长30min(1800s),不添加则正常回复(只有对方回复你才会被唤醒)\n" - "例如'5'-->等待5秒后再次唤醒.\n" + "最短10s最长30min(1800s),不添加则正常回复(只有对方回复你才会被唤醒)\n" + "例如'15'-->等待15秒后再次唤醒.\n" "使用场景:希望对方在多久之后能回你,或者希望这个话题能在短期内继续而不是因为对方的不回复就夭折\n" "例如如下场景:\n" "--\n" @@ -33,22 +35,15 @@ class MyPlugin(Star): "assistant:'你怎么不理我40'\n" "(系统时间:二十秒后,用户回复,定时任务取消)\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() async def pattern_extract(self,event:AstrMessageEvent,req:LLMResponse): - if not req.role == 'assistant': - 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))) + await timer(event,req) + async def terminate(self): """可选择实现异步的插件销毁方法,当插件被卸载/停用时会调用。""" diff --git a/service/timer.py b/service/timer.py new file mode 100644 index 0000000..ae2d768 --- /dev/null +++ b/service/timer.py @@ -0,0 +1,15 @@ +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 +from ..tool.pattern import Pattern + +# 解析模板并进行定时 +async def timer(event:AstrMessageEvent,req:LLMResponse): + if not req.role == 'assistant': + 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))) \ No newline at end of file diff --git a/tool/pattern.py b/tool/pattern.py index 04764cd..2af896c 100644 --- a/tool/pattern.py +++ b/tool/pattern.py @@ -1,5 +1,7 @@ import re from astrbot.api import logger + +# 输入内容,获取过滤后的字符串与定时秒数 class Pattern: _pattern = r'((?!wt).)*' @@ -22,7 +24,7 @@ class Pattern: def load(self,pattern_str:str)->int|None: content = pattern_str.strip().removeprefix("").removesuffix("").strip() - print(content) + try: num = int(content) return self.sec_range(num) @@ -50,4 +52,3 @@ class Pattern: "content":o_str, "second":timerSec } -print(Pattern("awdwadawdada"))