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

fix(proxy_server.py): check if prisma client is set before scheduling reset budget

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-07 18:14:26 -08:00
parent 68926c6524
commit d2dceb3537
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/proxy_server.py
View file

@ -2004,12 +2004,15 @@ async def startup_event():
)
### START BUDGET SCHEDULER ###
scheduler = AsyncIOScheduler()
interval = random.randint(
597, 605
) # random interval, so multiple workers avoid resetting budget at the same time
scheduler.add_job(reset_budget, "interval", seconds=interval, args=[prisma_client])
scheduler.start()
if prisma_client is not None:
scheduler = AsyncIOScheduler()
interval = random.randint(
597, 605
) # random interval, so multiple workers avoid resetting budget at the same time
scheduler.add_job(
reset_budget, "interval", seconds=interval, args=[prisma_client]
)
scheduler.start()
#### API ENDPOINTS ####