enkryptai-sdk 1.0.18__tar.gz → 1.0.20__tar.gz

{enkryptai_sdk-1.0.18/src/enkryptai_sdk.egg-info → enkryptai_sdk-1.0.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.18
+Version: 1.0.20
 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
@@ -275,6 +275,9 @@ sample_detectors = {
         },
         "bias": {
             "enabled": False
+        },
+        "sponge_attack": {
+            "enabled": False
         }
     }
 ```
@@ -984,6 +987,7 @@ print(batch_detect_response.to_dict())
 - `copyright_ip`: Check for copyright/IP violations ***(Coming soon)***
 - `system_prompt`: Detect system prompt leaks ***(Coming soon)***
 - `keyword_detector`: Check for specific keywords
+- `sponge_attack`: Detect infinite loops
 
 Each detector can be enabled/disabled and configured with specific options as documented in the [API docs](https://docs.enkryptai.com/guardrails-api-reference/introduction).
 
@@ -1058,6 +1062,12 @@ guardrails_config = GuardrailsConfig.copyright_ip()
 guardrails_config = GuardrailsConfig.system_prompt(index="system")
 ```
 
+### [Sponge Attack](https://docs.enkryptai.com/guardrails-api-reference/Sponge_Attack_Detector)
+
+```python Python
+guardrails_config = GuardrailsConfig.sponge_attack()
+```
+
 ## [Detect with config](https://docs.enkryptai.com/guardrails-api-reference/Detect)
 
 ```python Python
@@ -1656,7 +1666,7 @@ custom_headers = {
 
 # Example of making a request with custom headers
 response = client.chat.completions.create(
-    model='gpt-4o',
+    # model='gpt-4o', # Optional
     messages=[{'role': 'user', 'content': 'Hello!'}],
     extra_headers=custom_headers
 )

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/README.md

@@ -252,6 +252,9 @@ sample_detectors = {
         },
         "bias": {
             "enabled": False
+        },
+        "sponge_attack": {
+            "enabled": False
         }
     }
 ```
@@ -961,6 +964,7 @@ print(batch_detect_response.to_dict())
 - `copyright_ip`: Check for copyright/IP violations ***(Coming soon)***
 - `system_prompt`: Detect system prompt leaks ***(Coming soon)***
 - `keyword_detector`: Check for specific keywords
