enkryptai-sdk 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

enkryptai_sdk/ai_proxy.py

@@ -46,24 +46,41 @@ class AIProxyClient(BaseClient):
             "POST", "/ai-proxy/chat/completions", headers=headers, json=payload
         )
 
+        # print("Response from API: ", response)
+
         if response.get("error"):
             error_message = response["error"]
             is_json_str = error_message.startswith("{") and error_message.endswith("}")
             # Try to parse nested JSON error if it's a string
             if isinstance(error_message, str) and is_json_str:
+                import ast
                 import json
                 try:
-                    error_message = json.loads(error_message.replace("'", '"'))
-                    response["error"] = error_message.get("error", error_message)
+                    # # Using ast.literal_eval to safely evaluate the string as a Python literal
+                    # # As json.loads is not working with literal string representation of dict
+                    # parsed_error = json.loads(error_message.replace("'", '"'))
+                    # # Convert Python string representation to proper dict
+                    parsed_error = ast.literal_eval(response["error"])
+                    # # Preserve both error and enkrypt_policy_detections fields
+                    response["error"] = parsed_error.get("error", parsed_error)
+                    if "enkrypt_policy_detections" in parsed_error:
+                        response["enkrypt_policy_detections"] = parsed_error["enkrypt_policy_detections"]
                 except json.JSONDecodeError:
                     # If parsing fails, keep the original error
                     pass
                     
             # print("Error in response: ", response)
             if return_error:
-                if is_json_str:
+                # if is_json_str:
+                #     return ChatCompletionErrorResponse.from_dict(response)
+                # else:
+                #     return ChatCompletionDirectErrorResponse.from_dict(response)
+                try:
+                    # print("Error in response: ", response)
                     return ChatCompletionErrorResponse.from_dict(response)
-                else:
+                except (json.JSONDecodeError, TypeError, ValueError, SyntaxError):
+                    # Fallback to direct error if error object can't be parsed
+                    print("Failed to parse error response: ", response)
                     return ChatCompletionDirectErrorResponse.from_dict(response)
             raise AIProxyClientError(response["error"])
 

enkryptai_sdk/dto/ai_proxy.py

@@ -262,6 +262,7 @@ class ChatCompletionError(BaseDTO):
 @dataclass
 class ChatCompletionErrorResponse(BaseDTO):
     error: ChatCompletionError
+    enkrypt_policy_detections: Dict[str, Any] = field(default_factory=dict)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
     @classmethod
@@ -270,16 +271,18 @@ class ChatCompletionErrorResponse(BaseDTO):
         
         # Create a copy of the data without the fields we're explicitly processing
         extra_fields = {k: v for k, v in data.items() 
-                       if k not in ["error"]}
+                       if k not in ["error", "enkrypt_policy_detections"]}
         
         return cls(
             error=ChatCompletionError.from_dict(error_data),
+            enkrypt_policy_detections=data.get("enkrypt_policy_detections", {}),
             _extra_fields=extra_fields
         )
     
     def to_dict(self) -> Dict[str, Any]:
         result = {
-            "error": self.error.to_dict()
+            "error": self.error.to_dict(),
+            "enkrypt_policy_detections": self.enkrypt_policy_detections
         }
         
         # Add any extra fields

