Browse Source

fix: knowledge files issue

Timothy Jaeryang Baek 5 months ago
parent
commit
2b890cf747
1 changed files with 14 additions and 9 deletions
  1. 14 9
      backend/open_webui/apps/webui/routers/knowledge.py

+ 14 - 9
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
 
 
 ############################