dstack 0.18.43__py3-none-any.whl → 0.19.0rc1__py3-none-any.whl

dstack/_internal/core/backends/vastai/config.py

@@ -1,6 +0,0 @@
-from dstack._internal.core.backends.base.config import BackendConfig
-from dstack._internal.core.models.backends.vastai import AnyVastAICreds, VastAIStoredConfig
-
-
-class VastAIConfig(VastAIStoredConfig, BackendConfig):
-    creds: AnyVastAICreds

dstack/_internal/core/backends/vultr/config.py

@@ -1,9 +0,0 @@
-from dstack._internal.core.backends.base.config import BackendConfig
-from dstack._internal.core.models.backends.vultr import (
-    AnyVultrCreds,
-    VultrStoredConfig,
-)
-
-
-class VultrConfig(VultrStoredConfig, BackendConfig):
-    creds: AnyVultrCreds

dstack/_internal/core/models/backends/aws.py

@@ -1,86 +0,0 @@
-from typing import Dict
-
-from pydantic import Field
-from typing_extensions import Annotated, List, Literal, Optional, Union
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class AWSOSImage(CoreModel):
-    name: Annotated[str, Field(description="The AMI name")]
-    owner: Annotated[
-        str,
-        Field(regex=r"^(\d{12}|self)$", description="The AMI owner, account ID or `self`"),
-    ] = "self"
-    user: Annotated[str, Field(description="The OS user for provisioning")]
-
-
-class AWSOSImageConfig(CoreModel):
-    cpu: Annotated[Optional[AWSOSImage], Field(description="The AMI used for CPU instances")] = (
-        None
-    )
-    nvidia: Annotated[
-        Optional[AWSOSImage], Field(description="The AMI used for NVIDIA GPU instances")
-    ] = None
-
-
-class AWSConfigInfo(CoreModel):
-    type: Literal["aws"] = "aws"
-    regions: Optional[List[str]] = None
-    vpc_name: Optional[str] = None
-    vpc_ids: Optional[Dict[str, str]] = None
-    default_vpcs: Optional[bool] = None
-    public_ips: Optional[bool] = None
-    iam_instance_profile: Optional[str] = None
-    tags: Optional[Dict[str, str]] = None
-    os_images: Optional[AWSOSImageConfig] = None
-
-
-class AWSAccessKeyCreds(CoreModel):
-    type: Annotated[Literal["access_key"], Field(description="The type of credentials")] = (
-        "access_key"
-    )
-    access_key: Annotated[str, Field(description="The access key")]
-    secret_key: Annotated[str, Field(description="The secret key")]
-
-
-class AWSDefaultCreds(CoreModel):
-    type: Annotated[Literal["default"], Field(description="The type of credentials")] = "default"
-
-
-AnyAWSCreds = Union[AWSAccessKeyCreds, AWSDefaultCreds]
-
-
-class AWSCreds(CoreModel):
-    __root__: AnyAWSCreds = Field(..., discriminator="type")
-
-
-class AWSConfigInfoWithCreds(AWSConfigInfo):
-    creds: AnyAWSCreds
-
-
-AnyAWSConfigInfo = Union[AWSConfigInfo, AWSConfigInfoWithCreds]
-
-
-class AWSConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["aws"] = "aws"
-    creds: Optional[AnyAWSCreds]
-    regions: Optional[List[str]]
-    vpc_name: Optional[str]
-    vpc_ids: Optional[Dict[str, str]]
-    default_vpcs: Optional[bool]
-    public_ips: Optional[bool]
-    iam_instance_profile: Optional[str]
-    tags: Optional[Dict[str, str]]
-    os_images: Optional["AWSOSImageConfig"]
-
-
-class AWSConfigValues(CoreModel):
-    type: Literal["aws"] = "aws"
-    default_creds: bool = False
-    regions: Optional[ConfigMultiElement]
-
-
-class AWSStoredConfig(AWSConfigInfo):
-    pass

dstack/_internal/core/models/backends/azure.py

