crypticorn 2.16.0__py3-none-any.whl → 2.17.0__py3-none-any.whl

crypticorn/hive/client/api/data_api.py

@@ -11,14 +11,10 @@ Generated by OpenAPI Generator (https://openapi-generator.tech)
 Do not edit the class manually.
 """  # noqa: E501
 
-import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from pydantic import Field, StrictInt
-from typing import Optional
-from typing_extensions import Annotated
 from crypticorn.hive.client.models.data_download_response import DataDownloadResponse
 from crypticorn.hive.client.models.data_info import DataInfo
 from crypticorn.hive.client.models.data_version import DataVersion
@@ -28,6 +24,24 @@ from crypticorn.hive.client.api_client import ApiClient, RequestSerialized
 from crypticorn.hive.client.api_response import ApiResponse
 from crypticorn.hive.client.rest import RESTResponseType
 
+# Import async_to_sync for sync methods
+try:
+    from asgiref.sync import async_to_sync
+
+    _HAS_ASGIREF = True
+except ImportError:
+    _HAS_ASGIREF = False
+
+    def async_to_sync(async_func):
+        """Fallback decorator that raises an error if asgiref is not available."""
+
+        def wrapper(*args, **kwargs):
+            raise ImportError(
+                "asgiref is required for sync methods. Install with: pip install asgiref"
+            )
+
+        return wrapper
+
 
 class DataApi:
     """NOTE: This class is auto generated by OpenAPI Generator
@@ -36,13 +50,180 @@ class DataApi:
     Do not edit the class manually.
     """
 
-    def __init__(self, api_client=None) -> None:
+    def __init__(self, api_client=None, is_sync: bool = False) -> None:
         if api_client is None:
             api_client = ApiClient.get_default()
         self.api_client = api_client
+        self.is_sync = is_sync
 
     @validate_call
-    async def download_data(
+    def download_data(
+        self,
+        model_id: Annotated[
+            StrictInt, Field(description="The ID of the model to download data for.")
+        ],
+        version: Annotated[
+            Optional[DataVersion],
+            Field(
+                description="The version of the data to download. Default is the latest public version."
+            ),
+        ] = None,
+        feature_size: Annotated[
+            Optional[FeatureSize],
+            Field(
+                description="The number of features in the data. Default is `LARGE`."
+            ),
+        ] = None,
+        _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,
+    ) -> DataDownloadResponse:
+        """Download Data"""
+        if self.is_sync:
+            return self._download_data_sync(
+                model_id=model_id,
+                version=version,
+                feature_size=feature_size,
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+        else:
+            return self._download_data_async(
+                model_id=model_id,
+                version=version,
+                feature_size=feature_size,
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+    @validate_call
+    def download_data_with_http_info(
+        self,
+        model_id: Annotated[
+            StrictInt, Field(description="The ID of the model to download data for.")
+        ],
+        version: Annotated[
+            Optional[DataVersion],
+            Field(
+                description="The version of the data to download. Default is the latest public version."
+            ),
+        ] = None,
+        feature_size: Annotated[
+            Optional[FeatureSize],
+            Field(
+                description="The number of features in the data. Default is `LARGE`."
+            ),
+        ] = None,
+        _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[DataDownloadResponse]:
+        """Download Data with HTTP info"""
+        if self.is_sync:
+            return self._download_data_sync_with_http_info(
+                model_id=model_id,
+                version=version,
+                feature_size=feature_size,
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+        else:
+            return self._download_data_async_with_http_info(
+                model_id=model_id,
+                version=version,
+                feature_size=feature_size,
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+    @validate_call
+    def download_data_without_preload_content(
+        self,
+        model_id: Annotated[
+            StrictInt, Field(description="The ID of the model to download data for.")
+        ],
+        version: Annotated[
+            Optional[DataVersion],
+            Field(
+                description="The version of the data to download. Default is the latest public version."
+            ),
+        ] = None,
+        feature_size: Annotated[
+            Optional[FeatureSize],
+            Field(
+                description="The number of features in the data. Default is `LARGE`."
+            ),
+        ] = None,
+        _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:
+        """Download Data without preloading content"""
+        if self.is_sync:
+            return self._download_data_sync_without_preload_content(
+                model_id=model_id,
+                version=version,
+                feature_size=feature_size,
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+        else:
+            return self._download_data_async_without_preload_content(
+                model_id=model_id,
+                version=version,
+                feature_size=feature_size,
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+    # Private async implementation methods
+    @validate_call
+    async def _download_data_async(
         self,
         model_id: Annotated[
             StrictInt, Field(description="The ID of the model to download data for.")
@@ -126,7 +307,7 @@ class DataApi:
         ).data
 
     @validate_call
-    async def download_data_with_http_info(
+    async def _download_data_async_with_http_info(
         self,
         model_id: Annotated[
             StrictInt, Field(description="The ID of the model to download data for.")
@@ -205,12 +386,11 @@ class DataApi:
         )
         await response_data.read()
         return self.api_client.response_deserialize(
-            response_data=response_data,
-            response_types_map=_response_types_map,
+            response_data=response_data, response_types_map=_response_types_map
         )
 
     @validate_call
-    async def download_data_without_preload_content(
+    async def _download_data_async_without_preload_content(
         self,
         model_id: Annotated[
             StrictInt, Field(description="The ID of the model to download data for.")
@@ -287,7 +467,134 @@ class DataApi:
         response_data = await self.api_client.call_api(
             *_param, _request_timeout=_request_timeout
         )
-        return response_data.response
+        return response_data
+
+    # Private sync implementation methods
+    @validate_call
+    def _download_data_sync(
+        self,
+        model_id: Annotated[
+            StrictInt, Field(description="The ID of the model to download data for.")
+        ],
+        version: Annotated[
+            Optional[DataVersion],
+            Field(
+                description="The version of the data to download. Default is the latest public version."
+            ),
+        ] = None,
+        feature_size: Annotated[
+            Optional[FeatureSize],
+            Field(
+                description="The number of features in the data. Default is `LARGE`."
+            ),
+        ] = None,
+        _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,
+    ) -> DataDownloadResponse:
+        """Synchronous version of download_data"""
+        return async_to_sync(self._download_data_async)(
+            model_id=model_id,
+            version=version,
+            feature_size=feature_size,
+            _request_timeout=_request_timeout,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+    @validate_call
+    def _download_data_sync_with_http_info(
+        self,
+        model_id: Annotated[
+            StrictInt, Field(description="The ID of the model to download data for.")
+        ],
+        version: Annotated[
+            Optional[DataVersion],
+            Field(
+                description="The version of the data to download. Default is the latest public version."
+            ),
+        ] = None,
+        feature_size: Annotated[
+            Optional[FeatureSize],
+            Field(
+                description="The number of features in the data. Default is `LARGE`."
+            ),
+        ] = None,
+        _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[DataDownloadResponse]:
+        """Synchronous version of download_data_with_http_info"""
+        return async_to_sync(self._download_data_async_with_http_info)(
+            model_id=model_id,
+            version=version,
+            feature_size=feature_size,
+            _request_timeout=_request_timeout,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+    @validate_call
+    def _download_data_sync_without_preload_content(
+        self,
+        model_id: Annotated[
+            StrictInt, Field(description="The ID of the model to download data for.")
+        ],
+        version: Annotated[
+            Optional[DataVersion],
+            Field(
+                description="The version of the data to download. Default is the latest public version."
+            ),
+        ] = None,
+        feature_size: Annotated[
+            Optional[FeatureSize],
+            Field(
+                description="The number of features in the data. Default is `LARGE`."
+            ),
+        ] = None,
+        _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:
+        """Synchronous version of download_data_without_preload_content"""
+        return async_to_sync(self._download_data_async_without_preload_content)(
+            model_id=model_id,
+            version=version,
+            feature_size=feature_size,
+            _request_timeout=_request_timeout,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
 
     def _download_data_serialize(
         self,
@@ -356,7 +663,110 @@ class DataApi:
         )
 
     @validate_call
-    async def get_data_info(
+    def get_data_info(
+        self,
+        _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,
+    ) -> DataInfo:
+        """Get Data Info"""
+        if self.is_sync:
+            return self._get_data_info_sync(
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+        else:
+            return self._get_data_info_async(
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+    @validate_call
+    def get_data_info_with_http_info(
+        self,
+        _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[DataInfo]:
+        """Get Data Info with HTTP info"""
+        if self.is_sync:
+            return self._get_data_info_sync_with_http_info(
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+        else:
+            return self._get_data_info_async_with_http_info(
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+    @validate_call
+    def get_data_info_without_preload_content(
+        self,
+        _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:
+        """Get Data Info without preloading content"""
+        if self.is_sync:
+            return self._get_data_info_sync_without_preload_content(
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+        else:
+            return self._get_data_info_async_without_preload_content(
+                _request_timeout=_request_timeout,
+                _request_auth=_request_auth,
+                _content_type=_content_type,
+                _headers=_headers,
+                _host_index=_host_index,
+            )
+
+    # Private async implementation methods
+    @validate_call
+    async def _get_data_info_async(
         self,
         _request_timeout: Union[
             None,
@@ -416,7 +826,7 @@ class DataApi:
         ).data
 
     @validate_call
-    async def get_data_info_with_http_info(
+    async def _get_data_info_async_with_http_info(
         self,
         _request_timeout: Union[
             None,
@@ -471,12 +881,11 @@ class DataApi:
         )
         await response_data.read()
         return self.api_client.response_deserialize(
-            response_data=response_data,
-            response_types_map=_response_types_map,
+            response_data=response_data, response_types_map=_response_types_map
         )
 
     @validate_call
-    async def get_data_info_without_preload_content(
+    async def _get_data_info_async_without_preload_content(
         self,
         _request_timeout: Union[
             None,
@@ -529,7 +938,80 @@ class DataApi:
         response_data = await self.api_client.call_api(
             *_param, _request_timeout=_request_timeout
         )
-        return response_data.response
+        return response_data
+
+    # Private sync implementation methods
+    @validate_call
+    def _get_data_info_sync(
+        self,
+        _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,
+    ) -> DataInfo:
+        """Synchronous version of get_data_info"""
+        return async_to_sync(self._get_data_info_async)(
+            _request_timeout=_request_timeout,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+    @validate_call
+    def _get_data_info_sync_with_http_info(
+        self,
+        _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[DataInfo]:
+        """Synchronous version of get_data_info_with_http_info"""
+        return async_to_sync(self._get_data_info_async_with_http_info)(
+            _request_timeout=_request_timeout,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+    @validate_call
+    def _get_data_info_sync_without_preload_content(
+        self,
+        _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:
+        """Synchronous version of get_data_info_without_preload_content"""
+        return async_to_sync(self._get_data_info_async_without_preload_content)(
+            _request_timeout=_request_timeout,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
 
     def _get_data_info_serialize(
         self,