nucliadb-protos 6.3.7.post4116__py3-none-any.whl → 6.3.7.post4119__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.
- build.py +0 -3
- nucliadb_protos/audit_pb2.py +25 -38
- nucliadb_protos/knowledgebox_pb2.py +91 -97
- nucliadb_protos/knowledgebox_pb2.pyi +41 -20
- nucliadb_protos/train_pb2.py +0 -32
- nucliadb_protos/train_pb2.pyi +3 -1
- nucliadb_protos/train_pb2_grpc.pyi +3 -1
- nucliadb_protos/writer_pb2.py +135 -155
- nucliadb_protos/writer_pb2.pyi +4 -53
- nucliadb_protos/writer_pb2_grpc.pyi +3 -27
- {nucliadb_protos-6.3.7.post4116.dist-info → nucliadb_protos-6.3.7.post4119.dist-info}/METADATA +1 -1
- {nucliadb_protos-6.3.7.post4116.dist-info → nucliadb_protos-6.3.7.post4119.dist-info}/RECORD +14 -24
- nucliadb_protos/nodereader_pb2.py +0 -179
- nucliadb_protos/nodereader_pb2.pyi +0 -1523
- nucliadb_protos/nodereader_pb2_grpc.py +0 -579
- nucliadb_protos/nodereader_pb2_grpc.pyi +0 -288
- nucliadb_protos/noderesources_pb2.py +0 -155
- nucliadb_protos/noderesources_pb2.pyi +0 -993
- nucliadb_protos/nodewriter_pb2.py +0 -90
- nucliadb_protos/nodewriter_pb2.pyi +0 -348
- nucliadb_protos/nodewriter_pb2_grpc.py +0 -571
- nucliadb_protos/nodewriter_pb2_grpc.pyi +0 -256
- {nucliadb_protos-6.3.7.post4116.dist-info → nucliadb_protos-6.3.7.post4119.dist-info}/WHEEL +0 -0
- {nucliadb_protos-6.3.7.post4116.dist-info → nucliadb_protos-6.3.7.post4119.dist-info}/entry_points.txt +0 -0
build.py
CHANGED
|
@@ -30,13 +30,10 @@ def pdm_build_initialize(context):
|
|
|
30
30
|
|
|
31
31
|
# Compile protos
|
|
32
32
|
for proto, has_grpc in [
|
|
33
|
-
("nucliadb_protos/noderesources.proto", False),
|
|
34
33
|
("nucliadb_protos/utils.proto", False),
|
|
35
34
|
("nucliadb_protos/resources.proto", False),
|
|
36
35
|
("nucliadb_protos/knowledgebox.proto", False),
|
|
37
36
|
("nucliadb_protos/audit.proto", False),
|
|
38
|
-
("nucliadb_protos/nodewriter.proto", True),
|
|
39
|
-
("nucliadb_protos/nodereader.proto", True),
|
|
40
37
|
("nucliadb_protos/backups.proto", True),
|
|
41
38
|
("nucliadb_protos/writer.proto", True),
|
|
42
39
|
("nucliadb_protos/train.proto", True),
|
nucliadb_protos/audit_pb2.py
CHANGED
|
@@ -23,19 +23,6 @@ _sym_db = _symbol_database.Default()
|
|
|
23
23
|
|
|
24
24
|
|
|
25
25
|
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
26
|
-
from nucliadb_protos import nodereader_pb2 as nucliadb__protos_dot_nodereader__pb2
|
|
27
|
-
try:
|
|
28
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_nodereader__pb2.nucliadb__protos_dot_noderesources__pb2
|
|
29
|
-
except AttributeError:
|
|
30
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_nodereader__pb2.nucliadb_protos.noderesources_pb2
|
|
31
|
-
try:
|
|
32
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_nodereader__pb2.nucliadb__protos_dot_utils__pb2
|
|
33
|
-
except AttributeError:
|
|
34
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_nodereader__pb2.nucliadb_protos.utils_pb2
|
|
35
|
-
try:
|
|
36
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_nodereader__pb2.nucliadb__protos_dot_utils__pb2
|
|
37
|
-
except AttributeError:
|
|
38
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_nodereader__pb2.nucliadb_protos.utils_pb2
|
|
39
26
|
from nucliadb_protos import resources_pb2 as nucliadb__protos_dot_resources__pb2
|
|
40
27
|
try:
|
|
41
28
|
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_resources__pb2.nucliadb__protos_dot_utils__pb2
|
|
@@ -43,7 +30,7 @@ except AttributeError:
|
|
|
43
30
|
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_resources__pb2.nucliadb_protos.utils_pb2
|
|
44
31
|
|
|
45
32
|
|
|
46
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bnucliadb_protos/audit.proto\x12\x05\x61udit\x1a\x1fgoogle/protobuf/timestamp.proto\x1a
|
|
33
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bnucliadb_protos/audit.proto\x12\x05\x61udit\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1fnucliadb_protos/resources.proto\"\xe4\x01\n\nAuditField\x12-\n\x06\x61\x63tion\x18\x01 \x01(\x0e\x32\x1d.audit.AuditField.FieldAction\x12\x0c\n\x04size\x18\x02 \x01(\x05\x12\x16\n\nsize_delta\x18\x03 \x01(\x05\x42\x02\x18\x01\x12\x10\n\x08\x66ield_id\x18\x04 \x01(\t\x12(\n\nfield_type\x18\x05 \x01(\x0e\x32\x14.resources.FieldType\x12\x10\n\x08\x66ilename\x18\x06 \x01(\t\"3\n\x0b\x46ieldAction\x12\t\n\x05\x41\x44\x44\x45\x44\x10\x00\x12\x0c\n\x08MODIFIED\x10\x01\x12\x0b\n\x07\x44\x45LETED\x10\x02\"4\n\x0e\x41uditKBCounter\x12\x12\n\nparagraphs\x18\x02 \x01(\x03\x12\x0e\n\x06\x66ields\x18\x03 \x01(\x03\"+\n\x0b\x43hatContext\x12\x0e\n\x06\x61uthor\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\"7\n\x10RetrievedContext\x12\x15\n\rtext_block_id\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\"\xb6\x02\n\tChatAudit\x12\x10\n\x08question\x18\x01 \x01(\t\x12\x13\n\x06\x61nswer\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x1f\n\x12rephrased_question\x18\x03 \x01(\tH\x01\x88\x01\x01\x12(\n\x0c\x63hat_context\x18\x06 \x03(\x0b\x32\x12.audit.ChatContext\x12\x32\n\x11retrieved_context\x18\x08 \x03(\x0b\x32\x17.audit.RetrievedContext\x12\x18\n\x0blearning_id\x18\x05 \x01(\tH\x02\x88\x01\x01\x12\x13\n\x0bstatus_code\x18\t \x01(\x05\x12\x12\n\x05model\x18\n \x01(\tH\x03\x88\x01\x01\x42\t\n\x07_answerB\x15\n\x13_rephrased_questionB\x0e\n\x0c_learning_idB\x08\n\x06_modelJ\x04\x08\x04\x10\x05\"\xa3\x01\n\rFeedbackAudit\x12\x13\n\x0blearning_id\x18\x01 \x01(\t\x12\x0c\n\x04good\x18\x02 \x01(\x08\x12\x1d\n\x04task\x18\x03 \x01(\x0e\x32\x0f.audit.TaskType\x12\x15\n\x08\x66\x65\x65\x64\x62\x61\x63k\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x1a\n\rtext_block_id\x18\x05 \x01(\tH\x01\x88\x01\x01\x42\x0b\n\t_feedbackB\x10\n\x0e_text_block_id\"\x97\t\n\x0c\x41uditRequest\x12+\n\x04type\x18\x01 \x01(\x0e\x32\x1d.audit.AuditRequest.AuditType\x12\x0c\n\x04kbid\x18\x02 \x01(\t\x12\x0e\n\x06userid\x18\x04 \x01(\t\x12(\n\x04time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0e\n\x06\x66ields\x18\x06 \x03(\t\x12)\n\x06search\x18\x07 \x01(\x0b\x32\x19.audit.AuditSearchRequest\x12\x12\n\x06timeit\x18\x08 \x01(\x02\x42\x02\x18\x01\x12\x0e\n\x06origin\x18\t \x01(\t\x12\x0b\n\x03rid\x18\n \x01(\t\x12\x0c\n\x04task\x18\x0b \x01(\t\x12\x11\n\tresources\x18\x0c \x01(\x05\x12*\n\x0e\x66ield_metadata\x18\r \x03(\x0b\x32\x12.resources.FieldID\x12\'\n\x0c\x66ields_audit\x18\x0e \x03(\x0b\x32\x11.audit.AuditField\x12&\n\x0b\x63lient_type\x18\x10 \x01(\x0e\x32\x11.audit.ClientType\x12\x10\n\x08trace_id\x18\x11 \x01(\t\x12)\n\nkb_counter\x18\x12 \x01(\x0b\x32\x15.audit.AuditKBCounter\x12\x1e\n\x04\x63hat\x18\x13 \x01(\x0b\x32\x10.audit.ChatAudit\x12\x0f\n\x07success\x18\x14 \x01(\x08\x12\x14\n\x0crequest_time\x18\x15 \x01(\x02\x12\x1b\n\x0eretrieval_time\x18\x16 \x01(\x02H\x00\x88\x01\x01\x12#\n\x16generative_answer_time\x18\x17 \x01(\x02H\x01\x88\x01\x01\x12/\n\"generative_answer_first_chunk_time\x18\x18 \x01(\x02H\x02\x88\x01\x01\x12\x1a\n\rrephrase_time\x18\x19 \x01(\x02H\x03\x88\x01\x01\x12&\n\x08\x66\x65\x65\x64\x62\x61\x63k\x18\x1a \x01(\x0b\x32\x14.audit.FeedbackAudit\x12\x19\n\x0cuser_request\x18\x1b \x01(\tH\x04\x88\x01\x01\x12\x1a\n\rprocessing_id\x18\x1c \x01(\tH\x05\x88\x01\x01\x12)\n\x1cretrieval_rephrased_question\x18\x1d \x01(\tH\x06\x88\x01\x01\"\xbf\x01\n\tAuditType\x12\x0b\n\x07VISITED\x10\x00\x12\x0c\n\x08MODIFIED\x10\x01\x12\x0b\n\x07\x44\x45LETED\x10\x02\x12\x07\n\x03NEW\x10\x03\x12\x0b\n\x07STARTED\x10\x04\x12\x0b\n\x07STOPPED\x10\x05\x12\n\n\x06SEARCH\x10\x06\x12\r\n\tPROCESSED\x10\x07\x12\x12\n\nKB_DELETED\x10\x08\x1a\x02\x08\x01\x12\x0f\n\x07SUGGEST\x10\t\x1a\x02\x08\x01\x12\x0f\n\x07INDEXED\x10\n\x1a\x02\x08\x01\x12\x08\n\x04\x43HAT\x10\x0b\x12\x0c\n\x08\x46\x45\x45\x44\x42\x41\x43K\x10\x0c\x42\x11\n\x0f_retrieval_timeB\x19\n\x17_generative_answer_timeB%\n#_generative_answer_first_chunk_timeB\x10\n\x0e_rephrase_timeB\x0f\n\r_user_requestB\x10\n\x0e_processing_idB\x1f\n\x1d_retrieval_rephrased_question\"\xc1\x02\n\x12\x41uditSearchRequest\x12\x0c\n\x04\x62ody\x18\x03 \x01(\t\x12\x17\n\x0fresult_per_page\x18\x08 \x01(\x05\x12\x0e\n\x06vector\x18\n \x03(\x02\x12\x11\n\tvectorset\x18\x0f \x01(\t\x12\x1a\n\x12min_score_semantic\x18\x17 \x01(\x02\x12\x16\n\x0emin_score_bm25\x18\x19 \x01(\x02\x12&\n\x08security\x18\x18 \x01(\x0b\x32\x0f.utils.SecurityH\x00\x88\x01\x01\x12\x13\n\x06\x66ilter\x18\x1e \x01(\tH\x01\x88\x01\x01\x42\x0b\n\t_securityB\t\n\x07_filterJ\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x0c\x10\rJ\x04\x08\r\x10\x0eJ\x04\x08\x0e\x10\x0fJ\x04\x08\x10\x10\x11J\x04\x08\x12\x10\x13J\x04\x08\x14\x10\x15J\x04\x08\x15\x10\x16J\x04\x08\x1a\x10\x1bJ\x04\x08\x1b\x10\x1cJ\x04\x08\x1c\x10\x1dJ\x04\x08\x1d\x10\x1e*\\\n\nClientType\x12\x07\n\x03\x41PI\x10\x00\x12\x07\n\x03WEB\x10\x01\x12\n\n\x06WIDGET\x10\x02\x12\x0b\n\x07\x44\x45SKTOP\x10\x03\x12\r\n\tDASHBOARD\x10\x04\x12\x14\n\x10\x43HROME_EXTENSION\x10\x05*\x14\n\x08TaskType\x12\x08\n\x04\x43HAT\x10\x00\x62\x06proto3')
|
|
47
34
|
|
|
48
35
|
_globals = globals()
|
|
49
36
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
@@ -60,28 +47,28 @@ if not _descriptor._USE_C_DESCRIPTORS:
|
|
|
60
47
|
_globals['_AUDITREQUEST_AUDITTYPE'].values_by_name["INDEXED"]._serialized_options = b'\010\001'
|
|
61
48
|
_globals['_AUDITREQUEST'].fields_by_name['timeit']._loaded_options = None
|
|
62
49
|
_globals['_AUDITREQUEST'].fields_by_name['timeit']._serialized_options = b'\030\001'
|
|
63
|
-
_globals['_CLIENTTYPE']._serialized_start=
|
|
64
|
-
_globals['_CLIENTTYPE']._serialized_end=
|
|
65
|
-
_globals['_TASKTYPE']._serialized_start=
|
|
66
|
-
_globals['_TASKTYPE']._serialized_end=
|
|
67
|
-
_globals['_AUDITFIELD']._serialized_start=
|
|
68
|
-
_globals['_AUDITFIELD']._serialized_end=
|
|
69
|
-
_globals['_AUDITFIELD_FIELDACTION']._serialized_start=
|
|
70
|
-
_globals['_AUDITFIELD_FIELDACTION']._serialized_end=
|
|
71
|
-
_globals['_AUDITKBCOUNTER']._serialized_start=
|
|
72
|
-
_globals['_AUDITKBCOUNTER']._serialized_end=
|
|
73
|
-
_globals['_CHATCONTEXT']._serialized_start=
|
|
74
|
-
_globals['_CHATCONTEXT']._serialized_end=
|
|
75
|
-
_globals['_RETRIEVEDCONTEXT']._serialized_start=
|
|
76
|
-
_globals['_RETRIEVEDCONTEXT']._serialized_end=
|
|
77
|
-
_globals['_CHATAUDIT']._serialized_start=
|
|
78
|
-
_globals['_CHATAUDIT']._serialized_end=
|
|
79
|
-
_globals['_FEEDBACKAUDIT']._serialized_start=
|
|
80
|
-
_globals['_FEEDBACKAUDIT']._serialized_end=
|
|
81
|
-
_globals['_AUDITREQUEST']._serialized_start=
|
|
82
|
-
_globals['_AUDITREQUEST']._serialized_end=
|
|
83
|
-
_globals['_AUDITREQUEST_AUDITTYPE']._serialized_start=
|
|
84
|
-
_globals['_AUDITREQUEST_AUDITTYPE']._serialized_end=
|
|
85
|
-
_globals['_AUDITSEARCHREQUEST']._serialized_start=
|
|
86
|
-
_globals['_AUDITSEARCHREQUEST']._serialized_end=
|
|
50
|
+
_globals['_CLIENTTYPE']._serialized_start=2472
|
|
51
|
+
_globals['_CLIENTTYPE']._serialized_end=2564
|
|
52
|
+
_globals['_TASKTYPE']._serialized_start=2566
|
|
53
|
+
_globals['_TASKTYPE']._serialized_end=2586
|
|
54
|
+
_globals['_AUDITFIELD']._serialized_start=105
|
|
55
|
+
_globals['_AUDITFIELD']._serialized_end=333
|
|
56
|
+
_globals['_AUDITFIELD_FIELDACTION']._serialized_start=282
|
|
57
|
+
_globals['_AUDITFIELD_FIELDACTION']._serialized_end=333
|
|
58
|
+
_globals['_AUDITKBCOUNTER']._serialized_start=335
|
|
59
|
+
_globals['_AUDITKBCOUNTER']._serialized_end=387
|
|
60
|
+
_globals['_CHATCONTEXT']._serialized_start=389
|
|
61
|
+
_globals['_CHATCONTEXT']._serialized_end=432
|
|
62
|
+
_globals['_RETRIEVEDCONTEXT']._serialized_start=434
|
|
63
|
+
_globals['_RETRIEVEDCONTEXT']._serialized_end=489
|
|
64
|
+
_globals['_CHATAUDIT']._serialized_start=492
|
|
65
|
+
_globals['_CHATAUDIT']._serialized_end=802
|
|
66
|
+
_globals['_FEEDBACKAUDIT']._serialized_start=805
|
|
67
|
+
_globals['_FEEDBACKAUDIT']._serialized_end=968
|
|
68
|
+
_globals['_AUDITREQUEST']._serialized_start=971
|
|
69
|
+
_globals['_AUDITREQUEST']._serialized_end=2146
|
|
70
|
+
_globals['_AUDITREQUEST_AUDITTYPE']._serialized_start=1784
|
|
71
|
+
_globals['_AUDITREQUEST_AUDITTYPE']._serialized_end=1975
|
|
72
|
+
_globals['_AUDITSEARCHREQUEST']._serialized_start=2149
|
|
73
|
+
_globals['_AUDITSEARCHREQUEST']._serialized_end=2470
|
|
87
74
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -23,20 +23,10 @@ _sym_db = _symbol_database.Default()
|
|
|
23
23
|
|
|
24
24
|
|
|
25
25
|
from nucliadb_protos import utils_pb2 as nucliadb__protos_dot_utils__pb2
|
|
26
|
-
from nucliadb_protos import nodewriter_pb2 as nucliadb__protos_dot_nodewriter__pb2
|
|
27
|
-
try:
|
|
28
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_nodewriter__pb2.nucliadb__protos_dot_noderesources__pb2
|
|
29
|
-
except AttributeError:
|
|
30
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_nodewriter__pb2.nucliadb_protos.noderesources_pb2
|
|
31
|
-
try:
|
|
32
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_nodewriter__pb2.nucliadb__protos_dot_utils__pb2
|
|
33
|
-
except AttributeError:
|
|
34
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_nodewriter__pb2.nucliadb_protos.utils_pb2
|
|
35
26
|
|
|
36
27
|
from nucliadb_protos.utils_pb2 import *
|
|
37
|
-
from nucliadb_protos.nodewriter_pb2 import *
|
|
38
28
|
|
|
39
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"nucliadb_protos/knowledgebox.proto\x12\x0cknowledgebox\x1a\x1bnucliadb_protos/utils.proto\
|
|
29
|
+
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\"h\n\x14\x43reatePineconeConfig\x12\x0f\n\x07\x61pi_key\x18\x01 \x01(\t\x12?\n\x10serverless_cloud\x18\x02 \x01(\x0e\x32%.knowledgebox.PineconeServerlessCloud\"\x86\x01\n\x15PineconeIndexMetadata\x12\x12\n\nindex_name\x18\x01 \x01(\t\x12\x12\n\nindex_host\x18\x02 \x01(\t\x12\x18\n\x10vector_dimension\x18\x03 \x01(\x05\x12+\n\nsimilarity\x18\x04 \x01(\x0e\x32\x17.utils.VectorSimilarity\"\x89\x02\n\x14StoredPineconeConfig\x12\x19\n\x11\x65ncrypted_api_key\x18\x01 \x01(\t\x12@\n\x07indexes\x18\x02 \x03(\x0b\x32/.knowledgebox.StoredPineconeConfig.IndexesEntry\x12?\n\x10serverless_cloud\x18\x03 \x01(\x0e\x32%.knowledgebox.PineconeServerlessCloud\x1aS\n\x0cIndexesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x32\n\x05value\x18\x02 \x01(\x0b\x32#.knowledgebox.PineconeIndexMetadata:\x02\x38\x01\"\xa5\x01\n#CreateExternalIndexProviderMetadata\x12\x35\n\x04type\x18\x01 \x01(\x0e\x32\'.knowledgebox.ExternalIndexProviderType\x12=\n\x0fpinecone_config\x18\x02 \x01(\x0b\x32\".knowledgebox.CreatePineconeConfigH\x00\x42\x08\n\x06\x63onfig\"\xa5\x01\n#StoredExternalIndexProviderMetadata\x12\x35\n\x04type\x18\x01 \x01(\x0e\x32\'.knowledgebox.ExternalIndexProviderType\x12=\n\x0fpinecone_config\x18\x02 \x01(\x0b\x32\".knowledgebox.StoredPineconeConfigH\x00\x42\x08\n\x06\x63onfig\"\x8e\x03\n\x12KnowledgeBoxConfig\x12\r\n\x05title\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0c\n\x04slug\x18\x05 \x01(\t\x12\x19\n\x11migration_version\x18\x07 \x01(\x03\x12R\n\x17\x65xternal_index_provider\x18\t \x01(\x0b\x32\x31.knowledgebox.StoredExternalIndexProviderMetadata\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\x1b\n\x0f\x64isable_vectors\x18\x06 \x01(\x08\x42\x02\x18\x01\x12\x32\n\x0frelease_channel\x18\x08 \x01(\x0e\x32\x15.utils.ReleaseChannelB\x02\x18\x01\x12 \n\x18hidden_resources_enabled\x18\n \x01(\x08\x12)\n!hidden_resources_hide_on_creation\x18\x0b \x01(\x08\"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\"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\"\xbe\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\x18.knowledgebox.VectorType\x12\x1d\n\x10vector_dimension\x18\x04 \x01(\rH\x00\x88\x01\x01\x42\x13\n\x11_vector_dimension\"\x8e\x02\n\x0fVectorSetConfig\x12\x14\n\x0cvectorset_id\x18\x01 \x01(\t\x12?\n\x16vectorset_index_config\x18\x02 \x01(\x0b\x32\x1f.knowledgebox.VectorIndexConfig\x12\x1d\n\x15matryoshka_dimensions\x18\x03 \x03(\r\x12\x46\n\x10storage_key_kind\x18\x04 \x01(\x0e\x32,.knowledgebox.VectorSetConfig.StorageKeyKind\"=\n\x0eStorageKeyKind\x12\t\n\x05UNSET\x10\x00\x12\n\n\x06LEGACY\x10\x01\x12\x14\n\x10VECTORSET_PREFIX\x10\x02\"Q\n\x1cKnowledgeBoxVectorSetsConfig\x12\x31\n\nvectorsets\x18\x01 \x03(\x0b\x32\x1d.knowledgebox.VectorSetConfig\"X\n\x0eVectorSetPurge\x12\x46\n\x10storage_key_kind\x18\x01 \x01(\x0e\x32,.knowledgebox.VectorSetConfig.StorageKeyKind\" \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\"\xda\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\"\n\x11\x64\x65\x66\x61ult_min_score\x18\x03 \x01(\x02\x42\x02\x18\x01H\x01\x88\x01\x01\x12\x1d\n\x15matryoshka_dimensions\x18\x04 \x03(\rB\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*n\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\x03\x12!\n\x1d\x45XTERNAL_INDEX_PROVIDER_ERROR\x10\x04*4\n\x19\x45xternalIndexProviderType\x12\t\n\x05UNSET\x10\x00\x12\x0c\n\x08PINECONE\x10\x01*\x8e\x01\n\x17PineconeServerlessCloud\x12\x12\n\x0ePINECONE_UNSET\x10\x00\x12\x11\n\rAWS_US_EAST_1\x10\x01\x12\x11\n\rAWS_US_WEST_2\x10\x02\x12\x11\n\rAWS_EU_WEST_1\x10\x03\x12\x13\n\x0fGCP_US_CENTRAL1\x10\x04\x12\x11\n\rAZURE_EASTUS2\x10\x05*\x1b\n\nVectorType\x12\r\n\tDENSE_F32\x10\x00P\x00\x62\x06proto3')
|
|
40
30
|
|
|
41
31
|
_globals = globals()
|
|
42
32
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
@@ -69,90 +59,94 @@ if not _descriptor._USE_C_DESCRIPTORS:
|
|
|
69
59
|
_globals['_SYNONYMS_TERMSENTRY']._serialized_options = b'8\001'
|
|
70
60
|
_globals['_SEMANTICMODELMETADATA'].fields_by_name['default_min_score']._loaded_options = None
|
|
71
61
|
_globals['_SEMANTICMODELMETADATA'].fields_by_name['default_min_score']._serialized_options = b'\030\001'
|
|
72
|
-
_globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_start=
|
|
73
|
-
_globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_end=
|
|
74
|
-
_globals['_EXTERNALINDEXPROVIDERTYPE']._serialized_start=
|
|
75
|
-
_globals['_EXTERNALINDEXPROVIDERTYPE']._serialized_end=
|
|
76
|
-
_globals['_PINECONESERVERLESSCLOUD']._serialized_start=
|
|
77
|
-
_globals['_PINECONESERVERLESSCLOUD']._serialized_end=
|
|
78
|
-
_globals['
|
|
79
|
-
_globals['
|
|
80
|
-
_globals['
|
|
81
|
-
_globals['
|
|
82
|
-
_globals['
|
|
83
|
-
_globals['
|
|
84
|
-
_globals['
|
|
85
|
-
_globals['
|
|
86
|
-
_globals['
|
|
87
|
-
_globals['
|
|
88
|
-
_globals['
|
|
89
|
-
_globals['
|
|
90
|
-
_globals['
|
|
91
|
-
_globals['
|
|
92
|
-
_globals['
|
|
93
|
-
_globals['
|
|
94
|
-
_globals['
|
|
95
|
-
_globals['
|
|
96
|
-
_globals['
|
|
97
|
-
_globals['
|
|
98
|
-
_globals['
|
|
99
|
-
_globals['
|
|
100
|
-
_globals['
|
|
101
|
-
_globals['
|
|
102
|
-
_globals['
|
|
103
|
-
_globals['
|
|
104
|
-
_globals['
|
|
105
|
-
_globals['
|
|
106
|
-
_globals['
|
|
107
|
-
_globals['
|
|
108
|
-
_globals['
|
|
109
|
-
_globals['
|
|
110
|
-
_globals['
|
|
111
|
-
_globals['
|
|
112
|
-
_globals['
|
|
113
|
-
_globals['
|
|
114
|
-
_globals['
|
|
115
|
-
_globals['
|
|
116
|
-
_globals['
|
|
117
|
-
_globals['
|
|
118
|
-
_globals['
|
|
119
|
-
_globals['
|
|
120
|
-
_globals['
|
|
121
|
-
_globals['
|
|
122
|
-
_globals['
|
|
123
|
-
_globals['
|
|
124
|
-
_globals['
|
|
125
|
-
_globals['
|
|
126
|
-
_globals['
|
|
127
|
-
_globals['
|
|
128
|
-
_globals['
|
|
129
|
-
_globals['
|
|
130
|
-
_globals['
|
|
131
|
-
_globals['
|
|
132
|
-
_globals['
|
|
133
|
-
_globals['
|
|
134
|
-
_globals['
|
|
135
|
-
_globals['
|
|
136
|
-
_globals['
|
|
137
|
-
_globals['
|
|
138
|
-
_globals['
|
|
139
|
-
_globals['
|
|
140
|
-
_globals['
|
|
141
|
-
_globals['
|
|
142
|
-
_globals['
|
|
143
|
-
_globals['
|
|
144
|
-
_globals['
|
|
145
|
-
_globals['
|
|
146
|
-
_globals['
|
|
147
|
-
_globals['
|
|
148
|
-
_globals['
|
|
149
|
-
_globals['
|
|
150
|
-
_globals['
|
|
151
|
-
_globals['
|
|
152
|
-
_globals['
|
|
153
|
-
_globals['
|
|
154
|
-
_globals['
|
|
155
|
-
_globals['
|
|
156
|
-
_globals['
|
|
157
|
-
_globals['
|
|
62
|
+
_globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_start=4576
|
|
63
|
+
_globals['_KNOWLEDGEBOXRESPONSESTATUS']._serialized_end=4686
|
|
64
|
+
_globals['_EXTERNALINDEXPROVIDERTYPE']._serialized_start=4688
|
|
65
|
+
_globals['_EXTERNALINDEXPROVIDERTYPE']._serialized_end=4740
|
|
66
|
+
_globals['_PINECONESERVERLESSCLOUD']._serialized_start=4743
|
|
67
|
+
_globals['_PINECONESERVERLESSCLOUD']._serialized_end=4885
|
|
68
|
+
_globals['_VECTORTYPE']._serialized_start=4887
|
|
69
|
+
_globals['_VECTORTYPE']._serialized_end=4914
|
|
70
|
+
_globals['_KNOWLEDGEBOXID']._serialized_start=81
|
|
71
|
+
_globals['_KNOWLEDGEBOXID']._serialized_end=125
|
|
72
|
+
_globals['_CREATEPINECONECONFIG']._serialized_start=127
|
|
73
|
+
_globals['_CREATEPINECONECONFIG']._serialized_end=231
|
|
74
|
+
_globals['_PINECONEINDEXMETADATA']._serialized_start=234
|
|
75
|
+
_globals['_PINECONEINDEXMETADATA']._serialized_end=368
|
|
76
|
+
_globals['_STOREDPINECONECONFIG']._serialized_start=371
|
|
77
|
+
_globals['_STOREDPINECONECONFIG']._serialized_end=636
|
|
78
|
+
_globals['_STOREDPINECONECONFIG_INDEXESENTRY']._serialized_start=553
|
|
79
|
+
_globals['_STOREDPINECONECONFIG_INDEXESENTRY']._serialized_end=636
|
|
80
|
+
_globals['_CREATEEXTERNALINDEXPROVIDERMETADATA']._serialized_start=639
|
|
81
|
+
_globals['_CREATEEXTERNALINDEXPROVIDERMETADATA']._serialized_end=804
|
|
82
|
+
_globals['_STOREDEXTERNALINDEXPROVIDERMETADATA']._serialized_start=807
|
|
83
|
+
_globals['_STOREDEXTERNALINDEXPROVIDERMETADATA']._serialized_end=972
|
|
84
|
+
_globals['_KNOWLEDGEBOXCONFIG']._serialized_start=975
|
|
85
|
+
_globals['_KNOWLEDGEBOXCONFIG']._serialized_end=1373
|
|
86
|
+
_globals['_KNOWLEDGEBOXUPDATE']._serialized_start=1375
|
|
87
|
+
_globals['_KNOWLEDGEBOXUPDATE']._serialized_end=1473
|
|
88
|
+
_globals['_UPDATEKNOWLEDGEBOXRESPONSE']._serialized_start=1475
|
|
89
|
+
_globals['_UPDATEKNOWLEDGEBOXRESPONSE']._serialized_end=1575
|
|
90
|
+
_globals['_DELETEKNOWLEDGEBOXRESPONSE']._serialized_start=1577
|
|
91
|
+
_globals['_DELETEKNOWLEDGEBOXRESPONSE']._serialized_end=1663
|
|
92
|
+
_globals['_LABEL']._serialized_start=1665
|
|
93
|
+
_globals['_LABEL']._serialized_end=1731
|
|
94
|
+
_globals['_LABELSET']._serialized_start=1734
|
|
95
|
+
_globals['_LABELSET']._serialized_end=1958
|
|
96
|
+
_globals['_LABELSET_LABELSETKIND']._serialized_start=1882
|
|
97
|
+
_globals['_LABELSET_LABELSETKIND']._serialized_end=1958
|
|
98
|
+
_globals['_LABELS']._serialized_start=1961
|
|
99
|
+
_globals['_LABELS']._serialized_end=2096
|
|
100
|
+
_globals['_LABELS_LABELSETENTRY']._serialized_start=2025
|
|
101
|
+
_globals['_LABELS_LABELSETENTRY']._serialized_end=2096
|
|
102
|
+
_globals['_ENTITY']._serialized_start=2098
|
|
103
|
+
_globals['_ENTITY']._serialized_end=2174
|
|
104
|
+
_globals['_ENTITIESGROUPSUMMARY']._serialized_start=2176
|
|
105
|
+
_globals['_ENTITIESGROUPSUMMARY']._serialized_end=2244
|
|
106
|
+
_globals['_ENTITIESGROUP']._serialized_start=2247
|
|
107
|
+
_globals['_ENTITIESGROUP']._serialized_end=2440
|
|
108
|
+
_globals['_ENTITIESGROUP_ENTITIESENTRY']._serialized_start=2371
|
|
109
|
+
_globals['_ENTITIESGROUP_ENTITIESENTRY']._serialized_end=2440
|
|
110
|
+
_globals['_DELETEDENTITIESGROUPS']._serialized_start=2442
|
|
111
|
+
_globals['_DELETEDENTITIESGROUPS']._serialized_end=2490
|
|
112
|
+
_globals['_ENTITIESGROUPS']._serialized_start=2493
|
|
113
|
+
_globals['_ENTITIESGROUPS']._serialized_end=2668
|
|
114
|
+
_globals['_ENTITIESGROUPS_ENTITIESGROUPSENTRY']._serialized_start=2586
|
|
115
|
+
_globals['_ENTITIESGROUPS_ENTITIESGROUPSENTRY']._serialized_end=2668
|
|
116
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX']._serialized_start=2671
|
|
117
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX']._serialized_end=3170
|
|
118
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYDUPLICATES']._serialized_start=2786
|
|
119
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYDUPLICATES']._serialized_end=2824
|
|
120
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES']._serialized_start=2827
|
|
121
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES']._serialized_end=3052
|
|
122
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES_ENTITIESENTRY']._serialized_start=2947
|
|
123
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITYGROUPDUPLICATES_ENTITIESENTRY']._serialized_end=3052
|
|
124
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITIESGROUPSENTRY']._serialized_start=3054
|
|
125
|
+
_globals['_ENTITYGROUPDUPLICATEINDEX_ENTITIESGROUPSENTRY']._serialized_end=3170
|
|
126
|
+
_globals['_VECTORSET']._serialized_start=3172
|
|
127
|
+
_globals['_VECTORSET']._serialized_end=3247
|
|
128
|
+
_globals['_VECTORSETS']._serialized_start=3250
|
|
129
|
+
_globals['_VECTORSETS']._serialized_end=3400
|
|
130
|
+
_globals['_VECTORSETS_VECTORSETSENTRY']._serialized_start=3326
|
|
131
|
+
_globals['_VECTORSETS_VECTORSETSENTRY']._serialized_end=3400
|
|
132
|
+
_globals['_VECTORINDEXCONFIG']._serialized_start=3403
|
|
133
|
+
_globals['_VECTORINDEXCONFIG']._serialized_end=3593
|
|
134
|
+
_globals['_VECTORSETCONFIG']._serialized_start=3596
|
|
135
|
+
_globals['_VECTORSETCONFIG']._serialized_end=3866
|
|
136
|
+
_globals['_VECTORSETCONFIG_STORAGEKEYKIND']._serialized_start=3805
|
|
137
|
+
_globals['_VECTORSETCONFIG_STORAGEKEYKIND']._serialized_end=3866
|
|
138
|
+
_globals['_KNOWLEDGEBOXVECTORSETSCONFIG']._serialized_start=3868
|
|
139
|
+
_globals['_KNOWLEDGEBOXVECTORSETSCONFIG']._serialized_end=3949
|
|
140
|
+
_globals['_VECTORSETPURGE']._serialized_start=3951
|
|
141
|
+
_globals['_VECTORSETPURGE']._serialized_end=4039
|
|
142
|
+
_globals['_TERMSYNONYMS']._serialized_start=4041
|
|
143
|
+
_globals['_TERMSYNONYMS']._serialized_end=4073
|
|
144
|
+
_globals['_SYNONYMS']._serialized_start=4076
|
|
145
|
+
_globals['_SYNONYMS']._serialized_end=4210
|
|
146
|
+
_globals['_SYNONYMS_TERMSENTRY']._serialized_start=4138
|
|
147
|
+
_globals['_SYNONYMS_TERMSENTRY']._serialized_end=4210
|
|
148
|
+
_globals['_SEMANTICMODELMETADATA']._serialized_start=4213
|
|
149
|
+
_globals['_SEMANTICMODELMETADATA']._serialized_end=4431
|
|
150
|
+
_globals['_KBCONFIGURATION']._serialized_start=4434
|
|
151
|
+
_globals['_KBCONFIGURATION']._serialized_end=4574
|
|
158
152
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -9,7 +9,6 @@ import google.protobuf.descriptor
|
|
|
9
9
|
import google.protobuf.internal.containers
|
|
10
10
|
import google.protobuf.internal.enum_type_wrapper
|
|
11
11
|
import google.protobuf.message
|
|
12
|
-
import nucliadb_protos.nodewriter_pb2
|
|
13
12
|
import nucliadb_protos.utils_pb2
|
|
14
13
|
import sys
|
|
15
14
|
import typing
|
|
@@ -18,23 +17,6 @@ if sys.version_info >= (3, 10):
|
|
|
18
17
|
import typing as typing_extensions
|
|
19
18
|
else:
|
|
20
19
|
import typing_extensions
|
|
21
|
-
from nucliadb_protos.nodewriter_pb2 import (
|
|
22
|
-
CREATION as CREATION,
|
|
23
|
-
DELETION as DELETION,
|
|
24
|
-
DENSE_F32 as DENSE_F32,
|
|
25
|
-
GarbageCollectorResponse as GarbageCollectorResponse,
|
|
26
|
-
IndexMessage as IndexMessage,
|
|
27
|
-
IndexMessageSource as IndexMessageSource,
|
|
28
|
-
MergeResponse as MergeResponse,
|
|
29
|
-
NewShardRequest as NewShardRequest,
|
|
30
|
-
NewVectorSetRequest as NewVectorSetRequest,
|
|
31
|
-
OpStatus as OpStatus,
|
|
32
|
-
PROCESSOR as PROCESSOR,
|
|
33
|
-
TypeMessage as TypeMessage,
|
|
34
|
-
VectorIndexConfig as VectorIndexConfig,
|
|
35
|
-
VectorType as VectorType,
|
|
36
|
-
WRITER as WRITER,
|
|
37
|
-
)
|
|
38
20
|
from nucliadb_protos.utils_pb2 import (
|
|
39
21
|
COSINE as COSINE,
|
|
40
22
|
DOT as DOT,
|
|
@@ -119,6 +101,19 @@ GCP_US_CENTRAL1: PineconeServerlessCloud.ValueType # 4
|
|
|
119
101
|
AZURE_EASTUS2: PineconeServerlessCloud.ValueType # 5
|
|
120
102
|
global___PineconeServerlessCloud = PineconeServerlessCloud
|
|
121
103
|
|
|
104
|
+
class _VectorType:
|
|
105
|
+
ValueType = typing.NewType("ValueType", builtins.int)
|
|
106
|
+
V: typing_extensions.TypeAlias = ValueType
|
|
107
|
+
|
|
108
|
+
class _VectorTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VectorType.ValueType], builtins.type):
|
|
109
|
+
DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
|
|
110
|
+
DENSE_F32: _VectorType.ValueType # 0
|
|
111
|
+
|
|
112
|
+
class VectorType(_VectorType, metaclass=_VectorTypeEnumTypeWrapper): ...
|
|
113
|
+
|
|
114
|
+
DENSE_F32: VectorType.ValueType # 0
|
|
115
|
+
global___VectorType = VectorType
|
|
116
|
+
|
|
122
117
|
@typing.final
|
|
123
118
|
class KnowledgeBoxID(google.protobuf.message.Message):
|
|
124
119
|
"""ID"""
|
|
@@ -766,6 +761,32 @@ class VectorSets(google.protobuf.message.Message):
|
|
|
766
761
|
|
|
767
762
|
global___VectorSets = VectorSets
|
|
768
763
|
|
|
764
|
+
@typing.final
|
|
765
|
+
class VectorIndexConfig(google.protobuf.message.Message):
|
|
766
|
+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
|
|
767
|
+
|
|
768
|
+
SIMILARITY_FIELD_NUMBER: builtins.int
|
|
769
|
+
NORMALIZE_VECTORS_FIELD_NUMBER: builtins.int
|
|
770
|
+
VECTOR_TYPE_FIELD_NUMBER: builtins.int
|
|
771
|
+
VECTOR_DIMENSION_FIELD_NUMBER: builtins.int
|
|
772
|
+
similarity: nucliadb_protos.utils_pb2.VectorSimilarity.ValueType
|
|
773
|
+
normalize_vectors: builtins.bool
|
|
774
|
+
vector_type: global___VectorType.ValueType
|
|
775
|
+
vector_dimension: builtins.int
|
|
776
|
+
def __init__(
|
|
777
|
+
self,
|
|
778
|
+
*,
|
|
779
|
+
similarity: nucliadb_protos.utils_pb2.VectorSimilarity.ValueType = ...,
|
|
780
|
+
normalize_vectors: builtins.bool = ...,
|
|
781
|
+
vector_type: global___VectorType.ValueType = ...,
|
|
782
|
+
vector_dimension: builtins.int | None = ...,
|
|
783
|
+
) -> None: ...
|
|
784
|
+
def HasField(self, field_name: typing.Literal["_vector_dimension", b"_vector_dimension", "vector_dimension", b"vector_dimension"]) -> builtins.bool: ...
|
|
785
|
+
def ClearField(self, field_name: typing.Literal["_vector_dimension", b"_vector_dimension", "normalize_vectors", b"normalize_vectors", "similarity", b"similarity", "vector_dimension", b"vector_dimension", "vector_type", b"vector_type"]) -> None: ...
|
|
786
|
+
def WhichOneof(self, oneof_group: typing.Literal["_vector_dimension", b"_vector_dimension"]) -> typing.Literal["vector_dimension"] | None: ...
|
|
787
|
+
|
|
788
|
+
global___VectorIndexConfig = VectorIndexConfig
|
|
789
|
+
|
|
769
790
|
@typing.final
|
|
770
791
|
class VectorSetConfig(google.protobuf.message.Message):
|
|
771
792
|
"""Configuration values for a vectorset"""
|
|
@@ -800,7 +821,7 @@ class VectorSetConfig(google.protobuf.message.Message):
|
|
|
800
821
|
vectorset_id: builtins.str
|
|
801
822
|
storage_key_kind: global___VectorSetConfig.StorageKeyKind.ValueType
|
|
802
823
|
@property
|
|
803
|
-
def vectorset_index_config(self) ->
|
|
824
|
+
def vectorset_index_config(self) -> global___VectorIndexConfig: ...
|
|
804
825
|
@property
|
|
805
826
|
def matryoshka_dimensions(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
|
|
806
827
|
"""list of possible subdivisions of the matryoshka embeddings (if the model
|
|
@@ -811,7 +832,7 @@ class VectorSetConfig(google.protobuf.message.Message):
|
|
|
811
832
|
self,
|
|
812
833
|
*,
|
|
813
834
|
vectorset_id: builtins.str = ...,
|
|
814
|
-
vectorset_index_config:
|
|
835
|
+
vectorset_index_config: global___VectorIndexConfig | None = ...,
|
|
815
836
|
matryoshka_dimensions: collections.abc.Iterable[builtins.int] | None = ...,
|
|
816
837
|
storage_key_kind: global___VectorSetConfig.StorageKeyKind.ValueType = ...,
|
|
817
838
|
) -> None: ...
|
nucliadb_protos/train_pb2.py
CHANGED
|
@@ -24,18 +24,6 @@ _sym_db = _symbol_database.Default()
|
|
|
24
24
|
|
|
25
25
|
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
26
26
|
from nucliadb_protos import knowledgebox_pb2 as nucliadb__protos_dot_knowledgebox__pb2
|
|
27
|
-
try:
|
|
28
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb__protos_dot_utils__pb2
|
|
29
|
-
except AttributeError:
|
|
30
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb_protos.utils_pb2
|
|
31
|
-
try:
|
|
32
|
-
nucliadb__protos_dot_nodewriter__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb__protos_dot_nodewriter__pb2
|
|
33
|
-
except AttributeError:
|
|
34
|
-
nucliadb__protos_dot_nodewriter__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb_protos.nodewriter_pb2
|
|
35
|
-
try:
|
|
36
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb__protos_dot_noderesources__pb2
|
|
37
|
-
except AttributeError:
|
|
38
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb_protos.noderesources_pb2
|
|
39
27
|
try:
|
|
40
28
|
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_knowledgebox__pb2.nucliadb__protos_dot_utils__pb2
|
|
41
29
|
except AttributeError:
|
|
@@ -46,14 +34,6 @@ try:
|
|
|
46
34
|
except AttributeError:
|
|
47
35
|
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_resources__pb2.nucliadb_protos.utils_pb2
|
|
48
36
|
from nucliadb_protos import writer_pb2 as nucliadb__protos_dot_writer__pb2
|
|
49
|
-
try:
|
|
50
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_noderesources__pb2
|
|
51
|
-
except AttributeError:
|
|
52
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb_protos.noderesources_pb2
|
|
53
|
-
try:
|
|
54
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_utils__pb2
|
|
55
|
-
except AttributeError:
|
|
56
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb_protos.utils_pb2
|
|
57
37
|
try:
|
|
58
38
|
nucliadb__protos_dot_resources__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_resources__pb2
|
|
59
39
|
except AttributeError:
|
|
@@ -70,18 +50,6 @@ try:
|
|
|
70
50
|
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_utils__pb2
|
|
71
51
|
except AttributeError:
|
|
72
52
|
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb_protos.utils_pb2
|
|
73
|
-
try:
|
|
74
|
-
nucliadb__protos_dot_nodewriter__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_nodewriter__pb2
|
|
75
|
-
except AttributeError:
|
|
76
|
-
nucliadb__protos_dot_nodewriter__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb_protos.nodewriter_pb2
|
|
77
|
-
try:
|
|
78
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_noderesources__pb2
|
|
79
|
-
except AttributeError:
|
|
80
|
-
nucliadb__protos_dot_noderesources__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb_protos.noderesources_pb2
|
|
81
|
-
try:
|
|
82
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_utils__pb2
|
|
83
|
-
except AttributeError:
|
|
84
|
-
nucliadb__protos_dot_utils__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb_protos.utils_pb2
|
|
85
53
|
try:
|
|
86
54
|
nucliadb__protos_dot_audit__pb2 = nucliadb__protos_dot_writer__pb2.nucliadb__protos_dot_audit__pb2
|
|
87
55
|
except AttributeError:
|
nucliadb_protos/train_pb2.pyi
CHANGED
|
@@ -20,6 +20,7 @@ from nucliadb_protos.knowledgebox_pb2 import (
|
|
|
20
20
|
CONFLICT as CONFLICT,
|
|
21
21
|
CreateExternalIndexProviderMetadata as CreateExternalIndexProviderMetadata,
|
|
22
22
|
CreatePineconeConfig as CreatePineconeConfig,
|
|
23
|
+
DENSE_F32 as DENSE_F32,
|
|
23
24
|
DeleteKnowledgeBoxResponse as DeleteKnowledgeBoxResponse,
|
|
24
25
|
DeletedEntitiesGroups as DeletedEntitiesGroups,
|
|
25
26
|
ERROR as ERROR,
|
|
@@ -53,10 +54,12 @@ from nucliadb_protos.knowledgebox_pb2 import (
|
|
|
53
54
|
TermSynonyms as TermSynonyms,
|
|
54
55
|
UNSET as UNSET,
|
|
55
56
|
UpdateKnowledgeBoxResponse as UpdateKnowledgeBoxResponse,
|
|
57
|
+
VectorIndexConfig as VectorIndexConfig,
|
|
56
58
|
VectorSet as VectorSet,
|
|
57
59
|
VectorSetConfig as VectorSetConfig,
|
|
58
60
|
VectorSetPurge as VectorSetPurge,
|
|
59
61
|
VectorSets as VectorSets,
|
|
62
|
+
VectorType as VectorType,
|
|
60
63
|
)
|
|
61
64
|
from nucliadb_protos.resources_pb2 import (
|
|
62
65
|
AllFieldIDs as AllFieldIDs,
|
|
@@ -159,7 +162,6 @@ from nucliadb_protos.writer_pb2 import (
|
|
|
159
162
|
PROCESSOR as PROCESSOR,
|
|
160
163
|
SetEntitiesRequest as SetEntitiesRequest,
|
|
161
164
|
ShardObject as ShardObject,
|
|
162
|
-
ShardReplica as ShardReplica,
|
|
163
165
|
Shards as Shards,
|
|
164
166
|
SynonymsRequest as SynonymsRequest,
|
|
165
167
|
UNSET as UNSET,
|
|
@@ -18,6 +18,7 @@ from nucliadb_protos.knowledgebox_pb2 import (
|
|
|
18
18
|
CONFLICT as CONFLICT,
|
|
19
19
|
CreateExternalIndexProviderMetadata as CreateExternalIndexProviderMetadata,
|
|
20
20
|
CreatePineconeConfig as CreatePineconeConfig,
|
|
21
|
+
DENSE_F32 as DENSE_F32,
|
|
21
22
|
DeleteKnowledgeBoxResponse as DeleteKnowledgeBoxResponse,
|
|
22
23
|
DeletedEntitiesGroups as DeletedEntitiesGroups,
|
|
23
24
|
ERROR as ERROR,
|
|
@@ -51,10 +52,12 @@ from nucliadb_protos.knowledgebox_pb2 import (
|
|
|
51
52
|
TermSynonyms as TermSynonyms,
|
|
52
53
|
UNSET as UNSET,
|
|
53
54
|
UpdateKnowledgeBoxResponse as UpdateKnowledgeBoxResponse,
|
|
55
|
+
VectorIndexConfig as VectorIndexConfig,
|
|
54
56
|
VectorSet as VectorSet,
|
|
55
57
|
VectorSetConfig as VectorSetConfig,
|
|
56
58
|
VectorSetPurge as VectorSetPurge,
|
|
57
59
|
VectorSets as VectorSets,
|
|
60
|
+
VectorType as VectorType,
|
|
58
61
|
)
|
|
59
62
|
from nucliadb_protos.resources_pb2 import (
|
|
60
63
|
AllFieldIDs as AllFieldIDs,
|
|
@@ -157,7 +160,6 @@ from nucliadb_protos.writer_pb2 import (
|
|
|
157
160
|
PROCESSOR as PROCESSOR,
|
|
158
161
|
SetEntitiesRequest as SetEntitiesRequest,
|
|
159
162
|
ShardObject as ShardObject,
|
|
160
|
-
ShardReplica as ShardReplica,
|
|
161
163
|
Shards as Shards,
|
|
162
164
|
SynonymsRequest as SynonymsRequest,
|
|
163
165
|
UNSET as UNSET,
|