update celery beat scheduler time to env (#6352)

This commit is contained in:
Jyong
2024-07-17 02:31:30 +08:00
committed by GitHub
parent 06fcc0c650
commit 7c397f5722
3 changed files with 16 additions and 3 deletions

View File

@@ -43,15 +43,15 @@ def init_app(app: Flask) -> Celery:
"schedule.clean_embedding_cache_task",
"schedule.clean_unused_datasets_task",
]
day = app.config["CELERY_BEAT_SCHEDULER_TIME"]
beat_schedule = {
'clean_embedding_cache_task': {
'task': 'schedule.clean_embedding_cache_task.clean_embedding_cache_task',
'schedule': timedelta(days=1),
'schedule': timedelta(days=day),
},
'clean_unused_datasets_task': {
'task': 'schedule.clean_unused_datasets_task.clean_unused_datasets_task',
'schedule': timedelta(minutes=3),
'schedule': timedelta(days=day),
}
}
celery_app.conf.update(