from llm_classes.mistral_medium import MistralMedium
from llm_classes.pixtral_12b import Pixtral12b

print("Initialisation des modèles LLM...")

# Initialisation des modèles
try:
    text_model = MistralMedium()
    image_model = Pixtral12b()
    print("Modèles initialisés avec succès!")
except Exception as e:
    print(f"Erreur lors de l'initialisation des modèles: {str(e)}")
    exit(1)

# Test d'interrogation simple
try:
    question = "Quelle est la capitale de la France?"
    print(f"\nTest d'interrogation simple sur MistralMedium:")
    print(f"Question: {question}")
    response = text_model.interroger(question)
    print(f"Réponse: {response[:100]}...")
except Exception as e:
    print(f"Erreur lors de l'interrogation simple: {str(e)}")

print("\nTests terminés!")