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"))