From 7e94fcc9a8ca68672d7f42e3f6379e8a4eba34b1 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 2 Mar 2024 17:15:27 -0800 Subject: [PATCH] (fix) prisma test --- litellm/proxy/proxy_server.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 75f3a944dd..6a818666e1 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1814,6 +1814,9 @@ async def generate_key_helper_fn( key_soft_budget: Optional[ float ] = None, # key_soft_budget is used to Budget Per key + soft_budget: Optional[ + float + ] = None, # soft_budget is used to set soft Budgets Per user max_budget: Optional[float] = None, # max_budget is used to Budget Per user budget_duration: Optional[str] = None, # max_budget is used to Budget Per user token: Optional[str] = None,