Update app.py

app.py

@@ -39,6 +39,30 @@ class BasicAgent:
         #     llm=self.llm
         # )
         OpenAI_key = os.getenv("OPEN_AI_TOKEN")
+        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' to find factual information about topics, events, people, etc. (e.g., 'Use wikipedia_search to find the population of France').\n"
+                "- Use 'youtube_transcript' 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 'audio_transcriber' to transcribe uploaded audio files. (e.g., 'Use audio_transcriber to get the text from this audio recording').\n"
+                "- Use 'chess_image_solver' 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 'file_parser' 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_2022' 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' to extract total food sales from excel files. (e.g., 'Use excel_food_sales_sum to calculate the total food sales').\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"
+            ),
+            "max_iterations": 100,
+            "verbose": false
+        }
         self.llm = OpenAI(model="gpt-4o-mini", temperature=0.2, api_key=OpenAI_key)
         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],