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

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2026-10-10 21:29:13 +04:00
parent 79d376a1f3
commit 9202c7f100
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1886,7 +1886,6 @@ async def sync_knowledge_diff(
await _verify_knowledge_write_access(id, user, db)
# ── Index existing state ──
knowledge_files = await Knowledges.get_files_with_directory_ids(id, db=db)
existing_directories = await Knowledges.get_all_directories(id, db=db)
# Build directory path lookups
@ -1905,6 +1904,18 @@ async def sync_knowledge_diff(
directory_path_by_id[directory.id] = full_path
directory_id_by_path[full_path] = directory.id
# Retry uploads left pending by a restart after an hour.
pending_files = {
(
directory_path_by_id.get(file.meta.data.get('directory_id'), ''),
file.filename,
file.meta.model_dump().get('file_hash'),
)
for file in await Files.get_pending_files_for_knowledge(id, db=db)
if file.created_at >= int(time.time()) - 3600
}
knowledge_files = await Knowledges.get_files_with_directory_ids(id, db=db)
# Index existing files by (path, filename) → {file_id, checksum}
indexed_files: dict[tuple[str, str], dict] = {}
for file_model, directory_id in knowledge_files:
@ -1926,6 +1937,10 @@ async def sync_knowledge_diff(
key = (entry.path, entry.filename)
manifest_keys.add(key)
if key not in indexed_files and (*key, entry.checksum) in pending_files:
unmodified_count += 1
continue
if key not in indexed_files:
added.append({'filename': entry.filename, 'path': entry.path})
elif indexed_files[key]['checksum'] != entry.checksum: