Public Access
33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
import os
|
|
from google import genai
|
|
from modules.logger import log_event
|
|
|
|
def interroger_gemini(prompt_securise):
|
|
"""Interroge l'API Google Gemini via le nouveau SDK officiel (google-genai)."""
|
|
api_key = os.getenv("GEMINI_API_KEY")
|
|
if not api_key:
|
|
raise Exception("Clé API Gemini manquante dans l'environnement.")
|
|
|
|
try:
|
|
# Initialisation du client avec le nouveau SDK
|
|
client = genai.Client(api_key=api_key)
|
|
modeles = ["gemini-2.5-pro", "gemini-1.5-pro", "gemini-2.0-flash"]
|
|
|
|
for modele in modeles:
|
|
try:
|
|
# Appel avec configuration des instructions système
|
|
response = client.models.generate_content(
|
|
model=modele,
|
|
contents=prompt_securise,
|
|
)
|
|
if response and response.text:
|
|
log_event("INFO", f"GEMINI-{modele}", "Succès de la requête Gemini.")
|
|
return response.text, "#4285F4", "Google Gemini", modele
|
|
except Exception as sub_e:
|
|
log_event("WARNING", f"GEMINI-{modele}", f"Essai échoué pour {modele}: {sub_e}")
|
|
continue
|
|
|
|
raise Exception("Tous les modèles Gemini configurés ont échoué.")
|
|
except Exception as e:
|
|
log_event("ERROR", "GEMINI", f"Erreur critique Gemini : {e}")
|
|
raise e |