Skip to content

Commit 45bd0e3

Browse files
Fix guardrails orchestrator for 2.24 (#558) (#559)
* fix: change API schema for email address detection and fix image and logic * fix: typo in id of log Co-authored-by: Shelton Cyril <sheltoncyril@gmail.com>
1 parent 58828cd commit 45bd0e3

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/model_explainability/guardrails/test_guardrails.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ def test_guardrails_builtin_detectors_unsuitable_input(
173173
verify_builtin_detector_unsuitable_input_response(
174174
response=response,
175175
detector_id="regex",
176-
detection_name="EmailAddress",
176+
detection_name="email_address",
177177
detection_type="pii",
178178
detection_text=EXAMPLE_EMAIL_ADDRESS,
179179
)
@@ -194,7 +194,7 @@ def test_guardrails_builtin_detectors_unsuitable_output(
194194
)
195195

196196
verify_builtin_detector_unsuitable_output_response(
197-
response=response, detector_id="regex", detection_name="EmailAddress", detection_type="pii"
197+
response=response, detector_id="regex", detection_name="email_address", detection_type="pii"
198198
)
199199

200200
@pytest.mark.parametrize(
@@ -205,7 +205,7 @@ def test_guardrails_builtin_detectors_unsuitable_output(
205205
PII_ENDPOINT,
206206
id="harmless_input",
207207
),
208-
pytest.param(PROMPT_WITH_PII, "/passthrough", id="pastthrough_endpoint"),
208+
pytest.param(PROMPT_WITH_PII, "/passthrough", id="passthrough_endpoint"),
209209
],
210210
)
211211
def test_guardrails_builtin_detectors_negative_detection(
@@ -235,7 +235,7 @@ def test_guardrails_builtin_detectors_negative_detection(
235235
[
236236
pytest.param(
237237
{"name": "test-guardrails-huggingface"},
238-
MinIo.PodConfig.QWEN_MINIO_CONFIG,
238+
MinIo.PodConfig.QWEN_HAP_BPIV2_MINIO_CONFIG,
239239
{"bucket": "llms"},
240240
{
241241
"orchestrator_config_data": {
@@ -363,7 +363,7 @@ def test_guardrails_standalone_detector_endpoint(
363363

364364
assert "detections" in data
365365

366-
score = data.get("score")
366+
score = data["detections"][0]["score"]
367367
assert score > 0.9, f"Expected score > 0.9, got {score}"
368368

369369

0 commit comments

Comments
 (0)