mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 11:43:54 +00:00
feat - ui view spend per provider
This commit is contained in:
parent
829a190083
commit
443d88b744
2 changed files with 88 additions and 1 deletions
|
@ -994,6 +994,46 @@ export const adminTopEndUsersCall = async (
|
||||||
throw error;
|
throw error;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
export const adminspendByProvider = async (accessToken: String, keyToken: String | null, startTime: String | undefined, endTime: String | undefined) => {
|
||||||
|
try {
|
||||||
|
let url = proxyBaseUrl ? `${proxyBaseUrl}/global/spend/provider` : `/global/spend/provider`;
|
||||||
|
|
||||||
|
if (startTime && endTime) {
|
||||||
|
url += `?start_date=${startTime}&end_date=${endTime}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (keyToken) {
|
||||||
|
url += `&api_key=${keyToken}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
const requestOptions: {
|
||||||
|
method: string;
|
||||||
|
headers: {
|
||||||
|
Authorization: string;
|
||||||
|
};
|
||||||
|
} = {
|
||||||
|
method: "GET",
|
||||||
|
headers: {
|
||||||
|
Authorization: `Bearer ${accessToken}`,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
const response = await fetch(url, requestOptions);
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
const errorData = await response.text();
|
||||||
|
message.error(errorData, 10);
|
||||||
|
throw new Error("Network response was not ok");
|
||||||
|
}
|
||||||
|
|
||||||
|
const data = await response.json();
|
||||||
|
console.log(data);
|
||||||
|
return data;
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Failed to fetch spend data:", error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
export const adminTopModelsCall = async (accessToken: String) => {
|
export const adminTopModelsCall = async (accessToken: String) => {
|
||||||
try {
|
try {
|
||||||
|
|
|
@ -3,7 +3,7 @@ import { BarChart, BarList, Card, Title, Table, TableHead, TableHeaderCell, Tabl
|
||||||
import React, { useState, useEffect } from "react";
|
import React, { useState, useEffect } from "react";
|
||||||
|
|
||||||
import ViewUserSpend from "./view_user_spend";
|
import ViewUserSpend from "./view_user_spend";
|
||||||
import { Grid, Col, Text, LineChart, TabPanel, TabPanels, TabGroup, TabList, Tab, Select, SelectItem, DateRangePicker, DateRangePickerValue } from "@tremor/react";
|
import { Grid, Col, Text, LineChart, TabPanel, TabPanels, TabGroup, TabList, Tab, Select, SelectItem, DateRangePicker, DateRangePickerValue, DonutChart} from "@tremor/react";
|
||||||
import {
|
import {
|
||||||
userSpendLogsCall,
|
userSpendLogsCall,
|
||||||
keyInfoCall,
|
keyInfoCall,
|
||||||
|
@ -16,6 +16,7 @@ import {
|
||||||
modelMetricsCall,
|
modelMetricsCall,
|
||||||
modelAvailableCall,
|
modelAvailableCall,
|
||||||
modelInfoCall,
|
modelInfoCall,
|
||||||
|
adminspendByProvider,
|
||||||
} from "./networking";
|
} from "./networking";
|
||||||
import { start } from "repl";
|
import { start } from "repl";
|
||||||
|
|
||||||
|
@ -115,6 +116,7 @@ const UsagePage: React.FC<UsagePageProps> = ({
|
||||||
const [topTagsData, setTopTagsData] = useState<any[]>([]);
|
const [topTagsData, setTopTagsData] = useState<any[]>([]);
|
||||||
const [uniqueTeamIds, setUniqueTeamIds] = useState<any[]>([]);
|
const [uniqueTeamIds, setUniqueTeamIds] = useState<any[]>([]);
|
||||||
const [totalSpendPerTeam, setTotalSpendPerTeam] = useState<any[]>([]);
|
const [totalSpendPerTeam, setTotalSpendPerTeam] = useState<any[]>([]);
|
||||||
|
const [spendByProvider, setSpendByProvider] = useState<any[]>([]);
|
||||||
const [selectedKeyID, setSelectedKeyID] = useState<string | null>("");
|
const [selectedKeyID, setSelectedKeyID] = useState<string | null>("");
|
||||||
const [dateValue, setDateValue] = useState<DateRangePickerValue>({
|
const [dateValue, setDateValue] = useState<DateRangePickerValue>({
|
||||||
from: new Date(Date.now() - 7 * 24 * 60 * 60 * 1000),
|
from: new Date(Date.now() - 7 * 24 * 60 * 60 * 1000),
|
||||||
|
@ -210,6 +212,12 @@ const UsagePage: React.FC<UsagePageProps> = ({
|
||||||
if (userRole == "Admin" || userRole == "Admin Viewer") {
|
if (userRole == "Admin" || userRole == "Admin Viewer") {
|
||||||
const overall_spend = await adminSpendLogsCall(accessToken);
|
const overall_spend = await adminSpendLogsCall(accessToken);
|
||||||
setKeySpendData(overall_spend);
|
setKeySpendData(overall_spend);
|
||||||
|
|
||||||
|
const provider_spend = await adminspendByProvider(accessToken, token, startTime, endTime);
|
||||||
|
console.log("provider_spend", provider_spend);
|
||||||
|
setSpendByProvider(provider_spend);
|
||||||
|
|
||||||
|
|
||||||
const top_keys = await adminTopKeysCall(accessToken);
|
const top_keys = await adminTopKeysCall(accessToken);
|
||||||
const filtered_keys = top_keys.map((k: any) => ({
|
const filtered_keys = top_keys.map((k: any) => ({
|
||||||
key: (k["key_alias"] || k["key_name"] || k["api_key"]).substring(
|
key: (k["key_alias"] || k["key_name"] || k["api_key"]).substring(
|
||||||
|
@ -372,6 +380,44 @@ const UsagePage: React.FC<UsagePageProps> = ({
|
||||||
</Col>
|
</Col>
|
||||||
<Col numColSpan={1}>
|
<Col numColSpan={1}>
|
||||||
|
|
||||||
|
</Col>
|
||||||
|
<Col numColSpan={2}>
|
||||||
|
<Card className="mb-2">
|
||||||
|
<Title>✨ Spend by Provider</Title>
|
||||||
|
<Grid numItems={2}>
|
||||||
|
<Col numColSpan={1}>
|
||||||
|
<DonutChart
|
||||||
|
className="mt-4 h-40"
|
||||||
|
variant="pie"
|
||||||
|
data={spendByProvider}
|
||||||
|
index="provider"
|
||||||
|
category="spend"
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
<Col numColSpan={1}>
|
||||||
|
<Table>
|
||||||
|
<TableHead>
|
||||||
|
<TableRow>
|
||||||
|
<TableHeaderCell>Provider</TableHeaderCell>
|
||||||
|
<TableHeaderCell>Spend</TableHeaderCell>
|
||||||
|
</TableRow>
|
||||||
|
</TableHead>
|
||||||
|
<TableBody>
|
||||||
|
{spendByProvider.map((provider) => (
|
||||||
|
<TableRow key={provider.provider}>
|
||||||
|
<TableCell>{provider.provider}</TableCell>
|
||||||
|
<TableCell>
|
||||||
|
{parseFloat(provider.spend.toFixed(2)) < 0.00001
|
||||||
|
? "less than 0.00"
|
||||||
|
: provider.spend.toFixed(2)}
|
||||||
|
</TableCell>
|
||||||
|
</TableRow>
|
||||||
|
))}
|
||||||
|
</TableBody>
|
||||||
|
</Table>
|
||||||
|
</Col>
|
||||||
|
</Grid>
|
||||||
|
</Card>
|
||||||
</Col>
|
</Col>
|
||||||
</Grid>
|
</Grid>
|
||||||
</TabPanel>
|
</TabPanel>
|
||||||
|
@ -382,6 +428,7 @@ const UsagePage: React.FC<UsagePageProps> = ({
|
||||||
<Title>Total Spend Per Team</Title>
|
<Title>Total Spend Per Team</Title>
|
||||||
<BarList
|
<BarList
|
||||||
data={totalSpendPerTeam}
|
data={totalSpendPerTeam}
|
||||||
|
|
||||||
/>
|
/>
|
||||||
</Card>
|
</Card>
|
||||||
<Card>
|
<Card>
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue