Skip to content

Commit 9e05282

Browse files
committed
🚑 多轮对话仅允许会话发起者响应
1 parent ce10ea1 commit 9e05282

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

nonebot_plugin_deepseek/__init__.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
from nonebot import require
66
from nonebot.adapters import Event
77
from nonebot.params import Depends
8-
from nonebot.permission import SuperUser
8+
from nonebot.matcher import Matcher
9+
from nonebot.permission import User, SuperUser, Permission
910
from nonebot.plugin import PluginMetadata, inherit_supported_adapters
1011

1112
require("nonebot_plugin_waiter")
@@ -166,6 +167,8 @@ async def _(
166167

167168
@deepseek.handle()
168169
async def _(
170+
event: Event,
171+
matcher: Matcher,
169172
content: Match[tuple[str, ...]],
170173
model_name: Query[str] = Query("use-model.model", model_config.default_model),
171174
context_option: Query[bool] = Query("with-context.value"),
@@ -226,7 +229,8 @@ def handler(event: Event):
226229
return False
227230
return text
228231

229-
waiter = Waiter(waits=["message"], handler=handler, matcher=deepseek)
232+
permission = Permission(User.from_event(event, perm=matcher.permission))
233+
waiter = Waiter(waits=["message"], handler=handler, matcher=deepseek, permission=permission)
230234
waiter.future.set_result("")
231235

232236
async for resp in waiter(default=False):

0 commit comments

Comments
 (0)