AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-11 21:09:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor:cache in place & function name (#1001)

Browse Source
This commit is contained in:
yezhwi 2023-08-24 22:54:21 +08:00 committed by GitHub
parent a3fec11438
commit c1602aafc7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/services/dataset_service.py
View File

@ -464,7 +464,7 @@ class DocumentService:
data_source_info = { data_source_info = {
"upload_file_id": file_id, "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["data_source"]["type"],
document_data["doc_form"], document_data["doc_form"],
document_data["doc_language"], document_data["doc_language"],
@ -510,7 +510,7 @@ class DocumentService:
"notion_page_icon": page['page_icon'], "notion_page_icon": page['page_icon'],
"type": page['type'] "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["data_source"]["type"],
document_data["doc_form"], document_data["doc_form"],
document_data["doc_language"], document_data["doc_language"],
@ -534,7 +534,7 @@ class DocumentService:
return documents, batch return documents, batch
@staticmethod @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, document_language: str, data_source_info: dict, created_from: str, position: int,
account: Account, account: Account,
name: str, batch: str): name: str, batch: str):
@ -1007,10 +1007,11 @@ class SegmentService:
cache_result = redis_client.get(indexing_cache_key) cache_result = redis_client.get(indexing_cache_key)
if cache_result is not None: if cache_result is not None:
raise ValueError("Segment is deleting.") raise ValueError("Segment is deleting.")
# send delete segment index task
redis_client.setex(indexing_cache_key, 600, 1)
# enabled segment need to delete index # enabled segment need to delete index
if segment.enabled: 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) delete_segment_from_index_task.delay(segment.id, segment.index_node_id, dataset.id, document.id)
db.session.delete(segment) db.session.delete(segment)
db.session.commit() db.session.commit()