fix(chat): Remove @DuckBot in response (#79)

compsci-adl · Nov 10, 2024 · 6ac3314 · 6ac3314
1 parent 602831f
commit 6ac3314
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -195,7 +195,9 @@ 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)