dify/api/models/__init__.py

from .account import (
    Account,
    AccountIntegrate,
    AccountStatus,
    InvitationCode,
    Tenant,
    TenantAccountJoin,
    TenantAccountRole,
    TenantStatus,
)
from .api_based_extension import APIBasedExtension, APIBasedExtensionPoint
from .dataset import (
    AppDatasetJoin,
    Dataset,
    DatasetCollectionBinding,
    DatasetKeywordTable,
    DatasetPermission,
    DatasetPermissionEnum,
    DatasetProcessRule,
    DatasetQuery,
    Document,
    DocumentSegment,
    Embedding,
    ExternalKnowledgeApis,
    ExternalKnowledgeBindings,
    TidbAuthBinding,
    Whitelist,
)
from .enums import CreatorUserRole, UserFrom, WorkflowRunTriggeredFrom
from .model import (
    ApiRequest,
    ApiToken,
    App,
    AppAnnotationHitHistory,
    AppAnnotationSetting,
    AppMCPServer,
    AppMode,
    AppModelConfig,
    Conversation,
    DatasetRetrieverResource,
    DifySetup,
    EndUser,
    IconType,
    InstalledApp,
    Message,
    MessageAgentThought,
    MessageAnnotation,
    MessageChain,
    MessageFeedback,
    MessageFile,
    OperationLog,
    RecommendedApp,
    Site,
    Tag,
    TagBinding,
    TraceAppConfig,
    UploadFile,
)
from .provider import (
    LoadBalancingModelConfig,
    Provider,
    ProviderModel,
    ProviderModelSetting,
    ProviderOrder,
    ProviderQuotaType,
    ProviderType,
    TenantDefaultModel,
    TenantPreferredModelProvider,
)
from .source import DataSourceApiKeyAuthBinding, DataSourceOauthBinding
from .task import CeleryTask, CeleryTaskSet
from .tools import (
    ApiToolProvider,
    BuiltinToolProvider,
    ToolConversationVariables,
    ToolFile,
    ToolLabelBinding,
    ToolModelInvoke,
    WorkflowToolProvider,
)
from .web import PinnedConversation, SavedMessage
from .workflow import (
    ConversationVariable,
    Workflow,
    WorkflowAppLog,
    WorkflowAppLogCreatedFrom,
    WorkflowNodeExecutionModel,
    WorkflowNodeExecutionTriggeredFrom,
    WorkflowRun,
    WorkflowType,
)

__all__ = [
    "APIBasedExtension",
    "APIBasedExtensionPoint",
    "Account",
    "AccountIntegrate",
    "AccountStatus",
    "ApiRequest",
    "ApiToken",
    "ApiToolProvider",
    "App",
    "AppAnnotationHitHistory",
    "AppAnnotationSetting",
    "AppDatasetJoin",
    "AppMCPServer",  # Added
    "AppMode",
    "AppModelConfig",
    "BuiltinToolProvider",
    "CeleryTask",
    "CeleryTaskSet",
    "Conversation",
    "ConversationVariable",
    "CreatorUserRole",
    "DataSourceApiKeyAuthBinding",
    "DataSourceOauthBinding",
    "Dataset",
    "DatasetCollectionBinding",
    "DatasetKeywordTable",
    "DatasetPermission",
    "DatasetPermissionEnum",
    "DatasetProcessRule",
    "DatasetQuery",
    "DatasetRetrieverResource",
    "DifySetup",
    "Document",
    "DocumentSegment",
    "Embedding",
    "EndUser",
    "ExternalKnowledgeApis",
    "ExternalKnowledgeBindings",
    "IconType",
    "InstalledApp",
    "InvitationCode",
    "LoadBalancingModelConfig",
    "Message",
    "MessageAgentThought",
    "MessageAnnotation",
    "MessageChain",
    "MessageFeedback",
    "MessageFile",
    "OperationLog",
    "PinnedConversation",
    "Provider",
    "ProviderModel",
    "ProviderModelSetting",
    "ProviderOrder",
    "ProviderQuotaType",
    "ProviderType",
    "RecommendedApp",
    "SavedMessage",
    "Site",
    "Tag",
    "TagBinding",
    "Tenant",
    "TenantAccountJoin",
    "TenantAccountRole",
    "TenantDefaultModel",
    "TenantPreferredModelProvider",
    "TenantStatus",
    "TidbAuthBinding",
    "ToolConversationVariables",
    "ToolFile",
    "ToolLabelBinding",
    "ToolModelInvoke",
    "TraceAppConfig",
    "UploadFile",
    "UserFrom",
    "Whitelist",
    "Workflow",
    "WorkflowAppLog",
    "WorkflowAppLogCreatedFrom",
    "WorkflowNodeExecutionModel",
    "WorkflowNodeExecutionTriggeredFrom",
    "WorkflowRun",
    "WorkflowRunTriggeredFrom",
    "WorkflowToolProvider",
    "WorkflowType",
]