mirror of
https://github.com/Ladebeze66/llm_ticket3.git
synced 2025-12-16 06:28:02 +01:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from orchestrator import Orchestrator
|
|
from agents.agent_json_analyser import AgentJsonAnalyser
|
|
from agents.agent_image_sorter import AgentImageSorter
|
|
from agents.agent_image_analyser import AgentImageAnalyser
|
|
from agents.agent_report_generator import AgentReportGenerator
|
|
|
|
from llm_classes.mistral_large import MistralLarge
|
|
from llm_classes.pixtral_12b import Pixtral12b
|
|
from llm_classes.ollama import Ollama
|
|
|
|
def test_orchestrator():
|
|
# Initialisation des LLM
|
|
json_llm = MistralLarge()
|
|
image_sorter_llm = Pixtral12b()
|
|
image_analyser_llm = Ollama()
|
|
report_generator_llm = MistralLarge()
|
|
|
|
# Création des agents
|
|
json_agent = AgentJsonAnalyser(json_llm)
|
|
image_sorter = AgentImageSorter(image_sorter_llm)
|
|
image_analyser = AgentImageAnalyser(image_analyser_llm)
|
|
report_generator = AgentReportGenerator(report_generator_llm)
|
|
|
|
# Initialisation de l'orchestrateur avec les agents
|
|
orchestrator = Orchestrator(
|
|
output_dir="output/",
|
|
json_agent=json_agent,
|
|
image_sorter=image_sorter,
|
|
image_analyser=image_analyser,
|
|
report_generator=report_generator
|
|
)
|
|
|
|
# Exécution complète de l'orchestrateur
|
|
orchestrator.executer()
|
|
|
|
if __name__ == "__main__":
|
|
test_orchestrator()
|