Update app.py

app.py

@@ -39,12 +39,12 @@ class BasicAgent:
     FINAL ANSWER: [YOUR FINAL ANSWER]"
     """
     
-    def __init__(self, model_name: str = "gpt-4.1", api_key: str = None):
+    def __init__(self, model_name: str = "gpt-3.5-turbo", api_key: str = None):
         """
         Initialize the BasicAgent with the specified LLM.
         
         Args:
-            model_name: The name of the model to use (defaults to gpt-4.1)
+            model_name: The name of the model to use (defaults to gpt-3.5-turbo)
             api_key: OpenAI API key (defaults to environment variable)
         """
         # Use the provided API key or fall back to the environment variable
@@ -221,7 +221,7 @@ class BasicAgent:
             formatted_response += f"\n\nFINAL ANSWER: {final_answer}"
             
             print(f"Agent returning answer with final part: {final_answer}")
-            return formatted_response
+            return final_answer
         except Exception as e:
             print(f"Error running agent workflow: {e}")
             # Fallback response