Update app.py

app.py

@@ -122,7 +122,10 @@ def wikipedia_tool(query: str) -> str:
         return formatted_search_docs
     except Exception as e:
         return f"Wikipedia search error: {e}"
+    
 
+    
+    
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 
@@ -132,20 +135,11 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         token=os.environ["OPENROUTER_API_KEY"]  
-        self.system_prompt= """You are a helpful assistant.
-        You will answer questions based on the provided context.
-        You will always return a valid answer, even if the question is not clear or the context is insufficient. 
-        Always return a valid answer after validating the source.
-        
-        Your final answer should be in the following format:
-        "your answer here".
-        
-        If you are asked question related to numbers,return the answer in numeric format.
-        If you want more details about a topic,person, you can use the wikipedia tool.
-        If you want to perform web search to find the answer, use the web_search tool.
-        If you are given a task to create an image, you can use the image_generation_tool.
-        
-        Answer only what is asked, do not add any additional information.
+        self.system_prompt= """You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, 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.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
         
         """
         
@@ -155,7 +149,7 @@ class BasicAgent:
             provider="auto"  # Use openrouter as the provider
 )
         self.agent= CodeAgent(
-            tools = [add, subtract, multiply, divide, web_search, image_generation_tool,wikipedia_tool],
+            tools = [add, subtract, multiply, divide, web_search, image_generation_tool],
             model=model,
         )
     def __call__(self, question: str, context: str = "") -> str: