Update agent.py

agent.py

@@ -4,6 +4,7 @@ import os
 import re
 import pandas as pd
 from typing import Optional
+
 from token_bucket import Limiter, MemoryStorage
 import yaml
 from PIL import Image, ImageOps
@@ -207,6 +208,7 @@ class MagAgent:
             "bs4",
             "arxiv",
             "whisper",
+            "io"
         ]
 
         self.tools = [