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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.12
+Version: 1.0.14
 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
@@ -271,14 +271,7 @@ sample_detectors = {
         },
         "bias": {
             "enabled": False
-        },
-        "copyright_ip": {
-            "enabled": False
-        },
-        "system_prompt": {
-            "enabled": False,
-            "index": "system"
-        },
+        }
     }
 ```
 
@@ -701,8 +694,8 @@ print(batch_detect_response.to_dict())
 - `nsfw`: Filter inappropriate content
 - `toxicity`: Detect toxic language
 - `pii`: Detect personal information
-- `copyright_ip`: Check for copyright/IP violations
-- `system_prompt`: Detect system prompt leaks
+- `copyright_ip`: Check for copyright/IP violations *(Coming soon)*
+- `system_prompt`: Detect system prompt leaks *(Coming soon)*
 - `keyword_detector`: Check for specific keywords
 
 Each detector can be enabled/disabled and configured with specific options as documented in the [API docs](https://docs.enkryptai.com/guardrails-api-reference/introduction).
@@ -764,12 +757,16 @@ guardrails_config = GuardrailsConfig.keyword(keywords=["secret", "password"])
 
 ### [Copyright IP](https://docs.enkryptai.com/guardrails-api-reference/Copyright_IP_Leak_Detector)
 
+*(Coming soon)*
+
 ```python Python
 guardrails_config = GuardrailsConfig.copyright_ip()
 ```
 
 ### [System Prompt](https://docs.enkryptai.com/guardrails-api-reference/System_Prompt_Leak_Detector)
 
+*(Coming soon)*
+
 ```python Python
 guardrails_config = GuardrailsConfig.system_prompt(index="system")
 ```
@@ -972,6 +969,8 @@ print(relevancy_response.to_dict())
 
 ### [Check Hallucination](https://docs.enkryptai.com/guardrails-api-reference/Hallucination)
 
+*(Coming soon)*
+
 Detect hallucinations in an LLM's response:
 
 ```python Python
@@ -1057,7 +1056,7 @@ print(atomize_response.to_dict())
 # Add a code of conduct policy
 add_policy_response = coc_client.add_policy(
     policy_name=test_coc_policy_name,
-    policy_rules=example_coc_policy_rules,  # Can also be a list of rules
+    policy_rules=example_coc_policy_rules, # Can also be a list of rules
     total_rules=4,
     policy_file="/path/to/your/policy.pdf"
     # policy_text=example_coc_policy_text, # Optional: Use this if you want to add a policy text instead of a file

{enkryptai_sdk-1.0.12 → enkryptai_sdk-1.0.14}/README.md

@@ -248,14 +248,7 @@ sample_detectors = {
         },
         "bias": {
             "enabled": False
-        },
-        "copyright_ip": {
-            "enabled": False
-        },
-        "system_prompt": {
-            "enabled": False,
-            "index": "system"
-        },
+        }
     }
 ```
 
@@ -678,8 +671,8 @@ print(batch_detect_response.to_dict())
 - `nsfw`: Filter inappropriate content
 - `toxicity`: Detect toxic language
 - `pii`: Detect personal information
-- `copyright_ip`: Check for copyright/IP violations
-- `system_prompt`: Detect system prompt leaks
+- `copyright_ip`: Check for copyright/IP violations *(Coming soon)*
+- `system_prompt`: Detect system prompt leaks *(Coming soon)*
 - `keyword_detector`: Check for specific keywords
 
 Each detector can be enabled/disabled and configured with specific options as documented in the [API docs](https://docs.enkryptai.com/guardrails-api-reference/introduction).
@@ -741,12 +734,16 @@ guardrails_config = GuardrailsConfig.keyword(keywords=["secret", "password"])
 
 ### [Copyright IP](https://docs.enkryptai.com/guardrails-api-reference/Copyright_IP_Leak_Detector)
 
+*(Coming soon)*
+
 ```python Python
 guardrails_config = GuardrailsConfig.copyright_ip()
 ```
 
 ### [System Prompt](https://docs.enkryptai.com/guardrails-api-reference/System_Prompt_Leak_Detector)
 
+*(Coming soon)*
+
 ```python Python
 guardrails_config = GuardrailsConfig.system_prompt(index="system")
 ```
@@ -949,6 +946,8 @@ print(relevancy_response.to_dict())
 
 ### [Check Hallucination](https://docs.enkryptai.com/guardrails-api-reference/Hallucination)
 
+*(Coming soon)*
+
 Detect hallucinations in an LLM's response:
 
 ```python Python
