nidx-protos 6.9.4.post5356__py3-none-any.whl → 6.10.0.post5792__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of nidx-protos might be problematic. Click here for more details.

@@ -21,13 +21,10 @@ else:
21
21
  from nucliadb_protos.utils_pb2 import (
22
22
  COSINE as COSINE,
23
23
  DOT as DOT,
24
- EXPERIMENTAL as EXPERIMENTAL,
25
24
  ExtractedText as ExtractedText,
26
25
  Relation as Relation,
27
26
  RelationMetadata as RelationMetadata,
28
27
  RelationNode as RelationNode,
29
- ReleaseChannel as ReleaseChannel,
30
- STABLE as STABLE,
31
28
  Security as Security,
32
29
  UserVector as UserVector,
33
30
  UserVectorSet as UserVectorSet,
@@ -120,88 +117,14 @@ global___ShardIds = ShardIds
120
117
  class ShardCreated(google.protobuf.message.Message):
121
118
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
122
119
 
123
- class _DocumentService:
124
- ValueType = typing.NewType("ValueType", builtins.int)
125
- V: typing_extensions.TypeAlias = ValueType
126
-
127
- class _DocumentServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._DocumentService.ValueType], builtins.type):
128
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
129
- DOCUMENT_V0: ShardCreated._DocumentService.ValueType # 0
130
- DOCUMENT_V1: ShardCreated._DocumentService.ValueType # 1
131
- DOCUMENT_V2: ShardCreated._DocumentService.ValueType # 2
132
- DOCUMENT_V3: ShardCreated._DocumentService.ValueType # 3
133
-
134
- class DocumentService(_DocumentService, metaclass=_DocumentServiceEnumTypeWrapper): ...
135
- DOCUMENT_V0: ShardCreated.DocumentService.ValueType # 0
136
- DOCUMENT_V1: ShardCreated.DocumentService.ValueType # 1
137
- DOCUMENT_V2: ShardCreated.DocumentService.ValueType # 2
138
- DOCUMENT_V3: ShardCreated.DocumentService.ValueType # 3
139
-
140
- class _ParagraphService:
141
- ValueType = typing.NewType("ValueType", builtins.int)
142
- V: typing_extensions.TypeAlias = ValueType
143
-
144
- class _ParagraphServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._ParagraphService.ValueType], builtins.type):
145
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
146
- PARAGRAPH_V0: ShardCreated._ParagraphService.ValueType # 0
147
- PARAGRAPH_V1: ShardCreated._ParagraphService.ValueType # 1
148
- PARAGRAPH_V2: ShardCreated._ParagraphService.ValueType # 2
149
- PARAGRAPH_V3: ShardCreated._ParagraphService.ValueType # 3
150
-
151
- class ParagraphService(_ParagraphService, metaclass=_ParagraphServiceEnumTypeWrapper): ...
152
- PARAGRAPH_V0: ShardCreated.ParagraphService.ValueType # 0
153
- PARAGRAPH_V1: ShardCreated.ParagraphService.ValueType # 1
154
- PARAGRAPH_V2: ShardCreated.ParagraphService.ValueType # 2
155
- PARAGRAPH_V3: ShardCreated.ParagraphService.ValueType # 3
156
-
157
- class _VectorService:
158
- ValueType = typing.NewType("ValueType", builtins.int)
159
- V: typing_extensions.TypeAlias = ValueType
160
-
161
- class _VectorServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._VectorService.ValueType], builtins.type):
162
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
163
- VECTOR_V0: ShardCreated._VectorService.ValueType # 0
164
- VECTOR_V1: ShardCreated._VectorService.ValueType # 1
165
-
166
- class VectorService(_VectorService, metaclass=_VectorServiceEnumTypeWrapper): ...
167
- VECTOR_V0: ShardCreated.VectorService.ValueType # 0
168
- VECTOR_V1: ShardCreated.VectorService.ValueType # 1
169
-
170
- class _RelationService:
171
- ValueType = typing.NewType("ValueType", builtins.int)
172
- V: typing_extensions.TypeAlias = ValueType
173
-
174
- class _RelationServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ShardCreated._RelationService.ValueType], builtins.type):
175
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
176
- RELATION_V0: ShardCreated._RelationService.ValueType # 0
177
- RELATION_V1: ShardCreated._RelationService.ValueType # 1
178
- RELATION_V2: ShardCreated._RelationService.ValueType # 2
179
-
180
- class RelationService(_RelationService, metaclass=_RelationServiceEnumTypeWrapper): ...
181
- RELATION_V0: ShardCreated.RelationService.ValueType # 0
182
- RELATION_V1: ShardCreated.RelationService.ValueType # 1
183
- RELATION_V2: ShardCreated.RelationService.ValueType # 2
184
-
185
120
  ID_FIELD_NUMBER: builtins.int
186
- DOCUMENT_SERVICE_FIELD_NUMBER: builtins.int
187
- PARAGRAPH_SERVICE_FIELD_NUMBER: builtins.int
188
- VECTOR_SERVICE_FIELD_NUMBER: builtins.int
189
- RELATION_SERVICE_FIELD_NUMBER: builtins.int
190
121
  id: builtins.str
191
- document_service: global___ShardCreated.DocumentService.ValueType
192
- paragraph_service: global___ShardCreated.ParagraphService.ValueType
193
- vector_service: global___ShardCreated.VectorService.ValueType
194
- relation_service: global___ShardCreated.RelationService.ValueType
195
122
  def __init__(
196
123
  self,
197
124
  *,
198
125
  id: builtins.str = ...,
199
- document_service: global___ShardCreated.DocumentService.ValueType = ...,
200
- paragraph_service: global___ShardCreated.ParagraphService.ValueType = ...,
201
- vector_service: global___ShardCreated.VectorService.ValueType = ...,
202
- relation_service: global___ShardCreated.RelationService.ValueType = ...,
203
126
  ) -> None: ...
