# Tester un seul agent avec un seul prompt, sans variation
import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))

from utils.agent_manager import AgentManager

# Utilisation de l'AgentManager pour créer l'agent avec le modèle approprié
agent = AgentManager.create("test")  # Utilise l'agent "test" qui est configuré avec Mistral7B

print(f"Test de l'agent '{agent.agent}' avec le modèle '{agent.model}'")
response = agent.generate("Qu'est-ce qu'une API REST?")
print("-> Réponse générée.")