ernestmindres commited on
Commit
00a0107
·
verified ·
1 Parent(s): fda0c6e

Update model_handler.py

Browse files
Files changed (1) hide show
  1. model_handler.py +17 -2
model_handler.py CHANGED
@@ -3,7 +3,9 @@ import sys
3
  from google import genai
4
  from google.genai.errors import APIError
5
  from google.genai.types import Part
6
- from config import GEMINI_API_KEY_INTERNAL, GEMINI_MODEL_NAME, DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE, SYSTEM_PROMPT
 
 
7
 
8
  # --- Configuration du Client Gemini ---
9
 
@@ -13,9 +15,22 @@ MODEL_LOADED_SUCCESSFULLY = False
13
  def get_model_instance():
14
  """Initialise ou retourne l'instance du client Gemini."""
15
  global client, MODEL_LOADED_SUCCESSFULLY
 
16
  if client is None:
 
 
 
 
 
 
 
 
 
 
 
17
  try:
18
- client = genai.Client(api_key=GEMINI_API_KEY_INTERNAL)
 
19
  MODEL_LOADED_SUCCESSFULLY = True
20
  print(f"INFO: Client Gemini initialisé. Modèle configuré pour utiliser {GEMINI_MODEL_NAME}.")
21
  except Exception as e:
 
3
  from google import genai
4
  from google.genai.errors import APIError
5
  from google.genai.types import Part
6
+ # Importez os pour lire la variable d'environnement directement ici
7
+ import os
8
+ from config import GEMINI_MODEL_NAME, DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE, SYSTEM_PROMPT
9
 
10
  # --- Configuration du Client Gemini ---
11
 
 
15
  def get_model_instance():
16
  """Initialise ou retourne l'instance du client Gemini."""
17
  global client, MODEL_LOADED_SUCCESSFULLY
18
+
19
  if client is None:
20
+
21
+ # --- NOUVEAU: Récupérer la clé API au moment de l'initialisation ---
22
+ # On lit la variable d'environnement "GEMINI_API_KEY" directement ici.
23
+ # Cela garantit que la lecture est effectuée dans le contexte de la fonction.
24
+ api_key = os.environ.get("GEMINI_API_KEY")
25
+
26
+ if not api_key:
27
+ print("ALERTE: Variable d'environnement 'GEMINI_API_KEY' non trouvée.", file=sys.stderr)
28
+ MODEL_LOADED_SUCCESSFULLY = False
29
+ return
30
+
31
  try:
32
+ # Utilisez la clé API récupérée
33
+ client = genai.Client(api_key=api_key)
34
  MODEL_LOADED_SUCCESSFULLY = True
35
  print(f"INFO: Client Gemini initialisé. Modèle configuré pour utiliser {GEMINI_MODEL_NAME}.")
36
  except Exception as e: