ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix: complete file cleanup when removing from Knowledge Base

Browse source
This commit is contained in:
Juanan Pereira 2025-01-12 18:33:27 +01:00
parent b1bcca90b5
commit 913e0d391b
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
backend/open_webui/routers/knowledge.py
View file

@ -17,7 +17,7 @@ from open_webui.routers.retrieval import (
process_files_batch, process_files_batch,
BatchProcessFilesForm, BatchProcessFilesForm,
) )
from open_webui.storage.provider import Storage
from open_webui.constants import ERROR_MESSAGES from open_webui.constants import ERROR_MESSAGES
from open_webui.utils.auth import get_verified_user from open_webui.utils.auth import get_verified_user
@ -420,6 +420,18 @@ def remove_file_from_knowledge_by_id(
collection_name=knowledge.id, filter={"file_id": form_data.file_id} collection_name=knowledge.id, filter={"file_id": form_data.file_id}
) )
# Remove the file's collection from vector database
file_collection = f"file-{form_data.file_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)
if knowledge: if knowledge:
data = knowledge.data or {} data = knowledge.data or {}
file_ids = data.get("file_ids", []) file_ids = data.get("file_ids", [])