Update app.py

app.py

@@ -39,12 +39,12 @@ class BasicAgent:
     FINAL ANSWER: [YOUR FINAL ANSWER]"
     """
     
-    def __init__(self, model_name: str = "gpt-3.5-turbo", api_key: str = None):
+    def __init__(self, model_name: str = "gpt-4.1", api_key: str = None):
         """
         Initialize the BasicAgent with the specified LLM.
         
         Args:
-            model_name: The name of the model to use (defaults to gpt-3.5-turbo)
+            model_name: The name of the model to use (defaults to gpt-4.1)
             api_key: OpenAI API key (defaults to environment variable)
         """
         # Use the provided API key or fall back to the environment variable
@@ -144,34 +144,30 @@ class BasicAgent:
                 Reasoning:
                 {all_thoughts}
                 
-                Format your FINAL ANSWER following these strict rules:
+                Format your answer following these strict rules:
                 - For numbers: No commas in large numbers, no units unless explicitly requested.
                 - For strings: Minimal, no articles, no abbreviations.
                 - For lists: Comma-separated values.
                 - Use digits in words unless otherwise stated.
                 
-                IMPORTANT: Your answer must be concise and precise. Do not include explanations in your final answer.
+                IMPORTANT: Your answer must be concise, precise and EXACT. Do not include explanations.
+                Do not include a prefix like "FINAL ANSWER:" in your response. Just provide the exact answer.
                 
-                Your response must follow this exact format:
-                FINAL ANSWER: [concise answer following the formatting rules above]
+                For example, if the answer is "5", just respond with "5".
+                If the answer is "New York", just respond with "New York".
             """)
         ]
         
         try:
             response = self.llm.invoke(prompt)
+            answer = response.content.strip()
             
-            # Extract the final answer using regex
-            final_answer_match = re.search(r'FINAL ANSWER:\s*(.*?)(?:\n|$)', response.content)
-            if final_answer_match:
-                return final_answer_match.group(1).strip()
-            elif "FINAL ANSWER:" in response.content:
-                parts = response.content.split("FINAL ANSWER:")
-                if len(parts) > 1:
-                    return parts[1].strip()
-                else:
-                    return "Error: Malformed answer format."
-            else:
-                return response.content.strip()
+            # Remove any "FINAL ANSWER:" prefix if it exists
+            if "FINAL ANSWER:" in answer:
+                parts = answer.split("FINAL ANSWER:")
+                answer = parts[1].strip()
+            
+            return answer
         except Exception as e:
             return f"Error formatting answer: {str(e)}"
         
@@ -215,17 +211,12 @@ class BasicAgent:
             # Step 4: Format final answer
             final_answer = self.format_answer(question, thoughts)
             
-            # Format the response according to Gaia requirements
-            formatted_response = "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:\nFINAL ANSWER: [YOUR FINAL ANSWER]\n\n"
-            formatted_response += "\n\n".join(thoughts)
-            formatted_response += f"\n\nFINAL ANSWER: {final_answer}"
-            
-            print(f"Agent returning answer with final part: {final_answer}")
+            print(f"Agent returning answer: {final_answer}")
             return final_answer
         except Exception as e:
             print(f"Error running agent workflow: {e}")
-            # Fallback response
-            return "An error occurred while processing the question. FINAL ANSWER: Error"
+            # Fallback response - return just the error message without the prefix
+            return "Error"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """