From 6120e1c36df6c241a8c5eba97ef8395b543057cd Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 22 Jan 2024 12:50:14 -0800
Subject: [PATCH] (fix) dynamo max_budget per key

---
 litellm/proxy/_types.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 1615662012a..bb56ad6bf1b 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -297,6 +297,7 @@ class ConfigYAML(LiteLLMBase):
 class LiteLLM_VerificationToken(LiteLLMBase):
     token: str
     spend: float = 0.0
+    max_budget: Optional[float] = None
     expires: Union[str, None]
     models: List[str]
     aliases: Dict[str, str] = {}