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

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

@@ -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 → enkryptai_sdk-1.0.18}/setup.py

@@ -9,7 +9,7 @@ with open(os.path.join(here, "README.md"), encoding="utf-8") as fh:
 setup(
     name="enkryptai-sdk",  # This is the name of your package on PyPI
     # NOTE: Also change this in .github/workflows/test.yaml
-    version="1.0.17", # Update this for new versions
+    version="1.0.18", # Update this for new versions
     description="A Python SDK with guardrails and red teaming functionality for API interactions",
     long_description=long_description,
     long_description_content_type="text/markdown",

enkryptai_sdk-1.0.18/src/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-1.0.17 → enkryptai_sdk-1.0.18}/src/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-1.0.17 → enkryptai_sdk-1.0.18}/src/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 → enkryptai_sdk-1.0.18/src/enkryptai_sdk.egg-info}/PKG-INFO

@@ -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 → enkryptai_sdk-1.0.18}/src/enkryptai_sdk.egg-info/SOURCES.txt

@@ -22,6 +22,7 @@ src/enkryptai_sdk/dto/__init__.py
 src/enkryptai_sdk/dto/ai_proxy.py
 src/enkryptai_sdk/dto/base.py
 src/enkryptai_sdk/dto/coc.py
+src/enkryptai_sdk/dto/common.py
 src/enkryptai_sdk/dto/datasets.py
 src/enkryptai_sdk/dto/deployments.py
 src/enkryptai_sdk/dto/guardrails.py