Update app.py

app.py

@@ -33,12 +33,27 @@ class BasicAgent:
         print("BasicAgent initialized. . . .")
         
         #self.llm = OpenAI(model="gpt-4o-mini", temperature=0.2, api_key=OpenAI_key)
+        # self.system_prompt = (
+        # "You are a helpful AI assistant completing GAIA benchmark tasks.\n"
+        # "You MUST use the tools provided when needed.\n"
+        # "If you already have enough information, respond directly with:\n"
+        # "<answer>\n"
+        # "Once you output '<answer>', stop reasoning and do not call any tool.\n"
+        # )
         self.system_prompt = (
-        "You are a helpful AI assistant completing GAIA benchmark tasks.\n"
-        "You MUST use the tools provided when needed.\n"
-        "If you already have enough information, respond directly with:\n"
-        "<answer>\n"
-        "Once you output '<answer>', stop reasoning and do not call any tool.\n"
+        "You are a helpful assistant tasked with answering questions using a set of tools.\n"
+        "Your final answer must strictly follow this format:\n"
+        "FINAL ANSWER: [ANSWER]\n"
+        "Only write the answer in that exact format. Do not explain anything. Do not include any other text. \n"
+        "If you are provided with a similar question and its final answer, and the current question is **exactly the same**, then simply return the same final answer without using any tools. \n"
+        "Only use tools if the current question is different from the similar one. \n"
+        "Examples: \n"
+        "- FINAL ANSWER: FunkMonk \n"
+        "- FINAL ANSWER: Paris \n"
+        "- FINAL ANSWER: 128 \n"
+        " \n"
+        "Once you output 'FINAL ANSWER', stop reasoning and do not call any tool.\n"
+        "If you do not follow this format exactly, your response will be considered incorrect. \n"
         )
         self.llm = HuggingFaceInferenceAPI(
             model_name="deepseek-ai/DeepSeek-R1-0528",