Update agent.py

agent.py

@@ -212,7 +212,7 @@ class TestAgent:
         #                                         )
         #youtube_tools = MCPServerTool(server_params=server_parameters)
 
-        model = OpenAIServerModel(model_id="gpt-4o")
+        model = OpenAIServerModel(model_id="gpt-4.1")
         #model = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
         # Instantiate the agent
         self.agent = CodeAgent(
@@ -231,7 +231,7 @@ class TestAgent:
             model=model,
             max_steps=6,                              # V3 increase steps
             planning_interval=3,                      # V3 add structure
-            verbosity_level=2,
+            verbosity_level=0,
             use_structured_outputs_internally=True   # V3. Adds structure
         )
         # V3. add Guidance
@@ -240,7 +240,7 @@ class TestAgent:
 
         # V4. use prompt from the paper as guidance
         prompt = """\n\n
-                System: You are a general AI assistant. I will ask you a question. Report your thoughts, and
+                It is very important to remember the foillowing: 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.
@@ -250,8 +250,10 @@ class TestAgent:
                 digits in plain text unless specified otherwise.
                 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.
+                \n\n
+                Now it's your turn.
                 """
-        self.agent.prompt_templates['system_prompt'] =  prompt
+        self.agent.prompt_templates['system_prompt'] =  self.agent.prompt_templates['system_prompt']  + prompt
 
 
     def __call__(self, question: str) -> str: