cybrneko/ollama-discord-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(chat): do not respond with llm in private messages

Browse Source
This commit is contained in:
The0Mikkel 2024-05-28 01:09:52 +02:00
parent 886a45937d
commit 84c528b751
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bot.py
View File

@ -89,7 +89,7 @@ class Bot:
async def on_message(self, message): async def on_message(self, message):
if not self.ready: if not self.ready:
return return
string_channel_id = str(message.channel.id) string_channel_id = str(message.channel.id)
if self.discord.user == message.author: if self.discord.user == message.author:
@ -108,7 +108,14 @@ class Bot:
content = message.content.replace(f'<@{self.discord.user.id}>', self.bot_name.title()).strip() content = message.content.replace(f'<@{self.discord.user.id}>', self.bot_name.title()).strip()
if not content: if not content:
return return
# Do not respond with llm in private messages
if isinstance(message.channel, discord.DMChannel):
response = DiscordResponse(message)
await response.write(message, 'I am sorry, I am unable to respond in private messages.')
return
# Admin commands
if content == 'RESET' and str(message.author.id) == self.admin_id: if content == 'RESET' and str(message.author.id) == self.admin_id:
await self.flush_channel(str(message.channel.id)) await self.flush_channel(str(message.channel.id))
logging.info('Chat reset by admin in guild %s', message.channel.guild.name) logging.info('Chat reset by admin in guild %s', message.channel.guild.name)