Update BasicAgent.py

BasicAgent.py

@@ -15,17 +15,15 @@ from smolagents import (
 
 #*
 @tool
-def read_excel(file_path: str, sheet_name: Union[str, int] = 0) -> Union[str, pd.DataFrame]:
+def read_excel(file_path: str, sheet_name: str = "0") -> str:
     """
-    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.
+    Read an Excel file and return it as JSON-like string.
     """
     try:
-        return pd.read_excel(file_path, sheet_name=sheet_name)
+        # Try to convert sheet_name to int if it's a number
+        sheet = int(sheet_name) if sheet_name.isdigit() else sheet_name
+        df = pd.read_excel(file_path, sheet_name=sheet)
+        return df.to_json(orient="records")
     except Exception as e:
         return f"Error reading Excel file: {e}"
 #