refac

backend/open_webui/models/chats.py

@@ -694,13 +694,12 @@ class ChatTable:
             await session.commit()
             return result.rowcount or 0
 
-    async def mark_root_chats_read_by_user_id(self, user_id: str, db: AsyncSession | None = None) -> int:
+    async def mark_chats_read_by_user_id(self, user_id: str, db: AsyncSession | None = None) -> int:
         async with get_async_db_context(db) as session:
             result = await session.execute(
                 update(Chat)
                 .where(
                     Chat.user_id == user_id,
-                    Chat.folder_id.is_(None),
                     Chat.archived == False,
                     Chat.meta['internal'].as_boolean().is_not(True),
                 )

backend/open_webui/routers/chats.py

@@ -256,11 +256,14 @@ async def get_session_user_chat_list(
 
 
 @router.post('/read')
-async def mark_root_chats_read_by_user_id(
+async def mark_chats_read_by_user_id(
     user=Depends(get_verified_user),
     db: AsyncSession = Depends(get_async_session),
 ):
-    return {'updated_count': await Chats.mark_root_chats_read_by_user_id(user.id, db=db)}
+    return {
+        'updated_count': await Chats.mark_chats_read_by_user_id(user.id, db=db),
+        'folder_unread_counts': await get_folder_unread_counts(user.id, db=db),
+    }
 
 
 ############################