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

yandex/cloud/ai/assistants/v1/runs/run_service_pb2.py

@@ -31,7 +31,7 @@ from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
 from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4yandex/cloud/ai/assistants/v1/runs/run_service.proto\x12\"yandex.cloud.ai.assistants.v1.runs\x1a#yandex/cloud/ai/common/common.proto\x1a*yandex/cloud/ai/assistants/v1/common.proto\x1a\x33yandex/cloud/ai/assistants/v1/threads/message.proto\x1a,yandex/cloud/ai/assistants/v1/runs/run.proto\x1a\x1dyandex/cloud/validation.proto\x1a\x1cgoogle/api/annotations.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\x94\x04\n\x10\x43reateRunRequest\x12\x1a\n\x0c\x61ssistant_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x17\n\tthread_id\x18\x02 \x01(\tB\x04\xe8\xc7\x31\x01\x12P\n\x06labels\x18\x03 \x03(\x0b\x32@.yandex.cloud.ai.assistants.v1.runs.CreateRunRequest.LabelsEntry\x12O\n\x13\x61\x64\x64itional_messages\x18\x04 \x03(\x0b\x32\x32.yandex.cloud.ai.assistants.v1.threads.MessageData\x12`\n custom_prompt_truncation_options\x18\x05 \x01(\x0b\x32\x36.yandex.cloud.ai.assistants.v1.PromptTruncationOptions\x12S\n\x19\x63ustom_completion_options\x18\x06 \x01(\x0b\x32\x30.yandex.cloud.ai.assistants.v1.CompletionOptions\x12\x0e\n\x06stream\x18\x07 \x01(\x08\x12\x32\n\x05tools\x18\x08 \x03(\x0b\x32#.yandex.cloud.ai.assistants.v1.Tool\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"_\n\x10ListenRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x35\n\x10\x65vents_start_idx\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"\xb3\x01\n\x10\x41ttachRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x35\n\x10\x65vents_start_idx\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12I\n\x10tool_result_list\x18\x03 \x01(\x0b\x32-.yandex.cloud.ai.assistants.v1.ToolResultListH\x00\x42\x07\n\x05\x45vent\"%\n\rGetRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"4\n\x19GetLastRunByThreadRequest\x12\x17\n\tthread_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"Q\n\x0fListRunsRequest\x12\x17\n\tfolder_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x11\n\tpage_size\x18\x02 \x01(\x03\x12\x12\n\npage_token\x18\x03 \x01(\t\"b\n\x10ListRunsResponse\x12\x35\n\x04runs\x18\x01 \x03(\x0b\x32\'.yandex.cloud.ai.assistants.v1.runs.Run\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"K\n\x0cStreamCursor\x12\x19\n\x11\x63urrent_event_idx\x18\x01 \x01(\x03\x12 \n\x18num_user_events_received\x18\x02 \x01(\x03\"\xab\x04\n\x0bStreamEvent\x12M\n\nevent_type\x18\x01 \x01(\x0e\x32\x39.yandex.cloud.ai.assistants.v1.runs.StreamEvent.EventType\x12G\n\rstream_cursor\x18\x02 \x01(\x0b\x32\x30.yandex.cloud.ai.assistants.v1.runs.StreamCursor\x12.\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x1d.yandex.cloud.ai.common.ErrorH\x00\x12P\n\x0fpartial_message\x18\x04 \x01(\x0b\x32\x35.yandex.cloud.ai.assistants.v1.threads.MessageContentH\x00\x12K\n\x11\x63ompleted_message\x18\x05 \x01(\x0b\x32..yandex.cloud.ai.assistants.v1.threads.MessageH\x00\x12\x45\n\x0etool_call_list\x18\x06 \x01(\x0b\x32+.yandex.cloud.ai.assistants.v1.ToolCallListH\x00\"a\n\tEventType\x12\x1a\n\x16\x45VENT_TYPE_UNSPECIFIED\x10\x00\x12\x13\n\x0fPARTIAL_MESSAGE\x10\x01\x12\t\n\x05\x45RROR\x10\x02\x12\x08\n\x04\x44ONE\x10\x03\x12\x0e\n\nTOOL_CALLS\x10\x04\x42\x0b\n\tEventData2\xe3\x06\n\nRunService\x12\x87\x01\n\x06\x43reate\x12\x34.yandex.cloud.ai.assistants.v1.runs.CreateRunRequest\x1a\'.yandex.cloud.ai.assistants.v1.runs.Run\"\x1e\x82\xd3\xe4\x93\x02\x18\"\x13/assistants/v1/runs:\x01*\x12\x95\x01\n\x06Listen\x12\x34.yandex.cloud.ai.assistants.v1.runs.ListenRunRequest\x1a/.yandex.cloud.ai.assistants.v1.runs.StreamEvent\"\"\x82\xd3\xe4\x93\x02\x1c\x12\x1a/assistants/v1/runs/listen0\x01\x12s\n\x06\x41ttach\x12\x34.yandex.cloud.ai.assistants.v1.runs.AttachRunRequest\x1a/.yandex.cloud.ai.assistants.v1.runs.StreamEvent(\x01\x30\x01\x12\x87\x01\n\x03Get\x12\x31.yandex.cloud.ai.assistants.v1.runs.GetRunRequest\x1a\'.yandex.cloud.ai.assistants.v1.runs.Run\"$\x82\xd3\xe4\x93\x02\x1e\x12\x1c/assistants/v1/runs/{run_id}\x12\xa2\x01\n\x0fGetLastByThread\x12=.yandex.cloud.ai.assistants.v1.runs.GetLastRunByThreadRequest\x1a\'.yandex.cloud.ai.assistants.v1.runs.Run\"\'\x82\xd3\xe4\x93\x02!\x12\x1f/assistants/v1/runs:getByThread\x12\x8e\x01\n\x04List\x12\x33.yandex.cloud.ai.assistants.v1.runs.ListRunsRequest\x1a\x34.yandex.cloud.ai.assistants.v1.runs.ListRunsResponse\"\x1b\x82\xd3\xe4\x93\x02\x15\x12\x13/assistants/v1/runsBu\n&yandex.cloud.api.ai.assistants.v1.runsZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/runs;runsb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4yandex/cloud/ai/assistants/v1/runs/run_service.proto\x12\"yandex.cloud.ai.assistants.v1.runs\x1a#yandex/cloud/ai/common/common.proto\x1a*yandex/cloud/ai/assistants/v1/common.proto\x1a\x33yandex/cloud/ai/assistants/v1/threads/message.proto\x1a,yandex/cloud/ai/assistants/v1/runs/run.proto\x1a\x1dyandex/cloud/validation.proto\x1a\x1cgoogle/api/annotations.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\x94\x04\n\x10\x43reateRunRequest\x12\x1a\n\x0c\x61ssistant_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x17\n\tthread_id\x18\x02 \x01(\tB\x04\xe8\xc7\x31\x01\x12P\n\x06labels\x18\x03 \x03(\x0b\x32@.yandex.cloud.ai.assistants.v1.runs.CreateRunRequest.LabelsEntry\x12O\n\x13\x61\x64\x64itional_messages\x18\x04 \x03(\x0b\x32\x32.yandex.cloud.ai.assistants.v1.threads.MessageData\x12`\n custom_prompt_truncation_options\x18\x05 \x01(\x0b\x32\x36.yandex.cloud.ai.assistants.v1.PromptTruncationOptions\x12S\n\x19\x63ustom_completion_options\x18\x06 \x01(\x0b\x32\x30.yandex.cloud.ai.assistants.v1.CompletionOptions\x12\x0e\n\x06stream\x18\x07 \x01(\x08\x12\x32\n\x05tools\x18\x08 \x03(\x0b\x32#.yandex.cloud.ai.assistants.v1.Tool\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"_\n\x10ListenRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x35\n\x10\x65vents_start_idx\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"\xb3\x01\n\x10\x41ttachRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x35\n\x10\x65vents_start_idx\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12I\n\x10tool_result_list\x18\x03 \x01(\x0b\x32-.yandex.cloud.ai.assistants.v1.ToolResultListH\x00\x42\x07\n\x05\x45vent\"%\n\rGetRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"4\n\x19GetLastRunByThreadRequest\x12\x17\n\tthread_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"Q\n\x0fListRunsRequest\x12\x17\n\tfolder_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x11\n\tpage_size\x18\x02 \x01(\x03\x12\x12\n\npage_token\x18\x03 \x01(\t\"b\n\x10ListRunsResponse\x12\x35\n\x04runs\x18\x01 \x03(\x0b\x32\'.yandex.cloud.ai.assistants.v1.runs.Run\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"K\n\x0cStreamCursor\x12\x19\n\x11\x63urrent_event_idx\x18\x01 \x01(\x03\x12 \n\x18num_user_events_received\x18\x02 \x01(\x03\"\xab\x04\n\x0bStreamEvent\x12M\n\nevent_type\x18\x01 \x01(\x0e\x32\x39.yandex.cloud.ai.assistants.v1.runs.StreamEvent.EventType\x12G\n\rstream_cursor\x18\x02 \x01(\x0b\x32\x30.yandex.cloud.ai.assistants.v1.runs.StreamCursor\x12.\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x1d.yandex.cloud.ai.common.ErrorH\x00\x12P\n\x0fpartial_message\x18\x04 \x01(\x0b\x32\x35.yandex.cloud.ai.assistants.v1.threads.MessageContentH\x00\x12K\n\x11\x63ompleted_message\x18\x05 \x01(\x0b\x32..yandex.cloud.ai.assistants.v1.threads.MessageH\x00\x12\x45\n\x0etool_call_list\x18\x06 \x01(\x0b\x32+.yandex.cloud.ai.assistants.v1.ToolCallListH\x00\"a\n\tEventType\x12\x1a\n\x16\x45VENT_TYPE_UNSPECIFIED\x10\x00\x12\x13\n\x0fPARTIAL_MESSAGE\x10\x01\x12\t\n\x05\x45RROR\x10\x02\x12\x08\n\x04\x44ONE\x10\x03\x12\x0e\n\nTOOL_CALLS\x10\x04\x42\x0b\n\tEventData\"~\n\x12SubmitToRunRequest\x12\x14\n\x06run_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12I\n\x10tool_result_list\x18\x02 \x01(\x0b\x32-.yandex.cloud.ai.assistants.v1.ToolResultListH\x00\x42\x07\n\x05\x45vent\"\x15\n\x13SubmitToRunResponse2\x86\x08\n\nRunService\x12\x87\x01\n\x06\x43reate\x12\x34.yandex.cloud.ai.assistants.v1.runs.CreateRunRequest\x1a\'.yandex.cloud.ai.assistants.v1.runs.Run\"\x1e\x82\xd3\xe4\x93\x02\x18\"\x13/assistants/v1/runs:\x01*\x12\x95\x01\n\x06Listen\x12\x34.yandex.cloud.ai.assistants.v1.runs.ListenRunRequest\x1a/.yandex.cloud.ai.assistants.v1.runs.StreamEvent\"\"\x82\xd3\xe4\x93\x02\x1c\x12\x1a/assistants/v1/runs/listen0\x01\x12s\n\x06\x41ttach\x12\x34.yandex.cloud.ai.assistants.v1.runs.AttachRunRequest\x1a/.yandex.cloud.ai.assistants.v1.runs.StreamEvent(\x01\x30\x01\x12\x87\x01\n\x03Get\x12\x31.yandex.cloud.ai.assistants.v1.runs.GetRunRequest\x1a\'.yandex.cloud.ai.assistants.v1.runs.Run\"$\x82\xd3\xe4\x93\x02\x1e\x12\x1c/assistants/v1/runs/{run_id}\x12\xa2\x01\n\x0fGetLastByThread\x12=.yandex.cloud.ai.assistants.v1.runs.GetLastRunByThreadRequest\x1a\'.yandex.cloud.ai.assistants.v1.runs.Run\"\'\x82\xd3\xe4\x93\x02!\x12\x1f/assistants/v1/runs:getByThread\x12\x8e\x01\n\x04List\x12\x33.yandex.cloud.ai.assistants.v1.runs.ListRunsRequest\x1a\x34.yandex.cloud.ai.assistants.v1.runs.ListRunsResponse\"\x1b\x82\xd3\xe4\x93\x02\x15\x12\x13/assistants/v1/runs\x12\xa0\x01\n\x06Submit\x12\x36.yandex.cloud.ai.assistants.v1.runs.SubmitToRunRequest\x1a\x37.yandex.cloud.ai.assistants.v1.runs.SubmitToRunResponse\"%\x82\xd3\xe4\x93\x02\x1f\x32\x1a/assistants/v1/runs/submit:\x01*Bu\n&yandex.cloud.api.ai.assistants.v1.runsZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/runs;runsb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -55,6 +55,8 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_GETLASTRUNBYTHREADREQUEST'].fields_by_name['thread_id']._serialized_options = b'\350\3071\001'
   _globals['_LISTRUNSREQUEST'].fields_by_name['folder_id']._loaded_options = None
   _globals['_LISTRUNSREQUEST'].fields_by_name['folder_id']._serialized_options = b'\350\3071\001'
+  _globals['_SUBMITTORUNREQUEST'].fields_by_name['run_id']._loaded_options = None
+  _globals['_SUBMITTORUNREQUEST'].fields_by_name['run_id']._serialized_options = b'\350\3071\001'
   _globals['_RUNSERVICE'].methods_by_name['Create']._loaded_options = None
   _globals['_RUNSERVICE'].methods_by_name['Create']._serialized_options = b'\202\323\344\223\002\030\"\023/assistants/v1/runs:\001*'
   _globals['_RUNSERVICE'].methods_by_name['Listen']._loaded_options = None
@@ -65,6 +67,8 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_RUNSERVICE'].methods_by_name['GetLastByThread']._serialized_options = b'\202\323\344\223\002!\022\037/assistants/v1/runs:getByThread'
   _globals['_RUNSERVICE'].methods_by_name['List']._loaded_options = None
   _globals['_RUNSERVICE'].methods_by_name['List']._serialized_options = b'\202\323\344\223\002\025\022\023/assistants/v1/runs'
+  _globals['_RUNSERVICE'].methods_by_name['Submit']._loaded_options = None
+  _globals['_RUNSERVICE'].methods_by_name['Submit']._serialized_options = b'\202\323\344\223\002\0372\032/assistants/v1/runs/submit:\001*'
   _globals['_CREATERUNREQUEST']._serialized_start=366
   _globals['_CREATERUNREQUEST']._serialized_end=898
   _globals['_CREATERUNREQUEST_LABELSENTRY']._serialized_start=853
@@ -87,6 +91,10 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_STREAMEVENT']._serialized_end=2088
   _globals['_STREAMEVENT_EVENTTYPE']._serialized_start=1978
   _globals['_STREAMEVENT_EVENTTYPE']._serialized_end=2075
-  _globals['_RUNSERVICE']._serialized_start=2091
-  _globals['_RUNSERVICE']._serialized_end=2958
+  _globals['_SUBMITTORUNREQUEST']._serialized_start=2090
+  _globals['_SUBMITTORUNREQUEST']._serialized_end=2216
+  _globals['_SUBMITTORUNRESPONSE']._serialized_start=2218
+  _globals['_SUBMITTORUNRESPONSE']._serialized_end=2239
+  _globals['_RUNSERVICE']._serialized_start=2242
+  _globals['_RUNSERVICE']._serialized_end=3272
 # @@protoc_insertion_point(module_scope)

yandex/cloud/ai/assistants/v1/runs/run_service_pb2.pyi

@@ -344,3 +344,37 @@ class StreamEvent(google.protobuf.message.Message):
     def WhichOneof(self, oneof_group: typing.Literal["EventData", b"EventData"]) -> typing.Literal["error", "partial_message", "completed_message", "tool_call_list"] | None: ...
 
 global___StreamEvent = StreamEvent
+
+@typing.final
+class SubmitToRunRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    RUN_ID_FIELD_NUMBER: builtins.int
+    TOOL_RESULT_LIST_FIELD_NUMBER: builtins.int
+    run_id: builtins.str
+    """ID of the run to submit to."""
+    @property
+    def tool_result_list(self) -> yandex.cloud.ai.assistants.v1.common_pb2.ToolResultList:
+        """A list of tool results to submit to the run, such as the output of a function call."""
+
+    def __init__(
+        self,
+        *,
+        run_id: builtins.str = ...,
+        tool_result_list: yandex.cloud.ai.assistants.v1.common_pb2.ToolResultList | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["Event", b"Event", "tool_result_list", b"tool_result_list"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["Event", b"Event", "run_id", b"run_id", "tool_result_list", b"tool_result_list"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["Event", b"Event"]) -> typing.Literal["tool_result_list"] | None: ...
+
+global___SubmitToRunRequest = SubmitToRunRequest
+
+@typing.final
+class SubmitToRunResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___SubmitToRunResponse = SubmitToRunResponse

yandex/cloud/ai/assistants/v1/runs/run_service_pb2_grpc.py

@@ -66,6 +66,11 @@ class RunServiceStub(object):
                 request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.ListRunsRequest.SerializeToString,
                 response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.ListRunsResponse.FromString,
                 _registered_method=True)
+        self.Submit = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.runs.RunService/Submit',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.SubmitToRunRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.SubmitToRunResponse.FromString,
+                _registered_method=True)
 
 
 class RunServiceServicer(object):
@@ -118,6 +123,14 @@ class RunServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def Submit(self, request, context):
+        """Submit event to run
+        For example, submit function call results when the run is waiting for user input.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_RunServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -151,6 +164,11 @@ def add_RunServiceServicer_to_server(servicer, server):
                     request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.ListRunsRequest.FromString,
                     response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.ListRunsResponse.SerializeToString,
             ),
+            'Submit': grpc.unary_unary_rpc_method_handler(
+                    servicer.Submit,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.SubmitToRunRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.SubmitToRunResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'yandex.cloud.ai.assistants.v1.runs.RunService', rpc_method_handlers)
@@ -324,3 +342,30 @@ class RunService(object):
             timeout,
             metadata,
             _registered_method=True)
+
+    @staticmethod
+    def Submit(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.assistants.v1.runs.RunService/Submit',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.SubmitToRunRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_runs_dot_run__service__pb2.SubmitToRunResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

yandex/cloud/ai/assistants/v1/runs/run_service_pb2_grpc.pyi

@@ -64,6 +64,14 @@ class RunServiceStub:
     ]
     """List runs in a specific folder."""
 
+    Submit: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunRequest,
+        yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunResponse,
+    ]
+    """Submit event to run
+    For example, submit function call results when the run is waiting for user input.
+    """
+
 class RunServiceAsyncStub:
     """RunService provides operations for managing runs."""
 
@@ -109,6 +117,14 @@ class RunServiceAsyncStub:
     ]
     """List runs in a specific folder."""
 
+    Submit: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunRequest,
+        yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunResponse,
+    ]
+    """Submit event to run
+    For example, submit function call results when the run is waiting for user input.
+    """
+
 class RunServiceServicer(metaclass=abc.ABCMeta):
     """RunService provides operations for managing runs."""
 
@@ -166,4 +182,14 @@ class RunServiceServicer(metaclass=abc.ABCMeta):
     ) -> typing.Union[yandex.cloud.ai.assistants.v1.runs.run_service_pb2.ListRunsResponse, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.runs.run_service_pb2.ListRunsResponse]]:
         """List runs in a specific folder."""
 
+    @abc.abstractmethod
+    def Submit(
+        self,
+        request: yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunResponse, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.runs.run_service_pb2.SubmitToRunResponse]]:
+        """Submit event to run
+        For example, submit function call results when the run is waiting for user input.
+        """
+
 def add_RunServiceServicer_to_server(servicer: RunServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

yandex/cloud/ai/dataset/v1/dataset_service_pb2.py

@@ -29,7 +29,7 @@ from yandex.cloud.ai.dataset.v1 import dataset_pb2 as yandex_dot_cloud_dot_ai_do
 from google.protobuf import field_mask_pb2 as google_dot_protobuf_dot_field__mask__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n0yandex/cloud/ai/dataset/v1/dataset_service.proto\x12\x1ayandex.cloud.ai.dataset.v1\x1a\x1dyandex/cloud/validation.proto\x1a yandex/cloud/api/operation.proto\x1a&yandex/cloud/operation/operation.proto\x1a(yandex/cloud/ai/dataset/v1/dataset.proto\x1a google/protobuf/field_mask.proto\"2\n\x16\x44\x65scribeDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"S\n\x17\x44\x65scribeDatasetResponse\x12\x38\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\"2\n\x16ValidateDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"m\n\x17ValidateDatasetMetadata\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x12\n\nvalid_rows\x18\x02 \x01(\x03\x12\x16\n\x0eprocessed_rows\x18\x03 \x01(\x03\x12\x12\n\ntotal_rows\x18\x04 \x01(\x03\"|\n\x17ValidateDatasetResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x10\n\x08is_valid\x18\x02 \x01(\x08\x12;\n\x06\x65rrors\x18\x03 \x03(\x0b\x32+.yandex.cloud.ai.dataset.v1.ValidationError\"0\n\x14\x44\x65leteDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"\x17\n\x15\x44\x65leteDatasetResponse\"\xd5\x02\n\x14\x43reateDatasetRequest\x12\x1a\n\x04name\x18\x01 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=63\x12\x1f\n\tfolder_id\x18\x02 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=50\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x10\n\x08metadata\x18\x04 \x01(\t\x12\x1f\n\ttask_type\x18\x05 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=50\x12L\n\x06labels\x18\x06 \x03(\x0b\x32<.yandex.cloud.ai.dataset.v1.CreateDatasetRequest.LabelsEntry\x12#\n\rupload_format\x18\x07 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=50\x12\x16\n\x0e\x61llow_data_log\x18\x08 \x01(\x08\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"e\n\x15\x43reateDatasetResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x38\n\x07\x64\x61taset\x18\x02 \x01(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\"\x81\x02\n\x14UpdateDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12/\n\x0bupdate_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMask\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12L\n\x06labels\x18\x05 \x03(\x0b\x32<.yandex.cloud.ai.dataset.v1.UpdateDatasetRequest.LabelsEntry\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"Q\n\x15UpdateDatasetResponse\x12\x38\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\"P\n\x18GetUploadDraftUrlRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x1a\n\nsize_bytes\x18\x02 \x01(\x03\x42\x06\xfa\xc7\x31\x02>0\"C\n\x19GetUploadDraftUrlResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x12\n\nupload_url\x18\x02 \x01(\t\"g\n StartMultipartUploadDraftRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x1a\n\nsize_bytes\x18\x02 \x01(\x03\x42\x06\xfa\xc7\x31\x02>0\x12\r\n\x05parts\x18\x03 \x01(\x03\"V\n!StartMultipartUploadDraftResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x1d\n\x15multipart_upload_urls\x18\x02 \x03(\t\"E\n\x10UploadedPartInfo\x12\x1d\n\x08part_num\x18\x01 \x01(\x03\x42\x0b\xfa\xc7\x31\x07\x31-10000\x12\x12\n\x04\x65tag\x18\x02 \x01(\tB\x04\xe8\xc7\x31\x01\"\x83\x01\n!FinishMultipartUploadDraftRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x44\n\x0euploaded_parts\x18\x02 \x03(\x0b\x32,.yandex.cloud.ai.dataset.v1.UploadedPartInfo\"8\n\"FinishMultipartUploadDraftResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\"\xcd\x01\n\x13ListDatasetsRequest\x12\x17\n\tfolder_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12>\n\x06status\x18\x02 \x03(\x0e\x32..yandex.cloud.ai.dataset.v1.DatasetInfo.Status\x12\x1c\n\x14\x64\x61taset_name_pattern\x18\x03 \x01(\t\x12\x18\n\x10task_type_filter\x18\x04 \x03(\t\x12\x11\n\tpage_size\x18\x05 \x01(\x03\x12\x12\n\npage_token\x18\x06 \x01(\t\"j\n\x14ListDatasetsResponse\x12\x39\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"3\n\x18ListUploadFormatsRequest\x12\x17\n\ttask_type\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\",\n\x19ListUploadFormatsResponse\x12\x0f\n\x07\x66ormats\x18\x01 \x03(\t\"3\n\x18ListUploadSchemasRequest\x12\x17\n\ttask_type\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"]\n\x19ListUploadSchemasResponse\x12@\n\x07schemas\x18\x01 \x03(\x0b\x32/.yandex.cloud.ai.dataset.v1.DatasetUploadSchema\"\x12\n\x10ListTypesRequest\"\"\n\x11ListTypesResponse\x12\r\n\x05types\x18\x01 \x03(\t\"4\n\x18GetDatasetPreviewRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"F\n\x19GetDatasetPreviewResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x15\n\rpreview_lines\x18\x02 \x03(\t\"2\n\x16GetDownloadUrlsRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"x\n\x17GetDownloadUrlsResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12I\n\rdownload_urls\x18\x02 \x03(\x0b\x32\x32.yandex.cloud.ai.dataset.v1.DatasetFileDownloadUrl\".\n\x18ListOperationsIdsRequest\x12\x12\n\ndataset_id\x18\x01 \x03(\t\"\xd1\x01\n\x19ListOperationsIdsResponse\x12u\n\x1a\x64\x61taset_id_to_operation_id\x18\x01 \x03(\x0b\x32Q.yandex.cloud.ai.dataset.v1.ListOperationsIdsResponse.DatasetIdToOperationIdEntry\x1a=\n\x1b\x44\x61tasetIdToOperationIdEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x32\x84\x0f\n\x0e\x44\x61tasetService\x12s\n\x08\x44\x65scribe\x12\x32.yandex.cloud.ai.dataset.v1.DescribeDatasetRequest\x1a\x33.yandex.cloud.ai.dataset.v1.DescribeDatasetResponse\x12\x99\x01\n\x08Validate\x12\x32.yandex.cloud.ai.dataset.v1.ValidateDatasetRequest\x1a!.yandex.cloud.operation.Operation\"6\xb2\xd2*2\n\x17ValidateDatasetMetadata\x12\x17ValidateDatasetResponse\x12m\n\x06\x43reate\x12\x30.yandex.cloud.ai.dataset.v1.CreateDatasetRequest\x1a\x31.yandex.cloud.ai.dataset.v1.CreateDatasetResponse\x12m\n\x06Update\x12\x30.yandex.cloud.ai.dataset.v1.UpdateDatasetRequest\x1a\x31.yandex.cloud.ai.dataset.v1.UpdateDatasetResponse\x12m\n\x06\x44\x65lete\x12\x30.yandex.cloud.ai.dataset.v1.DeleteDatasetRequest\x1a\x31.yandex.cloud.ai.dataset.v1.DeleteDatasetResponse\x12i\n\x04List\x12/.yandex.cloud.ai.dataset.v1.ListDatasetsRequest\x1a\x30.yandex.cloud.ai.dataset.v1.ListDatasetsResponse\x12\x85\x01\n\x11ListUploadFormats\x12\x34.yandex.cloud.ai.dataset.v1.ListUploadFormatsRequest\x1a\x35.yandex.cloud.ai.dataset.v1.ListUploadFormatsResponse\"\x03\x88\x02\x01\x12\x80\x01\n\x11ListUploadSchemas\x12\x34.yandex.cloud.ai.dataset.v1.ListUploadSchemasRequest\x1a\x35.yandex.cloud.ai.dataset.v1.ListUploadSchemasResponse\x12\x80\x01\n\x11GetUploadDraftUrl\x12\x34.yandex.cloud.ai.dataset.v1.GetUploadDraftUrlRequest\x1a\x35.yandex.cloud.ai.dataset.v1.GetUploadDraftUrlResponse\x12z\n\x0fGetDownloadUrls\x12\x32.yandex.cloud.ai.dataset.v1.GetDownloadUrlsRequest\x1a\x33.yandex.cloud.ai.dataset.v1.GetDownloadUrlsResponse\x12\x98\x01\n\x19StartMultipartUploadDraft\x12<.yandex.cloud.ai.dataset.v1.StartMultipartUploadDraftRequest\x1a=.yandex.cloud.ai.dataset.v1.StartMultipartUploadDraftResponse\x12\x9b\x01\n\x1a\x46inishMultipartUploadDraft\x12=.yandex.cloud.ai.dataset.v1.FinishMultipartUploadDraftRequest\x1a>.yandex.cloud.ai.dataset.v1.FinishMultipartUploadDraftResponse\x12h\n\tListTypes\x12,.yandex.cloud.ai.dataset.v1.ListTypesRequest\x1a-.yandex.cloud.ai.dataset.v1.ListTypesResponse\x12y\n\nGetPreview\x12\x34.yandex.cloud.ai.dataset.v1.GetDatasetPreviewRequest\x1a\x35.yandex.cloud.ai.dataset.v1.GetDatasetPreviewResponse\x12\x80\x01\n\x11ListOperationsIds\x12\x34.yandex.cloud.ai.dataset.v1.ListOperationsIdsRequest\x1a\x35.yandex.cloud.ai.dataset.v1.ListOperationsIdsResponseBe\n\x1eyandex.cloud.api.ai.dataset.v1ZCgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/dataset/v1;fomob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n0yandex/cloud/ai/dataset/v1/dataset_service.proto\x12\x1ayandex.cloud.ai.dataset.v1\x1a\x1dyandex/cloud/validation.proto\x1a yandex/cloud/api/operation.proto\x1a&yandex/cloud/operation/operation.proto\x1a(yandex/cloud/ai/dataset/v1/dataset.proto\x1a google/protobuf/field_mask.proto\"2\n\x16\x44\x65scribeDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"S\n\x17\x44\x65scribeDatasetResponse\x12\x38\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\"2\n\x16ValidateDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"m\n\x17ValidateDatasetMetadata\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x12\n\nvalid_rows\x18\x02 \x01(\x03\x12\x16\n\x0eprocessed_rows\x18\x03 \x01(\x03\x12\x12\n\ntotal_rows\x18\x04 \x01(\x03\"|\n\x17ValidateDatasetResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x10\n\x08is_valid\x18\x02 \x01(\x08\x12;\n\x06\x65rrors\x18\x03 \x03(\x0b\x32+.yandex.cloud.ai.dataset.v1.ValidationError\"0\n\x14\x44\x65leteDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"\x17\n\x15\x44\x65leteDatasetResponse\"\xd5\x02\n\x14\x43reateDatasetRequest\x12\x1a\n\x04name\x18\x01 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=63\x12\x1f\n\tfolder_id\x18\x02 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=50\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x10\n\x08metadata\x18\x04 \x01(\t\x12\x1f\n\ttask_type\x18\x05 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=50\x12L\n\x06labels\x18\x06 \x03(\x0b\x32<.yandex.cloud.ai.dataset.v1.CreateDatasetRequest.LabelsEntry\x12#\n\rupload_format\x18\x07 \x01(\tB\x0c\xe8\xc7\x31\x01\x8a\xc8\x31\x04<=50\x12\x16\n\x0e\x61llow_data_log\x18\x08 \x01(\x08\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"e\n\x15\x43reateDatasetResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x38\n\x07\x64\x61taset\x18\x02 \x01(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\"\x81\x02\n\x14UpdateDatasetRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12/\n\x0bupdate_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMask\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12L\n\x06labels\x18\x05 \x03(\x0b\x32<.yandex.cloud.ai.dataset.v1.UpdateDatasetRequest.LabelsEntry\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"Q\n\x15UpdateDatasetResponse\x12\x38\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\"P\n\x18GetUploadDraftUrlRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x1a\n\nsize_bytes\x18\x02 \x01(\x03\x42\x06\xfa\xc7\x31\x02>0\"C\n\x19GetUploadDraftUrlResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x12\n\nupload_url\x18\x02 \x01(\t\"g\n StartMultipartUploadDraftRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x1a\n\nsize_bytes\x18\x02 \x01(\x03\x42\x06\xfa\xc7\x31\x02>0\x12\r\n\x05parts\x18\x03 \x01(\x03\"V\n!StartMultipartUploadDraftResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x1d\n\x15multipart_upload_urls\x18\x02 \x03(\t\"E\n\x10UploadedPartInfo\x12\x1d\n\x08part_num\x18\x01 \x01(\x03\x42\x0b\xfa\xc7\x31\x07\x31-10000\x12\x12\n\x04\x65tag\x18\x02 \x01(\tB\x04\xe8\xc7\x31\x01\"\x83\x01\n!FinishMultipartUploadDraftRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x44\n\x0euploaded_parts\x18\x02 \x03(\x0b\x32,.yandex.cloud.ai.dataset.v1.UploadedPartInfo\"8\n\"FinishMultipartUploadDraftResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\"\xcd\x01\n\x13ListDatasetsRequest\x12\x17\n\tfolder_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12>\n\x06status\x18\x02 \x03(\x0e\x32..yandex.cloud.ai.dataset.v1.DatasetInfo.Status\x12\x1c\n\x14\x64\x61taset_name_pattern\x18\x03 \x01(\t\x12\x18\n\x10task_type_filter\x18\x04 \x03(\t\x12\x11\n\tpage_size\x18\x05 \x01(\x03\x12\x12\n\npage_token\x18\x06 \x01(\t\"j\n\x14ListDatasetsResponse\x12\x39\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32\'.yandex.cloud.ai.dataset.v1.DatasetInfo\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"3\n\x18ListUploadFormatsRequest\x12\x17\n\ttask_type\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\",\n\x19ListUploadFormatsResponse\x12\x0f\n\x07\x66ormats\x18\x01 \x03(\t\"F\n\x18ListUploadSchemasRequest\x12\x17\n\ttask_type\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\x12\x11\n\tfolder_id\x18\x02 \x01(\t\"]\n\x19ListUploadSchemasResponse\x12@\n\x07schemas\x18\x01 \x03(\x0b\x32/.yandex.cloud.ai.dataset.v1.DatasetUploadSchema\"%\n\x10ListTypesRequest\x12\x11\n\tfolder_id\x18\x01 \x01(\t\"\"\n\x11ListTypesResponse\x12\r\n\x05types\x18\x01 \x03(\t\"4\n\x18GetDatasetPreviewRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"F\n\x19GetDatasetPreviewResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x15\n\rpreview_lines\x18\x02 \x03(\t\"2\n\x16GetDownloadUrlsRequest\x12\x18\n\ndataset_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"x\n\x17GetDownloadUrlsResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12I\n\rdownload_urls\x18\x02 \x03(\x0b\x32\x32.yandex.cloud.ai.dataset.v1.DatasetFileDownloadUrl\".\n\x18ListOperationsIdsRequest\x12\x12\n\ndataset_id\x18\x01 \x03(\t\"\xd1\x01\n\x19ListOperationsIdsResponse\x12u\n\x1a\x64\x61taset_id_to_operation_id\x18\x01 \x03(\x0b\x32Q.yandex.cloud.ai.dataset.v1.ListOperationsIdsResponse.DatasetIdToOperationIdEntry\x1a=\n\x1b\x44\x61tasetIdToOperationIdEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x32\x84\x0f\n\x0e\x44\x61tasetService\x12s\n\x08\x44\x65scribe\x12\x32.yandex.cloud.ai.dataset.v1.DescribeDatasetRequest\x1a\x33.yandex.cloud.ai.dataset.v1.DescribeDatasetResponse\x12\x99\x01\n\x08Validate\x12\x32.yandex.cloud.ai.dataset.v1.ValidateDatasetRequest\x1a!.yandex.cloud.operation.Operation\"6\xb2\xd2*2\n\x17ValidateDatasetMetadata\x12\x17ValidateDatasetResponse\x12m\n\x06\x43reate\x12\x30.yandex.cloud.ai.dataset.v1.CreateDatasetRequest\x1a\x31.yandex.cloud.ai.dataset.v1.CreateDatasetResponse\x12m\n\x06Update\x12\x30.yandex.cloud.ai.dataset.v1.UpdateDatasetRequest\x1a\x31.yandex.cloud.ai.dataset.v1.UpdateDatasetResponse\x12m\n\x06\x44\x65lete\x12\x30.yandex.cloud.ai.dataset.v1.DeleteDatasetRequest\x1a\x31.yandex.cloud.ai.dataset.v1.DeleteDatasetResponse\x12i\n\x04List\x12/.yandex.cloud.ai.dataset.v1.ListDatasetsRequest\x1a\x30.yandex.cloud.ai.dataset.v1.ListDatasetsResponse\x12\x85\x01\n\x11ListUploadFormats\x12\x34.yandex.cloud.ai.dataset.v1.ListUploadFormatsRequest\x1a\x35.yandex.cloud.ai.dataset.v1.ListUploadFormatsResponse\"\x03\x88\x02\x01\x12\x80\x01\n\x11ListUploadSchemas\x12\x34.yandex.cloud.ai.dataset.v1.ListUploadSchemasRequest\x1a\x35.yandex.cloud.ai.dataset.v1.ListUploadSchemasResponse\x12\x80\x01\n\x11GetUploadDraftUrl\x12\x34.yandex.cloud.ai.dataset.v1.GetUploadDraftUrlRequest\x1a\x35.yandex.cloud.ai.dataset.v1.GetUploadDraftUrlResponse\x12z\n\x0fGetDownloadUrls\x12\x32.yandex.cloud.ai.dataset.v1.GetDownloadUrlsRequest\x1a\x33.yandex.cloud.ai.dataset.v1.GetDownloadUrlsResponse\x12\x98\x01\n\x19StartMultipartUploadDraft\x12<.yandex.cloud.ai.dataset.v1.StartMultipartUploadDraftRequest\x1a=.yandex.cloud.ai.dataset.v1.StartMultipartUploadDraftResponse\x12\x9b\x01\n\x1a\x46inishMultipartUploadDraft\x12=.yandex.cloud.ai.dataset.v1.FinishMultipartUploadDraftRequest\x1a>.yandex.cloud.ai.dataset.v1.FinishMultipartUploadDraftResponse\x12h\n\tListTypes\x12,.yandex.cloud.ai.dataset.v1.ListTypesRequest\x1a-.yandex.cloud.ai.dataset.v1.ListTypesResponse\x12y\n\nGetPreview\x12\x34.yandex.cloud.ai.dataset.v1.GetDatasetPreviewRequest\x1a\x35.yandex.cloud.ai.dataset.v1.GetDatasetPreviewResponse\x12\x80\x01\n\x11ListOperationsIds\x12\x34.yandex.cloud.ai.dataset.v1.ListOperationsIdsRequest\x1a\x35.yandex.cloud.ai.dataset.v1.ListOperationsIdsResponseBe\n\x1eyandex.cloud.api.ai.dataset.v1ZCgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/dataset/v1;fomob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -136,27 +136,27 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_LISTUPLOADFORMATSRESPONSE']._serialized_start=2528
   _globals['_LISTUPLOADFORMATSRESPONSE']._serialized_end=2572
   _globals['_LISTUPLOADSCHEMASREQUEST']._serialized_start=2574
-  _globals['_LISTUPLOADSCHEMASREQUEST']._serialized_end=2625
-  _globals['_LISTUPLOADSCHEMASRESPONSE']._serialized_start=2627
-  _globals['_LISTUPLOADSCHEMASRESPONSE']._serialized_end=2720
-  _globals['_LISTTYPESREQUEST']._serialized_start=2722
-  _globals['_LISTTYPESREQUEST']._serialized_end=2740
-  _globals['_LISTTYPESRESPONSE']._serialized_start=2742
-  _globals['_LISTTYPESRESPONSE']._serialized_end=2776
-  _globals['_GETDATASETPREVIEWREQUEST']._serialized_start=2778
-  _globals['_GETDATASETPREVIEWREQUEST']._serialized_end=2830
-  _globals['_GETDATASETPREVIEWRESPONSE']._serialized_start=2832
-  _globals['_GETDATASETPREVIEWRESPONSE']._serialized_end=2902
-  _globals['_GETDOWNLOADURLSREQUEST']._serialized_start=2904
-  _globals['_GETDOWNLOADURLSREQUEST']._serialized_end=2954
-  _globals['_GETDOWNLOADURLSRESPONSE']._serialized_start=2956
-  _globals['_GETDOWNLOADURLSRESPONSE']._serialized_end=3076
-  _globals['_LISTOPERATIONSIDSREQUEST']._serialized_start=3078
-  _globals['_LISTOPERATIONSIDSREQUEST']._serialized_end=3124
-  _globals['_LISTOPERATIONSIDSRESPONSE']._serialized_start=3127
-  _globals['_LISTOPERATIONSIDSRESPONSE']._serialized_end=3336
-  _globals['_LISTOPERATIONSIDSRESPONSE_DATASETIDTOOPERATIONIDENTRY']._serialized_start=3275
-  _globals['_LISTOPERATIONSIDSRESPONSE_DATASETIDTOOPERATIONIDENTRY']._serialized_end=3336
-  _globals['_DATASETSERVICE']._serialized_start=3339
-  _globals['_DATASETSERVICE']._serialized_end=5263
+  _globals['_LISTUPLOADSCHEMASREQUEST']._serialized_end=2644
+  _globals['_LISTUPLOADSCHEMASRESPONSE']._serialized_start=2646
+  _globals['_LISTUPLOADSCHEMASRESPONSE']._serialized_end=2739
+  _globals['_LISTTYPESREQUEST']._serialized_start=2741
+  _globals['_LISTTYPESREQUEST']._serialized_end=2778
+  _globals['_LISTTYPESRESPONSE']._serialized_start=2780
+  _globals['_LISTTYPESRESPONSE']._serialized_end=2814
+  _globals['_GETDATASETPREVIEWREQUEST']._serialized_start=2816
+  _globals['_GETDATASETPREVIEWREQUEST']._serialized_end=2868
+  _globals['_GETDATASETPREVIEWRESPONSE']._serialized_start=2870
+  _globals['_GETDATASETPREVIEWRESPONSE']._serialized_end=2940
+  _globals['_GETDOWNLOADURLSREQUEST']._serialized_start=2942
+  _globals['_GETDOWNLOADURLSREQUEST']._serialized_end=2992
+  _globals['_GETDOWNLOADURLSRESPONSE']._serialized_start=2994
+  _globals['_GETDOWNLOADURLSRESPONSE']._serialized_end=3114
+  _globals['_LISTOPERATIONSIDSREQUEST']._serialized_start=3116
+  _globals['_LISTOPERATIONSIDSREQUEST']._serialized_end=3162
+  _globals['_LISTOPERATIONSIDSRESPONSE']._serialized_start=3165
+  _globals['_LISTOPERATIONSIDSRESPONSE']._serialized_end=3374
+  _globals['_LISTOPERATIONSIDSRESPONSE_DATASETIDTOOPERATIONIDENTRY']._serialized_start=3313
+  _globals['_LISTOPERATIONSIDSRESPONSE_DATASETIDTOOPERATIONIDENTRY']._serialized_end=3374
+  _globals['_DATASETSERVICE']._serialized_start=3377
+  _globals['_DATASETSERVICE']._serialized_end=5301
 # @@protoc_insertion_point(module_scope)

yandex/cloud/ai/dataset/v1/dataset_service_pb2.pyi

@@ -563,14 +563,18 @@ class ListUploadSchemasRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_TYPE_FIELD_NUMBER: builtins.int
+    FOLDER_ID_FIELD_NUMBER: builtins.int
     task_type: builtins.str
     """Dataset task type to list schemas."""
+    folder_id: builtins.str
+    """Folder to search task type (if empty only public types allowed)"""
     def __init__(
         self,
         *,
         task_type: builtins.str = ...,
+        folder_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["task_type", b"task_type"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["folder_id", b"folder_id", "task_type", b"task_type"]) -> None: ...
 
 global___ListUploadSchemasRequest = ListUploadSchemasRequest
 
@@ -596,9 +600,15 @@ global___ListUploadSchemasResponse = ListUploadSchemasResponse
 class ListTypesRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    FOLDER_ID_FIELD_NUMBER: builtins.int
+    folder_id: builtins.str
+    """Folder to search task types (if empty only public types will be returned)"""
     def __init__(
         self,
+        *,
+        folder_id: builtins.str = ...,
     ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["folder_id", b"folder_id"]) -> None: ...
 
 global___ListTypesRequest = ListTypesRequest
 

yandex/cloud/ai/foundation_models/v1/batch_inference_task_status_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: yandex/cloud/ai/foundation_models/v1/batch_inference_task_status.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/foundation_models/v1/batch_inference_task_status.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nFyandex/cloud/ai/foundation_models/v1/batch_inference_task_status.proto\x12$yandex.cloud.ai.foundation_models.v1*\x9b\x01\n\x18\x42\x61tchInferenceTaskStatus\x12+\n\'BATCH_INFERENCE_TASK_STATUS_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\x86\x01\n(yandex.cloud.api.ai.foundation_models.v1ZZgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1;foundation_modelsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'yandex.cloud.ai.foundation_models.v1.batch_inference_task_status_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n(yandex.cloud.api.ai.foundation_models.v1ZZgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1;foundation_models'
+  _globals['_BATCHINFERENCETASKSTATUS']._serialized_start=113
+  _globals['_BATCHINFERENCETASKSTATUS']._serialized_end=268
+# @@protoc_insertion_point(module_scope)

yandex/cloud/ai/foundation_models/v1/batch_inference_task_status_pb2.pyi

@@ -0,0 +1,43 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _BatchInferenceTaskStatus:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _BatchInferenceTaskStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BatchInferenceTaskStatus.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    BATCH_INFERENCE_TASK_STATUS_UNSPECIFIED: _BatchInferenceTaskStatus.ValueType  # 0
+    CREATED: _BatchInferenceTaskStatus.ValueType  # 1
+    PENDING: _BatchInferenceTaskStatus.ValueType  # 2
+    IN_PROGRESS: _BatchInferenceTaskStatus.ValueType  # 3
+    COMPLETED: _BatchInferenceTaskStatus.ValueType  # 4
+    FAILED: _BatchInferenceTaskStatus.ValueType  # 5
+    CANCELED: _BatchInferenceTaskStatus.ValueType  # 6
+
+class BatchInferenceTaskStatus(_BatchInferenceTaskStatus, metaclass=_BatchInferenceTaskStatusEnumTypeWrapper):
+    """Enum representing the status of a batch inference."""
+
+BATCH_INFERENCE_TASK_STATUS_UNSPECIFIED: BatchInferenceTaskStatus.ValueType  # 0
+CREATED: BatchInferenceTaskStatus.ValueType  # 1
+PENDING: BatchInferenceTaskStatus.ValueType  # 2
+IN_PROGRESS: BatchInferenceTaskStatus.ValueType  # 3
+COMPLETED: BatchInferenceTaskStatus.ValueType  # 4
+FAILED: BatchInferenceTaskStatus.ValueType  # 5
+CANCELED: BatchInferenceTaskStatus.ValueType  # 6
+global___BatchInferenceTaskStatus = BatchInferenceTaskStatus

yandex/cloud/ai/foundation_models/v1/batch_inference_task_status_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/foundation_models/v1/batch_inference_task_status_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/foundation_models/v1/batch_inference_task_status_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/foundation_models/v1/embedding/embedding_service_pb2.py

@@ -23,9 +23,10 @@ _sym_db = _symbol_database.Default()
 
 
 from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nFyandex/cloud/ai/foundation_models/v1/embedding/embedding_service.proto\x12$yandex.cloud.ai.foundation_models.v1\x1a\x1cgoogle/api/annotations.proto\"7\n\x14TextEmbeddingRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\"U\n\x15TextEmbeddingResponse\x12\x11\n\tembedding\x18\x01 \x03(\x01\x12\x12\n\nnum_tokens\x18\x02 \x01(\x03\x12\x15\n\rmodel_version\x18\x03 \x01(\t2\xcd\x01\n\x11\x45mbeddingsService\x12\xb7\x01\n\rTextEmbedding\x12:.yandex.cloud.ai.foundation_models.v1.TextEmbeddingRequest\x1a;.yandex.cloud.ai.foundation_models.v1.TextEmbeddingResponse\"-\x82\xd3\xe4\x93\x02\'\"\"/foundationModels/v1/textEmbedding:\x01*B\x90\x01\n(yandex.cloud.api.ai.foundation_models.v1Zdgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/embedding;foundation_modelsb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nFyandex/cloud/ai/foundation_models/v1/embedding/embedding_service.proto\x12$yandex.cloud.ai.foundation_models.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1egoogle/protobuf/wrappers.proto\"a\n\x14TextEmbeddingRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\x12(\n\x03\x64im\x18\x03 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"U\n\x15TextEmbeddingResponse\x12\x11\n\tembedding\x18\x01 \x03(\x01\x12\x12\n\nnum_tokens\x18\x02 \x01(\x03\x12\x15\n\rmodel_version\x18\x03 \x01(\t2\xcd\x01\n\x11\x45mbeddingsService\x12\xb7\x01\n\rTextEmbedding\x12:.yandex.cloud.ai.foundation_models.v1.TextEmbeddingRequest\x1a;.yandex.cloud.ai.foundation_models.v1.TextEmbeddingResponse\"-\x82\xd3\xe4\x93\x02\'\"\"/foundationModels/v1/textEmbedding:\x01*B\x90\x01\n(yandex.cloud.api.ai.foundation_models.v1Zdgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/embedding;foundation_modelsb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -35,10 +36,10 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._serialized_options = b'\n(yandex.cloud.api.ai.foundation_models.v1Zdgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/embedding;foundation_models'
   _globals['_EMBEDDINGSSERVICE'].methods_by_name['TextEmbedding']._loaded_options = None
   _globals['_EMBEDDINGSSERVICE'].methods_by_name['TextEmbedding']._serialized_options = b'\202\323\344\223\002\'\"\"/foundationModels/v1/textEmbedding:\001*'
-  _globals['_TEXTEMBEDDINGREQUEST']._serialized_start=142
-  _globals['_TEXTEMBEDDINGREQUEST']._serialized_end=197
-  _globals['_TEXTEMBEDDINGRESPONSE']._serialized_start=199
-  _globals['_TEXTEMBEDDINGRESPONSE']._serialized_end=284
-  _globals['_EMBEDDINGSSERVICE']._serialized_start=287
-  _globals['_EMBEDDINGSSERVICE']._serialized_end=492
+  _globals['_TEXTEMBEDDINGREQUEST']._serialized_start=174
+  _globals['_TEXTEMBEDDINGREQUEST']._serialized_end=271
+  _globals['_TEXTEMBEDDINGRESPONSE']._serialized_start=273
+  _globals['_TEXTEMBEDDINGRESPONSE']._serialized_end=358
+  _globals['_EMBEDDINGSSERVICE']._serialized_start=361
+  _globals['_EMBEDDINGSSERVICE']._serialized_end=566
 # @@protoc_insertion_point(module_scope)

yandex/cloud/ai/foundation_models/v1/embedding/embedding_service_pb2.pyi

@@ -8,6 +8,7 @@ import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
+import google.protobuf.wrappers_pb2
 import typing
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
@@ -20,17 +21,24 @@ class TextEmbeddingRequest(google.protobuf.message.Message):
 
     MODEL_URI_FIELD_NUMBER: builtins.int
     TEXT_FIELD_NUMBER: builtins.int
+    DIM_FIELD_NUMBER: builtins.int
     model_uri: builtins.str
     """The [model URI](/docs/foundation-models/concepts/embeddings) to be used for obtaining text embeddings."""
     text: builtins.str
     """The input text for which the embedding is requested."""
+    @property
+    def dim(self) -> google.protobuf.wrappers_pb2.Int64Value:
+        """Optional parameter to specify embedding dimension for models that support multi-dimensional outputs"""
+
     def __init__(
         self,
         *,
         model_uri: builtins.str = ...,
         text: builtins.str = ...,
+        dim: google.protobuf.wrappers_pb2.Int64Value | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["model_uri", b"model_uri", "text", b"text"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["dim", b"dim"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["dim", b"dim", "model_uri", b"model_uri", "text", b"text"]) -> None: ...
 
 global___TextEmbeddingRequest = TextEmbeddingRequest
 

yandex/cloud/ai/foundation_models/v1/text_generation/text_generation_service_pb2.py

@@ -24,11 +24,12 @@ _sym_db = _symbol_database.Default()
 
 from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
 from yandex.cloud.ai.foundation_models.v1 import text_common_pb2 as yandex_dot_cloud_dot_ai_dot_foundation__models_dot_v1_dot_text__common__pb2
+from yandex.cloud.ai.foundation_models.v1 import batch_inference_task_status_pb2 as yandex_dot_cloud_dot_ai_dot_foundation__models_dot_v1_dot_batch__inference__task__status__pb2
 from yandex.cloud.api import operation_pb2 as yandex_dot_cloud_dot_api_dot_operation__pb2
 from yandex.cloud.operation import operation_pb2 as yandex_dot_cloud_dot_operation_dot_operation__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nRyandex/cloud/ai/foundation_models/v1/text_generation/text_generation_service.proto\x12$yandex.cloud.ai.foundation_models.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x36yandex/cloud/ai/foundation_models/v1/text_common.proto\x1a yandex/cloud/api/operation.proto\x1a&yandex/cloud/operation/operation.proto\"\xe9\x02\n\x11\x43ompletionRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12S\n\x12\x63ompletion_options\x18\x02 \x01(\x0b\x32\x37.yandex.cloud.ai.foundation_models.v1.CompletionOptions\x12?\n\x08messages\x18\x03 \x03(\x0b\x32-.yandex.cloud.ai.foundation_models.v1.Message\x12\x39\n\x05tools\x18\x04 \x03(\x0b\x32*.yandex.cloud.ai.foundation_models.v1.Tool\x12\x15\n\x0bjson_object\x18\x05 \x01(\x08H\x00\x12G\n\x0bjson_schema\x18\x06 \x01(\x0b\x32\x30.yandex.cloud.ai.foundation_models.v1.JsonSchemaH\x00\x42\x10\n\x0eResponseFormat\"\xb7\x01\n\x12\x43ompletionResponse\x12G\n\x0c\x61lternatives\x18\x01 \x03(\x0b\x32\x31.yandex.cloud.ai.foundation_models.v1.Alternative\x12\x41\n\x05usage\x18\x02 \x01(\x0b\x32\x32.yandex.cloud.ai.foundation_models.v1.ContentUsage\x12\x15\n\rmodel_version\x18\x03 \x01(\t\"2\n\x0fTokenizeRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\"f\n\x10TokenizeResponse\x12;\n\x06tokens\x18\x01 \x03(\x0b\x32+.yandex.cloud.ai.foundation_models.v1.Token\x12\x15\n\rmodel_version\x18\x02 \x01(\t2\xc7\x01\n\x15TextGenerationService\x12\xad\x01\n\nCompletion\x12\x37.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a\x38.yandex.cloud.ai.foundation_models.v1.CompletionResponse\"*\x82\xd3\xe4\x93\x02$\"\x1f/foundationModels/v1/completion:\x01*0\x01\x32\xd0\x01\n\x1aTextGenerationAsyncService\x12\xb1\x01\n\nCompletion\x12\x37.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a!.yandex.cloud.operation.Operation\"G\xb2\xd2*\x14\x12\x12\x43ompletionResponse\x82\xd3\xe4\x93\x02)\"$/foundationModels/v1/completionAsync:\x01*2\xf4\x02\n\x10TokenizerService\x12\xa3\x01\n\x08Tokenize\x12\x35.yandex.cloud.ai.foundation_models.v1.TokenizeRequest\x1a\x36.yandex.cloud.ai.foundation_models.v1.TokenizeResponse\"(\x82\xd3\xe4\x93\x02\"\"\x1d/foundationModels/v1/tokenize:\x01*\x12\xb9\x01\n\x12TokenizeCompletion\x12\x37.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a\x36.yandex.cloud.ai.foundation_models.v1.TokenizeResponse\"2\x82\xd3\xe4\x93\x02,\"\'/foundationModels/v1/tokenizeCompletion:\x01*B\x96\x01\n(yandex.cloud.api.ai.foundation_models.v1Zjgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/text_generation;foundation_modelsb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nRyandex/cloud/ai/foundation_models/v1/text_generation/text_generation_service.proto\x12$yandex.cloud.ai.foundation_models.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x36yandex/cloud/ai/foundation_models/v1/text_common.proto\x1a\x46yandex/cloud/ai/foundation_models/v1/batch_inference_task_status.proto\x1a yandex/cloud/api/operation.proto\x1a&yandex/cloud/operation/operation.proto\"\xe9\x02\n\x11\x43ompletionRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12S\n\x12\x63ompletion_options\x18\x02 \x01(\x0b\x32\x37.yandex.cloud.ai.foundation_models.v1.CompletionOptions\x12?\n\x08messages\x18\x03 \x03(\x0b\x32-.yandex.cloud.ai.foundation_models.v1.Message\x12\x39\n\x05tools\x18\x04 \x03(\x0b\x32*.yandex.cloud.ai.foundation_models.v1.Tool\x12\x15\n\x0bjson_object\x18\x05 \x01(\x08H\x00\x12G\n\x0bjson_schema\x18\x06 \x01(\x0b\x32\x30.yandex.cloud.ai.foundation_models.v1.JsonSchemaH\x00\x42\x10\n\x0eResponseFormat\"\xb7\x01\n\x12\x43ompletionResponse\x12G\n\x0c\x61lternatives\x18\x01 \x03(\x0b\x32\x31.yandex.cloud.ai.foundation_models.v1.Alternative\x12\x41\n\x05usage\x18\x02 \x01(\x0b\x32\x32.yandex.cloud.ai.foundation_models.v1.ContentUsage\x12\x15\n\rmodel_version\x18\x03 \x01(\t\"\xaf\x01\n\x16\x42\x61tchCompletionRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12S\n\x12\x63ompletion_options\x18\x02 \x01(\x0b\x32\x37.yandex.cloud.ai.foundation_models.v1.CompletionOptions\x12\x1b\n\x11source_dataset_id\x18\x03 \x01(\tH\x00\x42\x10\n\x0erequest_format\"\xb1\x01\n\x17\x42\x61tchCompletionMetadata\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12S\n\x0btask_status\x18\x02 \x01(\x0e\x32>.yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus\x12\x19\n\x11\x63ompleted_batches\x18\x03 \x01(\x03\x12\x15\n\rtotal_batches\x18\x04 \x01(\x03\"\x9a\x01\n\x17\x42\x61tchCompletionResponse\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12S\n\x0btask_status\x18\x02 \x01(\x0e\x32>.yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus\x12\x19\n\x11result_dataset_id\x18\x03 \x01(\t\"2\n\x0fTokenizeRequest\x12\x11\n\tmodel_uri\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\"f\n\x10TokenizeResponse\x12;\n\x06tokens\x18\x01 \x03(\x0b\x32+.yandex.cloud.ai.foundation_models.v1.Token\x12\x15\n\rmodel_version\x18\x02 \x01(\t2\xc7\x01\n\x15TextGenerationService\x12\xad\x01\n\nCompletion\x12\x37.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a\x38.yandex.cloud.ai.foundation_models.v1.CompletionResponse\"*\x82\xd3\xe4\x93\x02$\"\x1f/foundationModels/v1/completion:\x01*0\x01\x32\xd0\x01\n\x1aTextGenerationAsyncService\x12\xb1\x01\n\nCompletion\x12\x37.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a!.yandex.cloud.operation.Operation\"G\xb2\xd2*\x14\x12\x12\x43ompletionResponse\x82\xd3\xe4\x93\x02)\"$/foundationModels/v1/completionAsync:\x01*2\xf3\x01\n\x1aTextGenerationBatchService\x12\xd4\x01\n\nCompletion\x12<.yandex.cloud.ai.foundation_models.v1.BatchCompletionRequest\x1a!.yandex.cloud.operation.Operation\"e\xb2\xd2*2\n\x17\x42\x61tchCompletionMetadata\x12\x17\x42\x61tchCompletionResponse\x82\xd3\xe4\x93\x02)\"$/foundationModels/v1/completionBatch:\x01*2\xf4\x02\n\x10TokenizerService\x12\xa3\x01\n\x08Tokenize\x12\x35.yandex.cloud.ai.foundation_models.v1.TokenizeRequest\x1a\x36.yandex.cloud.ai.foundation_models.v1.TokenizeResponse\"(\x82\xd3\xe4\x93\x02\"\"\x1d/foundationModels/v1/tokenize:\x01*\x12\xb9\x01\n\x12TokenizeCompletion\x12\x37.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a\x36.yandex.cloud.ai.foundation_models.v1.TokenizeResponse\"2\x82\xd3\xe4\x93\x02,\"\'/foundationModels/v1/tokenizeCompletion:\x01*B\x96\x01\n(yandex.cloud.api.ai.foundation_models.v1Zjgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/text_generation;foundation_modelsb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -40,22 +41,32 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_TEXTGENERATIONSERVICE'].methods_by_name['Completion']._serialized_options = b'\202\323\344\223\002$\"\037/foundationModels/v1/completion:\001*'
   _globals['_TEXTGENERATIONASYNCSERVICE'].methods_by_name['Completion']._loaded_options = None
   _globals['_TEXTGENERATIONASYNCSERVICE'].methods_by_name['Completion']._serialized_options = b'\262\322*\024\022\022CompletionResponse\202\323\344\223\002)\"$/foundationModels/v1/completionAsync:\001*'
+  _globals['_TEXTGENERATIONBATCHSERVICE'].methods_by_name['Completion']._loaded_options = None
+  _globals['_TEXTGENERATIONBATCHSERVICE'].methods_by_name['Completion']._serialized_options = b'\262\322*2\n\027BatchCompletionMetadata\022\027BatchCompletionResponse\202\323\344\223\002)\"$/foundationModels/v1/completionBatch:\001*'
   _globals['_TOKENIZERSERVICE'].methods_by_name['Tokenize']._loaded_options = None
   _globals['_TOKENIZERSERVICE'].methods_by_name['Tokenize']._serialized_options = b'\202\323\344\223\002\"\"\035/foundationModels/v1/tokenize:\001*'
   _globals['_TOKENIZERSERVICE'].methods_by_name['TokenizeCompletion']._loaded_options = None
   _globals['_TOKENIZERSERVICE'].methods_by_name['TokenizeCompletion']._serialized_options = b'\202\323\344\223\002,\"\'/foundationModels/v1/tokenizeCompletion:\001*'
-  _globals['_COMPLETIONREQUEST']._serialized_start=285
-  _globals['_COMPLETIONREQUEST']._serialized_end=646
-  _globals['_COMPLETIONRESPONSE']._serialized_start=649
-  _globals['_COMPLETIONRESPONSE']._serialized_end=832
-  _globals['_TOKENIZEREQUEST']._serialized_start=834
-  _globals['_TOKENIZEREQUEST']._serialized_end=884
-  _globals['_TOKENIZERESPONSE']._serialized_start=886
-  _globals['_TOKENIZERESPONSE']._serialized_end=988
-  _globals['_TEXTGENERATIONSERVICE']._serialized_start=991
-  _globals['_TEXTGENERATIONSERVICE']._serialized_end=1190
-  _globals['_TEXTGENERATIONASYNCSERVICE']._serialized_start=1193
-  _globals['_TEXTGENERATIONASYNCSERVICE']._serialized_end=1401
-  _globals['_TOKENIZERSERVICE']._serialized_start=1404
-  _globals['_TOKENIZERSERVICE']._serialized_end=1776
+  _globals['_COMPLETIONREQUEST']._serialized_start=357
+  _globals['_COMPLETIONREQUEST']._serialized_end=718
+  _globals['_COMPLETIONRESPONSE']._serialized_start=721
+  _globals['_COMPLETIONRESPONSE']._serialized_end=904
+  _globals['_BATCHCOMPLETIONREQUEST']._serialized_start=907
+  _globals['_BATCHCOMPLETIONREQUEST']._serialized_end=1082
+  _globals['_BATCHCOMPLETIONMETADATA']._serialized_start=1085
+  _globals['_BATCHCOMPLETIONMETADATA']._serialized_end=1262
+  _globals['_BATCHCOMPLETIONRESPONSE']._serialized_start=1265
+  _globals['_BATCHCOMPLETIONRESPONSE']._serialized_end=1419
+  _globals['_TOKENIZEREQUEST']._serialized_start=1421
+  _globals['_TOKENIZEREQUEST']._serialized_end=1471
+  _globals['_TOKENIZERESPONSE']._serialized_start=1473
+  _globals['_TOKENIZERESPONSE']._serialized_end=1575
+  _globals['_TEXTGENERATIONSERVICE']._serialized_start=1578
+  _globals['_TEXTGENERATIONSERVICE']._serialized_end=1777
+  _globals['_TEXTGENERATIONASYNCSERVICE']._serialized_start=1780
+  _globals['_TEXTGENERATIONASYNCSERVICE']._serialized_end=1988
+  _globals['_TEXTGENERATIONBATCHSERVICE']._serialized_start=1991
+  _globals['_TEXTGENERATIONBATCHSERVICE']._serialized_end=2234
+  _globals['_TOKENIZERSERVICE']._serialized_start=2237
+  _globals['_TOKENIZERSERVICE']._serialized_end=2609
 # @@protoc_insertion_point(module_scope)