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

enkryptai_sdk/ai_proxy.py

@@ -60,7 +60,7 @@ class AIProxyClient(BaseClient):
                     # # 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"])
+                    parsed_error = ast.literal_eval(f"API Error: {str(response)}")
                     # # Preserve both error and enkrypt_policy_detections fields
                     response["error"] = parsed_error.get("error", parsed_error)
                     if "enkrypt_policy_detections" in parsed_error:
@@ -84,6 +84,6 @@ class AIProxyClient(BaseClient):
                     # 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"])
+            raise AIProxyClientError(f"API Error: {str(response)}")
 
         return ChatCompletionResponse.from_dict(response)

enkryptai_sdk/datasets.py

@@ -19,6 +19,32 @@ class DatasetClient(BaseClient):
     def __init__(self, api_key: str, base_url: str = "https://api.enkryptai.com"):
         super().__init__(api_key, base_url)
 
+    @staticmethod
+    def prepare_dataset_payload(config: DatasetConfig | dict, is_custom: bool = False) -> dict:
+        """
+        Prepare the payload for dataset operations from a config object.
+        
+        Args:
+            config (Union[DatasetConfig, dict]): Configuration object or dictionary containing dataset details
+            
+        Returns:
+            dict: Processed payload ready for API submission
+        """
+        if isinstance(config, dict):
+            config = DatasetConfig.from_dict(config)
+        
+        payload = config.to_dict()
+
+        if not is_custom:
+            # Remove empty tools configuration
+            if (payload.get("tools") is None or 
+                payload["tools"] == [] or 
+                payload["tools"] == [{}] or
+                payload["tools"] == [{"name": "", "description": ""}]):
+                del payload["tools"]
+        
+        return payload
+
     def add_dataset(self, config: DatasetConfig):
         """
         Add a new dataset to the system.
@@ -34,14 +60,7 @@ class DatasetClient(BaseClient):
         if isinstance(config, dict):
             config = DatasetConfig.from_dict(config)
 
-        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"]
+        payload = self.prepare_dataset_payload(config)
 
         # Print payload
         # print(f"\nAdd Dataset Payload: {payload}")
@@ -54,7 +73,7 @@ class DatasetClient(BaseClient):
         # print(f"\nAdd Dataset Response: {response}")
 
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         return DatasetAddTaskResponse.from_dict(response)
     
     def get_dataset_task_status(self, dataset_name: str):
@@ -70,7 +89,7 @@ class DatasetClient(BaseClient):
         headers = {"X-Enkrypt-Dataset": dataset_name}
         response = self._request("GET", "/datasets/task-status", headers=headers)
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         response["dataset_name"] = dataset_name
         return DatasetTaskStatus.from_dict(response)
 
@@ -87,7 +106,7 @@ class DatasetClient(BaseClient):
         headers = {"X-Enkrypt-Dataset": dataset_name}
         response = self._request("GET", "/datasets/get-task", headers=headers)
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         response["dataset_name"] = dataset_name
         return DatasetTask.from_dict(response)
 
@@ -104,7 +123,7 @@ class DatasetClient(BaseClient):
         headers = {"X-Enkrypt-Dataset": dataset_name}
         response = self._request("GET", "/datasets/get-datacard", headers=headers)
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         response["dataset_name"] = dataset_name
         return DatasetCard.from_dict(response)
 
@@ -121,7 +140,7 @@ class DatasetClient(BaseClient):
         headers = {"X-Enkrypt-Dataset": dataset_name}
         response = self._request("GET", "/datasets/get-dataset", headers=headers)
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         response["dataset_name"] = dataset_name
         return DatasetResponse.from_dict(response)
 
@@ -138,7 +157,7 @@ class DatasetClient(BaseClient):
         headers = {"X-Enkrypt-Dataset": dataset_name}
         response = self._request("GET", "/datasets/get-summary", headers=headers)
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         response["dataset_name"] = dataset_name
         return DatasetSummary.from_dict(response)
 
@@ -158,5 +177,5 @@ class DatasetClient(BaseClient):
             url += f"?status={status}"
         response = self._request("GET", url)
         if response.get("error"):
-            raise DatasetClientError(response["error"])
+            raise DatasetClientError(f"API Error: {str(response)}")
         return DatasetCollection.from_dict(response)

enkryptai_sdk/deployments.py

@@ -46,7 +46,7 @@ class DeploymentClient(BaseClient):
             "POST", "/deployments/add-deployment", headers=headers, json=payload
         )
         if response.get("error"):
-            raise DeploymentClientError(response["error"])
+            raise DeploymentClientError(f"API Error: {str(response)}")
         return DeploymentAddTaskResponse.from_dict(response)
 
     def get_deployment(self, deployment_name: str, refresh_cache: bool = False):
@@ -64,7 +64,7 @@ class DeploymentClient(BaseClient):
         headers["X-Enkrypt-Refresh-Cache"] = "true" if refresh_cache else "false"
         response = self._request("GET", "/deployments/get-deployment", headers=headers)
         if response.get("error"):
-            raise DeploymentClientError(response["error"])
+            raise DeploymentClientError(f"API Error: {str(response)}")
         return GetDeploymentResponse.from_dict(response)
     
     def modify_deployment(self, deployment_name: str, config: DeploymentInput):
@@ -88,7 +88,7 @@ class DeploymentClient(BaseClient):
             "PATCH", "/deployments/modify-deployment", headers=headers, json=payload
         )
         if response.get("error"):
-            raise DeploymentClientError(response["error"])
+            raise DeploymentClientError(f"API Error: {str(response)}")
         return ModifyDeploymentResponse.from_dict(response)
 
     def delete_deployment(self, deployment_name: str):
@@ -104,7 +104,7 @@ class DeploymentClient(BaseClient):
         headers = {"X-Enkrypt-Deployment": deployment_name}
         response = self._request("DELETE", "/deployments/delete-deployment", headers=headers)
         if response.get("error"):
-            raise DeploymentClientError(response["error"])
+            raise DeploymentClientError(f"API Error: {str(response)}")
         return DeleteDeploymentResponse.from_dict(response)
 
     def list_deployments(self):
@@ -117,5 +117,5 @@ class DeploymentClient(BaseClient):
 
         response = self._request("GET", "/deployments/list-deployments")
         if response.get("error"):
-            raise DeploymentClientError(response["error"])
+            raise DeploymentClientError(f"API Error: {str(response)}")
         return DeploymentCollection.from_dict(response)

enkryptai_sdk/dto/__init__.py

@@ -22,6 +22,8 @@ __all__ = [
     "Metadata",
     "DEFAULT_REDTEAM_CONFIG",
     "DEFAULT_REDTEAM_CONFIG_WITH_SAVED_MODEL",
+    "DEFAULT_CUSTOM_REDTEAM_CONFIG",
+    "DEFAULT_CUSTOM_REDTEAM_CONFIG_WITH_SAVED_MODEL",
     "ADVANCED_REDTEAM_TESTS",
     "DETAIL_MODEL_CONFIG",
     "DatasetConfig",

enkryptai_sdk/dto/datasets.py

@@ -21,17 +21,20 @@ class Tool(BaseDTO):
             "description": self.description
         }
 
+
 @dataclass
 class DatasetConfig(BaseDTO):
-    dataset_name: str
-    system_description: str
-    policy_description: str = ""
-    tools: List[Tool] = field(default_factory=list)
-    info_pdf_url: str = ""
+    system_description: str = None
+    dataset_name: str = None
+    policy_description: str = None
+    risk_categories: str = None
+    tools: List[Tool] = None
+    info_pdf_url: str = None
     max_prompts: int = 100
     scenarios: int = 2
     categories: int = 2
     depth: int = 2
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
 
 @dataclass
@@ -51,6 +54,7 @@ class DatasetDataPoint(BaseDTO):
     category: str
     prompt: str
     source: str
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
 
 @dataclass

enkryptai_sdk/dto/models.py

@@ -142,8 +142,8 @@ class CustomHeader(BaseDTO):
 
 @dataclass
 class ModelConfigDetails(BaseDTO):
-    model_id: str = ""
-    model_source: str = ""
+    model_id: str = None
+    model_source: str = None
     # model_provider: str = "openai"
     model_provider: ModelProviders = ModelProviders.OPENAI
     model_api_value: str = ""
@@ -259,14 +259,14 @@ class ModelConfigDetails(BaseDTO):
 
 @dataclass
 class ModelConfig(BaseDTO):
-    created_at: str = ""
-    updated_at: str = ""
-    model_id: str = ""
-    model_saved_name: str = "Model Name"
-    model_version: str = "v1"
+    created_at: str = None
+    updated_at: str = None
+    model_id: str = None
+    model_saved_name: str = None
+    model_version: str = None
     testing_for: str = "foundationModels"
     # modality: Modality = Modality.TEXT
-    project_name: str = ""
+    project_name: str = None
     model_name: Optional[str] = "gpt-4o-mini"
     certifications: List[str] = field(default_factory=list)
     model_config: ModelConfigDetails = field(default_factory=ModelConfigDetails)

enkryptai_sdk/dto/red_team.py

@@ -3,6 +3,8 @@ from enum import Enum
 from .base import BaseDTO
 from typing import Dict, List, Optional, Any
 from dataclasses import dataclass, field, asdict
+from .datasets import DatasetConfig
+from .models import ModelConfig
 
 
 @dataclass
@@ -24,6 +26,7 @@ class RedteamHealthResponse(BaseDTO):
 
 @dataclass
 class RedTeamResponse(BaseDTO):
+    status: Optional[str] = None
     task_id: Optional[str] = None
     message: Optional[str] = None
     data: Optional[Dict[str, Any]] = None
@@ -300,7 +303,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)
+    custom_test: TestConfig = field(default=None)
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
 
     @classmethod
@@ -482,6 +485,83 @@ class RedTeamConfigWithSavedModel(BaseDTO):
         )
 
 
+@dataclass
+class RedTeamCustomConfig(BaseDTO):
+    test_name: str = "Test Name"
+
+    redteam_test_configurations: RedTeamTestConfigurations = field(
+        default_factory=RedTeamTestConfigurations
+    )
+    dataset_configuration: DatasetConfig = field(
+        default_factory=DatasetConfig
+    )
+    endpoint_configuration: ModelConfig = field(
+        default_factory=ModelConfig
+    )
+
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+
+    def to_dict(self) -> dict:
+        d = asdict(self)
+        d["redteam_test_configurations"] = self.redteam_test_configurations.to_dict()
+        d["dataset_configuration"] = self.dataset_configuration.to_dict()
+        d["endpoint_configuration"] = self.endpoint_configuration.to_dict()
+        return d
+
+    @classmethod
+    def from_dict(cls, data: dict):
+        data = data.copy()
+        test_configs = RedTeamTestConfigurations.from_dict(
+            data.pop("redteam_test_configurations", {})
+        )
+        dataset_config = DatasetConfig.from_dict(
+            data.pop("dataset_configuration", {})
+        )
+        endpoint_config = ModelConfig.from_dict(
+            data.pop("endpoint_configuration", {})
+        )
+        return cls(
+            **data,
+            redteam_test_configurations=test_configs,
+            dataset_configuration=dataset_config,
+            endpoint_configuration=endpoint_config,
+        )
+
+@dataclass
+class RedTeamCustomConfigWithSavedModel(BaseDTO):
+    test_name: str = "Test Name"
+
+    redteam_test_configurations: RedTeamTestConfigurations = field(
+        default_factory=RedTeamTestConfigurations
+    )
+    dataset_configuration: DatasetConfig = field(
+        default_factory=DatasetConfig
+    )
+
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+
+    def to_dict(self) -> dict:
+        d = asdict(self)
+        d["redteam_test_configurations"] = self.redteam_test_configurations.to_dict()
+        d["dataset_configuration"] = self.dataset_configuration.to_dict()
+        return d
+
+    @classmethod
+    def from_dict(cls, data: dict):
+        data = data.copy()
+        test_configs = RedTeamTestConfigurations.from_dict(
+            data.pop("redteam_test_configurations", {})
+        )
+        dataset_config = DatasetConfig.from_dict(
+            data.pop("dataset_configuration", {})
+        )
+        return cls(
+            **data,
+            redteam_test_configurations=test_configs,
+            dataset_configuration=dataset_config,
+        )
+
+
 @dataclass
 class RedTeamTaskList(BaseDTO):
     tasks: List[RedTeamTaskDetails] = field(default_factory=list)
@@ -494,3 +574,6 @@ class RedTeamTaskList(BaseDTO):
 # Default configurations
 DEFAULT_REDTEAM_CONFIG = RedTeamConfig()
 DEFAULT_REDTEAM_CONFIG_WITH_SAVED_MODEL = RedTeamConfigWithSavedModel()
+
+DEFAULT_CUSTOM_REDTEAM_CONFIG = RedTeamCustomConfig()
+DEFAULT_CUSTOM_REDTEAM_CONFIG_WITH_SAVED_MODEL = RedTeamCustomConfigWithSavedModel()

enkryptai_sdk/guardrails.py

@@ -92,7 +92,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("GET", "/guardrails/health")
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsHealthResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -104,7 +104,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("GET", "/guardrails/status")
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsHealthResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -116,7 +116,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("GET", "/guardrails/models")
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsModelsResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -152,7 +152,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/detect", json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsDetectResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -188,7 +188,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/batch/detect", json=payload)
             if isinstance(response, dict) and response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsBatchDetectResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -207,7 +207,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/pii", json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsPIIResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -225,7 +225,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/hallucination", json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsHallucinationResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -242,7 +242,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/adherence", json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsAdherenceResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -259,7 +259,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/relevancy", json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsRelevancyResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -293,7 +293,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/add-policy", json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsaPolicyResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -307,7 +307,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("GET", "/guardrails/get-policy", headers=headers)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsPolicyData.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -335,7 +335,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("PATCH", "/guardrails/modify-policy", headers=headers, json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsaPolicyResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -349,7 +349,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("DELETE", "/guardrails/delete-policy", headers=headers)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsDeletePolicyResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -364,7 +364,7 @@ class GuardrailsClient(BaseClient):
         try:
             response = self._request("POST", "/guardrails/policy/detect", headers=headers, json=payload)
             if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsDetectResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))
@@ -376,8 +376,8 @@ class GuardrailsClient(BaseClient):
 
         try:
             response = self._request("GET", "/guardrails/list-policies")
-            if response.get("error"):
-                raise GuardrailsClientError(response["error"])
+            if isinstance(response, dict) and response.get("error"):
+                raise GuardrailsClientError(f"API Error: {str(response)}")
             return GuardrailsListPoliciesResponse.from_dict(response)
         except Exception as e:
             raise GuardrailsClientError(str(e))

enkryptai_sdk/models.py

@@ -15,18 +15,20 @@ class ModelClient(BaseClient):
     def __init__(self, api_key: str, base_url: str = "https://api.enkryptai.com:443"):
         super().__init__(api_key, base_url)
 
-    def add_model(self, config: ModelConfig) -> ModelResponse:
+    @staticmethod
+    def prepare_model_payload(config: ModelConfig | dict, is_custom: bool = False) -> dict:
         """
