1 files changed, 2 insertions, 8 deletions
diff --git a/ATRI/plugins/rich/__init__.py b/ATRI/plugins/rich/__init__.py
index 3b49750..2d8f8c6 100644
--- a/ATRI/plugins/rich/__init__.py
+++ b/ATRI/plugins/rich/__init__.py
@@ -1,20 +1,15 @@
 from nonebot.adapters.onebot.v11 import MessageEvent
+from nonebot.adapters.onebot.v11.helpers import Cooldown
 
-from ATRI.utils.limit import FreqLimiter
 from ATRI.log import logger as log
 from .data_source import Rich
 
-_rich_flmt = FreqLimiter(3)
 
 bili_rich = Rich().on_message("小程序检测", "小程序爪巴", block=False)
 
 
-@bili_rich.handle()
+@bili_rich.handle([Cooldown(3)])
 async def _fk_bili(event: MessageEvent):
-    user_id = event.get_user_id()
-    if not _rich_flmt.check(user_id):
-        return
-
     msg = str(event.message)
     try:
         result, is_ok = await Rich().fk_bili(msg)
@@ -23,5 +18,4 @@ async def _fk_bili(event: MessageEvent):
     log.debug(result, is_ok)
     if not is_ok:
         return
-    _rich_flmt.start_cd(user_id)
     await bili_rich.finish(result)