UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 21:13:59 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #6433 from open-webui/dev

Browse Source 
fix: knowledge issue
This commit is contained in:
Timothy Jaeryang Baek 2024-10-26 01:19:38 -07:00 committed by GitHub
commit 1cd7d83f50
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -50,29 +50,32 @@ async def get_knowledge_items(
knowledge_bases = [] knowledge_bases = []
for knowledge in Knowledges.get_knowledge_items(): 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 files = []
if len(files) != len(knowledge.data.get("file_ids", [])): if knowledge.data:
missing_files = list( files = Files.get_file_metadatas_by_ids(
set(knowledge.data.get("file_ids", [])) 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: # Check if all files exist
file_ids.remove(missing_file) if len(files) != len(knowledge.data.get("file_ids", [])):
missing_files = list(
data["file_ids"] = file_ids set(knowledge.data.get("file_ids", []))
Knowledges.update_knowledge_by_id( - set([file.id for file in files])
id=knowledge.id, form_data=KnowledgeUpdateForm(data=data)
) )
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( knowledge_bases.append(
KnowledgeResponse( KnowledgeResponse(