feat: type cast to dict of SmartDataframe (#532)

pandasai/smart_dataframe/__init__.py

@@ -34,7 +34,7 @@
 from ..helpers.logger import Logger
 from ..helpers.df_config_manager import DfConfigManager
 from ..helpers.from_google_sheets import from_google_sheets
-from typing import List, Union, Optional
+from typing import List, Union, Optional, Literal
 from ..middlewares.base import Middleware
 from ..helpers.df_info import DataFrameType, df_type
 from .abstract_df import DataframeAbstract
@@ -248,6 +248,16 @@ def validate(self, schema: pydantic.BaseModel):
         df_validator = DfValidator(self.original_import)
         return df_validator.validate(schema)
 
+    def to_dict(
+        self,
+        orient: Literal[
+            "dict", "list", "series", "split", "tight", "records", "index"
+        ] = "dict",
+        into: type[dict] = dict,
+        index: bool = True,
+    ) -> dict | list[dict]:
+        pass
+
     @property
     def datalake(self):
         return self._dl

pandasai/smart_dataframe/abstract_df.py

@@ -1,3 +1,6 @@
+from typing import overload
+
+
 class DataframeAbstract:
     # Columns
     @property
@@ -152,6 +155,7 @@ def to_json(self, path):
     def to_sql(self, name, con):
         raise NotImplementedError
 
+    @overload
     def to_dict(self, orient):
         raise NotImplementedError