204
- def ClearField(self, field_name: typing.Literal["document_service", b"document_service", "id", b"id", "paragraph_service", b"paragraph_service", "relation_service", b"relation_service", "vector_service", b"vector_service"]) -> None: ...
127
+ def ClearField(self, field_name: typing.Literal["id", b"id"]) -> None: ...
205
128
 
206
129
  global___ShardCreated = ShardCreated
207
130
 
@@ -280,7 +203,10 @@ class Position(google.protobuf.message.Message):
280
203
  start: builtins.int
281
204
  end: builtins.int
282
205
  page_number: builtins.int
283
- """For pdfs/documents only"""
206
+ """For pdfs/documents only
207
+ FIXME: this should be an optional, otherwise the Python bindings always
208
+ output a default page 0, false information in most of the cases
209
+ """
284
210
  in_page: builtins.bool
285
211
  @property
286
212
  def start_seconds(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
@@ -694,42 +620,6 @@ class Resource(google.protobuf.message.Message):
694
620
  def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
695
621
  def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
696
622
 
697
- @typing.final
698
- class VectorsEntry(google.protobuf.message.Message):
699
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
700
-
701
- KEY_FIELD_NUMBER: builtins.int
702
- VALUE_FIELD_NUMBER: builtins.int
703
- key: builtins.str
704
- @property
705
- def value(self) -> nucliadb_protos.utils_pb2.UserVectors: ...
706
- def __init__(
707
- self,
708
- *,
709
- key: builtins.str = ...,
710
- value: nucliadb_protos.utils_pb2.UserVectors | None = ...,
711
- ) -> None: ...
712
- def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
713
- def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
714
-
715
- @typing.final
716
- class VectorsToDeleteEntry(google.protobuf.message.Message):
717
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
718
-
719
- KEY_FIELD_NUMBER: builtins.int
720
- VALUE_FIELD_NUMBER: builtins.int
721
- key: builtins.str
722
- @property
723
- def value(self) -> nucliadb_protos.utils_pb2.UserVectorsList: ...
724
- def __init__(
725
- self,
726
- *,
727
- key: builtins.str = ...,
728
- value: nucliadb_protos.utils_pb2.UserVectorsList | None = ...,
729
- ) -> None: ...
730
- def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
731
- def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
732
-
733
623
  RESOURCE_FIELD_NUMBER: builtins.int
734
624
  METADATA_FIELD_NUMBER: builtins.int
735
625
  TEXTS_FIELD_NUMBER: builtins.int
@@ -737,17 +627,17 @@ class Resource(google.protobuf.message.Message):
737
627
  STATUS_FIELD_NUMBER: builtins.int
738
628
  PARAGRAPHS_FIELD_NUMBER: builtins.int
739
629
  PARAGRAPHS_TO_DELETE_FIELD_NUMBER: builtins.int
740
- SENTENCES_TO_DELETE_FIELD_NUMBER: builtins.int
630
+ VECTORS_TO_DELETE_IN_ALL_VECTORSETS_FIELD_NUMBER: builtins.int
741
631
  VECTOR_PREFIXES_TO_DELETE_FIELD_NUMBER: builtins.int
742
632
  RELATION_FIELDS_TO_DELETE_FIELD_NUMBER: builtins.int
743
633
  TEXTS_TO_DELETE_FIELD_NUMBER: builtins.int
744
634
  FIELD_RELATIONS_FIELD_NUMBER: builtins.int
745
635
  SHARD_ID_FIELD_NUMBER: builtins.int
746
- VECTORS_FIELD_NUMBER: builtins.int
747
- VECTORS_TO_DELETE_FIELD_NUMBER: builtins.int
748
636
  SECURITY_FIELD_NUMBER: builtins.int
749
637
  SKIP_TEXTS_FIELD_NUMBER: builtins.int
750
638
  SKIP_PARAGRAPHS_FIELD_NUMBER: builtins.int
639
+ RELATION_NODE_VECTORS_FIELD_NUMBER: builtins.int
640
+ RELATION_EDGE_VECTORS_FIELD_NUMBER: builtins.int
751
641
  status: global___Resource.ResourceStatus.ValueType
752
642
  """Tantivy doc"""
753
643
  shard_id: builtins.str
@@ -780,14 +670,14 @@ class Resource(google.protobuf.message.Message):
780
670
  @property
781
671
  def paragraphs_to_delete(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
782
672
  @property
783
- def sentences_to_delete(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
784
- """DEPRECATED: replaced by vector_prefixes_to_delete in order to support
785
- updates on a subset of the available vectorsets
786
- """
673
+ def vectors_to_delete_in_all_vectorsets(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
674
+ """Delete in all vectorsets (e.g: resource deletion)"""
787
675
 
788
676
  @property
789
677
  def vector_prefixes_to_delete(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___StringList]:
790
- """vectorset_id -> [vector_key_prefix, ...]"""
678
+ """Delete in some vectorsets (e.g: updating some vectors)
679
+ vectorset_id -> [vector_key_prefix, ...]
680
+ """
791
681
 
792
682
  @property
793
683
  def relation_fields_to_delete(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
@@ -798,15 +688,11 @@ class Resource(google.protobuf.message.Message):
798
688
  """Relations"""
799
689
 
800
690
  @property
801
- def vectors(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, nucliadb_protos.utils_pb2.UserVectors]:
802
- """vectorset is the key"""
803
-
691
+ def security(self) -> nucliadb_protos.utils_pb2.Security: ...
804
692
  @property
805
- def vectors_to_delete(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, nucliadb_protos.utils_pb2.UserVectorsList]:
806
- """Vectorset prefix vector id"""
807
-
693
+ def relation_node_vectors(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RelationNodeVector]: ...
808
694
  @property
809
- def security(self) -> nucliadb_protos.utils_pb2.Security: ...
695
+ def relation_edge_vectors(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RelationEdgeVector]: ...
810
696
  def __init__(
811
697
  self,
812
698
  *,
@@ -817,39 +703,79 @@ class Resource(google.protobuf.message.Message):
817
703
  status: global___Resource.ResourceStatus.ValueType = ...,
818
704
  paragraphs: collections.abc.Mapping[builtins.str, global___IndexParagraphs] | None = ...,
819
705
  paragraphs_to_delete: collections.abc.Iterable[builtins.str] | None = ...,
820
- sentences_to_delete: collections.abc.Iterable[builtins.str] | None = ...,
706
+ vectors_to_delete_in_all_vectorsets: collections.abc.Iterable[builtins.str] | None = ...,
821
707
  vector_prefixes_to_delete: collections.abc.Mapping[builtins.str, global___StringList] | None = ...,
822
708
  relation_fields_to_delete: collections.abc.Iterable[builtins.str] | None = ...,
823
709
  texts_to_delete: collections.abc.Iterable[builtins.str] | None = ...,
824
710
  field_relations: collections.abc.Mapping[builtins.str, global___IndexRelations] | None = ...,
825
711
  shard_id: builtins.str = ...,
826
- vectors: collections.abc.Mapping[builtins.str, nucliadb_protos.utils_pb2.UserVectors] | None = ...,
827
- vectors_to_delete: collections.abc.Mapping[builtins.str, nucliadb_protos.utils_pb2.UserVectorsList] | None = ...,
828
712
  security: nucliadb_protos.utils_pb2.Security | None = ...,
829
713
  skip_texts: builtins.bool = ...,
830
714
  skip_paragraphs: builtins.bool = ...,
715
+ relation_node_vectors: collections.abc.Iterable[global___RelationNodeVector] | None = ...,
716
+ relation_edge_vectors: collections.abc.Iterable[global___RelationEdgeVector] | None = ...,
831
717
  ) -> None: ...
832
718
  def HasField(self, field_name: typing.Literal["_security", b"_security", "metadata", b"metadata", "resource", b"resource", "security", b"security"]) -> builtins.bool: ...
833
- def ClearField(self, field_name: typing.Literal["_security", b"_security", "field_relations", b"field_relations", "labels", b"labels", "metadata", b"metadata", "paragraphs", b"paragraphs", "paragraphs_to_delete", b"paragraphs_to_delete", "relation_fields_to_delete", b"relation_fields_to_delete", "resource", b"resource", "security", b"security", "sentences_to_delete", b"sentences_to_delete", "shard_id", b"shard_id", "skip_paragraphs", b"skip_paragraphs", "skip_texts", b"skip_texts", "status", b"status", "texts", b"texts", "texts_to_delete", b"texts_to_delete", "vector_prefixes_to_delete", b"vector_prefixes_to_delete", "vectors", b"vectors", "vectors_to_delete", b"vectors_to_delete"]) -> None: ...
719
+ def ClearField(self, field_name: typing.Literal["_security", b"_security", "field_relations", b"field_relations", "labels", b"labels", "metadata", b"metadata", "paragraphs", b"paragraphs", "paragraphs_to_delete", b"paragraphs_to_delete", "relation_edge_vectors", b"relation_edge_vectors", "relation_fields_to_delete", b"relation_fields_to_delete", "relation_node_vectors", b"relation_node_vectors", "resource", b"resource", "security", b"security", "shard_id", b"shard_id", "skip_paragraphs", b"skip_paragraphs", "skip_texts", b"skip_texts", "status", b"status", "texts", b"texts", "texts_to_delete", b"texts_to_delete", "vector_prefixes_to_delete", b"vector_prefixes_to_delete", "vectors_to_delete_in_all_vectorsets", b"vectors_to_delete_in_all_vectorsets"]) -> None: ...
834
720
  def WhichOneof(self, oneof_group: typing.Literal["_security", b"_security"]) -> typing.Literal["security"] | None: ...
835
721
 
836
722
  global___Resource = Resource
837
723
 
724
+ @typing.final
725
+ class RelationNodeVector(google.protobuf.message.Message):
726
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
727
+
728
+ NODE_FIELD_NUMBER: builtins.int
729
+ VECTOR_FIELD_NUMBER: builtins.int
730
+ @property
731
+ def node(self) -> nucliadb_protos.utils_pb2.RelationNode: ...
732
+ @property
733
+ def vector(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ...
734
+ def __init__(
735
+ self,
736
+ *,
737
+ node: nucliadb_protos.utils_pb2.RelationNode | None = ...,
738
+ vector: collections.abc.Iterable[builtins.float] | None = ...,
739
+ ) -> None: ...
740
+ def HasField(self, field_name: typing.Literal["node", b"node"]) -> builtins.bool: ...
741
+ def ClearField(self, field_name: typing.Literal["node", b"node", "vector", b"vector"]) -> None: ...
742
+
743
+ global___RelationNodeVector = RelationNodeVector
744
+
745
+ @typing.final
746
+ class RelationEdgeVector(google.protobuf.message.Message):
747
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
748
+
749
+ RELATION_TYPE_FIELD_NUMBER: builtins.int
750
+ RELATION_LABEL_FIELD_NUMBER: builtins.int
751
+ VECTOR_FIELD_NUMBER: builtins.int
752
+ relation_type: nucliadb_protos.utils_pb2.Relation.RelationType.ValueType
753
+ relation_label: builtins.str
754
+ @property
755
+ def vector(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ...
756
+ def __init__(
757
+ self,
758
+ *,
759
+ relation_type: nucliadb_protos.utils_pb2.Relation.RelationType.ValueType = ...,
760
+ relation_label: builtins.str = ...,
761
+ vector: collections.abc.Iterable[builtins.float] | None = ...,
762
+ ) -> None: ...
763
+ def ClearField(self, field_name: typing.Literal["relation_label", b"relation_label", "relation_type", b"relation_type", "vector", b"vector"]) -> None: ...
764
+
765
+ global___RelationEdgeVector = RelationEdgeVector
766
+
838
767
  @typing.final
839
768
  class ShardMetadata(google.protobuf.message.Message):
840
769
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
841
770
 
842
771
  KBID_FIELD_NUMBER: builtins.int
843
- RELEASE_CHANNEL_FIELD_NUMBER: builtins.int
844
772
  kbid: builtins.str
845
- release_channel: nucliadb_protos.utils_pb2.ReleaseChannel.ValueType
846
773
  def __init__(
847
774
  self,
848
775
  *,
849
776
  kbid: builtins.str = ...,
850
- release_channel: nucliadb_protos.utils_pb2.ReleaseChannel.ValueType = ...,
851
777
  ) -> None: ...
852
- def ClearField(self, field_name: typing.Literal["kbid", b"kbid", "release_channel", b"release_channel"]) -> None: ...
778
+ def ClearField(self, field_name: typing.Literal["kbid", b"kbid"]) -> None: ...
853
779
 
854
780
  global___ShardMetadata = ShardMetadata
855
781
 
@@ -2,7 +2,7 @@
2
2
  # Generated by the protocol buffer compiler. DO NOT EDIT!
3
3
  # NO CHECKED-IN PROTOBUF GENCODE
4
4
  # source: nidx_protos/nodewriter.proto
5
- # Protobuf Python Version: 5.29.0
5
+ # Protobuf Python Version: 6.31.1
6
6
  """Generated protocol buffer code."""
7
7
  from google.protobuf import descriptor as _descriptor
8
8
  from google.protobuf import descriptor_pool as _descriptor_pool
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
11
11
  from google.protobuf.internal import builder as _builder
12
12
  _runtime_version.ValidateProtobufRuntimeVersion(
13
13
  _runtime_version.Domain.PUBLIC,
14
- 5,
15
- 29,
16
- 0,
14
+ 6,
15
+ 31,
16
+ 1,
17
17
  '',
18
18
  'nidx_protos/nodewriter.proto'
19
19
  )
@@ -30,51 +30,33 @@ except AttributeError:
30
30
 
31
31
  from nidx_protos.noderesources_pb2 import *
32
32
 
33
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cnidx_protos/nodewriter.proto\x12\nnodewriter\x1a\x1fnidx_protos/noderesources.proto\"\xd9\x01\n\x08OpStatus\x12+\n\x06status\x18\x01 \x01(\x0e\x32\x1b.nodewriter.OpStatus.Status\x12\x0e\n\x06\x64\x65tail\x18\x02 \x01(\t\x12\x17\n\x0b\x66ield_count\x18\x03 \x01(\x04\x42\x02\x18\x01\x12\x1b\n\x0fparagraph_count\x18\x05 \x01(\x04\x42\x02\x18\x01\x12\x1a\n\x0esentence_count\x18\x06 \x01(\x04\x42\x02\x18\x01\x12\x14\n\x08shard_id\x18\x04 \x01(\tB\x02\x18\x01\"(\n\x06Status\x12\x06\n\x02OK\x10\x00\x12\x0b\n\x07WARNING\x10\x01\x12\t\n\x05\x45RROR\x10\x02\"\x86\x02\n\x0cIndexMessage\x12\x0c\n\x04node\x18\x01 \x01(\t\x12\r\n\x05shard\x18\x02 \x01(\t\x12\x0c\n\x04txid\x18\x03 \x01(\x04\x12\x10\n\x08resource\x18\x04 \x01(\t\x12,\n\x0btypemessage\x18\x05 \x01(\x0e\x32\x17.nodewriter.TypeMessage\x12\x12\n\nreindex_id\x18\x06 \x01(\t\x12\x16\n\tpartition\x18\x07 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x0bstorage_key\x18\x08 \x01(\t\x12\x0c\n\x04kbid\x18\t \x01(\t\x12.\n\x06source\x18\n \x01(\x0e\x32\x1e.nodewriter.IndexMessageSourceB\x0c\n\n_partition\"\xbc\x01\n\x11VectorIndexConfig\x12+\n\nsimilarity\x18\x01 \x01(\x0e\x32\x17.utils.VectorSimilarity\x12\x19\n\x11normalize_vectors\x18\x02 \x01(\x08\x12+\n\x0bvector_type\x18\x03 \x01(\x0e\x32\x16.nodewriter.VectorType\x12\x1d\n\x10vector_dimension\x18\x04 \x01(\rH\x00\x88\x01\x01\x42\x13\n\x11_vector_dimension\"\x94\x03\n\x0fNewShardRequest\x12/\n\nsimilarity\x18\x01 \x01(\x0e\x32\x17.utils.VectorSimilarityB\x02\x18\x01\x12\x0c\n\x04kbid\x18\x02 \x01(\t\x12.\n\x0frelease_channel\x18\x03 \x01(\x0e\x32\x15.utils.ReleaseChannel\x12\x1d\n\x11normalize_vectors\x18\x04 \x01(\x08\x42\x02\x18\x01\x12\x31\n\x06\x63onfig\x18\x05 \x01(\x0b\x32\x1d.nodewriter.VectorIndexConfigB\x02\x18\x01\x12N\n\x12vectorsets_configs\x18\x06 \x03(\x0b\x32\x32.nodewriter.NewShardRequest.VectorsetsConfigsEntry\x12\x17\n\x0fprewarm_enabled\x18\x07 \x01(\x08\x1aW\n\x16VectorsetsConfigsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x1d.nodewriter.VectorIndexConfig:\x02\x38\x01\"\xbc\x01\n\x13NewVectorSetRequest\x12&\n\x02id\x18\x01 \x01(\x0b\x32\x1a.noderesources.VectorSetID\x12/\n\nsimilarity\x18\x02 \x01(\x0e\x32\x17.utils.VectorSimilarityB\x02\x18\x01\x12\x1d\n\x11normalize_vectors\x18\x03 \x01(\x08\x42\x02\x18\x01\x12-\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x1d.nodewriter.VectorIndexConfig*)\n\x0bTypeMessage\x12\x0c\n\x08\x43REATION\x10\x00\x12\x0c\n\x08\x44\x45LETION\x10\x01*/\n\x12IndexMessageSource\x12\r\n\tPROCESSOR\x10\x00\x12\n\n\x06WRITER\x10\x01*\x1b\n\nVectorType\x12\r\n\tDENSE_F32\x10\x00P\x00\x62\x06proto3')
33
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cnidx_protos/nodewriter.proto\x12\nnodewriter\x1a\x1fnidx_protos/noderesources.proto\"\x89\x01\n\x08OpStatus\x12+\n\x06status\x18\x01 \x01(\x0e\x32\x1b.nodewriter.OpStatus.Status\x12\x0e\n\x06\x64\x65tail\x18\x02 \x01(\t\"(\n\x06Status\x12\x06\n\x02OK\x10\x00\x12\x0b\n\x07WARNING\x10\x01\x12\t\n\x05\x45RROR\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07\"\x86\x02\n\x0cIndexMessage\x12\x0c\n\x04node\x18\x01 \x01(\t\x12\r\n\x05shard\x18\x02 \x01(\t\x12\x0c\n\x04txid\x18\x03 \x01(\x04\x12\x10\n\x08resource\x18\x04 \x01(\t\x12,\n\x0btypemessage\x18\x05 \x01(\x0e\x32\x17.nodewriter.TypeMessage\x12\x12\n\nreindex_id\x18\x06 \x01(\t\x12\x16\n\tpartition\x18\x07 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x0bstorage_key\x18\x08 \x01(\t\x12\x0c\n\x04kbid\x18\t \x01(\t\x12.\n\x06source\x18\n \x01(\x0e\x32\x1e.nodewriter.IndexMessageSourceB\x0c\n\n_partition\"\xbc\x01\n\x11VectorIndexConfig\x12+\n\nsimilarity\x18\x01 \x01(\x0e\x32\x17.utils.VectorSimilarity\x12\x19\n\x11normalize_vectors\x18\x02 \x01(\x08\x12+\n\x0bvector_type\x18\x03 \x01(\x0e\x32\x16.nodewriter.VectorType\x12\x1d\n\x10vector_dimension\x18\x04 \x01(\rH\x00\x88\x01\x01\x42\x13\n\x11_vector_dimension\"\xf9\x01\n\x0fNewShardRequest\x12\x0c\n\x04kbid\x18\x02 \x01(\t\x12N\n\x12vectorsets_configs\x18\x06 \x03(\x0b\x32\x32.nodewriter.NewShardRequest.VectorsetsConfigsEntry\x12\x17\n\x0fprewarm_enabled\x18\x07 \x01(\x08\x1aW\n\x16VectorsetsConfigsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x1d.nodewriter.VectorIndexConfig:\x02\x38\x01J\x04\x08\x01\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06\"x\n\x13NewVectorSetRequest\x12&\n\x02id\x18\x01 \x01(\x0b\x32\x1a.noderesources.VectorSetID\x12-\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x1d.nodewriter.VectorIndexConfigJ\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04*)\n\x0bTypeMessage\x12\x0c\n\x08\x43REATION\x10\x00\x12\x0c\n\x08\x44\x45LETION\x10\x01*/\n\x12IndexMessageSource\x12\r\n\tPROCESSOR\x10\x00\x12\n\n\x06WRITER\x10\x01*\x1b\n\nVectorType\x12\r\n\tDENSE_F32\x10\x00P\x00\x62\x06proto3')
34
34
 
35
35
  _globals = globals()
36
36
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
37
37
  _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'nidx_protos.nodewriter_pb2', _globals)
38
38
  if not _descriptor._USE_C_DESCRIPTORS:
39
39
  DESCRIPTOR._loaded_options = None
40
- _globals['_OPSTATUS'].fields_by_name['field_count']._loaded_options = None
41
- _globals['_OPSTATUS'].fields_by_name['field_count']._serialized_options = b'\030\001'
42
- _globals['_OPSTATUS'].fields_by_name['paragraph_count']._loaded_options = None
43
- _globals['_OPSTATUS'].fields_by_name['paragraph_count']._serialized_options = b'\030\001'
44
- _globals['_OPSTATUS'].fields_by_name['sentence_count']._loaded_options = None
45
- _globals['_OPSTATUS'].fields_by_name['sentence_count']._serialized_options = b'\030\001'
46
- _globals['_OPSTATUS'].fields_by_name['shard_id']._loaded_options = None
47
- _globals['_OPSTATUS'].fields_by_name['shard_id']._serialized_options = b'\030\001'
48
40
  _globals['_NEWSHARDREQUEST_VECTORSETSCONFIGSENTRY']._loaded_options = None
49
41
  _globals['_NEWSHARDREQUEST_VECTORSETSCONFIGSENTRY']._serialized_options = b'8\001'
50
- _globals['_NEWSHARDREQUEST'].fields_by_name['similarity']._loaded_options = None
51
- _globals['_NEWSHARDREQUEST'].fields_by_name['similarity']._serialized_options = b'\030\001'
52
- _globals['_NEWSHARDREQUEST'].fields_by_name['normalize_vectors']._loaded_options = None
53
- _globals['_NEWSHARDREQUEST'].fields_by_name['normalize_vectors']._serialized_options = b'\030\001'
54
- _globals['_NEWSHARDREQUEST'].fields_by_name['config']._loaded_options = None
55
- _globals['_NEWSHARDREQUEST'].fields_by_name['config']._serialized_options = b'\030\001'
56
- _globals['_NEWVECTORSETREQUEST'].fields_by_name['similarity']._loaded_options = None
57
- _globals['_NEWVECTORSETREQUEST'].fields_by_name['similarity']._serialized_options = b'\030\001'
58
- _globals['_NEWVECTORSETREQUEST'].fields_by_name['normalize_vectors']._loaded_options = None
59
- _globals['_NEWVECTORSETREQUEST'].fields_by_name['normalize_vectors']._serialized_options = b'\030\001'
60
- _globals['_TYPEMESSAGE']._serialized_start=1351
61
- _globals['_TYPEMESSAGE']._serialized_end=1392
62
- _globals['_INDEXMESSAGESOURCE']._serialized_start=1394
63
- _globals['_INDEXMESSAGESOURCE']._serialized_end=1441
64
- _globals['_VECTORTYPE']._serialized_start=1443
65
- _globals['_VECTORTYPE']._serialized_end=1470
42
+ _globals['_TYPEMESSAGE']._serialized_start=1047
43
+ _globals['_TYPEMESSAGE']._serialized_end=1088
44
+ _globals['_INDEXMESSAGESOURCE']._serialized_start=1090
45
+ _globals['_INDEXMESSAGESOURCE']._serialized_end=1137
46
+ _globals['_VECTORTYPE']._serialized_start=1139
47
+ _globals['_VECTORTYPE']._serialized_end=1166
66
48
  _globals['_OPSTATUS']._serialized_start=78
67
- _globals['_OPSTATUS']._serialized_end=295
68
- _globals['_OPSTATUS_STATUS']._serialized_start=255
69
- _globals['_OPSTATUS_STATUS']._serialized_end=295
70
- _globals['_INDEXMESSAGE']._serialized_start=298
71
- _globals['_INDEXMESSAGE']._serialized_end=560
72
- _globals['_VECTORINDEXCONFIG']._serialized_start=563
73
- _globals['_VECTORINDEXCONFIG']._serialized_end=751
74
- _globals['_NEWSHARDREQUEST']._serialized_start=754
75
- _globals['_NEWSHARDREQUEST']._serialized_end=1158
76
- _globals['_NEWSHARDREQUEST_VECTORSETSCONFIGSENTRY']._serialized_start=1071
77
- _globals['_NEWSHARDREQUEST_VECTORSETSCONFIGSENTRY']._serialized_end=1158
78
- _globals['_NEWVECTORSETREQUEST']._serialized_start=1161
79
- _globals['_NEWVECTORSETREQUEST']._serialized_end=1349
49
+ _globals['_OPSTATUS']._serialized_end=215
50
+ _globals['_OPSTATUS_STATUS']._serialized_start=151
51
+ _globals['_OPSTATUS_STATUS']._serialized_end=191
52
+ _globals['_INDEXMESSAGE']._serialized_start=218
53
+ _globals['_INDEXMESSAGE']._serialized_end=480
54
+ _globals['_VECTORINDEXCONFIG']._serialized_start=483
55
+ _globals['_VECTORINDEXCONFIG']._serialized_end=671
56
+ _globals['_NEWSHARDREQUEST']._serialized_start=674
57
+ _globals['_NEWSHARDREQUEST']._serialized_end=923
58
+ _globals['_NEWSHARDREQUEST_VECTORSETSCONFIGSENTRY']._serialized_start=812
59
+ _globals['_NEWSHARDREQUEST_VECTORSETSCONFIGSENTRY']._serialized_end=899
60
+ _globals['_NEWVECTORSETREQUEST']._serialized_start=925
61
+ _globals['_NEWVECTORSETREQUEST']._serialized_end=1045
80
62
  # @@protoc_insertion_point(module_scope)
@@ -27,6 +27,8 @@ from nidx_protos.noderesources_pb2 import (
27
27
  IndexRelations as IndexRelations,
28
28
  ParagraphMetadata as ParagraphMetadata,
29
29
  Position as Position,
30
+ RelationEdgeVector as RelationEdgeVector,
31
+ RelationNodeVector as RelationNodeVector,
30
32
  Representation as Representation,
31
33
  Resource as Resource,
32
34
  ResourceID as ResourceID,
@@ -111,27 +113,15 @@ class OpStatus(google.protobuf.message.Message):
111
113
 
112
114
  STATUS_FIELD_NUMBER: builtins.int
113
115
  DETAIL_FIELD_NUMBER: builtins.int
114
- FIELD_COUNT_FIELD_NUMBER: builtins.int
115
- PARAGRAPH_COUNT_FIELD_NUMBER: builtins.int
116
- SENTENCE_COUNT_FIELD_NUMBER: builtins.int
117
- SHARD_ID_FIELD_NUMBER: builtins.int
118
116
  status: global___OpStatus.Status.ValueType
119
117
  detail: builtins.str
120
- field_count: builtins.int
121
- paragraph_count: builtins.int
122
- sentence_count: builtins.int
123
- shard_id: builtins.str
124
118
  def __init__(
125
119
  self,
126
120
  *,
127
121
  status: global___OpStatus.Status.ValueType = ...,
128
122
  detail: builtins.str = ...,
129
- field_count: builtins.int = ...,
130
- paragraph_count: builtins.int = ...,
131
- sentence_count: builtins.int = ...,
132
- shard_id: builtins.str = ...,
133
123
  ) -> None: ...
134
- def ClearField(self, field_name: typing.Literal["detail", b"detail", "field_count", b"field_count", "paragraph_count", b"paragraph_count", "sentence_count", b"sentence_count", "shard_id", b"shard_id", "status", b"status"]) -> None: ...
124
+ def ClearField(self, field_name: typing.Literal["detail", b"detail", "status", b"status"]) -> None: ...
135
125
 
136
126
  global___OpStatus = OpStatus
137
127
 
@@ -228,36 +218,21 @@ class NewShardRequest(google.protobuf.message.Message):
228
218
  def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
229
219
  def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
230
220
 
231
- SIMILARITY_FIELD_NUMBER: builtins.int
232
221
  KBID_FIELD_NUMBER: builtins.int
233
- RELEASE_CHANNEL_FIELD_NUMBER: builtins.int
234
- NORMALIZE_VECTORS_FIELD_NUMBER: builtins.int
235
- CONFIG_FIELD_NUMBER: builtins.int
236
222
  VECTORSETS_CONFIGS_FIELD_NUMBER: builtins.int
237
223
  PREWARM_ENABLED_FIELD_NUMBER: builtins.int
238
- similarity: nucliadb_protos.utils_pb2.VectorSimilarity.ValueType
239
224
  kbid: builtins.str
240
- release_channel: nucliadb_protos.utils_pb2.ReleaseChannel.ValueType
241
- normalize_vectors: builtins.bool
242
- """indicates whether the shard should normalize vectors on indexing or not"""
243
225
  prewarm_enabled: builtins.bool
244
226
  @property
245
- def config(self) -> global___VectorIndexConfig: ...
246
- @property
247
227
  def vectorsets_configs(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___VectorIndexConfig]: ...
248
228
  def __init__(
249
229
  self,
250
230
  *,
251
- similarity: nucliadb_protos.utils_pb2.VectorSimilarity.ValueType = ...,
252
231
  kbid: builtins.str = ...,
253
- release_channel: nucliadb_protos.utils_pb2.ReleaseChannel.ValueType = ...,
254
- normalize_vectors: builtins.bool = ...,
255
- config: global___VectorIndexConfig | None = ...,
256
232
  vectorsets_configs: collections.abc.Mapping[builtins.str, global___VectorIndexConfig] | None = ...,
257
233
  prewarm_enabled: builtins.bool = ...,
258
234
  ) -> None: ...
259
- def HasField(self, field_name: typing.Literal["config", b"config"]) -> builtins.bool: ...
260
- def ClearField(self, field_name: typing.Literal["config", b"config", "kbid", b"kbid", "normalize_vectors", b"normalize_vectors", "prewarm_enabled", b"prewarm_enabled", "release_channel", b"release_channel", "similarity", b"similarity", "vectorsets_configs", b"vectorsets_configs"]) -> None: ...
235
+ def ClearField(self, field_name: typing.Literal["kbid", b"kbid", "prewarm_enabled", b"prewarm_enabled", "vectorsets_configs", b"vectorsets_configs"]) -> None: ...
261
236
 
262
237
  global___NewShardRequest = NewShardRequest
263
238
 
@@ -266,11 +241,7 @@ class NewVectorSetRequest(google.protobuf.message.Message):
266
241
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
267
242
 
268
243
  ID_FIELD_NUMBER: builtins.int
269
- SIMILARITY_FIELD_NUMBER: builtins.int
270
- NORMALIZE_VECTORS_FIELD_NUMBER: builtins.int
271
244
  CONFIG_FIELD_NUMBER: builtins.int
272
- similarity: nucliadb_protos.utils_pb2.VectorSimilarity.ValueType
273
- normalize_vectors: builtins.bool
274
245
  @property
275
246
  def id(self) -> nidx_protos.noderesources_pb2.VectorSetID: ...
276
247
  @property
@@ -279,11 +250,9 @@ class NewVectorSetRequest(google.protobuf.message.Message):
279
250
  self,
280
251
  *,
281
252
  id: nidx_protos.noderesources_pb2.VectorSetID | None = ...,
282
- similarity: nucliadb_protos.utils_pb2.VectorSimilarity.ValueType = ...,
283
- normalize_vectors: builtins.bool = ...,
284
253
  config: global___VectorIndexConfig | None = ...,
285
254
  ) -> None: ...
286
255
  def HasField(self, field_name: typing.Literal["config", b"config", "id", b"id"]) -> builtins.bool: ...
287
- def ClearField(self, field_name: typing.Literal["config", b"config", "id", b"id", "normalize_vectors", b"normalize_vectors", "similarity", b"similarity"]) -> None: ...
256
+ def ClearField(self, field_name: typing.Literal["config", b"config", "id", b"id"]) -> None: ...
288
257
 
289
258
  global___NewVectorSetRequest = NewVectorSetRequest
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nidx_protos
3
- Version: 6.9.4.post5356
3
+ Version: 6.10.0.post5792
4
4
  Summary: Protobuf definitions for nucliadb/nidx
5
5
  Author-Email: Nuclia <nucliadb@nuclia.com>
6
6
  License-Expression: AGPL-3.0-or-later
@@ -14,4 +14,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
14
14
  Project-URL: Homepage, https://nuclia.com
15
15
  Project-URL: Repository, https://github.com/nuclia/nucliadb
16
16
  Requires-Python: <4,>=3.10
17
+ Requires-Dist: protobuf<7.0.0,>=6.31.1
18
+ Provides-Extra: grpc
19
+ Requires-Dist: grpcio>=1.76.0; extra == "grpc"
17
20
 
@@ -0,0 +1,15 @@
1
+ .gitignore,sha256=zbyuFRBda3geYggTx5x-hodA1OnMU85vX8u8Ejh630s,2
2
+ nidx_protos-6.10.0.post5792.dist-info/METADATA,sha256=3i2wNkNH53cVIXcAXC0j0U2fTDp9neO89k-mKAbFLgw,799
3
+ nidx_protos-6.10.0.post5792.dist-info/WHEEL,sha256=tsUv_t7BDeJeRHaSrczbGeuK-TtDpGsWi_JfpzD255I,90
4
+ nidx_protos-6.10.0.post5792.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
5
+ nidx_protos/nidx_pb2.py,sha256=HlgiGIeAE8oPZksy3DD3OWjDcxwhubCLLENuPfLrEqM,5145
6
+ nidx_protos/nidx_pb2.pyi,sha256=PBfwmt-_ox4zu1oURkAnELc8w4WPjSt4yJXL-_1JutI,3368
7
+ nidx_protos/nidx_pb2_grpc.py,sha256=--MhbAbsJLb6hZOyQ4yje4CzK-bbPS5JMg9ecIgeWFI,28353
8
+ nidx_protos/nodereader_pb2.py,sha256=aFrP1lg4ArXMFNyjEhsJLVXDs-gXfZ-jMCgDLIwSpmQ,21702
9
+ nidx_protos/nodereader_pb2.pyi,sha256=sSFc25TEBRUttNXATID5lO-lCQiQluSdVirIY7mTZL4,64659
10
+ nidx_protos/noderesources_pb2.py,sha256=6G4eB0sHVF8PvA8Pxy8dbM_ZvdQgStrn1Z7nzpLkRbI,12772
11
+ nidx_protos/noderesources_pb2.pyi,sha256=RUbHBEp-SxQ2bptVHCVH8Wep3NSF1t9uSrKM8AJ0PFQ,33578
12
+ nidx_protos/nodewriter_pb2.py,sha256=kq_8pkPQzEM68ldytycYj-UftX6nQaHxGJgxL-7xV04,4811
13
+ nidx_protos/nodewriter_pb2.pyi,sha256=MBO_En4GDvJXE9iumXDRrSVOUw1b3tWl4MapdtxOiGs,10288
14
+ nidx_protos/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
+ nidx_protos-6.10.0.post5792.dist-info/RECORD,,
@@ -1,15 +0,0 @@
1
- .gitignore,sha256=zbyuFRBda3geYggTx5x-hodA1OnMU85vX8u8Ejh630s,2
2
- nidx_protos-6.9.4.post5356.dist-info/METADATA,sha256=vWapcvhBk60UHvXa-IiR5tGxyl6PR9B8KagRNMOQ46E,691
3
- nidx_protos-6.9.4.post5356.dist-info/WHEEL,sha256=tsUv_t7BDeJeRHaSrczbGeuK-TtDpGsWi_JfpzD255I,90
4
- nidx_protos-6.9.4.post5356.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
5
- nidx_protos/nidx_pb2.py,sha256=xDsk6acoTEr7emWhcCLD-P59kS85mHtWel4vdsEuHZE,5145
6
- nidx_protos/nidx_pb2.pyi,sha256=PBfwmt-_ox4zu1oURkAnELc8w4WPjSt4yJXL-_1JutI,3368
7
- nidx_protos/nidx_pb2_grpc.py,sha256=sQFtyJFRoTG0D6Oy-7hR75DQD9bK-A4BbpRtzDj-f8o,28354
8
- nidx_protos/nodereader_pb2.py,sha256=1bPETL9rWGYgJEfn6bLnQ2Q45IQzOAy85g3PHxo6lF8,21097
9
- nidx_protos/nodereader_pb2.pyi,sha256=t8YKB0kn0PyrOBi7ZdNER0cs8tEq8KpGRuRTosZvwXE,61374
10
- nidx_protos/noderesources_pb2.py,sha256=Ckxv6d2GvIZi-X01KdaXPvpWz_ISeirE9rG76y0dzsg,14951
11
- nidx_protos/noderesources_pb2.pyi,sha256=FTNXlS_P9RNSb5ayNs7VZl1CKboK0o8tPA6wsYin6MQ,37796
12
- nidx_protos/nodewriter_pb2.py,sha256=9-GAcJvYDD5qyThPMI8lkr437pN5NqV464kBZRz5yzc,6895
13
- nidx_protos/nodewriter_pb2.pyi,sha256=e_Bx6wgskF9Y_o2XRSxDQMlGOh4Sh0G_vpzCJxdMpqk,12193
14
- nidx_protos/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
- nidx_protos-6.9.4.post5356.dist-info/RECORD,,