-        Add a new model configuration to the system.
-
+        Prepare the payload for model operations from a config object.
+        
         Args:
-            config (ModelConfig): Configuration object containing model details
-
+            config (Union[ModelConfig, dict]): Configuration object or dictionary containing model details
+            
         Returns:
-            dict: Response from the API containing the added model details
+            dict: Processed payload ready for API submission
         """
-        headers = {"Content-Type": "application/json"}
-        config = ModelConfig.from_dict(config)
+        if isinstance(config, dict):
+            config = ModelConfig.from_dict(config)
+        
         # Parse endpoint_url into components
         parsed_url = urlparse(config.model_config.endpoint_url)
         path_parts = parsed_url.path.strip("/").split("/")
@@ -43,15 +45,11 @@ class ModelClient(BaseClient):
             paths = config.model_config.paths.to_dict()
         else:
             paths = {
-                "completions": (
-                    f"/{remaining_path.split('/')[-1]}" if remaining_path else ""
-                ),
+                "completions": f"/{remaining_path.split('/')[-1]}" if remaining_path else "",
                 "chat": f"/{remaining_path}" if remaining_path else "",
             }
 
         payload = {
-            "model_saved_name": config.model_saved_name,
-            "model_version": config.model_version,
             "testing_for": config.testing_for,
             "model_name": config.model_name,
             "certifications": config.certifications,
@@ -63,9 +61,8 @@ class ModelClient(BaseClient):
                 "endpoint": {
                     "scheme": parsed_url.scheme,
                     "host": parsed_url.hostname,
-                    "port": parsed_url.port
-                    or (443 if parsed_url.scheme == "https" else 80),
-                    "base_path": f"/{base_path}",  # Just v1
+                    "port": parsed_url.port or (443 if parsed_url.scheme == "https" else 80),
+                    "base_path": f"/{base_path}",
                 },
                 "paths": paths,
                 "auth_data": {
@@ -73,9 +70,7 @@ class ModelClient(BaseClient):
                     "header_prefix": config.model_config.auth_data.header_prefix,
                     "space_after_prefix": config.model_config.auth_data.space_after_prefix,
                 },
-                "apikeys": (
-                    [config.model_config.apikey] if config.model_config.apikey else []
-                ),
+                "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,
@@ -87,12 +82,32 @@ class ModelClient(BaseClient):
                 "default_request_options": config.model_config.default_request_options,
             },
         }
+
+        if not is_custom:
+            payload["model_saved_name"] = config.model_saved_name
+            payload["model_version"] = config.model_version
+
+        return payload
+
+    def add_model(self, config: ModelConfig) -> ModelResponse:
+        """
+        Add a new model configuration to the system.
+
+        Args:
+            config (ModelConfig): Configuration object containing model details
+
+        Returns:
+            dict: Response from the API containing the added model details
+        """
+        headers = {"Content-Type": "application/json"}
+        payload = self.prepare_model_payload(config)
+
         try:
             response = self._request(
                 "POST", "/models/add-model", headers=headers, json=payload
             )
             if response.get("error"):
-                raise ModelClientError(response["error"])
+                raise ModelClientError(f"API Error: {str(response)}")
             return ModelResponse.from_dict(response)
         except Exception as e:
             raise ModelClientError(str(e))
@@ -112,7 +127,7 @@ class ModelClient(BaseClient):
         response = self._request("GET", "/models/get-model", headers=headers)
         # print(response)
         if response.get("error"):
-            raise ModelClientError(response["error"])
+            raise ModelClientError(f"API Error: {str(response)}")
         return ModelConfig.from_dict(response)
 
     def get_model_list(self):
@@ -124,8 +139,8 @@ class ModelClient(BaseClient):
         """
         try:
             response = self._request("GET", "/models/list-models")
