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.
@@ -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] = ""
@@ -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.12
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, # Can also be a list of 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=EA2jwdoYqOjQ8GuRawsZ3syIvVinwhvodTLQlvt51Mo,11399
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=08ilpQkp_tpUDzM5Jt-mrWdHlbBgYst0plzzT4G-Gds,14184
22
- enkryptai_sdk/dto/red_team.py,sha256=UNqiPxKqk8TJ-hLtBCWRvtoXyGCFiz3X7bKpx8sTQxo,18298
23
- enkryptai_sdk-1.0.12.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
- enkryptai_sdk-1.0.12.dist-info/METADATA,sha256=i6FJiWp2gJs3mZZ4YkQWGIvmHKCyD6PLpvvHzGNUKLQ,62092
25
- enkryptai_sdk-1.0.12.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
26
- enkryptai_sdk-1.0.12.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
27
- enkryptai_sdk-1.0.12.dist-info/RECORD,,
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,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.8.0)
2
+ Generator: setuptools (80.9.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5