dify/api/models/__init__.py

from .account import (
    Account,
    AccountIntegrate,
    AccountStatus,
    InvitationCode,
    Tenant,
    TenantAccountJoin,
    TenantAccountRole,
    TenantStatus,
)
from .agent import (
    Agent,
    AgentConfigDraft,
    AgentConfigDraftType,
    AgentConfigRevision,
    AgentConfigRevisionOperation,
    AgentConfigSnapshot,
    AgentConfigVersionKind,
    AgentDebugConversation,
    AgentHomeSnapshot,
    AgentIconType,
    AgentKind,
    AgentScope,
    AgentSource,
    AgentStatus,
    AgentWorkingResourceStatus,
    AgentWorkspace,
    AgentWorkspaceBinding,
    AgentWorkspaceOwnerType,
    WorkflowAgentBindingType,
    WorkflowAgentNodeBinding,
)
from .agent_sandbox_usage import AgentSandboxExecution, AgentSandboxUsageEvent
from .api_based_extension import APIBasedExtension, APIBasedExtensionPoint
from .application_trace import ApplicationTraceBatch, ApplicationTraceInvocation, ApplicationTraceRun
from .comment import (
    WorkflowComment,
    WorkflowCommentMention,
    WorkflowCommentReply,
)
from .credential_permission import CredentialPermission
from .credential_permission import CredentialType as CredentialPermissionType
from .dataset import (
    AppDatasetJoin,
    Dataset,
    DatasetCollectionBinding,
    DatasetKeywordTable,
    DatasetPermission,
    DatasetPermissionEnum,
    DatasetProcessRule,
    DatasetQuery,
    Document,
    DocumentSegment,
    Embedding,
    ExternalKnowledgeApis,
    ExternalKnowledgeBindings,
    TidbAuthBinding,
    Whitelist,
)
from .enums import (
    AppTriggerStatus,
    AppTriggerType,
    CreatorUserRole,
    PermissionEnum,
    WorkflowRunTriggeredFrom,
    WorkflowTriggerStatus,
)
from .execution_extra_content import ExecutionExtraContent, HumanInputContent
from .human_input import HumanInputForm, HumanInputFormUploadFile, HumanInputFormUploadToken
from .knowledge_fs import (
    AppKnowledgeFSSpaceJoin,
    KnowledgeFSApiCredential,
    KnowledgeFSApiCredentialStatus,
    KnowledgeFSAppSpaceJoinStatus,
    KnowledgeFSAppSpaceJoinType,
    KnowledgeFSAuthorizationRevision,
    KnowledgeFSCapabilityIssuanceAudit,
    KnowledgeFSCapabilityIssuanceReservation,
    KnowledgeFSCapabilityIssuanceReservationStatus,
    KnowledgeFSControlSpace,
    KnowledgeFSControlSpacePermission,
    KnowledgeFSControlSpacePermissionRole,
    KnowledgeFSControlSpacePermissionStatus,
    KnowledgeFSControlSpaceState,
    KnowledgeFSControlSpaceVisibility,
    KnowledgeFSExternalAccessPolicy,
    KnowledgeFSLifecycleOperation,
    KnowledgeFSLifecycleOutbox,
    KnowledgeFSLifecycleOutboxStatus,
    KnowledgeFSSpaceTagBinding,
    KnowledgeFSStagedUpload,
    KnowledgeFSStagedUploadStatus,
    KnowledgeFSUpgradeDocument,
    KnowledgeFSUpgradeFileLease,
    KnowledgeFSUpgradeFileLeaseStatus,
    KnowledgeFSUpgradeItemStatus,
    KnowledgeFSUpgradeJob,
    KnowledgeFSUpgradeJobStatus,
    KnowledgeFSUpgradeSource,
    KnowledgeFSUpgradeStage,
)
from .knowledge_fs_cleanup import (
    KnowledgeFSCleanupAuthorization,
    KnowledgeFSCleanupAuthorizationStatus,
    KnowledgeFSCleanupTarget,
)
from .knowledge_fs_cutover import (
    KnowledgeFSMigrationIssue,
    KnowledgeFSMigrationIssueKind,
    KnowledgeFSMigrationIssueStatus,
    KnowledgeFSMigrationQuarantine,
    KnowledgeFSMigrationQuarantineDisposition,
    KnowledgeFSMigrationQuarantineKind,
    KnowledgeFSShadowAuthorizationDecision,
    KnowledgeFSShadowAuthorizationDiff,
    KnowledgeFSShadowAuthorizationObservation,
    KnowledgeFSWorkspaceCutoverLedger,
    KnowledgeFSWorkspaceCutoverPhase,
)
from .model import (
    AccountTrialAppRecord,
    ApiRequest,
    ApiToken,
    App,
    AppAnnotationHitHistory,
    AppAnnotationSetting,
    AppMCPServer,
    AppMode,
    AppModelConfig,
    AppStar,
    Conversation,
    DatasetApiTokenBinding,
    DatasetRetrieverResource,
    DifySetup,
    EndUser,
    ExporleBanner,
    IconType,
    InstalledApp,
    Message,
    MessageAgentThought,
    MessageAnnotation,
    MessageChain,
    MessageFeedback,
    MessageFile,
    OperationLog,
    RecommendedApp,
    Site,
    Tag,
    TagBinding,
    TenantCreditPool,
    TraceAppConfig,
    TrialApp,
    UploadFile,
)
from .model_billing import TenantModelBillingProfile
from .model_billing_migration import TenantModelBillingMigration
from .oauth import DatasourceOauthParamConfig, DatasourceProvider, OAuthAccessToken
from .onboarding import AccountStepByStepTourState
from .provider import (
    LoadBalancingModelConfig,
    Provider,
    ProviderModel,
    ProviderModelSetting,
    ProviderOrder,
    ProviderQuotaType,
    ProviderType,
    TenantDefaultModel,
    TenantPreferredModelProvider,
)
from .resource_access_token import (
    ResourceAccessToken,
    ResourceAccessTokenRelation,
    ResourceAccessTokenResourceType,
)
from .skill import AgentSkillBinding, Skill, SkillDraftFile, SkillFileKind, SkillFileStorage, SkillVersion
from .snippet import CustomizedSnippet, SnippetType
from .source import DataSourceApiKeyAuthBinding, DataSourceOauthBinding
from .task import CeleryTask, CeleryTaskSet
from .tokener import TenantTokenerIntegration, TenantTokenerIntegrationStatus
from .tools import (
    ApiToolProvider,
    BuiltinToolProvider,
    ToolConversationVariables,
    ToolFile,
    ToolLabelBinding,
    ToolModelInvoke,
    WorkflowToolProvider,
)
from .trigger import (
    AppTrigger,
    TriggerOAuthSystemClient,
    TriggerOAuthTenantClient,
    TriggerSubscription,
    WorkflowSchedulePlan,
)
from .web import PinnedConversation, SavedMessage
from .workflow import (
    ConversationVariable,
    Workflow,
    WorkflowAppLog,
    WorkflowAppLogCreatedFrom,
    WorkflowArchiveLog,
    WorkflowKind,
    WorkflowNodeExecutionModel,
    WorkflowNodeExecutionOffload,
    WorkflowNodeExecutionTriggeredFrom,
    WorkflowPause,
    WorkflowRun,
    WorkflowRunArchiveBundle,
    WorkflowType,
    WorkflowVersionCounter,
    resolve_workflow_kind,
)

