From ea8f6672c53feec27eaf7c655d5f4b36ee204466 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 26 Mar 2024 15:22:05 -0700 Subject: [PATCH] test(test_jwt.py): fix test --- litellm/tests/test_jwt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_jwt.py b/litellm/tests/test_jwt.py index ee1c67fdd27..fe5a70b9ca9 100644 --- a/litellm/tests/test_jwt.py +++ b/litellm/tests/test_jwt.py @@ -28,7 +28,7 @@ public_key = { def test_load_config_with_custom_role_names(): config = { "general_settings": { - "litellm_proxy_roles": {"proxy_admin": "litellm-proxy-admin"} + "litellm_proxy_roles": {"admin_jwt_scope": "litellm-proxy-admin"} } } @@ -38,7 +38,7 @@ def test_load_config_with_custom_role_names(): print(f"proxy_roles: {proxy_roles}") - assert proxy_roles.proxy_admin == "litellm-proxy-admin" + assert proxy_roles.admin_jwt_scope == "litellm-proxy-admin" # test_load_config_with_custom_role_names()