Public Access
27 lines
1.3 KiB
Python
27 lines
1.3 KiB
Python
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é.") |