feat(cluster-types): add OpenAPI cluster types service

Co-authored-by: imagene-shahar <imagene-shahar@users.noreply.github.com>

Author: cursoragent

datacrunch_compat/datacrunch/__init__.py

@@ -5,6 +5,7 @@
     __version__,
     authentication,
     balance,
+    cluster_types,
     constants,
     container_types,
     containers,
@@ -29,6 +30,7 @@
     '__version__',
     'authentication',
     'balance',
+    'cluster_types',
     'constants',
     'container_types',
     'containers',

datacrunch_compat/datacrunch/datacrunch.py

@@ -4,6 +4,7 @@
 from verda._version import __version__
 from verda.authentication import AuthenticationService
 from verda.balance import BalanceService
+from verda.cluster_types import ClusterTypesService
 from verda.constants import Constants
 from verda.container_types import ContainerTypesService
 from verda.containers import ContainersService
@@ -21,6 +22,7 @@
 __all__ = [
     'AuthenticationService',
     'BalanceService',
+    'ClusterTypesService',
     'Constants',
     'ContainerTypesService',
     'ContainersService',

docs/source/api/services/cluster_types.rst

@@ -0,0 +1,8 @@
+Cluster Types
+=============
+
+.. autoclass:: verda.cluster_types.ClusterTypesService
+   :members:
+
+.. autoclass:: verda.cluster_types.ClusterType
+   :members:

tests/unit_tests/cluster_types/test_cluster_types.py

@@ -0,0 +1,51 @@
+import responses  # https://github.com/getsentry/responses
+
+from verda.cluster_types import ClusterType, ClusterTypesService
+
+CLUSTER_TYPE_ID = 'cluster-c0de-a5d2-4972-ae4e-d429115d055b'
+
+
+@responses.activate
+def test_cluster_types(http_client):
+    endpoint = http_client._base_url + '/cluster-types?currency=usd'
+    responses.add(
+        responses.GET,
+        endpoint,
+        json=[
+            {
+                'id': CLUSTER_TYPE_ID,
+                'model': 'H200',
+                'name': 'H200 Cluster',
+                'cluster_type': '16H200',
+                'cpu': {'description': '64 CPU', 'number_of_cores': 64},
+                'gpu': {'description': '16x H200', 'number_of_gpus': 16},
+                'gpu_memory': {'description': '2.2TB VRAM', 'size_in_gigabytes': 2200},
+                'memory': {'description': '4TB RAM', 'size_in_gigabytes': 4096},
+                'price_per_hour': '45.50',
+                'currency': 'usd',
+                'manufacturer': 'NVIDIA',
+                'node_details': ['2x 8 GPU nodes'],
+                'supported_os': ['ubuntu-24.04-cuda-12.8-cluster'],
+            }
+        ],
+        status=200,
+    )
+
+    service = ClusterTypesService(http_client)
+
+    cluster_types = service.get()
+    cluster_type = cluster_types[0]
+
+    assert isinstance(cluster_types, list)
+    assert len(cluster_types) == 1
+    assert isinstance(cluster_type, ClusterType)
+    assert cluster_type.id == CLUSTER_TYPE_ID
+    assert cluster_type.model == 'H200'
+    assert cluster_type.name == 'H200 Cluster'
+    assert cluster_type.cluster_type == '16H200'
+    assert cluster_type.price_per_hour == 45.5
+    assert cluster_type.currency == 'usd'
+    assert cluster_type.manufacturer == 'NVIDIA'
+    assert cluster_type.node_details == ['2x 8 GPU nodes']
+    assert cluster_type.supported_os == ['ubuntu-24.04-cuda-12.8-cluster']
+    assert responses.assert_call_count(endpoint, 1) is True

tests/unit_tests/test_client.py

@@ -27,6 +27,7 @@ def test_client(self):
         # assert
         assert client.constants.base_url == BASE_URL
         assert hasattr(client, 'container_types')
+        assert hasattr(client, 'cluster_types')
 
     @responses.activate
     def test_client_with_default_base_url(self):

verda/_verda.py

@@ -1,6 +1,7 @@
 from verda._version import __version__
 from verda.authentication import AuthenticationService
 from verda.balance import BalanceService
+from verda.cluster_types import ClusterTypesService
 from verda.clusters import ClustersService
 from verda.container_types import ContainerTypesService
 from verda.constants import Constants
@@ -87,5 +88,8 @@ def __init__(
         self.clusters: ClustersService = ClustersService(self._http_client)
         """Clusters service. Create and manage compute clusters"""
 
+        self.cluster_types: ClusterTypesService = ClusterTypesService(self._http_client)
+        """Cluster types service. Get available cluster SKUs"""
+
 
 __all__ = ['VerdaClient']

verda/cluster_types/__init__.py

@@ -0,0 +1,3 @@
+from verda.cluster_types._cluster_types import ClusterType, ClusterTypesService
+
+__all__ = ['ClusterType', 'ClusterTypesService']

verda/cluster_types/_cluster_types.py

@@ -0,0 +1,60 @@
+from dataclasses import dataclass
+from typing import Literal
+
+from dataclasses_json import dataclass_json
+
+CLUSTER_TYPES_ENDPOINT = '/cluster-types'
+
+Currency = Literal['usd', 'eur']
+
+
+@dataclass_json
+@dataclass
+class ClusterType:
+    """Cluster type returned by the public API."""
+
+    id: str
+    model: str
+    name: str
+    cluster_type: str
+    cpu: dict
+    gpu: dict
+    gpu_memory: dict
+    memory: dict
+    price_per_hour: float
+    currency: Currency
+    manufacturer: str
+    node_details: list[str]
+    supported_os: list[str]
+
+
+class ClusterTypesService:
+    """Service for interacting with cluster types."""
+
+    def __init__(self, http_client) -> None:
+        self._http_client = http_client
+
+    def get(self, currency: Currency = 'usd') -> list[ClusterType]:
+        """Return all available cluster types."""
+        cluster_types = self._http_client.get(
+            CLUSTER_TYPES_ENDPOINT,
+            params={'currency': currency},
+        ).json()
+        return [
+            ClusterType(
+                id=cluster_type['id'],
+                model=cluster_type['model'],
+                name=cluster_type['name'],
+                cluster_type=cluster_type['cluster_type'],
+                cpu=cluster_type['cpu'],
+                gpu=cluster_type['gpu'],
+                gpu_memory=cluster_type['gpu_memory'],
+                memory=cluster_type['memory'],
+                price_per_hour=float(cluster_type['price_per_hour']),
+                currency=cluster_type['currency'],
+                manufacturer=cluster_type['manufacturer'],
+                node_details=cluster_type['node_details'],
+                supported_os=cluster_type['supported_os'],
+            )
+            for cluster_type in cluster_types
+        ]