Add data clean schedule (#1859)

Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong
2024-01-02 15:29:18 +08:00
committed by GitHub
parent 06d2d8cea3
commit 595e9b25ba
5 changed files with 124 additions and 1 deletions

View File

@@ -1,3 +1,5 @@
from datetime import timedelta
from celery import Task, Celery
from flask import Flask
@@ -35,4 +37,25 @@ def init_app(app: Flask) -> Celery:
celery_app.set_default()
app.extensions["celery"] = celery_app
imports = [
"schedule.clean_embedding_cache_task",
"schedule.clean_unused_datasets_task",
]
beat_schedule = {
'clean_embedding_cache_task': {
'task': 'schedule.clean_embedding_cache_task.clean_embedding_cache_task',
'schedule': timedelta(minutes=1),
},
'clean_unused_datasets_task': {
'task': 'schedule.clean_unused_datasets_task.clean_unused_datasets_task',
'schedule': timedelta(minutes=10),
}
}
celery_app.conf.update(
beat_schedule=beat_schedule,
imports=imports
)
return celery_app