phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix /global/activity/model

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-29 08:58:25 -07:00
parent 34819549c4
commit 5958288ebb
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/proxy_server.py
View file

@ -6595,14 +6595,13 @@ async def get_global_activity_model(
sql_query = """ sql_query = """
SELECT SELECT
model_group AS model, model_group,
date_trunc('day', "startTime") AS date, date_trunc('day', "startTime") AS date,
COUNT(*) AS api_requests, COUNT(*) AS api_requests,
SUM(total_tokens) AS total_tokens SUM(total_tokens) AS total_tokens
FROM "LiteLLM_SpendLogs" FROM "LiteLLM_SpendLogs"
WHERE "startTime" BETWEEN $1::date AND $2::date + interval '1 day' WHERE "startTime" BETWEEN $1::date AND $2::date + interval '1 day'
GROUP BY model, date_trunc('day', "startTime") GROUP BY model_group, date_trunc('day', "startTime")
""" """
db_response = await prisma_client.db.query_raw( db_response = await prisma_client.db.query_raw(
sql_query, start_date_obj, end_date_obj sql_query, start_date_obj, end_date_obj
@ -6615,7 +6614,7 @@ async def get_global_activity_model(
) # {"gpt-4": {"daily_data": [], "sum_api_requests": 0, "sum_total_tokens": 0}} ) # {"gpt-4": {"daily_data": [], "sum_api_requests": 0, "sum_total_tokens": 0}}
for row in db_response: for row in db_response:
_model = row["model"] _model = row["model_group"]
if _model not in model_ui_data: if _model not in model_ui_data:
model_ui_data[_model] = { model_ui_data[_model] = {
"daily_data": [], "daily_data": [],
@ -6655,7 +6654,7 @@ async def get_global_activity_model(
except Exception as e: except Exception as e:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail={"error": str(e)}, detail={"error": str(e)},
) )