refactor(models): relocate and update enums usage

- Moved enums to a dedicated 'models.enums' module for better organization.
- Updated all import statements across the project to use the new location.
- Removed unnecessary docstring placeholders from enum classes.

api/controllers/console/app/workflow_statistic.py

@@ -10,10 +10,10 @@
 from controllers.console.app.wraps import get_app_model
 from controllers.console.setup import setup_required
 from controllers.console.wraps import account_initialization_required
-from enums import WorkflowRunTriggeredFrom
 from extensions.ext_database import db
 from libs.helper import DatetimeString
 from libs.login import login_required
+from models.enums import WorkflowRunTriggeredFrom
 from models.model import AppMode
 
 

api/core/app/apps/advanced_chat/app_generator.py

@@ -23,10 +23,10 @@
 from core.app.entities.task_entities import ChatbotAppBlockingResponse, ChatbotAppStreamResponse
 from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
 from core.ops.ops_trace_manager import TraceQueueManager
-from enums import CreatedByRole
 from extensions.ext_database import db
 from factories import file_factory
 from models.account import Account
+from models.enums import CreatedByRole
 from models.model import App, Conversation, EndUser, Message
 from models.workflow import Workflow
 

api/core/app/apps/advanced_chat/app_runner.py

@@ -20,8 +20,8 @@
 from core.workflow.entities.variable_pool import VariablePool
 from core.workflow.enums import SystemVariableKey
 from core.workflow.workflow_entry import WorkflowEntry
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.model import App, Conversation, EndUser, Message
 from models.workflow import ConversationVariable, WorkflowType
 

api/core/app/apps/advanced_chat/generate_task_pipeline.py

@@ -52,11 +52,11 @@
 from core.workflow.enums import SystemVariableKey
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes import NodeType
-from enums import CreatedByRole
 from events.message_event import message_was_created
 from extensions.ext_database import db
 from models import Conversation, EndUser, Message, MessageFile
 from models.account import Account
