From 41bf11a93042c3f7807e6d00882c8c02cbec2042 Mon Sep 17 00:00:00 2001
From: yucheng <yucheng@berri.ai>
Date: Wed, 22 Jul 2026 00:08:01 +0000
Subject: [PATCH] test(ui): preserve guardrail helper exports in panel mock

---
 .../_components/GuardrailsPanel.test.tsx         | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx
index 2ad332ae9ed..6d05dfaba58 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx
@@ -48,12 +48,16 @@ vi.mock("@/utils/roles", () => ({
   isAdminRole: vi.fn((role: string) => role === "admin"),
 }));
 
-vi.mock("./guardrail_info_helpers", () => ({
-  getGuardrailLogoAndName: vi.fn(() => ({
-    logo: null,
-    displayName: "Test Provider",
-  })),
-}));
+vi.mock("./guardrail_info_helpers", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("./guardrail_info_helpers")>();
+  return {
+    ...actual,
+    getGuardrailLogoAndName: vi.fn(() => ({
+      logo: null,
+      displayName: "Test Provider",
+    })),
+  };
+});
 
 beforeAll(() => {
   Object.defineProperty(window, "matchMedia", {