+- `sponge_attack`: Detect infinite loops
 
 Each detector can be enabled/disabled and configured with specific options as documented in the [API docs](https://docs.enkryptai.com/guardrails-api-reference/introduction).
 
@@ -1035,6 +1039,12 @@ guardrails_config = GuardrailsConfig.copyright_ip()
 guardrails_config = GuardrailsConfig.system_prompt(index="system")
 ```
 
+### [Sponge Attack](https://docs.enkryptai.com/guardrails-api-reference/Sponge_Attack_Detector)
+
+```python Python
+guardrails_config = GuardrailsConfig.sponge_attack()
+```
+
 ## [Detect with config](https://docs.enkryptai.com/guardrails-api-reference/Detect)
 
 ```python Python
@@ -1633,7 +1643,7 @@ custom_headers = {
 
 # Example of making a request with custom headers
 response = client.chat.completions.create(
-    model='gpt-4o',
+    # model='gpt-4o', # Optional
     messages=[{'role': 'user', 'content': 'Hello!'}],
     extra_headers=custom_headers
 )

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/setup.py

@@ -9,7 +9,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
     # NOTE: Also change this in .github/workflows/test.yaml
-    version="1.0.18", # Update this for new versions
+    version="1.0.20", # 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.18 → enkryptai_sdk-1.0.20}/src/enkryptai_sdk/config.py

@@ -16,6 +16,7 @@ DEFAULT_GUARDRAILS_CONFIG = {
     "bias": {"enabled": False},
     "copyright_ip": {"enabled": False},
     "system_prompt": {"enabled": False, "index": "system"},
+    "sponge_attack": {"enabled": False},
 }
 
 
@@ -151,6 +152,15 @@ class GuardrailsConfig:
         config = copy.deepcopy(DEFAULT_GUARDRAILS_CONFIG)
         config["system_prompt"] = {"enabled": True, "index": index}
         return cls(config)
+    
+    @classmethod
+    def sponge_attack(cls):
+        """
+        Returns a configuration instance pre-configured for sponge attack detection.
+        """
+        config = copy.deepcopy(DEFAULT_GUARDRAILS_CONFIG)
+        config["sponge_attack"] = {"enabled": True}
+        return cls(config)
 
     def update(self, **kwargs):
         """

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/src/enkryptai_sdk/dto/deployments.py

@@ -15,6 +15,7 @@ class InputGuardrailBlockType(str, Enum):
     BIAS = "bias"
     SYSTEM_PROMPT = "system_prompt"
     COPYRIGHT_IP = "copyright_ip"
+    SPONGE_ATTACK = "sponge_attack"
 
 
 class OutputGuardrailBlockType(str, Enum):
@@ -28,6 +29,7 @@ class OutputGuardrailBlockType(str, Enum):
     BIAS = "bias"
     SYSTEM_PROMPT = "system_prompt"
     COPYRIGHT_IP = "copyright_ip"
+    SPONGE_ATTACK = "sponge_attack"
     HALLUCINATION = "hallucination"
     ADHERENCE = "adherence"
     RELEVANCY = "relevancy"

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/src/enkryptai_sdk/dto/guardrails.py

@@ -243,6 +243,23 @@ class SystemPromptDetector(BaseDTO):
         }
 
 
+@dataclass
+class SpongeAttackDetector(BaseDTO):
+    enabled: bool = False
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "SpongeAttackDetector":
+        return cls(
+            enabled=data.get("enabled", False)
+        )
+    
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "enabled": self.enabled
+        }
+
+
 class GuardrailDetectorsEnum(str, Enum):
     TOPIC_DETECTOR = "topic_detector"
     NSFW = "nsfw"
@@ -254,6 +271,7 @@ class GuardrailDetectorsEnum(str, Enum):
     BIAS = "bias"
     COPYRIGHT_IP = "copyright_ip"
     SYSTEM_PROMPT = "system_prompt"
+    SPONGE_ATTACK = "sponge_attack"
 
 
 @dataclass
@@ -268,6 +286,7 @@ class GuardrailDetectors(BaseDTO):
     bias: BiasDetector = field(default_factory=BiasDetector)
     copyright_ip: CopyrightIPDetector = field(default_factory=CopyrightIPDetector)
     system_prompt: SystemPromptDetector = field(default_factory=SystemPromptDetector)
+    sponge_attack: SpongeAttackDetector = field(default_factory=SpongeAttackDetector)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
     @classmethod
@@ -282,7 +301,8 @@ class GuardrailDetectors(BaseDTO):
             policy_violation=PolicyViolationDetector.from_dict(data.get("policy_violation", {})),
             bias=BiasDetector.from_dict(data.get("bias", {})),
             copyright_ip=CopyrightIPDetector.from_dict(data.get("copyright_ip", {})),
-            system_prompt=SystemPromptDetector.from_dict(data.get("system_prompt", {}))
+            system_prompt=SystemPromptDetector.from_dict(data.get("system_prompt", {})),
+            sponge_attack=SpongeAttackDetector.from_dict(data.get("sponge_attack", {}))
         )
     
     def to_dict(self) -> Dict[str, Any]:
@@ -296,7 +316,8 @@ class GuardrailDetectors(BaseDTO):
             "policy_violation": self.policy_violation.to_dict(),
             "bias": self.bias.to_dict(),
             "copyright_ip": self.copyright_ip.to_dict(),
-            "system_prompt": self.system_prompt.to_dict()
+            "system_prompt": self.system_prompt.to_dict(),
+            "sponge_attack": self.sponge_attack.to_dict()
         }
 
 
@@ -608,6 +629,25 @@ class SystemPromptDetail(BaseDTO):
         return result
 
 
+@dataclass
+class SpongeAttackDetail(BaseDTO):
+    sponge_attack_detected: bool = False
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "SpongeAttackDetail":
+        return cls(
+            sponge_attack_detected=data.get("sponge_attack_detected", False)
+        )
+    
+    def to_dict(self) -> Dict[str, Any]:
+        result = {
+            "sponge_attack_detected": self.sponge_attack_detected
+        }
+        result.update(self._extra_fields)
+        return result
+
+
 @dataclass
 class DetectResponseSummary(BaseDTO):
     on_topic: int = 0
@@ -620,6 +660,7 @@ class DetectResponseSummary(BaseDTO):
     bias: int = 0
     copyright_ip_similarity: int = 0
     system_prompt_similarity: int = 0
+    sponge_attack: int = 0
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
     @classmethod
@@ -634,7 +675,8 @@ class DetectResponseSummary(BaseDTO):
             policy_violation=data.get("policy_violation", 0),
             bias=data.get("bias", 0),
             copyright_ip_similarity=data.get("copyright_ip_similarity", 0),
-            system_prompt_similarity=data.get("system_prompt_similarity", 0)
+            system_prompt_similarity=data.get("system_prompt_similarity", 0),
+            sponge_attack=data.get("sponge_attack", 0)
         )
     
     def to_dict(self) -> Dict[str, Any]:
@@ -648,7 +690,8 @@ class DetectResponseSummary(BaseDTO):
             "policy_violation": self.policy_violation,
             "bias": self.bias,
             "copyright_ip_similarity": self.copyright_ip_similarity,
-            "system_prompt_similarity": self.system_prompt_similarity
+            "system_prompt_similarity": self.system_prompt_similarity,
+            "sponge_attack": self.sponge_attack
         }
         result.update(self._extra_fields)
         return result
@@ -666,6 +709,7 @@ class DetectResponseDetails(BaseDTO):
     bias: BiasDetail = field(default_factory=BiasDetail)
     copyright_ip: CopyrightIPDetail = field(default_factory=CopyrightIPDetail)
     system_prompt: SystemPromptDetail = field(default_factory=SystemPromptDetail)
+    sponge_attack: SpongeAttackDetail = field(default_factory=SpongeAttackDetail)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
     @classmethod
@@ -680,7 +724,8 @@ class DetectResponseDetails(BaseDTO):
             policy_violation=PolicyViolationDetail.from_dict(data.get("policy_violation", {})),
             bias=BiasDetail.from_dict(data.get("bias", {})),
             copyright_ip=CopyrightIPDetail.from_dict(data.get("copyright_ip", {})),
-            system_prompt=SystemPromptDetail.from_dict(data.get("system_prompt", {}))
+            system_prompt=SystemPromptDetail.from_dict(data.get("system_prompt", {})),
+            sponge_attack=SpongeAttackDetail.from_dict(data.get("sponge_attack", {}))
         )
     
     def to_dict(self) -> Dict[str, Any]:
@@ -694,7 +739,8 @@ class DetectResponseDetails(BaseDTO):
             "policy_violation": self.policy_violation.to_dict(),
             "bias": self.bias.to_dict(),
             "copyright_ip": self.copyright_ip.to_dict(),
-            "system_prompt": self.system_prompt.to_dict()
+            "system_prompt": self.system_prompt.to_dict(),
+            "sponge_attack": self.sponge_attack.to_dict()
         }
         result.update(self._extra_fields)
         return result

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/src/enkryptai_sdk/dto/red_team.py

@@ -20,6 +20,7 @@ class RiskGuardrailDetectorsEnum(str, Enum):
     # Below are not yet supported by Guardrails. So, also not supported by Risk Mitigation.
     # COPYRIGHT_IP = "copyright_ip"
     # SYSTEM_PROMPT = "system_prompt"
+    # SPONGE_ATTACK = "sponge_attack"
 
 
 @dataclass
@@ -360,6 +361,28 @@ class TestConfig(BaseDTO):
         return cls(**data, attack_methods=attack_methods)
 
 
+@dataclass
+class CompetitorTestConfig(BaseDTO):
+    system_description: str
+    competitors_list: List[str]
+    sample_percentage: int = 1
+    attack_methods: AttackMethods = field(default_factory=AttackMethods)
+
+    def to_dict(self) -> dict:
+        return {
+            "sample_percentage": self.sample_percentage,
+            "competitors_list": self.competitors_list,
+            "system_description": self.system_description,
+            "attack_methods": self.attack_methods.to_dict(),
+        }
+
+    @classmethod
+    def from_dict(cls, data: dict):
+        data = data.copy()
+        attack_methods = AttackMethods.from_dict(data.pop("attack_methods", {}))
+        return cls(**data, attack_methods=attack_methods)
+
+
 @dataclass
 class RedTeamTestConfigurations(BaseDTO):
     # Basic tests
@@ -368,6 +391,12 @@ class RedTeamTestConfigurations(BaseDTO):
     insecure_code_test: TestConfig = field(default=None)
     toxicity_test: TestConfig = field(default=None)
     harmful_test: TestConfig = field(default=None)
+    pii_test: TestConfig = field(default=None)
+    copyright_test: TestConfig = field(default=None)
+    misinformation_test: TestConfig = field(default=None)
+    system_prompt_extractions_test: TestConfig = field(default=None)
+    sponge_test: TestConfig = field(default=None)
+    competitor_test: CompetitorTestConfig = field(default=None)
     # Advanced tests
     adv_info_test: TestConfig = field(default=None)
     adv_bias_test: TestConfig = field(default=None)
@@ -386,6 +415,14 @@ class RedTeamTestConfigurations(BaseDTO):
     access_control_test: TestConfig = field(default=None)
     physical_and_actuation_safety_test: TestConfig = field(default=None)
     reliability_and_monitoring_test: TestConfig = field(default=None)
+    governance_test: TestConfig = field(default=None)
+    agent_output_quality_test: TestConfig = field(default=None)
+    tool_misuse_test: TestConfig = field(default=None)
+    privacy_test: TestConfig = field(default=None)
+    reliability_and_observability_test: TestConfig = field(default=None)
+    agent_behaviour_test: TestConfig = field(default=None)
+    access_control_and_permissions_test: TestConfig = field(default=None)
+    tool_extraction_test: TestConfig = field(default=None)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
     @classmethod

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20/src/enkryptai_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.18
+Version: 1.0.20
 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
@@ -275,6 +275,9 @@ sample_detectors = {
         },
         "bias": {
             "enabled": False
+        },
+        "sponge_attack": {
+            "enabled": False
         }
     }
 ```
@@ -984,6 +987,7 @@ print(batch_detect_response.to_dict())
 - `copyright_ip`: Check for copyright/IP violations ***(Coming soon)***
 - `system_prompt`: Detect system prompt leaks ***(Coming soon)***
 - `keyword_detector`: Check for specific keywords
+- `sponge_attack`: Detect infinite loops
 
 Each detector can be enabled/disabled and configured with specific options as documented in the [API docs](https://docs.enkryptai.com/guardrails-api-reference/introduction).
 
@@ -1058,6 +1062,12 @@ guardrails_config = GuardrailsConfig.copyright_ip()
 guardrails_config = GuardrailsConfig.system_prompt(index="system")
 ```
 
+### [Sponge Attack](https://docs.enkryptai.com/guardrails-api-reference/Sponge_Attack_Detector)
+
+```python Python
+guardrails_config = GuardrailsConfig.sponge_attack()
+```
+
 ## [Detect with config](https://docs.enkryptai.com/guardrails-api-reference/Detect)
 
 ```python Python
@@ -1656,7 +1666,7 @@ custom_headers = {
 
 # Example of making a request with custom headers
 response = client.chat.completions.create(
-    model='gpt-4o',
+    # model='gpt-4o', # Optional
     messages=[{'role': 'user', 'content': 'Hello!'}],
     extra_headers=custom_headers
 )

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/tests/test_all_v2.py

@@ -156,6 +156,9 @@ def sample_detectors():
             "enabled": False,
             "index": "system"
         },
+        "sponge_attack": {
+            "enabled": False
+        },
     }
 
 

{enkryptai_sdk-1.0.18 → enkryptai_sdk-1.0.20}/tests/test_guardrails.py

@@ -75,6 +75,9 @@ def sample_detectors():
             "enabled": False,
             "index": "system"
         },
+        "sponge_attack": {
+            "enabled": False
+        },
     }