{enkryptai_sdk-1.0.1.dist-info → enkryptai_sdk-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.1
+Version: 1.0.2
 Summary: A Python SDK with guardrails and red teaming functionality for API interactions
 Home-page: https://github.com/enkryptai/enkryptai-sdk
 Author: Enkrypt AI Team
@@ -21,6 +21,7 @@ Dynamic: license-file
 Dynamic: requires-python
 Dynamic: summary
 
+![Python SDK test](https://github.com/enkryptai/enkryptai-sdk/actions/workflows/test.yaml/badge.svg)
 # Enkrypt AI Python SDK
 
 A Python SDK with Guardrails, Models, Deployments, AI Proxy, Datasets and Red Team functionality for API interactions.

{enkryptai_sdk-1.0.1.dist-info → enkryptai_sdk-1.0.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 enkryptai_sdk/__init__.py,sha256=rP6PtntJogJauj1lKWK8DkiBr3uYjireIUamr6aflu0,763
-enkryptai_sdk/ai_proxy.py,sha256=nwndIHNtCdvCOJwhkTOxtDHocbUta4cwtF1NrLkVg_I,2572
+enkryptai_sdk/ai_proxy.py,sha256=QnfRlyKKjUfDuJ19rTM8dtbWvBaHgLyb2Z9AOjK2PFA,3723
 enkryptai_sdk/base.py,sha256=MlEDcEIjXo35kat9XkGUu7VB2fIvJk38C94wAeO9bEw,1304
 enkryptai_sdk/config.py,sha256=IpB8_aO4zXdvv061v24oh83oyJ5Tp1QBQTzeuW4h9QY,8828
 enkryptai_sdk/datasets.py,sha256=xekcdY9wIniw2TnylaK6o1RNZ5DRoluNOGawBkVgaM0,4881
@@ -11,15 +11,15 @@ enkryptai_sdk/models.py,sha256=3y6-7vaDrEmDZpc3zHmIsBs7DFjDprf_EA7PU2yPU0c,8976
 enkryptai_sdk/red_team.py,sha256=DbJRpWwZBHljR3o247PvqCzJm3_TV9UXaxISeUmEJ5w,14188
 enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
 enkryptai_sdk/dto/__init__.py,sha256=kKBw4rkfqMBuK8nXRDtD6Sd0_uqLKgbcHrqzuSGJpr0,2310
-enkryptai_sdk/dto/ai_proxy.py,sha256=1Y5B0O3ooS4PNnvDkzf4e0zGyr8HoWeN5am3fg5sjrA,11024
+enkryptai_sdk/dto/ai_proxy.py,sha256=clwMN4xdH8Zr55dnhilHbs-qaHRlCOrLPrij0Zd1Av0,11283
 enkryptai_sdk/dto/base.py,sha256=6VWTkoNZ7uILqn_iYsPS21cVa2xLYpw5bjDIsRCS5tk,2389
 enkryptai_sdk/dto/datasets.py,sha256=E3hvHvGZ94iMvCslTcYM3VCKszVQq_xtu93nlm4dZhI,4444
 enkryptai_sdk/dto/deployments.py,sha256=lsKdG09C-rceIjGvEyYOBf5zBjrk7ma8NpPfgrAgdfM,10829
 enkryptai_sdk/dto/guardrails.py,sha256=XMFco-KlEqI4TYoJAyxxTrk-OFixtEcftBpG1SXFH4k,39583
 enkryptai_sdk/dto/models.py,sha256=O4gVhVTenlsytNJIvk2gO5530KZWMye6FCVCtF5IW-A,11700
 enkryptai_sdk/dto/red_team.py,sha256=BoOPYFjIONIC0XPuyJtkx_qLVYi2kLGEA4CzlySgbJA,13829
-enkryptai_sdk-1.0.1.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.1.dist-info/METADATA,sha256=jwRXWQw3EbYvzN3lmiPUcua-5m1dqPNT-Wi4uvllYEk,43106
-enkryptai_sdk-1.0.1.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-enkryptai_sdk-1.0.1.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.1.dist-info/RECORD,,
+enkryptai_sdk-1.0.2.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.2.dist-info/METADATA,sha256=erxYdcKCSpIFnjCvMQWFRETJk8xa7ErFuMlqv9dPA6I,43207
+enkryptai_sdk-1.0.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+enkryptai_sdk-1.0.2.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.2.dist-info/RECORD,,

{enkryptai_sdk-1.0.1.dist-info → enkryptai_sdk-1.0.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.3)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any