diff --git a/api/tasks/image_generation_task.py b/api/tasks/image_generation_task.py index 147adf0632..2997e1b259 100644 --- a/api/tasks/image_generation_task.py +++ b/api/tasks/image_generation_task.py @@ -4,6 +4,7 @@ import time import click from celery import shared_task # type: ignore + from configs import dify_config from core.app.entities.app_invoke_entities import InvokeFrom from extensions.ext_database import db diff --git a/api/tasks/phone_sms_code_login.py b/api/tasks/phone_sms_code_login.py index 03109cfd48..a2bfeffa14 100644 --- a/api/tasks/phone_sms_code_login.py +++ b/api/tasks/phone_sms_code_login.py @@ -3,6 +3,7 @@ import time import click from celery import shared_task # type: ignore + from extensions.ext_phone_sms import phone_sms