(feat) /spend/users endpoint

This commit is contained in:
ishaan-jaff 2024-01-24 11:34:28 -08:00
parent d3848b6e6c
commit 2692afca75
2 changed files with 10 additions and 3 deletions

View file

@ -2435,7 +2435,7 @@ async def spend_user_fn():
Example Request:
```
curl -X GET "http://0.0.0.0:8000/spend/keys" \
curl -X GET "http://0.0.0.0:8000/spend/users" \
-H "Authorization: Bearer sk-1234"
```
"""
@ -2446,9 +2446,11 @@ async def spend_user_fn():
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")
user_info = await prisma_client.get_data(
table_name="user", query_type="find_all"
)
return key_info
return user_info
except Exception as e:
raise HTTPException(

View file

@ -432,6 +432,11 @@ class PrismaClient:
}
)
return response
elif table_name == "user" and query_type == "find_all":
response = await self.db.litellm_usertable.find_many( # type: ignore
order={"spend": "desc"},
)
return response
elif table_name == "spend":
verbose_proxy_logger.debug(
f"PrismaClient: get_data: table_name == 'spend'"