yujunjun/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2025-09-04 06:43:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add message status enum (#21825)

Browse Source 
Co-authored-by: 刘江波 <jiangbo721@163.com>
This commit is contained in:
jiangbo721 2025-07-02 21:22:28 +08:00 committed by GitHub
parent 4e7c9dd2ae
commit e17b33e004
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/app/task_pipeline/based_generate_task_pipeline.py
View File

@ -19,7 +19,7 @@ from core.app.entities.task_entities import (
from core.errors.error import QuotaExceededError from core.errors.error import QuotaExceededError
from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
from core.moderation.output_moderation import ModerationRule, OutputModeration from core.moderation.output_moderation import ModerationRule, OutputModeration
from models.model import Message from models.model import Message, MessageStatus
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -62,7 +62,7 @@ class BasedGenerateTaskPipeline:
return err return err
err_desc = self._error_to_desc(err) err_desc = self._error_to_desc(err)
message.status = "error" message.status = MessageStatus.ERROR
message.error = err_desc message.error = err_desc
return err return err

9
api/models/model.py
View File

@ -892,6 +892,15 @@ class Conversation(Base):
} }
class MessageStatus(StrEnum):
"""
Message Status Enum
"""
NORMAL = "normal"
ERROR = "error"
class Message(Base): class Message(Base):
__tablename__ = "messages" __tablename__ = "messages"
__table_args__ = ( __table_args__ = (

4
api/services/audio_service.py
View File

@ -8,7 +8,7 @@ from werkzeug.datastructures import FileStorage
from constants import AUDIO_EXTENSIONS from constants import AUDIO_EXTENSIONS
from core.model_manager import ModelManager from core.model_manager import ModelManager
from core.model_runtime.entities.model_entities import ModelType from core.model_runtime.entities.model_entities import ModelType
from models.model import App, AppMode, AppModelConfig, Message from models.model import App, AppMode, AppModelConfig, Message, MessageStatus
from services.errors.audio import ( from services.errors.audio import (
AudioTooLargeServiceError, AudioTooLargeServiceError,
NoAudioUploadedServiceError, NoAudioUploadedServiceError,
@ -132,7 +132,7 @@ class AudioService:
message = db.session.query(Message).filter(Message.id == message_id).first() message = db.session.query(Message).filter(Message.id == message_id).first()
if message is None: if message is None:
return None return None
if message.answer == "" and message.status == "normal": if message.answer == "" and message.status == MessageStatus.NORMAL:
return None return None
else: else: