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

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

@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6yandex/cloud/ai/assistants/v1/searchindex/common.proto\x12)yandex.cloud.ai.assistants.v1.searchindex\x1a\x1egoogle/protobuf/wrappers.proto\"U\n\x16StaticChunkingStrategy\x12\x1d\n\x15max_chunk_size_tokens\x18\x01 \x01(\x03\x12\x1c\n\x14\x63hunk_overlap_tokens\x18\x02 \x01(\x03\"|\n\x10\x43hunkingStrategy\x12\\\n\x0fstatic_strategy\x18\x01 \x01(\x0b\x32\x41.yandex.cloud.ai.assistants.v1.searchindex.StaticChunkingStrategyH\x00\x42\n\n\x08Strategy\"\x9c\x02\n\x17MeanCombinationStrategy\x12}\n\x19mean_evaluation_technique\x18\x01 \x01(\x0e\x32Z.yandex.cloud.ai.assistants.v1.searchindex.MeanCombinationStrategy.MeanEvaluationTechnique\x12\x0f\n\x07weights\x18\x02 \x03(\x01\"q\n\x17MeanEvaluationTechnique\x12)\n%MEAN_EVALUATION_TECHNIQUE_UNSPECIFIED\x10\x00\x12\x0e\n\nARITHMETIC\x10\x01\x12\r\n\tGEOMETRIC\x10\x02\x12\x0c\n\x08HARMONIC\x10\x03\"Q\n\'ReciprocalRankFusionCombinationStrategy\x12&\n\x01k\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"\xf0\x01\n\x13\x43ombinationStrategy\x12^\n\x10mean_combination\x18\x01 \x01(\x0b\x32\x42.yandex.cloud.ai.assistants.v1.searchindex.MeanCombinationStrategyH\x00\x12m\n\x0frrf_combination\x18\x02 \x01(\x0b\x32R.yandex.cloud.ai.assistants.v1.searchindex.ReciprocalRankFusionCombinationStrategyH\x00\x42\n\n\x08Strategy\"n\n\x0eNgramTokenizer\x12-\n\x08min_gram\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12-\n\x08max_gram\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value*T\n\x15NormalizationStrategy\x12&\n\"NORMALIZATION_STRATEGY_UNSPECIFIED\x10\x00\x12\x0b\n\x07MIN_MAX\x10\x01\x12\x06\n\x02L2\x10\x02\x42\x8a\x01\n-yandex.cloud.api.ai.assistants.v1.searchindexZYgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/searchindex;searchindexb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6yandex/cloud/ai/assistants/v1/searchindex/common.proto\x12)yandex.cloud.ai.assistants.v1.searchindex\x1a\x1egoogle/protobuf/wrappers.proto\"U\n\x16StaticChunkingStrategy\x12\x1d\n\x15max_chunk_size_tokens\x18\x01 \x01(\x03\x12\x1c\n\x14\x63hunk_overlap_tokens\x18\x02 \x01(\x03\"|\n\x10\x43hunkingStrategy\x12\\\n\x0fstatic_strategy\x18\x01 \x01(\x0b\x32\x41.yandex.cloud.ai.assistants.v1.searchindex.StaticChunkingStrategyH\x00\x42\n\n\x08Strategy\"\x9c\x02\n\x17MeanCombinationStrategy\x12}\n\x19mean_evaluation_technique\x18\x01 \x01(\x0e\x32Z.yandex.cloud.ai.assistants.v1.searchindex.MeanCombinationStrategy.MeanEvaluationTechnique\x12\x0f\n\x07weights\x18\x02 \x03(\x01\"q\n\x17MeanEvaluationTechnique\x12)\n%MEAN_EVALUATION_TECHNIQUE_UNSPECIFIED\x10\x00\x12\x0e\n\nARITHMETIC\x10\x01\x12\r\n\tGEOMETRIC\x10\x02\x12\x0c\n\x08HARMONIC\x10\x03\"Q\n\'ReciprocalRankFusionCombinationStrategy\x12&\n\x01k\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"\xf0\x01\n\x13\x43ombinationStrategy\x12^\n\x10mean_combination\x18\x01 \x01(\x0b\x32\x42.yandex.cloud.ai.assistants.v1.searchindex.MeanCombinationStrategyH\x00\x12m\n\x0frrf_combination\x18\x02 \x01(\x0b\x32R.yandex.cloud.ai.assistants.v1.searchindex.ReciprocalRankFusionCombinationStrategyH\x00\x42\n\n\x08Strategy\"n\n\x0eNgramTokenizer\x12-\n\x08min_gram\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12-\n\x08max_gram\x18\x02 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\"\x13\n\x11StandardTokenizer\"\x12\n\x10StandardAnalyzer\"\x16\n\x14YandexLemmerAnalyzer*T\n\x15NormalizationStrategy\x12&\n\"NORMALIZATION_STRATEGY_UNSPECIFIED\x10\x00\x12\x0b\n\x07MIN_MAX\x10\x01\x12\x06\n\x02L2\x10\x02\x42\x8a\x01\n-yandex.cloud.api.ai.assistants.v1.searchindexZYgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/searchindex;searchindexb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -33,8 +33,8 @@ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'yandex.cloud.ai.assistants.
 if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
   _globals['DESCRIPTOR']._serialized_options = b'\n-yandex.cloud.api.ai.assistants.v1.searchindexZYgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/searchindex;searchindex'
-  _globals['_NORMALIZATIONSTRATEGY']._serialized_start=1071
-  _globals['_NORMALIZATIONSTRATEGY']._serialized_end=1155
+  _globals['_NORMALIZATIONSTRATEGY']._serialized_start=1136
+  _globals['_NORMALIZATIONSTRATEGY']._serialized_end=1220
   _globals['_STATICCHUNKINGSTRATEGY']._serialized_start=133
   _globals['_STATICCHUNKINGSTRATEGY']._serialized_end=218
   _globals['_CHUNKINGSTRATEGY']._serialized_start=220
@@ -49,4 +49,10 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_COMBINATIONSTRATEGY']._serialized_end=957
   _globals['_NGRAMTOKENIZER']._serialized_start=959
   _globals['_NGRAMTOKENIZER']._serialized_end=1069
+  _globals['_STANDARDTOKENIZER']._serialized_start=1071
+  _globals['_STANDARDTOKENIZER']._serialized_end=1090
+  _globals['_STANDARDANALYZER']._serialized_start=1092
+  _globals['_STANDARDANALYZER']._serialized_end=1110
+  _globals['_YANDEXLEMMERANALYZER']._serialized_start=1112
+  _globals['_YANDEXLEMMERANALYZER']._serialized_end=1134
 # @@protoc_insertion_point(module_scope)

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

@@ -222,3 +222,49 @@ class NgramTokenizer(google.protobuf.message.Message):
     def ClearField(self, field_name: typing.Literal["max_gram", b"max_gram", "min_gram", b"min_gram"]) -> None: ...
 
 global___NgramTokenizer = NgramTokenizer
+
+@typing.final
+class StandardTokenizer(google.protobuf.message.Message):
+    """A standard tokenizer that splits text on word boundaries and removes punctuation.
+    It follows the Unicode Text Segmentation rules as specified in Unicode Standard Annex #29.
+
+    Example:
+    Input text: `Hello, world! How are you?`
+    Output tokens: `[Hello, world, How, are, you]`
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___StandardTokenizer = StandardTokenizer
+
+@typing.final
+class StandardAnalyzer(google.protobuf.message.Message):
+    """A standard analyzer that uses StandardTokenizer."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___StandardAnalyzer = StandardAnalyzer
+
+@typing.final
+class YandexLemmerAnalyzer(google.protobuf.message.Message):
+    """A specialized analyzer that uses Yandex's lemmatization technology to reduce words to their base forms.
+    Particularly effective for Russian and other Slavic languages, handling their complex morphology.
+    For more information, see:
+    https://yandex.cloud/en/docs/tutorials/dataplatform/opensearch-yandex-lemmer
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___YandexLemmerAnalyzer = YandexLemmerAnalyzer

yandex/cloud/ai/assistants/v1/searchindex/search_index_pb2.py

@@ -27,7 +27,7 @@ from yandex.cloud.ai.assistants.v1.searchindex import common_pb2 as yandex_dot_c
 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/searchindex/search_index.proto\x12)yandex.cloud.ai.assistants.v1.searchindex\x1a#yandex/cloud/ai/common/common.proto\x1a\x36yandex/cloud/ai/assistants/v1/searchindex/common.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xef\x05\n\x0bSearchIndex\x12\n\n\x02id\x18\x01 \x01(\t\x12\x11\n\tfolder_id\x18\x02 \x01(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x12\n\ncreated_by\x18\x05 \x01(\t\x12.\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\nupdated_by\x18\x07 \x01(\t\x12.\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x43\n\x11\x65xpiration_config\x18\t \x01(\x0b\x32(.yandex.cloud.ai.common.ExpirationConfig\x12.\n\nexpires_at\x18\n \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12R\n\x06labels\x18\x0b \x03(\x0b\x32\x42.yandex.cloud.ai.assistants.v1.searchindex.SearchIndex.LabelsEntry\x12W\n\x11text_search_index\x18\x0c \x01(\x0b\x32:.yandex.cloud.ai.assistants.v1.searchindex.TextSearchIndexH\x00\x12[\n\x13vector_search_index\x18\r \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.VectorSearchIndexH\x00\x12[\n\x13hybrid_search_index\x18\x0e \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.HybridSearchIndexH\x00\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0b\n\tIndexType\"\xd0\x01\n\x0fTextSearchIndex\x12V\n\x11\x63hunking_strategy\x18\x01 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.ChunkingStrategy\x12T\n\x0fngram_tokenizer\x18\x02 \x01(\x0b\x32\x39.yandex.cloud.ai.assistants.v1.searchindex.NgramTokenizerH\x00\x42\x0f\n\rTextTokenizer\"\xa1\x01\n\x11VectorSearchIndex\x12\x18\n\x10\x64oc_embedder_uri\x18\x01 \x01(\t\x12\x1a\n\x12query_embedder_uri\x18\x02 \x01(\t\x12V\n\x11\x63hunking_strategy\x18\x03 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.ChunkingStrategy\"\xdd\x03\n\x11HybridSearchIndex\x12U\n\x11text_search_index\x18\x01 \x01(\x0b\x32:.yandex.cloud.ai.assistants.v1.searchindex.TextSearchIndex\x12Y\n\x13vector_search_index\x18\x02 \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.VectorSearchIndex\x12V\n\x11\x63hunking_strategy\x18\x03 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.ChunkingStrategy\x12`\n\x16normalization_strategy\x18\x04 \x01(\x0e\x32@.yandex.cloud.ai.assistants.v1.searchindex.NormalizationStrategy\x12\\\n\x14\x63ombination_strategy\x18\x05 \x01(\x0b\x32>.yandex.cloud.ai.assistants.v1.searchindex.CombinationStrategyB\x8a\x01\n-yandex.cloud.api.ai.assistants.v1.searchindexZYgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/searchindex;searchindexb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n<yandex/cloud/ai/assistants/v1/searchindex/search_index.proto\x12)yandex.cloud.ai.assistants.v1.searchindex\x1a#yandex/cloud/ai/common/common.proto\x1a\x36yandex/cloud/ai/assistants/v1/searchindex/common.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xef\x05\n\x0bSearchIndex\x12\n\n\x02id\x18\x01 \x01(\t\x12\x11\n\tfolder_id\x18\x02 \x01(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x12\n\ncreated_by\x18\x05 \x01(\t\x12.\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\nupdated_by\x18\x07 \x01(\t\x12.\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x43\n\x11\x65xpiration_config\x18\t \x01(\x0b\x32(.yandex.cloud.ai.common.ExpirationConfig\x12.\n\nexpires_at\x18\n \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12R\n\x06labels\x18\x0b \x03(\x0b\x32\x42.yandex.cloud.ai.assistants.v1.searchindex.SearchIndex.LabelsEntry\x12W\n\x11text_search_index\x18\x0c \x01(\x0b\x32:.yandex.cloud.ai.assistants.v1.searchindex.TextSearchIndexH\x00\x12[\n\x13vector_search_index\x18\r \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.VectorSearchIndexH\x00\x12[\n\x13hybrid_search_index\x18\x0e \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.HybridSearchIndexH\x00\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0b\n\tIndexType\"\xfd\x03\n\x0fTextSearchIndex\x12V\n\x11\x63hunking_strategy\x18\x01 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.ChunkingStrategy\x12X\n\x0fngram_tokenizer\x18\x02 \x01(\x0b\x32\x39.yandex.cloud.ai.assistants.v1.searchindex.NgramTokenizerB\x02\x18\x01H\x00\x12Z\n\x12standard_tokenizer\x18\x03 \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.StandardTokenizerH\x00\x12X\n\x11standard_analyzer\x18\x04 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.StandardAnalyzerH\x01\x12\x61\n\x16yandex_lemmer_analyzer\x18\x05 \x01(\x0b\x32?.yandex.cloud.ai.assistants.v1.searchindex.YandexLemmerAnalyzerH\x01\x42\x0f\n\rTextTokenizerB\x0e\n\x0cTextAnalyzer\"\xa1\x01\n\x11VectorSearchIndex\x12\x18\n\x10\x64oc_embedder_uri\x18\x01 \x01(\t\x12\x1a\n\x12query_embedder_uri\x18\x02 \x01(\t\x12V\n\x11\x63hunking_strategy\x18\x03 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.ChunkingStrategy\"\xdd\x03\n\x11HybridSearchIndex\x12U\n\x11text_search_index\x18\x01 \x01(\x0b\x32:.yandex.cloud.ai.assistants.v1.searchindex.TextSearchIndex\x12Y\n\x13vector_search_index\x18\x02 \x01(\x0b\x32<.yandex.cloud.ai.assistants.v1.searchindex.VectorSearchIndex\x12V\n\x11\x63hunking_strategy\x18\x03 \x01(\x0b\x32;.yandex.cloud.ai.assistants.v1.searchindex.ChunkingStrategy\x12`\n\x16normalization_strategy\x18\x04 \x01(\x0e\x32@.yandex.cloud.ai.assistants.v1.searchindex.NormalizationStrategy\x12\\\n\x14\x63ombination_strategy\x18\x05 \x01(\x0b\x32>.yandex.cloud.ai.assistants.v1.searchindex.CombinationStrategyB\x8a\x01\n-yandex.cloud.api.ai.assistants.v1.searchindexZYgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/searchindex;searchindexb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -37,14 +37,16 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._serialized_options = b'\n-yandex.cloud.api.ai.assistants.v1.searchindexZYgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/searchindex;searchindex'
   _globals['_SEARCHINDEX_LABELSENTRY']._loaded_options = None
   _globals['_SEARCHINDEX_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_TEXTSEARCHINDEX'].fields_by_name['ngram_tokenizer']._loaded_options = None
+  _globals['_TEXTSEARCHINDEX'].fields_by_name['ngram_tokenizer']._serialized_options = b'\030\001'
   _globals['_SEARCHINDEX']._serialized_start=234
   _globals['_SEARCHINDEX']._serialized_end=985
   _globals['_SEARCHINDEX_LABELSENTRY']._serialized_start=927
   _globals['_SEARCHINDEX_LABELSENTRY']._serialized_end=972
   _globals['_TEXTSEARCHINDEX']._serialized_start=988
-  _globals['_TEXTSEARCHINDEX']._serialized_end=1196
-  _globals['_VECTORSEARCHINDEX']._serialized_start=1199
-  _globals['_VECTORSEARCHINDEX']._serialized_end=1360
-  _globals['_HYBRIDSEARCHINDEX']._serialized_start=1363
-  _globals['_HYBRIDSEARCHINDEX']._serialized_end=1840
+  _globals['_TEXTSEARCHINDEX']._serialized_end=1497
+  _globals['_VECTORSEARCHINDEX']._serialized_start=1500
+  _globals['_VECTORSEARCHINDEX']._serialized_end=1661
+  _globals['_HYBRIDSEARCHINDEX']._serialized_start=1664
+  _globals['_HYBRIDSEARCHINDEX']._serialized_end=2141
 # @@protoc_insertion_point(module_scope)

yandex/cloud/ai/assistants/v1/searchindex/search_index_pb2.pyi

@@ -133,6 +133,9 @@ class TextSearchIndex(google.protobuf.message.Message):
 
     CHUNKING_STRATEGY_FIELD_NUMBER: builtins.int
     NGRAM_TOKENIZER_FIELD_NUMBER: builtins.int
+    STANDARD_TOKENIZER_FIELD_NUMBER: builtins.int
+    STANDARD_ANALYZER_FIELD_NUMBER: builtins.int
+    YANDEX_LEMMER_ANALYZER_FIELD_NUMBER: builtins.int
     @property
     def chunking_strategy(self) -> yandex.cloud.ai.assistants.v1.searchindex.common_pb2.ChunkingStrategy:
         """Chunking strategy used to split text into smaller chunks before indexing.
@@ -143,15 +146,35 @@ class TextSearchIndex(google.protobuf.message.Message):
     def ngram_tokenizer(self) -> yandex.cloud.ai.assistants.v1.searchindex.common_pb2.NgramTokenizer:
         """Tokenizer that generates n-grams."""
 
+    @property
+    def standard_tokenizer(self) -> yandex.cloud.ai.assistants.v1.searchindex.common_pb2.StandardTokenizer:
+        """Tokenizer that generates words."""
+
+    @property
+    def standard_analyzer(self) -> yandex.cloud.ai.assistants.v1.searchindex.common_pb2.StandardAnalyzer:
+        """Standard analyzer that performs common text processing operations to normalize text."""
+
+    @property
+    def yandex_lemmer_analyzer(self) -> yandex.cloud.ai.assistants.v1.searchindex.common_pb2.YandexLemmerAnalyzer:
+        """Specialized analyzer that uses Yandex's lemmatization technology,
+        particularly effective for Russian and other Slavic languages.
+        """
+
     def __init__(
         self,
         *,
         chunking_strategy: yandex.cloud.ai.assistants.v1.searchindex.common_pb2.ChunkingStrategy | None = ...,
         ngram_tokenizer: yandex.cloud.ai.assistants.v1.searchindex.common_pb2.NgramTokenizer | None = ...,
+        standard_tokenizer: yandex.cloud.ai.assistants.v1.searchindex.common_pb2.StandardTokenizer | None = ...,
+        standard_analyzer: yandex.cloud.ai.assistants.v1.searchindex.common_pb2.StandardAnalyzer | None = ...,
+        yandex_lemmer_analyzer: yandex.cloud.ai.assistants.v1.searchindex.common_pb2.YandexLemmerAnalyzer | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing.Literal["TextTokenizer", b"TextTokenizer", "chunking_strategy", b"chunking_strategy", "ngram_tokenizer", b"ngram_tokenizer"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["TextTokenizer", b"TextTokenizer", "chunking_strategy", b"chunking_strategy", "ngram_tokenizer", b"ngram_tokenizer"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing.Literal["TextTokenizer", b"TextTokenizer"]) -> typing.Literal["ngram_tokenizer"] | None: ...
+    def HasField(self, field_name: typing.Literal["TextAnalyzer", b"TextAnalyzer", "TextTokenizer", b"TextTokenizer", "chunking_strategy", b"chunking_strategy", "ngram_tokenizer", b"ngram_tokenizer", "standard_analyzer", b"standard_analyzer", "standard_tokenizer", b"standard_tokenizer", "yandex_lemmer_analyzer", b"yandex_lemmer_analyzer"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["TextAnalyzer", b"TextAnalyzer", "TextTokenizer", b"TextTokenizer", "chunking_strategy", b"chunking_strategy", "ngram_tokenizer", b"ngram_tokenizer", "standard_analyzer", b"standard_analyzer", "standard_tokenizer", b"standard_tokenizer", "yandex_lemmer_analyzer", b"yandex_lemmer_analyzer"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["TextAnalyzer", b"TextAnalyzer"]) -> typing.Literal["standard_analyzer", "yandex_lemmer_analyzer"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["TextTokenizer", b"TextTokenizer"]) -> typing.Literal["ngram_tokenizer", "standard_tokenizer"] | None: ...
 
 global___TextSearchIndex = TextSearchIndex
 

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

@@ -0,0 +1,67 @@
+# -*- 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_service.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_service.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from yandex.cloud import validation_pb2 as yandex_dot_cloud_dot_validation__pb2
+from yandex.cloud.ai.batch_inference.v1 import batch_inference_task_pb2 as yandex_dot_cloud_dot_ai_dot_batch__inference_dot_v1_dot_batch__inference__task__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n@yandex/cloud/ai/batch_inference/v1/batch_inference_service.proto\x12\"yandex.cloud.ai.batch_inference.v1\x1a\x1dyandex/cloud/validation.proto\x1a=yandex/cloud/ai/batch_inference/v1/batch_inference_task.proto\"\xaf\x01\n\x16\x42\x61tchInferenceMetadata\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12R\n\x0btask_status\x18\x02 \x01(\x0e\x32=.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask.Status\x12\x19\n\x11\x63ompleted_batches\x18\x03 \x01(\x03\x12\x15\n\rtotal_batches\x18\x04 \x01(\x03\"^\n\x16\x42\x61tchInferenceResponse\x12\x44\n\x04task\x18\x01 \x01(\x0b\x32\x36.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask\"6\n\x1d\x44\x65scribeBatchInferenceRequest\x12\x15\n\x07task_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"f\n\x1e\x44\x65scribeBatchInferenceResponse\x12\x44\n\x04task\x18\x01 \x01(\x0b\x32\x36.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask\"\xab\x01\n\x1aListBatchInferencesRequest\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\x12M\n\x06status\x18\x04 \x01(\x0e\x32=.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask.Status\"}\n\x1bListBatchInferencesResponse\x12\x45\n\x05tasks\x18\x01 \x03(\x0b\x32\x36.yandex.cloud.ai.batch_inference.v1.BatchInferenceTask\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"4\n\x1b\x43\x61ncelBatchInferenceRequest\x12\x15\n\x07task_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"/\n\x1c\x43\x61ncelBatchInferenceResponse\x12\x0f\n\x07task_id\x18\x01 \x01(\t\"4\n\x1b\x44\x65leteBatchInferenceRequest\x12\x15\n\x07task_id\x18\x01 \x01(\tB\x04\xe8\xc7\x31\x01\"/\n\x1c\x44\x65leteBatchInferenceResponse\x12\x0f\n\x07task_id\x18\x01 \x01(\t2\xd1\x04\n\x15\x42\x61tchInferenceService\x12\x91\x01\n\x08\x44\x65scribe\x12\x41.yandex.cloud.ai.batch_inference.v1.DescribeBatchInferenceRequest\x1a\x42.yandex.cloud.ai.batch_inference.v1.DescribeBatchInferenceResponse\x12\x87\x01\n\x04List\x12>.yandex.cloud.ai.batch_inference.v1.ListBatchInferencesRequest\x1a?.yandex.cloud.ai.batch_inference.v1.ListBatchInferencesResponse\x12\x8b\x01\n\x06\x43\x61ncel\x12?.yandex.cloud.ai.batch_inference.v1.CancelBatchInferenceRequest\x1a@.yandex.cloud.ai.batch_inference.v1.CancelBatchInferenceResponse\x12\x8b\x01\n\x06\x44\x65lete\x12?.yandex.cloud.ai.batch_inference.v1.DeleteBatchInferenceRequest\x1a@.yandex.cloud.ai.batch_inference.v1.DeleteBatchInferenceResponseBu\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_service_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['_DESCRIBEBATCHINFERENCEREQUEST'].fields_by_name['task_id']._loaded_options = None
+  _globals['_DESCRIBEBATCHINFERENCEREQUEST'].fields_by_name['task_id']._serialized_options = b'\350\3071\001'
+  _globals['_LISTBATCHINFERENCESREQUEST'].fields_by_name['folder_id']._loaded_options = None
+  _globals['_LISTBATCHINFERENCESREQUEST'].fields_by_name['folder_id']._serialized_options = b'\350\3071\001'
+  _globals['_CANCELBATCHINFERENCEREQUEST'].fields_by_name['task_id']._loaded_options = None
+  _globals['_CANCELBATCHINFERENCEREQUEST'].fields_by_name['task_id']._serialized_options = b'\350\3071\001'
+  _globals['_DELETEBATCHINFERENCEREQUEST'].fields_by_name['task_id']._loaded_options = None
+  _globals['_DELETEBATCHINFERENCEREQUEST'].fields_by_name['task_id']._serialized_options = b'\350\3071\001'
+  _globals['_BATCHINFERENCEMETADATA']._serialized_start=199
+  _globals['_BATCHINFERENCEMETADATA']._serialized_end=374
+  _globals['_BATCHINFERENCERESPONSE']._serialized_start=376
+  _globals['_BATCHINFERENCERESPONSE']._serialized_end=470
+  _globals['_DESCRIBEBATCHINFERENCEREQUEST']._serialized_start=472
+  _globals['_DESCRIBEBATCHINFERENCEREQUEST']._serialized_end=526
+  _globals['_DESCRIBEBATCHINFERENCERESPONSE']._serialized_start=528
+  _globals['_DESCRIBEBATCHINFERENCERESPONSE']._serialized_end=630
+  _globals['_LISTBATCHINFERENCESREQUEST']._serialized_start=633
+  _globals['_LISTBATCHINFERENCESREQUEST']._serialized_end=804
+  _globals['_LISTBATCHINFERENCESRESPONSE']._serialized_start=806
+  _globals['_LISTBATCHINFERENCESRESPONSE']._serialized_end=931
+  _globals['_CANCELBATCHINFERENCEREQUEST']._serialized_start=933
+  _globals['_CANCELBATCHINFERENCEREQUEST']._serialized_end=985
+  _globals['_CANCELBATCHINFERENCERESPONSE']._serialized_start=987
+  _globals['_CANCELBATCHINFERENCERESPONSE']._serialized_end=1034
+  _globals['_DELETEBATCHINFERENCEREQUEST']._serialized_start=1036
+  _globals['_DELETEBATCHINFERENCEREQUEST']._serialized_end=1088
+  _globals['_DELETEBATCHINFERENCERESPONSE']._serialized_start=1090
+  _globals['_DELETEBATCHINFERENCERESPONSE']._serialized_end=1137
+  _globals['_BATCHINFERENCESERVICE']._serialized_start=1140
+  _globals['_BATCHINFERENCESERVICE']._serialized_end=1733
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,195 @@
+"""
+@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 typing
+import yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class BatchInferenceMetadata(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    TASK_STATUS_FIELD_NUMBER: builtins.int
+    COMPLETED_BATCHES_FIELD_NUMBER: builtins.int
+    TOTAL_BATCHES_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    task_status: yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask.Status.ValueType
+    completed_batches: builtins.int
+    total_batches: builtins.int
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        task_status: yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask.Status.ValueType = ...,
+        completed_batches: builtins.int = ...,
+        total_batches: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["completed_batches", b"completed_batches", "task_id", b"task_id", "task_status", b"task_status", "total_batches", b"total_batches"]) -> None: ...
+
+global___BatchInferenceMetadata = BatchInferenceMetadata
+
+@typing.final
+class BatchInferenceResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_FIELD_NUMBER: builtins.int
+    @property
+    def task(self) -> yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask: ...
+    def __init__(
+        self,
+        *,
+        task: yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["task", b"task"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["task", b"task"]) -> None: ...
+
+global___BatchInferenceResponse = BatchInferenceResponse
+
+@typing.final
+class DescribeBatchInferenceRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    """Required task id"""
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...
+
+global___DescribeBatchInferenceRequest = DescribeBatchInferenceRequest
+
+@typing.final
+class DescribeBatchInferenceResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_FIELD_NUMBER: builtins.int
+    @property
+    def task(self) -> yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask: ...
+    def __init__(
+        self,
+        *,
+        task: yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["task", b"task"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["task", b"task"]) -> None: ...
+
+global___DescribeBatchInferenceResponse = DescribeBatchInferenceResponse
+
+@typing.final
+class ListBatchInferencesRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FOLDER_ID_FIELD_NUMBER: builtins.int
+    PAGE_SIZE_FIELD_NUMBER: builtins.int
+    PAGE_TOKEN_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    folder_id: builtins.str
+    """Folder ID for which the list of tasks will be provided."""
+    page_size: builtins.int
+    page_token: builtins.str
+    status: yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask.Status.ValueType
+    """Batch inference status for filtering"""
+    def __init__(
+        self,
+        *,
+        folder_id: builtins.str = ...,
+        page_size: builtins.int = ...,
+        page_token: builtins.str = ...,
+        status: yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask.Status.ValueType = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["folder_id", b"folder_id", "page_size", b"page_size", "page_token", b"page_token", "status", b"status"]) -> None: ...
+
+global___ListBatchInferencesRequest = ListBatchInferencesRequest
+
+@typing.final
+class ListBatchInferencesResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASKS_FIELD_NUMBER: builtins.int
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: builtins.int
+    next_page_token: builtins.str
+    @property
+    def tasks(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask]: ...
+    def __init__(
+        self,
+        *,
+        tasks: collections.abc.Iterable[yandex.cloud.ai.batch_inference.v1.batch_inference_task_pb2.BatchInferenceTask] | None = ...,
+        next_page_token: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["next_page_token", b"next_page_token", "tasks", b"tasks"]) -> None: ...
+
+global___ListBatchInferencesResponse = ListBatchInferencesResponse
+
+@typing.final
+class CancelBatchInferenceRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    """Task ID that should be canceled."""
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...
+
+global___CancelBatchInferenceRequest = CancelBatchInferenceRequest
+
+@typing.final
+class CancelBatchInferenceResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...
+
+global___CancelBatchInferenceResponse = CancelBatchInferenceResponse
+
+@typing.final
+class DeleteBatchInferenceRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    """Task ID that should be deleted."""
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...
+
+global___DeleteBatchInferenceRequest = DeleteBatchInferenceRequest
+
+@typing.final
+class DeleteBatchInferenceResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...
+
+global___DeleteBatchInferenceResponse = DeleteBatchInferenceResponse