enkryptai-sdk 1.0.1__tar.gz → 1.0.3__tar.gz

{enkryptai_sdk-1.0.1/src/enkryptai_sdk.egg-info → enkryptai_sdk-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.1
+Version: 1.0.3
 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 → enkryptai_sdk-1.0.3}/README.md

@@ -1,3 +1,4 @@
+![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 → enkryptai_sdk-1.0.3}/setup.py

@@ -8,7 +8,7 @@ with open(os.path.join(here, "README.md"), encoding="utf-8") as fh:
 
 setup(
     name="enkryptai-sdk",  # This is the name of your package on PyPI
-    version="1.0.1", # Update this for new versions
+    version="1.0.3", # Update this for new versions
     description="A Python SDK with guardrails and red teaming functionality for API interactions",
     long_description=long_description,
     long_description_content_type="text/markdown",

{enkryptai_sdk-1.0.1 → enkryptai_sdk-1.0.3}/src/enkryptai_sdk/ai_proxy.py

@@ -46,24 +46,43 @@ 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 json
+                import ast
+                # import json
                 try:
-                    error_message = json.loads(error_message.replace("'", '"'))
-                    response["error"] = error_message.get("error", error_message)
-                except json.JSONDecodeError:
+                    # # 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:
+                except Exception:
                     # 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):
+                except Exception:
+                    # 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-1.0.1 → enkryptai_sdk-1.0.3}/src/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 → enkryptai_sdk-1.0.3/src/enkryptai_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.1
+Version: 1.0.3
 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 → enkryptai_sdk-1.0.3}/tests/test_all.py

@@ -784,6 +784,7 @@ def test_add_dataset_success(dataset_client, sample_dataset_config):
     assert hasattr(response, "message")
     response.message == "Dataset task has been added successfully"
     # Sleep for a while to let the task complete
+    # This is also useful to avoid rate limiting issues
     import time
     print("\nSleeping for 60 seconds to let the task complete if possible ...")
     time.sleep(60)
@@ -926,6 +927,7 @@ def test_add_task_with_target_model(redteam_client, sample_redteam_target_config
     assert hasattr(response, "message")
     response.message == "Redteam task has been added successfully"
     # Sleep for a while to let the task complete
+    # This is also useful to avoid rate limiting issues
     import time
     print("\nSleeping for 60 seconds to let the task complete if possible ...")
     time.sleep(60)
@@ -940,6 +942,7 @@ def test_add_task_with_saved_model(redteam_client, sample_redteam_model_config):
     assert hasattr(response, "message")
     response.message == "Redteam task has been added successfully"
     # Sleep for a while to let the task complete
+    # This is also useful to avoid rate limiting issues
     import time
     print("\nSleeping for 60 seconds to let the task complete if possible ...")
     time.sleep(60)

{enkryptai_sdk-1.0.1 → enkryptai_sdk-1.0.3}/tests/test_datasets.py

@@ -77,6 +77,7 @@ def test_add_dataset_success(dataset_client, sample_dataset_config):
     assert hasattr(response, "message")
     response.message == "Dataset task has been added successfully"
     # Sleep for a while to let the task complete
+    # This is also useful to avoid rate limiting issues
     import time
     print("\nSleeping for 60 seconds to let the task complete if possible ...")
     time.sleep(60)

{enkryptai_sdk-1.0.1 → enkryptai_sdk-1.0.3}/tests/test_redteam.py

@@ -194,6 +194,7 @@ def test_add_task_with_target_model(redteam_client, sample_redteam_target_config
     assert hasattr(response, "message")
     response.message == "Redteam task has been added successfully"
     # Sleep for a while to let the task complete
+    # This is also useful to avoid rate limiting issues
     import time
     print("\nSleeping for 60 seconds to let the task complete if possible ...")
     time.sleep(60)
@@ -208,6 +209,7 @@ def test_add_task_with_saved_model(redteam_client, sample_redteam_model_config):
     assert hasattr(response, "message")
     response.message == "Redteam task has been added successfully"
     # Sleep for a while to let the task complete
+    # This is also useful to avoid rate limiting issues
     import time
     print("\nSleeping for 60 seconds to let the task complete if possible ...")
     time.sleep(60)