Update main.py

compsci-adl · Nov 9, 2024 · a1f7a34 · a1f7a34
1 parent 602831f
commit a1f7a34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -195,7 +195,7 @@ async def on_message(message: Message):
         bot_response = await client.gemini_model.query(
             author_id=message.author.id,
             author=message.author.display_name,
-            message=message.clean_content.replace("d.chat", ""),
+            message=message.content.replace("d.chat", "").replace(f"<@{client.user.id}>", "").strip(),
             attachment=attachment,
         )
         await message.reply(embeds=bot_response, mention_author=False)