rapidata 2.21.1__py3-none-any.whl → 2.21.2__py3-none-any.whl

rapidata/rapidata_client/api/rapidata_exception.py

@@ -0,0 +1,105 @@
+from typing import Optional, Any
+from rapidata.api_client.api_client import ApiClient, rest, ApiResponse, ApiResponseT
+from rapidata.api_client.exceptions import ApiException
+import json
+
+
+class RapidataError(Exception):
+    """Custom error class for Rapidata API errors."""
+    
+    def __init__(
+        self, 
+        status_code: Optional[int] = None, 
+        message: str | None = None, 
+        original_exception: Exception | None = None,
+        details: Any = None
+    ):
+        self.status_code = status_code
+        self.message = message
+        self.original_exception = original_exception
+        self.details = details
+        
+        # Create a nice error message
+        error_msg = "Rapidata API Error"
+        if status_code:
+            error_msg += f" ({status_code})"
+        if message:
+            error_msg += f": {message}"
+            
+        super().__init__(error_msg)
+
+    def __str__(self):
+        """Return a string representation of the error."""        
+        # Extract information from message if available
+        title = None
+        errors = None
+        trace_id = None
+        
+        # Try to extract from details if available and is a dict
+        if self.details and isinstance(self.details, dict):
+            title = self.details.get('title')
+            errors = self.details.get('errors')
+            trace_id = self.details.get('traceId')
+        
+        # Build the error string
+        error_parts = []
+        
+        # Main error line
+        if title:
+            error_parts.append(f"{title}")
+        else:
+            error_parts.append(f"{self.message or 'Unknown error'}")
+        
+        # Reasons
+        if errors:
+            if isinstance(errors, dict):
+                error_parts.append(f"Reasons: {json.dumps({'errors': errors})}")
+            else:
+                error_parts.append(f"Reasons: {errors}")
+        
+        # Trace ID
+        if trace_id:
+            error_parts.append(f"Trace Id: {trace_id}")
+        else:
+            error_parts.append("Trace Id: N/A")
+        
+        return "\n".join(error_parts)
+
+class RapidataApiClient(ApiClient):
+    """Custom API client that wraps errors in RapidataError."""
+
+    def response_deserialize(
+        self,
+        response_data: rest.RESTResponse,
+        response_types_map: Optional[dict[str, ApiResponseT]] = None
+    ) -> ApiResponse[ApiResponseT]:
+        """Override the response_deserialize method to catch and convert exceptions."""
+        try:
+            return super().response_deserialize(response_data, response_types_map)
+        except ApiException as e:
+            status_code = getattr(e, 'status', None)
+            message = str(e)
+            details = None
+            
+            # Extract more detailed error message from response body if available
+            if hasattr(e, 'body') and e.body:
+                try:
+                    body_json = json.loads(e.body)
+                    if isinstance(body_json, dict):
+                        if 'message' in body_json:
+                            message = body_json['message']
+                        elif 'error' in body_json:
+                            message = body_json['error']
+                        
+                        # Store the full error details for debugging
+                        details = body_json
+                except (json.JSONDecodeError, AttributeError):
+                    # If we can't parse the body as JSON, use the original message
+                    pass
+            
+            raise RapidataError(
+                status_code=status_code, 
+                message=message, 
+                original_exception=e,
+                details=details
+            ) from None

rapidata/rapidata_client/validation/rapids/rapids_manager.py

@@ -43,6 +43,9 @@ class RapidsManager:
         else:
             raise ValueError(f"Unsupported data type: {data_type}")
 
+        if not isinstance(truths, list):
+            raise ValueError("Truths must be a list of strings")
+        
         if not all(truth in answer_options for truth in truths):
             raise ValueError("Truths must be part of the answer options")
 

rapidata/service/openapi_service.py

@@ -8,9 +8,9 @@ from rapidata.api_client.api.pipeline_api import PipelineApi
 from rapidata.api_client.api.rapid_api import RapidApi
 from rapidata.api_client.api.validation_api import ValidationApi
 from rapidata.api_client.api.workflow_api import WorkflowApi
-from rapidata.api_client.api_client import ApiClient
 from rapidata.api_client.configuration import Configuration
 from rapidata.service.credential_manager import CredentialManager
+from rapidata.rapidata_client.api.rapidata_exception import RapidataApiClient
 
 
 class OpenAPIService:
@@ -36,7 +36,7 @@ class OpenAPIService:
         )
 
         client_configuration = Configuration(host=endpoint, ssl_ca_cert=cert_path)
