From f1c123c81937f0facb48843d80c9b4e3f12d4ebc Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sat, 26 Oct 2024 01:19:17 -0700 Subject: [PATCH] fix: knowledge issue --- .../apps/webui/routers/knowledge.py | 39 ++++++++++--------- 1 file changed, 21 insertions(+), 18 deletions(-) diff --git a/backend/open_webui/apps/webui/routers/knowledge.py b/backend/open_webui/apps/webui/routers/knowledge.py index fe5d27e3f..1b5381a74 100644 --- a/backend/open_webui/apps/webui/routers/knowledge.py +++ b/backend/open_webui/apps/webui/routers/knowledge.py @@ -50,29 +50,32 @@ async def get_knowledge_items( 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 [] - ) - # 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]) + files = [] + if knowledge.data: + files = Files.get_file_metadatas_by_ids( + knowledge.data.get("file_ids", []) ) - 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) + # 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", []) - files = Files.get_file_metadatas_by_ids(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(