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

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

Author: cursoragent

datacrunch_compat/datacrunch/__init__.py

@@ -6,6 +6,7 @@
     authentication,
     balance,
     constants,
+    container_types,
     containers,
     exceptions,
     helpers,
@@ -29,6 +30,7 @@
     'authentication',
     'balance',
     'constants',
+    'container_types',
     'containers',
     'datacrunch',
     'exceptions',

datacrunch_compat/datacrunch/datacrunch.py

@@ -5,6 +5,7 @@
 from verda.authentication import AuthenticationService
 from verda.balance import BalanceService
 from verda.constants import Constants
+from verda.container_types import ContainerTypesService
 from verda.containers import ContainersService
 from verda.http_client import HTTPClient
 from verda.images import ImagesService
@@ -21,6 +22,7 @@
     'AuthenticationService',
     'BalanceService',
     'Constants',
+    'ContainerTypesService',
     'ContainersService',
     'DataCrunchClient',
     'HTTPClient',

docs/source/api/services/container_types.rst

@@ -0,0 +1,8 @@
+Container Types
+===============
+
+.. autoclass:: verda.container_types.ContainerTypesService
+   :members:
+
+.. autoclass:: verda.container_types.ContainerType
+   :members:

tests/unit_tests/container_types/test_container_types.py

@@ -0,0 +1,49 @@
+import responses  # https://github.com/getsentry/responses
+
+from verda.container_types import ContainerType, ContainerTypesService
+
+CONTAINER_TYPE_ID = 'type-c0de-a5d2-4972-ae4e-d429115d055b'
+
+
+@responses.activate
+def test_container_types(http_client):
+    endpoint = http_client._base_url + '/container-types?currency=eur'
+    responses.add(
+        responses.GET,
+        endpoint,
+        json=[
+            {
+                'id': CONTAINER_TYPE_ID,
+                'model': 'H100',
+                'name': 'H100 SXM5 80GB',
+                'instance_type': '1H100.80S.22V',
+                'cpu': {'description': '22 CPU', 'number_of_cores': 22},
+                'gpu': {'description': '1x H100 SXM5 80GB', 'number_of_gpus': 1},
+                'gpu_memory': {'description': '80GB GPU RAM', 'size_in_gigabytes': 80},
+                'memory': {'description': '187GB RAM', 'size_in_gigabytes': 187},
+                'serverless_price': '1.75',
+                'serverless_spot_price': '0.87',
+                'currency': 'eur',
+                'manufacturer': 'NVIDIA',
+            }
+        ],
+        status=200,
+    )
+
+    service = ContainerTypesService(http_client)
+
+    container_types = service.get(currency='eur')
+    container_type = container_types[0]
+
+    assert isinstance(container_types, list)
+    assert len(container_types) == 1
+    assert isinstance(container_type, ContainerType)
+    assert container_type.id == CONTAINER_TYPE_ID
+    assert container_type.model == 'H100'
+    assert container_type.name == 'H100 SXM5 80GB'
+    assert container_type.instance_type == '1H100.80S.22V'
+    assert container_type.serverless_price == 1.75
+    assert container_type.serverless_spot_price == 0.87
+    assert container_type.currency == 'eur'
+    assert container_type.manufacturer == 'NVIDIA'
+    assert responses.assert_call_count(endpoint, 1) is True

tests/unit_tests/test_client.py

@@ -16,6 +16,7 @@
 
 
 class TestVerdaClient:
+    @responses.activate
     def test_client(self):
         # arrange - add response mock
         responses.add(responses.POST, BASE_URL + '/oauth2/token', json=response_json, status=200)
@@ -25,7 +26,9 @@ def test_client(self):
 
         # assert
         assert client.constants.base_url == BASE_URL
+        assert hasattr(client, 'container_types')
 
+    @responses.activate
     def test_client_with_default_base_url(self):
         # arrange - add response mock
         DEFAULT_BASE_URL = 'https://api.verda.com/v1'
@@ -42,6 +45,7 @@ def test_client_with_default_base_url(self):
         # assert
         assert client.constants.base_url == DEFAULT_BASE_URL
 
+    @responses.activate
     def test_invalid_client_credentials(self):
         # arrange - add response mock
         responses.add(

verda/_verda.py

@@ -2,6 +2,7 @@
 from verda.authentication import AuthenticationService
 from verda.balance import BalanceService
 from verda.clusters import ClustersService
+from verda.container_types import ContainerTypesService
 from verda.constants import Constants
 from verda.containers import ContainersService
 from verda.http_client import HTTPClient
@@ -80,6 +81,9 @@ def __init__(
         self.containers: ContainersService = ContainersService(self._http_client, inference_key)
         """Containers service. Deploy, manage, and monitor container deployments"""
 
+        self.container_types: ContainerTypesService = ContainerTypesService(self._http_client)
+        """Container types service. Get available serverless container SKUs"""
+
         self.clusters: ClustersService = ClustersService(self._http_client)
         """Clusters service. Create and manage compute clusters"""
 

verda/container_types/__init__.py

@@ -0,0 +1,3 @@
+from verda.container_types._container_types import ContainerType, ContainerTypesService
+
+__all__ = ['ContainerType', 'ContainerTypesService']

verda/container_types/_container_types.py

@@ -0,0 +1,58 @@
+from dataclasses import dataclass
+from typing import Literal
+
+from dataclasses_json import dataclass_json
+
+CONTAINER_TYPES_ENDPOINT = '/container-types'
+
+Currency = Literal['usd', 'eur']
+
+
+@dataclass_json
+@dataclass
+class ContainerType:
+    """Container type returned by the public API."""
+
+    id: str
+    model: str
+    name: str
+    instance_type: str
+    cpu: dict
+    gpu: dict
+    gpu_memory: dict
+    memory: dict
+    serverless_price: float
+    serverless_spot_price: float
+    currency: Currency
+    manufacturer: str
+
+
+class ContainerTypesService:
+    """Service for interacting with container types."""
+
+    def __init__(self, http_client) -> None:
+        self._http_client = http_client
+
+    def get(self, currency: Currency = 'usd') -> list[ContainerType]:
+        """Return all available container types."""
+        container_types = self._http_client.get(
+            CONTAINER_TYPES_ENDPOINT,
+            params={'currency': currency},
+        ).json()
+        return [
+            ContainerType(
+                id=container_type['id'],
+                model=container_type['model'],
+                name=container_type['name'],
+                instance_type=container_type['instance_type'],
+                cpu=container_type['cpu'],
+                gpu=container_type['gpu'],
+                gpu_memory=container_type['gpu_memory'],
+                memory=container_type['memory'],
+                serverless_price=float(container_type['serverless_price']),
+                serverless_spot_price=float(container_type['serverless_spot_price']),
+                currency=container_type['currency'],
+                manufacturer=container_type['manufacturer'],
+            )
+            for container_type in container_types
+        ]