@@ -1,68 +0,0 @@
-from typing import Dict
-
-from pydantic import Field
-from typing_extensions import Annotated, List, Literal, Optional, Union
-
-from dstack._internal.core.models.backends.base import ConfigElement, ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class AzureConfigInfo(CoreModel):
-    type: Literal["azure"] = "azure"
-    tenant_id: str
-    subscription_id: str
-    resource_group: Optional[str] = None
-    locations: Optional[List[str]] = None
-    vpc_ids: Optional[Dict[str, str]] = None
-    public_ips: Optional[bool] = None
-    tags: Optional[Dict[str, str]] = None
-
-
-class AzureClientCreds(CoreModel):
-    type: Annotated[Literal["client"], Field(description="The type of credentials")] = "client"
-    client_id: Annotated[str, Field(description="The client ID")]
-    client_secret: Annotated[str, Field(description="The client secret")]
-    # if tenant_id is missing, it will be populated from config info
-    tenant_id: Optional[str]
-
-
-class AzureDefaultCreds(CoreModel):
-    type: Annotated[Literal["default"], Field(description="The type of credentials")] = "default"
-
-
-AnyAzureCreds = Union[AzureClientCreds, AzureDefaultCreds]
-
-
-class AzureCreds(CoreModel):
-    __root__: AnyAzureCreds = Field(..., discriminator="type")
-
-
-class AzureConfigInfoWithCreds(AzureConfigInfo):
-    creds: AnyAzureCreds
-
-
-AnyAzureConfigInfo = Union[AzureConfigInfo, AzureConfigInfoWithCreds]
-
-
-class AzureConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["azure"] = "azure"
-    creds: Optional[AnyAzureCreds]
-    tenant_id: Optional[str]
-    subscription_id: Optional[str]
-    resource_group: Optional[str]
-    locations: Optional[List[str]]
-    vpc_ids: Optional[Dict[str, str]]
-    public_ips: Optional[bool]
-    tags: Optional[Dict[str, str]]
-
-
-class AzureConfigValues(CoreModel):
-    type: Literal["azure"] = "azure"
-    default_creds: bool = False
-    tenant_id: Optional[ConfigElement]
-    subscription_id: Optional[ConfigElement]
-    locations: Optional[ConfigMultiElement]
-
-
-class AzureStoredConfig(AzureConfigInfo):
-    resource_group: str = ""

dstack/_internal/core/models/backends/cudo.py

@@ -1,43 +0,0 @@
-from typing import List, Optional
-
-from pydantic.fields import Field
-from typing_extensions import Annotated, Literal
-
-from dstack._internal.core.models.backends.base import ConfigElement, ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class CudoConfigInfo(CoreModel):
-    type: Literal["cudo"] = "cudo"
-    project_id: str
-    regions: Optional[List[str]] = None
-
-
-class CudoStoredConfig(CudoConfigInfo):
-    pass
-
-
-class CudoAPIKeyCreds(CoreModel):
-    type: Annotated[Literal["api_key"], Field(description="The type of credentials")] = "api_key"
-    api_key: Annotated[str, Field(description="The API key")]
-
-
-AnyCudoCreds = CudoAPIKeyCreds
-CudoCreds = AnyCudoCreds
-
-
-class CudoConfigInfoWithCreds(CudoConfigInfo):
-    creds: AnyCudoCreds
-
-
-class CudoConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["cudo"] = "cudo"
-    creds: Optional[AnyCudoCreds]
-    project_id: Optional[str]
-    regions: Optional[List[str]]
-
-
-class CudoConfigValues(CoreModel):
-    type: Literal["cudo"] = "cudo"
-    regions: Optional[ConfigMultiElement]
-    project_id: Optional[ConfigElement]

dstack/_internal/core/models/backends/datacrunch.py

@@ -1,44 +0,0 @@
-from pydantic.fields import Field
-from typing_extensions import Annotated, List, Literal, Optional, Union
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class DataCrunchConfigInfo(CoreModel):
-    type: Literal["datacrunch"] = "datacrunch"
-    regions: Optional[List[str]] = None
-
-
-class DataCrunchAPIKeyCreds(CoreModel):
-    type: Annotated[Literal["api_key"], Field(description="The type of credentials")] = "api_key"
-    client_id: Annotated[str, Field(description="The client ID")]
-    client_secret: Annotated[str, Field(description="The client secret")]
-
-
-AnyDataCrunchCreds = DataCrunchAPIKeyCreds
-
-
-DataCrunchCreds = AnyDataCrunchCreds
-
-
-class DataCrunchConfigInfoWithCreds(DataCrunchConfigInfo):
-    creds: AnyDataCrunchCreds
-
-
-AnyDataCrunchConfigInfo = Union[DataCrunchConfigInfo, DataCrunchConfigInfoWithCreds]
-
-
-class DataCrunchConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["datacrunch"] = "datacrunch"
-    creds: Optional[AnyDataCrunchCreds]
-    regions: Optional[List[str]]
-
-
-class DataCrunchConfigValues(CoreModel):
-    type: Literal["datacrunch"] = "datacrunch"
-    regions: Optional[ConfigMultiElement]
-
-
-class DataCrunchStoredConfig(DataCrunchConfigInfo):
-    pass

