From 2b890cf747eeb06e84479cd937f8e48bbba2ce52 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Wed, 20 Nov 2024 10:02:14 -0800 Subject: [PATCH] fix: knowledge files issue --- .../apps/webui/routers/knowledge.py | 23 +++++++++++-------- 1 file changed, 14 insertions(+), 9 deletions(-) diff --git a/backend/open_webui/apps/webui/routers/knowledge.py b/backend/open_webui/apps/webui/routers/knowledge.py index 30fc5d9d9..1b063cda2 100644 --- a/backend/open_webui/apps/webui/routers/knowledge.py +++ b/backend/open_webui/apps/webui/routers/knowledge.py @@ -43,6 +43,7 @@ async def get_knowledge(user=Depends(get_verified_user)): knowledge_bases = Knowledges.get_knowledge_bases_by_user_id(user.id, "read") # Get files for each knowledge base + knowledge_with_files = [] for knowledge_base in knowledge_bases: files = [] if knowledge_base.data: @@ -70,12 +71,14 @@ async def get_knowledge(user=Depends(get_verified_user)): files = Files.get_file_metadatas_by_ids(file_ids) - knowledge_base = KnowledgeResponse( - **knowledge_base.model_dump(), - files=files, + knowledge_with_files.append( + KnowledgeUserResponse( + **knowledge_base.model_dump(), + files=files, + ) ) - return knowledge_bases + return knowledge_with_files @router.get("/list", response_model=list[KnowledgeUserResponse]) @@ -88,6 +91,7 @@ async def get_knowledge_list(user=Depends(get_verified_user)): knowledge_bases = Knowledges.get_knowledge_bases_by_user_id(user.id, "write") # Get files for each knowledge base + knowledge_with_files = [] for knowledge_base in knowledge_bases: files = [] if knowledge_base.data: @@ -115,12 +119,13 @@ async def get_knowledge_list(user=Depends(get_verified_user)): files = Files.get_file_metadatas_by_ids(file_ids) - knowledge_base = KnowledgeResponse( - **knowledge_base.model_dump(), - files=files, + knowledge_with_files.append( + KnowledgeUserResponse( + **knowledge_base.model_dump(), + files=files, + ) ) - - return knowledge_bases + return knowledge_with_files ############################