From cd56501b3de3c22984df15d0036f806c2492c74c Mon Sep 17 00:00:00 2001 From: Oceane Bordeau Date: Mon, 23 Mar 2026 18:08:29 -0400 Subject: [PATCH 1/3] Fix SDS cassette --- .../vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json b/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json index e327cd01c02..ab7b69abd48 100644 --- a/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json +++ b/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json @@ -43,6 +43,6 @@ "x-ratelimit-reset": "22", "x-ratelimit-name": "ai_guard_evaluate_per_org" }, - "body": "{\"data\":{\"id\":\"21df3f84-982c-43f5-a168-d5adf68dba3e\",\"type\":\"evaluations\",\"attributes\":{\"action\":\"ALLOW\",\"is_blocking_enabled\":true,\"reason\":\"No rule match.\",\"sds_findings\":[{\"rule_display_name\":\"Standard Email Address Scanner\",\"rule_tag\":\"email_address\",\"category\":\"email_address\",\"matched_text\":\"john.smith@acmebank.com\",\"location\":{\"path\":\"messages[0].content\",\"start_index\":35,\"end_index_exclusive\":58}},{\"rule_display_name\":\"US Social Security Number Scanner\",\"rule_tag\":\"us_ssn\",\"category\":\"pii\",\"matched_text\":\"456-78-9012\",\"location\":{\"path\":\"messages[0].content\",\"start_index\":73,\"end_index_exclusive\":84}}],\"tag_probs\":{\"authority-override\":1.9361263070560852e-7,\"data-exfiltration\":0.008578038945358801,\"denial-of-service-tool-call\":9.088342229901514e-7,\"destructive-tool-call\":1.9361263070560852e-7,\"indirect-prompt-injection\":1.9361263070560852e-7,\"instruction-override\":1.9361263070560852e-7,\"jailbreak\":4.3201989441410404e-7,\"obfuscation\":3.12816276770711e-7,\"role-play\":0,\"security-exploit\":0,\"system-prompt-extraction\":0.00001676278181728108},\"tags\":[]}}}" + "body": "{\"data\":{\"id\":\"3cb5f17b-b7d8-49f4-9bac-cc3f49c22960\",\"type\":\"evaluations\",\"attributes\":{\"action\":\"ALLOW\",\"is_blocking_enabled\":true,\"reason\":\"No rule match.\",\"sds_findings\":[{\"rule_display_name\":\"Standard Email Address Scanner\",\"rule_tag\":\"email_address\",\"category\":\"email_address\",\"location\":{\"path\":\"messages[0].content\",\"start_index\":35,\"end_index_exclusive\":58}},{\"rule_display_name\":\"US Social Security Number Scanner\",\"rule_tag\":\"us_ssn\",\"category\":\"pii\",\"location\":{\"path\":\"messages[0].content\",\"start_index\":73,\"end_index_exclusive\":84}}],\"tag_probs\":{\"authority-override\":1.9361263070560852e-7,\"data-exfiltration\":0.022978067534292212,\"denial-of-service-tool-call\":4.3201989441410404e-7,\"destructive-tool-call\":1.9361263070560852e-7,\"indirect-prompt-injection\":1.9361263070560852e-7,\"instruction-override\":0,\"jailbreak\":6.704270871793483e-7,\"obfuscation\":5.512234979576647e-7,\"role-play\":0,\"security-exploit\":0,\"system-prompt-extraction\":0.000058363296655428165},\"tags\":[]}}}" } } \ No newline at end of file From 96bc6e012c0ecf0f2e1a048991ca32b1eb67a1c6 Mon Sep 17 00:00:00 2001 From: Oceane Bordeau Date: Mon, 23 Mar 2026 18:13:42 -0400 Subject: [PATCH 2/3] Remove content encoding gzip --- .../vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json b/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json index ab7b69abd48..ec0caf40a47 100644 --- a/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json +++ b/utils/build/docker/vcr/cassettes/aiguard/aiguard_evaluate_post_ee2b240f.json @@ -29,7 +29,6 @@ "message": "OK" }, "headers": { - "content-encoding": "gzip", "content-type": "application/vnd.api+json", "vary": "Accept-Encoding", "x-frame-options": "SAMEORIGIN", From 55955b53a805f64ccb59a7b5b37e42f47c697e90 Mon Sep 17 00:00:00 2001 From: Oceane Bordeau Date: Tue, 24 Mar 2026 10:15:38 -0400 Subject: [PATCH 3/3] Fix ST for AI Guard --- tests/ai_guard/test_ai_guard_sdk.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/ai_guard/test_ai_guard_sdk.py b/tests/ai_guard/test_ai_guard_sdk.py index f9fbfc78538..7813044d2c9 100644 --- a/tests/ai_guard/test_ai_guard_sdk.py +++ b/tests/ai_guard/test_ai_guard_sdk.py @@ -353,7 +353,6 @@ def validate(span: DataDogLibrarySpan): assert _assert_key(sd, "rule_display_name") assert _assert_key(sd, "rule_tag") assert _assert_key(sd, "category") - assert _assert_key(sd, "matched_text") location = _assert_key(sd, "location") assert _assert_key(location, "start_index") is not None assert _assert_key(location, "end_index_exclusive") is not None @@ -391,7 +390,6 @@ def test_sds_in_response(self): assert _assert_key(finding, "rule_display_name") assert _assert_key(finding, "rule_tag") assert _assert_key(finding, "category") - assert _assert_key(finding, "matched_text") location = _assert_key(finding, "location") assert _assert_key(location, "start_index") is not None assert _assert_key(location, "end_index_exclusive") is not None