From 85d417effed1e5975218fc90da788b3332d2b41e Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 15 Sep 2026 23:16:21 +0000
Subject: [PATCH] refactor(ui): build PII entity category map without mutation

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../guardrails/_components/pii_configuration.tsx | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

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 609899aa95a..84b60f1e039 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
@@ -28,16 +28,12 @@ const PiiConfiguration: React.FC<PiiConfigurationProps> = ({
       ? [...entityCategories, { category: "Custom", entities: customEntities }]
       : entityCategories;
 
-  // Create a lookup map to quickly find an entity's category
-  const entityToCategoryMap = new Map<string, string>();
-  entityCategories.forEach((category) => {
-    category.entities.forEach((entity) => {
-      entityToCategoryMap.set(entity, category.category);
-    });
-  });
-  customEntities.forEach((entity) => {
-    entityToCategoryMap.set(entity, "Custom");
-  });
+  const entityToCategoryMap = new Map<string, string>([
+    ...entityCategories.flatMap((category) =>
+      category.entities.map((entity): [string, string] => [entity, category.category]),
+    ),
+    ...customEntities.map((entity): [string, string] => [entity, "Custom"]),
+  ]);
 
   // Filter entities based on selected categories
   const filteredEntities = allEntities.filter((entity) => {