|
@@ -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()
|