From f32929400501345d4773e20bde1c09496c24d665 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Tue, 17 Feb 2026 16:57:31 -0800
Subject: [PATCH] fix(ui): show category count badge in guardrail selection
 modal

Category-based guardrails (like EU AI Act) now display an orange
tag showing how many categories they contain, matching the existing
pattern count tag for pattern-based guardrails.

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 .../src/components/policies/guardrail_selection_modal.tsx    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/ui/litellm-dashboard/src/components/policies/guardrail_selection_modal.tsx b/ui/litellm-dashboard/src/components/policies/guardrail_selection_modal.tsx
index 973c8722bba..2984478ba5a 100644
--- a/ui/litellm-dashboard/src/components/policies/guardrail_selection_modal.tsx
+++ b/ui/litellm-dashboard/src/components/policies/guardrail_selection_modal.tsx
@@ -201,6 +201,11 @@ const GuardrailSelectionModal: React.FC<GuardrailSelectionModalProps> = ({
                         {guardrail.definition.litellm_params.patterns.length} pattern(s)
                       </Tag>
                     )}
+                    {guardrail.definition?.litellm_params?.categories && (
+                      <Tag className="text-xs" color="orange">
+                        {guardrail.definition.litellm_params.categories.length} category/categories
+                      </Tag>
+                    )}
                   </div>
                 </div>
               </div>