add create tidb serverless job control (#10467)

Co-authored-by: crazywoola <427733928@qq.com>
2024-11-08 18:48:12 +08:00 · 2024-11-08 18:48:12 +08:00 · 445dcfe4d0
commit 445dcfe4d0
parent 25ca0278dd
6 changed files with 14 additions and 0 deletions
--- a/api/schedule/create_tidb_serverless_task.py
+++ b/api/schedule/create_tidb_serverless_task.py
@ -12,6 +12,8 @@ from models.dataset import TidbAuthBinding
@app.celery.task(queue="dataset")
 def create_tidb_serverless_task():
    click.echo(click.style("Start create tidb serverless task.", fg="green"))
+    if not dify_config.CREATE_TIDB_SERVICE_JOB_ENABLED:
+        return
    tidb_serverless_number = dify_config.TIDB_SERVERLESS_NUMBER
    start_at = time.perf_counter()
    while True: