feat/enhance the multi-modal support (#8818)

api/models/__init__.py

@@ -1,29 +1,55 @@
-from enum import Enum
+from .account import Account, AccountIntegrate, InvitationCode, Tenant
+from .dataset import Dataset, DatasetProcessRule, Document, DocumentSegment
+from .model import (
+    ApiToken,
+    App,
+    AppMode,
+    Conversation,
+    EndUser,
+    FileUploadConfig,
+    InstalledApp,
+    Message,
+    MessageAnnotation,
+    MessageFile,
+    RecommendedApp,
+    Site,
+    UploadFile,
+)
+from .source import DataSourceOauthBinding
+from .tools import ToolFile
+from .workflow import (
+    ConversationVariable,
+    Workflow,
+    WorkflowAppLog,
+    WorkflowRun,
+)
 
-from .model import App, AppMode, Message
-from .types import StringUUID
-from .workflow import ConversationVariable, Workflow, WorkflowNodeExecutionStatus
-
-__all__ = ["ConversationVariable", "StringUUID", "AppMode", "WorkflowNodeExecutionStatus", "Workflow", "App", "Message"]
-
-
-class CreatedByRole(Enum):
-    """
-    Enum class for createdByRole
-    """
-
-    ACCOUNT = "account"
-    END_USER = "end_user"
-
-    @classmethod
-    def value_of(cls, value: str) -> "CreatedByRole":
-        """
-        Get value of given mode.
-
-        :param value: mode value
-        :return: mode
-        """
-        for role in cls:
-            if role.value == value:
-                return role
-        raise ValueError(f"invalid createdByRole value {value}")
+__all__ = [
+    "ConversationVariable",
+    "Document",
+    "Dataset",
+    "DatasetProcessRule",
+    "DocumentSegment",
+    "DataSourceOauthBinding",
+    "AppMode",
+    "Workflow",
+    "App",
+    "Message",
+    "EndUser",
+    "MessageFile",
+    "UploadFile",
+    "Account",
+    "WorkflowAppLog",
+    "WorkflowRun",
+    "Site",
+    "InstalledApp",
+    "RecommendedApp",
+    "ApiToken",
+    "AccountIntegrate",
+    "InvitationCode",
+    "Tenant",
+    "Conversation",
+    "MessageAnnotation",
+    "FileUploadConfig",
+    "ToolFile",
+]