refactor(types): annotate user request metadata and check_user_info validators (#45808)

Part of LIT-9423
This commit is contained in:
nate-berri 2026-10-10 12:08:31 -07:00 • committed by GitHub
parent b7d48ecd28
commit a859d844ed
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1905,7 +1905,7 @@ class UpdateUserRequestNoUserIDorEmail(GenerateRequestBase): # shared with Bulk
# repr=False keeps the plaintext out of management-endpoint alerts, which str() the request model
password: str | None = Field(default=None, repr=False)
spend: float | None = None
metadata: dict | None = None
metadata: dict[object, object] | None = None
user_alias: str | None = None
user_role: (
Literal[
@ -1927,7 +1927,7 @@ class UpdateUserRequest(UpdateUserRequestNoUserIDorEmail):
@model_validator(mode="before")
@classmethod
def check_user_info(cls, values):
def check_user_info(cls, values: Mapping[str, object]) -> Mapping[str, object]:
if values.get("user_id") is None and values.get("user_email") is None:
raise ValueError("Either user id or user email must be provided")
return values
@ -4588,7 +4588,7 @@ class MemberDeleteRequest(LiteLLMPydanticObjectBase):
@model_validator(mode="before")
@classmethod
def check_user_info(cls, values):
def check_user_info(cls, values: Mapping[str, object]) -> Mapping[str, object]:
if values.get("user_id") is None and values.get("user_email") is None:
raise ValueError("Either user id or user email must be provided")
return values