diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index df8b63b64..9b21aa880 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -1133,6 +1133,7 @@ async def generate_key_helper_fn(
     key_max_budget: Optional[float] = None,  # key_max_budget is used to Budget Per key
     key_budget_duration: Optional[str] = None,
     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,
     user_id: Optional[str] = None,
     team_id: Optional[str] = None,
diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index c06bed7fa..0f132d79b 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -399,7 +399,7 @@ class PrismaClient:
                     and reset_at is not None
                 ):
                     response = await self.db.litellm_verificationtoken.find_many(
-                        where={
+                        where={  # type:ignore
                             "OR": [
                                 {"expires": None},
                                 {"expires": {"gt": expires}},