Update app.py

app.py

@@ -36,34 +36,32 @@ class BasicAgent:
         #     [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
         #     llm=self.llm,
         # )
-        self.agent_kwargs = {
-            "template": (
-                "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"
-            ),
-            "input_variables": ["input", "agent_scratchpad"]
-        }
+        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,
-            **self.agent_kwargs,
+            system_prompt=system_prompt,
+            verbose=True,
         )
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")