forked from phoenix/litellm-mirror
(v0)
This commit is contained in:
parent
d9fd8f28d6
commit
d3848b6e6c
1 changed files with 35 additions and 2 deletions
|
@ -2393,7 +2393,7 @@ async def info_key_fn(
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/spend/keys",
|
"/spend/keys",
|
||||||
tags=["Budget & Spend Tracking"],
|
tags=["budget & spend Tracking"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def spend_key_fn():
|
async def spend_key_fn():
|
||||||
|
@ -2424,9 +2424,42 @@ async def spend_key_fn():
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get(
|
||||||
|
"/spend/users",
|
||||||
|
tags=["budget & spend Tracking"],
|
||||||
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
|
)
|
||||||
|
async def spend_user_fn():
|
||||||
|
"""
|
||||||
|
View all users created, ordered by spend
|
||||||
|
|
||||||
|
Example Request:
|
||||||
|
```
|
||||||
|
curl -X GET "http://0.0.0.0:8000/spend/keys" \
|
||||||
|
-H "Authorization: Bearer sk-1234"
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
global prisma_client
|
||||||
|
try:
|
||||||
|
if prisma_client is None:
|
||||||
|
raise Exception(
|
||||||
|
f"Database not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keys"
|
||||||
|
)
|
||||||
|
|
||||||
|
key_info = await prisma_client.get_data(table_name="key", query_type="find_all")
|
||||||
|
|
||||||
|
return key_info
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=status.HTTP_400_BAD_REQUEST,
|
||||||
|
detail={"error": str(e)},
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/spend/logs",
|
"/spend/logs",
|
||||||
tags=["Budget & Spend Tracking"],
|
tags=["budget & spend Tracking"],
|
||||||
dependencies=[Depends(user_api_key_auth)],
|
dependencies=[Depends(user_api_key_auth)],
|
||||||
)
|
)
|
||||||
async def view_spend_logs(
|
async def view_spend_logs(
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue