feat(sqlalchemy): remove session reference from router

backend/apps/webui/routers/files.py

@@ -20,7 +20,6 @@ from fastapi.responses import StreamingResponse, JSONResponse, FileResponse
 from pydantic import BaseModel
 import json
 
-from apps.webui.internal.db import get_db
 from apps.webui.models.files import (
     Files,
     FileForm,
@@ -53,8 +52,7 @@ router = APIRouter()
 @router.post("/")
 def upload_file(
     file: UploadFile = File(...),
-    user=Depends(get_verified_user),
-    db=Depends(get_db)
+    user=Depends(get_verified_user)
 ):
     log.info(f"file.content_type: {file.content_type}")
     try:
@@ -72,7 +70,6 @@ def upload_file(
             f.close()
 
         file = Files.insert_new_file(
-            db,
             user.id,
             FileForm(
                 **{
@@ -109,8 +106,8 @@ def upload_file(
 
 
 @router.get("/", response_model=List[FileModel])
-async def list_files(user=Depends(get_verified_user), db=Depends(get_db)):
-    files = Files.get_files(db)
+async def list_files(user=Depends(get_verified_user)):
+    files = Files.get_files()
     return files
 
 
@@ -120,8 +117,8 @@ async def list_files(user=Depends(get_verified_user), db=Depends(get_db)):
 
 
 @router.delete("/all")
-async def delete_all_files(user=Depends(get_admin_user), db=Depends(get_db)):
-    result = Files.delete_all_files(db)
+async def delete_all_files(user=Depends(get_admin_user)):
+    result = Files.delete_all_files()
 
     if result:
         folder = f"{UPLOAD_DIR}"
@@ -157,8 +154,8 @@ async def delete_all_files(user=Depends(get_admin_user), db=Depends(get_db)):
 
 
 @router.get("/{id}", response_model=Optional[FileModel])
-async def get_file_by_id(id: str, user=Depends(get_verified_user), db=Depends(get_db)):
-    file = Files.get_file_by_id(db, id)
+async def get_file_by_id(id: str, user=Depends(get_verified_user)):
+    file = Files.get_file_by_id(id)
 
     if file:
         return file
@@ -175,8 +172,8 @@ async def get_file_by_id(id: str, user=Depends(get_verified_user), db=Depends(ge
 
 
 @router.get("/{id}/content", response_model=Optional[FileModel])
-async def get_file_content_by_id(id: str, user=Depends(get_verified_user), db=Depends(get_db)):
-    file = Files.get_file_by_id(db, id)
+async def get_file_content_by_id(id: str, user=Depends(get_verified_user)):
+    file = Files.get_file_by_id(id)
 
     if file:
         file_path = Path(file.meta["path"])
@@ -226,11 +223,11 @@ async def get_file_content_by_id(id: str, user=Depends(get_verified_user)):
 
 
 @router.delete("/{id}")
-async def delete_file_by_id(id: str, user=Depends(get_verified_user), db=Depends(get_db)):
-    file = Files.get_file_by_id(db, id)
+async def delete_file_by_id(id: str, user=Depends(get_verified_user)):
+    file = Files.get_file_by_id(id)
 
     if file:
-        result = Files.delete_file_by_id(db, id)
+        result = Files.delete_file_by_id(id)
         if result:
             return {"message": "File deleted successfully"}
         else: