diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/agents/useAgents.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/agents/useAgents.test.ts
index 44fc6a96836..6137e35b5b9 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/agents/useAgents.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/agents/useAgents.test.ts
@@ -81,7 +81,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -144,7 +143,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAgents(), { wrapper });
@@ -168,7 +166,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAgents(), { wrapper });
@@ -192,7 +189,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAgents(), { wrapper });
@@ -216,7 +212,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAgents(), { wrapper });
@@ -240,7 +235,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAgents(), { wrapper });
@@ -283,7 +277,6 @@ describe("useAgents", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAgents(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/credentials/useCredentials.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/credentials/useCredentials.test.ts
index ee903628f08..3e2d72df1ce 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/credentials/useCredentials.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/credentials/useCredentials.test.ts
@@ -66,7 +66,6 @@ describe("useCredentials", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -129,7 +128,6 @@ describe("useCredentials", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCredentials(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/customers/useCustomers.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/customers/useCustomers.test.ts
index 716d6f75399..8e32c063e76 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/customers/useCustomers.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/customers/useCustomers.test.ts
@@ -83,7 +83,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -146,7 +145,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCustomers(), { wrapper });
@@ -170,7 +168,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCustomers(), { wrapper });
@@ -194,7 +191,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCustomers(), { wrapper });
@@ -218,7 +214,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCustomers(), { wrapper });
@@ -242,7 +237,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCustomers(), { wrapper });
@@ -285,7 +279,6 @@ describe("useCustomers", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCustomers(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/guardrails/useGuardrails.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/guardrails/useGuardrails.test.ts
index 8db520eecd0..b05b10b318a 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/guardrails/useGuardrails.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/guardrails/useGuardrails.test.ts
@@ -51,7 +51,6 @@ describe("useGuardrails", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -114,7 +113,6 @@ describe("useGuardrails", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useGuardrails(), { wrapper });
@@ -138,7 +136,6 @@ describe("useGuardrails", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useGuardrails(), { wrapper });
@@ -162,7 +159,6 @@ describe("useGuardrails", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useGuardrails(), { wrapper });
@@ -186,7 +182,6 @@ describe("useGuardrails", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useGuardrails(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/keys/useKeys.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/keys/useKeys.test.ts
index 1e700e572d0..bce0d970233 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/keys/useKeys.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/keys/useKeys.test.ts
@@ -182,7 +182,6 @@ describe("useKeys", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
// Reset fetch mock
@@ -273,7 +272,6 @@ describe("useKeys", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useKeys(1, 10), { wrapper });
@@ -485,7 +483,6 @@ describe("useDeletedKeys", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
// Reset fetch mock
@@ -596,7 +593,6 @@ describe("useDeletedKeys", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useDeletedKeys(1, 10), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
index 2539cc63f95..3a352aef337 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
@@ -67,7 +67,6 @@ describe("useModelsInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -164,7 +163,6 @@ describe("useModelsInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useModelsInfo(), { wrapper });
@@ -184,7 +182,6 @@ describe("useModelsInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useModelsInfo(), { wrapper });
@@ -204,7 +201,6 @@ describe("useModelsInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useModelsInfo(), { wrapper });
@@ -224,7 +220,6 @@ describe("useModelsInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useModelsInfo(), { wrapper });
@@ -258,7 +253,6 @@ describe("useModelHub", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -322,7 +316,6 @@ describe("useModelHub", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useModelHub(), { wrapper });
@@ -356,7 +349,6 @@ describe("useAllProxyModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -428,7 +420,6 @@ describe("useAllProxyModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAllProxyModels(), { wrapper });
@@ -448,7 +439,6 @@ describe("useAllProxyModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAllProxyModels(), { wrapper });
@@ -468,7 +458,6 @@ describe("useAllProxyModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useAllProxyModels(), { wrapper });
@@ -502,7 +491,6 @@ describe("useSelectedTeamModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -574,7 +562,6 @@ describe("useSelectedTeamModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSelectedTeamModels("team-1"), { wrapper });
@@ -594,7 +581,6 @@ describe("useSelectedTeamModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSelectedTeamModels("team-1"), { wrapper });
@@ -614,7 +600,6 @@ describe("useSelectedTeamModels", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSelectedTeamModels("team-1"), { wrapper });
@@ -673,7 +658,6 @@ describe("useInfiniteModelInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -802,7 +786,6 @@ describe("useInfiniteModelInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useInfiniteModelInfo(), { wrapper });
@@ -822,7 +805,6 @@ describe("useInfiniteModelInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useInfiniteModelInfo(), { wrapper });
@@ -842,7 +824,6 @@ describe("useInfiniteModelInfo", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useInfiniteModelInfo(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/organizations/useOrganizations.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/organizations/useOrganizations.test.ts
index 66c005f37c4..57f580101ff 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/organizations/useOrganizations.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/organizations/useOrganizations.test.ts
@@ -82,7 +82,6 @@ describe("useOrganizations", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -145,7 +144,6 @@ describe("useOrganizations", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useOrganizations(), { wrapper });
@@ -169,7 +167,6 @@ describe("useOrganizations", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useOrganizations(), { wrapper });
@@ -193,7 +190,6 @@ describe("useOrganizations", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useOrganizations(), { wrapper });
@@ -217,7 +213,6 @@ describe("useOrganizations", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useOrganizations(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/proxyConfig/useProxyConfig.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/proxyConfig/useProxyConfig.test.ts
index 4ba80df5c6c..3a808903803 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/proxyConfig/useProxyConfig.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/proxyConfig/useProxyConfig.test.ts
@@ -130,7 +130,6 @@ describe("useProxyConfig", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
fetchSpy = vi.fn();
@@ -218,7 +217,6 @@ describe("useProxyConfig", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useProxyConfig(ConfigType.GENERAL_SETTINGS), { wrapper });
@@ -301,7 +299,6 @@ describe("useDeleteProxyConfigField", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
fetchSpy = vi.fn();
@@ -388,7 +385,6 @@ describe("useDeleteProxyConfigField", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useDeleteProxyConfigField(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/router/useRouterFields.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/router/useRouterFields.test.ts
index a9f5e1047ac..4b986e587eb 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/router/useRouterFields.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/router/useRouterFields.test.ts
@@ -77,7 +77,6 @@ describe("useRouterFields", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -159,7 +158,6 @@ describe("useRouterFields", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useRouterFields(), { wrapper });
@@ -182,7 +180,6 @@ describe("useRouterFields", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useRouterFields(), { wrapper });
@@ -205,7 +202,6 @@ describe("useRouterFields", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useRouterFields(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useEditSSOSettings.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useEditSSOSettings.test.ts
index 8b2fee6cf30..ebfa1d5834b 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useEditSSOSettings.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useEditSSOSettings.test.ts
@@ -44,7 +44,6 @@ describe("useEditSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -113,7 +112,6 @@ describe("useEditSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useEditSSOSettings(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useSSOSettings.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useSSOSettings.test.ts
index 4e8d892b5d8..ca5f0c01803 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useSSOSettings.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/sso/useSSOSettings.test.ts
@@ -79,7 +79,6 @@ describe("useSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -143,7 +142,6 @@ describe("useSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSSOSettings(), { wrapper });
@@ -164,7 +162,6 @@ describe("useSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSSOSettings(), { wrapper });
@@ -185,7 +182,6 @@ describe("useSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSSOSettings(), { wrapper });
@@ -206,7 +202,6 @@ describe("useSSOSettings", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useSSOSettings(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/storeModelInDB/useStoreModelInDB.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/storeModelInDB/useStoreModelInDB.test.ts
index dd69e8c8791..14e6fb6f0f5 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/storeModelInDB/useStoreModelInDB.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/storeModelInDB/useStoreModelInDB.test.ts
@@ -93,7 +93,6 @@ describe("useStoreModelInDB", () => {
userEmail: null,
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
} as any);
const { result } = renderHook(() => useStoreModelInDB(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/tags/useTags.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/tags/useTags.test.ts
index a1751339568..5bd279f7b4d 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/tags/useTags.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/tags/useTags.test.ts
@@ -83,7 +83,6 @@ describe("useTags", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -146,7 +145,6 @@ describe("useTags", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTags(), { wrapper });
@@ -170,7 +168,6 @@ describe("useTags", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTags(), { wrapper });
@@ -194,7 +191,6 @@ describe("useTags", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTags(), { wrapper });
@@ -218,7 +214,6 @@ describe("useTags", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTags(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/teams/useTeams.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/teams/useTeams.test.ts
index 20f034ada36..96f4796edea 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/teams/useTeams.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/teams/useTeams.test.ts
@@ -79,7 +79,6 @@ describe("useTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -150,7 +149,6 @@ describe("useTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTeams(), { wrapper });
@@ -174,7 +172,6 @@ describe("useTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTeams(), { wrapper });
@@ -250,7 +247,6 @@ describe("useTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTeams(), { wrapper });
@@ -276,7 +272,6 @@ describe("useTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTeams(), { wrapper });
@@ -312,7 +307,6 @@ describe("useTeam", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -376,7 +370,6 @@ describe("useTeam", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useTeam("team-1"), { wrapper });
@@ -433,7 +426,6 @@ describe("useTeam", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
// Import useQueryClient to get access to query client
@@ -465,7 +457,6 @@ describe("useTeam", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const testQueryFnMissingTeamId = async () => {
@@ -651,7 +642,6 @@ describe("useDeletedTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
global.fetch = vi.fn();
@@ -719,7 +709,6 @@ describe("useDeletedTeams", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useDeletedTeams(1, 10, {}), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.test.ts
index e435d9ac242..6ecb33f0e02 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.test.ts
@@ -126,7 +126,6 @@ describe("useAuthorized", () => {
expect(result.current.userRole).toBe("Admin");
expect(result.current.premiumUser).toBe(true);
expect(result.current.disabledPersonalKeyCreation).toBe(false);
- expect(result.current.showSSOBanner).toBe(true);
expect(replaceMock).not.toHaveBeenCalled();
expect(clearTokenCookiesMock).not.toHaveBeenCalled();
});
@@ -163,9 +162,15 @@ describe("useAuthorized", () => {
expect(clearTokenCookiesMock).toHaveBeenCalled();
expect(result.current.isAuthorized).toBe(false);
expect(result.current.token).toBeNull();
- expect(result.current.accessToken).toBe("api-key-123");
- expect(result.current.userId).toBe("user-1");
- expect(result.current.userEmail).toBe("user@example.com");
+
+ // clearAuth must reset the context, not just the cookie, so no consumer
+ // can keep acting on the revoked session until a hard reload.
+ await waitFor(() => {
+ expect(result.current.accessToken).toBeNull();
+ });
+ expect(result.current.userId).toBeNull();
+ expect(result.current.userEmail).toBeNull();
+ expect(result.current.userRole).toBe("Undefined Role");
});
it("should redirect when token is missing", async () => {
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.ts
index 622f96c2ce0..c256c7466b3 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/useAuthorized.ts
@@ -1,7 +1,6 @@
"use client";
import { getProxyBaseUrl } from "@/components/networking";
-import { clearTokenCookies } from "@/utils/cookieUtils";
import { buildLoginUrlWithReturn, storeReturnUrl } from "@/utils/returnUrlUtils";
import { useRouter } from "next/navigation";
import { useCallback, useEffect } from "react";
@@ -30,7 +29,7 @@ const useAuthorized = () => {
accessToken,
premiumUser,
disabledPersonalKeyCreation,
- showSSOBanner,
+ clearAuth,
} = useAuth();
const isLoading = authLoading || isUIConfigLoading;
@@ -47,11 +46,11 @@ const useAuthorized = () => {
if (!isAuthorized) {
if (token) {
- clearTokenCookies();
+ clearAuth();
}
redirectToLogin();
}
- }, [isLoading, isAuthorized, token, redirectToLogin]);
+ }, [isLoading, isAuthorized, token, clearAuth, redirectToLogin]);
return {
isLoading,
@@ -63,7 +62,6 @@ const useAuthorized = () => {
userRole,
premiumUser: premiumUser as LegacyDecodedField,
disabledPersonalKeyCreation: disabledPersonalKeyCreation as LegacyDecodedField,
- showSSOBanner,
};
};
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useCurrentUser.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useCurrentUser.test.ts
index 0b37b605460..c7954886664 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useCurrentUser.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useCurrentUser.test.ts
@@ -70,7 +70,6 @@ describe("useCurrentUser", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
@@ -134,7 +133,6 @@ describe("useCurrentUser", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCurrentUser(), { wrapper });
@@ -158,7 +156,6 @@ describe("useCurrentUser", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCurrentUser(), { wrapper });
@@ -182,7 +179,6 @@ describe("useCurrentUser", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
const { result } = renderHook(() => useCurrentUser(), { wrapper });
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useUsers.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useUsers.test.ts
index 537e2c5378a..0d43b346c93 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useUsers.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/users/useUsers.test.ts
@@ -33,7 +33,6 @@ const DEFAULT_AUTH = {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
};
const buildUserListResponse = (page: number, totalPages: number, userCount = 2): UserListResponse => ({
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/models-and-endpoints/components/AllModelsTab.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/models-and-endpoints/components/AllModelsTab.test.tsx
index 045bf0a5f44..e8dce9e2037 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/models-and-endpoints/components/AllModelsTab.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/models-and-endpoints/components/AllModelsTab.test.tsx
@@ -119,7 +119,6 @@ describe("AllModelsTab", () => {
userRole: "Admin",
premiumUser: true,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
};
beforeEach(() => {
diff --git a/ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/TopKeyView.test.tsx b/ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/TopKeyView.test.tsx
index 126bf51bd36..b7c69720ee3 100644
--- a/ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/TopKeyView.test.tsx
+++ b/ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/TopKeyView.test.tsx
@@ -42,7 +42,6 @@ describe("TopKeyView", () => {
userRole: "admin",
premiumUser: true,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
};
const mockSetTopKeysLimit = vi.fn();
diff --git a/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx b/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx
index 7d5aec5f0b8..8f7ec393144 100644
--- a/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx
+++ b/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx
@@ -517,7 +517,6 @@ describe("UsagePage", () => {
userRole: "Admin",
premiumUser: true,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
mockUseCurrentUser.mockReturnValue({
data: {
@@ -651,7 +650,6 @@ describe("UsagePage", () => {
userRole: "internal_user",
premiumUser: true,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
renderWithProviders();
@@ -839,7 +837,6 @@ describe("UsagePage", () => {
userRole: "Internal User",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
renderWithProviders();
@@ -865,7 +862,6 @@ describe("UsagePage", () => {
userRole: "Internal User",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
renderWithProviders();
diff --git a/ui/litellm-dashboard/src/components/add_model/AddModelForm.test.tsx b/ui/litellm-dashboard/src/components/add_model/AddModelForm.test.tsx
index b4c22a834a4..b8f7c265823 100644
--- a/ui/litellm-dashboard/src/components/add_model/AddModelForm.test.tsx
+++ b/ui/litellm-dashboard/src/components/add_model/AddModelForm.test.tsx
@@ -113,7 +113,6 @@ const mockAuthorizedUser = (userRole: string, userId: string, premiumUser: boole
userRole,
premiumUser,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
const testTeam: Team = {
diff --git a/ui/litellm-dashboard/src/components/cache_settings/CacheFieldGroup.test.tsx b/ui/litellm-dashboard/src/components/cache_settings/CacheFieldGroup.test.tsx
index 6eba0718948..c0dfb9d1254 100644
--- a/ui/litellm-dashboard/src/components/cache_settings/CacheFieldGroup.test.tsx
+++ b/ui/litellm-dashboard/src/components/cache_settings/CacheFieldGroup.test.tsx
@@ -12,7 +12,6 @@ describe("CacheFieldGroup", () => {
userRole: "Admin",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
}),
}));
diff --git a/ui/litellm-dashboard/src/components/cache_settings/CacheFieldRenderer.test.tsx b/ui/litellm-dashboard/src/components/cache_settings/CacheFieldRenderer.test.tsx
index 3cfacac7fda..df0799d528e 100644
--- a/ui/litellm-dashboard/src/components/cache_settings/CacheFieldRenderer.test.tsx
+++ b/ui/litellm-dashboard/src/components/cache_settings/CacheFieldRenderer.test.tsx
@@ -12,7 +12,6 @@ vi.mock("@/app/(dashboard)/hooks/useAuthorized", () => ({
userRole: "Admin",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
}),
}));
diff --git a/ui/litellm-dashboard/src/components/leftnav.test.tsx b/ui/litellm-dashboard/src/components/leftnav.test.tsx
index 0e88a1a3603..4038ee0f297 100644
--- a/ui/litellm-dashboard/src/components/leftnav.test.tsx
+++ b/ui/litellm-dashboard/src/components/leftnav.test.tsx
@@ -23,7 +23,6 @@ const { mockUseAuthorized, mockUseOrganizations } = vi.hoisted(() => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
}));
const mockUseOrganizations = vi.fn(() => ({
@@ -148,7 +147,6 @@ describe("Sidebar (leftnav)", () => {
userEmail: "viewer@example.com",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
};
it("hides Playground from Admin Viewer (cost-incurring action)", () => {
@@ -192,7 +190,6 @@ describe("Sidebar (leftnav)", () => {
userEmail: "orgadmin@example.com",
premiumUser: false,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
mockUseOrganizations.mockReturnValueOnce({
diff --git a/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx b/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx
index ac0ae16a44f..a0acb3f443a 100644
--- a/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx
+++ b/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx
@@ -107,7 +107,6 @@ describe("TeamMembersComponent", () => {
userEmail: "test@example.com",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
});
});
diff --git a/ui/litellm-dashboard/src/components/templates/KeyInfoView.handleKeyUpdate.test.tsx b/ui/litellm-dashboard/src/components/templates/KeyInfoView.handleKeyUpdate.test.tsx
index abbc92f0210..2ab3f5fb96d 100644
--- a/ui/litellm-dashboard/src/components/templates/KeyInfoView.handleKeyUpdate.test.tsx
+++ b/ui/litellm-dashboard/src/components/templates/KeyInfoView.handleKeyUpdate.test.tsx
@@ -322,7 +322,6 @@ const renderView = (premiumUser: boolean) => {
token: "token_123",
userEmail: "test@example.com",
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
return render(
@@ -353,7 +352,6 @@ describe("KeyInfoView handleKeyUpdate guardrails guard", () => {
token: "token_123",
userEmail: "test@example.com",
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
render(
diff --git a/ui/litellm-dashboard/src/components/templates/key_info_view.budget_display.test.tsx b/ui/litellm-dashboard/src/components/templates/key_info_view.budget_display.test.tsx
index 77abde3d870..d7cdf705a10 100644
--- a/ui/litellm-dashboard/src/components/templates/key_info_view.budget_display.test.tsx
+++ b/ui/litellm-dashboard/src/components/templates/key_info_view.budget_display.test.tsx
@@ -102,7 +102,6 @@ const baseAuthorized = {
token: "test-token",
userEmail: null,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
};
describe("KeyInfoView overview budget display (LIT-2845)", () => {
diff --git a/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx b/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx
index c762e1ac14b..866bb3f661b 100644
--- a/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx
+++ b/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx
@@ -143,7 +143,6 @@ describe("KeyInfoView", () => {
token: "test-token",
userEmail: null,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
};
it("should render tags", async () => {
diff --git a/ui/litellm-dashboard/src/contexts/AuthContext.tsx b/ui/litellm-dashboard/src/contexts/AuthContext.tsx
index 591f6ad1a21..623f7224785 100644
--- a/ui/litellm-dashboard/src/contexts/AuthContext.tsx
+++ b/ui/litellm-dashboard/src/contexts/AuthContext.tsx
@@ -1,19 +1,12 @@
"use client";
-import React, { createContext, useContext, useEffect, useState } from "react";
+import React, { createContext, useCallback, useContext, useEffect, useMemo, useState } from "react";
import { jwtDecode } from "jwt-decode";
import { clearTokenCookies, getCookie } from "@/utils/cookieUtils";
import { isJwtExpired } from "@/utils/jwtUtils";
import { formatUserRole } from "@/utils/roles";
import { getUiConfig, setGlobalLitellmHeaderName } from "@/components/networking";
-function deleteCookie(name: string, path = "/") {
- document.cookie = `${name}=; Max-Age=0; Path=${path}`;
- if (name === "token") {
- clearTokenCookies();
- }
-}
-
type AuthContextValue = {
authLoading: boolean;
token: string | null;
@@ -23,8 +16,8 @@ type AuthContextValue = {
accessToken: string | null;
premiumUser: boolean;
disabledPersonalKeyCreation: boolean;
- showSSOBanner: boolean;
+ clearAuth: () => void;
setUserRole: React.Dispatch>;
setUserEmail: React.Dispatch>;
};
@@ -34,13 +27,22 @@ const AuthContext = createContext(null);
export function AuthProvider({ children }: { children: React.ReactNode }) {
const [authLoading, setAuthLoading] = useState(true);
const [token, setToken] = useState(null);
- const [userID, setUserID] = useState(null);
const [userRole, setUserRole] = useState(formatUserRole(""));
const [userEmail, setUserEmail] = useState(null);
- const [accessToken, setAccessToken] = useState(null);
- const [premiumUser, setPremiumUser] = useState(false);
- const [disabledPersonalKeyCreation, setDisabledPersonalKeyCreation] = useState(false);
- const [showSSOBanner, setShowSSOBanner] = useState(false);
+
+ const decoded = useMemo<{ [k: string]: any } | null>(() => {
+ if (!token || isJwtExpired(token)) return null;
+ try {
+ return jwtDecode(token);
+ } catch {
+ return null;
+ }
+ }, [token]);
+
+ const clearAuth = useCallback(() => {
+ clearTokenCookies();
+ setToken(null);
+ }, []);
// Load runtime UI config (populates proxyBaseUrl etc.) before clearing
// authLoading, so any consumer that builds proxy-rooted URLs from authLoading=false
@@ -63,7 +65,7 @@ export function AuthProvider({ children }: { children: React.ReactNode }) {
// Clear expired/invalid token so downstream code doesn't keep trying to use it.
if (raw && !valid) {
- deleteCookie("token", "/");
+ clearTokenCookies();
}
setToken(valid);
@@ -75,60 +77,42 @@ export function AuthProvider({ children }: { children: React.ReactNode }) {
};
}, []);
- // Decode JWT and populate derived auth state whenever the token changes.
+ // Side effects of a token change: drop tokens that fail to decode, sync the
+ // mutable user fields, and apply any custom auth header. The reset branch runs
+ // after any commit that nulls the token, so user fields cannot survive clearAuth
+ // even if a stale sync was queued in the same commit.
useEffect(() => {
- if (!token) {
+ if (token && !decoded) {
+ clearAuth();
return;
}
-
- if (isJwtExpired(token)) {
- deleteCookie("token", "/");
- setToken(null);
+ if (!decoded) {
+ setUserRole(formatUserRole(""));
+ setUserEmail(null);
return;
}
- let decoded: { [k: string]: any } | null = null;
- try {
- decoded = jwtDecode(token);
- } catch {
- deleteCookie("token", "/");
- setToken(null);
- return;
- }
-
- if (!decoded) return;
-
- setAccessToken(decoded.key);
- setDisabledPersonalKeyCreation(decoded.disabled_non_admin_personal_key_creation);
- setShowSSOBanner(decoded.login_method === "username_password");
-
if (decoded.user_role) {
setUserRole(formatUserRole(decoded.user_role));
}
if (decoded.user_email) {
setUserEmail(decoded.user_email);
}
- if (decoded.premium_user) {
- setPremiumUser(decoded.premium_user);
- }
if (decoded.auth_header_name) {
setGlobalLitellmHeaderName(decoded.auth_header_name);
}
- if (decoded.user_id) {
- setUserID(decoded.user_id);
- }
- }, [token]);
+ }, [token, decoded, clearAuth]);
const value: AuthContextValue = {
authLoading,
token,
- userID,
+ userID: decoded?.user_id ?? null,
userRole,
userEmail,
- accessToken,
- premiumUser,
- disabledPersonalKeyCreation,
- showSSOBanner,
+ accessToken: decoded?.key ?? null,
+ premiumUser: !!decoded?.premium_user,
+ disabledPersonalKeyCreation: !!decoded?.disabled_non_admin_personal_key_creation,
+ clearAuth,
setUserRole,
setUserEmail,
};
diff --git a/ui/litellm-dashboard/tests/setupTests.ts b/ui/litellm-dashboard/tests/setupTests.ts
index c38ab71a4eb..c3bdb6893f8 100644
--- a/ui/litellm-dashboard/tests/setupTests.ts
+++ b/ui/litellm-dashboard/tests/setupTests.ts
@@ -147,7 +147,6 @@ vi.mock("@/app/(dashboard)/hooks/useAuthorized", () => ({
userRole: "Admin",
premiumUser: false,
disabledPersonalKeyCreation: null,
- showSSOBanner: false,
}),
}));
diff --git a/ui/litellm-dashboard/tests/top_key_view.test.tsx b/ui/litellm-dashboard/tests/top_key_view.test.tsx
index 51662b8f453..8f33658c8da 100644
--- a/ui/litellm-dashboard/tests/top_key_view.test.tsx
+++ b/ui/litellm-dashboard/tests/top_key_view.test.tsx
@@ -72,7 +72,6 @@ describe("TopKeyView", () => {
userRole: mockProps.userRole,
premiumUser: mockProps.premiumUser,
disabledPersonalKeyCreation: false,
- showSSOBanner: false,
});
});