diff --git a/fjerkroa_bot/discord_bot.py b/fjerkroa_bot/discord_bot.py index b7cd4e0..9c59bc4 100644 --- a/fjerkroa_bot/discord_bot.py +++ b/fjerkroa_bot/discord_bot.py @@ -205,7 +205,8 @@ class FjerkroaBot(commands.Bot): if user is not None: message_content = re.sub(f'[<][@][!]? *{uid} *[>]', f'@{user.name}', message_content) channel_name = self.get_channel_name(message.channel) - msg = AIMessage(message.author.name, message_content, channel_name, self.user in message.mentions) + msg = AIMessage(message.author.name, message_content, channel_name, + self.user in message.mentions or isinstance(message.channel, DMChannel)) if message.attachments: for attachment in message.attachments: if not msg.urls: diff --git a/openai_chat.dat b/openai_chat.dat index 586ed71..b792dfa 100644 Binary files a/openai_chat.dat and b/openai_chat.dat differ