OpenWebAI-DiscordBots/OpenWebUI-Discordbot
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding back the dm chat
All checks were successful
OpenWebUI Discord Bot / Build-and-Push (push) Successful in 58s
Details

Browse Source
This commit is contained in:
Josh Knapp 2024-12-30 21:52:32 -08:00
parent aa9d6e9765
commit 35839395f4
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/discordbot.py
View File

@ -64,8 +64,13 @@ async def on_message(message):
if message.author == bot.user: if message.author == bot.user:
return return
if bot.user in message.mentions: # Respond to DMs or when mentioned in a server
prompt = message.content.replace(f'<@{bot.user.id}>', '').strip() if isinstance(message.channel, discord.DMChannel) or bot.user in message.mentions:
# For mentions, remove the bot mention from the message
if bot.user in message.mentions:
prompt = message.content.replace(f'<@{bot.user.id}>', '').strip()
else:
prompt = message.content.strip()
if not prompt: if not prompt:
await message.channel.send("Hello! How can I help you?") await message.channel.send("Hello! How can I help you?")