OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-24 06:44:24 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac: knowledge file ac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-03-31 01:10:18 -07:00
parent 021e25ade1
commit a5dbd2e8dd
1 changed files with 49 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
backend/open_webui/routers/files.py
View File

@ -24,6 +24,8 @@ from open_webui.models.files import (
FileModelResponse, FileModelResponse,
Files, Files,
) )
from open_webui.models.knowledge import Knowledges
from open_webui.routers.knowledge import get_knowledge, get_knowledge_list from open_webui.routers.knowledge import get_knowledge, get_knowledge_list
from open_webui.routers.retrieval import ProcessFileForm, process_file from open_webui.routers.retrieval import ProcessFileForm, process_file
from open_webui.routers.audio import transcribe from open_webui.routers.audio import transcribe
@ -37,10 +39,15 @@ log.setLevel(SRC_LOG_LEVELS["MODELS"])
router = APIRouter() router = APIRouter()
############################ ############################
# Check if the current user has access to a file through any knowledge bases the user may be in. # Check if the current user has access to a file through any knowledge bases the user may be in.
############################ ############################
async def check_user_has_access_to_file_via_any_knowledge_base(file_id: Optional[str], access_type: str, user=Depends(get_verified_user)) -> bool:
def has_access_to_file(
file_id: Optional[str], access_type: str, user=Depends(get_verified_user)
) -> bool:
file = Files.get_file_by_id(file_id) file = Files.get_file_by_id(file_id)
log.debug(f"Checking if user has {access_type} access to file") log.debug(f"Checking if user has {access_type} access to file")
@ -52,28 +59,19 @@ async def check_user_has_access_to_file_via_any_knowledge_base(file_id: Optional
has_access = False has_access = False
knowledge_base_id = file.meta.get("collection_name") if file.meta else None knowledge_base_id = file.meta.get("collection_name") if file.meta else None
log.debug(f"Knowledge base associated with file: {knowledge_base_id}")
if knowledge_base_id:
if access_type == "read":
user_access = await get_knowledge(user=user) # get_knowledge checks for read access
elif access_type == "write":
user_access = await get_knowledge_list(user=user) # get_knowledge_list checks for write access
else:
user_access = list()
for knowledge_base in user_access: if knowledge_base_id:
knowledge_bases = Knowledges.get_knowledge_bases_by_user_id(
user.id, access_type
)
for knowledge_base in knowledge_bases:
if knowledge_base.id == knowledge_base_id: if knowledge_base.id == knowledge_base_id:
log.debug(f"User knowledge base with {access_type} access {knowledge_base.id} == File knowledge base {knowledge_base_id}")
has_access = True has_access = True
break break
log.debug(f"Does user have {access_type} access to file: {has_access}")
return has_access return has_access
############################ ############################
# Upload File # Upload File
############################ ############################
@ -213,9 +211,11 @@ async def get_file_by_id(id: str, user=Depends(get_verified_user)):
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_read_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "read", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_read_access: or user.role == "admin"
or has_access_to_file(id, "read", user)
):
return file return file
else: else:
raise HTTPException( raise HTTPException(
@ -239,9 +239,11 @@ async def get_file_data_content_by_id(id: str, user=Depends(get_verified_user)):
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_read_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "read", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_read_access: or user.role == "admin"
or has_access_to_file(id, "read", user)
):
return {"content": file.data.get("content", "")} return {"content": file.data.get("content", "")}
else: else:
raise HTTPException( raise HTTPException(
@ -271,9 +273,11 @@ async def update_file_data_content_by_id(
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_write_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "write", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_write_access: or user.role == "admin"
or has_access_to_file(id, "write", user)
):
try: try:
process_file( process_file(
request, request,
@ -310,9 +314,11 @@ async def get_file_content_by_id(
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_read_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "read", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_read_access: or user.role == "admin"
or has_access_to_file(id, "read", user)
):
try: try:
file_path = Storage.get_file(file.path) file_path = Storage.get_file(file.path)
file_path = Path(file_path) file_path = Path(file_path)
@ -376,9 +382,11 @@ async def get_html_file_content_by_id(id: str, user=Depends(get_verified_user)):
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_read_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "read", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_read_access: or user.role == "admin"
or has_access_to_file(id, "read", user)
):
try: try:
file_path = Storage.get_file(file.path) file_path = Storage.get_file(file.path)
file_path = Path(file_path) file_path = Path(file_path)
@ -416,9 +424,11 @@ async def get_file_content_by_id(id: str, user=Depends(get_verified_user)):
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_read_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "read", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_read_access: or user.role == "admin"
or has_access_to_file(id, "read", user)
):
file_path = file.path file_path = file.path
# Handle Unicode filenames # Handle Unicode filenames
@ -476,9 +486,11 @@ async def delete_file_by_id(id: str, user=Depends(get_verified_user)):
detail=ERROR_MESSAGES.NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND,
) )
has_write_access: bool = await check_user_has_access_to_file_via_any_knowledge_base(id, "write", user) if (
file.user_id == user.id
if file.user_id == user.id or user.role == "admin" or has_write_access: or user.role == "admin"
or has_access_to_file(id, "write", user)
):
# We should add Chroma cleanup here # We should add Chroma cleanup here
result = Files.delete_file_by_id(id) result = Files.delete_file_by_id(id)