Add pandas import to BasicAgent and change ReadExcelFileBytes output type to object

agent/agents.py

@@ -82,6 +82,7 @@ class BasicAgent:
                 wiki_retriever,
             ],
             model=model,
+            additional_authorized_imports=["pandas"],
             step_callbacks=STEP_CALLBACKS,
             max_steps=15,
         )

agent/tools.py

@@ -229,7 +229,7 @@ class ReadExcelFileBytes(Tool):
             "description": "the bytes content of the Excel file",
         },
     }
-    output_type = "string"
+    output_type = "object"  # pandas DataFrame
 
     def forward(self, excel_bytes: str):
         from io import BytesIO
@@ -237,4 +237,5 @@ class ReadExcelFileBytes(Tool):
         import pandas as pd
 
         df = pd.read_excel(BytesIO(excel_bytes))
-        return df.to_string()
+        print(f"Read {len(df)} rows from the Excel file as pandas DataFrame.")
+        return df