refac

backend/open_webui/routers/files.py

@@ -31,6 +31,7 @@ from open_webui.models.channels import Channels
 from open_webui.models.users import Users
 from open_webui.models.files import (
     FileForm,
+    FileListResponse,
     FileModel,
     FileModelResponse,
     Files,
@@ -326,23 +327,31 @@ def upload_file_handler(
 ############################
 
 
-@router.get("/", response_model=list[FileModelResponse])
+PAGE_SIZE = 50
+
+
+@router.get("/", response_model=FileListResponse)
 async def list_files(
     user=Depends(get_verified_user),
+    page: int = Query(1, ge=1, description="Page number (1-indexed)"),
     content: bool = Query(True),
     db: Session = Depends(get_session),
 ):
-    if user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL:
-        files = Files.get_files(db=db)
-    else:
-        files = Files.get_files_by_user_id(user.id, db=db)
+    skip = (page - 1) * PAGE_SIZE
+    user_id = (
+        None if (user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL) else user.id
+    )
+
+    result = Files.get_file_list(
+        user_id=user_id, skip=skip, limit=PAGE_SIZE, db=db
+    )
 
     if not content:
-        for file in files:
-            if "content" in file.data:
+        for file in result.items:
+            if file.data and "content" in file.data:
                 del file.data["content"]
 
-    return files
+    return result
 
 
 ############################