fix: uploaded file should not be deleted

This commit is contained in:
Timothy Jaeryang Baek 2025-01-29 14:20:51 -08:00
parent 40a4443949
commit 4abede9a2b

View File

@ -264,7 +264,8 @@ def add_file_to_knowledge_by_id(
detail=ERROR_MESSAGES.NOT_FOUND,
)
if (knowledge.user_id != user.id
if (
knowledge.user_id != user.id
and not has_access(user.id, "write", knowledge.access_control)
and user.role != "admin"
):
@ -441,10 +442,6 @@ def remove_file_from_knowledge_by_id(
if VECTOR_DB_CLIENT.has_collection(collection_name=file_collection):
VECTOR_DB_CLIENT.delete_collection(collection_name=file_collection)
# Delete physical file
if file.path:
Storage.delete_file(file.path)
# Delete file from database
Files.delete_file_by_id(form_data.file_id)