Return df to keep consistency

Author: ZoranPandovski

mindsdb_sdk/connectors/rest_api.py

@@ -136,7 +136,7 @@ def objects_tree(self, item='', with_schemas=False):
         r = self.session.get(self.url + f'/api/tree/{item}', params=params)
         _raise_for_status(r)
 
-        return r.json()  # Return raw JSON instead of DataFrame for more flexible processing
+        return pd.DataFrame(r.json())
 
     @staticmethod
     def read_file_as_bytes(file_path: str):

mindsdb_sdk/databases.py

@@ -80,8 +80,8 @@ def tree(self, with_schemas: bool = False) -> List[TreeNode]:
         ...     else:
         ...         print(f"Table: {item.name}, Type: {item.type}")
         """
-        tree_data = self.api.objects_tree(self.name, with_schemas=with_schemas)
-        return [TreeNode.from_dict(item) for item in tree_data]
+        df = self.api.objects_tree(self.name, with_schemas=with_schemas)
+        return [TreeNode.from_dict(row.to_dict()) for _, row in df.iterrows()]
 
 
 class Databases(CollectionBase):

mindsdb_sdk/server.py

@@ -82,8 +82,8 @@ def tree(self) -> List[TreeNode]:
         >>> for db in tree:
         ...     print(f"Database: {db.name}, Type: {db.type}, Engine: {db.engine}")
         """
-        tree_data = self.api.objects_tree('')
-        return [TreeNode.from_dict(item) for item in tree_data]
+        df = self.api.objects_tree('')
+        return [TreeNode.from_dict(row.to_dict()) for _, row in df.iterrows()]
 
     def __repr__(self):
         return f'{self.__class__.__name__}({self.api.url})'