zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-04 01:29:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: preserve existing links when bulk linking files to knowledge bases (#16587)

Browse Source
This commit is contained in:
euvre
2026-07-03 13:17:19 +08:00
committed by GitHub
parent 6a4b9be426
commit e65bac238e
1 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
api/apps/restful_apis/file2document_api.py
View File

@@ -33,26 +33,28 @@ logger = logging.getLogger(__name__)
def _convert_files(file_ids, kb_ids, user_id):
"""Synchronous worker: delete old docs and insert new ones for the given file/kb pairs."""
for id in file_ids:
informs = File2DocumentService.get_by_file_id(id)
for inform in informs:
doc_id = inform.document_id
e, doc = DocumentService.get_by_id(doc_id)
if not e:
continue
tenant_id = DocumentService.get_tenant_id(doc_id)
if not tenant_id:
logging.warning("tenant_id not found for doc_id=%s, skipping remove_document", doc_id)
continue
DocumentService.remove_document(doc, tenant_id)
File2DocumentService.delete_by_file_id(id)
"""Synchronous worker: add new docs for the given file/kb pairs while preserving existing links.
Previously this function replaced all existing links with the new ones, which caused
multi-select "link to knowledge base" to overwrite previous links. Now it only creates
documents for knowledge bases that are not already linked to the file, and leaves
existing links untouched.
"""
for id in file_ids:
e, file = FileService.get_by_id(id)
if not e:
continue
existing_links = {inform.document_id for inform in File2DocumentService.get_by_file_id(id)}
existing_kb_ids = set()
for doc_id in existing_links:
e, doc = DocumentService.get_by_id(doc_id)
if e and doc:
existing_kb_ids.add(doc.kb_id)
for kb_id in kb_ids:
if kb_id in existing_kb_ids:
continue
e, kb = KnowledgebaseService.get_by_id(kb_id)
if not e:
continue