__all__ = [
    "APIBasedExtension",
    "APIBasedExtensionPoint",
    "Account",
    "AccountIntegrate",
    "AccountStatus",
    "AccountStepByStepTourState",
    "AccountTrialAppRecord",
    "Agent",
    "AgentConfigDraft",
    "AgentConfigDraftType",
    "AgentConfigRevision",
    "AgentConfigRevisionOperation",
    "AgentConfigSnapshot",
    "AgentConfigVersionKind",
    "AgentDebugConversation",
    "AgentHomeSnapshot",
    "AgentIconType",
    "AgentKind",
    "AgentSandboxExecution",
    "AgentSandboxUsageEvent",
    "AgentScope",
    "AgentSkillBinding",
    "AgentSource",
    "AgentStatus",
    "AgentWorkingResourceStatus",
    "AgentWorkspace",
    "AgentWorkspaceBinding",
    "AgentWorkspaceOwnerType",
    "ApiRequest",
    "ApiToken",
    "ApiToolProvider",
    "App",
    "AppAnnotationHitHistory",
    "AppAnnotationSetting",
    "AppDatasetJoin",
    "AppKnowledgeFSSpaceJoin",
    "AppMCPServer",
    "AppMode",
    "AppModelConfig",
    "AppStar",
    "AppTrigger",
    "AppTriggerStatus",
    "AppTriggerType",
    "ApplicationTraceBatch",
    "ApplicationTraceInvocation",
    "ApplicationTraceRun",
    "BuiltinToolProvider",
    "CeleryTask",
    "CeleryTaskSet",
    "Conversation",
    "ConversationVariable",
    "CreatorUserRole",
    "CredentialPermission",
    "CredentialPermissionType",
    "CustomizedSnippet",
    "DataSourceApiKeyAuthBinding",
    "DataSourceOauthBinding",
    "Dataset",
    "DatasetApiTokenBinding",
    "DatasetCollectionBinding",
    "DatasetKeywordTable",
    "DatasetPermission",
    "DatasetPermissionEnum",
    "DatasetProcessRule",
    "DatasetQuery",
    "DatasetRetrieverResource",
    "DatasourceOauthParamConfig",
    "DatasourceProvider",
    "DifySetup",
    "Document",
    "DocumentSegment",
    "Embedding",
    "EndUser",
    "ExecutionExtraContent",
    "ExporleBanner",
    "ExternalKnowledgeApis",
    "ExternalKnowledgeBindings",
    "HumanInputContent",
    "HumanInputForm",
    "HumanInputFormUploadFile",
    "HumanInputFormUploadToken",
    "IconType",
    "InstalledApp",
    "InvitationCode",
    "KnowledgeFSApiCredential",
    "KnowledgeFSApiCredentialStatus",
    "KnowledgeFSAppSpaceJoinStatus",
    "KnowledgeFSAppSpaceJoinType",
    "KnowledgeFSAuthorizationRevision",
    "KnowledgeFSCapabilityIssuanceAudit",
    "KnowledgeFSCapabilityIssuanceReservation",
    "KnowledgeFSCapabilityIssuanceReservationStatus",
    "KnowledgeFSCleanupAuthorization",
    "KnowledgeFSCleanupAuthorizationStatus",
    "KnowledgeFSCleanupTarget",
    "KnowledgeFSControlSpace",
    "KnowledgeFSControlSpacePermission",
    "KnowledgeFSControlSpacePermissionRole",
    "KnowledgeFSControlSpacePermissionStatus",
    "KnowledgeFSControlSpaceState",
    "KnowledgeFSControlSpaceVisibility",
    "KnowledgeFSExternalAccessPolicy",
    "KnowledgeFSLifecycleOperation",
    "KnowledgeFSLifecycleOutbox",
    "KnowledgeFSLifecycleOutboxStatus",
    "KnowledgeFSMigrationIssue",
    "KnowledgeFSMigrationIssueKind",
    "KnowledgeFSMigrationIssueStatus",
    "KnowledgeFSMigrationQuarantine",
    "KnowledgeFSMigrationQuarantineDisposition",
    "KnowledgeFSMigrationQuarantineKind",
    "KnowledgeFSShadowAuthorizationDecision",
    "KnowledgeFSShadowAuthorizationDiff",
    "KnowledgeFSShadowAuthorizationObservation",
    "KnowledgeFSSpaceTagBinding",
    "KnowledgeFSStagedUpload",
    "KnowledgeFSStagedUploadStatus",
    "KnowledgeFSUpgradeDocument",
    "KnowledgeFSUpgradeFileLease",
    "KnowledgeFSUpgradeFileLeaseStatus",
    "KnowledgeFSUpgradeItemStatus",
    "KnowledgeFSUpgradeJob",
    "KnowledgeFSUpgradeJobStatus",
    "KnowledgeFSUpgradeSource",
    "KnowledgeFSUpgradeStage",
    "KnowledgeFSWorkspaceCutoverLedger",
    "KnowledgeFSWorkspaceCutoverPhase",
    "LoadBalancingModelConfig",
    "Message",
    "MessageAgentThought",
    "MessageAnnotation",
    "MessageChain",
    "MessageFeedback",
    "MessageFile",
    "OAuthAccessToken",
    "OperationLog",
    "PermissionEnum",
    "PinnedConversation",
    "Provider",
    "ProviderModel",
    "ProviderModelSetting",
    "ProviderOrder",
    "ProviderQuotaType",
    "ProviderType",
    "RecommendedApp",
    "ResourceAccessToken",
    "ResourceAccessTokenRelation",
    "ResourceAccessTokenResourceType",
    "SavedMessage",
    "Site",
    "Skill",
    "SkillDraftFile",
    "SkillFileKind",
    "SkillFileStorage",
    "SkillVersion",
    "SnippetType",
    "Tag",
    "TagBinding",
    "Tenant",
    "TenantAccountJoin",
    "TenantAccountRole",
    "TenantCreditPool",
    "TenantDefaultModel",
    "TenantModelBillingMigration",
    "TenantModelBillingProfile",
    "TenantPreferredModelProvider",
    "TenantStatus",
    "TenantTokenerIntegration",
    "TenantTokenerIntegrationStatus",
    "TidbAuthBinding",
    "ToolConversationVariables",
    "ToolFile",
    "ToolLabelBinding",
    "ToolModelInvoke",
    "TraceAppConfig",
    "TrialApp",
    "TriggerOAuthSystemClient",
    "TriggerOAuthTenantClient",
    "TriggerSubscription",
    "UploadFile",
    "Whitelist",
    "Workflow",
    "WorkflowAgentBindingType",
    "WorkflowAgentNodeBinding",
    "WorkflowAppLog",
    "WorkflowAppLogCreatedFrom",
    "WorkflowArchiveLog",
    "WorkflowComment",
    "WorkflowCommentMention",
    "WorkflowCommentReply",
    "WorkflowKind",
    "WorkflowNodeExecutionModel",
    "WorkflowNodeExecutionOffload",
    "WorkflowNodeExecutionTriggeredFrom",
    "WorkflowPause",
    "WorkflowRun",
    "WorkflowRunArchiveBundle",
    "WorkflowRunTriggeredFrom",
    "WorkflowSchedulePlan",
    "WorkflowToolProvider",
    "WorkflowTriggerStatus",
    "WorkflowType",
    "WorkflowVersionCounter",
    "resolve_workflow_kind",
]