feat: add queue to celery task (#688)

2023-07-31 13:13:08 +08:00
parent 396197e881
commit 0c18cab111
19 changed files with 20 additions and 19 deletions
--- a/api/tasks/deal_dataset_vector_index_task.py
+++ b/api/tasks/deal_dataset_vector_index_task.py
@@ -11,7 +11,7 @@ from models.dataset import DocumentSegment, Dataset
 from models.dataset import Document as DatasetDocument


-@shared_task
+@shared_task(queue='dataset')
 def deal_dataset_vector_index_task(dataset_id: str, action: str):
    """
    Async deal dataset from index