enkryptai-sdk 1.0.4__py3-none-any.whl → 1.0.6__py3-none-any.whl

enkryptai_sdk/datasets.py

@@ -36,9 +36,23 @@ class DatasetClient(BaseClient):
 
         payload = config.to_dict()
 
+        # If payload["tools"] is None or is an empty list [] or [{}], remove it from the payload
+        if (payload.get("tools") is None or 
+            payload["tools"] == [] or 
+            payload["tools"] == [{}] or
+            payload["tools"] == [{"name": "", "description": ""}]):
+            del payload["tools"]
+
+        # Print payload
+        # print(f"\nAdd Dataset Payload: {payload}")
+
         response = self._request(
             "POST", "/datasets/add-task", headers=headers, json=payload
         )
+
+        # Print response
+        # print(f"\nAdd Dataset Response: {response}")
+
         if response.get("error"):
             raise DatasetClientError(response["error"])
         return DatasetAddTaskResponse.from_dict(response)

enkryptai_sdk/dto/__init__.py

@@ -62,10 +62,13 @@ __all__ = [
     "GuardrailsModelsResponse",
     "GuardrailDetectors",
     "GuardrailsDetectRequest",
+    "GuardrailsBatchDetectRequest",
     "GuardrailsPolicyDetectRequest",
     "DetectResponseSummary",
     "DetectResponseDetails",
     "GuardrailsDetectResponse",
+    "BatchDetectResponseItem",
+    "GuardrailsBatchDetectResponse",
     "GuardrailsPIIRequest",
     "GuardrailsPIIResponse",
     "GuardrailsHallucinationRequest",

enkryptai_sdk/dto/datasets.py

@@ -3,17 +3,35 @@ from dataclasses import dataclass, field
 from typing import Dict, List, Optional, Any
 
 
+@dataclass
+class Tool(BaseDTO):
+    name: str
+    description: str
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "Tool":
+        return cls(
+            name=data.get("name", ""),
+            description=data.get("description", "")
+        )
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "name": self.name,
+            "description": self.description
+        }
+
 @dataclass
 class DatasetConfig(BaseDTO):
     dataset_name: str
     system_description: str
     policy_description: str = ""
-    tools: List[str] = field(default_factory=list)
+    tools: List[Tool] = field(default_factory=list)
     info_pdf_url: str = ""
     max_prompts: int = 100
-    # scenarios: int = 2
-    # categories: int = 2
-    # depth: int = 2
+    scenarios: int = 2
+    categories: int = 2
+    depth: int = 2
 
 
 @dataclass

enkryptai_sdk/dto/deployments.py

@@ -141,6 +141,7 @@ class OutputGuardrailsPolicy(BaseDTO):
 class DeploymentInput(BaseDTO):
     name: str
     model_saved_name: str
+    model_version: str
     input_guardrails_policy: InputGuardrailsPolicy
     output_guardrails_policy: OutputGuardrailsPolicy
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
@@ -153,6 +154,7 @@ class DeploymentInput(BaseDTO):
         return cls(
             name=data.get("name", ""),
             model_saved_name=data.get("model_saved_name", ""),
+            model_version=data.get("model_version", ""),
             input_guardrails_policy=InputGuardrailsPolicy.from_dict(input_policy_data),
             output_guardrails_policy=OutputGuardrailsPolicy.from_dict(output_policy_data)
         )
@@ -161,6 +163,7 @@ class DeploymentInput(BaseDTO):
         result = {
             "name": self.name,
             "model_saved_name": self.model_saved_name,
+            "model_version": self.model_version,
             "input_guardrails_policy": self.input_guardrails_policy.to_dict(),
             "output_guardrails_policy": self.output_guardrails_policy.to_dict()
         }
@@ -172,6 +175,7 @@ class DeploymentInput(BaseDTO):
 class GetDeploymentResponse(BaseDTO):
     name: str
     model_saved_name: str
+    model_version: str
     input_guardrails_policy: InputGuardrailsPolicy
     output_guardrails_policy: OutputGuardrailsPolicy
     created_at: str
@@ -186,6 +190,7 @@ class GetDeploymentResponse(BaseDTO):
         return cls(
             name=data.get("name", ""),
             model_saved_name=data.get("model_saved_name", ""),
+            model_version=data.get("model_version", ""),
             input_guardrails_policy=InputGuardrailsPolicy.from_dict(input_policy_data),
             output_guardrails_policy=OutputGuardrailsPolicy.from_dict(output_policy_data),
             updated_at=data.get("updated_at", ""),
@@ -197,6 +202,7 @@ class GetDeploymentResponse(BaseDTO):
         return {
             "name": self.name,
             "model_saved_name": self.model_saved_name,
+            "model_version": self.model_version,
             "input_guardrails_policy": self.input_guardrails_policy.to_dict(),
             "output_guardrails_policy": self.output_guardrails_policy.to_dict(),
             "updated_at": self.updated_at,
@@ -293,6 +299,7 @@ class DeploymentSummary(BaseDTO):
     created_at: str
     updated_at: str
     model_saved_name: str
+    model_version: str
     project_name: str
     
     @classmethod
@@ -303,6 +310,7 @@ class DeploymentSummary(BaseDTO):
             created_at=data.get("created_at", ""),
             updated_at=data.get("updated_at", ""),
             model_saved_name=data.get("model_saved_name", ""),
+            model_version=data.get("model_version", ""),
             project_name=data.get("project_name", "")
         )
     
@@ -313,6 +321,7 @@ class DeploymentSummary(BaseDTO):
             "created_at": self.created_at,
             "updated_at": self.updated_at,
             "model_saved_name": self.model_saved_name,
+            "model_version": self.model_version,
             "project_name": self.project_name
         }
 

enkryptai_sdk/dto/guardrails.py

@@ -301,6 +301,27 @@ class GuardrailsDetectRequest(BaseDTO):
         }
 
 
