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

chore: apply static type checks on celery async task dispatches and imports (#24418)

Browse source
This commit is contained in:
Bowen Liang 2025-08-24 23:07:22 +08:00 • committed by GitHub
commit d8000251ff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
48 changed files with 101 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/extensions/ext_celery.py
View file

@ -3,8 +3,8 @@ from datetime import timedelta
from typing import Any, Optional
import pytz
from celery import Celery, Task # type: ignore
from celery.schedules import crontab # type: ignore
from celery import Celery, Task
from celery.schedules import crontab
from configs import dify_config
from dify_app import DifyApp
@ -66,7 +66,6 @@ def init_app(app: DifyApp) -> Celery:
task_cls=FlaskTask,
broker=dify_config.CELERY_BROKER_URL,
backend=dify_config.CELERY_BACKEND,
task_ignore_result=True,
)
celery_app.conf.update(
@ -77,6 +76,7 @@ def init_app(app: DifyApp) -> Celery:
worker_task_log_format=dify_config.LOG_FORMAT,
worker_hijack_root_logger=False,
timezone=pytz.timezone(dify_config.LOG_TZ or "UTC"),
task_ignore_result=True,
)
# Apply SSL configuration if enabled

2
api/extensions/ext_otel.py
View file

@ -8,7 +8,7 @@ import sys
from typing import Union
import flask
from celery.signals import worker_init # type: ignore
from celery.signals import worker_init
from flask_login import user_loaded_from_request, user_logged_in # type: ignore
from configs import dify_config