From e8dcf8fa130c6af87922b7c8b9911e5e0868d66e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 16 Feb 2024 17:00:23 -0800 Subject: [PATCH] (fix) setting model_max_budget --- litellm/proxy/proxy_server.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index dda0d59c96..7d30c959f1 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1763,6 +1763,11 @@ async def generate_key_helper_fn( saved_token["metadata"] = json.loads(saved_token["metadata"]) if isinstance(saved_token["permissions"], str): saved_token["permissions"] = json.loads(saved_token["permissions"]) + if isinstance(saved_token["model_max_budget"], str): + saved_token["model_max_budget"] = json.loads( + saved_token["model_max_budget"] + ) + if saved_token.get("expires", None) is not None and isinstance( saved_token["expires"], datetime ):