+@dataclass
+class GuardrailsBatchDetectRequest(BaseDTO):
+    texts: List[str] = field(default_factory=list)
+    detectors: GuardrailDetectors = field(default_factory=GuardrailDetectors)
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "GuardrailsBatchDetectRequest":
+        detectors_data = data.get("detectors", {})
+        return cls(
+            texts=data.get("texts", []),
+            detectors=GuardrailDetectors.from_dict(detectors_data)
+        )
+    
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "texts": self.texts,
+            "detectors": self.detectors.to_dict()
+        }
+
+
 @dataclass
 class GuardrailsPolicyDetectRequest(BaseDTO):
     text: str
@@ -750,6 +771,180 @@ class GuardrailsDetectResponse(BaseDTO):
         return f"Response Status: {status}\n{violation_str}"
 
 
+@dataclass
+class BatchDetectResponseItem(BaseDTO):
+    text: str
+    summary: DetectResponseSummary
+    details: DetectResponseDetails
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "BatchDetectResponseItem":
+        return cls(
+            text=data.get("text", ""),
+            summary=DetectResponseSummary.from_dict(data.get("summary", {})),
+            details=DetectResponseDetails.from_dict(data.get("details", {}))
+        )
+    
+    def to_dict(self) -> Dict[str, Any]:
+        result = {
+            "text": self.text,
+            "summary": self.summary.to_dict(),
+            "details": self.details.to_dict()
+        }
+        result.update(self._extra_fields)
+        return result
+
+    def has_violations(self) -> bool:
+        """
+        Check if any detectors found violations in the content.
+        
+        Returns:
+            bool: True if any detector reported a violation (score > 0), False otherwise
+        """
+        summary = self.summary.to_dict()
+        for key, value in summary.items():
+            if key == "toxicity" and isinstance(value, list) and len(value) > 0:
+                return True
+            elif isinstance(value, (int, float)) and value > 0:
+                return True
+        return False
+    
+    def get_violations(self) -> list[str]:
+        """
+        Get a list of detector names that found violations.
+        
+        Returns:
+            list[str]: Names of detectors that reported violations
+        """
+        summary = self.summary.to_dict()
+        violations = []
+        for detector, value in summary.items():
+            if detector == "toxicity" and isinstance(value, list) and len(value) > 0:
+                violations.append(detector)
+            elif isinstance(value, (int, float)) and value > 0:
+                violations.append(detector)
+        return violations
+
+    def is_safe(self) -> bool:
+        """
+        Check if the content is safe (no violations detected).
+        
+        Returns:
+            bool: True if no violations were detected, False otherwise
+        """
+        return not self.has_violations()
+    
+    def is_attack(self) -> bool:
+        """
+        Check if the content is attacked (violations detected).
+        
+        Returns:
+            bool: True if violations were detected, False otherwise
+        """
+        return self.has_violations()
+    
+    def __str__(self) -> str:
+        """
+        String representation of the response.
+        
+        Returns:
+            str: A formatted string showing summary and violation status
+        """
+        violations = self.get_violations()
+        status = "UNSAFE" if violations else "SAFE"
+        
+        if violations:
+            violation_str = f"Violations detected: {', '.join(violations)}"
+        else:
+            violation_str = "No violations detected"
+            
+        return f"Response Status: {status}\n{violation_str}"
+
+
+@dataclass
+class GuardrailsBatchDetectResponse(BaseDTO):
+    batch_detections: List[BatchDetectResponseItem] = field(default_factory=list)
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+    
+    @classmethod
+    def from_dict(cls, data: List[Dict[str, Any]]) -> "GuardrailsBatchDetectResponse":
+        return cls(
+            batch_detections=[BatchDetectResponseItem.from_dict(item) for item in data]
+        )
+    
+    def to_dict(self) -> List[Dict[str, Any]]:
+        return [response.to_dict() for response in self.batch_detections]
+
+    def has_violations(self) -> bool:
+        """
+        Check if any detectors found violations in any of the batch_detections.
+        
+        Returns:
+            bool: True if any detector reported a violation, False otherwise
+        """
+        for detection in self.batch_detections:         
+            summary = detection.summary.to_dict()
+            for key, value in summary.items():
+                if key == "toxicity" and isinstance(value, list) and len(value) > 0:
+                    return True
+                elif isinstance(value, (int, float)) and value > 0:
+                    return True
+        return False
+    
+    def get_violations(self) -> List[str]:
+        """
+        Get a list of texts that have violations.
+        
+        Returns:
+            List[str]: List of texts that have violations
+        """
+        violations = set()
+        for detection in self.batch_detections:
+            summary = detection.summary.to_dict()
+            for detector, value in summary.items():
+                if detector == "toxicity" and isinstance(value, list) and len(value) > 0:
+                    violations.add(detector)
+                elif isinstance(value, (int, float)) and value > 0:
+                    violations.add(detector)
+        return list(violations)
+
+    def is_safe(self) -> bool:
+        """
+        Check if all content is safe (no violations detected).
+        
+        Returns:
+            bool: True if no violations were detected in any response, False otherwise
+        """
+        return not self.has_violations()
+    
+    def is_attack(self) -> bool:
+        """
+        Check if any content is attacked (violations detected).
+        
+        Returns:
+            bool: True if violations were detected in any response, False otherwise
+        """
+        return self.has_violations()
+    
+    def __str__(self) -> str:
+        """
+        String representation of the batch response.
+        
+        Returns:
+            str: A formatted string showing violation status for all batch_detections
+        """
+        violations = self.get_violations()
+        status = "UNSAFE" if violations else "SAFE"
+        
+        if violations:
+            violation_str = f"Violations detected in texts:\n" + "\n".join(f"- {text}" for text in violations)
+        else:
+            violation_str = "No violations detected in any text"
+            
+        return f"Batch Response Status: {status}\n{violation_str}"
+
+
 # -------------------------------------
 # Guardrails PII
 # -------------------------------------

enkryptai_sdk/dto/models.py

@@ -7,14 +7,14 @@ from dataclasses import dataclass, field, asdict
 from typing import Optional, List, Set, Dict, Any
 
 
-class Modality(Enum):
-    TEXT = "text"
-    IMAGE = "image"
-    AUDIO = "audio"
-    VIDEO = "video"
+# class Modality(Enum):
+#     TEXT = "text"
+#     IMAGE = "image"
+#     AUDIO = "audio"
+#     VIDEO = "video"
 
