enkryptai-sdk 1.0.22__py3-none-any.whl → 1.0.23__py3-none-any.whl

enkryptai_sdk/dto/coc.py

@@ -7,6 +7,8 @@ from typing import List, Dict, Any, Set, Optional, BinaryIO
 @dataclass
 class CoCPolicyData(BaseDTO):
     created_at: str
+    created_by: str
+    updated_by: str
     name: str
     updated_at: str
     policy_id: int
@@ -20,6 +22,8 @@ class CoCPolicyData(BaseDTO):
     def from_dict(cls, data: Dict[str, Any]) -> "CoCPolicyData":
         return cls(
             created_at=data.get("created_at", ""),
+            created_by=data.get("created_by", ""),
+            updated_by=data.get("updated_by", ""),
             name=data.get("name", ""),
             updated_at=data.get("updated_at", ""),
             policy_id=data.get("policy_id", 0),
@@ -32,6 +36,8 @@ class CoCPolicyData(BaseDTO):
     def to_dict(self) -> Dict[str, Any]:
         result = {
             "created_at": self.created_at,
+            "created_by": self.created_by,
+            "updated_by": self.updated_by,
             "name": self.name,
             "updated_at": self.updated_at,
             "policy_id": self.policy_id,
@@ -65,6 +71,8 @@ class CoCPolicyData(BaseDTO):
             f"Policy Name: {self.name}\n"
             f"Created At: {self.created_at}\n"
             f"Updated At: {self.updated_at}\n"
+            f"Created By: {self.created_by}\n"
+            f"Updated By: {self.updated_by}\n"
             f"Policy ID: {self.policy_id}\n"
             f"total_rules: {self.total_rules}\n"
         )

enkryptai_sdk/dto/datasets.py

@@ -92,6 +92,7 @@ class DatasetContent(BaseDTO):
 class DatasetTaskData(BaseDTO):
     dataset_name: str
     created_at: str
+    created_by: str
     started_at: str
     status: str
     system_description: str
@@ -110,6 +111,7 @@ class DatasetTaskData(BaseDTO):
         return cls(
             dataset_name=data.get("dataset_name", ""),
             created_at=data.get("created_at", ""),
+            created_by=data.get("created_by", ""),
             started_at=data.get("started_at", ""),
             status=data.get("status", ""),
             system_description=data.get("system_description", ""),

enkryptai_sdk/dto/deployments.py

@@ -182,6 +182,8 @@ class GetDeploymentResponse(BaseDTO):
     output_guardrails_policy: OutputGuardrailsPolicy
     created_at: str
     updated_at: str
+    created_by: str
+    updated_by: str
     deployment_id: int
 
     @classmethod
@@ -197,7 +199,9 @@ class GetDeploymentResponse(BaseDTO):
             output_guardrails_policy=OutputGuardrailsPolicy.from_dict(output_policy_data),
             updated_at=data.get("updated_at", ""),
             deployment_id=data.get("deployment_id", 0),
-            created_at=data.get("created_at", "")
+            created_at=data.get("created_at", ""),
+            created_by=data.get("created_by", ""),
+            updated_by=data.get("updated_by", "")
         )
 
     def to_dict(self) -> Dict[str, Any]:
@@ -209,7 +213,9 @@ class GetDeploymentResponse(BaseDTO):
             "output_guardrails_policy": self.output_guardrails_policy.to_dict(),
             "updated_at": self.updated_at,
             "deployment_id": self.deployment_id,
-            "created_at": self.created_at
+            "created_at": self.created_at,
+            "created_by": self.created_by,
+            "updated_by": self.updated_by
         }
 
 
@@ -300,6 +306,8 @@ class DeploymentSummary(BaseDTO):
     name: str
     created_at: str
     updated_at: str
+    created_by: str
+    updated_by: str
     model_saved_name: str
     model_version: str
     project_name: str
@@ -311,6 +319,8 @@ class DeploymentSummary(BaseDTO):
             name=data.get("name", ""),
             created_at=data.get("created_at", ""),
             updated_at=data.get("updated_at", ""),
+            created_by=data.get("created_by", ""),
+            updated_by=data.get("updated_by", ""),
             model_saved_name=data.get("model_saved_name", ""),
             model_version=data.get("model_version", ""),
             project_name=data.get("project_name", "")
@@ -322,6 +332,8 @@ class DeploymentSummary(BaseDTO):
             "name": self.name,
             "created_at": self.created_at,
             "updated_at": self.updated_at,
+            "created_by": self.created_by,
+            "updated_by": self.updated_by,
             "model_saved_name": self.model_saved_name,
             "model_version": self.model_version,
             "project_name": self.project_name

enkryptai_sdk/dto/guardrails.py

@@ -1365,6 +1365,8 @@ class GuardrailsPolicyData(BaseDTO):
     created_at: str
     name: str
     updated_at: str
+    created_by: str
+    updated_by: str
     description: str
     policy_id: int
     project_name: str = ""
@@ -1379,6 +1381,8 @@ class GuardrailsPolicyData(BaseDTO):
             created_at=data.get("created_at", ""),
             name=data.get("name", ""),
             updated_at=data.get("updated_at", ""),
+            created_by=data.get("created_by", ""),
+            updated_by=data.get("updated_by", ""),
             description=data.get("description", ""),
             policy_id=data.get("policy_id", 0),
             project_name=data.get("project_name", ""),
