alterware/aw-bot
1
0
Fork 0
mirror of https://github.com/alterware/aw-bot.git synced 2025-10-26 14:15:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Make Grok generic

Browse Source
This commit is contained in:
6arelyFuture
2025-04-19 16:24:20 +02:00
parent acb065d3ef
commit f996297de9
2 changed files with 37 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
bot/events_handlers/message_events.py
View File

@@ -53,7 +53,7 @@ def fetch_image_from_message(message):
return image_object
async def handle_bot_mention(message, bot):
async def handle_bot_mention(message, bot, no_context=False):
staff_role = message.guild.get_role(STAFF_ROLE_ID)
member = message.guild.get_member(message.author.id)
if staff_role in member.roles:
@@ -81,7 +81,9 @@ async def handle_bot_mention(message, bot):
print(f"An error occurred while fetching the referenced message: {e}")
# Pass the reply content to forward_to_google_api
await forward_to_google_api(message, bot, image_object, reply_content)
await forward_to_google_api(
message, bot, image_object, reply_content, no_context
)
return True
return False
@@ -361,11 +363,15 @@ async def handle_message(message, bot):
await handle_dm(message)
return
grok_role = message.guild.get_role(GROK_ROLE_ID)
if grok_role in message.role_mentions or bot.user in message.mentions:
if bot.user in message.mentions:
if await handle_bot_mention(message, bot):
return
grok_role = message.guild.get_role(GROK_ROLE_ID)
if grok_role in message.role_mentions:
if await handle_bot_mention(message, bot, True):
return
# Too many mentions
if len(message.mentions) >= 3:
member = message.guild.get_member(message.author.id)