enkryptai-sdk 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

enkryptai_sdk/guardrails.py

@@ -57,7 +57,7 @@ class GuardrailsClient:
         """
         return self._request("GET", "/guardrails/models")
 
-    def detect(self, text, guardrails_config):
+    def detect(self, text, config):
         """
         Detects prompt injection, toxicity, NSFW content, PII, hallucination, and more.
 
@@ -70,12 +70,12 @@ class GuardrailsClient:
         - JSON response from the API.
         """
         # Allow passing in either a dict or a GuardrailsConfig instance.
-        if hasattr(guardrails_config, "as_dict"):
-            guardrails_config = guardrails_config.as_dict()
+        if hasattr(config, "as_dict"):
+            config = config.as_dict()
             
         payload = {
             "text": text,
-            "detectors": guardrails_config
+            "detectors": config
         }
         return self._request("POST", "/guardrails/detect", json=payload)
 
@@ -94,47 +94,47 @@ class GuardrailsClient:
     # Guardrails Policy Endpoints
     # ----------------------------
 
-    def add_policy(self, name, description, guardrails_config):
+    def add_policy(self, name, description, config):
         """
         Create a new policy with custom configurations.
         """
         payload = {
             "name": name,
             "description": description,
-            "detectors": guardrails_config
+            "detectors": config
         }
         return self._request("POST", "/guardrails/add-policy", json=payload)
 
-    def get_policy(self, x_enkrypt_policy):
+    def get_policy(self, policy_name):
         """
         Retrieve an existing policy by providing its header identifier.
         """
-        headers = {"X-Enkrypt-Policy": x_enkrypt_policy}
+        headers = {"X-Enkrypt-Policy": policy_name}
         return self._request("GET", "/guardrails/get-policy", headers=headers)
 
-    def modify_policy(self, x_enkrypt_policy, name, description, guardrails_config):
+    def modify_policy(self, policy_name, name, description, config):
         """
         Modify an existing policy.
         """
-        headers = {"X-Enkrypt-Policy": x_enkrypt_policy}
+        headers = {"X-Enkrypt-Policy": policy_name}
         payload = {
             "name": name,
             "description": description,
-            "detectors": guardrails_config
+            "detectors": config
         }
         return self._request("PATCH", "/guardrails/modify-policy", headers=headers, json=payload)
 
-    def delete_policy(self, x_enkrypt_policy):
+    def delete_policy(self, policy_name):
         """
         Delete a policy.
         """
-        headers = {"X-Enkrypt-Policy": x_enkrypt_policy}
+        headers = {"X-Enkrypt-Policy": policy_name}
         return self._request("DELETE", "/guardrails/delete-policy", headers=headers)
 
-    def policy_detect(self, x_enkrypt_policy, text):
+    def policy_detect(self, policy_name, text):
         """
         Apply a specific policy to detect and filter content.
         """
-        headers = {"X-Enkrypt-Policy": x_enkrypt_policy}
+        headers = {"X-Enkrypt-Policy": policy_name}
         payload = {"text": text}
         return self._request("POST", "/guardrails/policy/detect", headers=headers, json=payload)

{enkryptai_sdk-0.1.1.dist-info → enkryptai_sdk-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: enkryptai-sdk
-Version: 0.1.1
+Version: 0.1.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

enkryptai_sdk-0.1.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+enkryptai_sdk/__init__.py,sha256=git4pQzKT36zA-wIDz71crkRhp91FFIYrJXehUcnofg,141
+enkryptai_sdk/guardrails.py,sha256=toEYdYMU0qM6wmrpxfoR-DgzV7tug4NHtilu01CVpP8,4457
+enkryptai_sdk/guardrails_config.py,sha256=oFhCX2hJGVFQfGcaJqji4enc35gK5dTc95uSWungRPE,2487
+enkryptai_sdk/red_team.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-0.1.2.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-0.1.2.dist-info/METADATA,sha256=rJYlv7XSpSu5O7VzXYFx11--CJIymKM01N2Era_AuO8,2324
+enkryptai_sdk-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+enkryptai_sdk-0.1.2.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-0.1.2.dist-info/RECORD,,

enkryptai_sdk-0.1.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-enkryptai_sdk/__init__.py,sha256=git4pQzKT36zA-wIDz71crkRhp91FFIYrJXehUcnofg,141
-enkryptai_sdk/guardrails.py,sha256=rxUiHPAvdgIjX00XtAY5jPGde-_OEnpmV51jDxAZOSA,4596
-enkryptai_sdk/guardrails_config.py,sha256=oFhCX2hJGVFQfGcaJqji4enc35gK5dTc95uSWungRPE,2487
-enkryptai_sdk/red_team.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-0.1.1.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-0.1.1.dist-info/METADATA,sha256=DZDuoUjlb5ptOuGdMzTLzkZ561kYqis2VjjhajT3bnE,2324
-enkryptai_sdk-0.1.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-enkryptai_sdk-0.1.1.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-0.1.1.dist-info/RECORD,,