maas-base/gpustack/schemas/__init__.py

from gpustack.schemas.clusters import (
    CloudCredential,
    CloudCredentialCreate,
    CloudCredentialUpdate,
    CloudCredentialPublic,
    CloudCredentialsPublic,
    Cluster,
    ClusterCreate,
    ClusterUpdate,
    ClusterPublic,
    ClustersPublic,
    WorkerPool,
    WorkerPoolUpdate,
    WorkerPoolPublic,
    WorkerPoolsPublic,
)
from gpustack.schemas.inference_backend import (
    InferenceBackendPublic,
    InferenceBackendsPublic,
    InferenceBackendCreate,
    InferenceBackendUpdate,
)
from gpustack.schemas.models import (
    Model,
    ModelCreate,
    ModelUpdate,
    ModelPublic,
    ModelsPublic,
    ModelInstance,
    ModelInstanceCreate,
    ModelInstanceUpdate,
    ModelInstancePublic,
    ModelInstancesPublic,
    ModelInstanceLogOptions,
    ModelInstanceLogRestartEntry,
    ModelInstanceLogWorker,
    ModelInstanceLogWorkerOption,
    ServeLogOptionsResponse,
    ComputedResourceClaim,
)
from gpustack.schemas.model_files import (
    ModelFile,
    ModelFileCreate,
    ModelFileUpdate,
    ModelFilePublic,
    ModelFilesPublic,
)
from gpustack.schemas.workers import (
    Worker,
    WorkerCreate,
    WorkerUpdate,
    WorkerPublic,
    WorkersPublic,
    WorkerStatus,
    UptimeInfo,
    KernelInfo,
    OperatingSystemInfo,
    FileSystemInfo,
    MountPoint,
    GPUDevicesStatus,
    GPUDeviceStatus,
    CPUInfo,
    MemoryInfo,
)
from gpustack.schemas.users import User, UserCreate, UserUpdate, UserPublic, UsersPublic
from gpustack.schemas.api_keys import ApiKey, ApiKeyCreate, ApiKeyPublic, ApiKeysPublic
from gpustack.schemas.system_load import SystemLoad
from gpustack.schemas.model_usage import ModelUsage
from gpustack.schemas.model_usage_details import (
    ModelUsageDetails,
    ModelUsageDetailsArchive,
)
from gpustack.schemas.common import PaginatedList
from gpustack.schemas.benchmark import (
    ModelInstanceRuntimeInfo,
    ModelInstanceSnapshot,
    ModelInstanceSnapshots,
    WorkerSnapshot,
    WorkerSnapshots,
    GPUSnapshot,
    GPUSnapshots,
    BenchmarkSnapshot,
    BenchmarkMetrics,
    Benchmark,
    BenchmarkCreate,
    BenchmarkUpdate,
    BenchmarkPublic,
    BenchmarksPublic,
)
from gpustack.schemas.filesystem import (
    FileExistsResponse,
)
from gpustack.schemas.model_provider import (
    ModelProvider,
    ModelProviderCreate,
    ModelProviderUpdate,
    ModelProviderPublic,
    ModelProvidersPublic,
)
from gpustack.schemas.model_routes import (
    ModelRoute,
    ModelRouteCreate,
    ModelRouteUpdate,
    ModelRoutePublic,
    ModelRoutesPublic,
    ModelRouteTarget,
    ModelRouteTargetCreate,
    ModelRouteTargetUpdate,
    ModelRouteTargetPublic,
    ModelRouteTargetsPublic,
)
from gpustack.schemas.principals import (
    Principal,
    PrincipalMembership,
    PrincipalPublic,
    PrincipalsPublic,
)
from gpustack.schemas.organizations import (
    OrganizationCreate,
    OrganizationUpdate,
    OrganizationPublic,
    OrganizationsPublic,
    OrganizationMembershipPublic,
)
from gpustack.schemas.user_groups import (
    UserGroupCreate,
    UserGroupUpdate,
    UserGroupPublic,
    UserGroupsPublic,
    UserGroupMembershipPublic,
)
from gpustack.schemas.cluster_access import (
    ClusterAccess,
    ClusterAccessPublic,
)
from gpustack.schemas.tenant_quotas import (
    TenantQuota,
    TenantQuotaUpdate,
    TenantQuotaPublic,
    TenantQuotasPublic,
)
from gpustack.schemas.links import ModelRoutePrincipalLink  # noqa: F401


