1 files changed, 21 insertions, 21 deletions

diff --git a/cogs/jarvis.py b/cogs/assistant.py
index 866700e..71f5150 100644
--- a/cogs/jarvis.py
+++ b/cogs/assistant.py
@@ -10,48 +10,48 @@ import requests
 import json
 import logging
 import shutil
-from dotenv import load_dotenv
 import os
 from tempfile import NamedTemporaryFile
 from cogs.downloader import Downloader
 
 logger = logging.getLogger(__name__)
-load_dotenv()
 
 class Jarvis(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
+        self.keyword = os.getenv("ASSISTANT_KEYWORD", "jarvis").lower()
 
     @commands.Cog.listener()
     async def on_message(self, message):
         if message.author == self.bot.user:
             return
-        
+
         logger.info(message.content)
+
         if message.mentions:
             if message.mentions[0].id == self.bot.user.id:
-                await self.jarvis(message)
+                await self.assistant(message)
 
-        elif "jarvis" in message.content.lower():
-            await self.jarvis(message)
+        elif self.keyword in message.content.lower():
+            await self.assistant(message)
 
     @staticmethod
-    async def jarvis(message):
+    async def assistant(message):
         headers = {
             "Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
             "Content-Type": "application/json"
         }
-        
+
         chain = await Jarvis._build_message_chain(message)
         messages = Jarvis._build_messages(message, chain)
         payload = Jarvis._build_payload(messages)
-        
+
         response_data = await Jarvis._make_api_request(payload, headers)
         response_message, action = Jarvis._parse_response(response_data)
-        
+
         channel = message.channel
         await message.reply(response_message)
-        
+
         if action == "download":
             await Jarvis._handle_download_action(message, channel)
 
@@ -60,7 +60,7 @@ class Jarvis(commands.Cog):
         chain = []
         current = message
         max_depth = 5
-        
+
         while current and len(chain) < max_depth:
             chain.append(current)
             if not current.reference:
@@ -72,9 +72,9 @@ class Jarvis(commands.Cog):
                     current = await current.channel.fetch_message(current.reference.message_id)
             except discord.NotFound:
                 break
-        
+
         return list(reversed(chain))
-    
+
     @staticmethod
     def _build_messages(message, chain=None):
         messages = [{
@@ -85,16 +85,16 @@ class Jarvis(commands.Cog):
                        "zachowuj sie jak Jarvis, bądź osobistym asystentem użytkownika, nie używaj emotikon"
                        "nie pytaj sie użytkownika czy chcesz abyś coś zrobił, użytkownik ci sam powie abyś coś wykonał"
         }]
-        
+
         if chain is None:
             chain = [message]
-        
+
         for msg in chain:
             messages.append({
                 "role": "user",
                 "content": f"{msg.author.display_name}: {msg.content}"
             })
-            
+
         return messages
 
     @staticmethod
@@ -144,13 +144,13 @@ class Jarvis(commands.Cog):
         content = response_data['choices'][0]['message']['content']
         inner_json_str = content.strip('```json\n').strip('```').strip()
         inner_data = json.loads(inner_json_str)
-        
+
         response_message = inner_data['message']
         action = inner_data['action']
-        
+
         logger.info(response_message)
         logger.info(action)
-        
+
         return response_message, action
 
     @staticmethod
@@ -172,7 +172,7 @@ class Jarvis(commands.Cog):
     async def _process_download_response(response, channel):
         if not isinstance(response, dict):
             return
-            
+
         if response["status"] == "success":
             await Jarvis._handle_successful_download(response, channel)
         else: