crypticorn 2.5.3__py3-none-any.whl → 2.7.0__py3-none-any.whl

crypticorn/hive/client/api/data_api.py

@@ -48,9 +48,10 @@ class DataApi:
         version: Annotated[
             Optional[DataVersion],
             Field(description="Data version. Default is the latest public version."),
-        ],
+        ] = None,
         feature_size: Annotated[
-            Optional[FeatureSize], Field(description="Feature size")
+            Optional[FeatureSize],
+            Field(description="The number of features in the data. Default is LARGE."),
         ] = None,
         _request_timeout: Union[
             None,
@@ -70,9 +71,9 @@ class DataApi:
 
         :param model_id: Model ID (required)
         :type model_id: int
-        :param version: Data version. Default is the latest public version. (required)
+        :param version: Data version. Default is the latest public version.
         :type version: DataVersion
-        :param feature_size: Feature size
+        :param feature_size: The number of features in the data. Default is LARGE.
         :type feature_size: FeatureSize
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
@@ -125,9 +126,10 @@ class DataApi:
         version: Annotated[
             Optional[DataVersion],
             Field(description="Data version. Default is the latest public version."),
-        ],
+        ] = None,
         feature_size: Annotated[
-            Optional[FeatureSize], Field(description="Feature size")
+            Optional[FeatureSize],
+            Field(description="The number of features in the data. Default is LARGE."),
         ] = None,
         _request_timeout: Union[
             None,
@@ -147,9 +149,9 @@ class DataApi:
 
         :param model_id: Model ID (required)
         :type model_id: int
-        :param version: Data version. Default is the latest public version. (required)
+        :param version: Data version. Default is the latest public version.
         :type version: DataVersion
-        :param feature_size: Feature size
+        :param feature_size: The number of features in the data. Default is LARGE.
         :type feature_size: FeatureSize
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
@@ -202,9 +204,10 @@ class DataApi:
         version: Annotated[
             Optional[DataVersion],
             Field(description="Data version. Default is the latest public version."),
-        ],
+        ] = None,
         feature_size: Annotated[
-            Optional[FeatureSize], Field(description="Feature size")
+            Optional[FeatureSize],
+            Field(description="The number of features in the data. Default is LARGE."),
         ] = None,
         _request_timeout: Union[
             None,
@@ -224,9 +227,9 @@ class DataApi:
 
         :param model_id: Model ID (required)
         :type model_id: int
-        :param version: Data version. Default is the latest public version. (required)
+        :param version: Data version. Default is the latest public version.
         :type version: DataVersion
-        :param feature_size: Feature size
+        :param feature_size: The number of features in the data. Default is LARGE.
         :type feature_size: FeatureSize
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request

crypticorn/hive/client/api/models_api.py

@@ -16,7 +16,7 @@ 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, StrictStr
+from pydantic import Field, StrictBool, StrictInt, StrictStr
 from typing import Any, List, Optional
 from typing_extensions import Annotated
 from crypticorn.hive.client.models.data_version import DataVersion
@@ -829,8 +829,9 @@ class ModelsApi:
         )
 
     @validate_call
