Public Access
22 lines
1.1 KiB
Python
22 lines
1.1 KiB
Python
import os
|
|
from openai import OpenAI
|
|
from modules.logger import log_event
|
|
|
|
def interroger_cloudflare(prompt_securise):
|
|
account_id = os.getenv("CLOUDFLARE_ACCOUNT_ID")
|
|
api_key = os.getenv("CLOUDFLARE_API_KEY")
|
|
if not account_id or not api_key: raise Exception("Identifiants Cloudflare manquants.")
|
|
client = OpenAI(base_url=f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/v1", api_key=api_key)
|
|
modeles = ["@cf/meta/llama-3-8b-instruct"]
|
|
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"CF-{modele}", "Succès.")
|
|
return resp.choices[0].message.content, "#f38ba8", "CLOUDFLARE", modele
|
|
except Exception as e:
|
|
log_event("WARNING", f"CF-{modele}", f"Erreur: {e}")
|
|
raise Exception("Tous les modèles Cloudflare ont échoué.") |