dstack/_internal/core/models/backends/gcp.py

@@ -1,67 +0,0 @@
-from typing import Dict, List, Optional, Union
-
-from pydantic import Field
-from typing_extensions import Literal
-
-from dstack._internal.core.models.backends.base import ConfigElement, ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class GCPConfigInfo(CoreModel):
-    type: Literal["gcp"] = "gcp"
-    project_id: str
-    regions: Optional[List[str]] = None
-    vpc_name: Optional[str] = None
-    vpc_project_id: Optional[str] = None
-    public_ips: Optional[bool] = None
-    nat_check: Optional[bool] = None
-    vm_service_account: Optional[str] = None
-    tags: Optional[Dict[str, str]] = None
-
-
-class GCPServiceAccountCreds(CoreModel):
-    type: Literal["service_account"] = "service_account"
-    filename: str
-    data: str
-
-
-class GCPDefaultCreds(CoreModel):
-    type: Literal["default"] = "default"
-
-
-AnyGCPCreds = Union[GCPServiceAccountCreds, GCPDefaultCreds]
-
-
-class GCPCreds(CoreModel):
-    __root__: AnyGCPCreds = Field(..., discriminator="type")
-
-
-class GCPConfigInfoWithCreds(GCPConfigInfo):
-    creds: AnyGCPCreds
-
-
-AnyGCPConfigInfo = Union[GCPConfigInfo, GCPConfigInfoWithCreds]
-
-
-class GCPConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["gcp"] = "gcp"
-    creds: Optional[AnyGCPCreds]
-    project_id: Optional[str]
-    regions: Optional[List[str]]
-    vpc_name: Optional[str] = None
-    vpc_project_id: Optional[str] = None
-    public_ips: Optional[bool]
-    nat_check: Optional[bool] = None
-    vm_service_account: Optional[str] = None
-    tags: Optional[Dict[str, str]] = None
-
-
-class GCPConfigValues(CoreModel):
-    type: Literal["gcp"] = "gcp"
-    default_creds: bool = False
-    project_id: Optional[ConfigElement]
-    regions: Optional[ConfigMultiElement]
-
-
-class GCPStoredConfig(GCPConfigInfo):
-    pass

dstack/_internal/core/models/backends/kubernetes.py

@@ -1,40 +0,0 @@
-from pydantic.fields import Field
-from typing_extensions import Annotated, Literal, Optional, Union
-
-from dstack._internal.core.models.common import CoreModel
-
-
-class KubernetesNetworkingConfig(CoreModel):
-    ssh_host: Annotated[Optional[str], Field(description="The external IP address of any node")]
-    ssh_port: Annotated[
-        Optional[int], Field(description="Any port accessible outside of the cluster")
-    ]
-
-
-class KubernetesConfigInfo(CoreModel):
-    type: Literal["kubernetes"] = "kubernetes"
-    networking: KubernetesNetworkingConfig
-
-
-class KubeconfigConfig(CoreModel):
-    filename: str
-    data: str
-
-
-class KubernetesConfigInfoWithCreds(KubernetesConfigInfo):
-    kubeconfig: KubeconfigConfig
-
-
-AnyKubernetesConfigInfo = Union[KubernetesConfigInfo, KubernetesConfigInfoWithCreds]
-
-
-class KubernetesConfigInfoWithCredsPartial(KubernetesConfigInfoWithCreds):
-    pass
-
-
-class KubernetesConfigValues(CoreModel):
-    type: Literal["kubernetes"] = "kubernetes"
-
-
-class KubernetesStoredConfig(KubernetesConfigInfoWithCreds):
-    pass

dstack/_internal/core/models/backends/lambdalabs.py

