dstack 0.16.3__tar.gz → 0.16.4__tar.gz

{dstack-0.16.3 → dstack-0.16.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dstack
-Version: 0.16.3
+Version: 0.16.4
 Summary: dstack is an open-source engine for running GPU workloads across any cloud provider.
 Home-page: https://dstack.ai
 Author: Andrey Cheptsov

{dstack-0.16.3 → dstack-0.16.4}/setup.py

@@ -60,7 +60,7 @@ BASE_DEPS = [
     "docker>=6.0.0",
     "dnspython",
     "grpcio>=1.50",  # indirect
-    "gpuhunt==0.0.6",
+    "gpuhunt==0.0.8",
     "sentry-sdk[fastapi]",
     "httpx",
 ]

{dstack-0.16.3 → dstack-0.16.4}/src/dstack/_internal/core/models/backends/__init__.py

@@ -26,7 +26,11 @@ from dstack._internal.core.models.backends.datacrunch import (
     DataCrunchConfigInfoWithCredsPartial,
     DataCrunchConfigValues,
 )
-from dstack._internal.core.models.backends.dstack import DstackConfigInfo, DstackConfigValues
+from dstack._internal.core.models.backends.dstack import (
+    DstackBaseBackendConfigInfo,
+    DstackConfigInfo,
+    DstackConfigValues,
+)
 from dstack._internal.core.models.backends.gcp import (
     GCPConfigInfo,
     GCPConfigInfoWithCreds,
@@ -76,6 +80,7 @@ AnyConfigInfoWithoutCreds = Union[
     TensorDockConfigInfo,
     VastAIConfigInfo,
     DstackConfigInfo,
+    DstackBaseBackendConfigInfo,
 ]
 AnyConfigInfoWithCreds = Union[
     AWSConfigInfoWithCreds,

{dstack-0.16.3 → dstack-0.16.4}/src/dstack/_internal/core/models/backends/dstack.py

@@ -1,9 +1,16 @@
+from typing import List
+
 from pydantic import BaseModel
 from typing_extensions import Literal
 
 
 class DstackConfigInfo(BaseModel):
     type: Literal["dstack"] = "dstack"
+    base_backends: List[str]
+
+
+class DstackBaseBackendConfigInfo(BaseModel):
+    type: str
 
 
 class DstackConfigValues(BaseModel):

{dstack-0.16.3 → dstack-0.16.4}/src/dstack/_internal/server/routers/backends.py

@@ -22,6 +22,9 @@ root_router = APIRouter(prefix="/api/backends", tags=["backends"])
 project_router = APIRouter(prefix="/api/project/{project_name}/backends", tags=["backends"])
 
 
+# These routes are not currently in use since backend UI configuration is disabled.
+
+
 @root_router.post("/list_types")
 async def list_backend_types() -> List[BackendType]:
     return backends.list_available_backend_types()

{dstack-0.16.3 → dstack-0.16.4}/src/dstack/_internal/server/services/projects.py

@@ -6,6 +6,10 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from dstack._internal.core.errors import ForbiddenError, ResourceExistsError, ServerClientError
 from dstack._internal.core.models.backends import BackendInfo
+from dstack._internal.core.models.backends.dstack import (
+    DstackBaseBackendConfigInfo,
+    DstackConfigInfo,
+)
 from dstack._internal.core.models.projects import Member, Project
 from dstack._internal.core.models.users import GlobalRole, ProjectRole
 from dstack._internal.server.models import MemberModel, ProjectModel, UserModel
@@ -272,11 +276,20 @@ def project_model_to_project(project_model: ProjectModel) -> Project:
             logger.warning("Configurator for backend %s not found", b.type)
             continue
         config_info = configurator.get_config_info(model=b, include_creds=False)
-        backend_info = BackendInfo(
-            name=b.type,
-            config=config_info,
-        )
-        backends.append(backend_info)
+        if isinstance(config_info, DstackConfigInfo):
+            for backend_type in config_info.base_backends:
+                backends.append(
+                    BackendInfo(
+                        name=backend_type, config=DstackBaseBackendConfigInfo(type=backend_type)
+                    )
+                )
+        else:
+            backends.append(
+                BackendInfo(
+                    name=b.type,
+                    config=config_info,
+                )
+            )
     return Project(
         project_id=project_model.id,
         project_name=project_model.name,

{dstack-0.16.3 → dstack-0.16.4}/src/dstack/version.py

@@ -1,3 +1,3 @@
-__version__ = "0.16.3"
+__version__ = "0.16.4"
 __is_release__ = True
 base_image = "0.4rc4"

{dstack-0.16.3 → dstack-0.16.4}/src/dstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dstack
-Version: 0.16.3
+Version: 0.16.4
 Summary: dstack is an open-source engine for running GPU workloads across any cloud provider.
 Home-page: https://dstack.ai
 Author: Andrey Cheptsov

{dstack-0.16.3 → dstack-0.16.4}/src/dstack.egg-info/requires.txt

@@ -30,7 +30,7 @@ filelock
 docker>=6.0.0
 dnspython
 grpcio>=1.50
-gpuhunt==0.0.6
+gpuhunt==0.0.8
 sentry-sdk[fastapi]
 httpx