# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: needlestack/apis/indices.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from needlestack.apis import tensors_pb2 as needlestack_dot_apis_dot_tensors__pb2
from needlestack.apis import data_sources_pb2 as needlestack_dot_apis_dot_data__sources__pb2
DESCRIPTOR = _descriptor.FileDescriptor(
name='needlestack/apis/indices.proto',
package='',
syntax='proto3',
serialized_options=None,
serialized_pb=_b('\n\x1eneedlestack/apis/indices.proto\x1a\x1eneedlestack/apis/tensors.proto\x1a#needlestack/apis/data_sources.proto\"8\n\tBaseIndex\x12\"\n\x0b\x66\x61iss_index\x18\x01 \x01(\x0b\x32\x0b.FaissIndexH\x00\x42\x07\n\x05index\"b\n\nFaissIndex\x12\x14\n\x0cindex_binary\x18\x01 \x01(\x0c\x12\x1c\n\tmetadatas\x18\x02 \x03(\x0b\x32\t.Metadata\x12 \n\x0b\x64\x61ta_source\x18\x03 \x01(\x0b\x32\x0b.DataSource\"6\n\x08Metadata\x12\n\n\x02id\x18\x01 \x01(\t\x12\x1e\n\x06\x66ields\x18\x02 \x03(\x0b\x32\x0e.MetadataField\"\xa2\x01\n\rMetadataField\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\nstring_val\x18\x02 \x01(\tH\x00\x12\x14\n\ndouble_val\x18\x03 \x01(\x01H\x00\x12\x13\n\tfloat_val\x18\x04 \x01(\x02H\x00\x12\x12\n\x08long_val\x18\x05 \x01(\x03H\x00\x12\x11\n\x07int_val\x18\x06 \x01(\x05H\x00\x12\x12\n\x08\x62ool_val\x18\x07 \x01(\x08H\x00\x42\x07\n\x05value\"p\n\x10SearchResultItem\x12\x18\n\x0e\x66loat_distance\x18\x01 \x01(\x02H\x00\x12\x19\n\x0f\x64ouble_distance\x18\x02 \x01(\x01H\x00\x12\x1b\n\x08metadata\x18\x03 \x01(\x0b\x32\t.MetadataB\n\n\x08\x64istance\"L\n\x13RetrievalResultItem\x12\x18\n\x06vector\x18\x01 \x01(\x0b\x32\x08.NDArray\x12\x1b\n\x08metadata\x18\x02 \x01(\x0b\x32\t.Metadatab\x06proto3')
,
dependencies=[needlestack_dot_apis_dot_tensors__pb2.DESCRIPTOR,needlestack_dot_apis_dot_data__sources__pb2.DESCRIPTOR,])
_BASEINDEX = _descriptor.Descriptor(
name='BaseIndex',
full_name='BaseIndex',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='faiss_index', full_name='BaseIndex.faiss_index', index=0,
number=1, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto3',
extension_ranges=[],
oneofs=[
_descriptor.OneofDescriptor(
name='index', full_name='BaseIndex.index',
index=0, containing_type=None, fields=[]),
],
serialized_start=103,
serialized_end=159,
)
_FAISSINDEX = _descriptor.Descriptor(
name='FaissIndex',
full_name='FaissIndex',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='index_binary', full_name='FaissIndex.index_binary', index=0,
number=1, type=12, cpp_type=9, label=1,
has_default_value=False, default_value=_b(""),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='metadatas', full_name='FaissIndex.metadatas', index=1,
number=2, type=11, cpp_type=10, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='data_source', full_name='FaissIndex.data_source', index=2,
number=3, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto3',
extension_ranges=[],
oneofs=[
],
serialized_start=161,
serialized_end=259,
)
_METADATA = _descriptor.Descriptor(
name='Metadata',
full_name='Metadata',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='id', full_name='Metadata.id', index=0,
number=1, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='fields', full_name='Metadata.fields', index=1,
number=2, type=11, cpp_type=10, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto3',
extension_ranges=[],
oneofs=[
],
serialized_start=261,
serialized_end=315,
)
_METADATAFIELD = _descriptor.Descriptor(
name='MetadataField',
full_name='MetadataField',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='name', full_name='MetadataField.name', index=0,
number=1, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='string_val', full_name='MetadataField.string_val', index=1,
number=2, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='double_val', full_name='MetadataField.double_val', index=2,
number=3, type=1, cpp_type=5, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='float_val', full_name='MetadataField.float_val', index=3,
number=4, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='long_val', full_name='MetadataField.long_val', index=4,
number=5, type=3, cpp_type=2, label=1,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='int_val', full_name='MetadataField.int_val', index=5,
number=6, type=5, cpp_type=1, label=1,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='bool_val', full_name='MetadataField.bool_val', index=6,
number=7, type=8, cpp_type=7, label=1,
has_default_value=False, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto3',
extension_ranges=[],
oneofs=[
_descriptor.OneofDescriptor(
name='value', full_name='MetadataField.value',
index=0, containing_type=None, fields=[]),
],
serialized_start=318,
serialized_end=480,
)
_SEARCHRESULTITEM = _descriptor.Descriptor(
name='SearchResultItem',
full_name='SearchResultItem',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='float_distance', full_name='SearchResultItem.float_distance', index=0,
number=1, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='double_distance', full_name='SearchResultItem.double_distance', index=1,
number=2, type=1, cpp_type=5, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='metadata', full_name='SearchResultItem.metadata', index=2,
number=3, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto3',
extension_ranges=[],
oneofs=[
_descriptor.OneofDescriptor(
name='distance', full_name='SearchResultItem.distance',
index=0, containing_type=None, fields=[]),
],
serialized_start=482,
serialized_end=594,
)
_RETRIEVALRESULTITEM = _descriptor.Descriptor(
name='RetrievalResultItem',
full_name='RetrievalResultItem',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='vector', full_name='RetrievalResultItem.vector', index=0,
number=1, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='metadata', full_name='RetrievalResultItem.metadata', index=1,
number=2, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto3',
extension_ranges=[],
oneofs=[
],
serialized_start=596,
serialized_end=672,
)
_BASEINDEX.fields_by_name['faiss_index'].message_type = _FAISSINDEX
_BASEINDEX.oneofs_by_name['index'].fields.append(
_BASEINDEX.fields_by_name['faiss_index'])
_BASEINDEX.fields_by_name['faiss_index'].containing_oneof = _BASEINDEX.oneofs_by_name['index']
_FAISSINDEX.fields_by_name['metadatas'].message_type = _METADATA
_FAISSINDEX.fields_by_name['data_source'].message_type = needlestack_dot_apis_dot_data__sources__pb2._DATASOURCE
_METADATA.fields_by_name['fields'].message_type = _METADATAFIELD
_METADATAFIELD.oneofs_by_name['value'].fields.append(
_METADATAFIELD.fields_by_name['string_val'])
_METADATAFIELD.fields_by_name['string_val'].containing_oneof = _METADATAFIELD.oneofs_by_name['value']
_METADATAFIELD.oneofs_by_name['value'].fields.append(
_METADATAFIELD.fields_by_name['double_val'])
_METADATAFIELD.fields_by_name['double_val'].containing_oneof = _METADATAFIELD.oneofs_by_name['value']
_METADATAFIELD.oneofs_by_name['value'].fields.append(
_METADATAFIELD.fields_by_name['float_val'])
_METADATAFIELD.fields_by_name['float_val'].containing_oneof = _METADATAFIELD.oneofs_by_name['value']
_METADATAFIELD.oneofs_by_name['value'].fields.append(
_METADATAFIELD.fields_by_name['long_val'])
_METADATAFIELD.fields_by_name['long_val'].containing_oneof = _METADATAFIELD.oneofs_by_name['value']
_METADATAFIELD.oneofs_by_name['value'].fields.append(
_METADATAFIELD.fields_by_name['int_val'])
_METADATAFIELD.fields_by_name['int_val'].containing_oneof = _METADATAFIELD.oneofs_by_name['value']
_METADATAFIELD.oneofs_by_name['value'].fields.append(
_METADATAFIELD.fields_by_name['bool_val'])
_METADATAFIELD.fields_by_name['bool_val'].containing_oneof = _METADATAFIELD.oneofs_by_name['value']
_SEARCHRESULTITEM.fields_by_name['metadata'].message_type = _METADATA
_SEARCHRESULTITEM.oneofs_by_name['distance'].fields.append(
_SEARCHRESULTITEM.fields_by_name['float_distance'])
_SEARCHRESULTITEM.fields_by_name['float_distance'].containing_oneof = _SEARCHRESULTITEM.oneofs_by_name['distance']
_SEARCHRESULTITEM.oneofs_by_name['distance'].fields.append(
_SEARCHRESULTITEM.fields_by_name['double_distance'])
_SEARCHRESULTITEM.fields_by_name['double_distance'].containing_oneof = _SEARCHRESULTITEM.oneofs_by_name['distance']
_RETRIEVALRESULTITEM.fields_by_name['vector'].message_type = needlestack_dot_apis_dot_tensors__pb2._NDARRAY
_RETRIEVALRESULTITEM.fields_by_name['metadata'].message_type = _METADATA
DESCRIPTOR.message_types_by_name['BaseIndex'] = _BASEINDEX
DESCRIPTOR.message_types_by_name['FaissIndex'] = _FAISSINDEX
DESCRIPTOR.message_types_by_name['Metadata'] = _METADATA
DESCRIPTOR.message_types_by_name['MetadataField'] = _METADATAFIELD
DESCRIPTOR.message_types_by_name['SearchResultItem'] = _SEARCHRESULTITEM
DESCRIPTOR.message_types_by_name['RetrievalResultItem'] = _RETRIEVALRESULTITEM
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
BaseIndex = _reflection.GeneratedProtocolMessageType('BaseIndex', (_message.Message,), {
'DESCRIPTOR' : _BASEINDEX,
'__module__' : 'needlestack.apis.indices_pb2'
# @@protoc_insertion_point(class_scope:BaseIndex)
})
_sym_db.RegisterMessage(BaseIndex)
FaissIndex = _reflection.GeneratedProtocolMessageType('FaissIndex', (_message.Message,), {
'DESCRIPTOR' : _FAISSINDEX,
'__module__' : 'needlestack.apis.indices_pb2'
# @@protoc_insertion_point(class_scope:FaissIndex)
})
_sym_db.RegisterMessage(FaissIndex)
Metadata = _reflection.GeneratedProtocolMessageType('Metadata', (_message.Message,), {
'DESCRIPTOR' : _METADATA,
'__module__' : 'needlestack.apis.indices_pb2'
# @@protoc_insertion_point(class_scope:Metadata)
})
_sym_db.RegisterMessage(Metadata)
MetadataField = _reflection.GeneratedProtocolMessageType('MetadataField', (_message.Message,), {
'DESCRIPTOR' : _METADATAFIELD,
'__module__' : 'needlestack.apis.indices_pb2'
# @@protoc_insertion_point(class_scope:MetadataField)
})
_sym_db.RegisterMessage(MetadataField)
SearchResultItem = _reflection.GeneratedProtocolMessageType('SearchResultItem', (_message.Message,), {
'DESCRIPTOR' : _SEARCHRESULTITEM,
'__module__' : 'needlestack.apis.indices_pb2'
# @@protoc_insertion_point(class_scope:SearchResultItem)
})
_sym_db.RegisterMessage(SearchResultItem)
RetrievalResultItem = _reflection.GeneratedProtocolMessageType('RetrievalResultItem', (_message.Message,), {
'DESCRIPTOR' : _RETRIEVALRESULTITEM,
'__module__' : 'needlestack.apis.indices_pb2'
# @@protoc_insertion_point(class_scope:RetrievalResultItem)
})
_sym_db.RegisterMessage(RetrievalResultItem)
# @@protoc_insertion_point(module_scope)