From de2cac7b672d692cd9567f409fc281c29e092407 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 15 Sep 2026 22:38:42 +0000
Subject: [PATCH] fix(ui): default custom PII entities to MASK and classify
 test tier

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 ...ration.test.tsx => pii_configuration.integration.test.tsx} | 4 ++--
 .../(dashboard)/guardrails/_components/pii_configuration.tsx  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)
 rename ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/{pii_configuration.test.tsx => pii_configuration.integration.test.tsx} (95%)

diff --git a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.integration.test.tsx
similarity index 95%
rename from ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.test.tsx
rename to ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.integration.test.tsx
index b0f0265f067..969844c58ed 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.integration.test.tsx
@@ -33,13 +33,13 @@ describe("PiiConfiguration", () => {
     expect(screen.getByText("NO FODSELSNUMMER")).toBeInTheDocument();
   });
 
-  it("adds a normalized custom entity with the first action", () => {
+  it("adds a normalized custom entity with MASK", () => {
     const onEntitySelect = vi.fn();
     const onActionSelect = vi.fn();
     render(
       <PiiConfiguration
         entities={["PERSON"]}
-        actions={["MASK", "BLOCK"]}
+        actions={["BLOCK", "MASK"]}
         selectedEntities={[]}
         selectedActions={{}}
         onEntitySelect={onEntitySelect}
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx
index 4efb795bb6a..e6efdf53670 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx
@@ -78,7 +78,7 @@ const PiiConfiguration: React.FC<PiiConfigurationProps> = ({
     }
 
     onEntitySelect(normalizedEntityName);
-    onActionSelect(normalizedEntityName, actions[0] ?? "MASK");
+    onActionSelect(normalizedEntityName, "MASK");
   };
 
   return (