-            if response.get("error"):
-                raise ModelClientError(response["error"])
+            if isinstance(response, dict) and response.get("error"):
+                raise ModelClientError(f"API Error: {str(response)}")
             return ModelCollection.from_dict(response)
         except Exception as e:
             return {"error": str(e)}
@@ -217,7 +232,7 @@ class ModelClient(BaseClient):
                 "PATCH", "/models/modify-model", headers=headers, json=payload
             )
             if response.get("error"):
-                raise ModelClientError(response["error"])
+                raise ModelClientError(f"API Error: {str(response)}")
             return ModelResponse.from_dict(response)
         except Exception as e:
             raise ModelClientError(str(e))
@@ -238,7 +253,7 @@ class ModelClient(BaseClient):
         try:
             response = self._request("DELETE", "/models/delete-model", headers=headers)
             if response.get("error"):
-                raise ModelClientError(response["error"])
+                raise ModelClientError(f"API Error: {str(response)}")
             return ModelResponse.from_dict(response)
         except Exception as e:
             raise ModelClientError(str(e))

enkryptai_sdk/red_team.py

@@ -1,11 +1,15 @@
 import urllib3
 from .base import BaseClient
+from .models import ModelClient
+from .datasets import DatasetClient
 from .dto import (
     RedteamHealthResponse,
     RedTeamModelHealthConfig,
     RedteamModelHealthResponse,
     RedTeamConfig,
     RedTeamConfigWithSavedModel,
+    RedTeamCustomConfig,
+    RedTeamCustomConfigWithSavedModel,
     RedTeamResponse,
     RedTeamResultSummary,
     RedTeamResultDetails,
@@ -47,7 +51,7 @@ class RedTeamClient(BaseClient):
         try:
             response = self._request("GET", "/redteam/health")
             if response.get("error"):
-                raise RedTeamClientError(response["error"])
+                raise RedTeamClientError(f"API Error: {str(response)}")
             return RedteamHealthResponse.from_dict(response)
         except Exception as e:
             raise RedTeamClientError(str(e))
@@ -61,7 +65,7 @@ class RedTeamClient(BaseClient):
             response = self._request("POST", "/redteam/model-health", json=config.to_dict())
             # if response.get("error"):
             if response.get("error") not in [None, ""]:
-                raise RedTeamClientError(response["error"])
+                raise RedTeamClientError(f"API Error: {str(response)}")
             return RedteamModelHealthResponse.from_dict(response)
         except Exception as e:
             raise RedTeamClientError(str(e))
@@ -78,7 +82,7 @@ class RedTeamClient(BaseClient):
             response = self._request("POST", "/redteam/model/model-health", headers=headers)
             # if response.get("error"):
             if response.get("error") not in [None, ""]:
-                raise RedTeamClientError(response["error"])
+                raise RedTeamClientError(f"API Error: {str(response)}")
             return RedteamModelHealthResponse.from_dict(response)
         except Exception as e:
             raise RedTeamClientError(str(e))
@@ -115,7 +119,7 @@ class RedTeamClient(BaseClient):
                 json=payload,
             )
             if response.get("error"):
