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

fix: preserve chunks when updating compilation templates (#17035)

Browse Source
This commit is contained in:
buua436
2026-07-20 09:48:20 +08:00
committed by GitHub
parent 962fe9ff57
commit 99f13172ad
1 changed files with 3 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
api/apps/restful_apis/document_api.py
View File

@@ -79,14 +79,6 @@ from common.ssrf_guard import assert_url_is_safe
from rag.nlp import search
def _compilation_template_group_id_changed(old_config, new_config) -> bool:
from rag.svr.task_executor_refactor.chunk_post_processor import (
_parser_config_compilation_template_group_ids,
)
return _parser_config_compilation_template_group_ids(old_config) != _parser_config_compilation_template_group_ids(new_config)
def _normalize_parser_config_compilation_template_group_ids(parser_config) -> bool:
from rag.svr.task_executor_refactor.chunk_post_processor import (
_parser_config_compilation_template_group_ids,
@@ -260,16 +252,12 @@ async def update_document(tenant_id, dataset_id, document_id):
if "parser_id" in req and ((doc.type == FileType.VISUAL and req["parser_id"] != "picture") or (re.search(r"\.(ppt|pptx|pages)$", doc.name) and req["parser_id"] != "presentation")):
return get_data_error_result(message="Not supported yet!")
parser_config_template_group_changed = False
# parser config provided (already validated in UpdateDocumentReq), update it.
# Changing the document-scoped knowledge compilation template group
# affects parse output, so the document must be parsed again for it to
# execute.
# Changing the document-scoped knowledge compilation template group must
# not remove the existing chunks.
if update_doc_req.parser_config:
old_parser_config = dict(doc.parser_config or {})
req["parser_config"].update(update_doc_req.parser_config.ext)
parser_config_template_group_touched = _normalize_parser_config_compilation_template_group_ids(req["parser_config"])
parser_config_template_group_changed = parser_config_template_group_touched and _compilation_template_group_id_changed(old_parser_config, req["parser_config"])
_normalize_parser_config_compilation_template_group_ids(req["parser_config"])
DocumentService.update_parser_config(doc.id, req["parser_config"])
# A non-empty pipeline_id selects pipeline parsing; an explicitly empty
@@ -281,12 +269,6 @@ async def update_document(tenant_id, dataset_id, document_id):
elif update_doc_req.chunk_method:
if error := update_chunk_method(req, doc, tenant_id):
return error
if parser_config_template_group_changed and doc.parser_id.lower() == req["chunk_method"].lower():
if error := reset_document_for_reparse(doc, tenant_id, pipeline_id=""):
return error
elif parser_config_template_group_changed:
if error := reset_document_for_reparse(doc, tenant_id, pipeline_id=""):
return error
if "enabled" in req: # already checked in UpdateDocumentReq - it's int if present
# "enabled" flag provided, the update method will check if it's changed and then update if so