Files
aethas-free/modules/ai_local.py
T
2026-08-18 09:41:42 +02:00

16 lines
753 B
Python

from openai import OpenAI
from modules.logger import log_event
def interroger_local_ollama(prompt_securise, modele="qwen2.5:3b"):
client = OpenAI(base_url="http://192.168.1.20:11434/v1", api_key="ollama")
try:
resp = client.chat.completions.create(
model=modele,
messages=[{"role": "system", "content": "Réponds en français."}, {"role": "user", "content": prompt_securise}]
)
if resp and resp.choices:
log_event("INFO", f"LOCAL-{modele}", "Succès.")
return resp.choices[0].message.content, "#10b981", "SERVEUR LOCAL", modele
except Exception as e:
log_event("ERROR", f"LOCAL-{modele}", f"Erreur: {e}")
raise Exception("Le serveur local a échoué.")