yandexcloud 0.338.0__py3-none-any.whl → 0.340.0__py3-none-any.whl

yandex/cloud/ai/tuning/v1/tuning_service_pb2.pyi

@@ -827,3 +827,223 @@ class ListErrorsResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing.Literal["tuning_error", b"tuning_error"]) -> None: ...
 
 global___ListErrorsResponse = ListErrorsResponse
+
+@typing.final
+class CreateTuningDraftRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class LabelsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    BASE_MODEL_URI_FIELD_NUMBER: builtins.int
+    TRAIN_DATASETS_FIELD_NUMBER: builtins.int
+    VALIDATION_DATASETS_FIELD_NUMBER: builtins.int
+    TEXT_TO_TEXT_COMPLETION_FIELD_NUMBER: builtins.int
+    TEXT_CLASSIFICATION_MULTILABEL_FIELD_NUMBER: builtins.int
+    TEXT_CLASSIFICATION_MULTICLASS_FIELD_NUMBER: builtins.int
+    TEXT_EMBEDDING_PAIR_PARAMS_FIELD_NUMBER: builtins.int
+    TEXT_EMBEDDING_TRIPLET_PARAMS_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    LABELS_FIELD_NUMBER: builtins.int
+    base_model_uri: builtins.str
+    name: builtins.str
+    description: builtins.str
+    @property
+    def train_datasets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TuningRequest.WeightedDataset]: ...
+    @property
+    def validation_datasets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TuningRequest.WeightedDataset]: ...
+    @property
+    def text_to_text_completion(self) -> global___TextToTextCompletionTuningParams: ...
+    @property
+    def text_classification_multilabel(self) -> global___TextClassificationMultilabelParams: ...
+    @property
+    def text_classification_multiclass(self) -> global___TextClassificationMulticlassParams: ...
+    @property
+    def text_embedding_pair_params(self) -> global___TextEmbeddingPairParams: ...
+    @property
+    def text_embedding_triplet_params(self) -> global___TextEmbeddingTripletParams: ...
+    @property
+    def labels(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        base_model_uri: builtins.str = ...,
+        train_datasets: collections.abc.Iterable[global___TuningRequest.WeightedDataset] | None = ...,
+        validation_datasets: collections.abc.Iterable[global___TuningRequest.WeightedDataset] | None = ...,
+        text_to_text_completion: global___TextToTextCompletionTuningParams | None = ...,
+        text_classification_multilabel: global___TextClassificationMultilabelParams | None = ...,
+        text_classification_multiclass: global___TextClassificationMulticlassParams | None = ...,
+        text_embedding_pair_params: global___TextEmbeddingPairParams | None = ...,
+        text_embedding_triplet_params: global___TextEmbeddingTripletParams | None = ...,
+        name: builtins.str = ...,
+        description: builtins.str = ...,
+        labels: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["text_classification_multiclass", b"text_classification_multiclass", "text_classification_multilabel", b"text_classification_multilabel", "text_embedding_pair_params", b"text_embedding_pair_params", "text_embedding_triplet_params", b"text_embedding_triplet_params", "text_to_text_completion", b"text_to_text_completion", "tuning_params", b"tuning_params"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["base_model_uri", b"base_model_uri", "description", b"description", "labels", b"labels", "name", b"name", "text_classification_multiclass", b"text_classification_multiclass", "text_classification_multilabel", b"text_classification_multilabel", "text_embedding_pair_params", b"text_embedding_pair_params", "text_embedding_triplet_params", b"text_embedding_triplet_params", "text_to_text_completion", b"text_to_text_completion", "train_datasets", b"train_datasets", "tuning_params", b"tuning_params", "validation_datasets", b"validation_datasets"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["tuning_params", b"tuning_params"]) -> typing.Literal["text_to_text_completion", "text_classification_multilabel", "text_classification_multiclass", "text_embedding_pair_params", "text_embedding_triplet_params"] | None: ...
+
+global___CreateTuningDraftRequest = CreateTuningDraftRequest
+
+@typing.final
+class CreateTuningDraftResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TUNING_TASK_ID_FIELD_NUMBER: builtins.int
+    tuning_task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        tuning_task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tuning_task_id", b"tuning_task_id"]) -> None: ...
+
+global___CreateTuningDraftResponse = CreateTuningDraftResponse
+
+@typing.final
+class UpdateTuningDraftRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class LabelsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    TUNING_TASK_ID_FIELD_NUMBER: builtins.int
+    BASE_MODEL_URI_FIELD_NUMBER: builtins.int
+    TRAIN_DATASETS_FIELD_NUMBER: builtins.int
+    VALIDATION_DATASETS_FIELD_NUMBER: builtins.int
+    TEXT_TO_TEXT_COMPLETION_FIELD_NUMBER: builtins.int
+    TEXT_CLASSIFICATION_MULTILABEL_FIELD_NUMBER: builtins.int
+    TEXT_CLASSIFICATION_MULTICLASS_FIELD_NUMBER: builtins.int
+    TEXT_EMBEDDING_PAIR_PARAMS_FIELD_NUMBER: builtins.int
+    TEXT_EMBEDDING_TRIPLET_PARAMS_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    LABELS_FIELD_NUMBER: builtins.int
+    tuning_task_id: builtins.str
+    base_model_uri: builtins.str
+    name: builtins.str
+    description: builtins.str
+    @property
+    def train_datasets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TuningRequest.WeightedDataset]: ...
+    @property
+    def validation_datasets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TuningRequest.WeightedDataset]: ...
+    @property
+    def text_to_text_completion(self) -> global___TextToTextCompletionTuningParams: ...
+    @property
+    def text_classification_multilabel(self) -> global___TextClassificationMultilabelParams: ...
+    @property
+    def text_classification_multiclass(self) -> global___TextClassificationMulticlassParams: ...
+    @property
+    def text_embedding_pair_params(self) -> global___TextEmbeddingPairParams: ...
+    @property
+    def text_embedding_triplet_params(self) -> global___TextEmbeddingTripletParams: ...
+    @property
+    def labels(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        tuning_task_id: builtins.str = ...,
+        base_model_uri: builtins.str = ...,
+        train_datasets: collections.abc.Iterable[global___TuningRequest.WeightedDataset] | None = ...,
+        validation_datasets: collections.abc.Iterable[global___TuningRequest.WeightedDataset] | None = ...,
+        text_to_text_completion: global___TextToTextCompletionTuningParams | None = ...,
+        text_classification_multilabel: global___TextClassificationMultilabelParams | None = ...,
+        text_classification_multiclass: global___TextClassificationMulticlassParams | None = ...,
+        text_embedding_pair_params: global___TextEmbeddingPairParams | None = ...,
+        text_embedding_triplet_params: global___TextEmbeddingTripletParams | None = ...,
+        name: builtins.str = ...,
+        description: builtins.str = ...,
+        labels: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["text_classification_multiclass", b"text_classification_multiclass", "text_classification_multilabel", b"text_classification_multilabel", "text_embedding_pair_params", b"text_embedding_pair_params", "text_embedding_triplet_params", b"text_embedding_triplet_params", "text_to_text_completion", b"text_to_text_completion", "tuning_params", b"tuning_params"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["base_model_uri", b"base_model_uri", "description", b"description", "labels", b"labels", "name", b"name", "text_classification_multiclass", b"text_classification_multiclass", "text_classification_multilabel", b"text_classification_multilabel", "text_embedding_pair_params", b"text_embedding_pair_params", "text_embedding_triplet_params", b"text_embedding_triplet_params", "text_to_text_completion", b"text_to_text_completion", "train_datasets", b"train_datasets", "tuning_params", b"tuning_params", "tuning_task_id", b"tuning_task_id", "validation_datasets", b"validation_datasets"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["tuning_params", b"tuning_params"]) -> typing.Literal["text_to_text_completion", "text_classification_multilabel", "text_classification_multiclass", "text_embedding_pair_params", "text_embedding_triplet_params"] | None: ...
+
+global___UpdateTuningDraftRequest = UpdateTuningDraftRequest
+
+@typing.final
+class UpdateTuningDraftResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TUNING_TASK_ID_FIELD_NUMBER: builtins.int
+    tuning_task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        tuning_task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tuning_task_id", b"tuning_task_id"]) -> None: ...
+
+global___UpdateTuningDraftResponse = UpdateTuningDraftResponse
+
+@typing.final
+class DeleteTuningDraftRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TUNING_TASK_ID_FIELD_NUMBER: builtins.int
+    tuning_task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        tuning_task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tuning_task_id", b"tuning_task_id"]) -> None: ...
+
+global___DeleteTuningDraftRequest = DeleteTuningDraftRequest
+
+@typing.final
+class DeleteTuningDraftResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TUNING_TASK_ID_FIELD_NUMBER: builtins.int
+    tuning_task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        tuning_task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tuning_task_id", b"tuning_task_id"]) -> None: ...
+
+global___DeleteTuningDraftResponse = DeleteTuningDraftResponse
+
+@typing.final
+class TuneDraftRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TUNING_TASK_ID_FIELD_NUMBER: builtins.int
+    tuning_task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        tuning_task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tuning_task_id", b"tuning_task_id"]) -> None: ...
+
+global___TuneDraftRequest = TuneDraftRequest

yandex/cloud/ai/tuning/v1/tuning_service_pb2_grpc.py

@@ -70,6 +70,26 @@ class TuningServiceStub(object):
                 request_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.ListErrorsRequest.SerializeToString,
                 response_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.ListErrorsResponse.FromString,
                 _registered_method=True)
+        self.CreateDraft = channel.unary_unary(
+                '/yandex.cloud.ai.tuning.v1.TuningService/CreateDraft',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.CreateTuningDraftRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.CreateTuningDraftResponse.FromString,
+                _registered_method=True)
+        self.UpdateDraft = channel.unary_unary(
+                '/yandex.cloud.ai.tuning.v1.TuningService/UpdateDraft',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.UpdateTuningDraftRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.UpdateTuningDraftResponse.FromString,
+                _registered_method=True)
+        self.DeleteDraft = channel.unary_unary(
+                '/yandex.cloud.ai.tuning.v1.TuningService/DeleteDraft',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.DeleteTuningDraftRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.DeleteTuningDraftResponse.FromString,
+                _registered_method=True)
+        self.TuneDraft = channel.unary_unary(
+                '/yandex.cloud.ai.tuning.v1.TuningService/TuneDraft',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.TuneDraftRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+                _registered_method=True)
 
 
 class TuningServiceServicer(object):
@@ -117,6 +137,30 @@ class TuningServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CreateDraft(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UpdateDraft(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteDraft(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TuneDraft(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_TuningServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -155,6 +199,26 @@ def add_TuningServiceServicer_to_server(servicer, server):
                     request_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.ListErrorsRequest.FromString,
                     response_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.ListErrorsResponse.SerializeToString,
             ),
+            'CreateDraft': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateDraft,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.CreateTuningDraftRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.CreateTuningDraftResponse.SerializeToString,
+            ),
+            'UpdateDraft': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateDraft,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.UpdateTuningDraftRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.UpdateTuningDraftResponse.SerializeToString,
+            ),
+            'DeleteDraft': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteDraft,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.DeleteTuningDraftRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.DeleteTuningDraftResponse.SerializeToString,
+            ),
+            'TuneDraft': grpc.unary_unary_rpc_method_handler(
+                    servicer.TuneDraft,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.TuneDraftRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'yandex.cloud.ai.tuning.v1.TuningService', rpc_method_handlers)
@@ -354,3 +418,111 @@ class TuningService(object):
             timeout,
             metadata,
             _registered_method=True)
