mirror of
http://112.124.100.131/huang.ze/ebiz-dify-ai.git
synced 2025-12-09 02:46:52 +08:00
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user