Update BasicAgent.py

BasicAgent.py

@@ -1,4 +1,6 @@
-import smolagents, numpy, math, xlrd, os, pandas
+import smolagents, numpy, math, xlrd, os
+import pandas as pd
+from typing import Union
 from smolagents import (
     tool,
     CodeAgent,
@@ -13,11 +15,15 @@ from smolagents import (
 
 #*
 @tool
-def read_excel(file_path: str, sheet_name=0):
+def read_excel(file_path: str, sheet_name: Union[str, int] = 0) -> Union[str, pd.DataFrame]:
     """
     Read an Excel file into a pandas DataFrame.
+    Args:
+        file_path (str): Path to the Excel file.
+        sheet_name (Union[str, int]): Sheet name or index. Defaults to 0.
+    Returns:
+        DataFrame or error message string.
     """
-    import pandas as pd
     try:
         return pd.read_excel(file_path, sheet_name=sheet_name)
     except Exception as e: