fix: retrait de la dépendance au logger manquant et priorisation de Gemini Direct

This commit is contained in:
xavier committed 2026-10-07 14:39:40 +02:00
1 parent 6f7e3efd6e
commit eab77d2c7d
1 file changed
+1 -7
+1 -7
View File
@@ -1,12 +1,10 @@
from openai import OpenAI from openai import OpenAI
from fastapi import HTTPException from fastapi import HTTPException
from .models import SystemSettings from .models import SystemSettings
from .logger import system_logger
def get_ai_response(messages: list, settings: SystemSettings) -> str: def get_ai_response(messages: list, settings: SystemSettings) -> str:
"""Route la conversation vers le premier fournisseur IA disponible, priorité à Gemini Direct.""" """Route la conversation vers le premier fournisseur IA disponible, priorité à Gemini Direct."""
if not settings: if not settings:
system_logger.error("Configuration système introuvable lors de l'appel IA.")
raise HTTPException(status_code=500, detail="Configuration système introuvable.") raise HTTPException(status_code=500, detail="Configuration système introuvable.")
formatted_messages = [{"role": msg.role, "content": msg.content} for msg in messages] formatted_messages = [{"role": msg.role, "content": msg.content} for msg in messages]
@@ -20,14 +18,12 @@ def get_ai_response(messages: list, settings: SystemSettings) -> str:
) )
# Utilisation de l'identifiant standard API pour la version Flash # Utilisation de l'identifiant standard API pour la version Flash
model = "gemini-1.5-flash" model = "gemini-1.5-flash"
system_logger.info(f"Appel IA via Google Gemini direct (Modèle: {model})")
response = client.chat.completions.create(model=model, messages=formatted_messages) response = client.chat.completions.create(model=model, messages=formatted_messages)
# 2. Test OpenRouter (Fallback) # 2. Test OpenRouter (Fallback)
elif settings.openrouter_api_key: elif settings.openrouter_api_key:
client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=settings.openrouter_api_key) client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=settings.openrouter_api_key)
model = "google/gemini-1.5-pro" # Identifiant mis à jour et valide model = "google/gemini-1.5-pro" # Identifiant mis à jour et valide
system_logger.info(f"Appel IA via OpenRouter (Modèle: {model})")
response = client.chat.completions.create( response = client.chat.completions.create(
model=model, model=model,
messages=formatted_messages, messages=formatted_messages,
@@ -38,14 +34,12 @@ def get_ai_response(messages: list, settings: SystemSettings) -> str:
elif settings.deepseek_api_key: elif settings.deepseek_api_key:
client = OpenAI(base_url="https://api.deepseek.com/v1", api_key=settings.deepseek_api_key) client = OpenAI(base_url="https://api.deepseek.com/v1", api_key=settings.deepseek_api_key)
model = "deepseek-chat" model = "deepseek-chat"
system_logger.info(f"Appel IA via DeepSeek (Modèle: {model})")
response = client.chat.completions.create(model=model, messages=formatted_messages) response = client.chat.completions.create(model=model, messages=formatted_messages)
# 4. Test Groq # 4. Test Groq
elif settings.groq_api_key: elif settings.groq_api_key:
client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=settings.groq_api_key) client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=settings.groq_api_key)
model = "llama3-8b-8192" model = "llama3-8b-8192"
system_logger.info(f"Appel IA via Groq (Modèle: {model})")
response = client.chat.completions.create(model=model, messages=formatted_messages) response = client.chat.completions.create(model=model, messages=formatted_messages)
else: else:
@@ -54,5 +48,5 @@ def get_ai_response(messages: list, settings: SystemSettings) -> str:
return response.choices[0].message.content return response.choices[0].message.content
except Exception as e: except Exception as e:
system_logger.error(f"Erreur API IA : {str(e)}") print(f"Erreur API IA : {str(e)}")
raise HTTPException(status_code=502, detail=f"Détail du fournisseur : {str(e)}") raise HTTPException(status_code=502, detail=f"Détail du fournisseur : {str(e)}")