dstack 0.18.41__py3-none-any.whl → 0.18.43__py3-none-any.whl

dstack/_internal/server/testing/common.py

@@ -16,7 +16,7 @@ from dstack._internal.core.models.configurations import (
 )
 from dstack._internal.core.models.envs import Env
 from dstack._internal.core.models.fleets import FleetConfiguration, FleetSpec, FleetStatus
-from dstack._internal.core.models.gateways import GatewayStatus
+from dstack._internal.core.models.gateways import GatewayComputeConfiguration, GatewayStatus
 from dstack._internal.core.models.instances import (
     Disk,
     Gpu,
@@ -55,7 +55,7 @@ from dstack._internal.core.models.runs import (
 from dstack._internal.core.models.users import GlobalRole
 from dstack._internal.core.models.volumes import (
     Volume,
-    VolumeAttachmentData,
+    VolumeAttachment,
     VolumeConfiguration,
     VolumeProvisioningData,
     VolumeStatus,
@@ -76,6 +76,7 @@ from dstack._internal.server.models import (
     RepoModel,
     RunModel,
     UserModel,
+    VolumeAttachmentModel,
     VolumeModel,
 )
 from dstack._internal.server.services.jobs import get_job_specs_from_run_spec
@@ -419,6 +420,24 @@ async def create_gateway_compute(
     return gateway_compute
 
 
+def get_gateway_compute_configuration(
+    project_name: str = "test-project",
+    instance_name: str = "test-instance",
+    backend: BackendType = BackendType.AWS,
+    region: str = "us",
+    public_ip: bool = True,
+) -> GatewayComputeConfiguration:
+    return GatewayComputeConfiguration(
+        project_name=project_name,
+        instance_name=instance_name,
+        backend=backend,
+        region=region,
+        public_ip=public_ip,
+        ssh_key_pub="",
+        certificate=None,
+    )
+
+
 async def create_pool(
     session: AsyncSession,
     project: ProjectModel,
@@ -532,16 +551,13 @@ async def create_instance(
         requirements = Requirements(resources=ResourcesSpec(cpu=1))
 
     if instance_configuration is None:
-        instance_configuration = InstanceConfiguration(
-            project_name="test_proj",
-            instance_name="test_instance_name",
-            instance_id="test instance id",
-            ssh_keys=[],
-            user="test_user",
-        )
+        instance_configuration = get_instance_configuration()
 
     if volumes is None:
         volumes = []
+    volume_attachments = []
+    for volume in volumes:
+        volume_attachments.append(VolumeAttachmentModel(volume=volume))
 
     im = InstanceModel(
         id=instance_id,
@@ -566,7 +582,7 @@ async def create_instance(
         requirements=requirements.json(),
         instance_configuration=instance_configuration.json(),
         remote_connection_info=remote_connection_info.json() if remote_connection_info else None,
-        volumes=volumes,
+        volume_attachments=volume_attachments,
         total_blocks=total_blocks,
         busy_blocks=busy_blocks,
     )
@@ -577,6 +593,19 @@ async def create_instance(
     return im
 
 
+def get_instance_configuration(
+    project_name: str = "test-project",
+    instance_name: str = "test-instance",
+    user: str = "dstack-user",
+) -> InstanceConfiguration:
+    return InstanceConfiguration(
+        project_name=project_name,
+        instance_name=instance_name,
+        user=user,
+        ssh_keys=[],
+    )
+
+
 def get_instance_offer_with_availability(
     backend: BackendType = BackendType.AWS,
     region: str = "eu-west",
@@ -587,6 +616,7 @@ def get_instance_offer_with_availability(
     spot: bool = False,
     blocks: int = 1,
     total_blocks: int = 1,
+    availability_zones: Optional[List[str]] = None,
 ):
     gpus = [Gpu(name="T4", memory_mib=16384, vendor=gpuhunt.AcceleratorVendor.NVIDIA)] * gpu_count
     return InstanceOfferWithAvailability(
@@ -605,6 +635,7 @@ def get_instance_offer_with_availability(
         region=region,
         price=1,
         availability=InstanceAvailability.AVAILABLE,
+        availability_zones=availability_zones,
         blocks=blocks,
         total_blocks=total_blocks,
     )
@@ -669,7 +700,7 @@ async def create_volume(
         volume_provisioning_data=volume_provisioning_data.json()
         if volume_provisioning_data
         else None,
-        instances=[],
+        attachments=[],
         deleted_at=deleted_at,
         deleted=True if deleted_at else False,
     )
@@ -691,16 +722,14 @@ def get_volume(
     deleted: bool = False,
     volume_id: Optional[str] = None,
     provisioning_data: Optional[VolumeProvisioningData] = None,
-    attachment_data: Optional[VolumeAttachmentData] = None,
-    device_name: Optional[str] = None,
+    attachments: Optional[List[VolumeAttachment]] = None,
 ) -> Volume:
     if id_ is None:
         id_ = uuid.uuid4()
     if configuration is None:
         configuration = get_volume_configuration()
-    if device_name is not None:
-        assert attachment_data is None, "attachment_data and device_name are mutually exclusive"
-        attachment_data = VolumeAttachmentData(device_name=device_name)
+    if attachments is None:
+        attachments = []
     return Volume(
         id=id_,
         name=name,
@@ -714,7 +743,7 @@ def get_volume(
         deleted=deleted,
         volume_id=volume_id,
         provisioning_data=provisioning_data,
-        attachment_data=attachment_data,
+        attachments=attachments,
     )
 
 

dstack/api/_public/runs.py

@@ -215,7 +215,7 @@ class Run(ABC):
                         start_time=next_start_time,
                         end_time=None,
                         descending=False,
-                        limit=100,
+                        limit=1000,
                         diagnose=diagnose,
                     ),
                 )

dstack/version.py

@@ -1,3 +1,3 @@
-__version__ = "0.18.41"
+__version__ = "0.18.43"
 __is_release__ = True
-base_image = "0.6"
+base_image = "0.7"

{dstack-0.18.41.dist-info → dstack-0.18.43.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dstack
-Version: 0.18.41
+Version: 0.18.43
 Summary: dstack is an open-source orchestration engine for running AI workloads on any cloud or on-premises.
 Home-page: https://dstack.ai
 Author: Andrey Cheptsov
@@ -37,7 +37,8 @@ Requires-Dist: websocket-client
 Requires-Dist: python-multipart>=0.0.16
 Requires-Dist: filelock
 Requires-Dist: psutil
-Requires-Dist: gpuhunt<0.1.0,>=0.0.19
+Requires-Dist: gpuhunt<0.2.0,>=0.1.0
+Requires-Dist: argcomplete>=3.5.0
 Provides-Extra: all
 Requires-Dist: fastapi; extra == "all"
 Requires-Dist: starlette>=0.26.0; extra == "all"
@@ -307,9 +308,9 @@ for AI workloads both in the cloud and on-prem, speeding up the development, tra
 
 ## Major news ✨
 
+- [2025/02] [dstack 0.18.41: GPU blocks, Proxy jump, inactivity duration, and more](https://github.com/dstackai/dstack/releases/tag/0.18.41)
 - [2025/01] [dstack 0.18.38: Intel Gaudi](https://github.com/dstackai/dstack/releases/tag/0.18.38)
 - [2025/01] [dstack 0.18.35: Vultr](https://github.com/dstackai/dstack/releases/tag/0.18.35)
-- [2024/12] [dstack 0.18.32: TPU v6e](https://github.com/dstackai/dstack/releases/tag/0.18.32)
 - [2024/12] [dstack 0.18.30: AWS Capacity Reservations and Capacity Blocks](https://github.com/dstackai/dstack/releases/tag/0.18.30)
 - [2024/10] [dstack 0.18.21: Instance volumes](https://github.com/dstackai/dstack/releases/tag/0.18.21)
 - [2024/10] [dstack 0.18.18: Hardware metrics monitoring](https://github.com/dstackai/dstack/releases/tag/0.18.18)