+
+    @staticmethod
+    def CreateDraft(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ai.tuning.v1.TuningService/CreateDraft',
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.CreateTuningDraftRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.CreateTuningDraftResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def UpdateDraft(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ai.tuning.v1.TuningService/UpdateDraft',
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.UpdateTuningDraftRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.UpdateTuningDraftResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DeleteDraft(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ai.tuning.v1.TuningService/DeleteDraft',
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.DeleteTuningDraftRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.DeleteTuningDraftResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def TuneDraft(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/yandex.cloud.ai.tuning.v1.TuningService/TuneDraft',
+            yandex_dot_cloud_dot_ai_dot_tuning_dot_v1_dot_tuning__service__pb2.TuneDraftRequest.SerializeToString,
+            yandex_dot_cloud_dot_operation_dot_operation__pb2.Operation.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

yandex/cloud/ai/tuning/v1/tuning_service_pb2_grpc.pyi

@@ -55,6 +55,26 @@ class TuningServiceStub:
         yandex.cloud.ai.tuning.v1.tuning_service_pb2.ListErrorsResponse,
     ]
 
+    CreateDraft: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftRequest,
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftResponse,
+    ]
+
+    UpdateDraft: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftRequest,
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftResponse,
+    ]
+
+    DeleteDraft: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftRequest,
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftResponse,
+    ]
+
+    TuneDraft: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.TuneDraftRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+
 class TuningServiceAsyncStub:
     Tune: grpc.aio.UnaryUnaryMultiCallable[
         yandex.cloud.ai.tuning.v1.tuning_service_pb2.TuningRequest,
@@ -91,6 +111,26 @@ class TuningServiceAsyncStub:
         yandex.cloud.ai.tuning.v1.tuning_service_pb2.ListErrorsResponse,
     ]
 
