Update model_handler.py

model_handler.py

@@ -1,99 +1,99 @@
-import os
-import requests
-import json
-import time
-import sys
-from config import GEMINI_API_KEY_INTERNAL, GEMINI_MODEL_NAME, DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE
-
-# Le modèle est maintenant une API externe
-# On utilise requests pour simplifier l'intégration et simuler l'appel curl de l'utilisateur.
-
-# URL de l'API Gemini
-API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/{GEMINI_MODEL_NAME}:generateContent"
-
-# Indicateur de chargement du "modèle" (ici, de la clé API)
-MODEL_LOADED_SUCCESSFULLY = False
-
-# Vérification simple de la clé API
-if GEMINI_API_KEY_INTERNAL:
-    print(f"INFO: Clé API Gemini détectée. Modèle configuré pour utiliser {GEMINI_MODEL_NAME}.")
-    MODEL_LOADED_SUCCESSFULLY = True
-else:
-    print("ALERTE: Clé API Gemini non trouvée dans la configuration. Le service de génération ne fonctionnera pas.", file=sys.stderr)
-
-
-# Le prompt système (System Instruction) pour guider le modèle
-SYSTEM_PROMPT = "Vous êtes un assistant IA appler Ernestmind, expert en programmation, développement web (full-stack), et résolution de problèmes techniques. Répondez de manière concise, précise et professionnelle selon la langue utilisateur, en utilisant la mise en forme Markdown appropriée (blocs de code si nécessaire)."
-
-def get_model_instance():
-    """
-    Simule la récupération de l'instance du modèle, renvoyant simplement l'état du service API.
-    """
-    class GeminiAPIClient:
-        def __init__(self, model_name):
-            self.model_name = model_name
-            self.model_path = f"API Gemini ({self.model_name})"
-        
-    return GeminiAPIClient(GEMINI_MODEL_NAME)
-
-
-def generate_code_response(user_prompt, max_tokens, temperature):
-    """
-    Génère la réponse via l'API REST de Gemini.
-    """
-    if not MODEL_LOADED_SUCCESSFULLY:
-        return "Erreur de configuration: L'API Gemini n'est pas initialisée (clé manquante)."
-
-    # Payload de la requête API REST
-    payload = {
-        "contents": [
-            {
-                "role": "user",
-                "parts": [
-                    # Combinaison du System Prompt et du User Prompt pour une meilleure instruction
-                    {"text": f"SYSTEM INSTRUCTION: {SYSTEM_PROMPT}\n\nUSER QUERY: {user_prompt}"}
-                ]
-            }
-        ],
-        "config": {
-            "maxOutputTokens": max_tokens,
-            "temperature": temperature,
-        }
-    }
-    
-    # En-têtes HTTP
-    headers = {
-        'Content-Type': 'application/json',
-        # Utilisation de la clé dans l'en-tête, comme dans l'exemple curl de l'utilisateur.
-        'X-goog-api-key': GEMINI_API_KEY_INTERNAL,
-    }
-
-    try:
-        # Appel à l'API REST
-        # Augmentation du timeout pour les réponses potentiellement longues
-        response = requests.post(API_URL, headers=headers, data=json.dumps(payload), timeout=180)
-        
-        # Vérification du statut HTTP
-        if response.status_code != 200:
-            error_data = response.json()
-            error_details = error_data.get('error', {}).get('message', 'Détails non disponibles')
-            print(f"Erreur d'API Gemini (Code {response.status_code}): {error_details}", file=sys.stderr)
-            return f"Erreur de l'API Gemini: {error_details}"
-
-        # Extraction du texte
-        response_data = response.json()
-        
-        # Le texte est dans response_data['candidates'][0]['content']['parts'][0]['text']
-        candidates = response_data.get('candidates')
-        if candidates and candidates[0].get('content'):
-            text = candidates[0]['content']['parts'][0]['text']
-            return text.strip()
-        
-        return "Réponse vide reçue de l'API Gemini."
-
-    except requests.exceptions.RequestException as e:
-        print(f"Erreur réseau lors de l'appel API Gemini: {e}", file=sys.stderr)
-        return f"Erreur de communication avec l'API Gemini: {e}"
-    except Exception as e:
-        print(f"Erreur inattendue lors de la génération: {e}", file=sys.stderr)
-        return f"Erreur interne lors du traitement de la réponse: {e}"
+import os
+import requests
+import json
+import time
+import sys
+from config import GEMINI_API_KEY_INTERNAL, GEMINI_MODEL_NAME, DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE
+
+# Le modèle est maintenant une API externe
+# On utilise requests pour simplifier l'intégration et simuler l'appel curl de l'utilisateur.
+
+# URL de l'API Gemini
+API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/{GEMINI_MODEL_NAME}:generateContent"
+
+# Indicateur de chargement du "modèle" (ici, de la clé API)
+MODEL_LOADED_SUCCESSFULLY = False
+
+# Vérification simple de la clé API
+if GEMINI_API_KEY_INTERNAL:
+    print(f"INFO: Clé API Gemini détectée. Modèle configuré pour utiliser {GEMINI_MODEL_NAME}.")
+    MODEL_LOADED_SUCCESSFULLY = True
+else:
+    print("ALERTE: Clé API Gemini non trouvée dans la configuration. Le service de génération ne fonctionnera pas.", file=sys.stderr)
+
+
+# Le prompt système (System Instruction) pour guider le modèle
+SYSTEM_PROMPT = "Vous êtes un assistant IA appler Ernestmind, expert en programmation, développement web (full-stack), et résolution de problèmes techniques. Répondez de manière concise, précise et professionnelle selon la langue utilisateur, en utilisant la mise en forme Markdown appropriée (blocs de code si nécessaire)."
+
+def get_model_instance():
+    """
+    Simule la récupération de l'instance du modèle, renvoyant simplement l'état du service API.
+    """
+    class GeminiAPIClient:
+        def __init__(self, model_name):
+            self.model_name = model_name
+            self.model_path = f"API Gemini ({self.model_name})"
+        
+    return GeminiAPIClient(GEMINI_MODEL_NAME)
+
+
+def generate_code_response(user_prompt, max_tokens, temperature):
+    """
+    Génère la réponse via l'API REST de Gemini.
+    """
+    if not MODEL_LOADED_SUCCESSFULLY:
+        return "Erreur de configuration: L'API Gemini n'est pas initialisée (clé manquante)."
+
+    # Payload de la requête API REST
+    payload = {
+        "contents": [
+            {
+                "role": "user",
+                "parts": [
+                    # Combinaison du System Prompt et du User Prompt pour une meilleure instruction
+                    {"text": f"SYSTEM INSTRUCTION: {SYSTEM_PROMPT}\n\nUSER QUERY: {user_prompt}"}
+                ]
+            }
+        ],
+        "generationConfig": { # <--- C'EST LA CORRECTION
+            "maxOutputTokens": max_tokens,
+            "temperature": temperature,
+        }
+    }
+    
+    # En-têtes HTTP
+    headers = {
+        'Content-Type': 'application/json',
+        # Utilisation de la clé dans l'en-tête, comme dans l'exemple curl de l'utilisateur.
+        'X-goog-api-key': GEMINI_API_KEY_INTERNAL,
+    }
+
+    try:
+        # Appel à l'API REST
+        # Augmentation du timeout pour les réponses potentiellement longues
+        response = requests.post(API_URL, headers=headers, data=json.dumps(payload), timeout=180)
+        
+        # Vérification du statut HTTP
+        if response.status_code != 200:
+            error_data = response.json()
+            error_details = error_data.get('error', {}).get('message', 'Détails non disponibles')
+            print(f"Erreur d'API Gemini (Code {response.status_code}): {error_details}", file=sys.stderr)
+            return f"Erreur de l'API Gemini: {error_details}"
+
+        # Extraction du texte
+        response_data = response.json()
+        
+        # Le texte est dans response_data['candidates'][0]['content']['parts'][0]['text']
+        candidates = response_data.get('candidates')
+        if candidates and candidates[0].get('content'):
+            text = candidates[0]['content']['parts'][0]['text']
+            return text.strip()
+        
+        return "Réponse vide reçue de l'API Gemini."
+
+    except requests.exceptions.RequestException as e:
+        print(f"Erreur réseau lors de l'appel API Gemini: {e}", file=sys.stderr)
+        return f"Erreur de communication avec l'API Gemini: {e}"
+    except Exception as e:
+        print(f"Erreur inattendue lors de la génération: {e}", file=sys.stderr)
+        return f"Erreur interne lors du traitement de la réponse: {e}"