yandexcloud 0.317.0__py3-none-any.whl → 0.318.0__py3-none-any.whl

yandex/cloud/ai/assistants/v1/assistant_service_pb2_grpc.py

@@ -0,0 +1,232 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from yandex.cloud.ai.assistants.v1 import assistant_pb2 as yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2
+from yandex.cloud.ai.assistants.v1 import assistant_service_pb2 as yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2
+
+
+class AssistantServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Create = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.AssistantService/Create',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.CreateAssistantRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.FromString,
+                )
+        self.Get = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.AssistantService/Get',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.GetAssistantRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.FromString,
+                )
+        self.Update = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.AssistantService/Update',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.UpdateAssistantRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.FromString,
+                )
+        self.Delete = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.AssistantService/Delete',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.DeleteAssistantRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.DeleteAssistantResponse.FromString,
+                )
+        self.List = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.AssistantService/List',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantsRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantsResponse.FromString,
+                )
+        self.ListVersions = channel.unary_unary(
+                '/yandex.cloud.ai.assistants.v1.AssistantService/ListVersions',
+                request_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantVersionsRequest.SerializeToString,
+                response_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantVersionsResponse.FromString,
+                )
+
+
+class AssistantServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def Create(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Get(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Update(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Delete(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def List(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListVersions(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_AssistantServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Create': grpc.unary_unary_rpc_method_handler(
+                    servicer.Create,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.CreateAssistantRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.SerializeToString,
+            ),
+            'Get': grpc.unary_unary_rpc_method_handler(
+                    servicer.Get,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.GetAssistantRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.SerializeToString,
+            ),
+            'Update': grpc.unary_unary_rpc_method_handler(
+                    servicer.Update,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.UpdateAssistantRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.SerializeToString,
+            ),
+            'Delete': grpc.unary_unary_rpc_method_handler(
+                    servicer.Delete,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.DeleteAssistantRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.DeleteAssistantResponse.SerializeToString,
+            ),
+            'List': grpc.unary_unary_rpc_method_handler(
+                    servicer.List,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantsRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantsResponse.SerializeToString,
+            ),
+            'ListVersions': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListVersions,
+                    request_deserializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantVersionsRequest.FromString,
+                    response_serializer=yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantVersionsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'yandex.cloud.ai.assistants.v1.AssistantService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class AssistantService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def Create(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.AssistantService/Create',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.CreateAssistantRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def Get(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.AssistantService/Get',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.GetAssistantRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def Update(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.AssistantService/Update',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.UpdateAssistantRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__pb2.Assistant.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @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.assistants.v1.AssistantService/Delete',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.DeleteAssistantRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.DeleteAssistantResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @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.assistants.v1.AssistantService/List',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantsRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ListVersions(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.AssistantService/ListVersions',
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantVersionsRequest.SerializeToString,
+            yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_assistant__service__pb2.ListAssistantVersionsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

yandex/cloud/ai/assistants/v1/assistant_service_pb2_grpc.pyi

@@ -0,0 +1,127 @@
+"""
+@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.assistants.v1.assistant_pb2
+import yandex.cloud.ai.assistants.v1.assistant_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 AssistantServiceStub:
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
+    Create: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.CreateAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant,
+    ]
+
+    Get: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.GetAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant,
+    ]
+
+    Update: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.UpdateAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant,
+    ]
+
+    Delete: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantResponse,
+    ]
+
+    List: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsRequest,
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsResponse,
+    ]
+
+    ListVersions: grpc.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsRequest,
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsResponse,
+    ]
+
+class AssistantServiceAsyncStub:
+    Create: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.CreateAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant,
+    ]
+
+    Get: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.GetAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant,
+    ]
+
+    Update: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.UpdateAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant,
+    ]
+
+    Delete: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantRequest,
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantResponse,
+    ]
+
+    List: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsRequest,
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsResponse,
+    ]
+
+    ListVersions: grpc.aio.UnaryUnaryMultiCallable[
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsRequest,
+        yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsResponse,
+    ]
+
+class AssistantServiceServicer(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def Create(
+        self,
+        request: yandex.cloud.ai.assistants.v1.assistant_service_pb2.CreateAssistantRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant]]: ...
+
+    @abc.abstractmethod
+    def Get(
+        self,
+        request: yandex.cloud.ai.assistants.v1.assistant_service_pb2.GetAssistantRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant]]: ...
+
+    @abc.abstractmethod
+    def Update(
+        self,
+        request: yandex.cloud.ai.assistants.v1.assistant_service_pb2.UpdateAssistantRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.assistant_pb2.Assistant]]: ...
+
+    @abc.abstractmethod
+    def Delete(
+        self,
+        request: yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantResponse, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.assistant_service_pb2.DeleteAssistantResponse]]: ...
+
+    @abc.abstractmethod
+    def List(
+        self,
+        request: yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsResponse, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantsResponse]]: ...
+
+    @abc.abstractmethod
+    def ListVersions(
+        self,
+        request: yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsResponse, collections.abc.Awaitable[yandex.cloud.ai.assistants.v1.assistant_service_pb2.ListAssistantVersionsResponse]]: ...
+
+def add_AssistantServiceServicer_to_server(servicer: AssistantServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

yandex/cloud/ai/assistants/v1/common_pb2.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: yandex/cloud/ai/assistants/v1/common.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*yandex/cloud/ai/assistants/v1/common.proto\x12\x1dyandex.cloud.ai.assistants.v1\x1a\x1egoogle/protobuf/wrappers.proto\"Q\n\x17PromptTruncationOptions\x12\x36\n\x11max_prompt_tokens\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"w\n\x11\x43ompletionOptions\x12/\n\nmax_tokens\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12\x31\n\x0btemperature\x18\x03 \x01(\x0b\x32\x1c.google.protobuf.DoubleValue\"a\n\x0fSearchIndexTool\x12\x18\n\x10search_index_ids\x18\x01 \x03(\t\x12\x34\n\x0fmax_num_results\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"Z\n\x04Tool\x12\x46\n\x0csearch_index\x18\x01 \x01(\x0b\x32..yandex.cloud.ai.assistants.v1.SearchIndexToolH\x00\x42\n\n\x08ToolTypeBq\n!yandex.cloud.api.ai.assistants.v1ZLgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1;assistantsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'yandex.cloud.ai.assistants.v1.common_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!yandex.cloud.api.ai.assistants.v1ZLgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1;assistants'
+  _globals['_PROMPTTRUNCATIONOPTIONS']._serialized_start=109
+  _globals['_PROMPTTRUNCATIONOPTIONS']._serialized_end=190
+  _globals['_COMPLETIONOPTIONS']._serialized_start=192
+  _globals['_COMPLETIONOPTIONS']._serialized_end=311
+  _globals['_SEARCHINDEXTOOL']._serialized_start=313
+  _globals['_SEARCHINDEXTOOL']._serialized_end=410
+  _globals['_TOOL']._serialized_start=412
+  _globals['_TOOL']._serialized_end=502
+# @@protoc_insertion_point(module_scope)

yandex/cloud/ai/assistants/v1/common_pb2.pyi

@@ -0,0 +1,91 @@
+"""
+@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.message
+import google.protobuf.wrappers_pb2
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class PromptTruncationOptions(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MAX_PROMPT_TOKENS_FIELD_NUMBER: builtins.int
+    @property
+    def max_prompt_tokens(self) -> google.protobuf.wrappers_pb2.Int64Value: ...
+    def __init__(
+        self,
+        *,
+        max_prompt_tokens: google.protobuf.wrappers_pb2.Int64Value | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["max_prompt_tokens", b"max_prompt_tokens"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["max_prompt_tokens", b"max_prompt_tokens"]) -> None: ...
+
+global___PromptTruncationOptions = PromptTruncationOptions
+
+@typing.final
+class CompletionOptions(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MAX_TOKENS_FIELD_NUMBER: builtins.int
+    TEMPERATURE_FIELD_NUMBER: builtins.int
+    @property
+    def max_tokens(self) -> google.protobuf.wrappers_pb2.Int64Value: ...
+    @property
+    def temperature(self) -> google.protobuf.wrappers_pb2.DoubleValue: ...
+    def __init__(
+        self,
+        *,
+        max_tokens: google.protobuf.wrappers_pb2.Int64Value | None = ...,
+        temperature: google.protobuf.wrappers_pb2.DoubleValue | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["max_tokens", b"max_tokens", "temperature", b"temperature"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["max_tokens", b"max_tokens", "temperature", b"temperature"]) -> None: ...
+
+global___CompletionOptions = CompletionOptions
+
+@typing.final
+class SearchIndexTool(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SEARCH_INDEX_IDS_FIELD_NUMBER: builtins.int
+    MAX_NUM_RESULTS_FIELD_NUMBER: builtins.int
+    @property
+    def search_index_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    @property
+    def max_num_results(self) -> google.protobuf.wrappers_pb2.Int64Value: ...
+    def __init__(
+        self,
+        *,
+        search_index_ids: collections.abc.Iterable[builtins.str] | None = ...,
+        max_num_results: google.protobuf.wrappers_pb2.Int64Value | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["max_num_results", b"max_num_results"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["max_num_results", b"max_num_results", "search_index_ids", b"search_index_ids"]) -> None: ...
+
+global___SearchIndexTool = SearchIndexTool
+
+@typing.final
+class Tool(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SEARCH_INDEX_FIELD_NUMBER: builtins.int
+    @property
+    def search_index(self) -> global___SearchIndexTool: ...
+    def __init__(
+        self,
+        *,
+        search_index: global___SearchIndexTool | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["ToolType", b"ToolType", "search_index", b"search_index"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["ToolType", b"ToolType", "search_index", b"search_index"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["ToolType", b"ToolType"]) -> typing.Literal["search_index"] | None: ...
+
+global___Tool = Tool

yandex/cloud/ai/assistants/v1/common_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

yandex/cloud/ai/assistants/v1/common_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/assistants/v1/runs/run_pb2.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: yandex/cloud/ai/assistants/v1/runs/run.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from yandex.cloud.ai.common import common_pb2 as yandex_dot_cloud_dot_ai_dot_common_dot_common__pb2
+from yandex.cloud.ai.assistants.v1 import common_pb2 as yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_common__pb2
+from yandex.cloud.ai.assistants.v1.threads import message_pb2 as yandex_dot_cloud_dot_ai_dot_assistants_dot_v1_dot_threads_dot_message__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,yandex/cloud/ai/assistants/v1/runs/run.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\x1fgoogle/protobuf/timestamp.proto\"\xa7\x04\n\x03Run\x12\n\n\x02id\x18\x01 \x01(\t\x12\x14\n\x0c\x61ssistant_id\x18\x02 \x01(\t\x12\x11\n\tthread_id\x18\x03 \x01(\t\x12\x12\n\ncreated_by\x18\x04 \x01(\t\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x43\n\x06labels\x18\x06 \x03(\x0b\x32\x33.yandex.cloud.ai.assistants.v1.runs.Run.LabelsEntry\x12;\n\x05state\x18\x07 \x01(\x0b\x32,.yandex.cloud.ai.assistants.v1.runs.RunState\x12?\n\x05usage\x18\x08 \x01(\x0b\x32\x30.yandex.cloud.ai.assistants.v1.runs.ContentUsage\x12`\n custom_prompt_truncation_options\x18\t \x01(\x0b\x32\x36.yandex.cloud.ai.assistants.v1.PromptTruncationOptions\x12S\n\x19\x63ustom_completion_options\x18\n \x01(\x0b\x32\x30.yandex.cloud.ai.assistants.v1.CompletionOptions\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xbe\x02\n\x08RunState\x12\x46\n\x06status\x18\x01 \x01(\x0e\x32\x36.yandex.cloud.ai.assistants.v1.runs.RunState.RunStatus\x12.\n\x05\x65rror\x18\x02 \x01(\x0b\x32\x1d.yandex.cloud.ai.common.ErrorH\x00\x12K\n\x11\x63ompleted_message\x18\x03 \x01(\x0b\x32..yandex.cloud.ai.assistants.v1.threads.MessageH\x00\"`\n\tRunStatus\x12\x1a\n\x16RUN_STATUS_UNSPECIFIED\x10\x00\x12\x0b\n\x07PENDING\x10\x01\x12\x0f\n\x0bIN_PROGRESS\x10\x02\x12\n\n\x06\x46\x41ILED\x10\x03\x12\r\n\tCOMPLETED\x10\x04\x42\x0b\n\tStateData\"V\n\x0c\x43ontentUsage\x12\x15\n\rprompt_tokens\x18\x01 \x01(\x03\x12\x19\n\x11\x63ompletion_tokens\x18\x02 \x01(\x03\x12\x14\n\x0ctotal_tokens\x18\x03 \x01(\x03\x42u\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)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'yandex.cloud.ai.assistants.v1.runs.run_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n&yandex.cloud.api.ai.assistants.v1.runsZKgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/runs;runs'
+  _RUN_LABELSENTRY._options = None
+  _RUN_LABELSENTRY._serialized_options = b'8\001'
+  _globals['_RUN']._serialized_start=252
+  _globals['_RUN']._serialized_end=803
+  _globals['_RUN_LABELSENTRY']._serialized_start=758
+  _globals['_RUN_LABELSENTRY']._serialized_end=803
+  _globals['_RUNSTATE']._serialized_start=806
+  _globals['_RUNSTATE']._serialized_end=1124
+  _globals['_RUNSTATE_RUNSTATUS']._serialized_start=1015
+  _globals['_RUNSTATE_RUNSTATUS']._serialized_end=1111
+  _globals['_CONTENTUSAGE']._serialized_start=1126
+  _globals['_CONTENTUSAGE']._serialized_end=1212
+# @@protoc_insertion_point(module_scope)