# Tester plusieurs agents avec des combinations automatiques
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="Décris le fonctionnement du protocole HTTP.",
    agents=["formateur", "assistant_technique"],
    param_grid={
        "temperature": [0.5, 1.0, 1.5],
        "top_p": [0.5, 0.9, 1.0],
    },
    fixed_params={
        "num_predict": 300
    }
)