Johnounet vor 2 Monaten
Ursprung
Commit
62dd580c9d
1 geänderte Dateien mit 13 neuen und 2 gelöschten Zeilen
  1. 13 2
      chatbot.py

+ 13 - 2
chatbot.py

@@ -25,7 +25,7 @@ logger.addHandler(console_handler)
 load_dotenv()
 
 # Version du bot
-VERSION = "4.6.2"
+VERSION = "4.6.3"
 
 def get_env_variable(var_name, is_critical=True, default=None, var_type=str):
     value = os.getenv(var_name)
@@ -49,6 +49,15 @@ def get_env_variable(var_name, is_critical=True, default=None, var_type=str):
                 return default
             else:
                 raise ValueError(f"La variable d'environnement {var_name} doit être un entier.")
+    elif var_type == float:
+        try:
+            return float(value)
+        except ValueError:
+            logger.error(f"La variable d'environnement {var_name} doit être un nombre décimal. Valeur actuelle: {value}")
+            if default is not None:
+                return default
+            else:
+                raise ValueError(f"La variable d'environnement {var_name} doit être un nombre décimal.")
     return value
 
 try:
@@ -60,6 +69,7 @@ try:
     MAX_IMAGE_SIZE = get_env_variable('MAX_IMAGE_SIZE', is_critical=False, default=5*1024*1024, var_type=int)
     HISTORY_FILE = get_env_variable('HISTORY_FILE', is_critical=False, default="conversation_history.json")
     MISTRAL_MODEL = get_env_variable('MISTRAL_MODEL', is_critical=False, default="mistral-medium-latest")
+    TEMPERATURE = get_env_variable('TEMPERATURE', is_critical=False, default=0.7, var_type=float)
     logger.info("Toutes les variables d'environnement critiques ont été chargées avec succès.")
 except ValueError as e:
     logger.error(f"Erreur lors du chargement des variables d'environnement: {e}")
@@ -165,7 +175,8 @@ def call_mistral_api(prompt, history, image_url=None, user_id=None, username=Non
     data = {
         "model": MISTRAL_MODEL,
         "messages": messages,
-        "max_tokens": 128000
+        "max_tokens": 128000,
+        "temperature": TEMPERATURE
     }
     try:
         response = requests.post(MISTRAL_API_URL, headers=headers, data=json.dumps(data))