scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

ui add a check for isAdminOrAdminViewer

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-05 13:04:20 -07:00
parent 491e50f381
commit 8344bbd3af
1 changed files with 27 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
ui/litellm-dashboard/src/components/usage.tsx
View file

@ -132,6 +132,13 @@ type DataDict = { [key: string]: unknown };
type UserData = { user_id: string; spend: number };
const isAdminOrAdminViewer = (role: string | null): boolean => {
if (role === null) return false;
return role === 'Admin' || role === 'Admin Viewer';
};
const UsagePage: React.FC<UsagePageProps> = ({
accessToken,
token,
@ -379,16 +386,21 @@ const UsagePage: React.FC<UsagePageProps> = ({
useEffect(() => {
if (accessToken && token && userRole && userID) {
fetchOverallSpend();
fetchProviderSpend();
fetchTopKeys();
fetchTopModels();
fetchTeamSpend();
fetchTagNames();
fetchTopTags();
fetchTopEndUsers();
fetchGlobalActivity();
fetchGlobalActivityPerModel();
if (isAdminOrAdminViewer(userRole)) {
fetchTeamSpend();
fetchTagNames();
fetchTopTags();
fetchTopEndUsers();
}
}
}, [accessToken, token, userRole, userID, startTime, endTime]);
@ -399,9 +411,17 @@ const UsagePage: React.FC<UsagePageProps> = ({
<TabGroup>
<TabList className="mt-2">
<Tab>All Up</Tab>
<Tab>Team Based Usage</Tab>
<Tab>Customer Usage</Tab>
<Tab>Tag Based Usage</Tab>
{isAdminOrAdminViewer(userRole) ? (
<>
<Tab>Team Based Usage</Tab>
<Tab>Customer Usage</Tab>
<Tab>Tag Based Usage</Tab>
</>
) : (
<><div></div>
</>
)}
</TabList>
<TabPanels>
<TabPanel>