diff --git a/api/services/dataset_service.py b/api/services/dataset_service.py index 892ec52d63..e64fa881f0 100644 --- a/api/services/dataset_service.py +++ b/api/services/dataset_service.py @@ -464,7 +464,7 @@ class DocumentService: data_source_info = { "upload_file_id": file_id, } - document = DocumentService.save_document(dataset, dataset_process_rule.id, + document = DocumentService.build_document(dataset, dataset_process_rule.id, document_data["data_source"]["type"], document_data["doc_form"], document_data["doc_language"], @@ -510,7 +510,7 @@ class DocumentService: "notion_page_icon": page['page_icon'], "type": page['type'] } - document = DocumentService.save_document(dataset, dataset_process_rule.id, + document = DocumentService.build_document(dataset, dataset_process_rule.id, document_data["data_source"]["type"], document_data["doc_form"], document_data["doc_language"], @@ -534,7 +534,7 @@ class DocumentService: return documents, batch @staticmethod - def save_document(dataset: Dataset, process_rule_id: str, data_source_type: str, document_form: str, + def build_document(dataset: Dataset, process_rule_id: str, data_source_type: str, document_form: str, document_language: str, data_source_info: dict, created_from: str, position: int, account: Account, name: str, batch: str): @@ -1007,10 +1007,11 @@ class SegmentService: cache_result = redis_client.get(indexing_cache_key) if cache_result is not None: raise ValueError("Segment is deleting.") - # send delete segment index task - redis_client.setex(indexing_cache_key, 600, 1) + # enabled segment need to delete index if segment.enabled: + # send delete segment index task + redis_client.setex(indexing_cache_key, 600, 1) delete_segment_from_index_task.delay(segment.id, segment.index_node_id, dataset.id, document.id) db.session.delete(segment) db.session.commit()