@@ -1,43 +0,0 @@
-from pydantic.fields import Field
-from typing_extensions import Annotated, List, Literal, Optional, Union
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class LambdaConfigInfo(CoreModel):
-    type: Literal["lambda"] = "lambda"
-    regions: Optional[List[str]] = None
-
-
-class LambdaAPIKeyCreds(CoreModel):
-    type: Annotated[Literal["api_key"], Field(description="The type of credentials")] = "api_key"
-    api_key: Annotated[str, Field(description="The API key")]
-
-
-AnyLambdaCreds = LambdaAPIKeyCreds
-
-
-LambdaCreds = AnyLambdaCreds
-
-
-class LambdaConfigInfoWithCreds(LambdaConfigInfo):
-    creds: AnyLambdaCreds
-
-
-AnyLambdaConfigInfo = Union[LambdaConfigInfo, LambdaConfigInfoWithCreds]
-
-
-class LambdaConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["lambda"] = "lambda"
-    creds: Optional[AnyLambdaCreds]
-    regions: Optional[List[str]]
-
-
-class LambdaConfigValues(CoreModel):
-    type: Literal["lambda"] = "lambda"
-    regions: Optional[ConfigMultiElement]
-
-
-class LambdaStoredConfig(LambdaConfigInfo):
-    pass

dstack/_internal/core/models/backends/nebius.py

@@ -1,54 +0,0 @@
-from typing import List, Optional, Union
-
-from typing_extensions import Literal
-
-from dstack._internal.core.models.backends.base import ConfigElement, ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class NebiusConfigInfo(CoreModel):
-    type: Literal["nebius"] = "nebius"
-    cloud_id: str
-    folder_id: str
-    network_id: str
-    regions: Optional[List[str]] = None
-
-
-class NebiusServiceAccountCreds(CoreModel):
-    type: Literal["service_account"] = "service_account"
-    filename: str
-    data: str
-
-
-AnyNebiusCreds = NebiusServiceAccountCreds
-
-
-NebiusCreds = AnyNebiusCreds
-
-
-class NebiusConfigInfoWithCreds(NebiusConfigInfo):
-    creds: AnyNebiusCreds
-
-
-AnyNebiusConfigInfo = Union[NebiusConfigInfo, NebiusConfigInfoWithCreds]
-
-
-class NebiusConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["nebius"] = "nebius"
-    creds: Optional[AnyNebiusCreds]
-    cloud_id: Optional[str]
-    folder_id: Optional[str]
-    network_id: Optional[str]
-    regions: Optional[List[str]]
-
-
-class NebiusConfigValues(CoreModel):
-    type: Literal["nebius"] = "nebius"
-    cloud_id: Optional[ConfigElement]
-    folder_id: Optional[ConfigElement]
-    network_id: Optional[ConfigElement]
-    regions: Optional[ConfigMultiElement]
-
-
-class NebiusStoredConfig(NebiusConfigInfo):
-    pass

dstack/_internal/core/models/backends/runpod.py

@@ -1,40 +0,0 @@
-from typing import Annotated, List, Optional
-
-from pydantic import Field
-from typing_extensions import Literal
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class RunpodConfigInfo(CoreModel):
-    type: Literal["runpod"] = "runpod"
-    regions: Optional[List[str]] = None
-
-
-class RunpodStoredConfig(RunpodConfigInfo):
-    pass
-
-
-class RunpodAPIKeyCreds(CoreModel):
-    type: Literal["api_key"] = "api_key"
-    api_key: Annotated[str, Field(description="The API key")]
-
-
-AnyRunpodCreds = RunpodAPIKeyCreds
-RunpodCreds = AnyRunpodCreds
-
-
-class RunpodConfigInfoWithCreds(RunpodConfigInfo):
-    creds: AnyRunpodCreds
-
-
-class RunpodConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["runpod"] = "runpod"
-    creds: Optional[AnyRunpodCreds]
-    regions: Optional[List[str]]
-
-
-class RunpodConfigValues(CoreModel):
-    type: Literal["runpod"] = "runpod"
-    regions: Optional[ConfigMultiElement]

dstack/_internal/core/models/backends/tensordock.py

@@ -1,44 +0,0 @@
-from pydantic.fields import Field
-from typing_extensions import Annotated, List, Literal, Optional, Union
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class TensorDockConfigInfo(CoreModel):
-    type: Literal["tensordock"] = "tensordock"
-    regions: Optional[List[str]] = None
-
-
-class TensorDockAPIKeyCreds(CoreModel):
-    type: Annotated[Literal["api_key"], Field(description="The type of credentials")] = "api_key"
-    api_key: Annotated[str, Field(description="The API key")]
-    api_token: Annotated[str, Field(description="The API token")]
-
-
-AnyTensorDockCreds = TensorDockAPIKeyCreds
-
-
-TensorDockCreds = AnyTensorDockCreds
-
-
-class TensorDockConfigInfoWithCreds(TensorDockConfigInfo):
-    creds: AnyTensorDockCreds
-
-
-AnyTensorDockConfigInfo = Union[TensorDockConfigInfo, TensorDockConfigInfoWithCreds]
-
-
-class TensorDockConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["tensordock"] = "tensordock"
-    creds: Optional[AnyTensorDockCreds]
-    regions: Optional[List[str]]
-
-
-class TensorDockConfigValues(CoreModel):
-    type: Literal["tensordock"] = "tensordock"
-    regions: Optional[ConfigMultiElement]
-
-
-class TensorDockStoredConfig(TensorDockConfigInfo):
-    pass