@@ -1034,7 +1033,7 @@ print(atomize_response.to_dict())
 # Add a code of conduct policy
 add_policy_response = coc_client.add_policy(
     policy_name=test_coc_policy_name,
-    policy_rules=example_coc_policy_rules,  # Can also be a list of rules
+    policy_rules=example_coc_policy_rules, # Can also be a list of rules
     total_rules=4,
     policy_file="/path/to/your/policy.pdf"
     # policy_text=example_coc_policy_text, # Optional: Use this if you want to add a policy text instead of a file

{enkryptai_sdk-1.0.12 → enkryptai_sdk-1.0.14}/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.14", # 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.14}/src/enkryptai_sdk/dto/models.py

@@ -39,6 +39,7 @@ class ModelProviders(str, Enum):
     ANTHROPIC_COMPATIBLE = "anthropic_compatible"
     CUSTOM = "custom"
     HR = "hr"
+    URL = "url"
 
 
 @dataclass
@@ -166,6 +167,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.14}/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.14}/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.14/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.14
 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
@@ -271,14 +271,7 @@ sample_detectors = {
         },
         "bias": {
             "enabled": False
-        },
-        "copyright_ip": {
-            "enabled": False
-        },
-        "system_prompt": {
-            "enabled": False,
-            "index": "system"
-        },
+        }
     }
 ```
 
@@ -701,8 +694,8 @@ print(batch_detect_response.to_dict())
 - `nsfw`: Filter inappropriate content
 - `toxicity`: Detect toxic language
 - `pii`: Detect personal information
-- `copyright_ip`: Check for copyright/IP violations
-- `system_prompt`: Detect system prompt leaks
+- `copyright_ip`: Check for copyright/IP violations *(Coming soon)*
+- `system_prompt`: Detect system prompt leaks *(Coming soon)*
 - `keyword_detector`: Check for specific keywords
 
 Each detector can be enabled/disabled and configured with specific options as documented in the [API docs](https://docs.enkryptai.com/guardrails-api-reference/introduction).
@@ -764,12 +757,16 @@ guardrails_config = GuardrailsConfig.keyword(keywords=["secret", "password"])
 
 ### [Copyright IP](https://docs.enkryptai.com/guardrails-api-reference/Copyright_IP_Leak_Detector)
 
+*(Coming soon)*
+
 ```python Python
 guardrails_config = GuardrailsConfig.copyright_ip()
 ```
 
 ### [System Prompt](https://docs.enkryptai.com/guardrails-api-reference/System_Prompt_Leak_Detector)
 
+*(Coming soon)*
+
 ```python Python
 guardrails_config = GuardrailsConfig.system_prompt(index="system")
 ```
@@ -972,6 +969,8 @@ print(relevancy_response.to_dict())
 
 ### [Check Hallucination](https://docs.enkryptai.com/guardrails-api-reference/Hallucination)
 
+*(Coming soon)*
+
 Detect hallucinations in an LLM's response:
 
 ```python Python
@@ -1057,7 +1056,7 @@ print(atomize_response.to_dict())
 # Add a code of conduct policy
 add_policy_response = coc_client.add_policy(
     policy_name=test_coc_policy_name,
-    policy_rules=example_coc_policy_rules,  # Can also be a list of rules
+    policy_rules=example_coc_policy_rules, # Can also be a list of rules
     total_rules=4,
     policy_file="/path/to/your/policy.pdf"
     # policy_text=example_coc_policy_text, # Optional: Use this if you want to add a policy text instead of a file

{enkryptai_sdk-1.0.12 → enkryptai_sdk-1.0.14}/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)