frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/vectors/v1/vector_pb2.pyi

@@ -0,0 +1,142 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class DoubleVector(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ELEMENT_FIELD_NUMBER: builtins.int
+    @property
+    def element(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ...
+    def __init__(
+        self,
+        *,
+        element: collections.abc.Iterable[builtins.float] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["element", b"element"]) -> None: ...
+
+global___DoubleVector = DoubleVector
+
+class Property(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    STRING_VAL_FIELD_NUMBER: builtins.int
+    INT_VAL_FIELD_NUMBER: builtins.int
+    BOOL_VAL_FIELD_NUMBER: builtins.int
+    DOUBLE_VAL_FIELD_NUMBER: builtins.int
+    TIMESTAMP_VAL_FIELD_NUMBER: builtins.int
+    IS_EMPTY_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    string_val: builtins.str
+    int_val: builtins.int
+    """we treat all ints as 64 - no int/long distinction"""
+    bool_val: builtins.bool
+    double_val: builtins.float
+    """not distinguishing double vs float"""
+    @property
+    def timestamp_val(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
+    is_empty: builtins.bool
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        string_val: builtins.str = ...,
+        int_val: builtins.int = ...,
+        bool_val: builtins.bool = ...,
+        double_val: builtins.float = ...,
+        timestamp_val: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        is_empty: builtins.bool = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["bool_val", b"bool_val", "double_val", b"double_val", "int_val", b"int_val", "string_val", b"string_val", "timestamp_val", b"timestamp_val", "value_type", b"value_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["bool_val", b"bool_val", "double_val", b"double_val", "int_val", b"int_val", "is_empty", b"is_empty", "name", b"name", "string_val", b"string_val", "timestamp_val", b"timestamp_val", "value_type", b"value_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["value_type", b"value_type"]) -> typing_extensions.Literal["string_val", "int_val", "bool_val", "double_val", "timestamp_val"] | None: ...
+
+global___Property = Property
+
+class SearchResult(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    VECTOR_FIELD_NUMBER: builtins.int
+    PROPERTIES_FIELD_NUMBER: builtins.int
+    DISTANCE_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    @property
+    def vector(self) -> global___DoubleVector: ...
+    @property
+    def properties(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Property]: ...
+    distance: builtins.float
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        vector: global___DoubleVector | None = ...,
+        properties: collections.abc.Iterable[global___Property] | None = ...,
+        distance: builtins.float = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["vector", b"vector"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["distance", b"distance", "id", b"id", "properties", b"properties", "vector", b"vector"]) -> None: ...
+
+global___SearchResult = SearchResult
+
+class StoredVector(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    VECTOR_FIELD_NUMBER: builtins.int
+    PROPERTIES_FIELD_NUMBER: builtins.int
+    VECTOR_IDENTIFIER_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    @property
+    def vector(self) -> global___DoubleVector: ...
+    @property
+    def properties(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Property]: ...
+    @property
+    def vector_identifier(self) -> global___VectorIdentifier: ...
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        vector: global___DoubleVector | None = ...,
+        properties: collections.abc.Iterable[global___Property] | None = ...,
+        vector_identifier: global___VectorIdentifier | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["vector", b"vector", "vector_identifier", b"vector_identifier"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["id", b"id", "properties", b"properties", "vector", b"vector", "vector_identifier", b"vector_identifier"]) -> None: ...
+
+global___StoredVector = StoredVector
+
+class VectorIdentifier(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VECTOR_ID_FIELD_NUMBER: builtins.int
+    TENANT_ID_FIELD_NUMBER: builtins.int
+    vector_id: builtins.str
+    tenant_id: builtins.str
+    def __init__(
+        self,
+        *,
+        vector_id: builtins.str = ...,
+        tenant_id: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["optional_tenant", b"optional_tenant", "tenant_id", b"tenant_id"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["optional_tenant", b"optional_tenant", "tenant_id", b"tenant_id", "vector_id", b"vector_id"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tenant", b"optional_tenant"]) -> typing_extensions.Literal["tenant_id"] | None: ...
+
+global___VectorIdentifier = VectorIdentifier

_frogml_proto/qwak/vectors/v1/vector_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_frogml_proto/qwak/vectors/v1/vector_service_pb2.py

@@ -0,0 +1,134 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.vectors/v1/vector_service.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+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 _frogml_proto.qwak.vectors.v1 import vector_pb2 as qwak_dot_vectors_dot_v1_dot_vector__pb2
+from _frogml_proto.qwak.vectors.v1 import filters_pb2 as qwak_dot_vectors_dot_v1_dot_filters__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$qwak/vectors/v1/vector_service.proto\x12\x1cqwak.vector.store.serving.v1\x1a\x1cqwak/vectors/v1/vector.proto\x1a\x1dqwak/vectors/v1/filters.proto\"\xd9\x02\n\x1bSearchSimilarVectorsRequest\x12\x17\n\x0f\x63ollection_name\x18\x01 \x01(\t\x12\x44\n\x10reference_vector\x18\x02 \x01(\x0b\x32*.qwak.vector.store.serving.v1.DoubleVector\x12\x12\n\nproperties\x18\x03 \x03(\t\x12\x36\n\x06\x66ilter\x18\x04 \x01(\x0b\x32$.qwak.vector.store.serving.v1.FilterH\x00\x12\x13\n\x0bmax_results\x18\x05 \x01(\x05\x12\x16\n\x0einclude_vector\x18\x06 \x01(\x08\x12\x18\n\x10include_distance\x18\x07 \x01(\x08\x12\x12\n\ninclude_id\x18\x08 \x01(\x08\x12\x13\n\ttenant_id\x18\t \x01(\tH\x01\x42\x0c\n\nfilter_optB\x11\n\x0foptional_tenant\"b\n\x1cSearchSimilarVectorsResponse\x12\x42\n\x0esearch_results\x18\x01 \x03(\x0b\x32*.qwak.vector.store.serving.v1.SearchResult\"\x86\x01\n\x16GetVectorSampleRequest\x12\x17\n\x0f\x63ollection_name\x18\x01 \x01(\t\x12\x13\n\x0bnum_samples\x18\x02 \x01(\x05\x12\x16\n\x0einclude_vector\x18\x03 \x01(\x08\x12\x13\n\ttenant_id\x18\t \x01(\tH\x00\x42\x11\n\x0foptional_tenant\"V\n\x17GetVectorSampleResponse\x12;\n\x07vectors\x18\x01 \x03(\x0b\x32*.qwak.vector.store.serving.v1.StoredVector\"k\n\x14UpsertVectorsRequest\x12\x17\n\x0f\x63ollection_name\x18\x01 \x01(\t\x12:\n\x06vector\x18\x02 \x03(\x0b\x32*.qwak.vector.store.serving.v1.StoredVector\"-\n\x15UpsertVectorsResponse\x12\x14\n\x0cnum_affected\x18\x01 \x01(\x05\"\x92\x01\n\x14\x44\x65leteVectorsRequest\x12\x17\n\x0f\x63ollection_name\x18\x01 \x01(\t\x12\x15\n\tvector_id\x18\x02 \x03(\tB\x02\x18\x01\x12J\n\x12vector_identifiers\x18\x03 \x03(\x0b\x32..qwak.vector.store.serving.v1.VectorIdentifier\"4\n\x15\x44\x65leteVectorsResponse\x12\x1b\n\x13num_vectors_deleted\x18\x01 \x01(\x03\"\x8f\x01\n\x12\x46\x65tchVectorRequest\x12\x17\n\x0f\x63ollection_name\x18\x01 \x01(\t\x12\x15\n\tvector_id\x18\x02 \x01(\tB\x02\x18\x01\x12I\n\x11vector_identifier\x18\x03 \x01(\x0b\x32..qwak.vector.store.serving.v1.VectorIdentifier\"Q\n\x13\x46\x65tchVectorResponse\x12:\n\x06vector\x18\x01 \x01(\x0b\x32*.qwak.vector.store.serving.v1.StoredVector2\x87\x05\n\rVectorService\x12\x8d\x01\n\x14SearchSimilarVectors\x12\x39.qwak.vector.store.serving.v1.SearchSimilarVectorsRequest\x1a:.qwak.vector.store.serving.v1.SearchSimilarVectorsResponse\x12~\n\x0fGetVectorSample\x12\x34.qwak.vector.store.serving.v1.GetVectorSampleRequest\x1a\x35.qwak.vector.store.serving.v1.GetVectorSampleResponse\x12r\n\x0b\x46\x65tchVector\x12\x30.qwak.vector.store.serving.v1.FetchVectorRequest\x1a\x31.qwak.vector.store.serving.v1.FetchVectorResponse\x12x\n\rUpsertVectors\x12\x32.qwak.vector.store.serving.v1.UpsertVectorsRequest\x1a\x33.qwak.vector.store.serving.v1.UpsertVectorsResponse\x12x\n\rDeleteVectors\x12\x32.qwak.vector.store.serving.v1.DeleteVectorsRequest\x1a\x33.qwak.vector.store.serving.v1.DeleteVectorsResponseB\x1e\n\x1a\x63om.qwak.ai.vectors.api.v1P\x01\x62\x06proto3')
+
+
+
+_SEARCHSIMILARVECTORSREQUEST = DESCRIPTOR.message_types_by_name['SearchSimilarVectorsRequest']
+_SEARCHSIMILARVECTORSRESPONSE = DESCRIPTOR.message_types_by_name['SearchSimilarVectorsResponse']
+_GETVECTORSAMPLEREQUEST = DESCRIPTOR.message_types_by_name['GetVectorSampleRequest']
+_GETVECTORSAMPLERESPONSE = DESCRIPTOR.message_types_by_name['GetVectorSampleResponse']
+_UPSERTVECTORSREQUEST = DESCRIPTOR.message_types_by_name['UpsertVectorsRequest']
+_UPSERTVECTORSRESPONSE = DESCRIPTOR.message_types_by_name['UpsertVectorsResponse']
+_DELETEVECTORSREQUEST = DESCRIPTOR.message_types_by_name['DeleteVectorsRequest']
+_DELETEVECTORSRESPONSE = DESCRIPTOR.message_types_by_name['DeleteVectorsResponse']
+_FETCHVECTORREQUEST = DESCRIPTOR.message_types_by_name['FetchVectorRequest']
+_FETCHVECTORRESPONSE = DESCRIPTOR.message_types_by_name['FetchVectorResponse']
+SearchSimilarVectorsRequest = _reflection.GeneratedProtocolMessageType('SearchSimilarVectorsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _SEARCHSIMILARVECTORSREQUEST,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.SearchSimilarVectorsRequest)
+  })
+_sym_db.RegisterMessage(SearchSimilarVectorsRequest)
+
+SearchSimilarVectorsResponse = _reflection.GeneratedProtocolMessageType('SearchSimilarVectorsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _SEARCHSIMILARVECTORSRESPONSE,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.SearchSimilarVectorsResponse)
+  })
+_sym_db.RegisterMessage(SearchSimilarVectorsResponse)
+
+GetVectorSampleRequest = _reflection.GeneratedProtocolMessageType('GetVectorSampleRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETVECTORSAMPLEREQUEST,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.GetVectorSampleRequest)
+  })
+_sym_db.RegisterMessage(GetVectorSampleRequest)
+
+GetVectorSampleResponse = _reflection.GeneratedProtocolMessageType('GetVectorSampleResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETVECTORSAMPLERESPONSE,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.GetVectorSampleResponse)
+  })
+_sym_db.RegisterMessage(GetVectorSampleResponse)
+
+UpsertVectorsRequest = _reflection.GeneratedProtocolMessageType('UpsertVectorsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _UPSERTVECTORSREQUEST,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.UpsertVectorsRequest)
+  })
+_sym_db.RegisterMessage(UpsertVectorsRequest)
+
+UpsertVectorsResponse = _reflection.GeneratedProtocolMessageType('UpsertVectorsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _UPSERTVECTORSRESPONSE,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.UpsertVectorsResponse)
+  })
+_sym_db.RegisterMessage(UpsertVectorsResponse)
+
+DeleteVectorsRequest = _reflection.GeneratedProtocolMessageType('DeleteVectorsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _DELETEVECTORSREQUEST,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.DeleteVectorsRequest)
+  })
+_sym_db.RegisterMessage(DeleteVectorsRequest)
+
+DeleteVectorsResponse = _reflection.GeneratedProtocolMessageType('DeleteVectorsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _DELETEVECTORSRESPONSE,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.DeleteVectorsResponse)
+  })
+_sym_db.RegisterMessage(DeleteVectorsResponse)
+
+FetchVectorRequest = _reflection.GeneratedProtocolMessageType('FetchVectorRequest', (_message.Message,), {
+  'DESCRIPTOR' : _FETCHVECTORREQUEST,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.FetchVectorRequest)
+  })
+_sym_db.RegisterMessage(FetchVectorRequest)
+
+FetchVectorResponse = _reflection.GeneratedProtocolMessageType('FetchVectorResponse', (_message.Message,), {
+  'DESCRIPTOR' : _FETCHVECTORRESPONSE,
+  '__module__' : 'qwak.vectors.v1.vector_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.serving.v1.FetchVectorResponse)
+  })
+_sym_db.RegisterMessage(FetchVectorResponse)
+
+_VECTORSERVICE = DESCRIPTOR.services_by_name['VectorService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.qwak.ai.vectors.api.v1P\001'
+  _DELETEVECTORSREQUEST.fields_by_name['vector_id']._options = None
+  _DELETEVECTORSREQUEST.fields_by_name['vector_id']._serialized_options = b'\030\001'
+  _FETCHVECTORREQUEST.fields_by_name['vector_id']._options = None
+  _FETCHVECTORREQUEST.fields_by_name['vector_id']._serialized_options = b'\030\001'
+  _SEARCHSIMILARVECTORSREQUEST._serialized_start=132
+  _SEARCHSIMILARVECTORSREQUEST._serialized_end=477
+  _SEARCHSIMILARVECTORSRESPONSE._serialized_start=479
+  _SEARCHSIMILARVECTORSRESPONSE._serialized_end=577
+  _GETVECTORSAMPLEREQUEST._serialized_start=580
+  _GETVECTORSAMPLEREQUEST._serialized_end=714
+  _GETVECTORSAMPLERESPONSE._serialized_start=716
+  _GETVECTORSAMPLERESPONSE._serialized_end=802
+  _UPSERTVECTORSREQUEST._serialized_start=804
+  _UPSERTVECTORSREQUEST._serialized_end=911
+  _UPSERTVECTORSRESPONSE._serialized_start=913
+  _UPSERTVECTORSRESPONSE._serialized_end=958
+  _DELETEVECTORSREQUEST._serialized_start=961
+  _DELETEVECTORSREQUEST._serialized_end=1107
+  _DELETEVECTORSRESPONSE._serialized_start=1109
+  _DELETEVECTORSRESPONSE._serialized_end=1161
+  _FETCHVECTORREQUEST._serialized_start=1164
+  _FETCHVECTORREQUEST._serialized_end=1307
+  _FETCHVECTORRESPONSE._serialized_start=1309
+  _FETCHVECTORRESPONSE._serialized_end=1390
+  _VECTORSERVICE._serialized_start=1393
+  _VECTORSERVICE._serialized_end=2040
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/vectors/v1/vector_service_pb2.pyi

@@ -0,0 +1,243 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import _frogml_proto.qwak.vectors.v1.filters_pb2
+import _frogml_proto.qwak.vectors.v1.vector_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class SearchSimilarVectorsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    COLLECTION_NAME_FIELD_NUMBER: builtins.int
+    REFERENCE_VECTOR_FIELD_NUMBER: builtins.int
+    PROPERTIES_FIELD_NUMBER: builtins.int
+    FILTER_FIELD_NUMBER: builtins.int
+    MAX_RESULTS_FIELD_NUMBER: builtins.int
+    INCLUDE_VECTOR_FIELD_NUMBER: builtins.int
+    INCLUDE_DISTANCE_FIELD_NUMBER: builtins.int
+    INCLUDE_ID_FIELD_NUMBER: builtins.int
+    TENANT_ID_FIELD_NUMBER: builtins.int
+    collection_name: builtins.str
+    """Name of the collection to search in"""
+    @property
+    def reference_vector(self) -> _frogml_proto.qwak.vectors.v1.vector_pb2.DoubleVector:
+        """Reference vector"""
+    @property
+    def properties(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """properties to retrieve"""
+    @property
+    def filter(self) -> _frogml_proto.qwak.vectors.v1.filters_pb2.Filter: ...
+    max_results: builtins.int
+    """the "k" in top_k"""
+    include_vector: builtins.bool
+    """whether to include the actual result vectors in the response"""
+    include_distance: builtins.bool
+    """whether to include the distance in the result"""
+    include_id: builtins.bool
+    """whether to include ids in the result"""
+    tenant_id: builtins.str
+    def __init__(
+        self,
+        *,
+        collection_name: builtins.str = ...,
+        reference_vector: _frogml_proto.qwak.vectors.v1.vector_pb2.DoubleVector | None = ...,
+        properties: collections.abc.Iterable[builtins.str] | None = ...,
+        filter: _frogml_proto.qwak.vectors.v1.filters_pb2.Filter | None = ...,
+        max_results: builtins.int = ...,
+        include_vector: builtins.bool = ...,
+        include_distance: builtins.bool = ...,
+        include_id: builtins.bool = ...,
+        tenant_id: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["filter", b"filter", "filter_opt", b"filter_opt", "optional_tenant", b"optional_tenant", "reference_vector", b"reference_vector", "tenant_id", b"tenant_id"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "filter", b"filter", "filter_opt", b"filter_opt", "include_distance", b"include_distance", "include_id", b"include_id", "include_vector", b"include_vector", "max_results", b"max_results", "optional_tenant", b"optional_tenant", "properties", b"properties", "reference_vector", b"reference_vector", "tenant_id", b"tenant_id"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["filter_opt", b"filter_opt"]) -> typing_extensions.Literal["filter"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tenant", b"optional_tenant"]) -> typing_extensions.Literal["tenant_id"] | None: ...
+
+global___SearchSimilarVectorsRequest = SearchSimilarVectorsRequest
+
+class SearchSimilarVectorsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SEARCH_RESULTS_FIELD_NUMBER: builtins.int
+    @property
+    def search_results(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.SearchResult]: ...
+    def __init__(
+        self,
+        *,
+        search_results: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.SearchResult] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["search_results", b"search_results"]) -> None: ...
+
+global___SearchSimilarVectorsResponse = SearchSimilarVectorsResponse
+
+class GetVectorSampleRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    COLLECTION_NAME_FIELD_NUMBER: builtins.int
+    NUM_SAMPLES_FIELD_NUMBER: builtins.int
+    INCLUDE_VECTOR_FIELD_NUMBER: builtins.int
+    TENANT_ID_FIELD_NUMBER: builtins.int
+    collection_name: builtins.str
+    """Name of the collection to search in"""
+    num_samples: builtins.int
+    """Number of samples to retrieve"""
+    include_vector: builtins.bool
+    """Whether to include the actual result vectors in the response"""
+    tenant_id: builtins.str
+    def __init__(
+        self,
+        *,
+        collection_name: builtins.str = ...,
+        num_samples: builtins.int = ...,
+        include_vector: builtins.bool = ...,
+        tenant_id: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["optional_tenant", b"optional_tenant", "tenant_id", b"tenant_id"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "include_vector", b"include_vector", "num_samples", b"num_samples", "optional_tenant", b"optional_tenant", "tenant_id", b"tenant_id"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_tenant", b"optional_tenant"]) -> typing_extensions.Literal["tenant_id"] | None: ...
+
+global___GetVectorSampleRequest = GetVectorSampleRequest
+
+class GetVectorSampleResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VECTORS_FIELD_NUMBER: builtins.int
+    @property
+    def vectors(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.StoredVector]: ...
+    def __init__(
+        self,
+        *,
+        vectors: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.StoredVector] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["vectors", b"vectors"]) -> None: ...
+
+global___GetVectorSampleResponse = GetVectorSampleResponse
+
+class UpsertVectorsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    COLLECTION_NAME_FIELD_NUMBER: builtins.int
+    VECTOR_FIELD_NUMBER: builtins.int
+    collection_name: builtins.str
+    @property
+    def vector(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.StoredVector]: ...
+    def __init__(
+        self,
+        *,
+        collection_name: builtins.str = ...,
+        vector: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.StoredVector] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "vector", b"vector"]) -> None: ...
+
+global___UpsertVectorsRequest = UpsertVectorsRequest
+
+class UpsertVectorsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NUM_AFFECTED_FIELD_NUMBER: builtins.int
+    num_affected: builtins.int
+    """number of vectors that were affected in the collection"""
+    def __init__(
+        self,
+        *,
+        num_affected: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["num_affected", b"num_affected"]) -> None: ...
+
+global___UpsertVectorsResponse = UpsertVectorsResponse
+
+class DeleteVectorsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    COLLECTION_NAME_FIELD_NUMBER: builtins.int
+    VECTOR_ID_FIELD_NUMBER: builtins.int
+    VECTOR_IDENTIFIERS_FIELD_NUMBER: builtins.int
+    collection_name: builtins.str
+    @property
+    def vector_id(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """passing explicit id is now deprecated, pass a VectorIdentifier"""
+    @property
+    def vector_identifiers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.vectors.v1.vector_pb2.VectorIdentifier]: ...
+    def __init__(
+        self,
+        *,
+        collection_name: builtins.str = ...,
+        vector_id: collections.abc.Iterable[builtins.str] | None = ...,
+        vector_identifiers: collections.abc.Iterable[qwak.vectors.v1.vector_pb2.VectorIdentifier] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "vector_id", b"vector_id", "vector_identifiers", b"vector_identifiers"]) -> None: ...
+
+global___DeleteVectorsRequest = DeleteVectorsRequest
+
+class DeleteVectorsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NUM_VECTORS_DELETED_FIELD_NUMBER: builtins.int
+    num_vectors_deleted: builtins.int
+    def __init__(
+        self,
+        *,
+        num_vectors_deleted: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["num_vectors_deleted", b"num_vectors_deleted"]) -> None: ...
+
+global___DeleteVectorsResponse = DeleteVectorsResponse
+
+class FetchVectorRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    COLLECTION_NAME_FIELD_NUMBER: builtins.int
+    VECTOR_ID_FIELD_NUMBER: builtins.int
+    VECTOR_IDENTIFIER_FIELD_NUMBER: builtins.int
+    collection_name: builtins.str
+    """Name of the collection to fetch from"""
+    vector_id: builtins.str
+    """Vector id to fetch - this field is deprecated, use VectorIdentifier"""
+    @property
+    def vector_identifier(self) -> _frogml_proto.qwak.vectors.v1.vector_pb2.VectorIdentifier:
+        """Vector identifier to fetch"""
+    def __init__(
+        self,
+        *,
+        collection_name: builtins.str = ...,
+        vector_id: builtins.str = ...,
+        vector_identifier: _frogml_proto.qwak.vectors.v1.vector_pb2.VectorIdentifier | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["vector_identifier", b"vector_identifier"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["collection_name", b"collection_name", "vector_id", b"vector_id", "vector_identifier", b"vector_identifier"]) -> None: ...
+
+global___FetchVectorRequest = FetchVectorRequest
+
+class FetchVectorResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VECTOR_FIELD_NUMBER: builtins.int
+    @property
+    def vector(self) -> _frogml_proto.qwak.vectors.v1.vector_pb2.StoredVector:
+        """Fetched vector"""
+    def __init__(
+        self,
+        *,
+        vector: _frogml_proto.qwak.vectors.v1.vector_pb2.StoredVector | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["vector", b"vector"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["vector", b"vector"]) -> None: ...
+
+global___FetchVectorResponse = FetchVectorResponse