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