enkryptai-sdk 1.0.3__tar.gz → 1.0.5__tar.gz

{enkryptai_sdk-1.0.3/src/enkryptai_sdk.egg-info → enkryptai_sdk-1.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.3
+Version: 1.0.5
 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
@@ -21,9 +21,10 @@ Dynamic: license-file
 Dynamic: requires-python
 Dynamic: summary
 
-![Python SDK test](https://github.com/enkryptai/enkryptai-sdk/actions/workflows/test.yaml/badge.svg)
 # Enkrypt AI Python SDK
 
+![Python SDK test](https://github.com/enkryptai/enkryptai-sdk/actions/workflows/test.yaml/badge.svg)
+
 A Python SDK with Guardrails, Models, Deployments, AI Proxy, Datasets and Red Team functionality for API interactions.
 
 See [https://pypi.org/project/enkryptai-sdk](https://pypi.org/project/enkryptai-sdk)
@@ -55,6 +56,7 @@ Also see the API documentation at [https://docs.enkryptai.com](https://docs.enkr
 - [Guardrails Quickstart](#guardrails-quickstart)
 - [Guardrails Response Objects](#guardrails-response-objects)
   - [GuardrailsDetectResponse](#guardrailsdetectresponse)
+  - [GuardrailsBatchDetectResponse](#guardrailsbatchdetectresponse)
 - [Available Guardrails Detectors](#available-guardrails-detectors)
 - [Guardrails Configs](#guardrails-configs)
   - [Injection Attack](#injection-attack)
@@ -181,6 +183,7 @@ redteam_client = RedTeamClient(api_key=ENKRYPT_API_KEY, base_url=ENKRYPT_BASE_UR
 ```python Python
 test_policy_name = "Test Policy"
 test_model_saved_name = "Test Model"
+test_model_version = "v1"
 test_deployment_name = "test-deployment"
 
 pii_original_text = "My email is example@example.com. My phone number is 123-456-7890."
@@ -257,14 +260,15 @@ sample_detectors = {
 ```python Python
 sample_model_config = {
         "model_saved_name": test_model_saved_name,
-        "testing_for": "LLM",
+        "model_version": test_model_version,
+        "testing_for": "foundationModels",
         "model_name": model_name,
-        "modality": "text",
         "model_config": {
-            "model_version": "",
             "model_provider": model_provider,
             "endpoint_url": model_endpoint_url,
             "apikey": OPENAI_API_KEY,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"],
         },
     }
 ```
@@ -275,6 +279,7 @@ sample_model_config = {
 sample_deployment_config = {
         "name": test_deployment_name,
         "model_saved_name": test_model_saved_name,
+        "model_version": test_model_version,
         "input_guardrails_policy": {
             "policy_name": test_policy_name,
             "enabled": True,
@@ -308,8 +313,16 @@ sample_dataset_config = {
         "dataset_name": dataset_name,
         "system_description": "- **Voter Eligibility**: To vote in U.S. elections, individuals must be U.S. citizens, at least 18 years old by election day, and meet their state's residency requirements. - **Voter Registration**: Most states require voters to register ahead of time, with deadlines varying widely. North Dakota is an exception, as it does not require voter registration. - **Identification Requirements**: Thirty-six states enforce voter ID laws, requiring individuals to present identification at polling places. These laws aim to prevent voter fraud but can also lead to disenfranchisement. - **Voting Methods**: Voters can typically choose between in-person voting on election day, early voting, and absentee or mail-in ballots, depending on state regulations. - **Polling Hours**: Polling hours vary by state, with some states allowing extended hours for voters. Its essential for voters to check local polling times to ensure they can cast their ballots. - **Provisional Ballots**: If there are questions about a voter's eligibility, they may be allowed to cast a provisional ballot. This ballot is counted once eligibility is confirmed. - **Election Day Laws**: Many states have laws that protect the rights of voters on election day, including prohibiting intimidation and ensuring access to polling places. - **Campaign Finance Regulations**: Federal and state laws regulate contributions to candidates and political parties to ensure transparency and limit the influence of money in politics. - **Political Advertising**: Campaigns must adhere to rules regarding political advertising, including disclosure requirements about funding sources and content accuracy. - **Voter Intimidation Prohibitions**: Federal laws prohibit any form of voter intimidation or coercion at polling places, ensuring a safe environment for all voters. - **Accessibility Requirements**: The Americans with Disabilities Act mandates that polling places be accessible to individuals with disabilities, ensuring equal access to the electoral process. - **Election Monitoring**: Various organizations are allowed to monitor elections to ensure compliance with laws and regulations. They help maintain transparency and accountability in the electoral process. - **Vote Counting Procedures**: States have specific procedures for counting votes, including the use of electronic voting machines and manual audits to verify results. - **Ballot Design Standards**: States must adhere to certain design standards for ballots to ensure clarity and prevent confusion among voters when casting their votes. - **Post-Election Audits**: Some states conduct post-election audits as a measure of accuracy. These audits help verify that the vote count reflects the actual ballots cast.",
         "policy_description": "",
-        "tools": [],
+        "tools": [
+            {
+                "name": "web_search",
+                "description": "The tool web search is used to search the web for information related to finance."
+            }
+        ],
         "info_pdf_url": "",
+        "scenarios": 1,
+        "categories": 1,
+        "depth": 1,
         "max_prompts": 100,
     }
 ```
@@ -320,16 +333,16 @@ sample_dataset_config = {
 sample_redteam_model_health_config = {
         "target_model_configuration": {
             "model_name": model_name,
-            "testing_for": "LLM",
-            "model_type": "text_2_text",
-            "model_version": "v1",
+            "testing_for": "foundationModels",
+            "model_version": test_model_version,
             "model_source": "https://openai.com",
             "model_provider": model_provider,
             "model_endpoint_url": model_endpoint_url,
             "model_api_key": OPENAI_API_KEY,
             "system_prompt": "",
-            "conversation_template": "",
-            "rate_per_min": 20
+            "rate_per_min": 20,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"]
         },
     }
 ```
@@ -364,16 +377,16 @@ sample_redteam_target_config = {
         },
         "target_model_configuration": {
             "model_name": model_name,
-            "testing_for": "LLM",
-            "model_type": "text_2_text",
-            "model_version": "v1",
+            "testing_for": "foundationModels",
+            "model_version": test_model_version,
             "model_source": "https://openai.com",
             "model_provider": model_provider,
             "model_endpoint_url": model_endpoint_url,
             "model_api_key": OPENAI_API_KEY,
             "system_prompt": "",
-            "conversation_template": "",
-            "rate_per_min": 20
+            "rate_per_min": 20,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"]
         },
     }
 ```
@@ -539,6 +552,43 @@ print(detect_response)
 print(detect_response.to_dict())
 ```
 
+### GuardrailsBatchDetectResponse
+
+The `GuardrailsBatchDetectResponse` class wraps `batch_detect` response:
+
+```python Python
+# Example usage of batch_detect with multiple texts
+batch_detect_response = guardrails_client.batch_detect(
+    texts=[safe_prompt, bomb_prompt], 
+    config=copy.deepcopy(sample_detectors)
+)
+
+# Batch checks
+print(f"Batch Response Is Safe: {batch_detect_response.is_safe()}")
+print(f"Batch Response Is Attack: {batch_detect_response.is_attack()}")
+print(f"Batch Response Has Violations: {batch_detect_response.has_violations()}")
+print(f"Batch Response All Violations: {batch_detect_response.get_violations()}")
+
+# Access results for individual texts
+for idx, detection in enumerate(batch_detect_response.batch_detections):
+    print(f"\nResults for text #{idx + 1}:")
+    
+    # Access specific detector results
+    if detection.details.injection_attack:
+        print(f"Injection Attack Safe: {detection.details.injection_attack.safe}")
+        print(f"Injection Attack Score: {detection.details.injection_attack.attack}")
+    
+    # Check safety status for this text
+    print(f"Is Safe: {detection.is_safe()}")
+    print(f"Is Attack: {detection.is_attack()}")
+    print(f"Has Violations: {detection.has_violations()}")
+    print(f"Violations: {detection.get_violations()}")
+
+# Convert entire batch response to dictionary
+print("\nComplete Batch Response Dictionary:")
+print(batch_detect_response.to_dict())
+```
+
 ## Available Guardrails Detectors
 
 - `injection_attack`: Detect prompt injection attempts
@@ -890,7 +940,7 @@ print(add_model_response.to_dict())
 
 ```python Python
 # Check Model Health
-check_saved_model_health = redteam_client.check_saved_model_health(model_saved_name=test_model_saved_name)
+check_saved_model_health = redteam_client.check_saved_model_health(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(check_saved_model_health)
 
@@ -901,11 +951,13 @@ assert check_saved_model_health.status == "healthy"
 
 ```python Python
 # Retrieve model details
-model_details = model_client.get_model(model_saved_name=test_model_saved_name)
+model_details = model_client.get_model(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(model_details)
 
 # Get other fields
+print(model_details.model_saved_name)
+print(model_details.model_version)
 print(model_details.model_name)
 print(model_details.model_config)
 print(model_details.model_config.model_provider)
@@ -940,13 +992,23 @@ new_model_config = copy.deepcopy(sample_model_config)
 new_model_config["model_name"] = "gpt-4o-mini"
 
 # Update the model_saved_name if needed
+# ---------------------------------------------------
+# NOTE:
+# To ensure current stuff does not break, please try creating a new model instead of modifying the existing one.
+# Later, you can delete the old model when changes are made to use the new model.
+# ---------------------------------------------------
 # new_model_config["model_saved_name"] = "New Model Name"
+# new_model_config["model_version"] = "v2"
 
 old_model_saved_name = None
 if new_model_config["model_saved_name"] != test_model_saved_name:
     old_model_saved_name = test_model_saved_name
 
-modify_response = model_client.modify_model(old_model_saved_name=old_model_saved_name, config=new_model_config)
+old_model_version = None
+if new_model_config["model_version"] != test_model_version:
+    old_model_version = test_model_version
+
+modify_response = model_client.modify_model(old_model_saved_name=old_model_saved_name, old_model_version=old_model_version, config=new_model_config)
 
 print(modify_response)
 
@@ -960,7 +1022,7 @@ print(modify_response.to_dict())
 
 ```python Python
 # Remove a model
-delete_response = model_client.delete_model(model_saved_name=test_model_saved_name)
+delete_response = model_client.delete_model(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(delete_response)
 
@@ -996,6 +1058,7 @@ print(deployment_details)
 
 # Get other fields
 print(deployment_details.model_saved_name)
+print(deployment_details.model_version)
 print(deployment_details.input_guardrails_policy)
 print(deployment_details.input_guardrails_policy.policy_name)
 
@@ -1219,7 +1282,7 @@ print(add_redteam_target_response.to_dict())
 
 ```python Python
 # Use a dictionary to configure a redteam task
-add_redteam_model_response = redteam_client.add_task_with_saved_model(config=copy.deepcopy(sample_redteam_model_config),model_saved_name=test_model_saved_name)
+add_redteam_model_response = redteam_client.add_task_with_saved_model(config=copy.deepcopy(sample_redteam_model_config),model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(add_redteam_model_response)
 

{enkryptai_sdk-1.0.3 → enkryptai_sdk-1.0.5}/README.md

@@ -1,6 +1,7 @@
-![Python SDK test](https://github.com/enkryptai/enkryptai-sdk/actions/workflows/test.yaml/badge.svg)
 # Enkrypt AI Python SDK
 
+![Python SDK test](https://github.com/enkryptai/enkryptai-sdk/actions/workflows/test.yaml/badge.svg)
+
 A Python SDK with Guardrails, Models, Deployments, AI Proxy, Datasets and Red Team functionality for API interactions.
 
 See [https://pypi.org/project/enkryptai-sdk](https://pypi.org/project/enkryptai-sdk)
@@ -32,6 +33,7 @@ Also see the API documentation at [https://docs.enkryptai.com](https://docs.enkr
 - [Guardrails Quickstart](#guardrails-quickstart)
 - [Guardrails Response Objects](#guardrails-response-objects)
   - [GuardrailsDetectResponse](#guardrailsdetectresponse)
+  - [GuardrailsBatchDetectResponse](#guardrailsbatchdetectresponse)
 - [Available Guardrails Detectors](#available-guardrails-detectors)
 - [Guardrails Configs](#guardrails-configs)
   - [Injection Attack](#injection-attack)
@@ -158,6 +160,7 @@ redteam_client = RedTeamClient(api_key=ENKRYPT_API_KEY, base_url=ENKRYPT_BASE_UR
 ```python Python
 test_policy_name = "Test Policy"
 test_model_saved_name = "Test Model"
+test_model_version = "v1"
 test_deployment_name = "test-deployment"
 
 pii_original_text = "My email is example@example.com. My phone number is 123-456-7890."
@@ -234,14 +237,15 @@ sample_detectors = {
 ```python Python
 sample_model_config = {
         "model_saved_name": test_model_saved_name,
-        "testing_for": "LLM",
+        "model_version": test_model_version,
+        "testing_for": "foundationModels",
         "model_name": model_name,
-        "modality": "text",
         "model_config": {
-            "model_version": "",
             "model_provider": model_provider,
             "endpoint_url": model_endpoint_url,
             "apikey": OPENAI_API_KEY,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"],
         },
     }
 ```
@@ -252,6 +256,7 @@ sample_model_config = {
 sample_deployment_config = {
         "name": test_deployment_name,
         "model_saved_name": test_model_saved_name,
+        "model_version": test_model_version,
         "input_guardrails_policy": {
             "policy_name": test_policy_name,
             "enabled": True,
@@ -285,8 +290,16 @@ sample_dataset_config = {
         "dataset_name": dataset_name,
         "system_description": "- **Voter Eligibility**: To vote in U.S. elections, individuals must be U.S. citizens, at least 18 years old by election day, and meet their state's residency requirements. - **Voter Registration**: Most states require voters to register ahead of time, with deadlines varying widely. North Dakota is an exception, as it does not require voter registration. - **Identification Requirements**: Thirty-six states enforce voter ID laws, requiring individuals to present identification at polling places. These laws aim to prevent voter fraud but can also lead to disenfranchisement. - **Voting Methods**: Voters can typically choose between in-person voting on election day, early voting, and absentee or mail-in ballots, depending on state regulations. - **Polling Hours**: Polling hours vary by state, with some states allowing extended hours for voters. Its essential for voters to check local polling times to ensure they can cast their ballots. - **Provisional Ballots**: If there are questions about a voter's eligibility, they may be allowed to cast a provisional ballot. This ballot is counted once eligibility is confirmed. - **Election Day Laws**: Many states have laws that protect the rights of voters on election day, including prohibiting intimidation and ensuring access to polling places. - **Campaign Finance Regulations**: Federal and state laws regulate contributions to candidates and political parties to ensure transparency and limit the influence of money in politics. - **Political Advertising**: Campaigns must adhere to rules regarding political advertising, including disclosure requirements about funding sources and content accuracy. - **Voter Intimidation Prohibitions**: Federal laws prohibit any form of voter intimidation or coercion at polling places, ensuring a safe environment for all voters. - **Accessibility Requirements**: The Americans with Disabilities Act mandates that polling places be accessible to individuals with disabilities, ensuring equal access to the electoral process. - **Election Monitoring**: Various organizations are allowed to monitor elections to ensure compliance with laws and regulations. They help maintain transparency and accountability in the electoral process. - **Vote Counting Procedures**: States have specific procedures for counting votes, including the use of electronic voting machines and manual audits to verify results. - **Ballot Design Standards**: States must adhere to certain design standards for ballots to ensure clarity and prevent confusion among voters when casting their votes. - **Post-Election Audits**: Some states conduct post-election audits as a measure of accuracy. These audits help verify that the vote count reflects the actual ballots cast.",
         "policy_description": "",
-        "tools": [],
+        "tools": [
+            {
+                "name": "web_search",
+                "description": "The tool web search is used to search the web for information related to finance."
+            }
+        ],
         "info_pdf_url": "",
+        "scenarios": 1,
+        "categories": 1,
+        "depth": 1,
         "max_prompts": 100,
     }
 ```
@@ -297,16 +310,16 @@ sample_dataset_config = {
 sample_redteam_model_health_config = {
         "target_model_configuration": {
             "model_name": model_name,
-            "testing_for": "LLM",
-            "model_type": "text_2_text",
-            "model_version": "v1",
+            "testing_for": "foundationModels",
+            "model_version": test_model_version,
             "model_source": "https://openai.com",
             "model_provider": model_provider,
             "model_endpoint_url": model_endpoint_url,
             "model_api_key": OPENAI_API_KEY,
             "system_prompt": "",
-            "conversation_template": "",
-            "rate_per_min": 20
+            "rate_per_min": 20,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"]
         },
     }
 ```
@@ -341,16 +354,16 @@ sample_redteam_target_config = {
         },
         "target_model_configuration": {
             "model_name": model_name,
-            "testing_for": "LLM",
-            "model_type": "text_2_text",
-            "model_version": "v1",
+            "testing_for": "foundationModels",
+            "model_version": test_model_version,
             "model_source": "https://openai.com",
             "model_provider": model_provider,
             "model_endpoint_url": model_endpoint_url,
             "model_api_key": OPENAI_API_KEY,
             "system_prompt": "",
-            "conversation_template": "",
-            "rate_per_min": 20
+            "rate_per_min": 20,
+            "input_modalities": ["text"],
+            "output_modalities": ["text"]
         },
     }
 ```
@@ -516,6 +529,43 @@ print(detect_response)
 print(detect_response.to_dict())
 ```
 
+### GuardrailsBatchDetectResponse
+
+The `GuardrailsBatchDetectResponse` class wraps `batch_detect` response:
+
+```python Python
+# Example usage of batch_detect with multiple texts
+batch_detect_response = guardrails_client.batch_detect(
+    texts=[safe_prompt, bomb_prompt], 
+    config=copy.deepcopy(sample_detectors)
+)
+
+# Batch checks
+print(f"Batch Response Is Safe: {batch_detect_response.is_safe()}")
+print(f"Batch Response Is Attack: {batch_detect_response.is_attack()}")
+print(f"Batch Response Has Violations: {batch_detect_response.has_violations()}")
+print(f"Batch Response All Violations: {batch_detect_response.get_violations()}")
+
+# Access results for individual texts
+for idx, detection in enumerate(batch_detect_response.batch_detections):
+    print(f"\nResults for text #{idx + 1}:")
+    
+    # Access specific detector results
+    if detection.details.injection_attack:
+        print(f"Injection Attack Safe: {detection.details.injection_attack.safe}")
+        print(f"Injection Attack Score: {detection.details.injection_attack.attack}")
+    
+    # Check safety status for this text
+    print(f"Is Safe: {detection.is_safe()}")
+    print(f"Is Attack: {detection.is_attack()}")
+    print(f"Has Violations: {detection.has_violations()}")
+    print(f"Violations: {detection.get_violations()}")
+
+# Convert entire batch response to dictionary
+print("\nComplete Batch Response Dictionary:")
+print(batch_detect_response.to_dict())
+```
+
 ## Available Guardrails Detectors
 
 - `injection_attack`: Detect prompt injection attempts
@@ -867,7 +917,7 @@ print(add_model_response.to_dict())
 
 ```python Python
 # Check Model Health
-check_saved_model_health = redteam_client.check_saved_model_health(model_saved_name=test_model_saved_name)
+check_saved_model_health = redteam_client.check_saved_model_health(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(check_saved_model_health)
 
@@ -878,11 +928,13 @@ assert check_saved_model_health.status == "healthy"
 
 ```python Python
 # Retrieve model details
-model_details = model_client.get_model(model_saved_name=test_model_saved_name)
+model_details = model_client.get_model(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(model_details)
 
 # Get other fields
+print(model_details.model_saved_name)
+print(model_details.model_version)
 print(model_details.model_name)
 print(model_details.model_config)
 print(model_details.model_config.model_provider)
@@ -917,13 +969,23 @@ new_model_config = copy.deepcopy(sample_model_config)
 new_model_config["model_name"] = "gpt-4o-mini"
 
 # Update the model_saved_name if needed
+# ---------------------------------------------------
+# NOTE:
+# To ensure current stuff does not break, please try creating a new model instead of modifying the existing one.
+# Later, you can delete the old model when changes are made to use the new model.
+# ---------------------------------------------------
 # new_model_config["model_saved_name"] = "New Model Name"
+# new_model_config["model_version"] = "v2"
 
 old_model_saved_name = None
 if new_model_config["model_saved_name"] != test_model_saved_name:
     old_model_saved_name = test_model_saved_name
 
-modify_response = model_client.modify_model(old_model_saved_name=old_model_saved_name, config=new_model_config)
+old_model_version = None
+if new_model_config["model_version"] != test_model_version:
+    old_model_version = test_model_version
+
+modify_response = model_client.modify_model(old_model_saved_name=old_model_saved_name, old_model_version=old_model_version, config=new_model_config)
 
 print(modify_response)
 
@@ -937,7 +999,7 @@ print(modify_response.to_dict())
 
 ```python Python
 # Remove a model
-delete_response = model_client.delete_model(model_saved_name=test_model_saved_name)
+delete_response = model_client.delete_model(model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(delete_response)
 
@@ -973,6 +1035,7 @@ print(deployment_details)
 
 # Get other fields
 print(deployment_details.model_saved_name)
+print(deployment_details.model_version)
 print(deployment_details.input_guardrails_policy)
 print(deployment_details.input_guardrails_policy.policy_name)
 
@@ -1196,7 +1259,7 @@ print(add_redteam_target_response.to_dict())
 
 ```python Python
 # Use a dictionary to configure a redteam task
-add_redteam_model_response = redteam_client.add_task_with_saved_model(config=copy.deepcopy(sample_redteam_model_config),model_saved_name=test_model_saved_name)
+add_redteam_model_response = redteam_client.add_task_with_saved_model(config=copy.deepcopy(sample_redteam_model_config),model_saved_name=test_model_saved_name, model_version=test_model_version)
 
 print(add_redteam_model_response)
 

{enkryptai_sdk-1.0.3 → enkryptai_sdk-1.0.5}/setup.py

@@ -8,7 +8,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
-    version="1.0.3", # Update this for new versions
+    version="1.0.5", # 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.3 → enkryptai_sdk-1.0.5}/src/enkryptai_sdk/datasets.py

@@ -36,9 +36,23 @@ class DatasetClient(BaseClient):
 
         payload = config.to_dict()
 
+        # If payload["tools"] is None or is an empty list [] or [{}], remove it from the payload
+        if (payload.get("tools") is None or 
+            payload["tools"] == [] or 
+            payload["tools"] == [{}] or
+            payload["tools"] == [{"name": "", "description": ""}]):
+            del payload["tools"]
+
+        # Print payload
+        # print(f"\nAdd Dataset Payload: {payload}")
+
         response = self._request(
             "POST", "/datasets/add-task", headers=headers, json=payload
         )
+
+        # Print response
+        # print(f"\nAdd Dataset Response: {response}")
+
         if response.get("error"):
             raise DatasetClientError(response["error"])
         return DatasetAddTaskResponse.from_dict(response)

{enkryptai_sdk-1.0.3 → enkryptai_sdk-1.0.5}/src/enkryptai_sdk/dto/__init__.py

@@ -62,10 +62,13 @@ __all__ = [
     "GuardrailsModelsResponse",
     "GuardrailDetectors",
     "GuardrailsDetectRequest",
+    "GuardrailsBatchDetectRequest",
     "GuardrailsPolicyDetectRequest",
     "DetectResponseSummary",
     "DetectResponseDetails",
     "GuardrailsDetectResponse",
+    "BatchDetectResponseItem",
+    "GuardrailsBatchDetectResponse",
     "GuardrailsPIIRequest",
     "GuardrailsPIIResponse",
     "GuardrailsHallucinationRequest",

{enkryptai_sdk-1.0.3 → enkryptai_sdk-1.0.5}/src/enkryptai_sdk/dto/datasets.py

@@ -3,17 +3,35 @@ from dataclasses import dataclass, field
 from typing import Dict, List, Optional, Any
 
 
+@dataclass
+class Tool(BaseDTO):
+    name: str
+    description: str
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "Tool":
+        return cls(
+            name=data.get("name", ""),
+            description=data.get("description", "")
+        )
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "name": self.name,
+            "description": self.description
+        }
+
 @dataclass
 class DatasetConfig(BaseDTO):
     dataset_name: str
     system_description: str
     policy_description: str = ""
-    tools: List[str] = field(default_factory=list)
+    tools: List[Tool] = field(default_factory=list)
     info_pdf_url: str = ""
     max_prompts: int = 100
-    # scenarios: int = 2
-    # categories: int = 2
-    # depth: int = 2
+    scenarios: int = 2
+    categories: int = 2
+    depth: int = 2
 
 
 @dataclass

{enkryptai_sdk-1.0.3 → enkryptai_sdk-1.0.5}/src/enkryptai_sdk/dto/deployments.py

@@ -141,6 +141,7 @@ class OutputGuardrailsPolicy(BaseDTO):
 class DeploymentInput(BaseDTO):
     name: str
     model_saved_name: str
+    model_version: str
     input_guardrails_policy: InputGuardrailsPolicy
     output_guardrails_policy: OutputGuardrailsPolicy
     _extra_fields: Dict[str, Any] = field(default_factory=dict)
@@ -153,6 +154,7 @@ class DeploymentInput(BaseDTO):
         return cls(
             name=data.get("name", ""),
             model_saved_name=data.get("model_saved_name", ""),
+            model_version=data.get("model_version", ""),
             input_guardrails_policy=InputGuardrailsPolicy.from_dict(input_policy_data),
             output_guardrails_policy=OutputGuardrailsPolicy.from_dict(output_policy_data)
         )
@@ -161,6 +163,7 @@ class DeploymentInput(BaseDTO):
         result = {
             "name": self.name,
             "model_saved_name": self.model_saved_name,
+            "model_version": self.model_version,
             "input_guardrails_policy": self.input_guardrails_policy.to_dict(),
             "output_guardrails_policy": self.output_guardrails_policy.to_dict()
         }
@@ -172,6 +175,7 @@ class DeploymentInput(BaseDTO):
 class GetDeploymentResponse(BaseDTO):
     name: str
     model_saved_name: str
+    model_version: str
     input_guardrails_policy: InputGuardrailsPolicy
     output_guardrails_policy: OutputGuardrailsPolicy
     created_at: str
@@ -186,6 +190,7 @@ class GetDeploymentResponse(BaseDTO):
         return cls(
             name=data.get("name", ""),
             model_saved_name=data.get("model_saved_name", ""),
+            model_version=data.get("model_version", ""),
             input_guardrails_policy=InputGuardrailsPolicy.from_dict(input_policy_data),
             output_guardrails_policy=OutputGuardrailsPolicy.from_dict(output_policy_data),
             updated_at=data.get("updated_at", ""),
@@ -197,6 +202,7 @@ class GetDeploymentResponse(BaseDTO):
         return {
             "name": self.name,
             "model_saved_name": self.model_saved_name,
+            "model_version": self.model_version,
             "input_guardrails_policy": self.input_guardrails_policy.to_dict(),
             "output_guardrails_policy": self.output_guardrails_policy.to_dict(),
             "updated_at": self.updated_at,
@@ -293,6 +299,7 @@ class DeploymentSummary(BaseDTO):
     created_at: str
     updated_at: str
     model_saved_name: str
+    model_version: str
     project_name: str
     
     @classmethod
@@ -303,6 +310,7 @@ class DeploymentSummary(BaseDTO):
             created_at=data.get("created_at", ""),
             updated_at=data.get("updated_at", ""),
             model_saved_name=data.get("model_saved_name", ""),
+            model_version=data.get("model_version", ""),
             project_name=data.get("project_name", "")
         )
     
@@ -313,6 +321,7 @@ class DeploymentSummary(BaseDTO):
             "created_at": self.created_at,
             "updated_at": self.updated_at,
             "model_saved_name": self.model_saved_name,
+            "model_version": self.model_version,
             "project_name": self.project_name
         }