scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-11-04 07:47:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

fix(kb): clean up files when resetting knowledge bases

Browse source
This commit is contained in:
Akhil-Jasson 2026-08-04 02:33:26 +05:30
parent 8faaf2cd1e
commit 6e07048587
1 changed files with 24 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1752,6 +1752,7 @@ async def reset_knowledge_by_id(
request: Request,
id: str,
include_directories: bool = Query(True),
delete_file: bool = Query(True),
user=Depends(get_verified_user),
db: AsyncSession = Depends(get_async_session),
):
@ -1780,6 +1781,8 @@ async def reset_knowledge_by_id(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
files = await Knowledges.get_files_by_id(knowledge_id=id, db=db)
try:
await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=id)
except Exception as e:
@ -1787,13 +1790,33 @@ async def reset_knowledge_by_id(
pass
knowledge = await Knowledges.reset_knowledge_by_id(id=id, include_directories=include_directories, db=db)
if knowledge and delete_file:
for file in files:
if file.user_id != user.id and user.role != 'admin':
continue
try:
file_collection = f'file-{file.id}'
if await ASYNC_VECTOR_DB_CLIENT.has_collection(collection_name=file_collection):
await ASYNC_VECTOR_DB_CLIENT.delete_collection(collection_name=file_collection)
except Exception as e:
log.debug('This was most likely caused by bypassing embedding processing')
log.debug(e)
await Files.delete_file_by_id(file.id, db=db)
try:
await asyncio.to_thread(Storage.delete_file, file.path)
except Exception as e:
log.debug('Error deleting reset knowledge file storage: %s', e)
if knowledge:
await publish_event(
request,
EVENTS.KNOWLEDGE_RESET,
actor=user,
subject_id=id,
data={'include_directories': include_directories},
data={'include_directories': include_directories, 'delete_file': delete_file},
)
return knowledge