enkryptai-sdk 1.0.9__py3-none-any.whl → 1.0.10__py3-none-any.whl

enkryptai_sdk/config.py

@@ -10,7 +10,7 @@ DEFAULT_GUARDRAILS_CONFIG = {
     "keyword_detector": {"enabled": False, "banned_keywords": []},
     "policy_violation": {
         "enabled": False,
-        "policy_text": "",
+        "policy_text": "Do not allow any illegal or immoral activities.",
         "need_explanation": False,
     },
     "bias": {"enabled": False},
@@ -42,16 +42,21 @@ class GuardrailsConfig:
         return cls(config)
 
     @classmethod
-    def policy_violation(cls, policy_text: str, need_explanation: bool = False):
+    def policy_violation(cls, policy_text: str = "", need_explanation: bool = False, coc_policy_name: str = ""):
         """
         Returns a configuration instance pre-configured for policy violation detection.
         """
         config = copy.deepcopy(DEFAULT_GUARDRAILS_CONFIG)
         config["policy_violation"] = {
             "enabled": True,
-            "policy_text": policy_text,
             "need_explanation": need_explanation,
         }
+
+        if policy_text:
+            config["policy_violation"]["policy_text"] = policy_text
+        if coc_policy_name:
+            config["policy_violation"]["coc_policy_name"] = coc_policy_name
+
         return cls(config)
 
     @classmethod

enkryptai_sdk/dto/guardrails.py

@@ -163,6 +163,7 @@ class PolicyViolationDetector(BaseDTO):
     enabled: bool = False
     policy_text: str = ""
     need_explanation: bool = False
+    coc_policy_name: str = ""
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
     @classmethod
@@ -170,16 +171,23 @@ class PolicyViolationDetector(BaseDTO):
         return cls(
             enabled=data.get("enabled", False),
             policy_text=data.get("policy_text", ""),
-            need_explanation=data.get("need_explanation", False)
+            need_explanation=data.get("need_explanation", False),
+            coc_policy_name=data.get("coc_policy_name", "")
         )
     
     def to_dict(self) -> Dict[str, Any]:
-        return {
+        res_dict = {
             "enabled": self.enabled,
-            "policy_text": self.policy_text,
             "need_explanation": self.need_explanation
         }
 
+        if self.policy_text:
+            res_dict["policy_text"] = self.policy_text
+        if self.coc_policy_name:
+            res_dict["coc_policy_name"] = self.coc_policy_name
+
+        return res_dict
+
 
 @dataclass
 class BiasDetector(BaseDTO):

{enkryptai_sdk-1.0.9.dist-info → enkryptai_sdk-1.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.9
+Version: 1.0.10
 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
@@ -264,7 +264,10 @@ sample_detectors = {
         "policy_violation": {
             "enabled": True,
             "need_explanation": True,
-            "policy_text": ""
+            "policy_text": "The model should not provide medical advice when asked about health symptoms."
+            # Or we can also give coc_policy_name of a saved Code of Conduct Policy
+            # Instead of policy_text
+            # "coc_policy_name": "Test CoC Policy"
         },
         "bias": {
             "enabled": False
@@ -718,6 +721,9 @@ guardrails_config = GuardrailsConfig.injection_attack()
 
 ```python Python
 guardrails_config = GuardrailsConfig.policy_violation(policy_text="You must not use hate speech", need_explanation=True)
+
+# Or we can also give coc_policy_name of a saved Code of Conduct Policy instead of policy_text
+guardrails_config = GuardrailsConfig.policy_violation(coc_policy_name="Test CoC Policy", need_explanation=True)
 ```
 
 ### [Toxicity](https://docs.enkryptai.com/guardrails-api-reference/Toxicity_Detector)

{enkryptai_sdk-1.0.9.dist-info → enkryptai_sdk-1.0.10.dist-info}/RECORD

@@ -2,7 +2,7 @@ enkryptai_sdk/__init__.py,sha256=8H5tznmjirTVdrTmrsyU6fNVRj3-UC0nrTGkzfFp0h0,845
 enkryptai_sdk/ai_proxy.py,sha256=ot1lqKk2LjcvlpnFm2kSA51vFThfquVlx86BhSbAzBo,3823
 enkryptai_sdk/base.py,sha256=8iQr5skm46-hGFIMOZsQgZkqL0RBrIJDYUENOybBrwg,3087
 enkryptai_sdk/coc.py,sha256=5rq9LhZX-uvCmX8fM6JEndIEvd8rSzsSfDFnTvSvTQE,7396
-enkryptai_sdk/config.py,sha256=IpB8_aO4zXdvv061v24oh83oyJ5Tp1QBQTzeuW4h9QY,8828
+enkryptai_sdk/config.py,sha256=PyyuJRKWuiuFUaI90M21M141wlfIBeO_QVDU36KHKCs,9065
 enkryptai_sdk/datasets.py,sha256=RQIR6spI2STXeVolYzBt6gPv6PD5AGh9krs16aKWdWA,6067
 enkryptai_sdk/deployments.py,sha256=A7XZ2JwrMod9V4_aV8bFY_Soh9E3jHdwaTuJ9BwXuyk,4215
 enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
@@ -17,11 +17,11 @@ enkryptai_sdk/dto/base.py,sha256=6VWTkoNZ7uILqn_iYsPS21cVa2xLYpw5bjDIsRCS5tk,238
 enkryptai_sdk/dto/coc.py,sha256=Lp2aat_24J4KuUg4BeJl9S39tEak8Bw15eJ4cQDrRQk,4749
 enkryptai_sdk/dto/datasets.py,sha256=RFA9CmbhD-QDDyweBq_k9iBd00b6I6SWmdP9DPNd9fc,5002
 enkryptai_sdk/dto/deployments.py,sha256=Aw4b8tDA3FYIomqDvCjblCXTagL4bT8Fx91X0SFXs40,11216
-enkryptai_sdk/dto/guardrails.py,sha256=7hWzhFH-tc6Rbwkl6ZzW_egWTvrLXv4K22M7dNbKQuE,49556
+enkryptai_sdk/dto/guardrails.py,sha256=wfKm2R0uJAq9nljLBPuuT05tj1nBqZraga-5lMHd3n8,49812
 enkryptai_sdk/dto/models.py,sha256=lNakMstGozgN_htY2_OItkjQnMQpw5tO9lCZVlywMIg,13442
 enkryptai_sdk/dto/red_team.py,sha256=QtsT4GCnDs24WOF_RMxFtYjvIUULaw5temZ70aduevI,17966
-enkryptai_sdk-1.0.9.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.9.dist-info/METADATA,sha256=GDQsqPQycozBDIOW3qCfq3wEzNBVPs_4XcmXg-V3kYo,61634
-enkryptai_sdk-1.0.9.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-enkryptai_sdk-1.0.9.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.9.dist-info/RECORD,,
+enkryptai_sdk-1.0.10.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.10.dist-info/METADATA,sha256=elvqsbmZjsswZaBA2cbm1Dmy4teinjb7mR2EPUvtOR0,62092
+enkryptai_sdk-1.0.10.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+enkryptai_sdk-1.0.10.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.10.dist-info/RECORD,,

{enkryptai_sdk-1.0.9.dist-info → enkryptai_sdk-1.0.10.dist-info}/WHEEL

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