From daba3d14cbfa506e71b236bd6ec80f3396e2788b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 21 Feb 2024 13:25:52 -0800 Subject: [PATCH] (fix) ui allow users to make ui chat requests --- litellm/proxy/utils.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 3cad1777c9..34a77a973f 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1465,6 +1465,20 @@ def _is_user_proxy_admin(user_id_information=None): and _user.get("user_role") == "proxy_admin" ): return True + + # if user_id_information contains litellm-proxy-budget + # get first user_id that is not litellm-proxy-budget + for user in user_id_information: + if user.get("user_id") != "litellm-proxy-budget": + _user = user + break + + if ( + _user.get("user_role", None) is not None + and _user.get("user_role") == "proxy_admin" + ): + return True + return False