eval-studio-client 1.0.3a1__py3-none-any.whl → 1.1.0__py3-none-any.whl

eval_studio_client/api/api/operation_service_api.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
@@ -21,6 +21,7 @@ from typing import List, Optional
 from typing_extensions import Annotated
 from eval_studio_client.api.models.required_the_operation_to_finalize import RequiredTheOperationToFinalize
 from eval_studio_client.api.models.required_the_operation_to_update import RequiredTheOperationToUpdate
+from eval_studio_client.api.models.v1_abort_operation_response import V1AbortOperationResponse
 from eval_studio_client.api.models.v1_batch_get_operations_response import V1BatchGetOperationsResponse
 from eval_studio_client.api.models.v1_finalize_operation_response import V1FinalizeOperationResponse
 from eval_studio_client.api.models.v1_get_operation_response import V1GetOperationResponse
@@ -45,6 +46,260 @@ class OperationServiceApi:
         self.api_client = api_client
 
 
+    @validate_call
+    def operation_service_abort_operation(
+        self,
+        name: Annotated[str, Field(strict=True, description="Required. The name of the Operation to abort.")],
+        _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,
+    ) -> V1AbortOperationResponse:
+        """operation_service_abort_operation
+
+
+        :param name: Required. The name of the Operation to abort. (required)
+        :type name: 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._operation_service_abort_operation_serialize(
+            name=name,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "V1AbortOperationResponse",
+        }
+        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 operation_service_abort_operation_with_http_info(
+        self,
+        name: Annotated[str, Field(strict=True, description="Required. The name of the Operation to abort.")],
+        _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[V1AbortOperationResponse]:
+        """operation_service_abort_operation
+
+
+        :param name: Required. The name of the Operation to abort. (required)
+        :type name: 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._operation_service_abort_operation_serialize(
+            name=name,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "V1AbortOperationResponse",
+        }
+        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 operation_service_abort_operation_without_preload_content(
+        self,
+        name: Annotated[str, Field(strict=True, description="Required. The name of the Operation to abort.")],
+        _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:
+        """operation_service_abort_operation
+
+
+        :param name: Required. The name of the Operation to abort. (required)
+        :type name: 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._operation_service_abort_operation_serialize(
+            name=name,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "V1AbortOperationResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _operation_service_abort_operation_serialize(
+        self,
+        name,
+        _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]] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if name is not None:
+            _path_params['name'] = name
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        _header_params['Accept'] = self.api_client.select_header_accept(
+            [
+                'application/json'
+            ]
+        )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+        ]
+
+        return self.api_client.param_serialize(
+            method='POST',
+            resource_path='/v1/{name}:abort',
+            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 operation_service_batch_get_operations(
         self,
@@ -587,7 +842,7 @@ class OperationServiceApi:
     @validate_call
     def operation_service_get_operation(
         self,
-        name_5: Annotated[str, Field(strict=True, description="Required. The name of the Operation to retrieve.")],
+        name_6: Annotated[str, Field(strict=True, description="Required. The name of the Operation to retrieve.")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -604,8 +859,8 @@ class OperationServiceApi:
         """operation_service_get_operation
 
 
-        :param name_5: Required. The name of the Operation to retrieve. (required)
-        :type name_5: str
+        :param name_6: Required. The name of the Operation to retrieve. (required)
+        :type name_6: 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
@@ -629,7 +884,7 @@ class OperationServiceApi:
         """ # noqa: E501
 
         _param = self._operation_service_get_operation_serialize(
-            name_5=name_5,
+            name_6=name_6,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -653,7 +908,7 @@ class OperationServiceApi:
     @validate_call
     def operation_service_get_operation_with_http_info(
         self,
-        name_5: Annotated[str, Field(strict=True, description="Required. The name of the Operation to retrieve.")],
+        name_6: Annotated[str, Field(strict=True, description="Required. The name of the Operation to retrieve.")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -670,8 +925,8 @@ class OperationServiceApi:
         """operation_service_get_operation
 
 
-        :param name_5: Required. The name of the Operation to retrieve. (required)
-        :type name_5: str
+        :param name_6: Required. The name of the Operation to retrieve. (required)
+        :type name_6: 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
@@ -695,7 +950,7 @@ class OperationServiceApi:
         """ # noqa: E501
 
         _param = self._operation_service_get_operation_serialize(
-            name_5=name_5,
+            name_6=name_6,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -719,7 +974,7 @@ class OperationServiceApi:
     @validate_call
     def operation_service_get_operation_without_preload_content(
         self,
-        name_5: Annotated[str, Field(strict=True, description="Required. The name of the Operation to retrieve.")],
+        name_6: Annotated[str, Field(strict=True, description="Required. The name of the Operation to retrieve.")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -736,8 +991,8 @@ class OperationServiceApi:
         """operation_service_get_operation
 
 
-        :param name_5: Required. The name of the Operation to retrieve. (required)
-        :type name_5: str
+        :param name_6: Required. The name of the Operation to retrieve. (required)
+        :type name_6: 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
@@ -761,7 +1016,7 @@ class OperationServiceApi:
         """ # noqa: E501
 
         _param = self._operation_service_get_operation_serialize(
-            name_5=name_5,
+            name_6=name_6,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -780,7 +1035,7 @@ class OperationServiceApi:
 
     def _operation_service_get_operation_serialize(
         self,
-        name_5,
+        name_6,
         _request_auth,
         _content_type,
         _headers,
@@ -800,8 +1055,8 @@ class OperationServiceApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if name_5 is not None:
-            _path_params['name_5'] = name_5
+        if name_6 is not None:
+            _path_params['name_6'] = name_6
         # process the query parameters
         # process the header parameters
         # process the form parameters
@@ -822,7 +1077,7 @@ class OperationServiceApi:
 
         return self.api_client.param_serialize(
             method='GET',
-            resource_path='/v1/{name_5}',
+            resource_path='/v1/{name_6}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,

eval_studio_client/api/api/perturbation_service_api.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)