enkryptai-sdk 1.0.12__tar.gz → 1.0.13__tar.gz

{enkryptai_sdk-1.0.12/src/enkryptai_sdk.egg-info → enkryptai_sdk-1.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.12
+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.12 → enkryptai_sdk-1.0.13}/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.12", # Update this for new versions
+    version="1.0.13", # 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.12 → enkryptai_sdk-1.0.13}/src/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-1.0.12 → enkryptai_sdk-1.0.13}/src/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-1.0.12 → enkryptai_sdk-1.0.13}/src/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-1.0.12 → enkryptai_sdk-1.0.13/src/enkryptai_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.12
+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.12 → enkryptai_sdk-1.0.13}/tests/test_all_v2.py

@@ -996,7 +996,7 @@ def test_list_models(model_client):
     # assert model_info["model_version"] == test_model_version
 
 def test_get_model(model_client):
-    print("\n\nTesting get_model with custom model")
+    print("\n\nTesting get_model")
     # global model_saved_name
     # if model_saved_name is None:
     #     print("\nModel saved name is None, fetching it from list_models")
@@ -1006,11 +1006,16 @@ def test_get_model(model_client):
     #     model_saved_name = model_info.model_saved_name
     #     assert model_saved_name is not None
     #     print("\nPicked model in get_model: ", model_saved_name)
-    model_saved_name = test_custom_model_saved_name
+    # -------------------------------
+    # NOTE: Using custom model here is breaking AI Proxy as model_saved_name is a global variable
+    # -------------------------------
+    # model_saved_name = test_custom_model_saved_name
+    model_saved_name = test_model_saved_name
     model_version = test_model_version
 
     # Now test the get_model method
     model = model_client.get_model(model_saved_name=model_saved_name, model_version=model_version)
+    # print("\nGET Model: ", model)
     assert model is not None
     # assert hasattr(model, "model_id")
     assert model.model_saved_name == model_saved_name
@@ -1040,7 +1045,7 @@ def test_modify_model(model_client, sample_custom_model_config):
 # Tests for Deployment Endpoints
 # ----------------------------
 
-def test_add_deployment_success(deployment_client, sample_deployment_config):
+def test_add_deployment(deployment_client, sample_deployment_config):
     response = deployment_client.add_deployment(config=sample_deployment_config)
     assert response is not None
     assert hasattr(response, "from_dict")
@@ -1131,7 +1136,7 @@ def test_401_err_chat(ai_proxy_err_client, sample_chat_body):
 # Tests for Dataset Endpoints
 # ----------------------------
 
-def test_add_dataset_success(dataset_client, sample_dataset_config):
+def test_add_dataset(dataset_client, sample_dataset_config):
     print("\n\nTesting add_dataset")
     response = dataset_client.add_dataset(config=sample_dataset_config)
     print("\nAdd Dataset Response: ", response)