Update app.py

app.py

@@ -36,11 +36,36 @@ 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 = {
+            "prefix": (
+                "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"
+            ),
+            "suffix": (
+                "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"]
+        }
         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
+            llm=self.llm,
+            **self.agent_kwargs,
         )
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")