enkryptai-sdk 1.0.23__tar.gz → 1.0.24__tar.gz

{enkryptai_sdk-1.0.23 → enkryptai_sdk-1.0.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.23
+Version: 1.0.24
 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.23 → enkryptai_sdk-1.0.24}/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.23", # Update this for new versions
+    version="1.0.24", # 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.23 → enkryptai_sdk-1.0.24}/src/enkryptai_sdk/dto/red_team.py

@@ -391,6 +391,7 @@ class RedTeamTestConfigurations(BaseDTO):
     # Basic tests
     bias_test: TestConfig = field(default=None)
     cbrn_test: TestConfig = field(default=None)
+    csem_test: TestConfig = field(default=None)
     insecure_code_test: TestConfig = field(default=None)
     toxicity_test: TestConfig = field(default=None)
     harmful_test: TestConfig = field(default=None)

{enkryptai_sdk-1.0.23 → enkryptai_sdk-1.0.24}/src/enkryptai_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.23
+Version: 1.0.24
 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.23 → enkryptai_sdk-1.0.24}/tests/test_redteam.py

@@ -132,6 +132,61 @@ def test_add_custom_task_with_saved_model(redteam_client, sample_custom_redteam_
     time.sleep(60)
 
 
+# def test_add_custom_task_with_saved_model_2(redteam_client, sample_custom_redteam_model_config):
+#     print("\n\nTesting adding a new custom redteam task with saved model (2)")
+#     response = redteam_client.add_custom_task_with_saved_model(config=sample_custom_redteam_model_config, model_saved_name=test_model_saved_name, model_version=test_model_version)
+#     print("\nResponse from adding a new custom redteam task with saved model (2): ", response)
+#     assert response is not None
+#     assert hasattr(response, "task_id")
+#     assert hasattr(response, "message")
+#     response.message == "Task submitted successfully"
+#     # Sleep for a while to let the task complete
+#     # This is also useful to avoid rate limiting issues
+#     # print("\nSleeping for 60 seconds to let the (2) task complete if possible ...")
+#     # time.sleep(60)
+
+
+# def test_add_custom_task_with_saved_model_3(redteam_client, sample_custom_redteam_model_config):
+#     print("\n\nTesting adding a new custom redteam task with saved model (3)")
+#     response = redteam_client.add_custom_task_with_saved_model(config=sample_custom_redteam_model_config, model_saved_name=test_model_saved_name, model_version=test_model_version)
+#     print("\nResponse from adding a new custom redteam task with saved model (3): ", response)
+#     assert response is not None
+#     assert hasattr(response, "task_id")
+#     assert hasattr(response, "message")
+#     response.message == "Task submitted successfully"
+#     # Sleep for a while to let the task complete
+#     # This is also useful to avoid rate limiting issues
+#     # print("\nSleeping for 60 seconds to let the (3) task complete if possible ...")
+#     # time.sleep(60)
+
+
+# def test_add_custom_task_with_saved_model_4(redteam_client, sample_custom_redteam_model_config):
+#     print("\n\nTesting adding a new custom redteam task with saved model (4)")
+#     response = redteam_client.add_custom_task_with_saved_model(config=sample_custom_redteam_model_config, model_saved_name=test_model_saved_name, model_version=test_model_version)
+#     print("\nResponse from adding a new custom redteam task with saved model (4): ", response)
+#     assert response is not None
+#     assert hasattr(response, "task_id")
+#     assert hasattr(response, "message")
+#     response.message == "Task submitted successfully"
+#     # Sleep for a while to let the task complete
+#     # This is also useful to avoid rate limiting issues
+#     # print("\nSleeping for 60 seconds to let the (4) task complete if possible ...")
+#     # time.sleep(60)
+
+
+# def test_add_custom_task_with_saved_model_5(redteam_client, sample_custom_redteam_model_config):
+#     print("\n\nTesting adding a new custom redteam task with saved model (5)")
+#     response = redteam_client.add_custom_task_with_saved_model(config=sample_custom_redteam_model_config, model_saved_name=test_model_saved_name, model_version=test_model_version)
+#     print("\nResponse from adding a new custom redteam task with saved model (5): ", response)
+#     assert response is not None
+#     assert hasattr(response, "task_id")
+#     assert hasattr(response, "message")
+#     response.message == "Task submitted successfully"
+#     # Sleep for a while to let the task complete
+#     # This is also useful to avoid rate limiting issues
+#     # print("\nSleeping for 60 seconds to let the (5) task complete if possible ...")
+#     # time.sleep(60)
+
 def test_list_redteams(redteam_client):
     print("\n\nTesting list_redteam tasks")
     redteams = redteam_client.get_task_list(status="Finished")