hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-24 18:23:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat/clean message records (#10588)

Browse Source
This commit is contained in:
Jyong
2024-11-18 16:57:39 +08:00
committed by GitHub
parent c49efc0c22
commit 128efc3193
6 changed files with 127 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/extensions/ext_celery.py
View File

@@ -68,6 +68,7 @@ def init_app(app: Flask) -> Celery:
"schedule.clean_unused_datasets_task",
"schedule.create_tidb_serverless_task",
"schedule.update_tidb_serverless_status_task",
"schedule.clean_messages",
]
day = dify_config.CELERY_BEAT_SCHEDULER_TIME
beat_schedule = {
@@ -87,6 +88,10 @@ def init_app(app: Flask) -> Celery:
"task": "schedule.update_tidb_serverless_status_task.update_tidb_serverless_status_task",
"schedule": crontab(minute="30", hour="*"),
},
"clean_messages": {
"task": "schedule.clean_messages.clean_messages",
"schedule": timedelta(days=day),
},
}
celery_app.conf.update(beat_schedule=beat_schedule, imports=imports)