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

Add mention cleaning

Browse Source
This commit is contained in:
The0Mikkel 2024-05-27 23:21:44 +02:00
parent add98c0202
commit 03646b1142
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
bot/discollama.py
View File

@ -39,7 +39,13 @@ class DiscordResponse:
logging.info('Empty response, not sending')
value = '*I don\'t have anything to say.*'
self.r = await self.channel.send(value, reference=message)
self.r = await self.channel.send(self.sanitize(value), reference=message)
def sanitize(self, s):
step1 = s.replace('@everyone', '@\u200beveryone').replace('@here', '@\u200bhere')
step2 = discord.utils.escape_mentions(step1)
return step2
class Bot: