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