Update app.py

app.py

@@ -28,51 +28,56 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized. . . .")
-        # self.agent = smart_agent()
-        # self.api_key = os.getenv("HF_TOKEN")
         OpenAI_key = os.getenv("OPEN_AI_TOKEN")
         #self.llm = OpenAI(model="gpt-4o-mini", temperature=0.2, api_key=OpenAI_key)
-        self.llm = OpenAI(model="gpt-4o", temperature=0.1, api_key=OpenAI_key)
         self.system_prompt = (
-            "You are a helpful AI assistant completing GAIA benchmark tasks.\n"
-            "You MUST use the tools provided to answer the user's question. Do not answer from your own knowledge.\n"
-            "Carefully analyze the question to determine the most appropriate tool to use.\n"
-            "Here are guidelines for using the tools:\n"
-            "- Use 'wikipedia_search_tool' to find factual information about topics, events, people, etc. (e.g., 'Use wikipedia_search to find the population of France').\n"
-            "- Use 'youtube_transcript_tool' to extract transcripts from YouTube videos when the question requires understanding the video content. (e.g., 'Use youtube_transcript to summarize the key points of this video').\n"
-            "- Use 'transcribe_audio_tool' to transcribe uploaded audio files. (e.g., 'Use audio_transcriber to get the text from this audio recording').\n"
-            "- Use 'solve_chess_image_tool' to analyze and solve chess puzzles from images. (e.g., 'Use chess_image_solver to determine the best move in this chess position').\n"
-            "- Use 'parse_file_and_summarize_tool' to parse and analyze data from Excel or CSV files. (e.g., 'Use file_parser to calculate the average sales from this data').\n"
-            "- Use 'vegetable_classifier_tool' to classify a list of food items and extract only the vegetables. (e.g., 'Use vegetable_classifier_2022 to get a list of the vegetables in this grocery list').\n"
-            "- Use 'excel_food_sales_sum_tool' to extract total food sales from excel files. (e.g., 'Use excel_food_sales_sum to calculate the total food sales').\n"
-            "- Use 'google_web_search_tool' to find factual information about topics, events, people, from the web if not spificied to be fund in wikipedia etc. (e.g., 'find the population of France').\n"
-            "Do NOT guess or make up answers. If a tool cannot provide the answer, truthfully respond that you were unable to find the information.\n"
-            "Use the tools to research or calculate the answer.\n"
-            "If a tool fails, explain the reason for the failure instead of hallucinating an answer.\n"
-            "Provide concise and direct answers as requested in the questions. Do not add extra information unless explicitly asked for.\n"
-            "For example, if asked for a number, return only the number. If asked for a list, return only the list.\n"
+        "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"
+        "FINAL ANSWER: <answer>\n"
+        "Once you output 'FINAL 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 = (
+        #     "You are a helpful AI assistant completing GAIA benchmark tasks.\n"
+        #     "You MUST use the tools provided to answer the user's question. Do not answer from your own knowledge.\n"
+        #     "Carefully analyze the question to determine the most appropriate tool to use.\n"
+        #     "Here are guidelines for using the tools:\n"
+        #     "- Use 'wikipedia_search_tool' to find factual information about topics, events, people, etc. (e.g., 'Use wikipedia_search to find the population of France').\n"
+        #     "- Use 'youtube_transcript_tool' to extract transcripts from YouTube videos when the question requires understanding the video content. (e.g., 'Use youtube_transcript to summarize the key points of this video').\n"
+        #     "- Use 'transcribe_audio_tool' to transcribe uploaded audio files. (e.g., 'Use audio_transcriber to get the text from this audio recording').\n"
+        #     "- Use 'solve_chess_image_tool' to analyze and solve chess puzzles from images. (e.g., 'Use chess_image_solver to determine the best move in this chess position').\n"
+        #     "- Use 'parse_file_and_summarize_tool' to parse and analyze data from Excel or CSV files. (e.g., 'Use file_parser to calculate the average sales from this data').\n"
+        #     "- Use 'vegetable_classifier_tool' to classify a list of food items and extract only the vegetables. (e.g., 'Use vegetable_classifier_2022 to get a list of the vegetables in this grocery list').\n"
+        #     "- Use 'excel_food_sales_sum_tool' to extract total food sales from excel files. (e.g., 'Use excel_food_sales_sum to calculate the total food sales').\n"
+        #     "- Use 'google_web_search_tool' to find factual information about topics, events, people, from the web if not spificied to be fund in wikipedia etc. (e.g., 'find the population of France').\n"
+        #     "Do NOT guess or make up answers. If a tool cannot provide the answer, truthfully respond that you were unable to find the information.\n"
+        #     "Use the tools to research or calculate the answer.\n"
+        #     "If a tool fails, explain the reason for the failure instead of hallucinating an answer.\n"
+        #     "Provide concise and direct answers as requested in the questions. Do not add extra information unless explicitly asked for.\n"
+        #     "For example, if asked for a number, return only the number. If asked for a list, return only the list.\n"
+        # )
         self.agent = AgentWorkflow.from_tools_or_functions(
             [
                 wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool,transcribe_audio_tool, excel_food_sales_sum_tool, parse_file_and_summarize_tool, solve_chess_image_tool, vegetable_classifier_tool
              ],
             llm=self.llm,
             system_prompt=self.system_prompt,
-            verbose=True,
         )
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # answer = await self.agent.run(question)
         answer = await self.agent.run(
             f"{question}\n\nIf you have enough information, respond with a concise final answer.",
-            max_iterations=50
+            max_iterations=10
         )
-        if hasattr(answer, "output"):
-            print(f"Agent returning answer: {answer}")
-            return str(answer.output)
-        else:
-            print(f"Agent returning answer: {answer}")
-            return str(answer)
+        return str(answer)
+        # if hasattr(answer, "output"):
+        #     print(f"Agent returning answer: {answer}")
+        #     return str(answer.output)
+        # else:
+        #     print(f"Agent returning answer: {answer}")
+        #     return str(answer)
         
     def __call__(self, question: str) -> str:
         return asyncio.run(self.run(question))