Fix: Correct type annotations in resource clients

Author: Deepthi-Chand

dataspace_sdk/resources/aimodels.py

@@ -156,7 +156,7 @@ def list_all(
         model_type: Optional[str] = None,
         limit: int = 10,
         offset: int = 0,
-    ) -> Dict[str, Any]:
+    ) -> Any:
         """
         List all AI models with pagination using GraphQL.
 
@@ -224,15 +224,15 @@ def list_all(
             raise DataSpaceAPIError(f"GraphQL error: {response['errors']}")
 
         data = response.get("data", {})
-        models_result: Dict[str, Any] = data.get("aiModels", []) if isinstance(data, dict) else []
+        models_result: Any = data.get("aiModels", []) if isinstance(data, dict) else []
         return models_result
 
     def get_organization_models(
         self,
         organization_id: str,
         limit: int = 10,
         offset: int = 0,
-    ) -> Dict[str, Any]:
+    ) -> Any:
         """
         Get AI models for a specific organization.
 

dataspace_sdk/resources/datasets.py

@@ -132,7 +132,7 @@ def list_all(
         organization_id: Optional[str] = None,
         limit: int = 10,
         offset: int = 0,
-    ) -> Dict[str, Any]:
+    ) -> Any:
         """
         List all datasets with pagination using GraphQL.
 
@@ -194,7 +194,7 @@ def list_all(
             raise DataSpaceAPIError(f"GraphQL error: {response['errors']}")
 
         data = response.get("data", {})
-        datasets_result: Dict[str, Any] = data.get("datasets", []) if isinstance(data, dict) else []
+        datasets_result: Any = data.get("datasets", []) if isinstance(data, dict) else []
         return datasets_result
 
     def get_trending(self, limit: int = 10) -> Dict[str, Any]:
@@ -214,7 +214,7 @@ def get_organization_datasets(
         organization_id: str,
         limit: int = 10,
         offset: int = 0,
-    ) -> Dict[str, Any]:
+    ) -> Any:
         """
         Get datasets for a specific organization.
 

dataspace_sdk/resources/usecases.py

@@ -146,7 +146,7 @@ def list_all(
         organization_id: Optional[str] = None,
         limit: int = 10,
         offset: int = 0,
-    ) -> Dict[str, Any]:
+    ) -> Any:
         """
         List all use cases with pagination using GraphQL.
 
@@ -221,15 +221,15 @@ def list_all(
             raise DataSpaceAPIError(f"GraphQL error: {response['errors']}")
 
         data = response.get("data", {})
-        usecases_result: Dict[str, Any] = data.get("useCases", []) if isinstance(data, dict) else []
+        usecases_result: Any = data.get("useCases", []) if isinstance(data, dict) else []
         return usecases_result
 
     def get_organization_usecases(
         self,
         organization_id: str,
         limit: int = 10,
         offset: int = 0,
-    ) -> Dict[str, Any]:
+    ) -> Any:
         """
         Get use cases for a specific organization.