From fd62527ef383b55adc65e6f859ba1a21630bec2e Mon Sep 17 00:00:00 2001 From: neatguycoding <15627489+NeatGuyCoding@users.noreply.github.com> Date: Thu, 3 Jul 2025 12:00:46 +0800 Subject: [PATCH] feat refactor: move MessageStatus to enums.py Signed-off-by: neatguycoding <15627489+NeatGuyCoding@users.noreply.github.com> --- api/core/app/task_pipeline/based_generate_task_pipeline.py | 2 +- api/models/model.py | 2 +- api/services/audio_service.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/core/app/task_pipeline/based_generate_task_pipeline.py b/api/core/app/task_pipeline/based_generate_task_pipeline.py index 5773eeb9b7..3ed0c3352f 100644 --- a/api/core/app/task_pipeline/based_generate_task_pipeline.py +++ b/api/core/app/task_pipeline/based_generate_task_pipeline.py @@ -19,8 +19,8 @@ from core.app.entities.task_entities import ( from core.errors.error import QuotaExceededError from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError from core.moderation.output_moderation import ModerationRule, OutputModeration -from models.model import Message from models.enums import MessageStatus +from models.model import Message logger = logging.getLogger(__name__) diff --git a/api/models/model.py b/api/models/model.py index 11c3a45710..93737043d5 100644 --- a/api/models/model.py +++ b/api/models/model.py @@ -29,7 +29,7 @@ from libs.helper import generate_string from .account import Account, Tenant from .base import Base from .engine import db -from .enums import CreatorUserRole, MessageStatus +from .enums import CreatorUserRole from .types import StringUUID if TYPE_CHECKING: diff --git a/api/services/audio_service.py b/api/services/audio_service.py index 5c9c5ea08b..e8923eb51b 100644 --- a/api/services/audio_service.py +++ b/api/services/audio_service.py @@ -11,8 +11,8 @@ from constants import AUDIO_EXTENSIONS from core.model_manager import ModelManager from core.model_runtime.entities.model_entities import ModelType from extensions.ext_database import db -from models.model import App, AppMode, AppModelConfig, Message from models.enums import MessageStatus +from models.model import App, AppMode, AppModelConfig, Message from services.errors.audio import ( AudioTooLargeServiceError, NoAudioUploadedServiceError,