+from models.enums import CreatedByRole
 from models.workflow import (
     Workflow,
     WorkflowNodeExecution,

api/core/app/apps/agent_chat/app_generator.py

@@ -20,10 +20,10 @@
 from core.app.entities.app_invoke_entities import AgentChatAppGenerateEntity, InvokeFrom
 from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
 from core.ops.ops_trace_manager import TraceQueueManager
-from enums import CreatedByRole
 from extensions.ext_database import db
 from factories import file_factory
 from models import Account, App, EndUser
+from models.enums import CreatedByRole
 
 logger = logging.getLogger(__name__)
 

api/core/app/apps/base_app_generator.py

@@ -7,7 +7,7 @@
 
 if TYPE_CHECKING:
     from core.app.app_config.entities import AppConfig, VariableEntity
-    from enums import CreatedByRole
+    from models.enums import CreatedByRole
 
 
 class BaseAppGenerator:

api/core/app/apps/chat/app_generator.py

@@ -20,10 +20,10 @@
 from core.app.entities.app_invoke_entities import ChatAppGenerateEntity, InvokeFrom
 from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
 from core.ops.ops_trace_manager import TraceQueueManager
-from enums import CreatedByRole
 from extensions.ext_database import db
 from factories import file_factory
 from models.account import Account
+from models.enums import CreatedByRole
 from models.model import App, EndUser
 
 logger = logging.getLogger(__name__)

api/core/app/apps/completion/app_generator.py

@@ -19,10 +19,10 @@
 from core.app.entities.app_invoke_entities import CompletionAppGenerateEntity, InvokeFrom
 from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
 from core.ops.ops_trace_manager import TraceQueueManager
-from enums import CreatedByRole
 from extensions.ext_database import db
 from factories import file_factory
 from models import Account, App, EndUser, Message
+from models.enums import CreatedByRole
 from services.errors.app import MoreLikeThisDisabledError
 from services.errors.message import MessageNotExistsError
 

api/core/app/apps/workflow/app_generator.py

@@ -22,10 +22,10 @@
 from core.app.entities.task_entities import WorkflowAppBlockingResponse, WorkflowAppStreamResponse
 from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
 from core.ops.ops_trace_manager import TraceQueueManager
-from enums import CreatedByRole
 from extensions.ext_database import db
 from factories import file_factory
 from models import Account, App, EndUser, Workflow
+from models.enums import CreatedByRole
 
 logger = logging.getLogger(__name__)
 

api/core/app/apps/workflow/app_runner.py

@@ -13,8 +13,8 @@
 from core.workflow.entities.variable_pool import VariablePool
 from core.workflow.enums import SystemVariableKey
 from core.workflow.workflow_entry import WorkflowEntry
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.model import App, EndUser
 from models.workflow import WorkflowType
 

api/core/app/task_pipeline/workflow_cycle_manage.py

@@ -37,9 +37,9 @@
 from core.workflow.nodes import NodeType
 from core.workflow.nodes.tool.entities import ToolNodeData
 from core.workflow.workflow_entry import WorkflowEntry
-from enums import CreatedByRole, WorkflowRunTriggeredFrom
 from extensions.ext_database import db
 from models.account import Account
+from models.enums import CreatedByRole, WorkflowRunTriggeredFrom
 from models.model import EndUser
 from models.workflow import (
     Workflow,

api/core/tools/tool_engine.py

@@ -26,8 +26,8 @@
 from core.tools.tool.tool import Tool
 from core.tools.tool.workflow_tool import WorkflowTool
 from core.tools.utils.message_transformer import ToolFileMessageTransformer
-from enums import CreatedByRole
 from extensions.ext_database import db
+from models.enums import CreatedByRole
 from models.model import Message, MessageFile
 
 

api/core/workflow/entities/workflow_entities.py

@@ -4,7 +4,7 @@
 
 from core.app.entities.app_invoke_entities import InvokeFrom
 from core.workflow.nodes.base import BaseIterationState, BaseNode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import Workflow, WorkflowType
 
 from .node_entities import NodeRunResult

api/core/workflow/graph_engine/entities/graph_init_params.py

@@ -4,7 +4,7 @@
 from pydantic import BaseModel, Field
 
 from core.app.entities.app_invoke_entities import InvokeFrom
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowType
 
 

api/core/workflow/graph_engine/graph_engine.py

@@ -38,8 +38,8 @@
 from core.workflow.nodes.end.end_stream_processor import EndStreamProcessor
 from core.workflow.nodes.event import RunCompletedEvent, RunRetrieverResourceEvent, RunStreamChunkEvent
 from core.workflow.nodes.node_mapping import node_type_classes_mapping
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 logger = logging.getLogger(__name__)

api/core/workflow/workflow_entry.py

@@ -22,7 +22,7 @@
 from core.workflow.nodes.event import NodeEvent
 from core.workflow.nodes.llm import LLMNodeData
 from core.workflow.nodes.node_mapping import node_type_classes_mapping
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import (
     Workflow,
     WorkflowType,

api/factories/file_factory.py

@@ -7,9 +7,9 @@
 from constants import AUDIO_EXTENSIONS, DOCUMENT_EXTENSIONS, IMAGE_EXTENSIONS, VIDEO_EXTENSIONS
 from core.file import File, FileBelongsTo, FileExtraConfig, FileTransferMethod, FileType
 from core.helper import ssrf_proxy
-from enums import CreatedByRole
 from extensions.ext_database import db
 from models import MessageFile, ToolFile, UploadFile
+from models.enums import CreatedByRole
 
 
 def build_from_message_files(

api/enums/__init__.py → api/models/enums.py

@@ -2,30 +2,15 @@
 
 
 class CreatedByRole(str, Enum):
-    """
-    TODO: Need to write docstring
-    """
-
     ACCOUNT = "account"
     END_USER = "end_user"
 
 
 class UserFrom(str, Enum):
-    """
-    TODO: Need to write docstring
-    """
-
     ACCOUNT = "account"
     END_USER = "end-user"
 
 
 class WorkflowRunTriggeredFrom(str, Enum):
     DEBUGGING = "debugging"
     APP_RUN = "app-run"
-
-
-__all__ = [
-    "CreatedByRole",
-    "UserFrom",
-    "WorkflowRunTriggeredFrom",
-]

api/models/model.py

@@ -16,9 +16,9 @@
 from core.file import FILE_MODEL_IDENTITY, File, FileExtraConfig, FileTransferMethod, FileType
 from core.file import helpers as file_helpers
 from core.file.tool_file_parser import ToolFileParser
-from enums import CreatedByRole
 from extensions.ext_database import db
 from libs.helper import generate_string
+from models.enums import CreatedByRole
 
 from .account import Account, Tenant
 from .types import StringUUID

api/models/workflow.py

@@ -11,10 +11,10 @@
 from constants import HIDDEN_VALUE
 from core.helper import encrypter
 from core.variables import SecretVariable, Variable
-from enums import CreatedByRole
 from extensions.ext_database import db
 from factories import variable_factory
 from libs import helper
+from models.enums import CreatedByRole
 
 from .account import Account
 from .types import StringUUID

api/services/workflow_app_service.py

@@ -3,9 +3,9 @@
 from flask_sqlalchemy.pagination import Pagination
 from sqlalchemy import and_, or_
 
-from enums import CreatedByRole
 from extensions.ext_database import db
 from models import App, EndUser, WorkflowAppLog, WorkflowRun
+from models.enums import CreatedByRole
 from models.workflow import WorkflowRunStatus
 
 

api/services/workflow_run_service.py

@@ -1,6 +1,6 @@
-from enums import WorkflowRunTriggeredFrom
 from extensions.ext_database import db
 from libs.infinite_scroll_pagination import InfiniteScrollPagination
+from models.enums import WorkflowRunTriggeredFrom
 from models.model import App
 from models.workflow import (
     WorkflowNodeExecution,

api/services/workflow_service.py

@@ -14,10 +14,10 @@
 from core.workflow.nodes.event import RunCompletedEvent
 from core.workflow.nodes.node_mapping import node_type_classes_mapping
 from core.workflow.workflow_entry import WorkflowEntry
-from enums import CreatedByRole
 from events.app_event import app_draft_workflow_was_synced, app_published_workflow_was_updated
 from extensions.ext_database import db
 from models.account import Account
+from models.enums import CreatedByRole
 from models.model import App, AppMode
 from models.workflow import (
     Workflow,

api/tests/integration_tests/workflow/nodes/test_code.py

@@ -14,7 +14,7 @@
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.code.code_node import CodeNode
 from core.workflow.nodes.code.entities import CodeNodeData
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 from tests.integration_tests.workflow.nodes.__mock.code_executor import setup_code_executor_mock
 

api/tests/integration_tests/workflow/nodes/test_http.py

@@ -11,7 +11,7 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.http_request.node import HttpRequestNode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowType
 from tests.integration_tests.workflow.nodes.__mock.http import setup_http_mock
 

api/tests/integration_tests/workflow/nodes/test_llm.py

@@ -20,8 +20,8 @@
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.event import RunCompletedEvent
 from core.workflow.nodes.llm.node import LLMNode
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.provider import ProviderType
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 

api/tests/integration_tests/workflow/nodes/test_parameter_extractor.py

@@ -18,8 +18,8 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.parameter_extractor.parameter_extractor_node import ParameterExtractorNode
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.provider import ProviderType
 
 """FOR MOCK FIXTURES, DO NOT REMOVE"""

api/tests/integration_tests/workflow/nodes/test_template_transform.py

@@ -10,7 +10,7 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.template_transform.template_transform_node import TemplateTransformNode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 from tests.integration_tests.workflow.nodes.__mock.code_executor import setup_code_executor_mock
 

api/tests/integration_tests/workflow/nodes/test_tool.py

@@ -9,7 +9,7 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.tool.tool_node import ToolNode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py

@@ -19,7 +19,7 @@
 from core.workflow.graph_engine.graph_engine import GraphEngine
 from core.workflow.nodes.event import RunCompletedEvent, RunStreamChunkEvent
 from core.workflow.nodes.llm.node import LLMNode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/nodes/answer/test_answer.py

@@ -9,8 +9,8 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.answer.answer_node import AnswerNode
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/nodes/iteration/test_iteration.py

@@ -12,7 +12,7 @@
 from core.workflow.nodes.event import RunCompletedEvent
 from core.workflow.nodes.iteration.iteration_node import IterationNode
 from core.workflow.nodes.template_transform.template_transform_node import TemplateTransformNode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/nodes/test_answer.py

@@ -9,8 +9,8 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.answer.answer_node import AnswerNode
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/nodes/test_http_request_node.py

@@ -15,7 +15,7 @@
     HttpRequestNodeData,
 )
 from core.workflow.nodes.http_request.executor import _plain_text_to_dict
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/nodes/test_if_else.py

@@ -13,8 +13,8 @@
 from core.workflow.nodes.if_else.entities import IfElseNodeData
 from core.workflow.nodes.if_else.if_else_node import IfElseNode
 from core.workflow.utils.condition.entities import Condition, SubCondition, SubVariableCondition
-from enums import UserFrom
 from extensions.ext_database import db
+from models.enums import UserFrom
 from models.workflow import WorkflowNodeExecutionStatus, WorkflowType
 
 

api/tests/unit_tests/core/workflow/nodes/test_variable_assigner.py

@@ -11,7 +11,7 @@
 from core.workflow.graph_engine.entities.graph_init_params import GraphInitParams
 from core.workflow.graph_engine.entities.graph_runtime_state import GraphRuntimeState
 from core.workflow.nodes.variable_assigner import VariableAssignerNode, WriteMode
-from enums import UserFrom
+from models.enums import UserFrom
 from models.workflow import WorkflowType
 
 DEFAULT_NODE_ID = "node_id"