enkryptai-sdk 1.0.12__py3-none-any.whl → 1.0.14__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- enkryptai_sdk/dto/models.py +2 -0
- enkryptai_sdk/dto/red_team.py +1 -0
- enkryptai_sdk/models.py +2 -0
- {enkryptai_sdk-1.0.12.dist-info → enkryptai_sdk-1.0.14.dist-info}/METADATA +11 -12
- {enkryptai_sdk-1.0.12.dist-info → enkryptai_sdk-1.0.14.dist-info}/RECORD +8 -8
- {enkryptai_sdk-1.0.12.dist-info → enkryptai_sdk-1.0.14.dist-info}/WHEEL +1 -1
- {enkryptai_sdk-1.0.12.dist-info → enkryptai_sdk-1.0.14.dist-info}/licenses/LICENSE +0 -0
- {enkryptai_sdk-1.0.12.dist-info → enkryptai_sdk-1.0.14.dist-info}/top_level.txt +0 -0
enkryptai_sdk/dto/models.py
CHANGED
|
@@ -39,6 +39,7 @@ class ModelProviders(str, Enum):
|
|
|
39
39
|
ANTHROPIC_COMPATIBLE = "anthropic_compatible"
|
|
40
40
|
CUSTOM = "custom"
|
|
41
41
|
HR = "hr"
|
|
42
|
+
URL = "url"
|
|
42
43
|
|
|
43
44
|
|
|
44
45
|
@dataclass
|
|
@@ -166,6 +167,7 @@ class ModelConfigDetails(BaseDTO):
|
|
|
166
167
|
auth_data: AuthData = field(default_factory=AuthData)
|
|
167
168
|
input_modalities: List[InputModality] = field(default_factory=list)
|
|
168
169
|
output_modalities: List[OutputModality] = field(default_factory=list)
|
|
170
|
+
custom_curl_command: Optional[str] = None
|
|
169
171
|
custom_headers: List[CustomHeader] = field(default_factory=list)
|
|
170
172
|
custom_payload: Dict[str, Any] = field(default_factory=dict)
|
|
171
173
|
custom_response_content_type: Optional[str] = ""
|
enkryptai_sdk/dto/red_team.py
CHANGED
|
@@ -359,6 +359,7 @@ class TargetModelConfiguration(BaseDTO):
|
|
|
359
359
|
model_api_key: Optional[str] = None
|
|
360
360
|
input_modalities: List[InputModality] = field(default_factory=list)
|
|
361
361
|
output_modalities: List[OutputModality] = field(default_factory=list)
|
|
362
|
+
custom_curl_command: Optional[str] = None
|
|
362
363
|
custom_headers: List[CustomHeader] = field(default_factory=list)
|
|
363
364
|
custom_payload: Dict[str, Any] = field(default_factory=dict)
|
|
364
365
|
custom_response_content_type: Optional[str] = ""
|
enkryptai_sdk/models.py
CHANGED
|
@@ -77,6 +77,7 @@ class ModelClient(BaseClient):
|
|
|
77
77
|
"tools": config.model_config.tools,
|
|
78
78
|
"input_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.input_modalities],
|
|
79
79
|
"output_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.output_modalities],
|
|
80
|
+
"custom_curl_command": config.model_config.custom_curl_command,
|
|
80
81
|
"custom_headers": custom_headers,
|
|
81
82
|
"custom_payload": config.model_config.custom_payload,
|
|
82
83
|
"custom_response_content_type": config.model_config.custom_response_content_type,
|
|
@@ -225,6 +226,7 @@ class ModelClient(BaseClient):
|
|
|
225
226
|
"tools": config.model_config.tools,
|
|
226
227
|
"input_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.input_modalities],
|
|
227
228
|
"output_modalities": [m.value if hasattr(m, 'value') else m for m in config.model_config.output_modalities],
|
|
229
|
+
"custom_curl_command": config.model_config.custom_curl_command,
|
|
228
230
|
"custom_headers": custom_headers,
|
|
229
231
|
"custom_payload": config.model_config.custom_payload,
|
|
230
232
|
"custom_response_content_type": config.model_config.custom_response_content_type,
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: enkryptai-sdk
|
|
3
|
-
Version: 1.0.
|
|
3
|
+
Version: 1.0.14
|
|
4
4
|
Summary: A Python SDK with guardrails and red teaming functionality for API interactions
|
|
5
5
|
Home-page: https://github.com/enkryptai/enkryptai-sdk
|
|
6
6
|
Author: Enkrypt AI Team
|
|
@@ -271,14 +271,7 @@ sample_detectors = {
|
|
|
271
271
|
},
|
|
272
272
|
"bias": {
|
|
273
273
|
"enabled": False
|
|
274
|
-
}
|
|
275
|
-
"copyright_ip": {
|
|
276
|
-
"enabled": False
|
|
277
|
-
},
|
|
278
|
-
"system_prompt": {
|
|
279
|
-
"enabled": False,
|
|
280
|
-
"index": "system"
|
|
281
|
-
},
|
|
274
|
+
}
|
|
282
275
|
}
|
|
283
276
|
```
|
|
284
277
|
|
|
@@ -701,8 +694,8 @@ print(batch_detect_response.to_dict())
|
|
|
701
694
|
- `nsfw`: Filter inappropriate content
|
|
702
695
|
- `toxicity`: Detect toxic language
|
|
703
696
|
- `pii`: Detect personal information
|
|
704
|
-
- `copyright_ip`: Check for copyright/IP violations
|
|
705
|
-
- `system_prompt`: Detect system prompt leaks
|
|
697
|
+
- `copyright_ip`: Check for copyright/IP violations *(Coming soon)*
|
|
698
|
+
- `system_prompt`: Detect system prompt leaks *(Coming soon)*
|
|
706
699
|
- `keyword_detector`: Check for specific keywords
|
|
707
700
|
|
|
708
701
|
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"])
|
|
|
764
757
|
|
|
765
758
|
### [Copyright IP](https://docs.enkryptai.com/guardrails-api-reference/Copyright_IP_Leak_Detector)
|
|
766
759
|
|
|
760
|
+
*(Coming soon)*
|
|
761
|
+
|
|
767
762
|
```python Python
|
|
768
763
|
guardrails_config = GuardrailsConfig.copyright_ip()
|
|
769
764
|
```
|
|
770
765
|
|
|
771
766
|
### [System Prompt](https://docs.enkryptai.com/guardrails-api-reference/System_Prompt_Leak_Detector)
|
|
772
767
|
|
|
768
|
+
*(Coming soon)*
|
|
769
|
+
|
|
773
770
|
```python Python
|
|
774
771
|
guardrails_config = GuardrailsConfig.system_prompt(index="system")
|
|
775
772
|
```
|
|
@@ -972,6 +969,8 @@ print(relevancy_response.to_dict())
|
|
|
972
969
|
|
|
973
970
|
### [Check Hallucination](https://docs.enkryptai.com/guardrails-api-reference/Hallucination)
|
|
974
971
|
|
|
972
|
+
*(Coming soon)*
|
|
973
|
+
|
|
975
974
|
Detect hallucinations in an LLM's response:
|
|
976
975
|
|
|
977
976
|
```python Python
|
|
@@ -1057,7 +1056,7 @@ print(atomize_response.to_dict())
|
|
|
1057
1056
|
# Add a code of conduct policy
|
|
1058
1057
|
add_policy_response = coc_client.add_policy(
|
|
1059
1058
|
policy_name=test_coc_policy_name,
|
|
1060
|
-
policy_rules=example_coc_policy_rules,
|
|
1059
|
+
policy_rules=example_coc_policy_rules, # Can also be a list of rules
|
|
1061
1060
|
total_rules=4,
|
|
1062
1061
|
policy_file="/path/to/your/policy.pdf"
|
|
1063
1062
|
# policy_text=example_coc_policy_text, # Optional: Use this if you want to add a policy text instead of a file
|
|
@@ -8,7 +8,7 @@ enkryptai_sdk/deployments.py,sha256=A7XZ2JwrMod9V4_aV8bFY_Soh9E3jHdwaTuJ9BwXuyk,
|
|
|
8
8
|
enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
|
|
9
9
|
enkryptai_sdk/guardrails.py,sha256=NluimOA0gM9N3S_q47LTUeG97t9PlYqPHlZahDPkJvI,16365
|
|
10
10
|
enkryptai_sdk/guardrails_old.py,sha256=SgzPZkTzbAPD9XfmYNG6M1-TrzbhDHpAkI3FjnVWS_s,6434
|
|
11
|
-
enkryptai_sdk/models.py,sha256=
|
|
11
|
+
enkryptai_sdk/models.py,sha256=rrLTT3i96flWidVrr67j6VZ6XmkdxwEzlF4S4aoVmOQ,11559
|
|
12
12
|
enkryptai_sdk/red_team.py,sha256=A1b6R5d3dtZMfb5npBrSu4RX2X-C9V4hubyO9g3BZlA,18441
|
|
13
13
|
enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
|
|
14
14
|
enkryptai_sdk/dto/__init__.py,sha256=wHgIv_OCnVMJOys-vqImF59ifogDrMcgxVRmfNayVvc,2761
|
|
@@ -18,10 +18,10 @@ enkryptai_sdk/dto/coc.py,sha256=Lp2aat_24J4KuUg4BeJl9S39tEak8Bw15eJ4cQDrRQk,4749
|
|
|
18
18
|
enkryptai_sdk/dto/datasets.py,sha256=RFA9CmbhD-QDDyweBq_k9iBd00b6I6SWmdP9DPNd9fc,5002
|
|
19
19
|
enkryptai_sdk/dto/deployments.py,sha256=Aw4b8tDA3FYIomqDvCjblCXTagL4bT8Fx91X0SFXs40,11216
|
|
20
20
|
enkryptai_sdk/dto/guardrails.py,sha256=wfKm2R0uJAq9nljLBPuuT05tj1nBqZraga-5lMHd3n8,49812
|
|
21
|
-
enkryptai_sdk/dto/models.py,sha256=
|
|
22
|
-
enkryptai_sdk/dto/red_team.py,sha256=
|
|
23
|
-
enkryptai_sdk-1.0.
|
|
24
|
-
enkryptai_sdk-1.0.
|
|
25
|
-
enkryptai_sdk-1.0.
|
|
26
|
-
enkryptai_sdk-1.0.
|
|
27
|
-
enkryptai_sdk-1.0.
|
|
21
|
+
enkryptai_sdk/dto/models.py,sha256=zldbvYV5zcg1J3UZh4UnaeM1cBx-_LCCyW-LtfBcjaQ,14246
|
|
22
|
+
enkryptai_sdk/dto/red_team.py,sha256=GD_HxFhoaEixyaZLVdoLHmHDXFRf9b-Wq6RT-UJNQfM,18344
|
|
23
|
+
enkryptai_sdk-1.0.14.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
24
|
+
enkryptai_sdk-1.0.14.dist-info/METADATA,sha256=m6amEeDX9HURaZab4Wn_LMISj8WmxUQ8pTRRtd3Ogwo,62009
|
|
25
|
+
enkryptai_sdk-1.0.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
26
|
+
enkryptai_sdk-1.0.14.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
|
|
27
|
+
enkryptai_sdk-1.0.14.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|