-        self.api_client = ApiClient(
+        self.api_client = RapidataApiClient(
             configuration=client_configuration,
             header_name="X-Client",
             header_value=f"RapidataPythonSDK/{self._get_rapidata_package_version()}",

{rapidata-2.21.1.dist-info → rapidata-2.21.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rapidata
-Version: 2.21.1
+Version: 2.21.2
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG

{rapidata-2.21.1.dist-info → rapidata-2.21.2.dist-info}/RECORD

@@ -415,6 +415,8 @@ rapidata/api_client/models/workflow_state.py,sha256=5LAK1se76RCoozeVB6oxMPb8p_5b
 rapidata/api_client/rest.py,sha256=WTkaOPZhB24TG2mV7Ih5Km76lo2ySQXFjR98nyFIGIM,9013
 rapidata/api_client_README.md,sha256=XxKjqTKsOlLbAyeUNmJWnQU2Kc7LYOsJLUZ0Q_SEIIo,55005
 rapidata/rapidata_client/__init__.py,sha256=XP9btoeEBcUfLP_4Hi-tqmIsy__L7Q0l4LY1GRQZSKk,974
+rapidata/rapidata_client/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rapidata/rapidata_client/api/rapidata_exception.py,sha256=pJDMX5Pk1SSkxyt6PE0MCP2PQ13374qsi9SBWJYXbJI,3726
 rapidata/rapidata_client/assets/__init__.py,sha256=hKgrOSn8gJcBSULaf4auYhH1S1N5AfcwIhBSq1BOKwQ,323
 rapidata/rapidata_client/assets/_base_asset.py,sha256=B2YWH1NgaeYUYHDW3OPpHM_bqawHbH4EjnRCE2BYwiM,298
 rapidata/rapidata_client/assets/_media_asset.py,sha256=9IKNKWarBJ-aAxfTjh80ScNsHlWGJnd55fsDbrf8x4s,10336
@@ -485,7 +487,7 @@ rapidata/rapidata_client/validation/rapidata_validation_set.py,sha256=GaatGGuJCH
 rapidata/rapidata_client/validation/rapids/__init__.py,sha256=WU5PPwtTJlte6U90MDakzx4I8Y0laj7siw9teeXj5R0,21
 rapidata/rapidata_client/validation/rapids/box.py,sha256=t3_Kn6doKXdnJdtbwefXnYKPiTKHneJl9E2inkDSqL8,589
 rapidata/rapidata_client/validation/rapids/rapids.py,sha256=uCKnoSn1RykNHgTFbrvCFlfzU8lF42cff-2I-Pd48w0,4620
-rapidata/rapidata_client/validation/rapids/rapids_manager.py,sha256=F00lPYBUx5fTPRw50iZuobtdbjFo6ZHevPMk101JdaY,14271
+rapidata/rapidata_client/validation/rapids/rapids_manager.py,sha256=s5VAq8H5CKACWfmIQuz9kHC8t2nd-xEHGGUj9pIfXKI,14386
 rapidata/rapidata_client/validation/validation_set_manager.py,sha256=GCFDyruCz6EIN67cc2fwRjzirUoGfs3KLTIg7n6wXbI,26722
 rapidata/rapidata_client/workflow/__init__.py,sha256=7nXcY91xkxjHudBc9H0fP35eBBtgwHGWTQKbb-M4h7Y,477
 rapidata/rapidata_client/workflow/_base_workflow.py,sha256=XyIZFKS_RxAuwIHS848S3AyLEHqd07oTD_5jm2oUbsw,762
@@ -501,8 +503,8 @@ rapidata/rapidata_client/workflow/_timestamp_workflow.py,sha256=tPi2zu1-SlE_ppbG
 rapidata/service/__init__.py,sha256=s9bS1AJZaWIhLtJX_ZA40_CK39rAAkwdAmymTMbeWl4,68
 rapidata/service/credential_manager.py,sha256=3x-Fb6tyqmgtpjI1MSOtXWW_SkzTK8Lo7I0SSL2YD7E,8602
 rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5AnfY14BW8,877
-rapidata/service/openapi_service.py,sha256=ORFPfHlb41zOUP5nDjYWZwO-ZcqNF_Mw2r71RitFtS0,4042
-rapidata-2.21.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.21.1.dist-info/METADATA,sha256=btZoH5tXjLg2m2xzVxGRTkbtYrRzNOX3y5tjW_hK1vU,1227
-rapidata-2.21.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-rapidata-2.21.1.dist-info/RECORD,,
+rapidata/service/openapi_service.py,sha256=nh7gAmNRdZ5qu3sG7khi-pZmXhfg5u8KHdmEDQd_z9U,4075
+rapidata-2.21.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.21.2.dist-info/METADATA,sha256=npRKolQ9iiiUYn9FbocwGWJCWYzjMTS9DJJOTLNR1Oc,1227
+rapidata-2.21.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+rapidata-2.21.2.dist-info/RECORD,,