update

app.py

@@ -218,12 +218,10 @@ class BasicAgent:
                 "PIL",
                 "chess",
                 "img2text",
-                "chess.pgn",
                 "PIL.Image",
                 "bytes",
                 "cv2",
                 "numpy",
-                "chess.engine",
                 "json",
                 "whisper",
                 "openpyxl",
@@ -235,11 +233,10 @@ class BasicAgent:
                     - Writing code to solve problem. 
                     - Browse and search the web to find information.
                     - Solving chess problems. 
-                    - Parsing Wikipedia pages.
+                    - Parsing and extracting data from Wikipedia pages.
                 This agent follows rules below:
-                1. Take the question literally! Do not add any additional information or assumptions.
-                2. `wikipedia` Python library is provided that makes it easy to to interact with Wikipedia pages.
-                3. `pandas` Python package is provided that makes it easy to extract table data from Wikipedia HTML pages.
+                1. `wikipedia` Python library is provided, we should use it to interact with Wikipedia pages.
+                2. `pandas` Python package is provided, we should use it to extract table data from Wikipedia HTML pages.
             """,
             verbosity_level=0,
             max_steps=10,
@@ -255,28 +252,12 @@ class BasicAgent:
                 self.multimodal_agent, 
                 self.code_agent],
             additional_authorized_imports=[
-                "requests",
-                "bs4",
                 "markdownify",
                 "wikipedia",
-                "pandas",
-                "io",
-                "PIL",
-                "chess",
-                "img2text",
-                "chess.pgn",
-                "PIL.Image",
-                "bytes",
-                "cv2",
-                "numpy",
-                "chess.engine",
-                "json",
-                "whisper",
-                "youtube_transcript_api",
-                "openpyxl",
+                "pandas"
             ],
             planning_interval=5,
-            max_steps=15,
+            max_steps=10,
         )
 
     def __call__(self, question: str, question_id: str, file_name: str) -> str:
@@ -287,7 +268,6 @@ class BasicAgent:
                "{question}""{file}"
             Please follow rules below:
                 1. Take the question literally! Do not add any additional information or assumptions.
-                2. `pandas` Python package is provided that makes it easy to extract table data from Wikipedia HTML pages.
         """
         result = self.manager_agent.run(prompt)
         print(f"Agent responded with: {result}")