Procházet zdrojové kódy

fix: knowledge files issue

Timothy Jaeryang Baek před 5 měsíci
rodič
revize
2b890cf747
1 změnil soubory, kde provedl 14 přidání a 9 odebrání
  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")
         knowledge_bases = Knowledges.get_knowledge_bases_by_user_id(user.id, "read")
 
 
     # Get files for each knowledge base
     # Get files for each knowledge base
+    knowledge_with_files = []
     for knowledge_base in knowledge_bases:
     for knowledge_base in knowledge_bases:
         files = []
         files = []
         if knowledge_base.data:
         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)
                     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])
 @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")
         knowledge_bases = Knowledges.get_knowledge_bases_by_user_id(user.id, "write")
 
 
     # Get files for each knowledge base
     # Get files for each knowledge base
+    knowledge_with_files = []
     for knowledge_base in knowledge_bases:
     for knowledge_base in knowledge_bases:
         files = []
         files = []
         if knowledge_base.data:
         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)
                     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
 
 
 
 
 ############################
 ############################