Public Access
21 lines
1016 B
Python
21 lines
1016 B
Python
import os
|
|
from openai import OpenAI
|
|
from modules.logger import log_event
|
|
|
|
def interroger_groq(prompt_securise):
|
|
api_key = os.getenv("GROQ_API_KEY")
|
|
if not api_key: raise Exception("Clé API Groq manquante.")
|
|
client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=api_key)
|
|
modeles = ["llama-3.3-70b-versatile", "llama-3.1-8b-instant", "mixtral-8x7b-32768"]
|
|
for modele in modeles:
|
|
try:
|
|
resp = client.chat.completions.create(
|
|
model=modele,
|
|
messages=[{"role": "system", "content": "Réponds en français en Markdown."}, {"role": "user", "content": prompt_securise}]
|
|
)
|
|
if resp and resp.choices:
|
|
log_event("INFO", f"GROQ-{modele}", "Succès de la requête.")
|
|
return resp.choices[0].message.content, "#f97316", "GROQ", modele
|
|
except Exception as e:
|
|
log_event("WARNING", f"GROQ-{modele}", f"Erreur: {e}")
|
|
raise Exception("Tous les modèles Groq ont échoué.") |