diff --git a/nucliadb/nucliadb/common/cluster/manager.py b/nucliadb/nucliadb/common/cluster/manager.py index 195ab8cd3c..8fc4ca5ba6 100644 --- a/nucliadb/nucliadb/common/cluster/manager.py +++ b/nucliadb/nucliadb/common/cluster/manager.py @@ -24,7 +24,6 @@ from typing import Any, Awaitable, Callable, Optional import backoff -from nucliadb_protos.knowledgebox_pb2 import SemanticModelMetadata # type: ignore from nucliadb_protos.nodewriter_pb2 import IndexMessage, IndexMessageSource, TypeMessage from nucliadb.common import datamanagers @@ -43,7 +42,6 @@ nodereader_pb2, noderesources_pb2, nodewriter_pb2, - utils_pb2, writer_pb2, ) from nucliadb_telemetry import errors @@ -183,8 +181,6 @@ async def create_shard_by_kbid( self, txn: Transaction, kbid: str, - semantic_model: SemanticModelMetadata, - release_channel: utils_pb2.ReleaseChannel.ValueType, ) -> writer_pb2.ShardObject: try: check_enough_nodes() @@ -197,18 +193,12 @@ async def create_shard_by_kbid( kb_shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) if kb_shards is None: - # First logic shard on the index - kb_shards = writer_pb2.Shards() - kb_shards.kbid = kbid - # B/c with Shards.actual - kb_shards.actual = -1 - kb_shards.similarity = semantic_model.similarity_function - kb_shards.model.CopyFrom(semantic_model) - else: - # New logic shard on an existing index - pass + msg = ( + "Attempting to create a shard for a KB when it has no stored shards in maindb", + ) + logger.error(msg, extra={"kbid": kbid}) + raise ShardsNotFound(msg) - kb_shards.release_channel = release_channel existing_kb_nodes = [ replica.node for shard in kb_shards.shards for replica in shard.replicas ] @@ -259,8 +249,8 @@ async def create_shard_by_kbid( # Append the created shard and make `actual` point to it. kb_shards.shards.append(shard) - # B/c with Shards.actual - kb_shards.actual += 1 + # B/c with Shards.actual - we only use last created shard + kb_shards.actual = len(kb_shards.shards) - 1 await datamanagers.cluster.update_kb_shards(txn, kbid=kbid, shards=kb_shards) @@ -371,7 +361,6 @@ async def maybe_create_new_shard( self, kbid: str, num_paragraphs: int, - release_channel: utils_pb2.ReleaseChannel.ValueType = utils_pb2.ReleaseChannel.STABLE, ): if not self.should_create_new_shard(num_paragraphs): return @@ -379,13 +368,7 @@ async def maybe_create_new_shard( logger.warning({"message": "Adding shard", "kbid": kbid}) async with datamanagers.with_transaction() as txn: - model = await datamanagers.kb.get_model_metadata(txn, kbid=kbid) - await self.create_shard_by_kbid( - txn, - kbid, - semantic_model=model, - release_channel=release_channel, - ) + await self.create_shard_by_kbid(txn, kbid) await txn.commit() @@ -417,7 +400,6 @@ async def _resource_change_event( await self.maybe_create_new_shard( kbid, shard_info.paragraphs, - shard_info.metadata.release_channel, ) await index_node.writer.GC(noderesources_pb2.ShardId(id=shard_id)) # type: ignore diff --git a/nucliadb/nucliadb/common/cluster/rebalance.py b/nucliadb/nucliadb/common/cluster/rebalance.py index d444cd1c85..94008de4df 100644 --- a/nucliadb/nucliadb/common/cluster/rebalance.py +++ b/nucliadb/nucliadb/common/cluster/rebalance.py @@ -75,14 +75,8 @@ async def maybe_add_shard(kbid: str) -> None: ): # create new shard async with datamanagers.with_transaction() as txn: - model = await datamanagers.kb.get_model_metadata(txn, kbid=kbid) sm = get_shard_manager() - await sm.create_shard_by_kbid( - txn, - kbid, - semantic_model=model, - release_channel=kb_shards.release_channel, - ) + await sm.create_shard_by_kbid(txn, kbid) await txn.commit() diff --git a/nucliadb/nucliadb/ingest/consumer/shard_creator.py b/nucliadb/nucliadb/ingest/consumer/shard_creator.py index 90a328fb2e..e30ea334d8 100644 --- a/nucliadb/nucliadb/ingest/consumer/shard_creator.py +++ b/nucliadb/nucliadb/ingest/consumer/shard_creator.py @@ -22,7 +22,7 @@ import uuid from functools import partial -from nucliadb.common import datamanagers, locking +from nucliadb.common import locking from nucliadb.common.cluster.manager import choose_node from nucliadb.common.cluster.utils import get_shard_manager from nucliadb.common.maindb.driver import Driver @@ -91,12 +91,12 @@ async def handle_message(self, raw_data) -> None: async def process_kb(self, kbid: str) -> None: logger.info({"message": "Processing notification for kbid", "kbid": kbid}) async with self.driver.transaction(read_only=True) as txn: - kb_shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) current_shard = await self.shard_manager.get_current_active_shard(txn, kbid) - if kb_shards is None or current_shard is None: + if current_shard is None: logger.error( - "Processing a notification for a nonexistent", extra={"kbid": kbid} + "Processing a notification for KB with no current shard", + extra={"kbid": kbid}, ) return @@ -109,8 +109,4 @@ async def process_kb(self, kbid: str) -> None: shard: nodereader_pb2.Shard = await node.reader.GetShard( nodereader_pb2.GetShardRequest(shard_id=noderesources_pb2.ShardId(id=shard_id)) # type: ignore ) - await self.shard_manager.maybe_create_new_shard( - kbid, - shard.paragraphs, - kb_shards.release_channel, - ) + await self.shard_manager.maybe_create_new_shard(kbid, shard.paragraphs) diff --git a/nucliadb/nucliadb/ingest/orm/knowledgebox.py b/nucliadb/nucliadb/ingest/orm/knowledgebox.py index 01eef678fb..1f382b74f3 100644 --- a/nucliadb/nucliadb/ingest/orm/knowledgebox.py +++ b/nucliadb/nucliadb/ingest/orm/knowledgebox.py @@ -167,14 +167,22 @@ async def create( failed = True if failed is False: + kb_shards = writer_pb2.Shards() + kb_shards.kbid = uuid + # B/c with Shards.actual + kb_shards.actual = -1 + # B/c with `Shards.similarity`, replaced by `model` + kb_shards.similarity = semantic_model.similarity_function + kb_shards.model.CopyFrom(semantic_model) + kb_shards.release_channel = release_channel + + await datamanagers.cluster.update_kb_shards( + txn, kbid=uuid, shards=kb_shards + ) + shard_manager = get_shard_manager() try: - await shard_manager.create_shard_by_kbid( - txn, - uuid, - semantic_model=semantic_model, - release_channel=release_channel, - ) + await shard_manager.create_shard_by_kbid(txn, uuid) except Exception as e: await storage.delete_kb(uuid) raise e diff --git a/nucliadb/nucliadb/ingest/orm/processor/__init__.py b/nucliadb/nucliadb/ingest/orm/processor/__init__.py index 38b1cde817..08b6477a54 100644 --- a/nucliadb/nucliadb/ingest/orm/processor/__init__.py +++ b/nucliadb/nucliadb/ingest/orm/processor/__init__.py @@ -398,19 +398,7 @@ async def index_resource( shard = await self.shard_manager.get_current_active_shard(txn, kbid) if shard is None: # no shard available, create a new one - model = await datamanagers.kb.get_model_metadata(txn, kbid=kbid) - config = await kb.get_config() - if config is not None: - release_channel = config.release_channel - else: - release_channel = utils_pb2.ReleaseChannel.STABLE - - shard = await self.shard_manager.create_shard_by_kbid( - txn, - kbid, - semantic_model=model, - release_channel=release_channel, - ) + shard = await self.shard_manager.create_shard_by_kbid(txn, kbid) await datamanagers.resources.set_resource_shard_id( txn, kbid=kbid, rid=uuid, shard=shard.shard ) diff --git a/nucliadb/nucliadb/ingest/service/writer.py b/nucliadb/nucliadb/ingest/service/writer.py index f0511e2ee4..b383d0e220 100644 --- a/nucliadb/nucliadb/ingest/service/writer.py +++ b/nucliadb/nucliadb/ingest/service/writer.py @@ -188,7 +188,6 @@ async def _create_kb_onprem(self, request: KnowledgeBoxNew) -> str: """ kbid = request.forceuuid or str(uuid.uuid4()) release_channel = get_release_channel(request) - request.config.release_channel = release_channel lconfig = await learning_proxy.get_configuration(kbid) lconfig_created = False if lconfig is None: @@ -235,7 +234,6 @@ async def _create_kb_hosted(self, request: KnowledgeBoxNew) -> str: """ kbid = request.forceuuid or str(uuid.uuid4()) release_channel = get_release_channel(request) - request.config.release_channel = release_channel await self.proc.create_kb( request.slug, request.config, @@ -777,11 +775,8 @@ async def ReIndex(self, request: IndexResource, context=None) -> IndexStatus: # ) if shard is None: # no shard currently exists, create one - model = await datamanagers.kb.get_model_metadata( - txn, kbid=request.kbid - ) shard = await self.shards_manager.create_shard_by_kbid( - txn, request.kbid, semantic_model=model + txn, request.kbid ) await datamanagers.resources.set_resource_shard_id( diff --git a/nucliadb/nucliadb/ingest/tests/integration/servicer/test_knowledgebox_servicer.py b/nucliadb/nucliadb/ingest/tests/integration/servicer/test_knowledgebox_servicer.py index 36876916ae..e38947288f 100644 --- a/nucliadb/nucliadb/ingest/tests/integration/servicer/test_knowledgebox_servicer.py +++ b/nucliadb/nucliadb/ingest/tests/integration/servicer/test_knowledgebox_servicer.py @@ -127,5 +127,4 @@ async def test_create_knowledgebox_release_channel( async with driver.transaction() as txn: shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) assert shards is not None - config = await get_kb_config(txn, kbid) - assert shards.release_channel == config.release_channel == release_channel + assert shards.release_channel == release_channel diff --git a/nucliadb/nucliadb/ingest/tests/unit/consumer/test_shard_creator.py b/nucliadb/nucliadb/ingest/tests/unit/consumer/test_shard_creator.py index c655d77b1f..e099ea2eb4 100644 --- a/nucliadb/nucliadb/ingest/tests/unit/consumer/test_shard_creator.py +++ b/nucliadb/nucliadb/ingest/tests/unit/consumer/test_shard_creator.py @@ -63,9 +63,6 @@ def shard_manager(reader): ), patch( "nucliadb.ingest.consumer.shard_creator.choose_node", return_value=(node, "shard_id"), - ), patch( - "nucliadb.ingest.consumer.shard_creator.datamanagers.cluster.get_kb_shards", - AsyncMock(return_value=shards), ), patch( "nucliadb.ingest.consumer.shard_creator.locking.distributed_lock", return_value=AsyncMock(), @@ -103,7 +100,7 @@ async def test_handle_message_create_new_shard( await shard_creator_handler.handle_message(notif.SerializeToString()) await asyncio.sleep(0.06) shard_manager.maybe_create_new_shard.assert_called_with( - "kbid", settings.max_shard_paragraphs + 1, 0 + "kbid", settings.max_shard_paragraphs + 1 ) diff --git a/nucliadb/nucliadb/tests/integration/common/cluster/test_manager.py b/nucliadb/nucliadb/tests/integration/common/cluster/test_manager.py index 7f2039130f..3bee002a9a 100644 --- a/nucliadb/nucliadb/tests/integration/common/cluster/test_manager.py +++ b/nucliadb/nucliadb/tests/integration/common/cluster/test_manager.py @@ -205,9 +205,8 @@ def node_new_shard(): yield mocked -@pytest.mark.parametrize("release_channel", ("EXPERIMENTAL", "STABLE")) async def test_create_shard_by_kbid_attempts_on_all_nodes( - shards, maindb_driver, fake_kbid, node_new_shard, release_channel + shards, maindb_driver, fake_kbid, node_new_shard ): shard_manager = manager.KBShardManager() async with maindb_driver.transaction() as txn: @@ -215,8 +214,6 @@ async def test_create_shard_by_kbid_attempts_on_all_nodes( await shard_manager.create_shard_by_kbid( txn, fake_kbid, - semantic_model=mock.MagicMock(), - release_channel=release_channel, ) assert node_new_shard.await_count == len(manager.get_index_nodes()) diff --git a/nucliadb/nucliadb/tests/unit/common/cluster/test_kb_shard_manager.py b/nucliadb/nucliadb/tests/unit/common/cluster/test_kb_shard_manager.py index 48ae252ca8..02b5424342 100644 --- a/nucliadb/nucliadb/tests/unit/common/cluster/test_kb_shard_manager.py +++ b/nucliadb/nucliadb/tests/unit/common/cluster/test_kb_shard_manager.py @@ -28,7 +28,7 @@ from nucliadb.common.cluster import manager from nucliadb.common.cluster.settings import settings from nucliadb.common.maindb.driver import Transaction -from nucliadb_protos import knowledgebox_pb2, utils_pb2, writer_pb2 +from nucliadb_protos import writer_pb2 def test_should_create_new_shard(): @@ -89,15 +89,20 @@ async def test_shard_creation(fake_index_nodes: list[str], txn: Transaction): """ index_nodes = set(fake_index_nodes) kbid = f"kbid:{test_shard_creation.__name__}" - semantic_model = knowledgebox_pb2.SemanticModelMetadata() - release_channel = utils_pb2.ReleaseChannel.STABLE sm = manager.KBShardManager() + # Fake KB shards instead of creating a KB to generate it shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) - assert shards is None + await datamanagers.cluster.update_kb_shards( + txn, + kbid=kbid, + shards=writer_pb2.Shards( + kbid=kbid, + ), + ) # create first shard - await sm.create_shard_by_kbid(txn, kbid, semantic_model, release_channel) + await sm.create_shard_by_kbid(txn, kbid) shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) assert shards is not None @@ -108,7 +113,7 @@ async def test_shard_creation(fake_index_nodes: list[str], txn: Transaction): assert set((replica.node for replica in shards.shards[0].replicas)) == index_nodes # adding a second shard will mark the first as read only - await sm.create_shard_by_kbid(txn, kbid, semantic_model, release_channel) + await sm.create_shard_by_kbid(txn, kbid) shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) assert shards is not None @@ -120,7 +125,7 @@ async def test_shard_creation(fake_index_nodes: list[str], txn: Transaction): assert set((replica.node for replica in shards.shards[1].replicas)) == index_nodes # adding a third one will be equivalent - await sm.create_shard_by_kbid(txn, kbid, semantic_model, release_channel) + await sm.create_shard_by_kbid(txn, kbid) shards = await datamanagers.cluster.get_kb_shards(txn, kbid=kbid) assert shards is not None diff --git a/nucliadb_protos/knowledgebox.proto b/nucliadb_protos/knowledgebox.proto index cea2fe5145..0857e29c4a 100644 --- a/nucliadb_protos/knowledgebox.proto +++ b/nucliadb_protos/knowledgebox.proto @@ -30,7 +30,8 @@ message KnowledgeBoxConfig { string slug = 5; bool disable_vectors = 6 [deprecated = true]; int64 migration_version = 7; - utils.ReleaseChannel release_channel = 8; + // DEPRECATED: duplicated field also stored in `writer.proto Shards` + utils.ReleaseChannel release_channel = 8 [deprecated = true]; } // NEW diff --git a/nucliadb_protos/python/nucliadb_protos/audit_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/audit_pb2.pyi index 62d164a553..ea46068959 100644 --- a/nucliadb_protos/python/nucliadb_protos/audit_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/audit_pb2.pyi @@ -52,7 +52,7 @@ class AuditField(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FieldActionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AuditField._FieldAction.ValueType], builtins.type): + class _FieldActionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AuditField._FieldAction.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor ADDED: AuditField._FieldAction.ValueType # 0 MODIFIED: AuditField._FieldAction.ValueType # 1 @@ -167,7 +167,7 @@ class AuditRequest(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _AuditTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AuditRequest._AuditType.ValueType], builtins.type): + class _AuditTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AuditRequest._AuditType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor VISITED: AuditRequest._AuditType.ValueType # 0 MODIFIED: AuditRequest._AuditType.ValueType # 1 diff --git a/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.py b/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.py index 16a8e8277e..e085731779 100644 --- a/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.py +++ b/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.py @@ -16,7 +16,7 @@ from nucliadb_protos.utils_pb2 import * -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"nucliadb_protos/knowledgebox.proto\x12\x0cknowledgebox\x1a\x1bnucliadb_protos/utils.proto\",\n\x0eKnowledgeBoxID\x12\x0c\n\x04slug\x18\x01 \x01(\t\x12\x0c\n\x04uuid\x18\x02 \x01(\t\"\xe9\x01\n\x12KnowledgeBoxConfig\x12\r\n\x05title\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x1b\n\x0f\x65nabled_filters\x18\x03 \x03(\tB\x02\x18\x01\x12\x1c\n\x10\x65nabled_insights\x18\x04 \x03(\tB\x02\x18\x01\x12\x0c\n\x04slug\x18\x05 \x01(\t\x12\x1b\n\x0f\x64isable_vectors\x18\x06 \x01(\x08\x42\x02\x18\x01\x12\x19\n\x11migration_version\x18\x07 \x01(\x03\x12.\n\x0frelease_channel\x18\x08 \x01(\x0e\x32\x15.utils.ReleaseChannel\"\xc4\x02\n\x0fKnowledgeBoxNew\x12\x0c\n\x04slug\x18\x01 \x01(\t\x12\x30\n\x06\x63onfig\x18\x02 \x01(\x0b\x32 .knowledgebox.KnowledgeBoxConfig\x12\x11\n\tforceuuid\x18\x03 \x01(\t\x12+\n\nsimilarity\x18\x04 \x01(\x0e\x32\x17.utils.VectorSimilarity\x12\x1d\n\x10vector_dimension\x18\x05 \x01(\x05H\x00\x88\x01\x01\x12\x1e\n\x11\x64\x65\x66\x61ult_min_score\x18\x06 \x01(\x02H\x01\x88\x01\x01\x12.\n\x0frelease_channel\x18\x07 \x01(\x0e\x32\x15.utils.ReleaseChannel\x12\x17\n\x0flearning_config\x18\x08 \x01(\tB\x13\n\x11_vector_dimensionB\x14\n\x12_default_min_score\"a\n\x17NewKnowledgeBoxResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32(.knowledgebox.KnowledgeBoxResponseStatus\x12\x0c\n\x04uuid\x18\x02 \x01(\t\"b\n\x12KnowledgeBoxUpdate\x12\x0c\n\x04slug\x18\x01 \x01(\t\x12\x0c\n\x04uuid\x18\x02 \x01(\t\x12\x30\n\x06\x63onfig\x18\x03 \x01(\x0b\x32 .knowledgebox.KnowledgeBoxConfig\"d\n\x1aUpdateKnowledgeBoxResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32(.knowledgebox.KnowledgeBoxResponseStatus\x12\x0c\n\x04uuid\x18\x02 \x01(\t\"\x18\n\x16GCKnowledgeBoxResponse\"V\n\x1a\x44\x65leteKnowledgeBoxResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32(.knowledgebox.KnowledgeBoxResponseStatus\"B\n\x05Label\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0f\n\x07related\x18\x03 \x01(\t\x12\x0c\n\x04text\x18\x04 \x01(\t\x12\x0b\n\x03uri\x18\x05 \x01(\t\"\xe0\x01\n\x08LabelSet\x12\r\n\x05title\x18\x01 \x01(\t\x12\r\n\x05\x63olor\x18\x02 \x01(\t\x12#\n\x06labels\x18\x03 \x03(\x0b\x32\x13.knowledgebox.Label\x12\x10\n\x08multiple\x18\x04 \x01(\x08\x12\x31\n\x04kind\x18\x05 \x03(\x0e\x32#.knowledgebox.LabelSet.LabelSetKind\"L\n\x0cLabelSetKind\x12\r\n\tRESOURCES\x10\x00\x12\x0e\n\nPARAGRAPHS\x10\x01\x12\r\n\tSENTENCES\x10\x02\x12\x0e\n\nSELECTIONS\x10\x03\"\x87\x01\n\x06Labels\x12\x34\n\x08labelset\x18\x01 \x03(\x0b\x32\".knowledgebox.Labels.LabelsetEntry\x1aG\n\rLabelsetEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.knowledgebox.LabelSet:\x02\x38\x01\"L\n\x06\x45ntity\x12\r\n\x05value\x18\x02 \x01(\t\x12\x12\n\nrepresents\x18\x04 \x03(\t\x12\x0e\n\x06merged\x18\x03 \x01(\x08\x12\x0f\n\x07\x64\x65leted\x18\x05 \x01(\x08\"D\n\x14\x45ntitiesGroupSummary\x12\r\n\x05title\x18\x02 \x01(\t\x12\r\n\x05\x63olor\x18\x03 \x01(\t\x12\x0e\n\x06\x63ustom\x18\x04 \x01(\x08\"\xc1\x01\n\rEntitiesGroup\x12;\n\x08\x65ntities\x18\x01 \x03(\x0b\x32).knowledgebox.EntitiesGroup.EntitiesEntry\x12\r\n\x05title\x18\x02 \x01(\t\x12\r\n\x05\x63olor\x18\x03 \x01(\t\x12\x0e\n\x06\x63ustom\x18\x04 \x01(\x08\x1a\x45\n\rEntitiesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12#\n\x05value\x18\x02 \x01(\x0b\x32\x14.knowledgebox.Entity:\x02\x38\x01\"0\n\x15\x44\x65letedEntitiesGroups\x12\x17\n\x0f\x65ntities_groups\x18\x01 \x03(\t\"\xaf\x01\n\x0e\x45ntitiesGroups\x12I\n\x0f\x65ntities_groups\x18\x01 \x03(\x0b\x32\x30.knowledgebox.EntitiesGroups.EntitiesGroupsEntry\x1aR\n\x13\x45ntitiesGroupsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x1b.knowledgebox.EntitiesGroup:\x02\x38\x01\"\xf3\x03\n\x19\x45ntityGroupDuplicateIndex\x12T\n\x0f\x65ntities_groups\x18\x01 \x03(\x0b\x32;.knowledgebox.EntityGroupDuplicateIndex.EntitiesGroupsEntry\x1a&\n\x10\x45ntityDuplicates\x12\x12\n\nduplicates\x18\x01 \x03(\t\x1a\xe1\x01\n\x15\x45ntityGroupDuplicates\x12]\n\x08\x65ntities\x18\x01 \x03(\x0b\x32K.knowledgebox.EntityGroupDuplicateIndex.EntityGroupDuplicates.EntitiesEntry\x1ai\n\rEntitiesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12G\n\x05value\x18\x02 \x01(\x0b\x32\x38.knowledgebox.EntityGroupDuplicateIndex.EntityDuplicates:\x02\x38\x01\x1at\n\x13\x45ntitiesGroupsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12L\n\x05value\x18\x02 \x01(\x0b\x32=.knowledgebox.EntityGroupDuplicateIndex.EntityGroupDuplicates:\x02\x38\x01\"K\n\tVectorSet\x12\x11\n\tdimension\x18\x01 \x01(\x05\x12+\n\nsimilarity\x18\x02 \x01(\x0e\x32\x17.utils.VectorSimilarity\"\x96\x01\n\nVectorSets\x12<\n\nvectorsets\x18\x01 \x03(\x0b\x32(.knowledgebox.VectorSets.VectorsetsEntry\x1aJ\n\x0fVectorsetsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.knowledgebox.VectorSet:\x02\x38\x01\" \n\x0cTermSynonyms\x12\x10\n\x08synonyms\x18\x01 \x03(\t\"\x86\x01\n\x08Synonyms\x12\x30\n\x05terms\x18\x01 \x03(\x0b\x32!.knowledgebox.Synonyms.TermsEntry\x1aH\n\nTermsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x1a.knowledgebox.TermSynonyms:\x02\x38\x01\"\xb7\x01\n\x15SemanticModelMetadata\x12\x34\n\x13similarity_function\x18\x01 \x01(\x0e\x32\x17.utils.VectorSimilarity\x12\x1d\n\x10vector_dimension\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x1e\n\x11\x64\x65\x66\x61ult_min_score\x18\x03 \x01(\x02H\x01\x88\x01\x01\x42\x13\n\x11_vector_dimensionB\x14\n\x12_default_min_score\"\x8c\x01\n\x0fKBConfiguration\x12\x16\n\x0esemantic_model\x18\x02 \x01(\t\x12\x18\n\x10generative_model\x18\x03 \x01(\t\x12\x11\n\tner_model\x18\x04 \x01(\t\x12\x1b\n\x13\x61nonymization_model\x18\x05 \x01(\t\x12\x17\n\x0fvisual_labeling\x18\x06 \x01(\t*K\n\x1aKnowledgeBoxResponseStatus\x12\x06\n\x02OK\x10\x00\x12\x0c\n\x08\x43ONFLICT\x10\x01\x12\x0c\n\x08NOTFOUND\x10\x02\x12\t\n\x05\x45RROR\x10\x03P\x00\x62\x06proto3') +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"nucliadb_protos/knowledgebox.proto\x12\x0cknowledgebox\x1a\x1bnucliadb_protos/utils.proto\",\n\x0eKnowledgeBoxID\x12\x0c\n\x04slug\x18\x01 \x01(\t\x12\x0c\n\x04uuid\x18\x02 \x01(\t\"\xed\x01\n\x12KnowledgeBoxConfig\x12\r\n\x05title\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x1b\n\x0f\x65nabled_filters\x18\x03 \x03(\tB\x02\x18\x01\x12\x1c\n\x10\x65nabled_insights\x18\x04 \x03(\tB\x02\x18\x01\x12\x0c\n\x04slug\x18\x05 \x01(\t\x12\x1b\n\x0f\x64isable_vectors\x18\x06 \x01(\x08\x42\x02\x18\x01\x12\x19\n\x11migration_version\x18\x07 \x01(\x03\x12\x32\n\x0frelease_channel\x18\x08 \x01(\x0e\x32\x15.utils.ReleaseChannelB\x02\x18\x01\"\xc4\x02\n\x0fKnowledgeBoxNew\x12\x0c\n\x04slug\x18\x01 \x01(\t\x12\x30\n\x06\x63onfig\x18\x02 \x01(\x0b\x32 .knowledgebox.KnowledgeBoxConfig\x12\x11\n\tforceuuid\x18\x03 \x01(\t\x12+\n\nsimilarity\x18\x04 \x01(\x0e\x32\x17.utils.VectorSimilarity\x12\x1d\n\x10vector_dimension\x18\x05 \x01(\x05H\x00\x88\x01\x01\x12\x1e\n\x11\x64\x65\x66\x61ult_min_score\x18\x06 \x01(\x02H\x01\x88\x01\x01\x12.\n\x0frelease_channel\x18\x07 \x01(\x0e\x32\x15.utils.ReleaseChannel\x12\x17\n\x0flearning_config\x18\x08 \x01(\tB\x13\n\x11_vector_dimensionB\x14\n\x12_default_min_score\"a\n\x17NewKnowledgeBoxResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32(.knowledgebox.KnowledgeBoxResponseStatus\x12\x0c\n\x04uuid\x18\x02 \x01(\t\"b\n\x12KnowledgeBoxUpdate\x12\x0c\n\x04slug\x18\x01 \x01(\t\x12\x0c\n\x04uuid\x18\x02 \x01(\t\x12\x30\n\x06\x63onfig\x18\x03 \x01(\x0b\x32 .knowledgebox.KnowledgeBoxConfig\"d\n\x1aUpdateKnowledgeBoxResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32(.knowledgebox.KnowledgeBoxResponseStatus\x12\x0c\n\x04uuid\x18\x02 \x01(\t\"\x18\n\x16GCKnowledgeBoxResponse\"V\n\x1a\x44\x65leteKnowledgeBoxResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32(.knowledgebox.KnowledgeBoxResponseStatus\"B\n\x05Label\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0f\n\x07related\x18\x03 \x01(\t\x12\x0c\n\x04text\x18\x04 \x01(\t\x12\x0b\n\x03uri\x18\x05 \x01(\t\"\xe0\x01\n\x08LabelSet\x12\r\n\x05title\x18\x01 \x01(\t\x12\r\n\x05\x63olor\x18\x02 \x01(\t\x12#\n\x06labels\x18\x03 \x03(\x0b\x32\x13.knowledgebox.Label\x12\x10\n\x08multiple\x18\x04 \x01(\x08\x12\x31\n\x04kind\x18\x05 \x03(\x0e\x32#.knowledgebox.LabelSet.LabelSetKind\"L\n\x0cLabelSetKind\x12\r\n\tRESOURCES\x10\x00\x12\x0e\n\nPARAGRAPHS\x10\x01\x12\r\n\tSENTENCES\x10\x02\x12\x0e\n\nSELECTIONS\x10\x03\"\x87\x01\n\x06Labels\x12\x34\n\x08labelset\x18\x01 \x03(\x0b\x32\".knowledgebox.Labels.LabelsetEntry\x1aG\n\rLabelsetEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.knowledgebox.LabelSet:\x02\x38\x01\"L\n\x06\x45ntity\x12\r\n\x05value\x18\x02 \x01(\t\x12\x12\n\nrepresents\x18\x04 \x03(\t\x12\x0e\n\x06merged\x18\x03 \x01(\x08\x12\x0f\n\x07\x64\x65leted\x18\x05 \x01(\x08\"D\n\x14\x45ntitiesGroupSummary\x12\r\n\x05title\x18\x02 \x01(\t\x12\r\n\x05\x63olor\x18\x03 \x01(\t\x12\x0e\n\x06\x63ustom\x18\x04 \x01(\x08\"\xc1\x01\n\rEntitiesGroup\x12;\n\x08\x65ntities\x18\x01 \x03(\x0b\x32).knowledgebox.EntitiesGroup.EntitiesEntry\x12\r\n\x05title\x18\x02 \x01(\t\x12\r\n\x05\x63olor\x18\x03 \x01(\t\x12\x0e\n\x06\x63ustom\x18\x04 \x01(\x08\x1a\x45\n\rEntitiesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12#\n\x05value\x18\x02 \x01(\x0b\x32\x14.knowledgebox.Entity:\x02\x38\x01\"0\n\x15\x44\x65letedEntitiesGroups\x12\x17\n\x0f\x65ntities_groups\x18\x01 \x03(\t\"\xaf\x01\n\x0e\x45ntitiesGroups\x12I\n\x0f\x65ntities_groups\x18\x01 \x03(\x0b\x32\x30.knowledgebox.EntitiesGroups.EntitiesGroupsEntry\x1aR\n\x13\x45ntitiesGroupsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x1b.knowledgebox.EntitiesGroup:\x02\x38\x01\"\xf3\x03\n\x19\x45ntityGroupDuplicateIndex\x12T\n\x0f\x65ntities_groups\x18\x01 \x03(\x0b\x32;.knowledgebox.EntityGroupDuplicateIndex.EntitiesGroupsEntry\x1a&\n\x10\x45ntityDuplicates\x12\x12\n\nduplicates\x18\x01 \x03(\t\x1a\xe1\x01\n\x15\x45ntityGroupDuplicates\x12]\n\x08\x65ntities\x18\x01 \x03(\x0b\x32K.knowledgebox.EntityGroupDuplicateIndex.EntityGroupDuplicates.EntitiesEntry\x1ai\n\rEntitiesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12G\n\x05value\x18\x02 \x01(\x0b\x32\x38.knowledgebox.EntityGroupDuplicateIndex.EntityDuplicates:\x02\x38\x01\x1at\n\x13\x45ntitiesGroupsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12L\n\x05value\x18\x02 \x01(\x0b\x32=.knowledgebox.EntityGroupDuplicateIndex.EntityGroupDuplicates:\x02\x38\x01\"K\n\tVectorSet\x12\x11\n\tdimension\x18\x01 \x01(\x05\x12+\n\nsimilarity\x18\x02 \x01(\x0e\x32\x17.utils.VectorSimilarity\"\x96\x01\n\nVectorSets\x12<\n\nvectorsets\x18\x01 \x03(\x0b\x32(.knowledgebox.VectorSets.VectorsetsEntry\x1aJ\n\x0fVectorsetsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.knowledgebox.VectorSet:\x02\x38\x01\" \n\x0cTermSynonyms\x12\x10\n\x08synonyms\x18\x01 \x03(\t\"\x86\x01\n\x08Synonyms\x12\x30\n\x05terms\x18\x01 \x03(\x0b\x32!.knowledgebox.Synonyms.TermsEntry\x1aH\n\nTermsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x1a.knowledgebox.TermSynonyms:\x02\x38\x01\"\xb7\x01\n\x15SemanticModelMetadata\x12\x34\n\x13similarity_function\x18\x01 \x01(\x0e\x32\x17.utils.VectorSimilarity\x12\x1d\n\x10vector_dimension\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x1e\n\x11\x64\x65\x66\x61ult_min_score\x18\x03 \x01(\x02H\x01\x88\x01\x01\x42\x13\n\x11_vector_dimensionB\x14\n\x12_default_min_score\"\x8c\x01\n\x0fKBConfiguration\x12\x16\n\x0esemantic_model\x18\x02 \x01(\t\x12\x18\n\x10generative_model\x18\x03 \x01(\t\x12\x11\n\tner_model\x18\x04 \x01(\t\x12\x1b\n\x13\x61nonymization_model\x18\x05 \x01(\t\x12\x17\n\x0fvisual_labeling\x18\x06 \x01(\t*K\n\x1aKnowledgeBoxResponseStatus\x12\x06\n\x02OK\x10\x00\x12\x0c\n\x08\x43ONFLICT\x10\x01\x12\x0c\n\x08NOTFOUND\x10\x02\x12\t\n\x05\x45RROR\x10\x03P\x00\x62\x06proto3') _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) @@ -29,6 +29,8 @@ _globals['_KNOWLEDGEBOXCONFIG'].fields_by_name['enabled_insights']._serialized_options = b'\030\001' _globals['_KNOWLEDGEBOXCONFIG'].fields_by_name['disable_vectors']._options = None _globals['_KNOWLEDGEBOXCONFIG'].fields_by_name['disable_vectors']._serialized_options = b'\030\001' + _globals['_KNOWLEDGEBOXCONFIG'].fields_by_name['release_channel']._options = None + _globals['_KNOWLEDGEBOXCONFIG'].fields_by_name['release_channel']._serialized_options = b'\030\001' _globals['_LABELS_LABELSETENTRY']._options = None _globals['_LABELS_LABELSETENTRY']._serialized_options = b'8\001' _globals['_ENTITIESGROUP_ENTITIESENTRY']._options = None @@ -43,72 +45,72 @@ _globals['_VECTORSETS_VECTORSETSENTRY']._serialized_options = b'8\001' _globals['_SYNONYMS_TERMSENTRY']._options = None _globals['_SYNONYMS_TERMSENTRY']._serialized_options = b'8\001' - _globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_start=3342 - _globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_end=3417 + _globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_start=3346 + _globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_end=3421 _globals['_KNOWLEDGEBOXID']._serialized_start=81 _globals['_KNOWLEDGEBOXID']._serialized_end=125 _globals['_KNOWLEDGEBOXCONFIG']._serialized_start=128 - _globals['_KNOWLEDGEBOXCONFIG']._serialized_end=361 - _globals['_KNOWLEDGEBOXNEW']._serialized_start=364 - _globals['_KNOWLEDGEBOXNEW']._serialized_end=688 - _globals['_NEWKNOWLEDGEBOXRESPONSE']._serialized_start=690 - _globals['_NEWKNOWLEDGEBOXRESPONSE']._serialized_end=787 - _globals['_KNOWLEDGEBOXUPDATE']._serialized_start=789 - _globals['_KNOWLEDGEBOXUPDATE']._serialized_end=887 - _globals['_UPDATEKNOWLEDGEBOXRESPONSE']._serialized_start=889 - _globals['_UPDATEKNOWLEDGEBOXRESPONSE']._serialized_end=989 - _globals['_GCKNOWLEDGEBOXRESPONSE']._serialized_start=991 - _globals['_GCKNOWLEDGEBOXRESPONSE']._serialized_end=1015 - _globals['_DELETEKNOWLEDGEBOXRESPONSE']._serialized_start=1017 - _globals['_DELETEKNOWLEDGEBOXRESPONSE']._serialized_end=1103 - _globals['_LABEL']._serialized_start=1105 - _globals['_LABEL']._serialized_end=1171 - _globals['_LABELSET']._serialized_start=1174 - _globals['_LABELSET']._serialized_end=1398 - _globals['_LABELSET_LABELSETKIND']._serialized_start=1322 - _globals['_LABELSET_LABELSETKIND']._serialized_end=1398 - _globals['_LABELS']._serialized_start=1401 - _globals['_LABELS']._serialized_end=1536 - _globals['_LABELS_LABELSETENTRY']._serialized_start=1465 - _globals['_LABELS_LABELSETENTRY']._serialized_end=1536 - _globals['_ENTITY']._serialized_start=1538 - _globals['_ENTITY']._serialized_end=1614 - _globals['_ENTITIESGROUPSUMMARY']._serialized_start=1616 - _globals['_ENTITIESGROUPSUMMARY']._serialized_end=1684 - _globals['_ENTITIESGROUP']._serialized_start=1687 - _globals['_ENTITIESGROUP']._serialized_end=1880 - _globals['_ENTITIESGROUP_ENTITIESENTRY']._serialized_start=1811 - _globals['_ENTITIESGROUP_ENTITIESENTRY']._serialized_end=1880 - _globals['_DELETEDENTITIESGROUPS']._serialized_start=1882 - _globals['_DELETEDENTITIESGROUPS']._serialized_end=1930 - _globals['_ENTITIESGROUPS']._serialized_start=1933 - _globals['_ENTITIESGROUPS']._serialized_end=2108 - _globals['_ENTITIESGROUPS_ENTITIESGROUPSENTRY']._serialized_start=2026 - _globals['_ENTITIESGROUPS_ENTITIESGROUPSENTRY']._serialized_end=2108 - _globals['_ENTITYGROUPDUPLICATEINDEX']._serialized_start=2111 - _globals['_ENTITYGROUPDUPLICATEINDEX']._serialized_end=2610 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYDUPLICATES']._serialized_start=2226 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYDUPLICATES']._serialized_end=2264 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES']._serialized_start=2267 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES']._serialized_end=2492 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES_ENTITIESENTRY']._serialized_start=2387 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES_ENTITIESENTRY']._serialized_end=2492 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITIESGROUPSENTRY']._serialized_start=2494 - _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITIESGROUPSENTRY']._serialized_end=2610 - _globals['_VECTORSET']._serialized_start=2612 - _globals['_VECTORSET']._serialized_end=2687 - _globals['_VECTORSETS']._serialized_start=2690 - _globals['_VECTORSETS']._serialized_end=2840 - _globals['_VECTORSETS_VECTORSETSENTRY']._serialized_start=2766 - _globals['_VECTORSETS_VECTORSETSENTRY']._serialized_end=2840 - _globals['_TERMSYNONYMS']._serialized_start=2842 - _globals['_TERMSYNONYMS']._serialized_end=2874 - _globals['_SYNONYMS']._serialized_start=2877 - _globals['_SYNONYMS']._serialized_end=3011 - _globals['_SYNONYMS_TERMSENTRY']._serialized_start=2939 - _globals['_SYNONYMS_TERMSENTRY']._serialized_end=3011 - _globals['_SEMANTICMODELMETADATA']._serialized_start=3014 - _globals['_SEMANTICMODELMETADATA']._serialized_end=3197 - _globals['_KBCONFIGURATION']._serialized_start=3200 - _globals['_KBCONFIGURATION']._serialized_end=3340 + _globals['_KNOWLEDGEBOXCONFIG']._serialized_end=365 + _globals['_KNOWLEDGEBOXNEW']._serialized_start=368 + _globals['_KNOWLEDGEBOXNEW']._serialized_end=692 + _globals['_NEWKNOWLEDGEBOXRESPONSE']._serialized_start=694 + _globals['_NEWKNOWLEDGEBOXRESPONSE']._serialized_end=791 + _globals['_KNOWLEDGEBOXUPDATE']._serialized_start=793 + _globals['_KNOWLEDGEBOXUPDATE']._serialized_end=891 + _globals['_UPDATEKNOWLEDGEBOXRESPONSE']._serialized_start=893 + _globals['_UPDATEKNOWLEDGEBOXRESPONSE']._serialized_end=993 + _globals['_GCKNOWLEDGEBOXRESPONSE']._serialized_start=995 + _globals['_GCKNOWLEDGEBOXRESPONSE']._serialized_end=1019 + _globals['_DELETEKNOWLEDGEBOXRESPONSE']._serialized_start=1021 + _globals['_DELETEKNOWLEDGEBOXRESPONSE']._serialized_end=1107 + _globals['_LABEL']._serialized_start=1109 + _globals['_LABEL']._serialized_end=1175 + _globals['_LABELSET']._serialized_start=1178 + _globals['_LABELSET']._serialized_end=1402 + _globals['_LABELSET_LABELSETKIND']._serialized_start=1326 + _globals['_LABELSET_LABELSETKIND']._serialized_end=1402 + _globals['_LABELS']._serialized_start=1405 + _globals['_LABELS']._serialized_end=1540 + _globals['_LABELS_LABELSETENTRY']._serialized_start=1469 + _globals['_LABELS_LABELSETENTRY']._serialized_end=1540 + _globals['_ENTITY']._serialized_start=1542 + _globals['_ENTITY']._serialized_end=1618 + _globals['_ENTITIESGROUPSUMMARY']._serialized_start=1620 + _globals['_ENTITIESGROUPSUMMARY']._serialized_end=1688 + _globals['_ENTITIESGROUP']._serialized_start=1691 + _globals['_ENTITIESGROUP']._serialized_end=1884 + _globals['_ENTITIESGROUP_ENTITIESENTRY']._serialized_start=1815 + _globals['_ENTITIESGROUP_ENTITIESENTRY']._serialized_end=1884 + _globals['_DELETEDENTITIESGROUPS']._serialized_start=1886 + _globals['_DELETEDENTITIESGROUPS']._serialized_end=1934 + _globals['_ENTITIESGROUPS']._serialized_start=1937 + _globals['_ENTITIESGROUPS']._serialized_end=2112 + _globals['_ENTITIESGROUPS_ENTITIESGROUPSENTRY']._serialized_start=2030 + _globals['_ENTITIESGROUPS_ENTITIESGROUPSENTRY']._serialized_end=2112 + _globals['_ENTITYGROUPDUPLICATEINDEX']._serialized_start=2115 + _globals['_ENTITYGROUPDUPLICATEINDEX']._serialized_end=2614 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYDUPLICATES']._serialized_start=2230 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYDUPLICATES']._serialized_end=2268 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES']._serialized_start=2271 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES']._serialized_end=2496 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES_ENTITIESENTRY']._serialized_start=2391 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES_ENTITIESENTRY']._serialized_end=2496 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITIESGROUPSENTRY']._serialized_start=2498 + _globals['_ENTITYGROUPDUPLICATEINDEX_ENTITIESGROUPSENTRY']._serialized_end=2614 + _globals['_VECTORSET']._serialized_start=2616 + _globals['_VECTORSET']._serialized_end=2691 + _globals['_VECTORSETS']._serialized_start=2694 + _globals['_VECTORSETS']._serialized_end=2844 + _globals['_VECTORSETS_VECTORSETSENTRY']._serialized_start=2770 + _globals['_VECTORSETS_VECTORSETSENTRY']._serialized_end=2844 + _globals['_TERMSYNONYMS']._serialized_start=2846 + _globals['_TERMSYNONYMS']._serialized_end=2878 + _globals['_SYNONYMS']._serialized_start=2881 + _globals['_SYNONYMS']._serialized_end=3015 + _globals['_SYNONYMS_TERMSENTRY']._serialized_start=2943 + _globals['_SYNONYMS_TERMSENTRY']._serialized_end=3015 + _globals['_SEMANTICMODELMETADATA']._serialized_start=3018 + _globals['_SEMANTICMODELMETADATA']._serialized_end=3201 + _globals['_KBCONFIGURATION']._serialized_start=3204 + _globals['_KBCONFIGURATION']._serialized_end=3344 # @@protoc_insertion_point(module_scope) diff --git a/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.pyi index ba56dd8207..c125c2e87f 100644 --- a/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/knowledgebox_pb2.pyi @@ -102,6 +102,7 @@ class KnowledgeBoxConfig(google.protobuf.message.Message): disable_vectors: builtins.bool migration_version: builtins.int release_channel: nucliadb_protos.utils_pb2.ReleaseChannel.ValueType + """DEPRECATED: duplicated field also stored in `writer.proto Shards`""" def __init__( self, *, @@ -286,7 +287,7 @@ class LabelSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _LabelSetKindEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[LabelSet._LabelSetKind.ValueType], builtins.type): + class _LabelSetKindEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[LabelSet._LabelSetKind.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor RESOURCES: LabelSet._LabelSetKind.ValueType # 0 PARAGRAPHS: LabelSet._LabelSetKind.ValueType # 1 diff --git a/nucliadb_protos/python/nucliadb_protos/nodereader_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/nodereader_pb2.pyi index 4d2a66e38b..8676f1be77 100644 --- a/nucliadb_protos/python/nucliadb_protos/nodereader_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/nodereader_pb2.pyi @@ -111,7 +111,7 @@ class StreamFilter(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ConjunctionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[StreamFilter._Conjunction.ValueType], builtins.type): + class _ConjunctionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[StreamFilter._Conjunction.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor AND: StreamFilter._Conjunction.ValueType # 0 OR: StreamFilter._Conjunction.ValueType # 1 @@ -161,7 +161,7 @@ class OrderBy(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _OrderTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OrderBy._OrderType.ValueType], builtins.type): + class _OrderTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OrderBy._OrderType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DESC: OrderBy._OrderType.ValueType # 0 ASC: OrderBy._OrderType.ValueType # 1 @@ -174,7 +174,7 @@ class OrderBy(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _OrderFieldEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OrderBy._OrderField.ValueType], builtins.type): + class _OrderFieldEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OrderBy._OrderField.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CREATED: OrderBy._OrderField.ValueType # 0 MODIFIED: OrderBy._OrderField.ValueType # 1 diff --git a/nucliadb_protos/python/nucliadb_protos/nodereader_pb2_grpc.pyi b/nucliadb_protos/python/nucliadb_protos/nodereader_pb2_grpc.pyi index be15024620..2b879f4852 100644 --- a/nucliadb_protos/python/nucliadb_protos/nodereader_pb2_grpc.pyi +++ b/nucliadb_protos/python/nucliadb_protos/nodereader_pb2_grpc.pyi @@ -5,10 +5,8 @@ isort:skip_file import abc import collections.abc import grpc -import grpc.aio import nucliadb_protos.nodereader_pb2 import nucliadb_protos.noderesources_pb2 -import typing from nucliadb_protos.noderesources_pb2 import ( EmptyQuery as EmptyQuery, EmptyResponse as EmptyResponse, @@ -53,18 +51,10 @@ from nucliadb_protos.utils_pb2 import ( Vectors as Vectors, ) -_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 - ... - class NodeReaderStub: """Implemented at nucliadb_object_storage""" - def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ... + def __init__(self, channel: grpc.Channel) -> None: ... GetShard: grpc.UnaryUnaryMultiCallable[ nucliadb_protos.nodereader_pb2.GetShardRequest, nucliadb_protos.noderesources_pb2.Shard, @@ -132,76 +122,6 @@ class NodeReaderStub: nucliadb_protos.nodereader_pb2.ShardFileChunk, ] -class NodeReaderAsyncStub: - """Implemented at nucliadb_object_storage""" - - GetShard: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.GetShardRequest, - nucliadb_protos.noderesources_pb2.Shard, - ] - DocumentSearch: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.DocumentSearchRequest, - nucliadb_protos.nodereader_pb2.DocumentSearchResponse, - ] - ParagraphSearch: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.ParagraphSearchRequest, - nucliadb_protos.nodereader_pb2.ParagraphSearchResponse, - ] - VectorSearch: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.VectorSearchRequest, - nucliadb_protos.nodereader_pb2.VectorSearchResponse, - ] - RelationSearch: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.RelationSearchRequest, - nucliadb_protos.nodereader_pb2.RelationSearchResponse, - ] - DocumentIds: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodereader_pb2.IdCollection, - ] - ParagraphIds: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodereader_pb2.IdCollection, - ] - VectorIds: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodereader_pb2.IdCollection, - ] - RelationIds: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodereader_pb2.IdCollection, - ] - RelationEdges: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodereader_pb2.EdgeList, - ] - Search: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.SearchRequest, - nucliadb_protos.nodereader_pb2.SearchResponse, - ] - Suggest: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.SuggestRequest, - nucliadb_protos.nodereader_pb2.SuggestResponse, - ] - Paragraphs: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.nodereader_pb2.StreamRequest, - nucliadb_protos.nodereader_pb2.ParagraphItem, - ] - """Streams""" - Documents: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.nodereader_pb2.StreamRequest, - nucliadb_protos.nodereader_pb2.DocumentItem, - ] - GetShardFiles: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodereader_pb2.GetShardFilesRequest, - nucliadb_protos.nodereader_pb2.ShardFileList, - ] - """Shard Download""" - DownloadShardFile: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.nodereader_pb2.DownloadShardFileRequest, - nucliadb_protos.nodereader_pb2.ShardFileChunk, - ] - class NodeReaderServicer(metaclass=abc.ABCMeta): """Implemented at nucliadb_object_storage""" @@ -209,99 +129,99 @@ class NodeReaderServicer(metaclass=abc.ABCMeta): def GetShard( self, request: nucliadb_protos.nodereader_pb2.GetShardRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.Shard, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.Shard]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.Shard: ... @abc.abstractmethod def DocumentSearch( self, request: nucliadb_protos.nodereader_pb2.DocumentSearchRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.DocumentSearchResponse, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.DocumentSearchResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.DocumentSearchResponse: ... @abc.abstractmethod def ParagraphSearch( self, request: nucliadb_protos.nodereader_pb2.ParagraphSearchRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.ParagraphSearchResponse, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.ParagraphSearchResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.ParagraphSearchResponse: ... @abc.abstractmethod def VectorSearch( self, request: nucliadb_protos.nodereader_pb2.VectorSearchRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.VectorSearchResponse, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.VectorSearchResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.VectorSearchResponse: ... @abc.abstractmethod def RelationSearch( self, request: nucliadb_protos.nodereader_pb2.RelationSearchRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.RelationSearchResponse, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.RelationSearchResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.RelationSearchResponse: ... @abc.abstractmethod def DocumentIds( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.IdCollection, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.IdCollection]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.IdCollection: ... @abc.abstractmethod def ParagraphIds( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.IdCollection, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.IdCollection]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.IdCollection: ... @abc.abstractmethod def VectorIds( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.IdCollection, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.IdCollection]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.IdCollection: ... @abc.abstractmethod def RelationIds( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.IdCollection, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.IdCollection]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.IdCollection: ... @abc.abstractmethod def RelationEdges( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.EdgeList, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.EdgeList]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.EdgeList: ... @abc.abstractmethod def Search( self, request: nucliadb_protos.nodereader_pb2.SearchRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.SearchResponse, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.SearchResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.SearchResponse: ... @abc.abstractmethod def Suggest( self, request: nucliadb_protos.nodereader_pb2.SuggestRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.SuggestResponse, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.SuggestResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.SuggestResponse: ... @abc.abstractmethod def Paragraphs( self, request: nucliadb_protos.nodereader_pb2.StreamRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.nodereader_pb2.ParagraphItem], collections.abc.AsyncIterator[nucliadb_protos.nodereader_pb2.ParagraphItem]]: + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.nodereader_pb2.ParagraphItem]: """Streams""" @abc.abstractmethod def Documents( self, request: nucliadb_protos.nodereader_pb2.StreamRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.nodereader_pb2.DocumentItem], collections.abc.AsyncIterator[nucliadb_protos.nodereader_pb2.DocumentItem]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.nodereader_pb2.DocumentItem]: ... @abc.abstractmethod def GetShardFiles( self, request: nucliadb_protos.nodereader_pb2.GetShardFilesRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodereader_pb2.ShardFileList, collections.abc.Awaitable[nucliadb_protos.nodereader_pb2.ShardFileList]]: + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodereader_pb2.ShardFileList: """Shard Download""" @abc.abstractmethod def DownloadShardFile( self, request: nucliadb_protos.nodereader_pb2.DownloadShardFileRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.nodereader_pb2.ShardFileChunk], collections.abc.AsyncIterator[nucliadb_protos.nodereader_pb2.ShardFileChunk]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.nodereader_pb2.ShardFileChunk]: ... -def add_NodeReaderServicer_to_server(servicer: NodeReaderServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ... +def add_NodeReaderServicer_to_server(servicer: NodeReaderServicer, server: grpc.Server) -> None: ... diff --git a/nucliadb_protos/python/nucliadb_protos/noderesources_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/noderesources_pb2.pyi index 477bf36f6a..0bbe3b42fb 100644 --- a/nucliadb_protos/python/nucliadb_protos/noderesources_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/noderesources_pb2.pyi @@ -121,7 +121,7 @@ class ShardCreated(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _DocumentServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._DocumentService.ValueType], builtins.type): + class _DocumentServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._DocumentService.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DOCUMENT_V0: ShardCreated._DocumentService.ValueType # 0 DOCUMENT_V1: ShardCreated._DocumentService.ValueType # 1 @@ -136,7 +136,7 @@ class ShardCreated(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ParagraphServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._ParagraphService.ValueType], builtins.type): + class _ParagraphServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._ParagraphService.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PARAGRAPH_V0: ShardCreated._ParagraphService.ValueType # 0 PARAGRAPH_V1: ShardCreated._ParagraphService.ValueType # 1 @@ -151,7 +151,7 @@ class ShardCreated(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _VectorServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._VectorService.ValueType], builtins.type): + class _VectorServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._VectorService.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor VECTOR_V0: ShardCreated._VectorService.ValueType # 0 VECTOR_V1: ShardCreated._VectorService.ValueType # 1 @@ -164,7 +164,7 @@ class ShardCreated(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _RelationServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._RelationService.ValueType], builtins.type): + class _RelationServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._RelationService.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor RELATION_V0: ShardCreated._RelationService.ValueType # 0 RELATION_V1: ShardCreated._RelationService.ValueType # 1 @@ -539,7 +539,7 @@ class Resource(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ResourceStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Resource._ResourceStatus.ValueType], builtins.type): + class _ResourceStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Resource._ResourceStatus.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PROCESSED: Resource._ResourceStatus.ValueType # 0 EMPTY: Resource._ResourceStatus.ValueType # 1 diff --git a/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2.pyi index 02c65dce59..106167e37a 100644 --- a/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2.pyi @@ -80,7 +80,7 @@ class OpStatus(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpStatus._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpStatus._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: OpStatus._Status.ValueType # 0 WARNING: OpStatus._Status.ValueType # 1 @@ -170,7 +170,7 @@ class GarbageCollectorResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GarbageCollectorResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GarbageCollectorResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: GarbageCollectorResponse._Status.ValueType # 0 TRY_LATER: GarbageCollectorResponse._Status.ValueType # 1 @@ -239,7 +239,7 @@ class MergeResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MergeStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[MergeResponse._MergeStatus.ValueType], builtins.type): + class _MergeStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[MergeResponse._MergeStatus.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: MergeResponse._MergeStatus.ValueType # 0 diff --git a/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2_grpc.pyi b/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2_grpc.pyi index 948d282e03..898124503c 100644 --- a/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2_grpc.pyi +++ b/nucliadb_protos/python/nucliadb_protos/nodewriter_pb2_grpc.pyi @@ -3,12 +3,9 @@ isort:skip_file """ import abc -import collections.abc import grpc -import grpc.aio import nucliadb_protos.noderesources_pb2 import nucliadb_protos.nodewriter_pb2 -import typing from nucliadb_protos.noderesources_pb2 import ( EmptyQuery as EmptyQuery, EmptyResponse as EmptyResponse, @@ -33,16 +30,8 @@ from nucliadb_protos.noderesources_pb2 import ( VectorSetList as VectorSetList, ) -_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 - ... - class NodeWriterStub: - def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ... + def __init__(self, channel: grpc.Channel) -> None: ... NewShard: grpc.UnaryUnaryMultiCallable[ nucliadb_protos.nodewriter_pb2.NewShardRequest, nucliadb_protos.noderesources_pb2.ShardCreated, @@ -88,118 +77,72 @@ class NodeWriterStub: nucliadb_protos.noderesources_pb2.NodeMetadata, ] -class NodeWriterAsyncStub: - NewShard: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodewriter_pb2.NewShardRequest, - nucliadb_protos.noderesources_pb2.ShardCreated, - ] - DeleteShard: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.noderesources_pb2.ShardId, - ] - ListShards: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.EmptyQuery, - nucliadb_protos.noderesources_pb2.ShardIds, - ] - GC: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodewriter_pb2.GarbageCollectorResponse, - ] - Merge: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.nodewriter_pb2.MergeResponse, - ] - SetResource: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.Resource, - nucliadb_protos.nodewriter_pb2.OpStatus, - ] - RemoveResource: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ResourceID, - nucliadb_protos.nodewriter_pb2.OpStatus, - ] - AddVectorSet: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.nodewriter_pb2.NewVectorSetRequest, - nucliadb_protos.nodewriter_pb2.OpStatus, - ] - RemoveVectorSet: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.VectorSetID, - nucliadb_protos.nodewriter_pb2.OpStatus, - ] - ListVectorSets: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.ShardId, - nucliadb_protos.noderesources_pb2.VectorSetList, - ] - GetMetadata: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.EmptyQuery, - nucliadb_protos.noderesources_pb2.NodeMetadata, - ] - class NodeWriterServicer(metaclass=abc.ABCMeta): @abc.abstractmethod def NewShard( self, request: nucliadb_protos.nodewriter_pb2.NewShardRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.ShardCreated, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.ShardCreated]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.ShardCreated: ... @abc.abstractmethod def DeleteShard( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.ShardId, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.ShardId]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.ShardId: ... @abc.abstractmethod def ListShards( self, request: nucliadb_protos.noderesources_pb2.EmptyQuery, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.ShardIds, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.ShardIds]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.ShardIds: ... @abc.abstractmethod def GC( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodewriter_pb2.GarbageCollectorResponse, collections.abc.Awaitable[nucliadb_protos.nodewriter_pb2.GarbageCollectorResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodewriter_pb2.GarbageCollectorResponse: ... @abc.abstractmethod def Merge( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodewriter_pb2.MergeResponse, collections.abc.Awaitable[nucliadb_protos.nodewriter_pb2.MergeResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodewriter_pb2.MergeResponse: ... @abc.abstractmethod def SetResource( self, request: nucliadb_protos.noderesources_pb2.Resource, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodewriter_pb2.OpStatus, collections.abc.Awaitable[nucliadb_protos.nodewriter_pb2.OpStatus]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodewriter_pb2.OpStatus: ... @abc.abstractmethod def RemoveResource( self, request: nucliadb_protos.noderesources_pb2.ResourceID, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodewriter_pb2.OpStatus, collections.abc.Awaitable[nucliadb_protos.nodewriter_pb2.OpStatus]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodewriter_pb2.OpStatus: ... @abc.abstractmethod def AddVectorSet( self, request: nucliadb_protos.nodewriter_pb2.NewVectorSetRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodewriter_pb2.OpStatus, collections.abc.Awaitable[nucliadb_protos.nodewriter_pb2.OpStatus]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodewriter_pb2.OpStatus: ... @abc.abstractmethod def RemoveVectorSet( self, request: nucliadb_protos.noderesources_pb2.VectorSetID, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.nodewriter_pb2.OpStatus, collections.abc.Awaitable[nucliadb_protos.nodewriter_pb2.OpStatus]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.nodewriter_pb2.OpStatus: ... @abc.abstractmethod def ListVectorSets( self, request: nucliadb_protos.noderesources_pb2.ShardId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.VectorSetList, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.VectorSetList]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.VectorSetList: ... @abc.abstractmethod def GetMetadata( self, request: nucliadb_protos.noderesources_pb2.EmptyQuery, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.NodeMetadata, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.NodeMetadata]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.NodeMetadata: ... -def add_NodeWriterServicer_to_server(servicer: NodeWriterServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ... +def add_NodeWriterServicer_to_server(servicer: NodeWriterServicer, server: grpc.Server) -> None: ... diff --git a/nucliadb_protos/python/nucliadb_protos/replication_pb2_grpc.pyi b/nucliadb_protos/python/nucliadb_protos/replication_pb2_grpc.pyi index 6a9b72f515..0113f30b2d 100644 --- a/nucliadb_protos/python/nucliadb_protos/replication_pb2_grpc.pyi +++ b/nucliadb_protos/python/nucliadb_protos/replication_pb2_grpc.pyi @@ -5,10 +5,8 @@ isort:skip_file import abc import collections.abc import grpc -import grpc.aio import nucliadb_protos.noderesources_pb2 import nucliadb_protos.replication_pb2 -import typing from nucliadb_protos.noderesources_pb2 import ( EmptyQuery as EmptyQuery, EmptyResponse as EmptyResponse, @@ -33,16 +31,8 @@ from nucliadb_protos.noderesources_pb2 import ( VectorSetList as VectorSetList, ) -_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 - ... - class ReplicationServiceStub: - def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ... + def __init__(self, channel: grpc.Channel) -> None: ... CheckReplicationState: grpc.UnaryUnaryMultiCallable[ nucliadb_protos.replication_pb2.SecondaryCheckReplicationStateRequest, nucliadb_protos.replication_pb2.PrimaryCheckReplicationStateResponse, @@ -57,40 +47,25 @@ class ReplicationServiceStub: nucliadb_protos.noderesources_pb2.NodeMetadata, ] -class ReplicationServiceAsyncStub: - CheckReplicationState: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.replication_pb2.SecondaryCheckReplicationStateRequest, - nucliadb_protos.replication_pb2.PrimaryCheckReplicationStateResponse, - ] - """Shard replication RPCs""" - ReplicateShard: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.replication_pb2.ReplicateShardRequest, - nucliadb_protos.replication_pb2.ReplicateShardResponse, - ] - GetMetadata: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.noderesources_pb2.EmptyQuery, - nucliadb_protos.noderesources_pb2.NodeMetadata, - ] - class ReplicationServiceServicer(metaclass=abc.ABCMeta): @abc.abstractmethod def CheckReplicationState( self, request: nucliadb_protos.replication_pb2.SecondaryCheckReplicationStateRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.replication_pb2.PrimaryCheckReplicationStateResponse, collections.abc.Awaitable[nucliadb_protos.replication_pb2.PrimaryCheckReplicationStateResponse]]: + context: grpc.ServicerContext, + ) -> nucliadb_protos.replication_pb2.PrimaryCheckReplicationStateResponse: """Shard replication RPCs""" @abc.abstractmethod def ReplicateShard( self, request: nucliadb_protos.replication_pb2.ReplicateShardRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.replication_pb2.ReplicateShardResponse], collections.abc.AsyncIterator[nucliadb_protos.replication_pb2.ReplicateShardResponse]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.replication_pb2.ReplicateShardResponse]: ... @abc.abstractmethod def GetMetadata( self, request: nucliadb_protos.noderesources_pb2.EmptyQuery, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.noderesources_pb2.NodeMetadata, collections.abc.Awaitable[nucliadb_protos.noderesources_pb2.NodeMetadata]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.noderesources_pb2.NodeMetadata: ... -def add_ReplicationServiceServicer_to_server(servicer: ReplicationServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ... +def add_ReplicationServiceServicer_to_server(servicer: ReplicationServiceServicer, server: grpc.Server) -> None: ... diff --git a/nucliadb_protos/python/nucliadb_protos/resources_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/resources_pb2.pyi index 026c1333f2..c59a99c65e 100644 --- a/nucliadb_protos/python/nucliadb_protos/resources_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/resources_pb2.pyi @@ -78,7 +78,7 @@ class CloudFile(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _SourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CloudFile._Source.ValueType], builtins.type): + class _SourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CloudFile._Source.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor FLAPS: CloudFile._Source.ValueType # 0 GCS: CloudFile._Source.ValueType # 1 @@ -156,7 +156,7 @@ class Basic(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _QueueTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Basic._QueueType.ValueType], builtins.type): + class _QueueTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Basic._QueueType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PRIVATE: Basic._QueueType.ValueType # 0 SHARED: Basic._QueueType.ValueType # 1 @@ -247,7 +247,7 @@ class Origin(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _SourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Origin._Source.ValueType], builtins.type): + class _SourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Origin._Source.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor WEB: Origin._Source.ValueType # 0 DESKTOP: Origin._Source.ValueType # 1 @@ -363,7 +363,7 @@ class MessageContent(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[MessageContent._Format.ValueType], builtins.type): + class _FormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[MessageContent._Format.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PLAIN: MessageContent._Format.ValueType # 0 HTML: MessageContent._Format.ValueType # 1 @@ -404,7 +404,7 @@ class Message(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MessageTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Message._MessageType.ValueType], builtins.type): + class _MessageTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Message._MessageType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSET: Message._MessageType.ValueType # 0 QUESTION: Message._MessageType.ValueType # 1 @@ -919,7 +919,7 @@ class Paragraph(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeParagraphEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Paragraph._TypeParagraph.ValueType], builtins.type): + class _TypeParagraphEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Paragraph._TypeParagraph.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TEXT: Paragraph._TypeParagraph.ValueType # 0 OCR: Paragraph._TypeParagraph.ValueType # 1 @@ -1302,7 +1302,7 @@ class Metadata(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Metadata._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Metadata._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PENDING: Metadata._Status.ValueType # 0 PROCESSED: Metadata._Status.ValueType # 1 @@ -1366,7 +1366,7 @@ class FieldText(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldText._Format.ValueType], builtins.type): + class _FormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldText._Format.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PLAIN: FieldText._Format.ValueType # 0 HTML: FieldText._Format.ValueType # 1 @@ -1410,7 +1410,7 @@ class Block(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeBlockEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Block._TypeBlock.ValueType], builtins.type): + class _TypeBlockEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Block._TypeBlock.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TITLE: Block._TypeBlock.ValueType # 0 DESCRIPTION: Block._TypeBlock.ValueType # 1 @@ -1509,7 +1509,7 @@ class FieldLayout(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldLayout._Format.ValueType], builtins.type): + class _FormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldLayout._Format.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NUCLIAv1: FieldLayout._Format.ValueType # 0 diff --git a/nucliadb_protos/python/nucliadb_protos/standalone_pb2_grpc.pyi b/nucliadb_protos/python/nucliadb_protos/standalone_pb2_grpc.pyi index 70abb952e9..db822104af 100644 --- a/nucliadb_protos/python/nucliadb_protos/standalone_pb2_grpc.pyi +++ b/nucliadb_protos/python/nucliadb_protos/standalone_pb2_grpc.pyi @@ -3,22 +3,11 @@ isort:skip_file """ import abc -import collections.abc import grpc -import grpc.aio import nucliadb_protos.standalone_pb2 -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 - ... class StandaloneClusterServiceStub: - def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ... + def __init__(self, channel: grpc.Channel) -> None: ... NodeAction: grpc.UnaryUnaryMultiCallable[ nucliadb_protos.standalone_pb2.NodeActionRequest, nucliadb_protos.standalone_pb2.NodeActionResponse, @@ -28,28 +17,18 @@ class StandaloneClusterServiceStub: nucliadb_protos.standalone_pb2.NodeInfoResponse, ] -class StandaloneClusterServiceAsyncStub: - NodeAction: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.standalone_pb2.NodeActionRequest, - nucliadb_protos.standalone_pb2.NodeActionResponse, - ] - NodeInfo: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.standalone_pb2.NodeInfoRequest, - nucliadb_protos.standalone_pb2.NodeInfoResponse, - ] - class StandaloneClusterServiceServicer(metaclass=abc.ABCMeta): @abc.abstractmethod def NodeAction( self, request: nucliadb_protos.standalone_pb2.NodeActionRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.standalone_pb2.NodeActionResponse, collections.abc.Awaitable[nucliadb_protos.standalone_pb2.NodeActionResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.standalone_pb2.NodeActionResponse: ... @abc.abstractmethod def NodeInfo( self, request: nucliadb_protos.standalone_pb2.NodeInfoRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.standalone_pb2.NodeInfoResponse, collections.abc.Awaitable[nucliadb_protos.standalone_pb2.NodeInfoResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.standalone_pb2.NodeInfoResponse: ... -def add_StandaloneClusterServiceServicer_to_server(servicer: StandaloneClusterServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ... +def add_StandaloneClusterServiceServicer_to_server(servicer: StandaloneClusterServiceServicer, server: grpc.Server) -> None: ... diff --git a/nucliadb_protos/python/nucliadb_protos/train_pb2_grpc.pyi b/nucliadb_protos/python/nucliadb_protos/train_pb2_grpc.pyi index f45798403b..0d3542bc4b 100644 --- a/nucliadb_protos/python/nucliadb_protos/train_pb2_grpc.pyi +++ b/nucliadb_protos/python/nucliadb_protos/train_pb2_grpc.pyi @@ -5,10 +5,8 @@ isort:skip_file import abc import collections.abc import grpc -import grpc.aio import nucliadb_protos.train_pb2 import nucliadb_protos.writer_pb2 -import typing from nucliadb_protos.knowledgebox_pb2 import ( CONFLICT as CONFLICT, DeleteKnowledgeBoxResponse as DeleteKnowledgeBoxResponse, @@ -172,16 +170,8 @@ from nucliadb_protos.writer_pb2 import ( WriterStatusResponse as WriterStatusResponse, ) -_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 - ... - class TrainStub: - def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ... + def __init__(self, channel: grpc.Channel) -> None: ... GetInfo: grpc.UnaryUnaryMultiCallable[ nucliadb_protos.train_pb2.GetInfoRequest, nucliadb_protos.train_pb2.TrainInfo, @@ -215,88 +205,54 @@ class TrainStub: nucliadb_protos.train_pb2.LabelsetsCount, ] -class TrainAsyncStub: - GetInfo: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.train_pb2.GetInfoRequest, - nucliadb_protos.train_pb2.TrainInfo, - ] - GetSentences: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.train_pb2.GetSentencesRequest, - nucliadb_protos.train_pb2.TrainSentence, - ] - GetParagraphs: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.train_pb2.GetParagraphsRequest, - nucliadb_protos.train_pb2.TrainParagraph, - ] - GetFields: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.train_pb2.GetFieldsRequest, - nucliadb_protos.train_pb2.TrainField, - ] - GetResources: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.train_pb2.GetResourcesRequest, - nucliadb_protos.train_pb2.TrainResource, - ] - GetOntology: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetLabelsRequest, - nucliadb_protos.writer_pb2.GetLabelsResponse, - ] - GetEntities: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetEntitiesRequest, - nucliadb_protos.writer_pb2.GetEntitiesResponse, - ] - GetOntologyCount: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.train_pb2.GetLabelsetsCountRequest, - nucliadb_protos.train_pb2.LabelsetsCount, - ] - class TrainServicer(metaclass=abc.ABCMeta): @abc.abstractmethod def GetInfo( self, request: nucliadb_protos.train_pb2.GetInfoRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.train_pb2.TrainInfo, collections.abc.Awaitable[nucliadb_protos.train_pb2.TrainInfo]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.train_pb2.TrainInfo: ... @abc.abstractmethod def GetSentences( self, request: nucliadb_protos.train_pb2.GetSentencesRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.train_pb2.TrainSentence], collections.abc.AsyncIterator[nucliadb_protos.train_pb2.TrainSentence]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.train_pb2.TrainSentence]: ... @abc.abstractmethod def GetParagraphs( self, request: nucliadb_protos.train_pb2.GetParagraphsRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.train_pb2.TrainParagraph], collections.abc.AsyncIterator[nucliadb_protos.train_pb2.TrainParagraph]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.train_pb2.TrainParagraph]: ... @abc.abstractmethod def GetFields( self, request: nucliadb_protos.train_pb2.GetFieldsRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.train_pb2.TrainField], collections.abc.AsyncIterator[nucliadb_protos.train_pb2.TrainField]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.train_pb2.TrainField]: ... @abc.abstractmethod def GetResources( self, request: nucliadb_protos.train_pb2.GetResourcesRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.train_pb2.TrainResource], collections.abc.AsyncIterator[nucliadb_protos.train_pb2.TrainResource]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.train_pb2.TrainResource]: ... @abc.abstractmethod def GetOntology( self, request: nucliadb_protos.writer_pb2.GetLabelsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetLabelsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetLabelsResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetLabelsResponse: ... @abc.abstractmethod def GetEntities( self, request: nucliadb_protos.writer_pb2.GetEntitiesRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetEntitiesResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetEntitiesResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetEntitiesResponse: ... @abc.abstractmethod def GetOntologyCount( self, request: nucliadb_protos.train_pb2.GetLabelsetsCountRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.train_pb2.LabelsetsCount, collections.abc.Awaitable[nucliadb_protos.train_pb2.LabelsetsCount]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.train_pb2.LabelsetsCount: ... -def add_TrainServicer_to_server(servicer: TrainServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ... +def add_TrainServicer_to_server(servicer: TrainServicer, server: grpc.Server) -> None: ... diff --git a/nucliadb_protos/python/nucliadb_protos/utils_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/utils_pb2.pyi index cd26946d23..c1ba45b153 100644 --- a/nucliadb_protos/python/nucliadb_protos/utils_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/utils_pb2.pyi @@ -60,7 +60,7 @@ class Relation(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _RelationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Relation._RelationType.ValueType], builtins.type): + class _RelationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Relation._RelationType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CHILD: Relation._RelationType.ValueType # 0 """Child resource""" @@ -164,7 +164,7 @@ class RelationNode(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _NodeTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RelationNode._NodeType.ValueType], builtins.type): + class _NodeTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RelationNode._NodeType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor ENTITY: RelationNode._NodeType.ValueType # 0 LABEL: RelationNode._NodeType.ValueType # 1 diff --git a/nucliadb_protos/python/nucliadb_protos/writer_pb2.pyi b/nucliadb_protos/python/nucliadb_protos/writer_pb2.pyi index ff475fd934..68d50f0ea8 100644 --- a/nucliadb_protos/python/nucliadb_protos/writer_pb2.pyi +++ b/nucliadb_protos/python/nucliadb_protos/writer_pb2.pyi @@ -176,7 +176,7 @@ class Audit(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _SourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Audit._Source.ValueType], builtins.type): + class _SourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Audit._Source.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor HTTP: Audit._Source.ValueType # 0 DASHBOARD: Audit._Source.ValueType # 1 @@ -217,7 +217,7 @@ class Error(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Error._ErrorCode.ValueType], builtins.type): + class _ErrorCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Error._ErrorCode.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor GENERIC: Error._ErrorCode.ValueType # 0 EXTRACT: Error._ErrorCode.ValueType # 1 @@ -256,7 +256,7 @@ class BrokerMessage(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MessageTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BrokerMessage._MessageType.ValueType], builtins.type): + class _MessageTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BrokerMessage._MessageType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor AUTOCOMMIT: BrokerMessage._MessageType.ValueType # 0 MULTI: BrokerMessage._MessageType.ValueType # 1 @@ -275,7 +275,7 @@ class BrokerMessage(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MessageSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BrokerMessage._MessageSource.ValueType], builtins.type): + class _MessageSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BrokerMessage._MessageSource.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor WRITER: BrokerMessage._MessageSource.ValueType # 0 PROCESSOR: BrokerMessage._MessageSource.ValueType # 1 @@ -687,7 +687,7 @@ class GetLabelsResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetLabelsResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetLabelsResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: GetLabelsResponse._Status.ValueType # 0 NOTFOUND: GetLabelsResponse._Status.ValueType # 1 @@ -765,7 +765,7 @@ class NewEntitiesGroupResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[NewEntitiesGroupResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[NewEntitiesGroupResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: NewEntitiesGroupResponse._Status.ValueType # 0 ERROR: NewEntitiesGroupResponse._Status.ValueType # 1 @@ -898,7 +898,7 @@ class UpdateEntitiesGroupResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[UpdateEntitiesGroupResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[UpdateEntitiesGroupResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: UpdateEntitiesGroupResponse._Status.ValueType # 0 ERROR: UpdateEntitiesGroupResponse._Status.ValueType # 1 @@ -947,7 +947,7 @@ class ListEntitiesGroupsResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ListEntitiesGroupsResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ListEntitiesGroupsResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: ListEntitiesGroupsResponse._Status.ValueType # 0 NOTFOUND: ListEntitiesGroupsResponse._Status.ValueType # 1 @@ -1016,7 +1016,7 @@ class GetEntitiesResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetEntitiesResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetEntitiesResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: GetEntitiesResponse._Status.ValueType # 0 NOTFOUND: GetEntitiesResponse._Status.ValueType # 1 @@ -1151,7 +1151,7 @@ class GetLabelSetResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetLabelSetResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetLabelSetResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: GetLabelSetResponse._Status.ValueType # 0 NOTFOUND: GetLabelSetResponse._Status.ValueType # 1 @@ -1208,7 +1208,7 @@ class GetEntitiesGroupResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetEntitiesGroupResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetEntitiesGroupResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: GetEntitiesGroupResponse._Status.ValueType # 0 KB_NOT_FOUND: GetEntitiesGroupResponse._Status.ValueType # 1 @@ -1266,7 +1266,7 @@ class GetVectorSetsResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetVectorSetsResponse._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GetVectorSetsResponse._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: GetVectorSetsResponse._Status.ValueType # 0 NOTFOUND: GetVectorSetsResponse._Status.ValueType # 1 @@ -1349,7 +1349,7 @@ class OpStatusWriter(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpStatusWriter._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpStatusWriter._Status.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OK: OpStatusWriter._Status.ValueType # 0 ERROR: OpStatusWriter._Status.ValueType # 1 @@ -1379,7 +1379,7 @@ class Notification(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ActionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Notification._Action.ValueType], builtins.type): + class _ActionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Notification._Action.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor COMMIT: Notification._Action.ValueType # 0 ABORT: Notification._Action.ValueType # 1 @@ -1394,7 +1394,7 @@ class Notification(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _WriteTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Notification._WriteType.ValueType], builtins.type): + class _WriteTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Notification._WriteType.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSET: Notification._WriteType.ValueType # 0 CREATED: Notification._WriteType.ValueType # 1 @@ -1457,7 +1457,7 @@ class Member(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Member._Type.ValueType], builtins.type): + class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Member._Type.ValueType], builtins.type): # noqa: F821 DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor IO: Member._Type.ValueType # 0 SEARCH: Member._Type.ValueType # 1 diff --git a/nucliadb_protos/python/nucliadb_protos/writer_pb2_grpc.pyi b/nucliadb_protos/python/nucliadb_protos/writer_pb2_grpc.pyi index 6650748664..95d6f8d515 100644 --- a/nucliadb_protos/python/nucliadb_protos/writer_pb2_grpc.pyi +++ b/nucliadb_protos/python/nucliadb_protos/writer_pb2_grpc.pyi @@ -5,10 +5,8 @@ isort:skip_file import abc import collections.abc import grpc -import grpc.aio import nucliadb_protos.knowledgebox_pb2 import nucliadb_protos.writer_pb2 -import typing from nucliadb_protos.knowledgebox_pb2 import ( CONFLICT as CONFLICT, DeleteKnowledgeBoxResponse as DeleteKnowledgeBoxResponse, @@ -136,16 +134,8 @@ from nucliadb_protos.resources_pb2 import ( VisualSelection as VisualSelection, ) -_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 - ... - class WriterStub: - def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ... + def __init__(self, channel: grpc.Channel) -> None: ... NewKnowledgeBox: grpc.UnaryUnaryMultiCallable[ nucliadb_protos.knowledgebox_pb2.KnowledgeBoxNew, nucliadb_protos.knowledgebox_pb2.NewKnowledgeBoxResponse, @@ -275,326 +265,196 @@ class WriterStub: nucliadb_protos.writer_pb2.FileUploaded, ] -class WriterAsyncStub: - NewKnowledgeBox: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.knowledgebox_pb2.KnowledgeBoxNew, - nucliadb_protos.knowledgebox_pb2.NewKnowledgeBoxResponse, - ] - DeleteKnowledgeBox: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - nucliadb_protos.knowledgebox_pb2.DeleteKnowledgeBoxResponse, - ] - UpdateKnowledgeBox: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.knowledgebox_pb2.KnowledgeBoxUpdate, - nucliadb_protos.knowledgebox_pb2.UpdateKnowledgeBoxResponse, - ] - GCKnowledgeBox: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - nucliadb_protos.knowledgebox_pb2.GCKnowledgeBoxResponse, - ] - SetVectors: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.SetVectorsRequest, - nucliadb_protos.writer_pb2.SetVectorsResponse, - ] - ResourceFieldExists: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.ResourceFieldId, - nucliadb_protos.writer_pb2.ResourceFieldExistsResponse, - ] - GetResourceId: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.ResourceIdRequest, - nucliadb_protos.writer_pb2.ResourceIdResponse, - ] - ProcessMessage: grpc.aio.StreamUnaryMultiCallable[ - nucliadb_protos.writer_pb2.BrokerMessage, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - GetLabels: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetLabelsRequest, - nucliadb_protos.writer_pb2.GetLabelsResponse, - ] - """Labels""" - GetLabelSet: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetLabelSetRequest, - nucliadb_protos.writer_pb2.GetLabelSetResponse, - ] - SetLabels: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.SetLabelsRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - DelLabels: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.DelLabelsRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - GetVectorSets: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetVectorSetsRequest, - nucliadb_protos.writer_pb2.GetVectorSetsResponse, - ] - """VectorSets""" - DelVectorSet: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.DelVectorSetRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - SetVectorSet: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.SetVectorSetRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - NewEntitiesGroup: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.NewEntitiesGroupRequest, - nucliadb_protos.writer_pb2.NewEntitiesGroupResponse, - ] - """Entities""" - GetEntities: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetEntitiesRequest, - nucliadb_protos.writer_pb2.GetEntitiesResponse, - ] - GetEntitiesGroup: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.GetEntitiesGroupRequest, - nucliadb_protos.writer_pb2.GetEntitiesGroupResponse, - ] - ListEntitiesGroups: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.ListEntitiesGroupsRequest, - nucliadb_protos.writer_pb2.ListEntitiesGroupsResponse, - ] - SetEntities: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.SetEntitiesRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - UpdateEntitiesGroup: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.UpdateEntitiesGroupRequest, - nucliadb_protos.writer_pb2.UpdateEntitiesGroupResponse, - ] - DelEntities: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.DelEntitiesRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - GetSynonyms: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - nucliadb_protos.writer_pb2.GetSynonymsResponse, - ] - """Synonyms""" - SetSynonyms: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.SetSynonymsRequest, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - DelSynonyms: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - nucliadb_protos.writer_pb2.OpStatusWriter, - ] - Status: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.WriterStatusRequest, - nucliadb_protos.writer_pb2.WriterStatusResponse, - ] - ListMembers: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.ListMembersRequest, - nucliadb_protos.writer_pb2.ListMembersResponse, - ] - Index: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.IndexResource, - nucliadb_protos.writer_pb2.IndexStatus, - ] - ReIndex: grpc.aio.UnaryUnaryMultiCallable[ - nucliadb_protos.writer_pb2.IndexResource, - nucliadb_protos.writer_pb2.IndexStatus, - ] - DownloadFile: grpc.aio.UnaryStreamMultiCallable[ - nucliadb_protos.writer_pb2.FileRequest, - nucliadb_protos.writer_pb2.BinaryData, - ] - UploadFile: grpc.aio.StreamUnaryMultiCallable[ - nucliadb_protos.writer_pb2.UploadBinaryData, - nucliadb_protos.writer_pb2.FileUploaded, - ] - class WriterServicer(metaclass=abc.ABCMeta): @abc.abstractmethod def NewKnowledgeBox( self, request: nucliadb_protos.knowledgebox_pb2.KnowledgeBoxNew, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.knowledgebox_pb2.NewKnowledgeBoxResponse, collections.abc.Awaitable[nucliadb_protos.knowledgebox_pb2.NewKnowledgeBoxResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.knowledgebox_pb2.NewKnowledgeBoxResponse: ... @abc.abstractmethod def DeleteKnowledgeBox( self, request: nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.knowledgebox_pb2.DeleteKnowledgeBoxResponse, collections.abc.Awaitable[nucliadb_protos.knowledgebox_pb2.DeleteKnowledgeBoxResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.knowledgebox_pb2.DeleteKnowledgeBoxResponse: ... @abc.abstractmethod def UpdateKnowledgeBox( self, request: nucliadb_protos.knowledgebox_pb2.KnowledgeBoxUpdate, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.knowledgebox_pb2.UpdateKnowledgeBoxResponse, collections.abc.Awaitable[nucliadb_protos.knowledgebox_pb2.UpdateKnowledgeBoxResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.knowledgebox_pb2.UpdateKnowledgeBoxResponse: ... @abc.abstractmethod def GCKnowledgeBox( self, request: nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.knowledgebox_pb2.GCKnowledgeBoxResponse, collections.abc.Awaitable[nucliadb_protos.knowledgebox_pb2.GCKnowledgeBoxResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.knowledgebox_pb2.GCKnowledgeBoxResponse: ... @abc.abstractmethod def SetVectors( self, request: nucliadb_protos.writer_pb2.SetVectorsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.SetVectorsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.SetVectorsResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.SetVectorsResponse: ... @abc.abstractmethod def ResourceFieldExists( self, request: nucliadb_protos.writer_pb2.ResourceFieldId, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.ResourceFieldExistsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.ResourceFieldExistsResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.ResourceFieldExistsResponse: ... @abc.abstractmethod def GetResourceId( self, request: nucliadb_protos.writer_pb2.ResourceIdRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.ResourceIdResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.ResourceIdResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.ResourceIdResponse: ... @abc.abstractmethod def ProcessMessage( self, - request_iterator: _MaybeAsyncIterator[nucliadb_protos.writer_pb2.BrokerMessage], - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + request_iterator: collections.abc.Iterator[nucliadb_protos.writer_pb2.BrokerMessage], + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def GetLabels( self, request: nucliadb_protos.writer_pb2.GetLabelsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetLabelsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetLabelsResponse]]: + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetLabelsResponse: """Labels""" @abc.abstractmethod def GetLabelSet( self, request: nucliadb_protos.writer_pb2.GetLabelSetRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetLabelSetResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetLabelSetResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetLabelSetResponse: ... @abc.abstractmethod def SetLabels( self, request: nucliadb_protos.writer_pb2.SetLabelsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def DelLabels( self, request: nucliadb_protos.writer_pb2.DelLabelsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def GetVectorSets( self, request: nucliadb_protos.writer_pb2.GetVectorSetsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetVectorSetsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetVectorSetsResponse]]: + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetVectorSetsResponse: """VectorSets""" @abc.abstractmethod def DelVectorSet( self, request: nucliadb_protos.writer_pb2.DelVectorSetRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def SetVectorSet( self, request: nucliadb_protos.writer_pb2.SetVectorSetRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def NewEntitiesGroup( self, request: nucliadb_protos.writer_pb2.NewEntitiesGroupRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.NewEntitiesGroupResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.NewEntitiesGroupResponse]]: + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.NewEntitiesGroupResponse: """Entities""" @abc.abstractmethod def GetEntities( self, request: nucliadb_protos.writer_pb2.GetEntitiesRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetEntitiesResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetEntitiesResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetEntitiesResponse: ... @abc.abstractmethod def GetEntitiesGroup( self, request: nucliadb_protos.writer_pb2.GetEntitiesGroupRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetEntitiesGroupResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetEntitiesGroupResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetEntitiesGroupResponse: ... @abc.abstractmethod def ListEntitiesGroups( self, request: nucliadb_protos.writer_pb2.ListEntitiesGroupsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.ListEntitiesGroupsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.ListEntitiesGroupsResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.ListEntitiesGroupsResponse: ... @abc.abstractmethod def SetEntities( self, request: nucliadb_protos.writer_pb2.SetEntitiesRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def UpdateEntitiesGroup( self, request: nucliadb_protos.writer_pb2.UpdateEntitiesGroupRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.UpdateEntitiesGroupResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.UpdateEntitiesGroupResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.UpdateEntitiesGroupResponse: ... @abc.abstractmethod def DelEntities( self, request: nucliadb_protos.writer_pb2.DelEntitiesRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def GetSynonyms( self, request: nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.GetSynonymsResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.GetSynonymsResponse]]: + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.GetSynonymsResponse: """Synonyms""" @abc.abstractmethod def SetSynonyms( self, request: nucliadb_protos.writer_pb2.SetSynonymsRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def DelSynonyms( self, request: nucliadb_protos.knowledgebox_pb2.KnowledgeBoxID, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.OpStatusWriter, collections.abc.Awaitable[nucliadb_protos.writer_pb2.OpStatusWriter]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.OpStatusWriter: ... @abc.abstractmethod def Status( self, request: nucliadb_protos.writer_pb2.WriterStatusRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.WriterStatusResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.WriterStatusResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.WriterStatusResponse: ... @abc.abstractmethod def ListMembers( self, request: nucliadb_protos.writer_pb2.ListMembersRequest, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.ListMembersResponse, collections.abc.Awaitable[nucliadb_protos.writer_pb2.ListMembersResponse]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.ListMembersResponse: ... @abc.abstractmethod def Index( self, request: nucliadb_protos.writer_pb2.IndexResource, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.IndexStatus, collections.abc.Awaitable[nucliadb_protos.writer_pb2.IndexStatus]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.IndexStatus: ... @abc.abstractmethod def ReIndex( self, request: nucliadb_protos.writer_pb2.IndexResource, - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.IndexStatus, collections.abc.Awaitable[nucliadb_protos.writer_pb2.IndexStatus]]: ... + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.IndexStatus: ... @abc.abstractmethod def DownloadFile( self, request: nucliadb_protos.writer_pb2.FileRequest, - context: _ServicerContext, - ) -> typing.Union[collections.abc.Iterator[nucliadb_protos.writer_pb2.BinaryData], collections.abc.AsyncIterator[nucliadb_protos.writer_pb2.BinaryData]]: ... + context: grpc.ServicerContext, + ) -> collections.abc.Iterator[nucliadb_protos.writer_pb2.BinaryData]: ... @abc.abstractmethod def UploadFile( self, - request_iterator: _MaybeAsyncIterator[nucliadb_protos.writer_pb2.UploadBinaryData], - context: _ServicerContext, - ) -> typing.Union[nucliadb_protos.writer_pb2.FileUploaded, collections.abc.Awaitable[nucliadb_protos.writer_pb2.FileUploaded]]: ... + request_iterator: collections.abc.Iterator[nucliadb_protos.writer_pb2.UploadBinaryData], + context: grpc.ServicerContext, + ) -> nucliadb_protos.writer_pb2.FileUploaded: ... -def add_WriterServicer_to_server(servicer: WriterServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ... +def add_WriterServicer_to_server(servicer: WriterServicer, server: grpc.Server) -> None: ... diff --git a/nucliadb_protos/rust/src/knowledgebox.rs b/nucliadb_protos/rust/src/knowledgebox.rs index 4541651acf..18f9d91240 100644 --- a/nucliadb_protos/rust/src/knowledgebox.rs +++ b/nucliadb_protos/rust/src/knowledgebox.rs @@ -28,6 +28,8 @@ pub struct KnowledgeBoxConfig { pub disable_vectors: bool, #[prost(int64, tag="7")] pub migration_version: i64, + /// DEPRECATED: duplicated field also stored in `writer.proto Shards` + #[deprecated] #[prost(enumeration="super::utils::ReleaseChannel", tag="8")] pub release_channel: i32, }