fix: retrait de la dépendance au logger manquant et priorisation de Gemini Direct
Build and Push Docker Image / build-and-push (push) Successful in 42s
Build and Push Docker Image / build-and-push (push) Successful in 42s
This commit is contained in:
1 parent
6f7e3efd6e
commit
ec37ac06fe
1 file changed
+1
-7
+1
-7
@@ -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)}")
|
||||||
Reference in new issue
Block a user