refac/enh: drop profile_image_url field in responses

This commit is contained in:
Timothy Jaeryang Baek
2025-11-26 21:47:20 -05:00
parent 3fe5a47050
commit 384753c6ca
4 changed files with 17 additions and 27 deletions
+3 -4
View File
@@ -16,9 +16,8 @@ from open_webui.models.auths import (
SigninResponse,
SignupForm,
UpdatePasswordForm,
UserResponse,
)
from open_webui.models.users import Users, UpdateProfileForm
from open_webui.models.users import UserProfileImageResponse, Users, UpdateProfileForm
from open_webui.models.groups import Groups
from open_webui.models.oauth_sessions import OAuthSessions
@@ -78,7 +77,7 @@ log.setLevel(SRC_LOG_LEVELS["MAIN"])
############################
class SessionUserResponse(Token, UserResponse):
class SessionUserResponse(Token, UserProfileImageResponse):
expires_at: Optional[int] = None
permissions: Optional[dict] = None
@@ -149,7 +148,7 @@ async def get_session_user(
############################
@router.post("/update/profile", response_model=UserResponse)
@router.post("/update/profile", response_model=UserProfileImageResponse)
async def update_profile(
form_data: UpdateProfileForm, session_user=Depends(get_verified_user)
):
+4 -5
View File
@@ -359,14 +359,14 @@ async def update_user_info_by_session_user(
############################
class UserResponse(BaseModel):
class UserActiveResponse(BaseModel):
name: str
profile_image_url: str
profile_image_url: Optional[str] = None
active: Optional[bool] = None
model_config = ConfigDict(extra="allow")
@router.get("/{user_id}", response_model=UserResponse)
@router.get("/{user_id}", response_model=UserActiveResponse)
async def get_user_by_id(user_id: str, user=Depends(get_verified_user)):
# Check if user_id is a shared chat
# If it is, get the user_id from the chat
@@ -384,11 +384,10 @@ async def get_user_by_id(user_id: str, user=Depends(get_verified_user)):
user = Users.get_user_by_id(user_id)
if user:
return UserResponse(
return UserActiveResponse(
**{
"id": user.id,
"name": user.name,
"profile_image_url": user.profile_image_url,
"active": get_active_status_by_user_id(user_id),
}
)