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

Revert "Merge pull request #2593 from BerriAI/litellm_reset_budget_fix"

Browse source 
This reverts commit afd363129f, reversing
changes made to c94bc94ad5.
This commit is contained in:
Krrish Dholakia 2024-03-19 20:25:41 -07:00
parent 3a866dd349
commit 2dfdc8dd69
4 changed files with 42 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/proxy_server.py
View file

@ -2661,12 +2661,6 @@ async def startup_event():
### START BUDGET SCHEDULER ###
if prisma_client is not None:
scheduler = AsyncIOScheduler()
verbose_proxy_logger.debug(
f"proxy_budget_rescheduler_max_time: {proxy_budget_rescheduler_max_time}"
)
verbose_proxy_logger.debug(
f"proxy_budget_rescheduler_min_time: {proxy_budget_rescheduler_min_time}"
)
interval = random.randint(
proxy_budget_rescheduler_min_time, proxy_budget_rescheduler_max_time
) # random interval, so multiple workers avoid resetting budget at the same time
@ -4999,7 +4993,6 @@ async def user_info(
```
"""
global prisma_client
verbose_proxy_logger.debug(f"Received `/user/info` call for user_id={user_id}")
try:
if prisma_client is None:
raise Exception(
@ -5007,9 +5000,7 @@ async def user_info(
)
## GET USER ROW ##
if user_id is not None:
verbose_proxy_logger.debug(f"Making get_data call for user_id={user_id}")
user_info = await prisma_client.get_data(user_id=user_id)
verbose_proxy_logger.debug(f"Received get_data for user_id={user_id}")
elif view_all == True:
if page is None:
page = 0
@ -5090,7 +5081,6 @@ async def user_info(
key = key.dict()
key.pop("token", None)
verbose_proxy_logger.debug(f"RETURNING RESPONSE FOR /USER/INFO call ")
response_data = {
"user_id": user_id,
"user_info": user_info,