nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Add interaction check to ollama view

Browse source
This commit is contained in:
Nexus 2024-01-10 16:04:58 +00:00
parent d0809a4881
commit b98e8024d4
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/cogs/ollama.py
View file

@ -6,6 +6,8 @@ import textwrap
import time import time
import typing import typing
import io import io
from discord import
from discord.ui import View, button from discord.ui import View, button
from fnmatch import fnmatch from fnmatch import fnmatch
@ -21,6 +23,9 @@ class OllamaView(View):
self.ctx = ctx self.ctx = ctx
self.cancel = asyncio.Event() self.cancel = asyncio.Event()
async def interaction_check(self, interaction: discord.Interaction) -> bool:
return interaction.user == self.ctx.user
@button(label="Stop", style=discord.ButtonStyle.danger, emoji="\N{wastebasket}\U0000fe0f") @button(label="Stop", style=discord.ButtonStyle.danger, emoji="\N{wastebasket}\U0000fe0f")
async def _stop(self, btn: discord.ui.Button, interaction: discord.Interaction): async def _stop(self, btn: discord.ui.Button, interaction: discord.Interaction):
self.cancel.set() self.cancel.set()