-    async def get_all_models(
+    async def get_model(
         self,
+        id: Annotated[StrictInt, Field(description="Model ID to get")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -842,11 +843,13 @@ class ModelsApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> List[Model]:
-        """Get All Models
+    ) -> Model:
+        """Get Model
 
-        List all models
+        Get a model by ID
 
+        :param id: Model ID to get (required)
+        :type id: int
         :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
@@ -869,7 +872,8 @@ class ModelsApi:
         :return: Returns the result object.
         """  # noqa: E501
 
-        _param = self._get_all_models_serialize(
+        _param = self._get_model_serialize(
+            id=id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -877,7 +881,7 @@ class ModelsApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "List[Model]",
+            "200": "Model",
         }
         response_data = await self.api_client.call_api(
             *_param, _request_timeout=_request_timeout
@@ -889,8 +893,9 @@ class ModelsApi:
         ).data
 
     @validate_call
-    async def get_all_models_with_http_info(
+    async def get_model_with_http_info(
         self,
+        id: Annotated[StrictInt, Field(description="Model ID to get")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -902,11 +907,13 @@ class ModelsApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[List[Model]]:
-        """Get All Models
+    ) -> ApiResponse[Model]:
+        """Get Model
 
-        List all models
+        Get a model by ID
 
+        :param id: Model ID to get (required)
+        :type id: int
         :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
@@ -929,7 +936,8 @@ class ModelsApi:
         :return: Returns the result object.
         """  # noqa: E501
 
-        _param = self._get_all_models_serialize(
+        _param = self._get_model_serialize(
+            id=id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -937,7 +945,7 @@ class ModelsApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "List[Model]",
+            "200": "Model",
         }
         response_data = await self.api_client.call_api(
             *_param, _request_timeout=_request_timeout
@@ -949,8 +957,9 @@ class ModelsApi:
         )
 
     @validate_call
-    async def get_all_models_without_preload_content(
+    async def get_model_without_preload_content(
         self,
+        id: Annotated[StrictInt, Field(description="Model ID to get")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -963,10 +972,12 @@ class ModelsApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Get All Models
+        """Get Model
 
-        List all models
+        Get a model by ID
 
+        :param id: Model ID to get (required)
+        :type id: int
         :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
@@ -989,7 +1000,8 @@ class ModelsApi:
         :return: Returns the result object.
         """  # noqa: E501
 
-        _param = self._get_all_models_serialize(
+        _param = self._get_model_serialize(
+            id=id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -997,15 +1009,16 @@ class ModelsApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "List[Model]",
+            "200": "Model",
         }
         response_data = await self.api_client.call_api(
             *_param, _request_timeout=_request_timeout
         )
         return response_data.response
 
-    def _get_all_models_serialize(
+    def _get_model_serialize(
         self,
+        id,
         _request_auth,
         _content_type,
         _headers,
@@ -1026,6 +1039,8 @@ class ModelsApi:
         _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
@@ -1042,7 +1057,7 @@ class ModelsApi:
 
         return self.api_client.param_serialize(
             method="GET",
-            resource_path="/model/all",
+            resource_path="/model/{id}",
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -1056,10 +1071,9 @@ class ModelsApi:
         )
 
     @validate_call
-    async def get_model(
+    async def get_model_by_name(
         self,
-        id: Optional[StrictInt] = None,
-        name: Optional[StrictStr] = None,
+        name: Annotated[StrictStr, Field(description="Model name to get")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1072,13 +1086,11 @@ class ModelsApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> Model:
-        """Get Model
+        """Get Model By Name
 
-        Get a model by ID or name
+        Get a model by name
 
-        :param id:
-        :type id: int
-        :param name:
+        :param name: Model name to get (required)
         :type name: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
@@ -1102,8 +1114,7 @@ class ModelsApi:
         :return: Returns the result object.
         """  # noqa: E501
 
-        _param = self._get_model_serialize(
-            id=id,
+        _param = self._get_model_by_name_serialize(
             name=name,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -1124,10 +1135,9 @@ class ModelsApi:
         ).data
 
     @validate_call
-    async def get_model_with_http_info(
+    async def get_model_by_name_with_http_info(
         self,
-        id: Optional[StrictInt] = None,
-        name: Optional[StrictStr] = None,
+        name: Annotated[StrictStr, Field(description="Model name to get")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1140,13 +1150,11 @@ class ModelsApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> ApiResponse[Model]:
-        """Get Model
+        """Get Model By Name
 
-        Get a model by ID or name
+        Get a model by name
 
-        :param id:
-        :type id: int
-        :param name:
+        :param name: Model name to get (required)
         :type name: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
@@ -1170,8 +1178,7 @@ class ModelsApi:
         :return: Returns the result object.
         """  # noqa: E501
 
-        _param = self._get_model_serialize(
-            id=id,
+        _param = self._get_model_by_name_serialize(
             name=name,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -1192,10 +1199,9 @@ class ModelsApi:
         )
 
     @validate_call
-    async def get_model_without_preload_content(
+    async def get_model_by_name_without_preload_content(
         self,
-        id: Optional[StrictInt] = None,
-        name: Optional[StrictStr] = None,
+        name: Annotated[StrictStr, Field(description="Model name to get")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1208,13 +1214,11 @@ class ModelsApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Get Model
+        """Get Model By Name
 
-        Get a model by ID or name
+        Get a model by name
 
-        :param id:
-        :type id: int
-        :param name:
+        :param name: Model name to get (required)
         :type name: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
@@ -1238,8 +1242,7 @@ class ModelsApi:
         :return: Returns the result object.
         """  # noqa: E501
 
-        _param = self._get_model_serialize(
-            id=id,
+        _param = self._get_model_by_name_serialize(
             name=name,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -1255,9 +1258,8 @@ class ModelsApi:
         )
         return response_data.response
 
-    def _get_model_serialize(
+    def _get_model_by_name_serialize(
         self,
-        id,
         name,
         _request_auth,
         _content_type,
@@ -1279,14 +1281,299 @@ class ModelsApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
+        if name is not None:
+            _path_params["name"] = name
         # process the query parameters
-        if id is not None:
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
 
-            _query_params.append(("id", id))
+        # set the HTTP header `Accept`
+        if "Accept" not in _header_params:
+            _header_params["Accept"] = self.api_client.select_header_accept(
+                ["application/json"]
+            )
 
-        if name is not None:
+        # authentication setting
+        _auth_settings: List[str] = ["APIKeyHeader", "HTTPBearer"]
+
+        return self.api_client.param_serialize(
+            method="GET",
+            resource_path="/model/by-name/{name}",
+            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
+    async def get_models(
+        self,
+        by_user: Annotated[
+            Optional[StrictBool],
+            Field(
+                description="Whether to get models by user. Else all models are returned."
+            ),
+        ] = None,
+        user_id: Annotated[
+            Optional[StrictStr],
+            Field(
+                description="User ID to get models for. Only used if by_user is true. Default is current user."
+            ),
+        ] = 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,
+    ) -> List[Model]:
+        """Get All Models
+
+        List all models
+
+        :param by_user: Whether to get models by user. Else all models are returned.
+        :type by_user: bool
+        :param user_id: User ID to get models for. Only used if by_user is true. Default is current user.
+        :type user_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_models_serialize(
+            by_user=by_user,
+            user_id=user_id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "List[Model]",
+        }
+        response_data = await self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        await response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    @validate_call
+    async def get_models_with_http_info(
+        self,
+        by_user: Annotated[
+            Optional[StrictBool],
+            Field(
+                description="Whether to get models by user. Else all models are returned."
+            ),
+        ] = None,
+        user_id: Annotated[
+            Optional[StrictStr],
+            Field(
+                description="User ID to get models for. Only used if by_user is true. Default is current user."
+            ),
+        ] = 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[List[Model]]:
+        """Get All Models
+
+        List all models
+
+        :param by_user: Whether to get models by user. Else all models are returned.
+        :type by_user: bool
+        :param user_id: User ID to get models for. Only used if by_user is true. Default is current user.
+        :type user_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_models_serialize(
+            by_user=by_user,
+            user_id=user_id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "List[Model]",
+        }
+        response_data = await self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        await response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+    @validate_call
+    async def get_models_without_preload_content(
+        self,
+        by_user: Annotated[
+            Optional[StrictBool],
+            Field(
+                description="Whether to get models by user. Else all models are returned."
+            ),
+        ] = None,
+        user_id: Annotated[
+            Optional[StrictStr],
+            Field(
+                description="User ID to get models for. Only used if by_user is true. Default is current user."
+            ),
+        ] = 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:
+        """Get All Models
+
+        List all models
+
+        :param by_user: Whether to get models by user. Else all models are returned.
+        :type by_user: bool
+        :param user_id: User ID to get models for. Only used if by_user is true. Default is current user.
+        :type user_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_models_serialize(
+            by_user=by_user,
+            user_id=user_id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "List[Model]",
+        }
+        response_data = await self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+    def _get_models_serialize(
+        self,
+        by_user,
+        user_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
+        # process the query parameters
+        if by_user is not None:
+
+            _query_params.append(("by_user", by_user))
+
+        if user_id is not None:
 
-            _query_params.append(("name", name))
+            _query_params.append(("user_id", user_id))
 
         # process the header parameters
         # process the form parameters

crypticorn/hive/client/models/__init__.py

@@ -21,6 +21,7 @@ from crypticorn.hive.client.models.data_value_value_value_inner import (
     DataValueValueValueInner,
 )
 from crypticorn.hive.client.models.data_version import DataVersion
+from crypticorn.hive.client.models.data_version_info import DataVersionInfo
 from crypticorn.hive.client.models.download_links import DownloadLinks
 from crypticorn.hive.client.models.evaluation import Evaluation
 from crypticorn.hive.client.models.evaluation_response import EvaluationResponse
@@ -31,4 +32,5 @@ from crypticorn.hive.client.models.model_create import ModelCreate
 from crypticorn.hive.client.models.model_status import ModelStatus
 from crypticorn.hive.client.models.model_update import ModelUpdate
 from crypticorn.hive.client.models.target import Target
+from crypticorn.hive.client.models.target_info import TargetInfo
 from crypticorn.hive.client.models.target_type import TargetType