AIagent/llm_classes/test_class.py

from llm_classes.mistral import Mistral
from llm_classes.pixtral import Pixtral

def tester_mistral():
    """Teste la classe Mistral et affiche les modèles disponibles"""
    print("\n" + "="*50)
    print("Test de la classe Mistral")
    print("="*50)
    
    # Création d'une instance de Mistral
    mistral = Mistral()
    
    print("Récupération des modèles disponibles via l'API Mistral...")
    
    try:
        # Récupération de la liste des modèles avec la méthode correcte
        modeles = mistral.model_list()
        
        print("\nModèles disponibles:")
        print("-" * 30)
        
        # Affichage de tous les modèles disponibles
        for i, modele in enumerate(modeles, 1):
            print(f"{i}. {modele}")
            
    except Exception as e:
        print(f"Erreur lors de la récupération des modèles: {e}")

def tester_pixtral():
    """Teste la classe Pixtral spécialisée pour le modèle pixtral-12b-latest"""
    print("\n" + "="*50)
    print("Test de la classe Pixtral")
    print("="*50)
    
    # Création d'une instance de Pixtral
    pixtral = Pixtral()
    
    try:
        # Afficher le modèle utilisé
        print(f"\nModèle utilisé par défaut: {pixtral.Modele}")
        
        # Faire une requête de test simple
        print("\nTest d'une requête simple...")
        pixtral.prompt_system = "Tu es un assistant utile, précis et concis."
        reponse = pixtral.Interroger("Présente-toi en une phrase.")
        
        print("\nRéponse reçue:")
        print("-" * 30)
        print(reponse)
        
    except Exception as e:
        print(f"Erreur lors du test de Pixtral: {e}")

def main():
    """Fonction principale qui exécute les tests"""
    # Test de la classe Mistral
    tester_mistral()
    
    # Test de la classe Pixtral
    tester_pixtral()
    
    print("\nTests terminés.")

if __name__ == "__main__":
    main()