mirror of
https://github.com/Ladebeze66/llm_lab.git
synced 2025-12-15 19:26:52 +01:00
13 lines
410 B
Python
13 lines
410 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 core.factory import LLMFactory
|
|
from agents.roles import AGENTS
|
|
|
|
model = LLMFactory.create("mistral7b")
|
|
model.set_role("formateur", AGENTS["formateur"])
|
|
response = model.generate("Qu'est-ce qu'une API REST?")
|
|
print("-> Réponse générée.")
|