Feat/clean message records (#10588)

2024-11-18 16:57:39 +08:00
parent c49efc0c22
commit 128efc3193
6 changed files with 127 additions and 8 deletions
--- a/api/extensions/ext_celery.py
+++ b/api/extensions/ext_celery.py
@@ -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)