gpuai-sdk 0.4.1__tar.gz → 0.4.2__tar.gz

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpuai-sdk
-Version: 0.4.1
+Version: 0.4.2
 Summary: GPU.ai Public Developer API
 Home-page: 
 Author: GPU.ai Support
@@ -27,7 +27,7 @@ the dashboard. See https://gpu.ai/docs for details.
 This Python package is automatically generated by the [OpenAPI Generator](https://openapi-generator.tech) project:
 
 - API version: 1.0.0
-- Package version: 0.4.1
+- Package version: 0.4.2
 - Generator version: 7.24.0
 - Build package: org.openapitools.codegen.languages.PythonClientCodegen
 
@@ -191,6 +191,7 @@ Class | Method | HTTP request | Description
 *InstancesApi* | [**create_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#create_instance) | **POST** /instances | Create an instance (async — returns 202 + Operation-Id header)
 *InstancesApi* | [**delete_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#delete_instance) | **DELETE** /instances/{id} | Terminate an instance (idempotent)
 *InstancesApi* | [**get_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#get_instance) | **GET** /instances/{id} | Get an instance
+*InstancesApi* | [**get_instance_gpu_suggestion**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#get_instance_gpu_suggestion) | **GET** /instances/{id}/gpu-suggestion | Suggest a lower-priced GPU for an instance
 *InstancesApi* | [**list_instances**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#list_instances) | **GET** /instances | List instances
 *InstancesApi* | [**update_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#update_instance) | **PATCH** /instances/{id} | Update an instance (rename only in v1)
 *MarketApi* | [**get_market_history**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/MarketApi.md#get_market_history) | **GET** /market-history | Get a GPU's price and availability over time
@@ -276,6 +277,9 @@ Class | Method | HTTP request | Description
  - [GPUType](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GPUType.md)
  - [GPUTypePage](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GPUTypePage.md)
  - [GetHealth200Response](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GetHealth200Response.md)
+ - [GpuSuggestion](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GpuSuggestion.md)
+ - [GpuSuggestionCheaper](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GpuSuggestionCheaper.md)
+ - [GpuSuggestionGpu](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GpuSuggestionGpu.md)
  - [ImagesGenerationsRequest](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/ImagesGenerationsRequest.md)
  - [ImagesResponse](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/ImagesResponse.md)
  - [ImagesResponseDataInner](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/ImagesResponseDataInner.md)
@@ -283,6 +287,7 @@ Class | Method | HTTP request | Description
  - [IncludeItem](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/IncludeItem.md)
  - [Instance](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/Instance.md)
  - [InstanceConnection](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstanceConnection.md)
+ - [InstanceGpuMemory](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstanceGpuMemory.md)
  - [InstancePage](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancePage.md)
  - [Invoice](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/Invoice.md)
  - [InvoiceList](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InvoiceList.md)

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/README.md

@@ -6,7 +6,7 @@ the dashboard. See https://gpu.ai/docs for details.
 This Python package is automatically generated by the [OpenAPI Generator](https://openapi-generator.tech) project:
 
 - API version: 1.0.0
-- Package version: 0.4.1
+- Package version: 0.4.2
 - Generator version: 7.24.0
 - Build package: org.openapitools.codegen.languages.PythonClientCodegen
 
@@ -170,6 +170,7 @@ Class | Method | HTTP request | Description
 *InstancesApi* | [**create_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#create_instance) | **POST** /instances | Create an instance (async — returns 202 + Operation-Id header)
 *InstancesApi* | [**delete_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#delete_instance) | **DELETE** /instances/{id} | Terminate an instance (idempotent)
 *InstancesApi* | [**get_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#get_instance) | **GET** /instances/{id} | Get an instance
+*InstancesApi* | [**get_instance_gpu_suggestion**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#get_instance_gpu_suggestion) | **GET** /instances/{id}/gpu-suggestion | Suggest a lower-priced GPU for an instance
 *InstancesApi* | [**list_instances**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#list_instances) | **GET** /instances | List instances
 *InstancesApi* | [**update_instance**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancesApi.md#update_instance) | **PATCH** /instances/{id} | Update an instance (rename only in v1)
 *MarketApi* | [**get_market_history**](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/MarketApi.md#get_market_history) | **GET** /market-history | Get a GPU's price and availability over time
@@ -255,6 +256,9 @@ Class | Method | HTTP request | Description
  - [GPUType](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GPUType.md)
  - [GPUTypePage](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GPUTypePage.md)
  - [GetHealth200Response](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GetHealth200Response.md)
+ - [GpuSuggestion](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GpuSuggestion.md)
+ - [GpuSuggestionCheaper](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GpuSuggestionCheaper.md)
+ - [GpuSuggestionGpu](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/GpuSuggestionGpu.md)
  - [ImagesGenerationsRequest](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/ImagesGenerationsRequest.md)
  - [ImagesResponse](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/ImagesResponse.md)
  - [ImagesResponseDataInner](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/ImagesResponseDataInner.md)
@@ -262,6 +266,7 @@ Class | Method | HTTP request | Description
  - [IncludeItem](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/IncludeItem.md)
  - [Instance](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/Instance.md)
  - [InstanceConnection](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstanceConnection.md)
+ - [InstanceGpuMemory](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstanceGpuMemory.md)
  - [InstancePage](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InstancePage.md)
  - [Invoice](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/Invoice.md)
  - [InvoiceList](https://github.com/gpuai-dev/gpuai-python-sdk/blob/main/docs/InvoiceList.md)

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/gpuai_sdk/__init__.py

@@ -15,7 +15,7 @@
 """  # noqa: E501
 
 
-__version__ = "0.4.1"
+__version__ = "0.4.2"
 
 # Define package exports
 __all__ = [
@@ -102,6 +102,9 @@ __all__ = [
     "GPUType",
     "GPUTypePage",
     "GetHealth200Response",
+    "GpuSuggestion",
+    "GpuSuggestionCheaper",
+    "GpuSuggestionGpu",
     "ImagesGenerationsRequest",
     "ImagesResponse",
     "ImagesResponseDataInner",
@@ -109,6 +112,7 @@ __all__ = [
     "IncludeItem",
     "Instance",
     "InstanceConnection",
+    "InstanceGpuMemory",
     "InstancePage",
     "Invoice",
     "InvoiceList",
@@ -248,6 +252,9 @@ from gpuai_sdk.models.function_version import FunctionVersion as FunctionVersion
 from gpuai_sdk.models.gpu_type import GPUType as GPUType
 from gpuai_sdk.models.gpu_type_page import GPUTypePage as GPUTypePage
 from gpuai_sdk.models.get_health200_response import GetHealth200Response as GetHealth200Response
+from gpuai_sdk.models.gpu_suggestion import GpuSuggestion as GpuSuggestion
+from gpuai_sdk.models.gpu_suggestion_cheaper import GpuSuggestionCheaper as GpuSuggestionCheaper
+from gpuai_sdk.models.gpu_suggestion_gpu import GpuSuggestionGpu as GpuSuggestionGpu
 from gpuai_sdk.models.images_generations_request import ImagesGenerationsRequest as ImagesGenerationsRequest
 from gpuai_sdk.models.images_response import ImagesResponse as ImagesResponse
 from gpuai_sdk.models.images_response_data_inner import ImagesResponseDataInner as ImagesResponseDataInner
@@ -255,6 +262,7 @@ from gpuai_sdk.models.images_response_usage import ImagesResponseUsage as Images
 from gpuai_sdk.models.include_item import IncludeItem as IncludeItem
 from gpuai_sdk.models.instance import Instance as Instance
 from gpuai_sdk.models.instance_connection import InstanceConnection as InstanceConnection
+from gpuai_sdk.models.instance_gpu_memory import InstanceGpuMemory as InstanceGpuMemory
 from gpuai_sdk.models.instance_page import InstancePage as InstancePage
 from gpuai_sdk.models.invoice import Invoice as Invoice
 from gpuai_sdk.models.invoice_list import InvoiceList as InvoiceList

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/gpuai_sdk/api/instances_api.py

@@ -20,6 +20,7 @@ from pydantic import Field, StrictStr, field_validator
 from typing import Optional
 from typing_extensions import Annotated
 from gpuai_sdk.models.create_instance_request import CreateInstanceRequest
+from gpuai_sdk.models.gpu_suggestion import GpuSuggestion
 from gpuai_sdk.models.instance import Instance
 from gpuai_sdk.models.instance_page import InstancePage
 from gpuai_sdk.models.operation import Operation
@@ -907,6 +908,277 @@ class InstancesApi:
 
 
 
+    @validate_call
+    def get_instance_gpu_suggestion(
+        self,
+        id: StrictStr,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> GpuSuggestion:
+        """Suggest a lower-priced GPU for an instance
+
+        Compares the highest GPU memory this instance has used (its `gpu_memory` peak, plus headroom) with the GPU models currently listed, and names the cheapest model that would have fit when the saving is worth a look. It never changes anything: it reports what the telemetry shows. Memory use can grow later, so treat it as a starting point.  `status` is `insufficient_data` until the instance has about an hour of telemetry, `no_cheaper_fit` when no cheaper listed model covers the peak, and `suggestion` when one does. Requires the `instances:read` scope. An instance owned by another organization returns 404. 
+
+        :param id: (required)
+        :type id: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._get_instance_gpu_suggestion_serialize(
+            id=id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "GpuSuggestion",
+            '404': "Problem",
+            '429': "Problem",
+            'default': "Problem",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def get_instance_gpu_suggestion_with_http_info(
+        self,
+        id: StrictStr,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[GpuSuggestion]:
+        """Suggest a lower-priced GPU for an instance
+
+        Compares the highest GPU memory this instance has used (its `gpu_memory` peak, plus headroom) with the GPU models currently listed, and names the cheapest model that would have fit when the saving is worth a look. It never changes anything: it reports what the telemetry shows. Memory use can grow later, so treat it as a starting point.  `status` is `insufficient_data` until the instance has about an hour of telemetry, `no_cheaper_fit` when no cheaper listed model covers the peak, and `suggestion` when one does. Requires the `instances:read` scope. An instance owned by another organization returns 404. 
+
+        :param id: (required)
+        :type id: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._get_instance_gpu_suggestion_serialize(
+            id=id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "GpuSuggestion",
+            '404': "Problem",
+            '429': "Problem",
+            'default': "Problem",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def get_instance_gpu_suggestion_without_preload_content(
+        self,
+        id: StrictStr,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Suggest a lower-priced GPU for an instance
+
+        Compares the highest GPU memory this instance has used (its `gpu_memory` peak, plus headroom) with the GPU models currently listed, and names the cheapest model that would have fit when the saving is worth a look. It never changes anything: it reports what the telemetry shows. Memory use can grow later, so treat it as a starting point.  `status` is `insufficient_data` until the instance has about an hour of telemetry, `no_cheaper_fit` when no cheaper listed model covers the peak, and `suggestion` when one does. Requires the `instances:read` scope. An instance owned by another organization returns 404. 
+
+        :param id: (required)
+        :type id: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._get_instance_gpu_suggestion_serialize(
+            id=id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "GpuSuggestion",
+            '404': "Problem",
+            '429': "Problem",
+            'default': "Problem",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _get_instance_gpu_suggestion_serialize(
+        self,
+        id,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if id is not None:
+            _path_params['id'] = id
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json', 
+                    'application/problem+json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'bearerAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/instances/{id}/gpu-suggestion',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def list_instances(
         self,

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/gpuai_sdk/api_client.py

@@ -92,7 +92,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'gpuai-sdk-python/0.4.1'
+        self.user_agent = 'gpuai-sdk-python/0.4.2'
         self.client_side_validation = configuration.client_side_validation
 
     def __enter__(self):

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/gpuai_sdk/configuration.py

@@ -552,7 +552,7 @@ class Configuration:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: 1.0.0\n"\
-               "SDK Package Version: 0.4.1".\
+               "SDK Package Version: 0.4.2".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

{gpuai_sdk-0.4.1 → gpuai_sdk-0.4.2}/gpuai_sdk/models/__init__.py

@@ -69,6 +69,9 @@ from gpuai_sdk.models.function_version import FunctionVersion
 from gpuai_sdk.models.gpu_type import GPUType
 from gpuai_sdk.models.gpu_type_page import GPUTypePage
 from gpuai_sdk.models.get_health200_response import GetHealth200Response
+from gpuai_sdk.models.gpu_suggestion import GpuSuggestion
+from gpuai_sdk.models.gpu_suggestion_cheaper import GpuSuggestionCheaper
+from gpuai_sdk.models.gpu_suggestion_gpu import GpuSuggestionGpu
 from gpuai_sdk.models.images_generations_request import ImagesGenerationsRequest
 from gpuai_sdk.models.images_response import ImagesResponse
 from gpuai_sdk.models.images_response_data_inner import ImagesResponseDataInner
@@ -76,6 +79,7 @@ from gpuai_sdk.models.images_response_usage import ImagesResponseUsage
 from gpuai_sdk.models.include_item import IncludeItem
 from gpuai_sdk.models.instance import Instance
 from gpuai_sdk.models.instance_connection import InstanceConnection
+from gpuai_sdk.models.instance_gpu_memory import InstanceGpuMemory
 from gpuai_sdk.models.instance_page import InstancePage
 from gpuai_sdk.models.invoice import Invoice
 from gpuai_sdk.models.invoice_list import InvoiceList

gpuai_sdk-0.4.2/gpuai_sdk/models/gpu_suggestion.py

@@ -0,0 +1,131 @@
+# coding: utf-8
+
+"""
+    GPU.ai Public Developer API
+
+    Programmatic API for GPU.ai. Authenticate with API keys minted from the dashboard. See https://gpu.ai/docs for details. 
+
+    The version of the OpenAPI document: 1.0.0
+    Contact: support@gpu.ai
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from gpuai_sdk.models.gpu_suggestion_cheaper import GpuSuggestionCheaper
+from gpuai_sdk.models.gpu_suggestion_gpu import GpuSuggestionGpu
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class GpuSuggestion(BaseModel):
+    """
+    Whether a lower-priced GPU model would have fit an instance's observed peak GPU memory. It names GPU models and prices only. 
+    """ # noqa: E501
+    status: StrictStr = Field(description="`suggestion` when a cheaper model fits and the saving is at least 10% and $0.05 per hour. `insufficient_data` until the instance has at least 30 samples over at least an hour. `no_cheaper_fit` otherwise. ")
+    peak_used_mb: Optional[StrictInt] = Field(description="Highest GPU memory seen, in MiB. Null with no telemetry.")
+    required_mb: Optional[StrictInt] = Field(description="The peak plus headroom (25% and 1024 MiB), in MiB: the memory a suggested model must have. Null until there is enough telemetry. ")
+    samples: StrictInt
+    observed_minutes: StrictInt = Field(description="Minutes between the first and the latest telemetry sample.")
+    current: GpuSuggestionGpu
+    suggestion: Optional[GpuSuggestionCheaper] = Field(description="Set only when status is `suggestion`.")
+    __properties: ClassVar[List[str]] = ["status", "peak_used_mb", "required_mb", "samples", "observed_minutes", "current", "suggestion"]
+
+    @field_validator('status')
+    def status_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['suggestion', 'insufficient_data', 'no_cheaper_fit']):
+            raise ValueError("must be one of enum values ('suggestion', 'insufficient_data', 'no_cheaper_fit')")
+        return value
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of GpuSuggestion from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of current
+        if self.current:
+            _dict['current'] = self.current.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of suggestion
+        if self.suggestion:
+            _dict['suggestion'] = self.suggestion.to_dict()
+        # set to None if peak_used_mb (nullable) is None
+        # and model_fields_set contains the field
+        if self.peak_used_mb is None and "peak_used_mb" in self.model_fields_set:
+            _dict['peak_used_mb'] = None
+
+        # set to None if required_mb (nullable) is None
+        # and model_fields_set contains the field
+        if self.required_mb is None and "required_mb" in self.model_fields_set:
+            _dict['required_mb'] = None
+
+        # set to None if suggestion (nullable) is None
+        # and model_fields_set contains the field
+        if self.suggestion is None and "suggestion" in self.model_fields_set:
+            _dict['suggestion'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of GpuSuggestion from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "status": obj.get("status"),
+            "peak_used_mb": obj.get("peak_used_mb"),
+            "required_mb": obj.get("required_mb"),
+            "samples": obj.get("samples"),
+            "observed_minutes": obj.get("observed_minutes"),
+            "current": GpuSuggestionGpu.from_dict(obj["current"]) if obj.get("current") is not None else None,
+            "suggestion": GpuSuggestionCheaper.from_dict(obj["suggestion"]) if obj.get("suggestion") is not None else None
+        })
+        return _obj
+
+