UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-22 08:07:55 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

enh: more robust knowledge file info retrieval

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-25 21:58:29 -07:00
parent 5d2714bc92
commit 21a28e3bf0
1 changed files with 33 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
backend/open_webui/apps/webui/routers/knowledge.py
View File

@ -47,15 +47,40 @@ async def get_knowledge_items(
detail=ERROR_MESSAGES.NOT_FOUND,
)
else:
return [
KnowledgeResponse(
**knowledge.model_dump(),
files=Files.get_file_metadatas_by_ids(
knowledge.data.get("file_ids", []) if knowledge.data else []
),
knowledge_bases = []
for knowledge in Knowledges.get_knowledge_items():
files = Files.get_file_metadatas_by_ids(
knowledge.data.get("file_ids", []) if knowledge.data else []
)
for knowledge in Knowledges.get_knowledge_items()
]
# Check if all files exist
if len(files) != len(knowledge.data.get("file_ids", [])):
missing_files = list(
set(knowledge.data.get("file_ids", []))
- set([file.id for file in files])
)
if missing_files:
data = knowledge.data or {}
file_ids = data.get("file_ids", [])
for missing_file in missing_files:
file_ids.remove(missing_file)
data["file_ids"] = file_ids
Knowledges.update_knowledge_by_id(
id=knowledge.id, form_data=KnowledgeUpdateForm(data=data)
)
files = Files.get_file_metadatas_by_ids(file_ids)
knowledge_bases.append(
KnowledgeResponse(
**knowledge.model_dump(),
files=files,
)
)
return knowledge_bases
############################