# Tester un seul agent, avec des paramètres qui varient
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="Donne une définition claire de l'intelligence artificielle.",
    agents=["juriste"],
    param_grid={
        "temperature": [0.1, 0.5, 1.0],
        "top_p": [0.7, 0.9]
    },
    fixed_params={
        "num_predict": 256
    }
)