from lightning_sdk.__version__ import __version__
from lightning_sdk.agents import Agent
from lightning_sdk.api.billing_api import (
    ActivityFileFormat,
    BillingActivity,
    BillingActivityFilterValues,
    BillingDailyUsage,
    BillingNamedFilterValue,
    BillingResourceUsage,
)
from lightning_sdk.api.teamspace_api import SecretType
from lightning_sdk.api.utils import FileEntry
from lightning_sdk.constants import __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__  # noqa: F401
from lightning_sdk.data_connection import BucketCredentials
from lightning_sdk.deployment import Deployment
from lightning_sdk.helpers import VersionChecker, set_tqdm_envvars_noninteractive
from lightning_sdk.job import Job
from lightning_sdk.k8s_cluster import K8sCluster
from lightning_sdk.machine import CloudProvider, Machine
from lightning_sdk.mmt import MMT
from lightning_sdk.organization import BillingActivityCursor, BillingActivityFilters, MonthlySummary, Organization
from lightning_sdk.status import Status
from lightning_sdk.studio import Studio
from lightning_sdk.teamspace import ConnectionType, FolderLocation, Teamspace
from lightning_sdk.user import User
from lightning_sdk.vm import VM

__all__ = [
    "MMT",
    "ActivityFileFormat",
    "Agent",
    "BillingActivity",
    "BillingActivityCursor",
    "BillingActivityFilterValues",
    "BillingActivityFilters",
    "BillingDailyUsage",
    "BillingNamedFilterValue",
    "BillingResourceUsage",
    "BucketCredentials",
    "CloudProvider",
    "ConnectionType",
    "Deployment",
    "FileEntry",
    "FolderLocation",
    "Job",
    "K8sCluster",
    "Machine",
    "MonthlySummary",
    "Organization",
    "SecretType",
    "Status",
    "Studio",
    "Teamspace",
    "User",
    "VM",
    "__version__",
]

_version_checker = VersionChecker()
_version_checker.check_and_prompt_upgrade(__version__)

set_tqdm_envvars_noninteractive()