-                raise RedTeamClientError(response["error"])
+                raise RedTeamClientError(f"API Error: {str(response)}")
             return RedTeamResponse.from_dict(response)
         else:
             raise RedTeamClientError(
@@ -163,7 +167,118 @@ class RedTeamClient(BaseClient):
             json=payload,
         )
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
+        return RedTeamResponse.from_dict(response)
+
+    def add_custom_task(
+        self,
+        config: RedTeamCustomConfig,
+        policy_name: str = None,
+    ):
+        """
+        Add a new custom red teaming task.
+        """
+        headers = {
+            "Content-Type": "application/json",
+        }
+
+        if policy_name is not None:
+            headers["X-Enkrypt-Policy"] = policy_name
+
+        config = RedTeamCustomConfig.from_dict(config)
+        test_configs = config.redteam_test_configurations.to_dict()
+        # Remove None or empty test configurations
+        test_configs = {k: v for k, v in test_configs.items() if v is not None}
+
+        payload = {
+            # "async": config.async_enabled,
+            "test_name": config.test_name,
+            "redteam_test_configurations": {
+                k: v.to_dict() for k, v in test_configs.items()
+            },
+        }
+
+        if config.dataset_configuration:
+            payload["dataset_configuration"] = DatasetClient.prepare_dataset_payload(
+                config.dataset_configuration, True)
+        else:
+            raise RedTeamClientError(
+                "Please provide a dataset configuration"
+            )
+
+        if config.endpoint_configuration:
+            payload["endpoint_configuration"] = ModelClient.prepare_model_payload(
+                config.endpoint_configuration, True)
+            # print(payload)
+
+            response = self._request(
+                "POST",
+                "/redteam/v2/add-custom-task",
+                headers=headers,
+                json=payload,
+            )
+            if response.get("error"):
+                raise RedTeamClientError(f"API Error: {str(response)}")
+            return RedTeamResponse.from_dict(response)
+        else:
+            raise RedTeamClientError(
+                "Please provide a endpoint configuration"
+            )
+
+    def add_custom_task_with_saved_model(
+        self,
+        config: RedTeamCustomConfigWithSavedModel,
+        model_saved_name: str,
+        model_version: str,
+        policy_name: str = None,
+    ):
+        """
+        Add a new red teaming custom 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 = RedTeamCustomConfigWithSavedModel.from_dict(config)
+        test_configs = config.redteam_test_configurations.to_dict()
+        # Remove None or empty test configurations
+        test_configs = {k: v for k, v in test_configs.items() if v is not None}
+
+        payload = {
+            # "async": config.async_enabled,
+            "test_name": config.test_name,
+            "redteam_test_configurations": {
+                k: v.to_dict() for k, v in test_configs.items()
+            },
+        }
+
+        if config.dataset_configuration:
+            payload["dataset_configuration"] = DatasetClient.prepare_dataset_payload(
+                config.dataset_configuration, True)
+        else:
+            raise RedTeamClientError(
+                "Please provide a dataset configuration"
+            )
+
+        headers = {
+            "X-Enkrypt-Model": model_saved_name,
+            "X-Enkrypt-Model-Version": model_version,
+            "Content-Type": "application/json",
+        }
+
+        if policy_name is not None:
+            headers["X-Enkrypt-Policy"] = policy_name
+
+        response = self._request(
+            "POST",
+            "/redteam/v2/model/add-custom-task",
+            headers=headers,
+            json=payload,
+        )
+        if response.get("error"):
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return RedTeamResponse.from_dict(response)
 
     def status(self, task_id: str = None, test_name: str = None):
@@ -191,7 +306,7 @@ class RedTeamClient(BaseClient):
 
         response = self._request("GET", "/redteam/task-status", headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return RedTeamTaskStatus.from_dict(response)
 
     def cancel_task(self, task_id: str = None, test_name: str = None):
@@ -224,7 +339,7 @@ class RedTeamClient(BaseClient):
 
         response = self._request("POST", "/redteam/cancel-task", headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return response
 
     def get_task(self, task_id: str = None, test_name: str = None):
@@ -252,7 +367,7 @@ class RedTeamClient(BaseClient):
 
         response = self._request("GET", "/redteam/get-task", headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return RedTeamTaskDetails.from_dict(response["data"])
 
     def get_result_summary(self, task_id: str = None, test_name: str = None):
@@ -280,7 +395,7 @@ class RedTeamClient(BaseClient):
 
         response = self._request("GET", "/redteam/results/summary", headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         # print(f"Response: {response}")
         return RedTeamResultSummary.from_dict(response)
     
@@ -314,7 +429,7 @@ class RedTeamClient(BaseClient):
         url = f"/redteam/v2/results/summary/{test_type}"
         response = self._request("GET", url, headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         # print(f"Response: {response}")
         return RedTeamResultSummary.from_dict(response)
 
@@ -343,7 +458,7 @@ class RedTeamClient(BaseClient):
             
         response = self._request("GET", "/redteam/v2/results/details", headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return RedTeamResultDetails.from_dict(response)
     
     def get_result_details_test_type(self, task_id: str = None, test_name: str = None, test_type: str = None):
@@ -377,7 +492,7 @@ class RedTeamClient(BaseClient):
         url = f"/redteam/v2/results/details/{test_type}"
         response = self._request("GET", url, headers=headers)
         if response.get("error"):
-            raise RedTeamClientError(response["error"])
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return RedTeamResultDetails.from_dict(response)
 
     def get_task_list(self, status: str = None):
@@ -395,6 +510,6 @@ class RedTeamClient(BaseClient):
             url += f"?status={status}"
 
         response = self._request("GET", url)
-        if response.get("error"):
-            raise RedTeamClientError(response["error"])
+        if isinstance(response, dict) and response.get("error"):
+            raise RedTeamClientError(f"API Error: {str(response)}")
         return RedTeamTaskList.from_dict(response)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.6
+Version: 1.0.8
 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
@@ -47,6 +47,8 @@ Also see the API documentation at [https://docs.enkryptai.com](https://docs.enkr
   - [Sample Redteam Model Health Config](#sample-redteam-model-health-config)
   - [Sample Redteam Target Config](#sample-redteam-target-config)
   - [Sample Redteam Model Config](#sample-redteam-model-config)
+  - [Sample Custom Redteam Target Config](#sample-custom-redteam-target-config)
+  - [Sample Custom Redteam Model Config](#sample-custom-redteam-model-config)
 - [Health Checks](#health-checks)
   - [Guardrails Health](#guardrails-health)
   - [Guardrails Status](#guardrails-status)
@@ -106,6 +108,8 @@ Also see the API documentation at [https://docs.enkryptai.com](https://docs.enkr
 - [Redteam](#redteam)
   - [Add a Redteam Task with Target Model Config](#add-a-redteam-task-with-target-model-config)
   - [Add a Redteam Task with a saved model](#add-a-redteam-task-with-a-saved-model)
+  - [Add a Redteam Custom Task with Endpoint Config](#add-a-redteam-custom-task-with-endpoint-config)
+  - [Add a Redteam Custom Task with a saved model](#add-a-redteam-custom-task-with-a-saved-model)
   - [Get Redteam Task Status](#get-redteam-task-status)
   - [Get Redteam Task](#get-redteam-task)
   - [List Redteam Tasks](#list-redteam-tasks)
@@ -193,6 +197,7 @@ pii_anonymized_text = None
 dataset_name = f"TestElectionDataset-{str(uuid.uuid4())[:8]}"
 
 redteam_test_name = f"Redteam Test {str(uuid.uuid4())[:6]}"
+custom_redteam_test_name = f"Custom Redteam Test {str(uuid.uuid4())[:6]}"
 
 model_name = "gpt-4o-mini"
 model_provider = "openai"
@@ -422,6 +427,82 @@ sample_redteam_model_config = {
     }
 ```
 
+### [Sample Custom Redteam Target Config](https://docs.enkryptai.com/redteam-api-reference/endpoint/add-custom-task)
+
+```python Python
+sample_custom_redteam_target_config = {
+        "test_name": custom_redteam_test_name,
+        "dataset_configuration": {
+          "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": "Do not generate any content that is illegal, harmful, or violates the rights of others.",
+          # "risk_categories": "",
+          "tools": [
+              {
+                  "name": "web_search",
+                  "description": "The tool web search is used to search the web for information related to finance."
+              }
+          ],
+          "max_prompts": 100,
+          "scenarios": 2,
+          "categories": 2,
+          "depth": 2,
+        },
+        "redteam_test_configurations": {
+            "harmful_test": {
+                "sample_percentage": 2,
+                "attack_methods": {"basic": ["basic"]},
+            },
+            "custom_test": {
+                "sample_percentage": 2,
+                "attack_methods": {"basic": ["basic"]},
+            },
+        },
+        "endpoint_configuration": {
+          "testing_for": "foundationModels",
+          "model_name": model_name,
+          "model_config": {
+              "model_provider": model_provider,
+              "endpoint_url": model_endpoint_url,
+              "apikey": OPENAI_API_KEY,
+              "input_modalities": ["text"],
+              "output_modalities": ["text"],
+          },
+        },
+    }
+```
+
+### [Sample Custom Redteam Model Config](https://docs.enkryptai.com/redteam-api-reference/endpoint/model-add-custom-task)
+
+```python Python
+sample_custom_redteam_model_config = {
+        "test_name": custom_redteam_test_name,
+        "dataset_configuration": {
+          "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": "Do not generate any content that is illegal, harmful, or violates the rights of others.",
+          "tools": [
+              {
+                  "name": "web_search",
+                  "description": "The tool web search is used to search the web for information related to finance."
+              }
+          ],
+          "max_prompts": 100,
+          "scenarios": 2,
+          "categories": 2,
+          "depth": 2,
+        },
+        "redteam_test_configurations": {
+            "harmful_test": {
+                "sample_percentage": 2,
+                "attack_methods": {"basic": ["basic"]},
+            },
+            "custom_test": {
+                "sample_percentage": 2,
+                "attack_methods": {"basic": ["basic"]},
+            },
+        },
+    }
+```
+
 ## Health Checks
 
 ### [Guardrails Health](https://docs.enkryptai.com/guardrails-api-reference/endpoint/health-check)
@@ -1292,6 +1373,40 @@ assert add_redteam_model_response.message == "Redteam task has been added succes
 print(add_redteam_model_response.to_dict())
 ```
 
+### [Add a Redteam Custom Task with Endpoint Config](https://docs.enkryptai.com/redteam-api-reference/endpoint/add-custom-task)
+
+```python Python
+# Use a dictionary to configure a redteam task
+add_custom_redteam_target_response = redteam_client.add_custom_task(config=copy.deepcopy(sample_custom_redteam_target_config))
+
+# If you already saved a Code of Conduct Policy and want to use it, then instead of passing `dataset_configuration.policy_description` in the body, you can use the SDK like this:
+add_custom_redteam_target_response = redteam_client.add_custom_task(config=copy.deepcopy(sample_custom_redteam_target_config), policy_name="Code of Conduct Policy")
+
+print(add_custom_redteam_target_response)
+
+assert add_custom_redteam_target_response.message == "Task submitted successfully"
+
+# Print as a dictionary
+print(add_custom_redteam_target_response.to_dict())
+```
+
+### [Add a Redteam Custom Task with a saved model](https://docs.enkryptai.com/redteam-api-reference/endpoint/model-add-custom-task)
+
+```python Python
+# Use a dictionary to configure a redteam task
+add_custom_redteam_target_response = redteam_client.add_custom_task_with_saved_model(config=copy.deepcopy(sample_custom_redteam_model_config),model_saved_name=test_model_saved_name,model_version="v1")
+
+# If you already saved a Code of Conduct Policy and want to use it, then instead of passing `dataset_configuration.policy_description` in the body, you can use the SDK like this:
+add_custom_redteam_target_response = redteam_client.add_custom_task_with_saved_model(config=copy.deepcopy(sample_custom_redteam_model_config),model_saved_name=test_model_saved_name,model_version="v1",policy_name="Code of Conduct Policy")
+
+print(add_custom_redteam_target_response)
+
+assert add_custom_redteam_target_response.message == "Task submitted successfully"
+
+# Print as a dictionary
+print(add_custom_redteam_target_response.to_dict())
+```
+
 ### [Get Redteam Task Status](https://docs.enkryptai.com/redteam-api-reference/endpoint/get-task-status)
 
 ```python Python

enkryptai_sdk-1.0.8.dist-info/RECORD

@@ -0,0 +1,25 @@
+enkryptai_sdk/__init__.py,sha256=rP6PtntJogJauj1lKWK8DkiBr3uYjireIUamr6aflu0,763
+enkryptai_sdk/ai_proxy.py,sha256=ot1lqKk2LjcvlpnFm2kSA51vFThfquVlx86BhSbAzBo,3823
+enkryptai_sdk/base.py,sha256=MlEDcEIjXo35kat9XkGUu7VB2fIvJk38C94wAeO9bEw,1304
+enkryptai_sdk/config.py,sha256=IpB8_aO4zXdvv061v24oh83oyJ5Tp1QBQTzeuW4h9QY,8828
+enkryptai_sdk/datasets.py,sha256=RQIR6spI2STXeVolYzBt6gPv6PD5AGh9krs16aKWdWA,6067
+enkryptai_sdk/deployments.py,sha256=A7XZ2JwrMod9V4_aV8bFY_Soh9E3jHdwaTuJ9BwXuyk,4215
+enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
+enkryptai_sdk/guardrails.py,sha256=I5D8OxvJzz7t2jXNqNUj0_1Kl5K0b8zyPsXWAwiZu2o,13853
+enkryptai_sdk/guardrails_old.py,sha256=SgzPZkTzbAPD9XfmYNG6M1-TrzbhDHpAkI3FjnVWS_s,6434
+enkryptai_sdk/models.py,sha256=2H316e28Ssi_MHck0pAoydThRYSdAeSOlwwAoU7q86Q,10941
+enkryptai_sdk/red_team.py,sha256=sJbhCuPfI9aowGvV71hRZslBp1HQa7bklbb89HU5tk8,18574
+enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
+enkryptai_sdk/dto/__init__.py,sha256=RMhr9QKMFuJVXQt3yN7uFwcFpX4PkRvqxjg3ZySUnjg,2505
+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=RFA9CmbhD-QDDyweBq_k9iBd00b6I6SWmdP9DPNd9fc,5002
+enkryptai_sdk/dto/deployments.py,sha256=Aw4b8tDA3FYIomqDvCjblCXTagL4bT8Fx91X0SFXs40,11216
+enkryptai_sdk/dto/guardrails.py,sha256=uolGPPF4v0l76MI5G0ofTtc9-r1l0_sqQqQkLEhAsf0,46305
+enkryptai_sdk/dto/models.py,sha256=lNakMstGozgN_htY2_OItkjQnMQpw5tO9lCZVlywMIg,13442
+enkryptai_sdk/dto/red_team.py,sha256=QtsT4GCnDs24WOF_RMxFtYjvIUULaw5temZ70aduevI,17966
+enkryptai_sdk-1.0.8.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.8.dist-info/METADATA,sha256=CXU0oqhE8FABilK7dug8IBcP8OjetArrUXXsFc7fXM8,56401
+enkryptai_sdk-1.0.8.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+enkryptai_sdk-1.0.8.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.8.dist-info/RECORD,,

enkryptai_sdk-1.0.6.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=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,,