fix literal usage

litellm/tests/test_key_generate_prisma.py

@@ -138,7 +138,7 @@ async def test_new_user_response(prisma_client):
                 team_id=_team_id,
             ),
             user_api_key_dict=UserAPIKeyAuth(
-                user_role=LitellmUserRoles.PROXY_ADMIN.value,
+                user_role=LitellmUserRoles.PROXY_ADMIN,
                 api_key="sk-1234",
                 user_id="1234",
             ),
@@ -367,9 +367,7 @@ async def test_call_with_valid_model_using_all_models(prisma_client):
 
         new_team_response = await new_team(
             data=team_request,
-            user_api_key_dict=UserAPIKeyAuth(
-                user_role=LitellmUserRoles.PROXY_ADMIN.value
-            ),
+            user_api_key_dict=UserAPIKeyAuth(user_role=LitellmUserRoles.PROXY_ADMIN),
         )
         print("new_team_response", new_team_response)
         created_team_id = new_team_response["team_id"]
@@ -928,7 +926,7 @@ def test_delete_key(prisma_client):
             # use generated key to auth in
             result = await user_api_key_auth(request=request, api_key=bearer_token)
             print(f"result: {result}")
-            result.user_role = LitellmUserRoles.PROXY_ADMIN.value
+            result.user_role = LitellmUserRoles.PROXY_ADMIN
             # delete the key
             result_delete_key = await delete_key_fn(
                 data=delete_key_request, user_api_key_dict=result
@@ -978,7 +976,7 @@ def test_delete_key_auth(prisma_client):
             # use generated key to auth in
             result = await user_api_key_auth(request=request, api_key=bearer_token)
             print(f"result: {result}")
-            result.user_role = LitellmUserRoles.PROXY_ADMIN.value
+            result.user_role = LitellmUserRoles.PROXY_ADMIN
 
             result_delete_key = await delete_key_fn(
                 data=delete_key_request, user_api_key_dict=result
@@ -1050,7 +1048,7 @@ def test_generate_and_call_key_info(prisma_client):
             # use generated key to auth in
             result = await user_api_key_auth(request=request, api_key=bearer_token)
             print(f"result: {result}")
-            result.user_role = LitellmUserRoles.PROXY_ADMIN.value
+            result.user_role = LitellmUserRoles.PROXY_ADMIN
 
             result_delete_key = await delete_key_fn(
                 data=delete_key_request, user_api_key_dict=result
@@ -1084,7 +1082,7 @@ def test_generate_and_update_key(prisma_client):
                     team_id=_team_1,
                 ),
                 user_api_key_dict=UserAPIKeyAuth(
-                    user_role=LitellmUserRoles.PROXY_ADMIN.value,
+                    user_role=LitellmUserRoles.PROXY_ADMIN,
                     api_key="sk-1234",
                     user_id="1234",
                 ),
@@ -1096,7 +1094,7 @@ def test_generate_and_update_key(prisma_client):
                     team_id=_team_2,
                 ),
                 user_api_key_dict=UserAPIKeyAuth(
-                    user_role=LitellmUserRoles.PROXY_ADMIN.value,
+                    user_role=LitellmUserRoles.PROXY_ADMIN,
                     api_key="sk-1234",
                     user_id="1234",
                 ),
@@ -1168,7 +1166,7 @@ def test_generate_and_update_key(prisma_client):
             # use generated key to auth in
             result = await user_api_key_auth(request=request, api_key=bearer_token)
             print(f"result: {result}")
-            result.user_role = LitellmUserRoles.PROXY_ADMIN.value
+            result.user_role = LitellmUserRoles.PROXY_ADMIN
 
             result_delete_key = await delete_key_fn(
                 data=delete_key_request, user_api_key_dict=result
@@ -2048,7 +2046,7 @@ async def test_master_key_hashing(prisma_client):
         await new_team(
             NewTeamRequest(team_id=_team_id),
             user_api_key_dict=UserAPIKeyAuth(
-                user_role=LitellmUserRoles.PROXY_ADMIN.value,
+                user_role=LitellmUserRoles.PROXY_ADMIN,
                 api_key="sk-1234",
                 user_id="1234",
             ),
@@ -2088,7 +2086,7 @@ async def test_reset_spend_authentication(prisma_client):
     """
     1. Test master key can access this route  -> ONLY MASTER KEY SHOULD BE ABLE TO RESET SPEND
     2. Test that non-master key gets rejected
-    3. Test that non-master key with role == LitellmUserRoles.PROXY_ADMIN.value or admin gets rejected
+    3. Test that non-master key with role == LitellmUserRoles.PROXY_ADMIN or admin gets rejected
     """
 
     print("prisma client=", prisma_client)
@@ -2133,10 +2131,10 @@ async def test_reset_spend_authentication(prisma_client):
             in e.message
         )
 
-    # Test 3 - Non-Master Key with role == LitellmUserRoles.PROXY_ADMIN.value or admin
+    # Test 3 - Non-Master Key with role == LitellmUserRoles.PROXY_ADMIN or admin
     _response = await new_user(
         data=NewUserRequest(
-            user_role=LitellmUserRoles.PROXY_ADMIN.value,
+            user_role=LitellmUserRoles.PROXY_ADMIN,
             tpm_limit=20,
         )
     )
@@ -2186,7 +2184,7 @@ async def test_create_update_team(prisma_client):
             rpm_limit=20,
         ),
         user_api_key_dict=UserAPIKeyAuth(
-            user_role=LitellmUserRoles.PROXY_ADMIN.value,
+            user_role=LitellmUserRoles.PROXY_ADMIN,
             api_key="sk-1234",
             user_id="1234",
         ),
@@ -2214,7 +2212,7 @@ async def test_create_update_team(prisma_client):
             rpm_limit=30,
         ),
         user_api_key_dict=UserAPIKeyAuth(
-            user_role=LitellmUserRoles.PROXY_ADMIN.value,
+            user_role=LitellmUserRoles.PROXY_ADMIN,
             api_key="sk-1234",
             user_id="1234",
         ),