phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(user_dashboard.tsx): check key health on login - if invalid -> redirect to login

Browse source 
handles invalid / expired key scenario
This commit is contained in:
Krrish Dholakia 2025-04-23 16:15:37 -07:00
parent be3276200b
commit a9af63ead5
2 changed files with 40 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ui/litellm-dashboard/src/components/networking.tsx
View file

@ -2493,6 +2493,9 @@ export const keyInfoCall = async (accessToken: String, keys: String[]) => {
if (!response.ok) { if (!response.ok) {
const errorData = await response.text(); const errorData = await response.text();
if (errorData.includes("Invalid proxy server token passed")) {
throw new Error("Invalid proxy server token passed");
}
handleError(errorData); handleError(errorData);
throw new Error("Network response was not ok"); throw new Error("Network response was not ok");
} }

46
ui/litellm-dashboard/src/components/user_dashboard.tsx
View file

@ -7,7 +7,8 @@ import {
getProxyUISettings, getProxyUISettings,
Organization, Organization,
organizationListCall, organizationListCall,
DEFAULT_ORGANIZATION DEFAULT_ORGANIZATION,
keyInfoCall
} from "./networking"; } from "./networking";
import { fetchTeams } from "./common_components/fetch_teams"; import { fetchTeams } from "./common_components/fetch_teams";
import { Grid, Col, Card, Text, Title } from "@tremor/react"; import { Grid, Col, Card, Text, Title } from "@tremor/react";
@ -192,6 +193,7 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
null, null,
null null
); );
setUserSpendData(response["user_info"]); setUserSpendData(response["user_info"]);
console.log(`userSpendData: ${JSON.stringify(userSpendData)}`) console.log(`userSpendData: ${JSON.stringify(userSpendData)}`)
@ -238,8 +240,11 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
"userModels" + userID, "userModels" + userID,
JSON.stringify(available_model_names) JSON.stringify(available_model_names)
); );
} catch (error) { } catch (error: any) {
console.error("There was an error fetching the data", error); console.error("There was an error fetching the data", error);
if (error.message.includes("Invalid proxy server token passed")) {
gotoLogin();
}
// Optionally, update your UI to reflect the error state here as well // Optionally, update your UI to reflect the error state here as well
} }
}; };
@ -249,6 +254,24 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
} }
}, [userID, token, accessToken, keys, userRole]); }, [userID, token, accessToken, keys, userRole]);
useEffect(() => {
// check key health - if it's invalid, redirect to login
if (accessToken) {
const fetchKeyInfo = async () => {
try {
const keyInfo = await keyInfoCall(accessToken, [accessToken]);
console.log("keyInfo: ", keyInfo);
} catch (error: any) {
if (error.message.includes("Invalid proxy server token passed")) {
gotoLogin();
}
}
}
fetchKeyInfo();
}
}, [accessToken]);
useEffect(() => { useEffect(() => {
console.log(`currentOrg: ${JSON.stringify(currentOrg)}, accessToken: ${accessToken}, userID: ${userID}, userRole: ${userRole}`) console.log(`currentOrg: ${JSON.stringify(currentOrg)}, accessToken: ${accessToken}, userID: ${userID}, userRole: ${userRole}`)
if (accessToken) { if (accessToken) {
@ -295,26 +318,31 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
) )
} }
function gotoLogin() {
if (token == null) {
// user is not logged in as yet
console.log("All cookies before redirect:", document.cookie);
// Clear token cookies using the utility function // Clear token cookies using the utility function
clearTokenCookies(); clearTokenCookies();
const url = proxyBaseUrl const url = proxyBaseUrl
? `${proxyBaseUrl}/sso/key/generate` ? `${proxyBaseUrl}/sso/key/generate`
: `/sso/key/generate`; : `/sso/key/generate`;
console.log("Full URL:", url); console.log("Full URL:", url);
window.location.href = url; window.location.href = url;
return null; return null;
}
if (token == null) {
// user is not logged in as yet
console.log("All cookies before redirect:", document.cookie);
// Clear token cookies using the utility function
gotoLogin();
} else { } else {
// Check if token is expired // Check if token is expired
try { try {
const decoded = jwtDecode(token) as { [key: string]: any }; const decoded = jwtDecode(token) as { [key: string]: any };
console.log("Decoded token:", decoded);
const expTime = decoded.exp; const expTime = decoded.exp;
const currentTime = Math.floor(Date.now() / 1000); const currentTime = Math.floor(Date.now() / 1000);