@@ -1390,6 +1394,8 @@ class GuardrailsPolicyData(BaseDTO):
             "created_at": self.created_at,
             "name": self.name,
             "updated_at": self.updated_at,
+            "created_by": self.created_by,
+            "updated_by": self.updated_by,
             "description": self.description,
             "policy_id": self.policy_id,
             "project_name": self.project_name,
@@ -1474,6 +1480,8 @@ class GuardrailsPolicyListItem(BaseDTO):
     description: str
     created_at: str
     updated_at: str
+    created_by: str
+    updated_by: str
     project_name: str = "default"
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
     
@@ -1485,6 +1493,8 @@ class GuardrailsPolicyListItem(BaseDTO):
             description=data.get("description", ""),
             created_at=data.get("created_at", ""),
             updated_at=data.get("updated_at", ""),
+            created_by=data.get("created_by", ""),
+            updated_by=data.get("updated_by", ""),
             project_name=data.get("project_name", "default")
         )
     
@@ -1495,6 +1505,8 @@ class GuardrailsPolicyListItem(BaseDTO):
             "description": self.description,
             "created_at": self.created_at,
             "updated_at": self.updated_at,
+            "created_by": self.created_by,
+            "updated_by": self.updated_by,
             "project_name": self.project_name
         }
         result.update(self._extra_fields)

enkryptai_sdk/dto/models.py

@@ -268,6 +268,8 @@ class ModelConfigDetails(BaseDTO):
 class ModelConfig(BaseDTO):
     created_at: str = None
     updated_at: str = None
+    created_by: str = None
+    updated_by: str = None
     model_id: str = None
     model_saved_name: str = None
     model_version: str = None

enkryptai_sdk/dto/red_team.py

@@ -88,6 +88,7 @@ class RedTeamTaskDetailsModelConfig(BaseDTO):
 @dataclass
 class RedTeamTaskDetails(BaseDTO):
     created_at: Optional[str] = None
+    created_by: Optional[str] = None
     model_saved_name: Optional[str] = None
     model_name: Optional[str] = None
     status: Optional[str] = None
@@ -101,6 +102,7 @@ class RedTeamTaskDetails(BaseDTO):
         # print(f"RedTeamTaskDetails data: {data}")
         return cls(
             created_at=data.get("created_at"),
+            created_by=data.get("created_by"),
             model_saved_name=data.get("model_saved_name"),
             model_name=data.get("model_name"),
             status=data.get("status"),
@@ -112,6 +114,7 @@ class RedTeamTaskDetails(BaseDTO):
     def to_dict(self) -> Dict:
         return {
             "created_at": self.created_at,
+            "created_by": self.created_by,
             "model_saved_name": self.model_saved_name,
             "model_name": self.model_name,
             "status": self.status,

{enkryptai_sdk-1.0.22.dist-info → enkryptai_sdk-1.0.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.22
+Version: 1.0.23
 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.22.dist-info → enkryptai_sdk-1.0.23.dist-info}/RECORD

@@ -14,15 +14,15 @@ enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,408
 enkryptai_sdk/dto/__init__.py,sha256=wHgIv_OCnVMJOys-vqImF59ifogDrMcgxVRmfNayVvc,2761
 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/coc.py,sha256=9D5mmSdmC_guV75ml48PPLZD_zFa5FjxRwlTqHrmdak,5071
 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=_tdSyRTJvthjLTbOxFO4f2P6vgirDsF3in5_gXE3a_U,11288
-enkryptai_sdk/dto/guardrails.py,sha256=cfL18qfDtNXG4SONCVONHYAKS9J3m0EAqUdTeeUoZK8,55510
-enkryptai_sdk/dto/models.py,sha256=4aeI9-iHPhSX5duRMTQRbkXhplAhN8OHw3gWdTxrafU,14531
-enkryptai_sdk/dto/red_team.py,sha256=KRR_gDBTrrN6VEiSLaXDIqciY7N7OelIAQNmzCAsZgU,28675
-enkryptai_sdk-1.0.22.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.22.dist-info/METADATA,sha256=EDFqSS5spOBzcyNAgtqm8DnUeGE7AtxXRsDEIdHWZTw,1644
-enkryptai_sdk-1.0.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-enkryptai_sdk-1.0.22.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.22.dist-info/RECORD,,
+enkryptai_sdk/dto/datasets.py,sha256=FiVwbEZ04jPajnlBfgw1Aj83iAhup3Ut7GOymq73OMg,5073
+enkryptai_sdk/dto/deployments.py,sha256=v--UrwkuXP4xTsPbmVruYj-g3JEQXepBRQfr-Gsv3aA,11744
+enkryptai_sdk/dto/guardrails.py,sha256=n73uPmK9fCIqqxv1rNecYUX4v5ePdIjxSBD4jIhwDe8,55966
+enkryptai_sdk/dto/models.py,sha256=AwqYMBZckJ0RdeU89I-3CYJ03c-VLdHf5zD5PSbU3JQ,14585
+enkryptai_sdk/dto/red_team.py,sha256=rV7bYVb5YSxnrhLwNiTlKWy0ZTXK4zhmPgVoLJQMMoo,28802
+enkryptai_sdk-1.0.23.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.23.dist-info/METADATA,sha256=CMSsEyEBXxGhIciL2JgGg2slVeEbxw9OeLRpVtUp1qM,1644
+enkryptai_sdk-1.0.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+enkryptai_sdk-1.0.23.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.23.dist-info/RECORD,,