#Tester plusieurs agents avec les mêmes paramètres
import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from core.factory import LLMFactory
from utils.parameter_tester import test_agents_on_prompt

test_agents_on_prompt(
    model_class=LLMFactory._registry["mistral7b"],
    prompt="Explique ce qu'est le machine learning.",
    agents=["formateur", "chercheur", "assistant_technique"],
    param_grid={},
    fixed_params={
        "temperature": 1.5,
        "top_p": 0.9,
        "num_predict": 300
    }
)