-    def to_dict(self):
-        return self.value
+#     def to_dict(self):
+#         return self.value
 
 
 class ModelProviders(str, Enum):
@@ -85,7 +85,6 @@ class AuthData(BaseDTO):
 
 @dataclass
 class ModelDetailConfig:
-    model_version: Optional[str] = None
     model_source: str = ""
     # model_provider: str = "openai"
     model_provider: ModelProviders = ModelProviders.OPENAI
@@ -99,18 +98,52 @@ class ModelDetailConfig:
 @dataclass
 class DetailModelConfig:
     model_saved_name: str = "Model Name"
-    testing_for: str = "LLM"
-    model_name: str = "gpt-4o-mini"
-    modality: Modality = Modality.TEXT
+    model_version: str = "v1"
+    testing_for: str = "foundationModels"
+    model_name: Optional[str] = "gpt-4o-mini"
+    # modality: Modality = Modality.TEXT
     model_config: ModelDetailConfig = field(default_factory=ModelDetailConfig)
 
 
+class InputModality(str, Enum):
+    text = "text"
+    image = "image"
+    audio = "audio"
+    # video = "video"
+    # code = "code"
+
+
+class OutputModality(str, Enum):
+    text = "text"
+    # image = "image"
+    # audio = "audio"
+    # video = "video"
+    # code = "code"
+
+
+@dataclass
+class CustomHeader(BaseDTO):
+    key: str
+    value: str
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "CustomHeader":
+        return cls(
+            key=data.get("key", ""),
+            value=data.get("value", "")
+        )
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "key": self.key,
+            "value": self.value
+        }
+
+
 @dataclass
 class ModelConfigDetails(BaseDTO):
     model_id: str = ""
-    model_version: Optional[str] = None
     model_source: str = ""
-    model_name: str = ""
     # model_provider: str = "openai"
     model_provider: ModelProviders = ModelProviders.OPENAI
     model_api_value: str = ""
@@ -119,15 +152,23 @@ class ModelConfigDetails(BaseDTO):
     model_api_key: str = ""
     model_endpoint_url: str = ""
     rate_per_min: int = 20
-    testing_for: str = "LLM"
+    testing_for: str = "foundationModels"
     headers: str = ""
     system_prompt: str = ""
-    conversation_template: str = ""
     hosting_type: str = "External"
     endpoint_url: str = "https://api.openai.com/v1/chat/completions"
+    model_name: Optional[str] = ""
+    apikey: Optional[str] = None
     paths: Optional[PathsConfig] = None
+    tools: List[Dict[str, str]] = field(default_factory=list)
     auth_data: AuthData = field(default_factory=AuthData)
-    apikey: Optional[str] = None
+    input_modalities: List[InputModality] = field(default_factory=list)
+    output_modalities: List[OutputModality] = field(default_factory=list)
+    custom_headers: List[CustomHeader] = field(default_factory=list)
+    custom_payload: Dict[str, Any] = field(default_factory=dict)
+    custom_response_content_type: Optional[str] = ""
+    custom_response_format: Optional[str] = ""
+    metadata: Dict[str, Any] = field(default_factory=dict)
     default_request_options: Dict[str, Any] = field(default_factory=dict)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
@@ -136,6 +177,17 @@ class ModelConfigDetails(BaseDTO):
         # Create a copy of the data to avoid modifying the original
         data = data.copy()
 
+        if "custom_headers" in data:
+            data["custom_headers"] = [CustomHeader.from_dict(h) for h in data["custom_headers"]]
+
+        # Convert input_modalities strings to enum values
+        if "input_modalities" in data:
+            data["input_modalities"] = [InputModality(m) for m in data["input_modalities"]]
+        
+        # Convert output_modalities strings to enum values
+        if "output_modalities" in data:
+            data["output_modalities"] = [OutputModality(m) for m in data["output_modalities"]]
+
         # Validate model_provider if present
         if "model_provider" in data:
             provider = data["model_provider"]
@@ -151,10 +203,10 @@ class ModelConfigDetails(BaseDTO):
                 valid_providers = [p.value for p in ModelProviders]
                 raise ValueError(f"Invalid model_provider type. Valid values: {valid_providers}")
 
-        # Remove known fields that we don't want in our model
-        unwanted_fields = ["queryParams"]
-        for field in unwanted_fields:
-            data.pop(field, None)
+        # # Remove known fields that we don't want in our model
+        # unwanted_fields = ["queryParams"]
+        # for field in unwanted_fields:
+        #     data.pop(field, None)
 
         # Handle apikeys to apikey conversion
         if "apikeys" in data:
@@ -211,14 +263,14 @@ class ModelConfig(BaseDTO):
     updated_at: str = ""
     model_id: str = ""
     model_saved_name: str = "Model Name"
-    testing_for: str = "LLM"
-    model_name: str = "gpt-4o-mini"
-    model_type: str = "text_2_text"
-    modality: Modality = Modality.TEXT
+    model_version: str = "v1"
+    testing_for: str = "foundationModels"
+    # modality: Modality = Modality.TEXT
     project_name: str = ""
+    model_name: Optional[str] = "gpt-4o-mini"
     certifications: List[str] = field(default_factory=list)
     model_config: ModelConfigDetails = field(default_factory=ModelConfigDetails)
-    is_sample: bool = False
+    is_sample: Optional[bool] = False
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
     @classmethod
@@ -231,16 +283,16 @@ class ModelConfig(BaseDTO):
         except ValueError as e:
             raise ValueError(f"Error in model_config: {str(e)}")
 
-        # Handle Modality enum
-        modality_value = data.pop("modality", "text")
-        modality = Modality(modality_value)
+        # # Handle Modality enum
+        # modality_value = data.pop("modality", "text")
+        # modality = Modality(modality_value)
 
-        return cls(**data, modality=modality, model_config=model_config)
+        return cls(**data, model_config=model_config)
 
     @classmethod
     def __str__(self):
         """String representation of the ModelConfig."""
-        return f"ModelConfig(name={self.model_saved_name}, model={self.model_name})"
+        return f"ModelConfig(name={self.model_saved_name}, version={self.model_version}, model={self.model_name})"
 
     def __repr__(self):
         """Detailed string representation of the ModelConfig."""
