Update app.py

app.py

@@ -56,11 +56,6 @@ class BasicAgent:
             #provider='nebius',
             token=os.getenv('HF_Token'))
         
-        with open("prompts.yaml", 'r') as stream:
-            prompts = yaml.safe_load(stream)
-
-        #prompt_templates.final_answer.pre_messages = prompts['final_answer']['pre_messages']
-        
         self.agent = CodeAgent(
             model=model,
             tools=[FinalAnswerTool(), DuckDuckGoSearchTool(), VisitWebpageTool()], ## add your tools here (don't remove final answer),
@@ -70,9 +65,18 @@ class BasicAgent:
             #step_callbacks=[delay_execution_10]
         )
 
-        print('*$%')
+        
+        print('*$% Original system prompt')
 
         print(self.agent.prompt_templates['system_prompt'])
+
+        # Customize the system prompt for GAIA format.
+        with open("prompts.yaml", 'r') as stream:
+            prompts = yaml.safe_load(stream)
+
+            print('*$% Customize system prompt')
+            self.agent.prompt_templates['system_prompt'] = prompts['system_prompt']
+        
         
         print("BasicAgent initialized.")