jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add tidb on qdrant type (#9831)

Browse source 
Co-authored-by: Zhaofeng Miao <522856232@qq.com>
This commit is contained in:
Jyong 2024-10-25 13:57:03 +08:00 • committed by GitHub
commit 18106a4fc6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
27 changed files with 1648 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/extensions/ext_celery.py
View file

@ -1,6 +1,7 @@
from datetime import timedelta
from celery import Celery, Task
from celery.schedules import crontab
from flask import Flask
from configs import dify_config
@ -55,6 +56,8 @@ def init_app(app: Flask) -> Celery:
imports = [
"schedule.clean_embedding_cache_task",
"schedule.clean_unused_datasets_task",
"schedule.create_tidb_serverless_task",
"schedule.update_tidb_serverless_status_task",
]
day = dify_config.CELERY_BEAT_SCHEDULER_TIME
beat_schedule = {
@ -66,6 +69,14 @@ def init_app(app: Flask) -> Celery:
"task": "schedule.clean_unused_datasets_task.clean_unused_datasets_task",
"schedule": timedelta(days=day),
},
"create_tidb_serverless_task": {
"task": "schedule.create_tidb_serverless_task.create_tidb_serverless_task",
"schedule": crontab(minute="0", hour="*"),
},
"update_tidb_serverless_status_task": {
"task": "schedule.update_tidb_serverless_status_task.update_tidb_serverless_status_task",
"schedule": crontab(minute="30", hour="*"),
},
}
celery_app.conf.update(beat_schedule=beat_schedule, imports=imports)