@@ -341,7 +393,7 @@ class Task(BaseDTO):
 
     task_id: str
     status: str
-    model_name: str
+    model_name: Optional[str] = None
     test_name: Optional[str] = None
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 

enkryptai_sdk/dto/red_team.py

@@ -1,4 +1,5 @@
 import pandas as pd
+from enum import Enum
 from .base import BaseDTO
 from typing import Dict, List, Optional, Any
 from dataclasses import dataclass, field, asdict
@@ -100,11 +101,11 @@ class ResultSummary(BaseDTO):
     test_date: str
     test_name: str
     dataset_name: str
-    model_name: str
     model_endpoint_url: str
     model_source: str
     model_provider: str
     risk_score: float
+    model_name: Optional[str]
     test_type: Dict[str, StatisticItem]
     nist_category: Dict[str, StatisticItem]
     scenario: Dict[str, StatisticItem]
@@ -257,9 +258,10 @@ class RedTeamResultDetails(BaseDTO):
 @dataclass
 class AttackMethods(BaseDTO):
     basic: List[str] = field(default_factory=lambda: ["basic"])
-    advanced: Dict[str, List[str]] = field(
-        default_factory=lambda: {"static": ["single_shot"], "dynamic": ["iterative"]}
-    )
+    # advanced: Dict[str, List[str]] = field(
+    #     default_factory=lambda: {"static": ["single_shot"], "dynamic": ["iterative"]}
+    # )
+    advanced: Dict[str, List[str]] = field(default_factory=dict)
 
     def to_dict(self) -> dict:
         return asdict(self)
@@ -271,7 +273,7 @@ class AttackMethods(BaseDTO):
 
 @dataclass
 class TestConfig(BaseDTO):
-    sample_percentage: int = 100
+    sample_percentage: int = 5
     attack_methods: AttackMethods = field(default_factory=AttackMethods)
 
     def to_dict(self) -> dict:
@@ -298,6 +300,7 @@ class RedTeamTestConfigurations(BaseDTO):
     adv_info_test: TestConfig = field(default=None)
     adv_bias_test: TestConfig = field(default=None)
     adv_command_test: TestConfig = field(default=None)
+    # custom_test: TestConfig = field(default=None)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
     @classmethod
@@ -305,19 +308,58 @@ class RedTeamTestConfigurations(BaseDTO):
         return cls(**{k: TestConfig.from_dict(v) for k, v in data.items()})
 
 
+class InputModality(str, Enum):
+    text = "text"
+    image = "image"
+    audio = "audio"
+    # video = "video"
+    # code = "code"
+
+
+class OutputModality(str, Enum):
+    text = "text"
+    # image = "image"
+    # audio = "audio"
+    # video = "video"
+    # code = "code"
+
+
+@dataclass
+class CustomHeader(BaseDTO):
+    key: str
+    value: str
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "CustomHeader":
+        return cls(
+            key=data.get("key", ""),
+            value=data.get("value", "")
+        )
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "key": self.key,
+            "value": self.value
+        }
+    
+
 @dataclass
 class TargetModelConfiguration(BaseDTO):
-    testing_for: str = "LLM"
-    model_name: str = "gpt-4o-mini"
-    model_type: str = "text_2_text"
-    model_version: Optional[str] = None
+    testing_for: str = "foundationModels"
     system_prompt: str = ""
-    conversation_template: str = ""
     model_source: str = ""
     model_provider: str = "openai"
     model_endpoint_url: str = "https://api.openai.com/v1/chat/completions"
-    model_api_key: Optional[str] = None
     rate_per_min: int = 20
+    model_name: Optional[str] = "gpt-4o-mini"
+    model_version: Optional[str] = None
+    model_api_key: Optional[str] = None
+    input_modalities: List[InputModality] = field(default_factory=list)
+    output_modalities: List[OutputModality] = field(default_factory=list)
+    custom_headers: List[CustomHeader] = field(default_factory=list)
+    custom_payload: Dict[str, Any] = field(default_factory=dict)
+    custom_response_content_type: Optional[str] = ""
+    custom_response_format: Optional[str] = ""
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
     @classmethod
@@ -351,19 +393,25 @@ class RedTeamModelHealthConfig(BaseDTO):
 @dataclass
 class RedteamModelHealthResponse(BaseDTO):
     status: str
+    message: str
     error: str
+    data: Optional[Dict[str, Any]] = field(default_factory=dict)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
     @classmethod
     def from_dict(cls, data: Dict[str, Any]) -> "RedteamModelHealthResponse":
         return cls(
             status=data.get("status", ""),
+            message=data.get("message", ""),
+            data=data.get("data", {}),
             error=data.get("error", "")
         )
     
     def to_dict(self) -> Dict[str, Any]:
         return {
             "status": self.status,
+            "message": self.message,
+            "data": self.data,
             "error": self.error
         }
 
@@ -409,6 +457,7 @@ class RedTeamConfigWithSavedModel(BaseDTO):
     test_name: str = "Test Name"
     dataset_name: str = "standard"
     model_saved_name: str = "gpt-4o-mini"