+    CreateDraft: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftRequest,
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftResponse,
+    ]
+
+    UpdateDraft: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftRequest,
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftResponse,
+    ]
+
+    DeleteDraft: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftRequest,
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftResponse,
+    ]
+
+    TuneDraft: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.tuning.v1.tuning_service_pb2.TuneDraftRequest,
+        yandex.cloud.operation.operation_pb2.Operation,
+    ]
+
 class TuningServiceServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
     def Tune(
@@ -141,4 +181,32 @@ class TuningServiceServicer(metaclass=abc.ABCMeta):
         context: _ServicerContext,
     ) -> typing.Union[yandex.cloud.ai.tuning.v1.tuning_service_pb2.ListErrorsResponse, collections.abc.Awaitable[yandex.cloud.ai.tuning.v1.tuning_service_pb2.ListErrorsResponse]]: ...
 
+    @abc.abstractmethod
+    def CreateDraft(
+        self,
+        request: yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftResponse, collections.abc.Awaitable[yandex.cloud.ai.tuning.v1.tuning_service_pb2.CreateTuningDraftResponse]]: ...
+
+    @abc.abstractmethod
+    def UpdateDraft(
+        self,
+        request: yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftResponse, collections.abc.Awaitable[yandex.cloud.ai.tuning.v1.tuning_service_pb2.UpdateTuningDraftResponse]]: ...
+
+    @abc.abstractmethod
+    def DeleteDraft(
+        self,
+        request: yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftResponse, collections.abc.Awaitable[yandex.cloud.ai.tuning.v1.tuning_service_pb2.DeleteTuningDraftResponse]]: ...
+
+    @abc.abstractmethod
+    def TuneDraft(
+        self,
+        request: yandex.cloud.ai.tuning.v1.tuning_service_pb2.TuneDraftRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.operation.operation_pb2.Operation, collections.abc.Awaitable[yandex.cloud.operation.operation_pb2.Operation]]: ...
+
 def add_TuningServiceServicer_to_server(servicer: TuningServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

yandex/cloud/ai/tuning/v1/tuning_task_pb2.py

@@ -26,7 +26,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 from yandex.cloud import validation_pb2 as yandex_dot_cloud_dot_validation__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+yandex/cloud/ai/tuning/v1/tuning_task.proto\x12\x19yandex.cloud.ai.tuning.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1dyandex/cloud/validation.proto\"\x8b\x05\n\nTuningTask\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0coperation_id\x18\x03 \x01(\t\x12<\n\x06status\x18\x04 \x01(\x0e\x32,.yandex.cloud.ai.tuning.v1.TuningTask.Status\x12\x11\n\tfolder_id\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12.\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nstarted_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0b\x66inished_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10source_model_uri\x18\n \x01(\t\x12\x18\n\x10target_model_uri\x18\x0b \x01(\t\x12)\n\x04name\x18\x0c \x01(\tB\x1b\xf2\xc7\x31\x17^[A-Za-z0-9 _-]{3,100}$\x12\x13\n\x0b\x64\x65scription\x18\r \x01(\t\x12\x41\n\x06labels\x18\x0e \x03(\x0b\x32\x31.yandex.cloud.ai.tuning.v1.TuningTask.LabelsEntry\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"t\n\x06Status\x12\x16\n\x12STATUS_UNSPECIFIED\x10\x00\x12\x0b\n\x07\x43REATED\x10\x01\x12\x0b\n\x07PENDING\x10\x02\x12\x0f\n\x0bIN_PROGRESS\x10\x03\x12\r\n\tCOMPLETED\x10\x04\x12\n\n\x06\x46\x41ILED\x10\x05\x12\x0c\n\x08\x43\x41NCELED\x10\x06J\x04\x08\x02\x10\x03\x42\x63\n\x1dyandex.cloud.api.ai.tuning.v1ZBgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/tuning/v1;fomob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+yandex/cloud/ai/tuning/v1/tuning_task.proto\x12\x19yandex.cloud.ai.tuning.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1dyandex/cloud/validation.proto\"\x96\x05\n\nTuningTask\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0coperation_id\x18\x03 \x01(\t\x12<\n\x06status\x18\x04 \x01(\x0e\x32,.yandex.cloud.ai.tuning.v1.TuningTask.Status\x12\x11\n\tfolder_id\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12.\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nstarted_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0b\x66inished_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10source_model_uri\x18\n \x01(\t\x12\x18\n\x10target_model_uri\x18\x0b \x01(\t\x12)\n\x04name\x18\x0c \x01(\tB\x1b\xf2\xc7\x31\x17^[A-Za-z0-9 _-]{3,100}$\x12\x13\n\x0b\x64\x65scription\x18\r \x01(\t\x12\x41\n\x06labels\x18\x0e \x03(\x0b\x32\x31.yandex.cloud.ai.tuning.v1.TuningTask.LabelsEntry\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x7f\n\x06Status\x12\x16\n\x12STATUS_UNSPECIFIED\x10\x00\x12\x0b\n\x07\x43REATED\x10\x01\x12\x0b\n\x07PENDING\x10\x02\x12\x0f\n\x0bIN_PROGRESS\x10\x03\x12\r\n\tCOMPLETED\x10\x04\x12\n\n\x06\x46\x41ILED\x10\x05\x12\x0c\n\x08\x43\x41NCELED\x10\x06\x12\t\n\x05\x44RAFT\x10\x07J\x04\x08\x02\x10\x03\x42\x63\n\x1dyandex.cloud.api.ai.tuning.v1ZBgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/tuning/v1;fomob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -39,9 +39,9 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_TUNINGTASK'].fields_by_name['name']._loaded_options = None
   _globals['_TUNINGTASK'].fields_by_name['name']._serialized_options = b'\362\3071\027^[A-Za-z0-9 _-]{3,100}$'
   _globals['_TUNINGTASK']._serialized_start=139
-  _globals['_TUNINGTASK']._serialized_end=790
+  _globals['_TUNINGTASK']._serialized_end=801
   _globals['_TUNINGTASK_LABELSENTRY']._serialized_start=621
   _globals['_TUNINGTASK_LABELSENTRY']._serialized_end=666
   _globals['_TUNINGTASK_STATUS']._serialized_start=668
-  _globals['_TUNINGTASK_STATUS']._serialized_end=784
+  _globals['_TUNINGTASK_STATUS']._serialized_end=795
 # @@protoc_insertion_point(module_scope)

yandex/cloud/ai/tuning/v1/tuning_task_pb2.pyi

@@ -37,6 +37,7 @@ class TuningTask(google.protobuf.message.Message):
         COMPLETED: TuningTask._Status.ValueType  # 4
         FAILED: TuningTask._Status.ValueType  # 5
         CANCELED: TuningTask._Status.ValueType  # 6
+        DRAFT: TuningTask._Status.ValueType  # 7
 
     class Status(_Status, metaclass=_StatusEnumTypeWrapper): ...
     STATUS_UNSPECIFIED: TuningTask.Status.ValueType  # 0
@@ -46,6 +47,7 @@ class TuningTask(google.protobuf.message.Message):
     COMPLETED: TuningTask.Status.ValueType  # 4
     FAILED: TuningTask.Status.ValueType  # 5
     CANCELED: TuningTask.Status.ValueType  # 6
+    DRAFT: TuningTask.Status.ValueType  # 7
 
     @typing.final
     class LabelsEntry(google.protobuf.message.Message):