Update app.py

app.py

@@ -14,7 +14,8 @@ from youtube_tool import youtube_transcript_tool, youtube_transcript_snippet_too
 #from multiple_tools import round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool, wikipedia_search_tool
 from multiple_tools import round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool, wikipedia_search_tool, transcribe_audio_tool, excel_food_sales_sum_tool, parse_file_and_summarize_tool, solve_chess_image_tool, vegetable_classifier_tool
 from agent import smart_agent
-from llama_index.llms.openai import OpenAI
+from llama_index.llms.openai import 
+import re
 #-----------------------------------------------------------------
 
 
@@ -33,7 +34,9 @@ class BasicAgent:
         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 and stop reasoning and do not call any tool.:\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.llm = OpenAI(model="gpt-4o", temperature=0.1, api_key=OpenAI_key)
         # self.system_prompt = (
@@ -62,6 +65,10 @@ class BasicAgent:
             llm=self.llm,
             system_prompt=self.system_prompt,
         )
+    def extract_answer(text: str) -> str:
+        match = re.search(r"(?<=<answer>)(.*?)(?=</answer>)", text)
+        return match.group(1) if match else ""
+    
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # answer = await self.agent.run(question)
@@ -69,7 +76,8 @@ class BasicAgent:
             f"{question}\n\nIf you have enough information, respond with a concise final answer.",
             max_iterations=10
         )
-        return str(answer)
+        #return str(answer)
+        return self.extract_answer(answer);
         # if hasattr(answer, "output"):
         #     print(f"Agent returning answer: {answer}")
         #     return str(answer.output)