From 8f0e35ba79c2831d7c2d72d82d71c4b0c743875d Mon Sep 17 00:00:00 2001 From: The0Mikkel Date: Tue, 28 May 2024 09:54:16 +0200 Subject: [PATCH] fix(chat): stop caching messages in DMs --- bot.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/bot.py b/bot.py index e3e6399..42875ce 100644 --- a/bot.py +++ b/bot.py @@ -109,7 +109,15 @@ class Bot: if self.discord.user == message.author: # don't respond to ourselves return + + # Do not respond with llm in private messages + if isinstance(message.channel, discord.DMChannel): + response = DiscordResponse(message) + if self.discord.user.mentioned_in(message): + await response.write(message, 'I am sorry, I am unable to respond in private messages.') + return + # Do not respond to messages that don't mention us if not self.discord.user.mentioned_in(message) or message.author.bot or '@everyone' in message.content or '@here' in message.content: # don't respond to messages that don't mention us, but save it for context await self.save_message(str(message.channel.id), self.message(message, message.content), 'user') @@ -119,16 +127,11 @@ class Bot: if (random.random() * 1000) > 0.1: return + # Clean message content = message.content.replace(f'<@{self.discord.user.id}>', self.bot_name.title()).strip() if not content: 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: await self.flush_channel(str(message.channel.id))