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

ui - filter tags by starttime and endtime

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-13 17:15:25 -07:00
parent 5845d6f958
commit 51961387a2
1 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
enterprise/utils.py
View file

@ -1,6 +1,7 @@
# Enterprise Proxy Util Endpoints # Enterprise Proxy Util Endpoints
from litellm._logging import verbose_logger from litellm._logging import verbose_logger
import collections import collections
from datetime import datetime
async def get_spend_by_tags(start_date=None, end_date=None, prisma_client=None): async def get_spend_by_tags(start_date=None, end_date=None, prisma_client=None):
@ -18,19 +19,25 @@ async def get_spend_by_tags(start_date=None, end_date=None, prisma_client=None):
return response return response
async def ui_get_spend_by_tags(start_date=None, end_date=None, prisma_client=None): async def ui_get_spend_by_tags(start_date: str, end_date: str, prisma_client):
response = await prisma_client.db.query_raw(
""" sql_query = """
SELECT SELECT
jsonb_array_elements_text(request_tags) AS individual_request_tag, jsonb_array_elements_text(request_tags) AS individual_request_tag,
DATE(s."startTime") AS spend_date, DATE(s."startTime") AS spend_date,
COUNT(*) AS log_count, COUNT(*) AS log_count,
SUM(spend) AS total_spend SUM(spend) AS total_spend
FROM "LiteLLM_SpendLogs" s FROM "LiteLLM_SpendLogs" s
WHERE s."startTime" >= current_date - interval '30 days' WHERE
DATE(s."startTime") >= $1::date
AND DATE(s."startTime") <= $2::date
GROUP BY individual_request_tag, spend_date GROUP BY individual_request_tag, spend_date
ORDER BY spend_date; ORDER BY spend_date;
""" """
response = await prisma_client.db.query_raw(
sql_query,
start_date,
end_date,
) )
# print("tags - spend") # print("tags - spend")