from openai import OpenAI
from modules.logger import log_event

def interroger_local_ollama(prompt_securise: str, modele: str = "qwen2.5-coder:3b"):
    """
    Interroge le serveur local Ollama avec repli automatique sur le modèle par défaut.
    """
    modele_effectif = modele if modele else "qwen2.5-coder:3b"
    client = OpenAI(base_url="http://192.168.1.20:11434/v1", api_key="ollama")
    
    try:
        resp = client.chat.completions.create(
            model=modele_effectif,
            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_effectif}", "Succès de la requête.")
            return resp.choices[0].message.content, "#10b981", "SERVEUR LOCAL", modele_effectif
            
    except Exception as e:
        log_event("ERROR", f"LOCAL-{modele_effectif}", f"Erreur : {e}")
        raise Exception("Le serveur local a échoué.")