From 620d36f915ce946d7d4420b12dc2e67d3096c3ef Mon Sep 17 00:00:00 2001
From: Migatu <mtuszowski@gmail.com>
Date: Tue, 29 Aug 2023 12:17:04 +0200
Subject: [PATCH] Fix 2.0

---
 bot.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/bot.py b/bot.py
index b931fe5..49ca713 100644
--- a/bot.py
+++ b/bot.py
@@ -278,22 +278,25 @@ async def on_message(message):
     if kondziu_mentioned or "conjurer:" in message.content:
         async with channel.typing():
             logger.debug("Procedura chatu")
-            message.content = message.content.replace("conjurer: ", "")
-            message.content = message.content.replace(str(client.user), "")
-            prompt = message.content
 
+            message.content = message.content.replace("conjurer: ", "")
             if message.author.nick:
                 username = message.author.nick
             else:
                 username = message.author.name
             vykidailo = False
             bartender = False
+            if kondziu_mentioned:
+                prompt = message.clean_content
+            else:
+                prompt = message.content
             for role in message.author.roles:
                 if role.name == "Vykidailo":
                     vykidailo = True
                 if role.name == "Bartender":
                     bartender = True
             global MESSAGE_TABLE  # pylint: disable=global-statement
+
             result, MESSAGE_TABLE = await handle_response(
                 prompt, vykidailo, bartender, MESSAGE_TABLE, username, False
             )