use gpt model and update prompt

app.py

@@ -1,9 +1,8 @@
 import os
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, OpenAIServerModel
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
@@ -11,7 +10,6 @@ from tools.web_search import DuckDuckGoSearchTool
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-agent_description = "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. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the 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."
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -21,25 +19,34 @@ class BasicAgent:
         final_answer = FinalAnswerTool()
         visit_webpage = VisitWebpageTool()
         web_search = DuckDuckGoSearchTool()
-        model = HfApiModel(
-            max_tokens=2096,
-            temperature=0.5,
-            model_id='Qwen/Qwen2.5-Coder-32B-Instruct', custom_role_conversions=None,
-        )
+        model = OpenAIServerModel(model_id="gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
         self.agent = CodeAgent(
             model=model,
             tools=[visit_webpage, web_search, final_answer],
-            max_steps=10,
+            max_steps=3,
             verbosity_level=1,
-            grammar=None,
-            planning_interval=None,
-            name=None,
-            description=agent_description,
             add_base_tools=True
         )
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question}")
-        answer = self.agent.run(question)
+        prompt = (f"""
+            You are a general AI assistant. I will ask you a question.
+            Report your thoughts, and finish your answer with the following template: [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.
+            If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+            If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the 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.
+            IMPORTANT NOTES TO LIMIT COSTS AND PREVENT ERRORS:
+            - Use web search sparingly and only when absolutely necessary.
+            - Limit to 1-2 web searches per question.
+            - If the search fails due to rate limiting, add a 3-5 second delay using time.sleep() before retrying with a different search term.
+            - Do not import libraries that aren't available - stick to basic Python and the tools provided.
+            - Focus on answering directly with what you already know when possible.
+            - If you have made more than 3 attempts to solve a problem, prioritize providing your best guess.
+            - Always add a delay of 2-3 seconds between web searches using time.sleep() to avoid rate limiting.
+            Remember to structure your response in Python code format using the final_answer() function.
+            The question is: {question}""")
+        answer = self.agent.run(prompt)
         print(f"Agent returning answer: {answer}")
         return answer