Update agent.py

agent.py

@@ -229,8 +229,8 @@ class TestAgent:
                    FinalAnswerTool()],
             additional_authorized_imports=["pandas","markdownify","requests","chess","os"],    # V2 add markdownify & requests V5 add chess and os
             model=model,
-            max_steps=4,                              # V3 increase steps
-            planning_interval=2,                      # V3 add structure
+            max_steps=6,                              # V3 increase steps
+            planning_interval=3,                      # V3 add structure
             verbosity_level=2,
             use_structured_outputs_internally=True   # V3. Adds structure
         )
@@ -240,7 +240,7 @@ class TestAgent:
 
         # V4. use prompt from the paper as guidance
         prompt = """\n\n
-                Remember: You are a general AI assistant. I will ask you a question. Report your thoughts, and
+                System: You are a general AI assistant. I will ask you a question. Report your thoughts, and
                 finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
                 YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated
                 list of numbers and/or strings.
@@ -251,7 +251,7 @@ class TestAgent:
                 If you are asked for a comma separated list, apply the above rules depending of whether the element
                 to be put in the list is a number or a string.
                 """
-        self.agent.prompt_templates['system_prompt'] = self.agent.prompt_templates['system_prompt'] + prompt
+        self.agent.prompt_templates['system_prompt'] =  prompt
 
 
     def __call__(self, question: str) -> str: