mirror of
https://github.com/Ladebeze66/llm_lab_perso.git
synced 2025-12-15 19:26:52 +01:00
14 lines
549 B
Python
14 lines
549 B
Python
# 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.")
|