__all__ = [
    "ApiKey",
    "ApiKeyCreate",
    "ApiKeyPublic",
    "ApiKeysPublic",
    "Worker",
    "WorkerCreate",
    "WorkerUpdate",
    "WorkerPublic",
    "WorkersPublic",
    "Model",
    "ModelCreate",
    "ModelUpdate",
    "ModelPublic",
    "ModelsPublic",
    "ModelInstance",
    "ModelInstanceCreate",
    "ModelInstanceUpdate",
    "ModelInstancePublic",
    "ModelInstancesPublic",
    "ModelInstanceLogWorker",
    "ModelInstanceLogWorkerOption",
    "ModelInstanceLogRestartEntry",
    "ModelInstanceLogOptions",
    "ServeLogOptionsResponse",
    "ComputedResourceClaim",
    "User",
    "UserCreate",
    "UserUpdate",
    "UserPublic",
    "UsersPublic",
    "WorkerStatus",
    "UptimeInfo",
    "KernelInfo",
    "OperatingSystemInfo",
    "FileSystemInfo",
    "PaginatedList",
    "MountPoint",
    "GPUDevicesStatus",
    "GPUDeviceStatus",
    "CPUInfo",
    "MemoryInfo",
    "SystemLoad",
    "ModelUsage",
    "ModelFile",
    "ModelFileCreate",
    "ModelFileUpdate",
    "ModelFilePublic",
    "ModelFilesPublic",
    "CloudCredential",
    "CloudCredentialCreate",
    "CloudCredentialUpdate",
    "CloudCredentialPublic",
    "CloudCredentialsPublic",
    "Cluster",
    "ClusterCreate",
    "ClusterUpdate",
    "ClusterPublic",
    "ClustersPublic",
    "WorkerPool",
    "WorkerPoolUpdate",
    "WorkerPoolPublic",
    "WorkerPoolsPublic",
    "InferenceBackendPublic",
    "InferenceBackendsPublic",
    "InferenceBackendCreate",
    "InferenceBackendUpdate",
    "Benchmark",
    "BenchmarkCreate",
    "BenchmarkUpdate",
    "BenchmarkPublic",
    "BenchmarksPublic",
    "ModelInstanceRuntimeInfo",
    "ModelInstanceSnapshot",
    "ModelInstanceSnapshots",
    "WorkerSnapshot",
    "WorkerSnapshots",
    "GPUSnapshot",
    "GPUSnapshots",
    "BenchmarkSnapshot",
    "BenchmarkMetrics",
    "FileExistsResponse",
    "ModelProvider",
    "ModelProviderCreate",
    "ModelProviderUpdate",
    "ModelProviderPublic",
    "ModelProvidersPublic",
    "ModelRoute",
    "ModelRouteCreate",
    "ModelRouteUpdate",
    "ModelRoutePublic",
    "ModelRoutesPublic",
    "ModelRouteTarget",
    "ModelRouteTargetCreate",
    "ModelRouteTargetUpdate",
    "ModelRouteTargetPublic",
    "ModelRouteTargetsPublic",
    "ModelUsageDetails",
    "ModelUsageDetailsArchive",
    "Principal",
    "PrincipalMembership",
    "PrincipalPublic",
    "PrincipalsPublic",
    "OrganizationCreate",
    "OrganizationUpdate",
    "OrganizationPublic",
    "OrganizationsPublic",
    "OrganizationMembershipPublic",
    "UserGroupCreate",
    "UserGroupUpdate",
    "UserGroupPublic",
    "UserGroupsPublic",
    "UserGroupMembershipPublic",
    "ClusterAccess",
    "ClusterAccessPublic",
    "TenantQuota",
    "TenantQuotaUpdate",
    "TenantQuotaPublic",
    "TenantQuotasPublic",
]