Skip to content

Commit a8cf646

Browse files
fix(mypy): use LitellmUserRoles enum instead of raw string in custom_auth_basic
Fixes mypy error: Argument 'user_role' has incompatible type 'str'; expected 'LitellmUserRoles | None' Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
1 parent d4bd46b commit a8cf646

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from fastapi import Request
22

3-
from litellm.proxy._types import UserAPIKeyAuth
3+
from litellm.proxy._types import LitellmUserRoles, UserAPIKeyAuth
44

55

66
async def user_api_key_auth(request: Request, api_key: str) -> UserAPIKeyAuth:
@@ -9,7 +9,7 @@ async def user_api_key_auth(request: Request, api_key: str) -> UserAPIKeyAuth:
99
api_key="best-api-key-ever",
1010
user_id="best-user-id-ever",
1111
team_id="best-team-id-ever",
12-
user_role="proxy_admin",
12+
user_role=LitellmUserRoles.PROXY_ADMIN,
1313
)
1414
except Exception:
1515
raise Exception

0 commit comments

Comments
 (0)