+    model_version: str = "v1"
 
     redteam_test_configurations: RedTeamTestConfigurations = field(
         default_factory=RedTeamTestConfigurations

enkryptai_sdk/guardrails.py

@@ -7,10 +7,13 @@ from .dto import (
     GuardrailsModelsResponse,
     # GuardrailDetectors,
     # GuardrailsDetectRequest,
+    # GuardrailsBatchDetectRequest,
     # GuardrailsPolicyDetectRequest,
     # DetectResponseSummary,
     # DetectResponseDetails,
     GuardrailsDetectResponse,
+    # BatchDetectResponseItem,
+    GuardrailsBatchDetectResponse,
     # GuardrailsPIIRequest,
     GuardrailsPIIResponse,
     # GuardrailsHallucinationRequest,
@@ -154,6 +157,42 @@ class GuardrailsClient(BaseClient):
         except Exception as e:
             raise GuardrailsClientError(str(e))
         
+    def batch_detect(self, texts, config=None):
+        """
+        Detects prompt injection, toxicity, NSFW content, PII, hallucination, and more in batch.
+
+        Parameters:
+        - texts (list): A list of texts to analyze.
+        - guardrails_config (dict or GuardrailsConfig, optional): A configuration for detectors.
+          If a GuardrailsConfig instance is provided, its underlying dictionary will be used.
+          If not provided, defaults to injection attack detection only.
+
+        Returns:
+        - Response from the API.
+        """
+        # Use injection attack config by default if none provided
+        if config is None:
+            config = GuardrailsConfig.injection_attack()
+
+        # Allow passing in either a dict or a GuardrailsConfig or GuardrailDetectors instance
+        if hasattr(config, "as_dict"):
+            config = config.as_dict()
+        if hasattr(config, "to_dict"):
+            config = config.to_dict()
+            
+        payload = {
+            "texts": texts,
+            "detectors": config
+        }
+
+        try:
+            response = self._request("POST", "/guardrails/batch/detect", json=payload)
+            if isinstance(response, dict) and response.get("error"):
+                raise GuardrailsClientError(response["error"])
+            return GuardrailsBatchDetectResponse.from_dict(response)
+        except Exception as e:
+            raise GuardrailsClientError(str(e))
+    
     def pii(self, text, mode="request", key="null", entities=None):
         """
         Detects Personally Identifiable Information (PII) and can de-anonymize it.

enkryptai_sdk/models.py

@@ -51,17 +51,15 @@ class ModelClient(BaseClient):
 
         payload = {
             "model_saved_name": config.model_saved_name,
+            "model_version": config.model_version,
             "testing_for": config.testing_for,
             "model_name": config.model_name,
-            "model_type": config.model_type,
             "certifications": config.certifications,
             "model_config": {
                 "model_provider": config.model_config.model_provider,
-                "model_version": config.model_config.model_version,
                 "hosting_type": config.model_config.hosting_type,
                 "model_source": config.model_config.model_source,
                 "system_prompt": config.model_config.system_prompt,
-                "conversation_template": config.model_config.conversation_template,
                 "endpoint": {
                     "scheme": parsed_url.scheme,
                     "host": parsed_url.hostname,
@@ -78,6 +76,14 @@ class ModelClient(BaseClient):
                 "apikeys": (
                     [config.model_config.apikey] if config.model_config.apikey else []
                 ),
+                "tools": config.model_config.tools,
+                "input_modalities": config.model_config.input_modalities,
+                "output_modalities": config.model_config.output_modalities,
+                "custom_headers": config.model_config.custom_headers,
+                "custom_payload": config.model_config.custom_payload,
+                "custom_response_content_type": config.model_config.custom_response_content_type,
+                "custom_response_format": config.model_config.custom_response_format,
+                "metadata": config.model_config.metadata,
                 "default_request_options": config.model_config.default_request_options,
             },
         }
@@ -91,17 +97,18 @@ class ModelClient(BaseClient):
         except Exception as e:
             raise ModelClientError(str(e))
 
-    def get_model(self, model_saved_name: str) -> ModelConfig:
+    def get_model(self, model_saved_name: str, model_version: str) -> ModelConfig:
         """
         Get model configuration by model saved name.
 
         Args:
             model_saved_name (str): Saved name of the model to retrieve
+            model_version (str): Version of the model to retrieve
 
         Returns:
             ModelConfig: Configuration object containing model details
         """
-        headers = {"X-Enkrypt-Model": model_saved_name}
+        headers = {"X-Enkrypt-Model": model_saved_name, "X-Enkrypt-Model-Version": model_version}
         response = self._request("GET", "/models/get-model", headers=headers)
         # print(response)
         if response.get("error"):
@@ -123,12 +130,13 @@ class ModelClient(BaseClient):
         except Exception as e:
             return {"error": str(e)}
 
-    def modify_model(self, config: ModelConfig, old_model_saved_name=None) -> ModelResponse:
+    def modify_model(self, config: ModelConfig, old_model_saved_name=None, old_model_version=None) -> ModelResponse:
         """
         Modify an existing model in the system.
 
         Args:
-            model_saved_name (str): The saved name of the model to modify
+            old_model_saved_name (str): The old saved name of the model to modify
+            old_model_version (str): The old version of the model to modify
             config (ModelConfig): Configuration object containing model details
 
         Returns:
@@ -137,7 +145,10 @@ class ModelClient(BaseClient):
         if old_model_saved_name is None:
             old_model_saved_name = config["model_saved_name"]
 
-        headers = {"Content-Type": "application/json", "X-Enkrypt-Model": old_model_saved_name}
+        if old_model_version is None:
+            old_model_version = config["model_version"]
+
+        headers = {"Content-Type": "application/json", "X-Enkrypt-Model": old_model_saved_name, "X-Enkrypt-Model-Version": old_model_version}
         # print(config)
         config = ModelConfig.from_dict(config)
         # Parse endpoint_url into components
@@ -165,17 +176,15 @@ class ModelClient(BaseClient):
 
         payload = {
             "model_saved_name": config.model_saved_name,
+            "model_version": config.model_version,
             "testing_for": config.testing_for,
             "model_name": config.model_name,
-            "model_type": config.model_type,
             "certifications": config.certifications,
             "model_config": {
                 "model_provider": config.model_config.model_provider,
-                "model_version": config.model_config.model_version,
                 "hosting_type": config.model_config.hosting_type,
                 "model_source": config.model_config.model_source,
                 "system_prompt": config.model_config.system_prompt,
-                "conversation_template": config.model_config.conversation_template,
                 "endpoint": {
                     "scheme": parsed_url.scheme,
                     "host": parsed_url.hostname,
@@ -192,6 +201,14 @@ class ModelClient(BaseClient):
                 "apikeys": (
                     [config.model_config.apikey] if config.model_config.apikey else []
                 ),
+                "tools": config.model_config.tools,
+                "input_modalities": config.model_config.input_modalities,
+                "output_modalities": config.model_config.output_modalities,
+                "custom_headers": config.model_config.custom_headers,
+                "custom_payload": config.model_config.custom_payload,
+                "custom_response_content_type": config.model_config.custom_response_content_type,
+                "custom_response_format": config.model_config.custom_response_format,
+                "metadata": config.model_config.metadata,
                 "default_request_options": config.model_config.default_request_options,
             },
         }
@@ -205,17 +222,18 @@ class ModelClient(BaseClient):
         except Exception as e:
             raise ModelClientError(str(e))
 
-    def delete_model(self, model_saved_name: str) -> ModelResponse:
+    def delete_model(self, model_saved_name: str, model_version: str) -> ModelResponse:
         """
         Delete a specific model from the system.
 
         Args:
             model_saved_name (str): The saved name of the model to delete
+            model_version (str): The version of the model to delete
 
         Returns:
             dict: Response from the API containing the deletion status
         """
-        headers = {"X-Enkrypt-Model": model_saved_name}
+        headers = {"X-Enkrypt-Model": model_saved_name, "X-Enkrypt-Model-Version": model_version}
 
         try:
             response = self._request("DELETE", "/models/delete-model", headers=headers)

enkryptai_sdk/red_team.py

@@ -66,13 +66,14 @@ class RedTeamClient(BaseClient):
         except Exception as e:
             raise RedTeamClientError(str(e))
         
-    def check_saved_model_health(self, model_saved_name: str):
+    def check_saved_model_health(self, model_saved_name: str, model_version: str):
         """
         Get the health status of a saved model.
         """
         try:
             headers = {
                 "X-Enkrypt-Model": model_saved_name,
+                "X-Enkrypt-Model-Version": model_version,
             }
             response = self._request("POST", "/redteam/model/model-health", headers=headers)
             # if response.get("error"):
@@ -125,12 +126,16 @@ class RedTeamClient(BaseClient):
         self,
         config: RedTeamConfigWithSavedModel,
         model_saved_name: str,
+        model_version: str,
     ):
         """
         Add a new red teaming task using a saved model.
         """
         if not model_saved_name:
             raise RedTeamClientError("Please provide a model_saved_name")
+        
+        if not model_version:
+            raise RedTeamClientError("Please provide a model_version. Default is 'v1'")
 
         config = RedTeamConfigWithSavedModel.from_dict(config)
         test_configs = config.redteam_test_configurations.to_dict()
@@ -148,6 +153,7 @@ class RedTeamClient(BaseClient):
 
         headers = {
             "X-Enkrypt-Model": model_saved_name,
+            "X-Enkrypt-Model-Version": model_version,
             "Content-Type": "application/json",
         }
         response = self._request(

{enkryptai_sdk-1.0.4.dist-info → enkryptai_sdk-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.4
+Version: 1.0.6
 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,9 +21,10 @@ 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
 
+![Python SDK test](https://github.com/enkryptai/enkryptai-sdk/actions/workflows/test.yaml/badge.svg)
+
 A Python SDK with Guardrails, Models, Deployments, AI Proxy, Datasets and Red Team functionality for API interactions.
 
 See [https://pypi.org/project/enkryptai-sdk](https://pypi.org/project/enkryptai-sdk)
@@ -55,6 +56,7 @@ Also see the API documentation at [https://docs.enkryptai.com](https://docs.enkr
 - [Guardrails Quickstart](#guardrails-quickstart)
 - [Guardrails Response Objects](#guardrails-response-objects)
   - [GuardrailsDetectResponse](#guardrailsdetectresponse)
+  - [GuardrailsBatchDetectResponse](#guardrailsbatchdetectresponse)
 - [Available Guardrails Detectors](#available-guardrails-detectors)
 - [Guardrails Configs](#guardrails-configs)
   - [Injection Attack](#injection-attack)
@@ -181,6 +183,7 @@ redteam_client = RedTeamClient(api_key=ENKRYPT_API_KEY, base_url=ENKRYPT_BASE_UR
 ```python Python
 test_policy_name = "Test Policy"
 test_model_saved_name = "Test Model"
+test_model_version = "v1"
 test_deployment_name = "test-deployment"
 
 pii_original_text = "My email is example@example.com. My phone number is 123-456-7890."
@@ -257,14 +260,15 @@ sample_detectors = {
 ```python Python
 sample_model_config = {
         "model_saved_name": test_model_saved_name,
-        "testing_for": "LLM",
+        "model_version": test_model_version,
+        "testing_for": "foundationModels",
         "model_name": model_name,
-        "modality": "text",
         "model_config": {
-            "model_version": "",
             "model_provider": model_provider,
             "endpoint_url": model_endpoint_url,
             "apikey": OPENAI_API_KEY,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"],
         },
     }
 ```
@@ -275,6 +279,7 @@ sample_model_config = {
 sample_deployment_config = {
         "name": test_deployment_name,
         "model_saved_name": test_model_saved_name,
+        "model_version": test_model_version,
         "input_guardrails_policy": {
             "policy_name": test_policy_name,
             "enabled": True,
@@ -308,8 +313,16 @@ sample_dataset_config = {
         "dataset_name": dataset_name,
         "system_description": "- **Voter Eligibility**: To vote in U.S. elections, individuals must be U.S. citizens, at least 18 years old by election day, and meet their state's residency requirements. - **Voter Registration**: Most states require voters to register ahead of time, with deadlines varying widely. North Dakota is an exception, as it does not require voter registration. - **Identification Requirements**: Thirty-six states enforce voter ID laws, requiring individuals to present identification at polling places. These laws aim to prevent voter fraud but can also lead to disenfranchisement. - **Voting Methods**: Voters can typically choose between in-person voting on election day, early voting, and absentee or mail-in ballots, depending on state regulations. - **Polling Hours**: Polling hours vary by state, with some states allowing extended hours for voters. Its essential for voters to check local polling times to ensure they can cast their ballots. - **Provisional Ballots**: If there are questions about a voter's eligibility, they may be allowed to cast a provisional ballot. This ballot is counted once eligibility is confirmed. - **Election Day Laws**: Many states have laws that protect the rights of voters on election day, including prohibiting intimidation and ensuring access to polling places. - **Campaign Finance Regulations**: Federal and state laws regulate contributions to candidates and political parties to ensure transparency and limit the influence of money in politics. - **Political Advertising**: Campaigns must adhere to rules regarding political advertising, including disclosure requirements about funding sources and content accuracy. - **Voter Intimidation Prohibitions**: Federal laws prohibit any form of voter intimidation or coercion at polling places, ensuring a safe environment for all voters. - **Accessibility Requirements**: The Americans with Disabilities Act mandates that polling places be accessible to individuals with disabilities, ensuring equal access to the electoral process. - **Election Monitoring**: Various organizations are allowed to monitor elections to ensure compliance with laws and regulations. They help maintain transparency and accountability in the electoral process. - **Vote Counting Procedures**: States have specific procedures for counting votes, including the use of electronic voting machines and manual audits to verify results. - **Ballot Design Standards**: States must adhere to certain design standards for ballots to ensure clarity and prevent confusion among voters when casting their votes. - **Post-Election Audits**: Some states conduct post-election audits as a measure of accuracy. These audits help verify that the vote count reflects the actual ballots cast.",
         "policy_description": "",
-        "tools": [],
+        "tools": [
+            {
+                "name": "web_search",
+                "description": "The tool web search is used to search the web for information related to finance."
+            }
+        ],
         "info_pdf_url": "",
+        "scenarios": 1,
+        "categories": 1,
+        "depth": 1,
         "max_prompts": 100,
     }
 ```
@@ -320,16 +333,16 @@ sample_dataset_config = {
 sample_redteam_model_health_config = {
         "target_model_configuration": {
             "model_name": model_name,
-            "testing_for": "LLM",
-            "model_type": "text_2_text",
-            "model_version": "v1",
+            "testing_for": "foundationModels",
+            "model_version": test_model_version,
             "model_source": "https://openai.com",
             "model_provider": model_provider,
             "model_endpoint_url": model_endpoint_url,
             "model_api_key": OPENAI_API_KEY,
             "system_prompt": "",
-            "conversation_template": "",
-            "rate_per_min": 20
+            "rate_per_min": 20,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"]
         },
     }
 ```
@@ -364,16 +377,16 @@ sample_redteam_target_config = {
         },
         "target_model_configuration": {
             "model_name": model_name,
-            "testing_for": "LLM",
-            "model_type": "text_2_text",
-            "model_version": "v1",
+            "testing_for": "foundationModels",
+            "model_version": test_model_version,
             "model_source": "https://openai.com",
             "model_provider": model_provider,
             "model_endpoint_url": model_endpoint_url,
             "model_api_key": OPENAI_API_KEY,
             "system_prompt": "",
-            "conversation_template": "",
-            "rate_per_min": 20
+            "rate_per_min": 20,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"]
         },
     }
 ```
@@ -539,6 +552,43 @@ print(detect_response)
 print(detect_response.to_dict())
 ```
 
+### GuardrailsBatchDetectResponse
+
+The `GuardrailsBatchDetectResponse` class wraps `batch_detect` response:
+
+```python Python
+# Example usage of batch_detect with multiple texts
+batch_detect_response = guardrails_client.batch_detect(
+    texts=[safe_prompt, bomb_prompt], 
+    config=copy.deepcopy(sample_detectors)
+)
+
+# Batch checks
+print(f"Batch Response Is Safe: {batch_detect_response.is_safe()}")
+print(f"Batch Response Is Attack: {batch_detect_response.is_attack()}")
+print(f"Batch Response Has Violations: {batch_detect_response.has_violations()}")
+print(f"Batch Response All Violations: {batch_detect_response.get_violations()}")
+
+# Access results for individual texts
+for idx, detection in enumerate(batch_detect_response.batch_detections):
+    print(f"\nResults for text #{idx + 1}:")
+    
+    # Access specific detector results
+    if detection.details.injection_attack:
+        print(f"Injection Attack Safe: {detection.details.injection_attack.safe}")
+        print(f"Injection Attack Score: {detection.details.injection_attack.attack}")
+    
+    # Check safety status for this text
+    print(f"Is Safe: {detection.is_safe()}")
+    print(f"Is Attack: {detection.is_attack()}")
+    print(f"Has Violations: {detection.has_violations()}")
+    print(f"Violations: {detection.get_violations()}")
+
+# Convert entire batch response to dictionary
+print("\nComplete Batch Response Dictionary:")
+print(batch_detect_response.to_dict())
+```
+
 ## Available Guardrails Detectors
 
 - `injection_attack`: Detect prompt injection attempts
@@ -890,7 +940,7 @@ print(add_model_response.to_dict())
 
 ```python Python
 # Check Model Health
-check_saved_model_health = redteam_client.check_saved_model_health(model_saved_name=test_model_saved_name)
+check_saved_model_health = redteam_client.check_saved_model_health(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(check_saved_model_health)
 
@@ -901,11 +951,13 @@ assert check_saved_model_health.status == "healthy"
 
 ```python Python
 # Retrieve model details
-model_details = model_client.get_model(model_saved_name=test_model_saved_name)
+model_details = model_client.get_model(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(model_details)
 
 # Get other fields
+print(model_details.model_saved_name)
+print(model_details.model_version)
 print(model_details.model_name)
 print(model_details.model_config)
 print(model_details.model_config.model_provider)
@@ -940,13 +992,23 @@ new_model_config = copy.deepcopy(sample_model_config)
 new_model_config["model_name"] = "gpt-4o-mini"
 
 # Update the model_saved_name if needed
+# ---------------------------------------------------
+# NOTE:
+# To ensure current stuff does not break, please try creating a new model instead of modifying the existing one.
+# Later, you can delete the old model when changes are made to use the new model.
+# ---------------------------------------------------
 # new_model_config["model_saved_name"] = "New Model Name"
+# new_model_config["model_version"] = "v2"
 
 old_model_saved_name = None
 if new_model_config["model_saved_name"] != test_model_saved_name:
     old_model_saved_name = test_model_saved_name
 
-modify_response = model_client.modify_model(old_model_saved_name=old_model_saved_name, config=new_model_config)
+old_model_version = None
+if new_model_config["model_version"] != test_model_version:
+    old_model_version = test_model_version
+
+modify_response = model_client.modify_model(old_model_saved_name=old_model_saved_name, old_model_version=old_model_version, config=new_model_config)
 
 print(modify_response)
 
@@ -960,7 +1022,7 @@ print(modify_response.to_dict())
 
 ```python Python
 # Remove a model
-delete_response = model_client.delete_model(model_saved_name=test_model_saved_name)
+delete_response = model_client.delete_model(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(delete_response)
 
@@ -996,6 +1058,7 @@ print(deployment_details)
 
 # Get other fields
 print(deployment_details.model_saved_name)
+print(deployment_details.model_version)
 print(deployment_details.input_guardrails_policy)
 print(deployment_details.input_guardrails_policy.policy_name)
 
@@ -1219,7 +1282,7 @@ print(add_redteam_target_response.to_dict())
 
 ```python Python
 # Use a dictionary to configure a redteam task
-add_redteam_model_response = redteam_client.add_task_with_saved_model(config=copy.deepcopy(sample_redteam_model_config),model_saved_name=test_model_saved_name)
+add_redteam_model_response = redteam_client.add_task_with_saved_model(config=copy.deepcopy(sample_redteam_model_config),model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(add_redteam_model_response)
 

enkryptai_sdk-1.0.6.dist-info/RECORD

@@ -0,0 +1,25 @@
+enkryptai_sdk/__init__.py,sha256=rP6PtntJogJauj1lKWK8DkiBr3uYjireIUamr6aflu0,763
+enkryptai_sdk/ai_proxy.py,sha256=pD6kPmD9H0gttN28cezgV7_IVelLXAHNR5cPeXCM8Ew,3799
+enkryptai_sdk/base.py,sha256=MlEDcEIjXo35kat9XkGUu7VB2fIvJk38C94wAeO9bEw,1304
+enkryptai_sdk/config.py,sha256=IpB8_aO4zXdvv061v24oh83oyJ5Tp1QBQTzeuW4h9QY,8828
+enkryptai_sdk/datasets.py,sha256=2rAU1qeqBZz6rXxG4n9XM_8_ZQYNmcNFW-jgyw7nLPI,5364
+enkryptai_sdk/deployments.py,sha256=qXoUQtLYRSqZsAAqPgxJAad1a3mgMAbPx09NiOyAdsw,4155
+enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
+enkryptai_sdk/guardrails.py,sha256=meR4iZuGV5Js9BRU0edJ82CGOrZMroaTcoaEoUn7k-s,13642
+enkryptai_sdk/guardrails_old.py,sha256=SgzPZkTzbAPD9XfmYNG6M1-TrzbhDHpAkI3FjnVWS_s,6434
+enkryptai_sdk/models.py,sha256=Bj07UAhy2yWAgDxOeInkCTqxDmJK5Ae74v7rNlkA6V4,10361
+enkryptai_sdk/red_team.py,sha256=Puo6BvYDrbERXTuSCFXLtXKhXcFiHuJrcRU5akvXps4,14479
+enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
+enkryptai_sdk/dto/__init__.py,sha256=vG5PmG-L99v89AGiKF1QAF63YILk-e9w0dd665M3U60,2414
+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=nI0VjgkViaAi49x01dTDfkzoynx5__zCfJLUsZG8kyw,4846
+enkryptai_sdk/dto/deployments.py,sha256=Aw4b8tDA3FYIomqDvCjblCXTagL4bT8Fx91X0SFXs40,11216
+enkryptai_sdk/dto/guardrails.py,sha256=uolGPPF4v0l76MI5G0ofTtc9-r1l0_sqQqQkLEhAsf0,46305
+enkryptai_sdk/dto/models.py,sha256=6BYF4XwzUwY8O_YgE9S7LPAvsQaRYtUV6ylkTYP7w10,13438
+enkryptai_sdk/dto/red_team.py,sha256=4VGxDGCsZsHi6yJsUZ64IXj6u84xZPcSQuME1PFSnSc,15243
+enkryptai_sdk-1.0.6.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.6.dist-info/METADATA,sha256=8htBFgQ36iHitRe08rcvEgB9nnu1pXWBoC3G_i0hdss,45959
+enkryptai_sdk-1.0.6.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+enkryptai_sdk-1.0.6.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.6.dist-info/RECORD,,

enkryptai_sdk-1.0.4.dist-info/RECORD

@@ -1,25 +0,0 @@
-enkryptai_sdk/__init__.py,sha256=rP6PtntJogJauj1lKWK8DkiBr3uYjireIUamr6aflu0,763
-enkryptai_sdk/ai_proxy.py,sha256=pD6kPmD9H0gttN28cezgV7_IVelLXAHNR5cPeXCM8Ew,3799
-enkryptai_sdk/base.py,sha256=MlEDcEIjXo35kat9XkGUu7VB2fIvJk38C94wAeO9bEw,1304
-enkryptai_sdk/config.py,sha256=IpB8_aO4zXdvv061v24oh83oyJ5Tp1QBQTzeuW4h9QY,8828
-enkryptai_sdk/datasets.py,sha256=xekcdY9wIniw2TnylaK6o1RNZ5DRoluNOGawBkVgaM0,4881
-enkryptai_sdk/deployments.py,sha256=qXoUQtLYRSqZsAAqPgxJAad1a3mgMAbPx09NiOyAdsw,4155
-enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
-enkryptai_sdk/guardrails.py,sha256=CtQwPBmpFFy71P22kG0gJbefYJflUT9aareYksPhNlQ,12096
-enkryptai_sdk/guardrails_old.py,sha256=SgzPZkTzbAPD9XfmYNG6M1-TrzbhDHpAkI3FjnVWS_s,6434
-enkryptai_sdk/models.py,sha256=FSdpfz-e1a0D7nuYCnQC859AdrOUVr60QbLR1mlZjp0,8978
-enkryptai_sdk/red_team.py,sha256=jlSf7oNPM7jJHXyv37ROQgkluuigxbSpeeAfvKqUiGI,14192
-enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
-enkryptai_sdk/dto/__init__.py,sha256=kKBw4rkfqMBuK8nXRDtD6Sd0_uqLKgbcHrqzuSGJpr0,2310
-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.4.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.4.dist-info/METADATA,sha256=2lqT-652SgRDCyjpMhmRfyUiF6dK1uL-kl0i1RbKhsc,43207
-enkryptai_sdk-1.0.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-enkryptai_sdk-1.0.4.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.4.dist-info/RECORD,,