diff --git a/src/backend/langflow/services/database/models/user/crud.py b/src/backend/langflow/services/database/models/user/crud.py index 93d5dd801..abc79f49f 100644 --- a/src/backend/langflow/services/database/models/user/crud.py +++ b/src/backend/langflow/services/database/models/user/crud.py @@ -1,13 +1,12 @@ from datetime import datetime, timezone from typing import Union from uuid import UUID -from fastapi import Depends, HTTPException +from fastapi import Depends, HTTPException, status from langflow.services.database.models.user.user import User, UserUpdate from langflow.services.utils import get_session from sqlalchemy.exc import IntegrityError from sqlmodel import Session - from sqlalchemy.orm.attributes import flag_modified @@ -37,7 +36,9 @@ def update_user( changed = True if not changed: - raise HTTPException(status_code=304, detail="Nothing to update") + raise HTTPException( + status_code=status.HTTP_304_NOT_MODIFIED, detail="Nothing to update" + ) user_db.updated_at = datetime.now(timezone.utc) flag_modified(user_db, "updated_at") diff --git a/tests/test_user.py b/tests/test_user.py index 4d99c5c9c..4f2307624 100644 --- a/tests/test_user.py +++ b/tests/test_user.py @@ -163,6 +163,14 @@ def test_patch_user(client, active_user, logged_in_headers): f"/api/v1/users/{user_id}", json=update_data.dict(), headers=logged_in_headers ) assert response.status_code == 304, response.json() + update_data = UserUpdate( + profile_image="new_image", + ) + + response = client.patch( + f"/api/v1/users/{user_id}", json=update_data.dict(), headers=logged_in_headers + ) + assert response.status_code == 200, response.json() def test_patch_reset_password(client, active_user, logged_in_headers):