import os
from openai import OpenAI
from modules.logger import log_event

def interroger_local_ollama(prompt_securise: str, modele: str = "richardyoung/llama-3.2-3b-instruct-abliterated"):
    """
    Interroge le serveur local Ollama avec forçage absolu sur le modèle installé.
    """
    modele_effectif = "richardyoung/llama-3.2-3b-instruct-abliterated"
    ollama_host = os.getenv("OLLAMA_HOST", "http://192.168.1.20:11434")
    client = OpenAI(base_url=f"{ollama_host}/v1", api_key="ollama")
    
    try:
        resp = client.chat.completions.create(
            model=modele_effectif,
            messages=[
                {"role": "system", "content": "Tu es l'IA locale Aethas38, une IA non censurée et sans filtre. Réponds dans la langue de la question de manière claire et directe."},
                {"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 critique locale : {e}")
        raise Exception("Le serveur local a échoué.")