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

fix(new_usage.tsx): increase page size + iterate through all pages if multiple pages

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-09 15:53:44 -07:00
parent 9ec1972926
commit b11c08bde3
2 changed files with 37 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1075,7 +1075,7 @@ export const organizationDeleteCall = async (
}; };
export const userDailyActivityCall = async (accessToken: String, startTime: Date, endTime: Date) => { export const userDailyActivityCall = async (accessToken: String, startTime: Date, endTime: Date, page: number = 1) => {
/** /**
* Get daily user activity on proxy * Get daily user activity on proxy
*/ */
@ -1084,6 +1084,8 @@ export const userDailyActivityCall = async (accessToken: String, startTime: Date
const queryParams = new URLSearchParams(); const queryParams = new URLSearchParams();
queryParams.append('start_date', startTime.toISOString()); queryParams.append('start_date', startTime.toISOString());
queryParams.append('end_date', endTime.toISOString()); queryParams.append('end_date', endTime.toISOString());
queryParams.append('page_size', '1000');
queryParams.append('page', page.toString());
const queryString = queryParams.toString(); const queryString = queryParams.toString();
if (queryString) { if (queryString) {
url += `?${queryString}`; url += `?${queryString}`;

39
ui/litellm-dashboard/src/components/new_usage.tsx
View file

@ -22,15 +22,13 @@ import ViewUserSpend from "./view_user_spend";
import TopKeyView from "./top_key_view"; import TopKeyView from "./top_key_view";
import { ActivityMetrics, processActivityData } from './activity_metrics'; import { ActivityMetrics, processActivityData } from './activity_metrics';
import { SpendMetrics, DailyData, ModelActivityData, MetricWithMetadata, KeyMetricWithMetadata } from './usage/types'; import { SpendMetrics, DailyData, ModelActivityData, MetricWithMetadata, KeyMetricWithMetadata } from './usage/types';
interface NewUsagePageProps { interface NewUsagePageProps {
accessToken: string | null; accessToken: string | null;
userRole: string | null; userRole: string | null;
userID: string | null; userID: string | null;
} }
const NewUsagePage: React.FC<NewUsagePageProps> = ({ const NewUsagePage: React.FC<NewUsagePageProps> = ({
accessToken, accessToken,
userRole, userRole,
@ -177,8 +175,39 @@ const NewUsagePage: React.FC<NewUsagePageProps> = ({
if (!accessToken || !dateValue.from || !dateValue.to) return; if (!accessToken || !dateValue.from || !dateValue.to) return;
const startTime = dateValue.from; const startTime = dateValue.from;
const endTime = dateValue.to; const endTime = dateValue.to;
const data = await userDailyActivityCall(accessToken, startTime, endTime);
setUserSpendData(data); try {
// Get first page
const firstPageData = await userDailyActivityCall(accessToken, startTime, endTime);
// Check if we need to fetch more pages
if (firstPageData.metadata.total_pages > 10) {
throw new Error("Too many pages of data (>10). Please select a smaller date range.");
}
// If only one page, just set the data
if (firstPageData.metadata.total_pages === 1) {
setUserSpendData(firstPageData);
return;
}
// Fetch all pages
const allResults = [...firstPageData.results];
for (let page = 2; page <= firstPageData.metadata.total_pages; page++) {
const pageData = await userDailyActivityCall(accessToken, startTime, endTime, page);
allResults.push(...pageData.results);
}
// Combine all results with the first page's metadata
setUserSpendData({
results: allResults,
metadata: firstPageData.metadata
});
} catch (error) {
console.error("Error fetching user spend data:", error);
throw error;
}
}; };
useEffect(() => { useEffect(() => {