Update agent.py

agent.py

@@ -11,6 +11,23 @@ from smolagents import CodeAgent, MultiStepAgent, AgentError, PythonInterpreterT
 from models import ModelManager
 from tools import search_web, scrape_website, read_file
 
+
+class OrchestratorAgent(MultiStepAgent):
+    """
+    Agent orchestrateur qui hérite de MultiStepAgent et implémente 
+    la méthode requise pour le prompt système.
+    """
+    def initialize_system_prompt(self) -> str:
+        """Définit le prompt système pour l'agent orchestrateur."""
+        return (
+            "You are an expert orchestrator agent. Your primary role is to understand a user's question, "
+            "devise a multi-step plan, and use the available tools to find the answer. "
+            "Think step-by-step. Analyze the results of each tool call. "
+            "When you have gathered all the necessary information, formulate a clear and comprehensive final answer. "
+            "Cite your sources if you used the search or scrape tools."
+        )
+
+
 class UltraAgent:
     """
     Agent ultra-puissant avec orchestration multi-modèles.
@@ -45,14 +62,16 @@ class UltraAgent:
     def _init_specialized_agents(self):
         """Initialise les agents spécialisés comme l'orchestrateur et l'agent de code."""
         try:
-            # Agent principal (orchestrateur)
-            self.orchestrator = MultiStepAgent(
+
+            # Utiliser la nouvelle classe OrchestratorAgent au lieu de MultiStepAgent
+            self.orchestrator = OrchestratorAgent(
                 model=self.model_manager.get_orchestrator(),
                 tools=self.tools,
                 max_steps=15,
                 planning_interval=3
             )
             
+            
             # Agent de code
             self.code_agent = CodeAgent(
                 model=self.model_manager.get_code_agent(),