yandexcloud 0.339.0__py3-none-any.whl → 0.341.0__py3-none-any.whl

yandex/cloud/ai/batch_inference/v1/batch_inference_service_pb2_grpc.py

@@ -0,0 +1,230 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from yandex.cloud.ai.batch_inference.v1 import batch_inference_service_pb2 as yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in yandex/cloud/ai/batch_inference/v1/batch_inference_service_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class BatchInferenceServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Describe = channel.unary_unary(
+                '/yandex.cloud.ai.batch_inference.v1.BatchInferenceService/Describe',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DescribeBatchInferenceRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DescribeBatchInferenceResponse.FromString,
+                _registered_method=True)
+        self.List = channel.unary_unary(
+                '/yandex.cloud.ai.batch_inference.v1.BatchInferenceService/List',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.ListBatchInferencesRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.ListBatchInferencesResponse.FromString,
+                _registered_method=True)
+        self.Cancel = channel.unary_unary(
+                '/yandex.cloud.ai.batch_inference.v1.BatchInferenceService/Cancel',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.CancelBatchInferenceRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.CancelBatchInferenceResponse.FromString,
+                _registered_method=True)
+        self.Delete = channel.unary_unary(
+                '/yandex.cloud.ai.batch_inference.v1.BatchInferenceService/Delete',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DeleteBatchInferenceRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DeleteBatchInferenceResponse.FromString,
+                _registered_method=True)
+
+
+class BatchInferenceServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def Describe(self, request, context):
+        """Describes a concrete task
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def List(self, request, context):
+        """Lists tasks in folder
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Cancel(self, request, context):
+        """Cancels a concrete task
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Delete(self, request, context):
+        """Deletes a concrete task
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_BatchInferenceServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Describe': grpc.unary_unary_rpc_method_handler(
+                    servicer.Describe,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DescribeBatchInferenceRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DescribeBatchInferenceResponse.SerializeToString,
+            ),
+            'List': grpc.unary_unary_rpc_method_handler(
+                    servicer.List,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.ListBatchInferencesRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.ListBatchInferencesResponse.SerializeToString,
+            ),
+            'Cancel': grpc.unary_unary_rpc_method_handler(
+                    servicer.Cancel,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.CancelBatchInferenceRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.CancelBatchInferenceResponse.SerializeToString,
+            ),
+            'Delete': grpc.unary_unary_rpc_method_handler(
+                    servicer.Delete,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DeleteBatchInferenceRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DeleteBatchInferenceResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'yandex.cloud.ai.batch_inference.v1.BatchInferenceService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('yandex.cloud.ai.batch_inference.v1.BatchInferenceService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class BatchInferenceService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def Describe(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.batch_inference.v1.BatchInferenceService/Describe',
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DescribeBatchInferenceRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DescribeBatchInferenceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def List(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.batch_inference.v1.BatchInferenceService/List',
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.ListBatchInferencesRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.ListBatchInferencesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Cancel(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.batch_inference.v1.BatchInferenceService/Cancel',
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.CancelBatchInferenceRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.CancelBatchInferenceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Delete(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.batch_inference.v1.BatchInferenceService/Delete',
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DeleteBatchInferenceRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__service__pb2.DeleteBatchInferenceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

yandex/cloud/ai/batch_inference/v1/batch_inference_service_pb2_grpc.pyi

@@ -0,0 +1,104 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import abc
+import collections.abc
+import grpc
+import grpc.aio
+import typing
+import yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...
+
+class BatchInferenceServiceStub:
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
+    Describe: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceResponse,
+    ]
+    """Describes a concrete task"""
+
+    List: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesResponse,
+    ]
+    """Lists tasks in folder"""
+
+    Cancel: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceResponse,
+    ]
+    """Cancels a concrete task"""
+
+    Delete: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceResponse,
+    ]
+    """Deletes a concrete task"""
+
+class BatchInferenceServiceAsyncStub:
+    Describe: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceResponse,
+    ]
+    """Describes a concrete task"""
+
+    List: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesResponse,
+    ]
+    """Lists tasks in folder"""
+
+    Cancel: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceResponse,
+    ]
+    """Cancels a concrete task"""
+
+    Delete: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceRequest,
+        yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceResponse,
+    ]
+    """Deletes a concrete task"""
+
+class BatchInferenceServiceServicer(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def Describe(
+        self,
+        request: yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceResponse, collections.abc.Awaitable[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DescribeBatchInferenceResponse]]:
+        """Describes a concrete task"""
+
+    @abc.abstractmethod
+    def List(
+        self,
+        request: yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesResponse, collections.abc.Awaitable[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.ListBatchInferencesResponse]]:
+        """Lists tasks in folder"""
+
+    @abc.abstractmethod
+    def Cancel(
+        self,
+        request: yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceResponse, collections.abc.Awaitable[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.CancelBatchInferenceResponse]]:
+        """Cancels a concrete task"""
+
+    @abc.abstractmethod
+    def Delete(
+        self,
+        request: yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceResponse, collections.abc.Awaitable[yandex.cloud.ai.batch_inference.v1.batch_inference_service_pb2.DeleteBatchInferenceResponse]]:
+        """Deletes a concrete task"""
+
+def add_BatchInferenceServiceServicer_to_server(servicer: BatchInferenceServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

yandex/cloud/ai/batch_inference/v1/batch_inference_task_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: yandex/cloud/ai/batch_inference/v1/batch_inference_task.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'yandex/cloud/ai/batch_inference/v1/batch_inference_task.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from yandex.cloud.ai.batch_inference.v1 import inference_options_pb2 as yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_inference__options__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n=yandex/cloud/ai/batch_inference/v1/batch_inference_task.proto\x12\"yandex.cloud.ai.batch_inference.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a:yandex/cloud/ai/batch_inference/v1/inference_options.proto\"\xe9\x05\n\x12\x42\x61tchInferenceTask\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0coperation_id\x18\x02 \x01(\t\x12\x11\n\tfolder_id\x18\x03 \x01(\t\x12\x11\n\tmodel_uri\x18\x04 \x01(\t\x12\x19\n\x11source_dataset_id\x18\x05 \x01(\t\x12X\n\x12\x63ompletion_request\x18\x06 \x01(\x0b\x32:.yandex.cloud.ai.batch_inference.v1.BatchCompletionRequestH\x00\x12M\n\x06status\x18\x07 \x01(\x0e\x32=.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask.Status\x12\x19\n\x11result_dataset_id\x18\x08 \x01(\t\x12R\n\x06labels\x18\t \x03(\x0b\x32\x42.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask.LabelsEntry\x12\x12\n\ncreated_by\x18\n \x01(\t\x12.\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nstarted_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0b\x66inished_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.Timestamp\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\x06\x42\t\n\x07requestBu\n&yandex.cloud.api.ai.batch_inference.v1ZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/batch_inference/v1;fomob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n&yandex.cloud.api.ai.batch_inference.v1ZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/batch_inference/v1;fomo'
+  _globals['_BATCHINFERENCETASK_LABELSENTRY']._loaded_options = None
+  _globals['_BATCHINFERENCETASK_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_BATCHINFERENCETASK']._serialized_start=195
+  _globals['_BATCHINFERENCETASK']._serialized_end=940
+  _globals['_BATCHINFERENCETASK_LABELSENTRY']._serialized_start=766
+  _globals['_BATCHINFERENCETASK_LABELSENTRY']._serialized_end=811
+  _globals['_BATCHINFERENCETASK_STATUS']._serialized_start=813
+  _globals['_BATCHINFERENCETASK_STATUS']._serialized_end=929
+# @@protoc_insertion_point(module_scope)

yandex/cloud/ai/batch_inference/v1/batch_inference_task_pb2.pyi

@@ -0,0 +1,119 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import sys
+import typing
+import yandex.cloud.ai.batch_inference.v1.inference_options_pb2
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class BatchInferenceTask(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _Status:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BatchInferenceTask._Status.ValueType], builtins.type):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        STATUS_UNSPECIFIED: BatchInferenceTask._Status.ValueType  # 0
+        CREATED: BatchInferenceTask._Status.ValueType  # 1
+        PENDING: BatchInferenceTask._Status.ValueType  # 2
+        IN_PROGRESS: BatchInferenceTask._Status.ValueType  # 3
+        COMPLETED: BatchInferenceTask._Status.ValueType  # 4
+        FAILED: BatchInferenceTask._Status.ValueType  # 5
+        CANCELED: BatchInferenceTask._Status.ValueType  # 6
+
+    class Status(_Status, metaclass=_StatusEnumTypeWrapper): ...
+    STATUS_UNSPECIFIED: BatchInferenceTask.Status.ValueType  # 0
+    CREATED: BatchInferenceTask.Status.ValueType  # 1
+    PENDING: BatchInferenceTask.Status.ValueType  # 2
+    IN_PROGRESS: BatchInferenceTask.Status.ValueType  # 3
+    COMPLETED: BatchInferenceTask.Status.ValueType  # 4
+    FAILED: BatchInferenceTask.Status.ValueType  # 5
+    CANCELED: BatchInferenceTask.Status.ValueType  # 6
+
+    @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: ...
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    OPERATION_ID_FIELD_NUMBER: builtins.int
+    FOLDER_ID_FIELD_NUMBER: builtins.int
+    MODEL_URI_FIELD_NUMBER: builtins.int
+    SOURCE_DATASET_ID_FIELD_NUMBER: builtins.int
+    COMPLETION_REQUEST_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    RESULT_DATASET_ID_FIELD_NUMBER: builtins.int
+    LABELS_FIELD_NUMBER: builtins.int
+    CREATED_BY_FIELD_NUMBER: builtins.int
+    CREATED_AT_FIELD_NUMBER: builtins.int
+    STARTED_AT_FIELD_NUMBER: builtins.int
+    FINISHED_AT_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    operation_id: builtins.str
+    folder_id: builtins.str
+    model_uri: builtins.str
+    source_dataset_id: builtins.str
+    status: global___BatchInferenceTask.Status.ValueType
+    result_dataset_id: builtins.str
+    created_by: builtins.str
+    @property
+    def completion_request(self) -> yandex.cloud.ai.batch_inference.v1.inference_options_pb2.BatchCompletionRequest: ...
+    @property
+    def labels(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]: ...
+    @property
+    def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
+    @property
+    def started_at(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
+    @property
+    def finished_at(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        operation_id: builtins.str = ...,
+        folder_id: builtins.str = ...,
+        model_uri: builtins.str = ...,
+        source_dataset_id: builtins.str = ...,
+        completion_request: yandex.cloud.ai.batch_inference.v1.inference_options_pb2.BatchCompletionRequest | None = ...,
+        status: global___BatchInferenceTask.Status.ValueType = ...,
+        result_dataset_id: builtins.str = ...,
+        labels: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        created_by: builtins.str = ...,
+        created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        started_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        finished_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["completion_request", b"completion_request", "created_at", b"created_at", "finished_at", b"finished_at", "request", b"request", "started_at", b"started_at"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["completion_request", b"completion_request", "created_at", b"created_at", "created_by", b"created_by", "finished_at", b"finished_at", "folder_id", b"folder_id", "labels", b"labels", "model_uri", b"model_uri", "operation_id", b"operation_id", "request", b"request", "result_dataset_id", b"result_dataset_id", "source_dataset_id", b"source_dataset_id", "started_at", b"started_at", "status", b"status", "task_id", b"task_id"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["request", b"request"]) -> typing.Literal["completion_request"] | None: ...
+
+global___BatchInferenceTask = BatchInferenceTask

yandex/cloud/ai/batch_inference/v1/batch_inference_task_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in yandex/cloud/ai/batch_inference/v1/batch_inference_task_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

yandex/cloud/ai/batch_inference/v1/batch_inference_task_pb2_grpc.pyi

@@ -0,0 +1,17 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import abc
+import collections.abc
+import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...

yandex/cloud/ai/batch_inference/v1/inference_options_pb2.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: yandex/cloud/ai/batch_inference/v1/inference_options.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'yandex/cloud/ai/batch_inference/v1/inference_options.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+from yandex.cloud import validation_pb2 as yandex_dot_cloud_dot_validation__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n:yandex/cloud/ai/batch_inference/v1/inference_options.proto\x12\"yandex.cloud.ai.batch_inference.v1\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1dyandex/cloud/validation.proto\"\xb5\x02\n\x16\x42\x61tchCompletionRequest\x12\x17\n\tmodel_uri\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x1f\n\x11source_dataset_id\x18\x02 \x01(\tB\x04\xe8\xc7\x31\x01\x12Q\n\x12\x63ompletion_options\x18\x03 \x01(\x0b\x32\x35.yandex.cloud.ai.batch_inference.v1.CompletionOptions\x12\x1c\n\x14\x64\x61ta_logging_enabled\x18\x04 \x01(\x08\x12\x15\n\x0bjson_object\x18\x05 \x01(\x08H\x00\x12\x45\n\x0bjson_schema\x18\x06 \x01(\x0b\x32..yandex.cloud.ai.batch_inference.v1.JsonSchemaH\x00\x42\x12\n\x10StructuredOutput\"\xc8\x01\n\x11\x43ompletionOptions\x12\x31\n\x0btemperature\x18\x01 \x01(\x0b\x32\x1c.google.protobuf.DoubleValue\x12/\n\nmax_tokens\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12O\n\x11reasoning_options\x18\x03 \x01(\x0b\x32\x34.yandex.cloud.ai.batch_inference.v1.ReasoningOptions\"\xb7\x01\n\x10ReasoningOptions\x12P\n\x04mode\x18\x01 \x01(\x0e\x32\x42.yandex.cloud.ai.batch_inference.v1.ReasoningOptions.ReasoningMode\"Q\n\rReasoningMode\x12\x1e\n\x1aREASONING_MODE_UNSPECIFIED\x10\x00\x12\x0c\n\x08\x44ISABLED\x10\x01\x12\x12\n\x0e\x45NABLED_HIDDEN\x10\x02\"5\n\nJsonSchema\x12\'\n\x06schema\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructBu\n&yandex.cloud.api.ai.batch_inference.v1ZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/batch_inference/v1;fomob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'yandex.cloud.ai.batch_inference.v1.inference_options_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n&yandex.cloud.api.ai.batch_inference.v1ZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/batch_inference/v1;fomo'
+  _globals['_BATCHCOMPLETIONREQUEST'].fields_by_name['model_uri']._loaded_options = None
+  _globals['_BATCHCOMPLETIONREQUEST'].fields_by_name['model_uri']._serialized_options = b'\350\3071\001'
+  _globals['_BATCHCOMPLETIONREQUEST'].fields_by_name['source_dataset_id']._loaded_options = None
+  _globals['_BATCHCOMPLETIONREQUEST'].fields_by_name['source_dataset_id']._serialized_options = b'\350\3071\001'
+  _globals['_BATCHCOMPLETIONREQUEST']._serialized_start=192
+  _globals['_BATCHCOMPLETIONREQUEST']._serialized_end=501
+  _globals['_COMPLETIONOPTIONS']._serialized_start=504
+  _globals['_COMPLETIONOPTIONS']._serialized_end=704
+  _globals['_REASONINGOPTIONS']._serialized_start=707
+  _globals['_REASONINGOPTIONS']._serialized_end=890
+  _globals['_REASONINGOPTIONS_REASONINGMODE']._serialized_start=809
+  _globals['_REASONINGOPTIONS_REASONINGMODE']._serialized_end=890
+  _globals['_JSONSCHEMA']._serialized_start=892
+  _globals['_JSONSCHEMA']._serialized_end=945
+# @@protoc_insertion_point(module_scope)