enkryptai-sdk 1.0.11__py3-none-any.whl → 1.0.13__py3-none-any.whl

enkryptai_sdk/dto/models.py

@@ -166,6 +166,7 @@ class ModelConfigDetails(BaseDTO):
     auth_data: AuthData = field(default_factory=AuthData)
     input_modalities: List[InputModality] = field(default_factory=list)
     output_modalities: List[OutputModality] = field(default_factory=list)
+    custom_curl_command: Optional[str] = None
     custom_headers: List[CustomHeader] = field(default_factory=list)
     custom_payload: Dict[str, Any] = field(default_factory=dict)
     custom_response_content_type: Optional[str] = ""

enkryptai_sdk/dto/red_team.py

@@ -359,6 +359,7 @@ class TargetModelConfiguration(BaseDTO):
     model_api_key: Optional[str] = None
     input_modalities: List[InputModality] = field(default_factory=list)
     output_modalities: List[OutputModality] = field(default_factory=list)
+    custom_curl_command: Optional[str] = None
     custom_headers: List[CustomHeader] = field(default_factory=list)
     custom_payload: Dict[str, Any] = field(default_factory=dict)
     custom_response_content_type: Optional[str] = ""

enkryptai_sdk/models.py

@@ -77,6 +77,7 @@ class ModelClient(BaseClient):
                 "tools": config.model_config.tools,
                 "input_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.input_modalities],
                 "output_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.output_modalities],
+                "custom_curl_command": config.model_config.custom_curl_command,
                 "custom_headers": custom_headers,
                 "custom_payload": config.model_config.custom_payload,
                 "custom_response_content_type": config.model_config.custom_response_content_type,
@@ -225,6 +226,7 @@ class ModelClient(BaseClient):
                 "tools": config.model_config.tools,
                 "input_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.input_modalities],
                 "output_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.output_modalities],
+                "custom_curl_command": config.model_config.custom_curl_command,
                 "custom_headers": custom_headers,
                 "custom_payload": config.model_config.custom_payload,
                 "custom_response_content_type": config.model_config.custom_response_content_type,

enkryptai_sdk/red_team.py

@@ -106,9 +106,7 @@ class RedTeamClient(BaseClient):
             # "async": config.async_enabled,
             "dataset_name": config.dataset_name,
             "test_name": config.test_name,
-            "redteam_test_configurations": {
-                k: v.to_dict() for k, v in test_configs.items()
-            },
+            "redteam_test_configurations": test_configs,
         }
 
         if config.target_model_configuration:
@@ -153,9 +151,7 @@ class RedTeamClient(BaseClient):
             # "async": config.async_enabled,
             "dataset_name": config.dataset_name,
             "test_name": config.test_name,
-            "redteam_test_configurations": {
-                k: v.to_dict() for k, v in test_configs.items()
-            },
+            "redteam_test_configurations": test_configs,
         }
 
         headers = {
@@ -196,9 +192,7 @@ class RedTeamClient(BaseClient):
         payload = {
             # "async": config.async_enabled,
             "test_name": config.test_name,
-            "redteam_test_configurations": {
-                k: v.to_dict() for k, v in test_configs.items()
-            },
+            "redteam_test_configurations": test_configs,
         }
 
         if config.dataset_configuration:
@@ -252,9 +246,7 @@ class RedTeamClient(BaseClient):
         payload = {
             # "async": config.async_enabled,
             "test_name": config.test_name,
-            "redteam_test_configurations": {
-                k: v.to_dict() for k, v in test_configs.items()
-            },
+            "redteam_test_configurations": test_configs,
         }
 
         if config.dataset_configuration:

{enkryptai_sdk-1.0.11.dist-info → enkryptai_sdk-1.0.13.dist-info}/METADATA

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

@@ -8,8 +8,8 @@ enkryptai_sdk/deployments.py,sha256=A7XZ2JwrMod9V4_aV8bFY_Soh9E3jHdwaTuJ9BwXuyk,
 enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
 enkryptai_sdk/guardrails.py,sha256=NluimOA0gM9N3S_q47LTUeG97t9PlYqPHlZahDPkJvI,16365
 enkryptai_sdk/guardrails_old.py,sha256=SgzPZkTzbAPD9XfmYNG6M1-TrzbhDHpAkI3FjnVWS_s,6434
-enkryptai_sdk/models.py,sha256=EA2jwdoYqOjQ8GuRawsZ3syIvVinwhvodTLQlvt51Mo,11399
-enkryptai_sdk/red_team.py,sha256=iTBSXNUr_JFFQppKzUzJ5-gWtlPPzY6BkKahhXppaFM,18709
+enkryptai_sdk/models.py,sha256=rrLTT3i96flWidVrr67j6VZ6XmkdxwEzlF4S4aoVmOQ,11559
+enkryptai_sdk/red_team.py,sha256=A1b6R5d3dtZMfb5npBrSu4RX2X-C9V4hubyO9g3BZlA,18441
 enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
 enkryptai_sdk/dto/__init__.py,sha256=wHgIv_OCnVMJOys-vqImF59ifogDrMcgxVRmfNayVvc,2761
 enkryptai_sdk/dto/ai_proxy.py,sha256=clwMN4xdH8Zr55dnhilHbs-qaHRlCOrLPrij0Zd1Av0,11283
@@ -18,10 +18,10 @@ enkryptai_sdk/dto/coc.py,sha256=Lp2aat_24J4KuUg4BeJl9S39tEak8Bw15eJ4cQDrRQk,4749
 enkryptai_sdk/dto/datasets.py,sha256=RFA9CmbhD-QDDyweBq_k9iBd00b6I6SWmdP9DPNd9fc,5002
 enkryptai_sdk/dto/deployments.py,sha256=Aw4b8tDA3FYIomqDvCjblCXTagL4bT8Fx91X0SFXs40,11216
 enkryptai_sdk/dto/guardrails.py,sha256=wfKm2R0uJAq9nljLBPuuT05tj1nBqZraga-5lMHd3n8,49812
-enkryptai_sdk/dto/models.py,sha256=08ilpQkp_tpUDzM5Jt-mrWdHlbBgYst0plzzT4G-Gds,14184
-enkryptai_sdk/dto/red_team.py,sha256=UNqiPxKqk8TJ-hLtBCWRvtoXyGCFiz3X7bKpx8sTQxo,18298
-enkryptai_sdk-1.0.11.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.11.dist-info/METADATA,sha256=adej6oT0uO297qJIDmyktoTx7CmgSZWiBMyPWAHYMro,62092
-enkryptai_sdk-1.0.11.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-enkryptai_sdk-1.0.11.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.11.dist-info/RECORD,,
+enkryptai_sdk/dto/models.py,sha256=h8eiDeT-cpCMS5IUFxWzwrLQHAAdnAFrvSuSFw7SiaI,14230
+enkryptai_sdk/dto/red_team.py,sha256=GD_HxFhoaEixyaZLVdoLHmHDXFRf9b-Wq6RT-UJNQfM,18344
+enkryptai_sdk-1.0.13.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.13.dist-info/METADATA,sha256=2rSIQzIvmj8ISa4HBKXGEEepdb4ASOxts6E1jo8U8Q0,62092
+enkryptai_sdk-1.0.13.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+enkryptai_sdk-1.0.13.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.13.dist-info/RECORD,,