dstack/_internal/core/models/backends/vastai.py

@@ -1,43 +0,0 @@
-from pydantic.fields import Field
-from typing_extensions import Annotated, List, Literal, Optional, Union
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class VastAIConfigInfo(CoreModel):
-    type: Literal["vastai"] = "vastai"
-    regions: Optional[List[str]] = None
-
-
-class VastAIAPIKeyCreds(CoreModel):
-    type: Annotated[Literal["api_key"], Field(description="The type of credentials")] = "api_key"
-    api_key: Annotated[str, Field(description="The API key")]
-
-
-AnyVastAICreds = VastAIAPIKeyCreds
-
-
-VastAICreds = AnyVastAICreds
-
-
-class VastAIConfigInfoWithCreds(VastAIConfigInfo):
-    creds: AnyVastAICreds
-
-
-AnyVastAIConfigInfo = Union[VastAIConfigInfo, VastAIConfigInfoWithCreds]
-
-
-class VastAIConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["vastai"] = "vastai"
-    creds: Optional[AnyVastAICreds]
-    regions: Optional[List[str]]
-
-
-class VastAIConfigValues(CoreModel):
-    type: Literal["vastai"] = "vastai"
-    regions: Optional[ConfigMultiElement]
-
-
-class VastAIStoredConfig(VastAIConfigInfo):
-    pass

dstack/_internal/core/models/backends/vultr.py

@@ -1,40 +0,0 @@
-from typing import List, Optional
-
-from pydantic.fields import Field
-from typing_extensions import Annotated, Literal
-
-from dstack._internal.core.models.backends.base import ConfigMultiElement
-from dstack._internal.core.models.common import CoreModel
-
-
-class VultrConfigInfo(CoreModel):
-    type: Literal["vultr"] = "vultr"
-    regions: Optional[List[str]] = None
-
-
-class VultrStoredConfig(VultrConfigInfo):
-    pass
-
-
-class VultrAPIKeyCreds(CoreModel):
-    type: Annotated[Literal["api_key"], Field(description="The type of credentials")] = "api_key"
-    api_key: Annotated[str, Field(description="The API key")]
-
-
-AnyVultrCreds = VultrAPIKeyCreds
-VultrCreds = AnyVultrCreds
-
-
-class VultrConfigInfoWithCreds(VultrConfigInfo):
-    creds: AnyVultrCreds
-
-
-class VultrConfigInfoWithCredsPartial(CoreModel):
-    type: Literal["vultr"] = "vultr"
-    creds: Optional[AnyVultrCreds]
-    regions: Optional[List[str]]
-
-
-class VultrConfigValues(CoreModel):
-    type: Literal["vultr"] = "vultr"
-    regions: Optional[ConfigMultiElement]

dstack/_internal/core/models/pools.py

@@ -1,43 +0,0 @@
-import datetime
-from typing import List, Optional
-from uuid import UUID
-
-from dstack._internal.core.models.backends.base import BackendType
-from dstack._internal.core.models.common import CoreModel
-from dstack._internal.core.models.instances import InstanceStatus, InstanceType
-
-
-class Pool(CoreModel):
-    name: str
-    default: bool
-    created_at: datetime.datetime
-    total_instances: int
-    available_instances: int
-
-
-class Instance(CoreModel):
-    id: UUID
-    project_name: str
-    backend: Optional[BackendType] = None
-    instance_type: Optional[InstanceType] = None
-    name: str
-    fleet_id: Optional[UUID] = None
-    fleet_name: Optional[str] = None
-    instance_num: int
-    pool_name: Optional[str] = None
-    job_name: Optional[str] = None  # deprecated, always None (instance can have more than one job)
-    hostname: Optional[str] = None
-    status: InstanceStatus
-    unreachable: bool = False
-    termination_reason: Optional[str] = None
-    created: datetime.datetime
-    region: Optional[str] = None
-    availability_zone: Optional[str] = None
-    price: Optional[float] = None
-    total_blocks: Optional[int] = None
-    busy_blocks: int = 0
-
-
-class PoolInstances(CoreModel):
-    name: str
-    instances: List[Instance]