dev:style 项目结构和分类
This commit is contained in:
@@ -1,10 +1,12 @@
|
|||||||
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 timer
|
||||||
|
|
||||||
|
|
||||||
@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):
|
||||||
@@ -13,7 +15,7 @@ class MyPlugin(Star):
|
|||||||
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,8 +24,8 @@ 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"
|
||||||
@@ -33,22 +35,15 @@ class MyPlugin(Star):
|
|||||||
"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_llm_response()
|
||||||
async def pattern_extract(self,event:AstrMessageEvent,req:LLMResponse):
|
async def pattern_extract(self,event:AstrMessageEvent,req:LLMResponse):
|
||||||
if not req.role == 'assistant':
|
await timer(event,req)
|
||||||
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):
|
||||||
"""可选择实现异步的插件销毁方法,当插件被卸载/停用时会调用。"""
|
"""可选择实现异步的插件销毁方法,当插件被卸载/停用时会调用。"""
|
||||||
|
|||||||
@@ -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)))
|
||||||
+3
-2
@@ -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>"))
|
|
||||||
|
|||||||
Reference in New Issue
Block a user