enkryptai-sdk 1.0.17__py3-none-any.whl → 1.0.18__py3-none-any.whl

enkryptai_sdk/dto/common.py

@@ -0,0 +1,64 @@
+# To avoid circular imports
+from enum import Enum
+from .base import BaseDTO
+from typing import Dict, List, Optional, Any
+from dataclasses import dataclass, field, asdict
+
+
+class ModelAuthTypeEnum(str, Enum):
+    APIKEY = "apikey"
+    JWT = "jwt"
+
+
+class ModelJwtMethodEnum(str, Enum):
+    POST = "POST"
+    GET = "GET"
+
+
+@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 ModelJwtConfig(BaseDTO):
+    jwt_method: ModelJwtMethodEnum = ModelJwtMethodEnum.POST
+    jwt_url: str = ""
+    jwt_headers: List[CustomHeader] = field(default_factory=list)
+    jwt_body: str = ""
+    jwt_response_key: str = ""
+    _extra_fields: Dict[str, Any] = field(default_factory=dict)
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "ModelJwtConfig":
+        return cls(
+            jwt_method=ModelJwtMethodEnum(data.get("jwt_method", ModelJwtMethodEnum.POST)),
+            jwt_url=data.get("jwt_url", ""),
+            jwt_headers=[CustomHeader.from_dict(header) for header in data.get("jwt_headers", [])],
+            jwt_body=data.get("jwt_body", ""),
+            jwt_response_key=data.get("jwt_response_key", ""),
+        )
+    
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "jwt_method": self.jwt_method.value,
+            "jwt_url": self.jwt_url,
+            "jwt_headers": [header.to_dict() for header in self.jwt_headers],
+            "jwt_body": self.jwt_body,
+            "jwt_response_key": self.jwt_response_key,
+        }
+

enkryptai_sdk/dto/models.py

@@ -5,7 +5,7 @@ from .base import BaseDTO
 from tabulate import tabulate
 from dataclasses import dataclass, field, asdict
 from typing import Optional, List, Set, Dict, Any
-from .red_team import ModelAuthTypeEnum, CustomHeader, ModelJwtConfig
+from .common import ModelAuthTypeEnum, CustomHeader, ModelJwtConfig
 
 
 # class Modality(Enum):

enkryptai_sdk/dto/red_team.py

@@ -3,9 +3,11 @@ 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
 from .guardrails import GuardrailDetectors
+from .common import ModelAuthTypeEnum, CustomHeader, ModelJwtConfig
 
 # The risk mitigation do not support all detectors, so we need to create a separate enum for them.
 class RiskGuardrailDetectorsEnum(str, Enum):
@@ -20,64 +22,6 @@ class RiskGuardrailDetectorsEnum(str, Enum):
     # SYSTEM_PROMPT = "system_prompt"
 
 
-class ModelAuthTypeEnum(str, Enum):
-    APIKEY = "apikey"
-    JWT = "jwt"
-
-
-class ModelJwtMethodEnum(str, Enum):
-    POST = "POST"
-    GET = "GET"
-
-
-@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 ModelJwtConfig(BaseDTO):
-    jwt_method: ModelJwtMethodEnum = ModelJwtMethodEnum.POST
-    jwt_url: str = ""
-    jwt_headers: List[CustomHeader] = field(default_factory=list)
-    jwt_body: str = ""
-    jwt_response_key: str = ""
-    _extra_fields: Dict[str, Any] = field(default_factory=dict)
-
-    @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> "ModelJwtConfig":
-        return cls(
-            jwt_method=ModelJwtMethodEnum(data.get("jwt_method", ModelJwtMethodEnum.POST)),
-            jwt_url=data.get("jwt_url", ""),
-            jwt_headers=[CustomHeader.from_dict(header) for header in data.get("jwt_headers", [])],
-            jwt_body=data.get("jwt_body", ""),
-            jwt_response_key=data.get("jwt_response_key", ""),
-        )
-    
-    def to_dict(self) -> Dict[str, Any]:
-        return {
-            "jwt_method": self.jwt_method.value,
-            "jwt_url": self.jwt_url,
-            "jwt_headers": [header.to_dict() for header in self.jwt_headers],
-            "jwt_body": self.jwt_body,
-            "jwt_response_key": self.jwt_response_key,
-        }
-
-
 @dataclass
 class RedteamHealthResponse(BaseDTO):
     status: str

{enkryptai_sdk-1.0.17.dist-info → enkryptai_sdk-1.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.17
+Version: 1.0.18
 Summary: A Python SDK with guardrails and red teaming functionality for API interactions
 Home-page: https://github.com/enkryptai/enkryptai-sdk
 Author: Enkrypt AI Team

{enkryptai_sdk-1.0.17.dist-info → enkryptai_sdk-1.0.18.dist-info}/RECORD

@@ -15,13 +15,14 @@ enkryptai_sdk/dto/__init__.py,sha256=wHgIv_OCnVMJOys-vqImF59ifogDrMcgxVRmfNayVvc
 enkryptai_sdk/dto/ai_proxy.py,sha256=clwMN4xdH8Zr55dnhilHbs-qaHRlCOrLPrij0Zd1Av0,11283
 enkryptai_sdk/dto/base.py,sha256=y77kQL1X7389ifSVNc0E7CUFNxACh5AM3ml9YPon1KY,2822
 enkryptai_sdk/dto/coc.py,sha256=Lp2aat_24J4KuUg4BeJl9S39tEak8Bw15eJ4cQDrRQk,4749
+enkryptai_sdk/dto/common.py,sha256=lrWMu4FKUGCN2dbS9fT4yNtfiPm1cNN16J4eCe4_tBM,1812
 enkryptai_sdk/dto/datasets.py,sha256=RFA9CmbhD-QDDyweBq_k9iBd00b6I6SWmdP9DPNd9fc,5002
 enkryptai_sdk/dto/deployments.py,sha256=Aw4b8tDA3FYIomqDvCjblCXTagL4bT8Fx91X0SFXs40,11216
 enkryptai_sdk/dto/guardrails.py,sha256=oJQqFhsdQd_yPU187AhKse-Y4xktgmVNwwKKkzFazbg,50167
-enkryptai_sdk/dto/models.py,sha256=Nk6ZQyMfDMbl5ITQWQJUZMPxQOTATA65QbKHIy7i3qM,14533
-enkryptai_sdk/dto/red_team.py,sha256=NrYj9NZX3EmrBpESiZYFUCzgXH3eConh4NhVRKT28LM,27362
-enkryptai_sdk-1.0.17.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.17.dist-info/METADATA,sha256=10vTRJNUYuEu7KgixIMOvgaweistehy8_s3kf3lkXiA,72860
-enkryptai_sdk-1.0.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-enkryptai_sdk-1.0.17.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.17.dist-info/RECORD,,
+enkryptai_sdk/dto/models.py,sha256=4aeI9-iHPhSX5duRMTQRbkXhplAhN8OHw3gWdTxrafU,14531
+enkryptai_sdk/dto/red_team.py,sha256=BAvjpz_Xxb4UW5GRmvfOLV3KvvciPyVGUyYgNDBzq2E,25790
+enkryptai_sdk-1.0.18.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.18.dist-info/METADATA,sha256=PWAkBaJJiqgxvUfRfSbO9wUSNaYsAczTiFwB8sycVo0,72860
+enkryptai_sdk-1.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+enkryptai_sdk-1.0.18.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.18.dist-info/RECORD,,