From 95959ebe67bc07e611b31b77cc24df907f099ae9 Mon Sep 17 00:00:00 2001 From: rohanshah18 Date: Thu, 26 Jun 2025 13:33:25 -0400 Subject: [PATCH 1/4] add support for grpc namespaces --- codegen/apis | 2 +- codegen/buf.yaml | 2 +- docs/grpc.rst | 11 + .../core/grpc/protos/db_data_2025_04_pb2.py | 146 ++++++ .../core/grpc/protos/db_data_2025_04_pb2.pyi | 295 ++++++++++++ .../grpc/protos/db_data_2025_04_pb2_grpc.py | 419 ++++++++++++++++++ pinecone/grpc/__init__.py | 5 +- pinecone/grpc/grpc_runner.py | 2 + pinecone/grpc/index_grpc.py | 154 ++++++- pinecone/grpc/sparse_values_factory.py | 2 +- pinecone/grpc/utils.py | 37 ++ pinecone/grpc/vector_factory_grpc.py | 2 +- tests/integration/data/test_namespace.py | 6 - .../data_grpc_futures/stub_backend.py | 4 +- .../test_grpc_index_describe_index_stats.py | 2 +- tests/unit_grpc/test_grpc_index_fetch.py | 2 +- tests/unit_grpc/test_grpc_index_namespace.py | 78 ++++ tests/unit_grpc/test_grpc_index_query.py | 2 +- tests/unit_grpc/test_grpc_index_update.py | 2 +- tests/unit_grpc/test_grpc_index_upsert.py | 2 +- 20 files changed, 1152 insertions(+), 23 deletions(-) create mode 100644 pinecone/core/grpc/protos/db_data_2025_04_pb2.py create mode 100644 pinecone/core/grpc/protos/db_data_2025_04_pb2.pyi create mode 100644 pinecone/core/grpc/protos/db_data_2025_04_pb2_grpc.py create mode 100644 tests/unit_grpc/test_grpc_index_namespace.py diff --git a/codegen/apis b/codegen/apis index 7e21ca9a..827d26f4 160000 --- a/codegen/apis +++ b/codegen/apis @@ -1 +1 @@ -Subproject commit 7e21ca9adb6a530ce11909d6209d69551f86e9bd +Subproject commit 827d26f4825902994a099595d49779d16fea3a0a diff --git a/codegen/buf.yaml b/codegen/buf.yaml index a979e0a5..988170bd 100644 --- a/codegen/buf.yaml +++ b/codegen/buf.yaml @@ -9,4 +9,4 @@ breaking: deps: - buf.build/googleapis/googleapis modules: - - path: apis/_build/2025-01 + - path: apis/_build/2025-04 diff --git a/docs/grpc.rst b/docs/grpc.rst index 2ced748d..1980e9c7 100644 --- a/docs/grpc.rst +++ b/docs/grpc.rst @@ -80,3 +80,14 @@ Vectors .. automethod:: pinecone.grpc::GRPCIndex.list .. automethod:: pinecone.grpc::GRPCIndex.list_paginated + +Namespaces +---------- + +.. automethod:: pinecone.grpc::GRPCIndex.list_namespaces + +.. automethod:: pinecone.grpc::GRPCIndex.list_namespaces_paginated + +.. automethod:: pinecone.grpc::GRPCIndex.describe_namespace + +.. automethod:: pinecone.grpc::GRPCIndex.delete_namespace diff --git a/pinecone/core/grpc/protos/db_data_2025_04_pb2.py b/pinecone/core/grpc/protos/db_data_2025_04_pb2.py new file mode 100644 index 00000000..caf7aa59 --- /dev/null +++ b/pinecone/core/grpc/protos/db_data_2025_04_pb2.py @@ -0,0 +1,146 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# NO CHECKED-IN PROTOBUF GENCODE +# source: db_data_2025-04.proto +# Protobuf Python Version: 5.29.0 +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import runtime_version as _runtime_version +from google.protobuf import symbol_database as _symbol_database +from google.protobuf.internal import builder as _builder +_runtime_version.ValidateProtobufRuntimeVersion( + _runtime_version.Domain.PUBLIC, + 5, + 29, + 0, + '', + 'db_data_2025-04.proto' +) +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2 +from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2 +from google.api import field_behavior_pb2 as google_dot_api_dot_field__behavior__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x64\x62_data_2025-04.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/api/field_behavior.proto\"J\n\x0cSparseValues\x12\x1d\n\x07indices\x18\x01 \x03(\rB\x03\xe0\x41\x02R\x07indices\x12\x1b\n\x06values\x18\x02 \x03(\x02\x42\x03\xe0\x41\x02R\x06values\"\x9e\x01\n\x06Vector\x12\x13\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x02R\x02id\x12\x16\n\x06values\x18\x02 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x04 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12\x33\n\x08metadata\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\"\xba\x01\n\x0cScoredVector\x12\x13\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x02R\x02id\x12\x14\n\x05score\x18\x02 \x01(\x02R\x05score\x12\x16\n\x06values\x18\x03 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x05 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12\x33\n\x08metadata\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\"\xa1\x01\n\x0cRequestUnion\x12(\n\x06upsert\x18\x01 \x01(\x0b\x32\x0e.UpsertRequestH\x00R\x06upsert\x12(\n\x06\x64\x65lete\x18\x02 \x01(\x0b\x32\x0e.DeleteRequestH\x00R\x06\x64\x65lete\x12(\n\x06update\x18\x03 \x01(\x0b\x32\x0e.UpdateRequestH\x00R\x06updateB\x13\n\x11RequestUnionInner\"U\n\rUpsertRequest\x12&\n\x07vectors\x18\x01 \x03(\x0b\x32\x07.VectorB\x03\xe0\x41\x02R\x07vectors\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\"7\n\x0eUpsertResponse\x12%\n\x0eupserted_count\x18\x01 \x01(\rR\rupsertedCount\"\x8f\x01\n\rDeleteRequest\x12\x10\n\x03ids\x18\x01 \x03(\tR\x03ids\x12\x1d\n\ndelete_all\x18\x02 \x01(\x08R\tdeleteAll\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12/\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\"\x10\n\x0e\x44\x65leteResponse\"C\n\x0c\x46\x65tchRequest\x12\x15\n\x03ids\x18\x01 \x03(\tB\x03\xe0\x41\x02R\x03ids\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\"\xd6\x01\n\rFetchResponse\x12\x35\n\x07vectors\x18\x01 \x03(\x0b\x32\x1b.FetchResponse.VectorsEntryR\x07vectors\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\x12!\n\x05usage\x18\x03 \x01(\x0b\x32\x06.UsageH\x00R\x05usage\x88\x01\x01\x1a\x43\n\x0cVectorsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x1d\n\x05value\x18\x02 \x01(\x0b\x32\x07.VectorR\x05value:\x02\x38\x01\x42\x08\n\x06_usage\"\xbd\x01\n\x0bListRequest\x12\x1b\n\x06prefix\x18\x01 \x01(\tH\x00R\x06prefix\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\rH\x01R\x05limit\x88\x01\x01\x12.\n\x10pagination_token\x18\x03 \x01(\tH\x02R\x0fpaginationToken\x88\x01\x01\x12\x1c\n\tnamespace\x18\x04 \x01(\tR\tnamespaceB\t\n\x07_prefixB\x08\n\x06_limitB\x13\n\x11_pagination_token\" \n\nPagination\x12\x12\n\x04next\x18\x01 \x01(\tR\x04next\"\x1a\n\x08ListItem\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\"\xbf\x01\n\x0cListResponse\x12#\n\x07vectors\x18\x01 \x03(\x0b\x32\t.ListItemR\x07vectors\x12\x30\n\npagination\x18\x02 \x01(\x0b\x32\x0b.PaginationH\x00R\npagination\x88\x01\x01\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12!\n\x05usage\x18\x04 \x01(\x0b\x32\x06.UsageH\x01R\x05usage\x88\x01\x01\x42\r\n\x0b_paginationB\x08\n\x06_usage\"\xbd\x01\n\x0bQueryVector\x12\x16\n\x06values\x18\x01 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x05 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12\x13\n\x05top_k\x18\x02 \x01(\rR\x04topK\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12/\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\"\xd1\x02\n\x0cQueryRequest\x12\x1c\n\tnamespace\x18\x01 \x01(\tR\tnamespace\x12\x18\n\x05top_k\x18\x02 \x01(\rB\x03\xe0\x41\x02R\x04topK\x12/\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\x12%\n\x0einclude_values\x18\x04 \x01(\x08R\rincludeValues\x12)\n\x10include_metadata\x18\x05 \x01(\x08R\x0fincludeMetadata\x12*\n\x07queries\x18\x06 \x03(\x0b\x32\x0c.QueryVectorB\x02\x18\x01R\x07queries\x12\x16\n\x06vector\x18\x07 \x03(\x02R\x06vector\x12\x32\n\rsparse_vector\x18\t \x01(\x0b\x32\r.SparseValuesR\x0csparseVector\x12\x0e\n\x02id\x18\x08 \x01(\tR\x02id\"[\n\x12SingleQueryResults\x12\'\n\x07matches\x18\x01 \x03(\x0b\x32\r.ScoredVectorR\x07matches\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\"\xb6\x01\n\rQueryResponse\x12\x31\n\x07results\x18\x01 \x03(\x0b\x32\x13.SingleQueryResultsB\x02\x18\x01R\x07results\x12\'\n\x07matches\x18\x02 \x03(\x0b\x32\r.ScoredVectorR\x07matches\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12!\n\x05usage\x18\x04 \x01(\x0b\x32\x06.UsageH\x00R\x05usage\x88\x01\x01\x42\x08\n\x06_usage\":\n\x05Usage\x12\"\n\nread_units\x18\x01 \x01(\rH\x00R\treadUnits\x88\x01\x01\x42\r\n\x0b_read_units\"\xca\x01\n\rUpdateRequest\x12\x13\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x02R\x02id\x12\x16\n\x06values\x18\x02 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x05 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12:\n\x0cset_metadata\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x0bsetMetadata\x12\x1c\n\tnamespace\x18\x04 \x01(\tR\tnamespace\"\x10\n\x0eUpdateResponse\"L\n\x19\x44\x65scribeIndexStatsRequest\x12/\n\x06\x66ilter\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\"5\n\x10NamespaceSummary\x12!\n\x0cvector_count\x18\x01 \x01(\rR\x0bvectorCount\"\x81\x01\n\x15ListNamespacesRequest\x12.\n\x10pagination_token\x18\x01 \x01(\tH\x00R\x0fpaginationToken\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\rH\x01R\x05limit\x88\x01\x01\x42\x13\n\x11_pagination_tokenB\x08\n\x06_limit\"\x90\x01\n\x16ListNamespacesResponse\x12\x35\n\nnamespaces\x18\x01 \x03(\x0b\x32\x15.NamespaceDescriptionR\nnamespaces\x12\x30\n\npagination\x18\x02 \x01(\x0b\x32\x0b.PaginationH\x00R\npagination\x88\x01\x01\x42\r\n\x0b_pagination\"8\n\x18\x44\x65scribeNamespaceRequest\x12\x1c\n\tnamespace\x18\x01 \x01(\tR\tnamespace\"M\n\x14NamespaceDescription\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12!\n\x0crecord_count\x18\x02 \x01(\x04R\x0brecordCount\"6\n\x16\x44\x65leteNamespaceRequest\x12\x1c\n\tnamespace\x18\x01 \x01(\tR\tnamespace\"\x9f\x03\n\x1a\x44\x65scribeIndexStatsResponse\x12K\n\nnamespaces\x18\x01 \x03(\x0b\x32+.DescribeIndexStatsResponse.NamespacesEntryR\nnamespaces\x12!\n\tdimension\x18\x02 \x01(\rH\x00R\tdimension\x88\x01\x01\x12%\n\x0eindex_fullness\x18\x03 \x01(\x02R\rindexFullness\x12,\n\x12total_vector_count\x18\x04 \x01(\rR\x10totalVectorCount\x12\x1b\n\x06metric\x18\x05 \x01(\tH\x01R\x06metric\x88\x01\x01\x12$\n\x0bvector_type\x18\x06 \x01(\tH\x02R\nvectorType\x88\x01\x01\x1aP\n\x0fNamespacesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\'\n\x05value\x18\x02 \x01(\x0b\x32\x11.NamespaceSummaryR\x05value:\x02\x38\x01\x42\x0c\n\n_dimensionB\t\n\x07_metricB\x0e\n\x0c_vector_type2\xd7\x06\n\rVectorService\x12\x45\n\x06Upsert\x12\x0e.UpsertRequest\x1a\x0f.UpsertResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/vectors/upsert:\x01*\x12X\n\x06\x44\x65lete\x12\x0e.DeleteRequest\x1a\x0f.DeleteResponse\"-\x82\xd3\xe4\x93\x02\'\"\x0f/vectors/delete:\x01*Z\x11*\x0f/vectors/delete\x12>\n\x05\x46\x65tch\x12\r.FetchRequest\x1a\x0e.FetchResponse\"\x16\x82\xd3\xe4\x93\x02\x10\x12\x0e/vectors/fetch\x12:\n\x04List\x12\x0c.ListRequest\x1a\r.ListResponse\"\x15\x82\xd3\xe4\x93\x02\x0f\x12\r/vectors/list\x12\x39\n\x05Query\x12\r.QueryRequest\x1a\x0e.QueryResponse\"\x11\x82\xd3\xe4\x93\x02\x0b\"\x06/query:\x01*\x12\x45\n\x06Update\x12\x0e.UpdateRequest\x1a\x0f.UpdateResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/vectors/update:\x01*\x12\x88\x01\n\x12\x44\x65scribeIndexStats\x12\x1a.DescribeIndexStatsRequest\x1a\x1b.DescribeIndexStatsResponse\"9\x82\xd3\xe4\x93\x02\x33\"\x15/describe_index_stats:\x01*Z\x17\x12\x15/describe_index_stats\x12V\n\x0eListNamespaces\x12\x16.ListNamespacesRequest\x1a\x17.ListNamespacesResponse\"\x13\x82\xd3\xe4\x93\x02\r\x12\x0b/namespaces\x12\x66\n\x11\x44\x65scribeNamespace\x12\x19.DescribeNamespaceRequest\x1a\x15.NamespaceDescription\"\x1f\x82\xd3\xe4\x93\x02\x19\x12\x17/namespaces/{namespace}\x12\\\n\x0f\x44\x65leteNamespace\x12\x17.DeleteNamespaceRequest\x1a\x0f.DeleteResponse\"\x1f\x82\xd3\xe4\x93\x02\x19*\x17/namespaces/{namespace}BS\n\x11io.pinecone.protoP\x01Z None: ... + +class Vector(_message.Message): + __slots__ = ("id", "values", "sparse_values", "metadata") + ID_FIELD_NUMBER: _ClassVar[int] + VALUES_FIELD_NUMBER: _ClassVar[int] + SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] + METADATA_FIELD_NUMBER: _ClassVar[int] + id: str + values: _containers.RepeatedScalarFieldContainer[float] + sparse_values: SparseValues + metadata: _struct_pb2.Struct + def __init__(self, id: _Optional[str] = ..., values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... + +class ScoredVector(_message.Message): + __slots__ = ("id", "score", "values", "sparse_values", "metadata") + ID_FIELD_NUMBER: _ClassVar[int] + SCORE_FIELD_NUMBER: _ClassVar[int] + VALUES_FIELD_NUMBER: _ClassVar[int] + SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] + METADATA_FIELD_NUMBER: _ClassVar[int] + id: str + score: float + values: _containers.RepeatedScalarFieldContainer[float] + sparse_values: SparseValues + metadata: _struct_pb2.Struct + def __init__(self, id: _Optional[str] = ..., score: _Optional[float] = ..., values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... + +class RequestUnion(_message.Message): + __slots__ = ("upsert", "delete", "update") + UPSERT_FIELD_NUMBER: _ClassVar[int] + DELETE_FIELD_NUMBER: _ClassVar[int] + UPDATE_FIELD_NUMBER: _ClassVar[int] + upsert: UpsertRequest + delete: DeleteRequest + update: UpdateRequest + def __init__(self, upsert: _Optional[_Union[UpsertRequest, _Mapping]] = ..., delete: _Optional[_Union[DeleteRequest, _Mapping]] = ..., update: _Optional[_Union[UpdateRequest, _Mapping]] = ...) -> None: ... + +class UpsertRequest(_message.Message): + __slots__ = ("vectors", "namespace") + VECTORS_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + vectors: _containers.RepeatedCompositeFieldContainer[Vector] + namespace: str + def __init__(self, vectors: _Optional[_Iterable[_Union[Vector, _Mapping]]] = ..., namespace: _Optional[str] = ...) -> None: ... + +class UpsertResponse(_message.Message): + __slots__ = ("upserted_count",) + UPSERTED_COUNT_FIELD_NUMBER: _ClassVar[int] + upserted_count: int + def __init__(self, upserted_count: _Optional[int] = ...) -> None: ... + +class DeleteRequest(_message.Message): + __slots__ = ("ids", "delete_all", "namespace", "filter") + IDS_FIELD_NUMBER: _ClassVar[int] + DELETE_ALL_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + FILTER_FIELD_NUMBER: _ClassVar[int] + ids: _containers.RepeatedScalarFieldContainer[str] + delete_all: bool + namespace: str + filter: _struct_pb2.Struct + def __init__(self, ids: _Optional[_Iterable[str]] = ..., delete_all: bool = ..., namespace: _Optional[str] = ..., filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... + +class DeleteResponse(_message.Message): + __slots__ = () + def __init__(self) -> None: ... + +class FetchRequest(_message.Message): + __slots__ = ("ids", "namespace") + IDS_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + ids: _containers.RepeatedScalarFieldContainer[str] + namespace: str + def __init__(self, ids: _Optional[_Iterable[str]] = ..., namespace: _Optional[str] = ...) -> None: ... + +class FetchResponse(_message.Message): + __slots__ = ("vectors", "namespace", "usage") + class VectorsEntry(_message.Message): + __slots__ = ("key", "value") + KEY_FIELD_NUMBER: _ClassVar[int] + VALUE_FIELD_NUMBER: _ClassVar[int] + key: str + value: Vector + def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Vector, _Mapping]] = ...) -> None: ... + VECTORS_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + USAGE_FIELD_NUMBER: _ClassVar[int] + vectors: _containers.MessageMap[str, Vector] + namespace: str + usage: Usage + def __init__(self, vectors: _Optional[_Mapping[str, Vector]] = ..., namespace: _Optional[str] = ..., usage: _Optional[_Union[Usage, _Mapping]] = ...) -> None: ... + +class ListRequest(_message.Message): + __slots__ = ("prefix", "limit", "pagination_token", "namespace") + PREFIX_FIELD_NUMBER: _ClassVar[int] + LIMIT_FIELD_NUMBER: _ClassVar[int] + PAGINATION_TOKEN_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + prefix: str + limit: int + pagination_token: str + namespace: str + def __init__(self, prefix: _Optional[str] = ..., limit: _Optional[int] = ..., pagination_token: _Optional[str] = ..., namespace: _Optional[str] = ...) -> None: ... + +class Pagination(_message.Message): + __slots__ = ("next",) + NEXT_FIELD_NUMBER: _ClassVar[int] + next: str + def __init__(self, next: _Optional[str] = ...) -> None: ... + +class ListItem(_message.Message): + __slots__ = ("id",) + ID_FIELD_NUMBER: _ClassVar[int] + id: str + def __init__(self, id: _Optional[str] = ...) -> None: ... + +class ListResponse(_message.Message): + __slots__ = ("vectors", "pagination", "namespace", "usage") + VECTORS_FIELD_NUMBER: _ClassVar[int] + PAGINATION_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + USAGE_FIELD_NUMBER: _ClassVar[int] + vectors: _containers.RepeatedCompositeFieldContainer[ListItem] + pagination: Pagination + namespace: str + usage: Usage + def __init__(self, vectors: _Optional[_Iterable[_Union[ListItem, _Mapping]]] = ..., pagination: _Optional[_Union[Pagination, _Mapping]] = ..., namespace: _Optional[str] = ..., usage: _Optional[_Union[Usage, _Mapping]] = ...) -> None: ... + +class QueryVector(_message.Message): + __slots__ = ("values", "sparse_values", "top_k", "namespace", "filter") + VALUES_FIELD_NUMBER: _ClassVar[int] + SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] + TOP_K_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + FILTER_FIELD_NUMBER: _ClassVar[int] + values: _containers.RepeatedScalarFieldContainer[float] + sparse_values: SparseValues + top_k: int + namespace: str + filter: _struct_pb2.Struct + def __init__(self, values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., top_k: _Optional[int] = ..., namespace: _Optional[str] = ..., filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... + +class QueryRequest(_message.Message): + __slots__ = ("namespace", "top_k", "filter", "include_values", "include_metadata", "queries", "vector", "sparse_vector", "id") + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + TOP_K_FIELD_NUMBER: _ClassVar[int] + FILTER_FIELD_NUMBER: _ClassVar[int] + INCLUDE_VALUES_FIELD_NUMBER: _ClassVar[int] + INCLUDE_METADATA_FIELD_NUMBER: _ClassVar[int] + QUERIES_FIELD_NUMBER: _ClassVar[int] + VECTOR_FIELD_NUMBER: _ClassVar[int] + SPARSE_VECTOR_FIELD_NUMBER: _ClassVar[int] + ID_FIELD_NUMBER: _ClassVar[int] + namespace: str + top_k: int + filter: _struct_pb2.Struct + include_values: bool + include_metadata: bool + queries: _containers.RepeatedCompositeFieldContainer[QueryVector] + vector: _containers.RepeatedScalarFieldContainer[float] + sparse_vector: SparseValues + id: str + def __init__(self, namespace: _Optional[str] = ..., top_k: _Optional[int] = ..., filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., include_values: bool = ..., include_metadata: bool = ..., queries: _Optional[_Iterable[_Union[QueryVector, _Mapping]]] = ..., vector: _Optional[_Iterable[float]] = ..., sparse_vector: _Optional[_Union[SparseValues, _Mapping]] = ..., id: _Optional[str] = ...) -> None: ... + +class SingleQueryResults(_message.Message): + __slots__ = ("matches", "namespace") + MATCHES_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + matches: _containers.RepeatedCompositeFieldContainer[ScoredVector] + namespace: str + def __init__(self, matches: _Optional[_Iterable[_Union[ScoredVector, _Mapping]]] = ..., namespace: _Optional[str] = ...) -> None: ... + +class QueryResponse(_message.Message): + __slots__ = ("results", "matches", "namespace", "usage") + RESULTS_FIELD_NUMBER: _ClassVar[int] + MATCHES_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + USAGE_FIELD_NUMBER: _ClassVar[int] + results: _containers.RepeatedCompositeFieldContainer[SingleQueryResults] + matches: _containers.RepeatedCompositeFieldContainer[ScoredVector] + namespace: str + usage: Usage + def __init__(self, results: _Optional[_Iterable[_Union[SingleQueryResults, _Mapping]]] = ..., matches: _Optional[_Iterable[_Union[ScoredVector, _Mapping]]] = ..., namespace: _Optional[str] = ..., usage: _Optional[_Union[Usage, _Mapping]] = ...) -> None: ... + +class Usage(_message.Message): + __slots__ = ("read_units",) + READ_UNITS_FIELD_NUMBER: _ClassVar[int] + read_units: int + def __init__(self, read_units: _Optional[int] = ...) -> None: ... + +class UpdateRequest(_message.Message): + __slots__ = ("id", "values", "sparse_values", "set_metadata", "namespace") + ID_FIELD_NUMBER: _ClassVar[int] + VALUES_FIELD_NUMBER: _ClassVar[int] + SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] + SET_METADATA_FIELD_NUMBER: _ClassVar[int] + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + id: str + values: _containers.RepeatedScalarFieldContainer[float] + sparse_values: SparseValues + set_metadata: _struct_pb2.Struct + namespace: str + def __init__(self, id: _Optional[str] = ..., values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., set_metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., namespace: _Optional[str] = ...) -> None: ... + +class UpdateResponse(_message.Message): + __slots__ = () + def __init__(self) -> None: ... + +class DescribeIndexStatsRequest(_message.Message): + __slots__ = ("filter",) + FILTER_FIELD_NUMBER: _ClassVar[int] + filter: _struct_pb2.Struct + def __init__(self, filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... + +class NamespaceSummary(_message.Message): + __slots__ = ("vector_count",) + VECTOR_COUNT_FIELD_NUMBER: _ClassVar[int] + vector_count: int + def __init__(self, vector_count: _Optional[int] = ...) -> None: ... + +class ListNamespacesRequest(_message.Message): + __slots__ = ("pagination_token", "limit") + PAGINATION_TOKEN_FIELD_NUMBER: _ClassVar[int] + LIMIT_FIELD_NUMBER: _ClassVar[int] + pagination_token: str + limit: int + def __init__(self, pagination_token: _Optional[str] = ..., limit: _Optional[int] = ...) -> None: ... + +class ListNamespacesResponse(_message.Message): + __slots__ = ("namespaces", "pagination") + NAMESPACES_FIELD_NUMBER: _ClassVar[int] + PAGINATION_FIELD_NUMBER: _ClassVar[int] + namespaces: _containers.RepeatedCompositeFieldContainer[NamespaceDescription] + pagination: Pagination + def __init__(self, namespaces: _Optional[_Iterable[_Union[NamespaceDescription, _Mapping]]] = ..., pagination: _Optional[_Union[Pagination, _Mapping]] = ...) -> None: ... + +class DescribeNamespaceRequest(_message.Message): + __slots__ = ("namespace",) + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + namespace: str + def __init__(self, namespace: _Optional[str] = ...) -> None: ... + +class NamespaceDescription(_message.Message): + __slots__ = ("name", "record_count") + NAME_FIELD_NUMBER: _ClassVar[int] + RECORD_COUNT_FIELD_NUMBER: _ClassVar[int] + name: str + record_count: int + def __init__(self, name: _Optional[str] = ..., record_count: _Optional[int] = ...) -> None: ... + +class DeleteNamespaceRequest(_message.Message): + __slots__ = ("namespace",) + NAMESPACE_FIELD_NUMBER: _ClassVar[int] + namespace: str + def __init__(self, namespace: _Optional[str] = ...) -> None: ... + +class DescribeIndexStatsResponse(_message.Message): + __slots__ = ("namespaces", "dimension", "index_fullness", "total_vector_count", "metric", "vector_type") + class NamespacesEntry(_message.Message): + __slots__ = ("key", "value") + KEY_FIELD_NUMBER: _ClassVar[int] + VALUE_FIELD_NUMBER: _ClassVar[int] + key: str + value: NamespaceSummary + def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[NamespaceSummary, _Mapping]] = ...) -> None: ... + NAMESPACES_FIELD_NUMBER: _ClassVar[int] + DIMENSION_FIELD_NUMBER: _ClassVar[int] + INDEX_FULLNESS_FIELD_NUMBER: _ClassVar[int] + TOTAL_VECTOR_COUNT_FIELD_NUMBER: _ClassVar[int] + METRIC_FIELD_NUMBER: _ClassVar[int] + VECTOR_TYPE_FIELD_NUMBER: _ClassVar[int] + namespaces: _containers.MessageMap[str, NamespaceSummary] + dimension: int + index_fullness: float + total_vector_count: int + metric: str + vector_type: str + def __init__(self, namespaces: _Optional[_Mapping[str, NamespaceSummary]] = ..., dimension: _Optional[int] = ..., index_fullness: _Optional[float] = ..., total_vector_count: _Optional[int] = ..., metric: _Optional[str] = ..., vector_type: _Optional[str] = ...) -> None: ... diff --git a/pinecone/core/grpc/protos/db_data_2025_04_pb2_grpc.py b/pinecone/core/grpc/protos/db_data_2025_04_pb2_grpc.py new file mode 100644 index 00000000..d733efcf --- /dev/null +++ b/pinecone/core/grpc/protos/db_data_2025_04_pb2_grpc.py @@ -0,0 +1,419 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +import pinecone.core.grpc.protos.db_data_2025_04_pb2 as db__data__2025__04__pb2 + + +class VectorServiceStub(object): + """The `VectorService` interface is exposed by Pinecone's vector index services. + This service could also be called a `gRPC` service or a `REST`-like api. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Upsert = channel.unary_unary( + '/VectorService/Upsert', + request_serializer=db__data__2025__04__pb2.UpsertRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.UpsertResponse.FromString, + ) + self.Delete = channel.unary_unary( + '/VectorService/Delete', + request_serializer=db__data__2025__04__pb2.DeleteRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.DeleteResponse.FromString, + ) + self.Fetch = channel.unary_unary( + '/VectorService/Fetch', + request_serializer=db__data__2025__04__pb2.FetchRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.FetchResponse.FromString, + ) + self.List = channel.unary_unary( + '/VectorService/List', + request_serializer=db__data__2025__04__pb2.ListRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.ListResponse.FromString, + ) + self.Query = channel.unary_unary( + '/VectorService/Query', + request_serializer=db__data__2025__04__pb2.QueryRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.QueryResponse.FromString, + ) + self.Update = channel.unary_unary( + '/VectorService/Update', + request_serializer=db__data__2025__04__pb2.UpdateRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.UpdateResponse.FromString, + ) + self.DescribeIndexStats = channel.unary_unary( + '/VectorService/DescribeIndexStats', + request_serializer=db__data__2025__04__pb2.DescribeIndexStatsRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.DescribeIndexStatsResponse.FromString, + ) + self.ListNamespaces = channel.unary_unary( + '/VectorService/ListNamespaces', + request_serializer=db__data__2025__04__pb2.ListNamespacesRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.ListNamespacesResponse.FromString, + ) + self.DescribeNamespace = channel.unary_unary( + '/VectorService/DescribeNamespace', + request_serializer=db__data__2025__04__pb2.DescribeNamespaceRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.NamespaceDescription.FromString, + ) + self.DeleteNamespace = channel.unary_unary( + '/VectorService/DeleteNamespace', + request_serializer=db__data__2025__04__pb2.DeleteNamespaceRequest.SerializeToString, + response_deserializer=db__data__2025__04__pb2.DeleteResponse.FromString, + ) + + +class VectorServiceServicer(object): + """The `VectorService` interface is exposed by Pinecone's vector index services. + This service could also be called a `gRPC` service or a `REST`-like api. + """ + + def Upsert(self, request, context): + """Upsert vectors + + Upsert vectors into a namespace. If a new value is upserted for an existing vector ID, it will overwrite the previous value. + + For guidance, examples, and limits, see [Upsert data](https://docs.pinecone.io/guides/index-data/upsert-data). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Delete(self, request, context): + """Delete vectors + + Delete vectors by id from a single namespace. + + For guidance and examples, see [Delete data](https://docs.pinecone.io/guides/manage-data/delete-data). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Fetch(self, request, context): + """Fetch vectors + + Look up and return vectors by ID from a single namespace. The returned vectors include the vector data and/or metadata. + + For guidance and examples, see [Fetch data](https://docs.pinecone.io/guides/manage-data/fetch-data). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def List(self, request, context): + """List vector IDs + + List the IDs of vectors in a single namespace of a serverless index. An optional prefix can be passed to limit the results to IDs with a common prefix. + + This returns up to 100 IDs at a time by default in sorted order (bitwise/"C" collation). If the `limit` parameter is set, `list` returns up to that number of IDs instead. Whenever there are additional IDs to return, the response also includes a `pagination_token` that you can use to get the next batch of IDs. When the response does not include a `pagination_token`, there are no more IDs to return. + + For guidance and examples, see [List record IDs](https://docs.pinecone.io/guides/manage-data/list-record-ids). + + **Note:** `list` is supported only for serverless indexes. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Query(self, request, context): + """Search with a vector + + Search a namespace with a query vector or record ID and return the IDs of the most similar records, along with their similarity scores. + + For guidance, examples, and limits, see [Search](https://docs.pinecone.io/guides/search/search-overview). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Update(self, request, context): + """Update a vector + + Update a vector in a namespace. If a value is included, it will overwrite the previous value. If a `set_metadata` is included, the values of the fields specified in it will be added or overwrite the previous value. + + For guidance and examples, see [Update data](https://docs.pinecone.io/guides/manage-data/update-data). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DescribeIndexStats(self, request, context): + """Get index stats + + Return statistics about the contents of an index, including the vector count per namespace, the number of dimensions, and the index fullness. + + Serverless indexes scale automatically as needed, so index fullness is relevant only for pod-based indexes. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ListNamespaces(self, request, context): + """List namespaces + + Get a list of all [namespaces](https://docs.pinecone.io/guides/index-data/indexing-overview#namespaces) in a serverless index. + + Up to 100 namespaces are returned at a time by default, in sorted order (bitwise “C” collation). If the `limit` parameter is set, up to that number of namespaces are returned instead. Whenever there are additional namespaces to return, the response also includes a `pagination_token` that you can use to get the next batch of namespaces. When the response does not include a `pagination_token`, there are no more namespaces to return. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DescribeNamespace(self, request, context): + """Describe a namespace + + Describe a [namespace](https://docs.pinecone.io/guides/index-data/indexing-overview#namespaces) in a serverless index, including the total number of vectors in the namespace. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DeleteNamespace(self, request, context): + """Delete a namespace + + Delete a namespace from an index. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_VectorServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Upsert': grpc.unary_unary_rpc_method_handler( + servicer.Upsert, + request_deserializer=db__data__2025__04__pb2.UpsertRequest.FromString, + response_serializer=db__data__2025__04__pb2.UpsertResponse.SerializeToString, + ), + 'Delete': grpc.unary_unary_rpc_method_handler( + servicer.Delete, + request_deserializer=db__data__2025__04__pb2.DeleteRequest.FromString, + response_serializer=db__data__2025__04__pb2.DeleteResponse.SerializeToString, + ), + 'Fetch': grpc.unary_unary_rpc_method_handler( + servicer.Fetch, + request_deserializer=db__data__2025__04__pb2.FetchRequest.FromString, + response_serializer=db__data__2025__04__pb2.FetchResponse.SerializeToString, + ), + 'List': grpc.unary_unary_rpc_method_handler( + servicer.List, + request_deserializer=db__data__2025__04__pb2.ListRequest.FromString, + response_serializer=db__data__2025__04__pb2.ListResponse.SerializeToString, + ), + 'Query': grpc.unary_unary_rpc_method_handler( + servicer.Query, + request_deserializer=db__data__2025__04__pb2.QueryRequest.FromString, + response_serializer=db__data__2025__04__pb2.QueryResponse.SerializeToString, + ), + 'Update': grpc.unary_unary_rpc_method_handler( + servicer.Update, + request_deserializer=db__data__2025__04__pb2.UpdateRequest.FromString, + response_serializer=db__data__2025__04__pb2.UpdateResponse.SerializeToString, + ), + 'DescribeIndexStats': grpc.unary_unary_rpc_method_handler( + servicer.DescribeIndexStats, + request_deserializer=db__data__2025__04__pb2.DescribeIndexStatsRequest.FromString, + response_serializer=db__data__2025__04__pb2.DescribeIndexStatsResponse.SerializeToString, + ), + 'ListNamespaces': grpc.unary_unary_rpc_method_handler( + servicer.ListNamespaces, + request_deserializer=db__data__2025__04__pb2.ListNamespacesRequest.FromString, + response_serializer=db__data__2025__04__pb2.ListNamespacesResponse.SerializeToString, + ), + 'DescribeNamespace': grpc.unary_unary_rpc_method_handler( + servicer.DescribeNamespace, + request_deserializer=db__data__2025__04__pb2.DescribeNamespaceRequest.FromString, + response_serializer=db__data__2025__04__pb2.NamespaceDescription.SerializeToString, + ), + 'DeleteNamespace': grpc.unary_unary_rpc_method_handler( + servicer.DeleteNamespace, + request_deserializer=db__data__2025__04__pb2.DeleteNamespaceRequest.FromString, + response_serializer=db__data__2025__04__pb2.DeleteResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'VectorService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class VectorService(object): + """The `VectorService` interface is exposed by Pinecone's vector index services. + This service could also be called a `gRPC` service or a `REST`-like api. + """ + + @staticmethod + def Upsert(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/Upsert', + db__data__2025__04__pb2.UpsertRequest.SerializeToString, + db__data__2025__04__pb2.UpsertResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Delete(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/Delete', + db__data__2025__04__pb2.DeleteRequest.SerializeToString, + db__data__2025__04__pb2.DeleteResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Fetch(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/Fetch', + db__data__2025__04__pb2.FetchRequest.SerializeToString, + db__data__2025__04__pb2.FetchResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def List(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/List', + db__data__2025__04__pb2.ListRequest.SerializeToString, + db__data__2025__04__pb2.ListResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Query(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/Query', + db__data__2025__04__pb2.QueryRequest.SerializeToString, + db__data__2025__04__pb2.QueryResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Update(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/Update', + db__data__2025__04__pb2.UpdateRequest.SerializeToString, + db__data__2025__04__pb2.UpdateResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DescribeIndexStats(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/DescribeIndexStats', + db__data__2025__04__pb2.DescribeIndexStatsRequest.SerializeToString, + db__data__2025__04__pb2.DescribeIndexStatsResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ListNamespaces(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/ListNamespaces', + db__data__2025__04__pb2.ListNamespacesRequest.SerializeToString, + db__data__2025__04__pb2.ListNamespacesResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DescribeNamespace(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/DescribeNamespace', + db__data__2025__04__pb2.DescribeNamespaceRequest.SerializeToString, + db__data__2025__04__pb2.NamespaceDescription.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DeleteNamespace(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/VectorService/DeleteNamespace', + db__data__2025__04__pb2.DeleteNamespaceRequest.SerializeToString, + db__data__2025__04__pb2.DeleteResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/pinecone/grpc/__init__.py b/pinecone/grpc/__init__.py index 307b2a20..92c10d0f 100644 --- a/pinecone/grpc/__init__.py +++ b/pinecone/grpc/__init__.py @@ -51,12 +51,14 @@ from pinecone.db_data.dataclasses import Vector, SparseValues -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import ( +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import ( Vector as GRPCVector, SparseValues as GRPCSparseValues, DeleteResponse as GRPCDeleteResponse, ) +from pinecone.core.openapi.db_data.models import ListNamespacesResponse + __all__ = [ "GRPCIndex", "PineconeGRPC", @@ -67,4 +69,5 @@ "Vector", "SparseValues", "PineconeGrpcFuture", + "ListNamespacesResponse", ] diff --git a/pinecone/grpc/grpc_runner.py b/pinecone/grpc/grpc_runner.py index 253a6b33..cc2e35d5 100644 --- a/pinecone/grpc/grpc_runner.py +++ b/pinecone/grpc/grpc_runner.py @@ -10,6 +10,7 @@ from pinecone.exceptions.exceptions import PineconeException from grpc import CallCredentials, Compression from google.protobuf.message import Message +from pinecone.openapi_support.api_version import API_VERSION class GrpcRunner: @@ -21,6 +22,7 @@ def __init__(self, index_name: str, config: Config, grpc_config: GRPCClientConfi "api-key": config.api_key, "service-name": index_name, "client-version": CLIENT_VERSION, + "x-pinecone-api-version": API_VERSION, } if self.grpc_client_config.additional_metadata: self.fixed_metadata.update(self.grpc_client_config.additional_metadata) diff --git a/pinecone/grpc/index_grpc.py b/pinecone/grpc/index_grpc.py index dae3db37..7548c7a6 100644 --- a/pinecone/grpc/index_grpc.py +++ b/pinecone/grpc/index_grpc.py @@ -4,6 +4,7 @@ from google.protobuf import json_format from pinecone.utils.tqdm import tqdm +from pinecone.utils import require_kwargs from concurrent.futures import as_completed, Future @@ -15,6 +16,8 @@ parse_upsert_response, parse_update_response, parse_delete_response, + parse_namespace_description, + parse_list_namespaces_response, ) from .vector_factory_grpc import VectorFactoryGRPC from .sparse_values_factory import SparseValuesFactory @@ -23,9 +26,11 @@ FetchResponse, QueryResponse, IndexDescription as DescribeIndexStatsResponse, + NamespaceDescription, + ListNamespacesResponse, ) from pinecone.db_control.models.list_response import ListResponse as SimpleListResponse, Pagination -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import ( +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import ( Vector as GRPCVector, QueryVector as GRPCQueryVector, UpsertRequest, @@ -39,10 +44,13 @@ DeleteResponse, UpdateResponse, SparseValues as GRPCSparseValues, + DescribeNamespaceRequest, + DeleteNamespaceRequest, + ListNamespacesRequest, ) -from pinecone import Vector, SparseValues +from pinecone import Vector from pinecone.db_data.query_results_aggregator import QueryNamespacesResults, QueryResultsAggregator -from pinecone.core.grpc.protos.db_data_2025_01_pb2_grpc import VectorServiceStub +from pinecone.core.grpc.protos.db_data_2025_04_pb2_grpc import VectorServiceStub from .base import GRPCIndexBase from .future import PineconeGrpcFuture from ..db_data.types import ( @@ -54,7 +62,7 @@ ) -__all__ = ["GRPCIndex", "GRPCVector", "GRPCQueryVector", "GRPCSparseValues"] +__all__ = ["GRPCIndex", "GRPCVector", "GRPCQueryVector", "GRPCSparseValues", "NamespaceDescription", "ListNamespacesResponse"] _logger = logging.getLogger(__name__) """ :meta private: """ @@ -358,7 +366,7 @@ def query( include_values: Optional[bool] = None, include_metadata: Optional[bool] = None, sparse_vector: Optional[ - Union[SparseValues, GRPCSparseValues, SparseVectorTypedDict] + Union[GRPCSparseValues, SparseVectorTypedDict] ] = None, async_req: Optional[bool] = False, **kwargs, @@ -681,6 +689,142 @@ def describe_index_stats( json_response = json_format.MessageToDict(response) return parse_stats_response(json_response) + @require_kwargs + def describe_namespace( + self, namespace: str, **kwargs + ) -> NamespaceDescription: + """ + The describe_namespace operation returns information about a specific namespace, + including the total number of vectors in the namespace. + + Examples: + + .. code-block:: python + + >>> index.describe_namespace(namespace='my_namespace') + + Args: + namespace (str): The namespace to describe. + + Returns: NamespaceDescription object which contains information about the namespace. + """ + timeout = kwargs.pop("timeout", None) + request = DescribeNamespaceRequest(namespace=namespace) + response = self.runner.run(self.stub.DescribeNamespace, request, timeout=timeout) + return parse_namespace_description(response) + + @require_kwargs + def delete_namespace( + self, namespace: str, **kwargs + ) -> Dict[str, Any]: + """ + The delete_namespace operation deletes a namespace from an index. + This operation is irreversible and will permanently delete all data in the namespace. + + Examples: + + .. code-block:: python + + >>> index.delete_namespace(namespace='my_namespace') + + Args: + namespace (str): The namespace to delete. + + Returns: Empty dictionary indicating successful deletion. + """ + timeout = kwargs.pop("timeout", None) + request = DeleteNamespaceRequest(namespace=namespace) + response = self.runner.run(self.stub.DeleteNamespace, request, timeout=timeout) + return parse_delete_response(response) + + @require_kwargs + def list_namespaces_paginated( + self, + limit: Optional[int] = None, + pagination_token: Optional[str] = None, + **kwargs, + ) -> ListNamespacesResponse: + """ + The list_namespaces_paginated operation returns a list of all namespaces in a serverless index. + It returns namespaces in a paginated form, with a pagination token to fetch the next page of results. + + Examples: + + .. code-block:: python + + >>> results = index.list_namespaces_paginated(limit=10) + >>> [ns.name for ns in results.namespaces] + ['namespace1', 'namespace2', 'namespace3'] + >>> results.pagination.next + eyJza2lwX3Bhc3QiOiI5OTMiLCJwcmVmaXgiOiI5OSJ9 + >>> next_results = index.list_namespaces_paginated(limit=10, pagination_token=results.pagination.next) + + Args: + limit (Optional[int]): The maximum number of namespaces to return. If unspecified, the server will use a default value. [optional] + pagination_token (Optional[str]): A token needed to fetch the next page of results. This token is returned + in the response if additional results are available. [optional] + + Returns: ListNamespacesResponse object which contains the list of namespaces and pagination information. + """ + args_dict = self._parse_non_empty_args( + [ + ("limit", limit), + ("pagination_token", pagination_token), + ] + ) + request = ListNamespacesRequest(**args_dict, **kwargs) + timeout = kwargs.pop("timeout", None) + response = self.runner.run(self.stub.ListNamespaces, request, timeout=timeout) + return parse_list_namespaces_response(response) + + @require_kwargs + def list_namespaces(self, limit: Optional[int] = None, **kwargs): + """ + The list_namespaces operation accepts all of the same arguments as list_namespaces_paginated, and returns a generator that yields + each namespace. It automatically handles pagination tokens on your behalf. + + Args: + limit (Optional[int]): The maximum number of namespaces to fetch in each network call. If unspecified, the server will use a default value. [optional] + + Returns: + Returns a generator that yields each namespace. It automatically handles pagination tokens on your behalf so you can + easily iterate over all results. The ``list_namespaces`` method accepts all of the same arguments as list_namespaces_paginated + + Examples: + + .. code-block:: python + + >>> for namespace in index.list_namespaces(): + >>> print(namespace.name) + namespace1 + namespace2 + namespace3 + + You can convert the generator into a list by wrapping the generator in a call to the built-in ``list`` function: + + .. code-block:: python + + namespaces = list(index.list_namespaces()) + + You should be cautious with this approach because it will fetch all namespaces at once, which could be a large number + of network calls and a lot of memory to hold the results. + """ + done = False + while not done: + try: + results = self.list_namespaces_paginated(limit=limit, **kwargs) + except Exception as e: + raise e + + if results.namespaces and len(results.namespaces) > 0: + for namespace in results.namespaces: + yield namespace + + if results.pagination and results.pagination.next: + kwargs.update({"pagination_token": results.pagination.next}) + else: + done = True + @staticmethod def _parse_non_empty_args(args: List[Tuple[str, Any]]) -> Dict[str, Any]: return {arg_name: val for arg_name, val in args if val is not None} diff --git a/pinecone/grpc/sparse_values_factory.py b/pinecone/grpc/sparse_values_factory.py index 5bb14685..06aa8e67 100644 --- a/pinecone/grpc/sparse_values_factory.py +++ b/pinecone/grpc/sparse_values_factory.py @@ -6,7 +6,7 @@ from ..db_data import SparseValuesTypeError, SparseValuesMissingKeysError from ..db_data.types import SparseVectorTypedDict -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import SparseValues as GRPCSparseValues +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import SparseValues as GRPCSparseValues from pinecone.core.openapi.db_data.models import SparseValues as OpenApiSparseValues from pinecone import SparseValues diff --git a/pinecone/grpc/utils.py b/pinecone/grpc/utils.py index 92f2ce15..1be98f5b 100644 --- a/pinecone/grpc/utils.py +++ b/pinecone/grpc/utils.py @@ -13,6 +13,9 @@ IndexDescription as DescribeIndexStatsResponse, UpsertResponse, NamespaceSummary, + NamespaceDescription, + ListNamespacesResponse, + Pagination, ) from pinecone.db_data.dataclasses import FetchResponse @@ -126,3 +129,37 @@ def parse_stats_response(response: dict): total_vector_count=total_vector_count, _check_type=False, ) + + +def parse_namespace_description(response: Message) -> NamespaceDescription: + json_response = json_format.MessageToDict(response) + return NamespaceDescription( + name=json_response.get("name", ""), + record_count=json_response.get("recordCount", 0), + _check_type=False, + ) + + +def parse_list_namespaces_response(response: Message) -> ListNamespacesResponse: + json_response = json_format.MessageToDict(response) + + namespaces = [] + for ns in json_response.get("namespaces", []): + namespaces.append(NamespaceDescription( + name=ns.get("name", ""), + record_count=ns.get("recordCount", 0), + _check_type=False, + )) + + pagination = None + if "pagination" in json_response and json_response["pagination"]: + pagination = Pagination( + next=json_response["pagination"].get("next", ""), + _check_type=False, + ) + + return ListNamespacesResponse( + namespaces=namespaces, + pagination=pagination, + _check_type=False, + ) diff --git a/pinecone/grpc/vector_factory_grpc.py b/pinecone/grpc/vector_factory_grpc.py index 22efd269..3af3add8 100644 --- a/pinecone/grpc/vector_factory_grpc.py +++ b/pinecone/grpc/vector_factory_grpc.py @@ -17,7 +17,7 @@ from ..db_data.types import VectorTuple, VectorTypedDict from .sparse_values_factory import SparseValuesFactory -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import ( +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import ( Vector as GRPCVector, SparseValues as GRPCSparseValues, ) diff --git a/tests/integration/data/test_namespace.py b/tests/integration/data/test_namespace.py index 459bfdfd..6c2099ee 100644 --- a/tests/integration/data/test_namespace.py +++ b/tests/integration/data/test_namespace.py @@ -1,9 +1,6 @@ -import os import time import logging -import pytest - from pinecone import NamespaceDescription logger = logging.getLogger(__name__) @@ -43,9 +40,6 @@ def delete_all_namespaces(index): except Exception as e: logger.error(f"Error in delete_all_namespaces: {e}") -@pytest.mark.skipif( - os.getenv("USE_GRPC") == "true", reason="Disable until grpc namespaces support is added" -) class TestNamespaceOperations: def test_describe_namespace(self, idx): """Test describing a namespace""" diff --git a/tests/integration/data_grpc_futures/stub_backend.py b/tests/integration/data_grpc_futures/stub_backend.py index d8983db9..85f400ea 100644 --- a/tests/integration/data_grpc_futures/stub_backend.py +++ b/tests/integration/data_grpc_futures/stub_backend.py @@ -2,8 +2,8 @@ import grpc import logging from concurrent import futures -import pinecone.core.grpc.protos.db_data_2025_01_pb2 as pb2 -import pinecone.core.grpc.protos.db_data_2025_01_pb2_grpc as pb2_grpc +import pinecone.core.grpc.protos.db_data_2025_04_pb2 as pb2 +import pinecone.core.grpc.protos.db_data_2025_04_pb2_grpc as pb2_grpc logger = logging.getLogger(__name__) diff --git a/tests/unit_grpc/test_grpc_index_describe_index_stats.py b/tests/unit_grpc/test_grpc_index_describe_index_stats.py index 0599badb..554fbd40 100644 --- a/tests/unit_grpc/test_grpc_index_describe_index_stats.py +++ b/tests/unit_grpc/test_grpc_index_describe_index_stats.py @@ -1,6 +1,6 @@ from pinecone import Config from pinecone.grpc import GRPCIndex -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import DescribeIndexStatsRequest +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import DescribeIndexStatsRequest from pinecone.grpc.utils import dict_to_proto_struct diff --git a/tests/unit_grpc/test_grpc_index_fetch.py b/tests/unit_grpc/test_grpc_index_fetch.py index ca5c9e7d..97291fe0 100644 --- a/tests/unit_grpc/test_grpc_index_fetch.py +++ b/tests/unit_grpc/test_grpc_index_fetch.py @@ -1,6 +1,6 @@ from pinecone import Config from pinecone.grpc import GRPCIndex -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import FetchRequest +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import FetchRequest class TestGrpcIndexFetch: diff --git a/tests/unit_grpc/test_grpc_index_namespace.py b/tests/unit_grpc/test_grpc_index_namespace.py new file mode 100644 index 00000000..2ae25f14 --- /dev/null +++ b/tests/unit_grpc/test_grpc_index_namespace.py @@ -0,0 +1,78 @@ +from pinecone import Config +from pinecone.grpc import GRPCIndex +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import ( + DescribeNamespaceRequest, + DeleteNamespaceRequest, + ListNamespacesRequest, +) + + +class TestGrpcIndexNamespace: + def setup_method(self): + self.config = Config(api_key="test-api-key", host="foo.pinecone.io") + self.index = GRPCIndex( + config=self.config, index_name="example-name", _endpoint_override="test-endpoint" + ) + + def test_describe_namespace(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.describe_namespace("test_namespace") + self.index.runner.run.assert_called_once_with( + self.index.stub.DescribeNamespace, + DescribeNamespaceRequest(namespace="test_namespace"), + timeout=None + ) + + def test_describe_namespace_with_timeout(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.describe_namespace("test_namespace", timeout=30) + self.index.runner.run.assert_called_once_with( + self.index.stub.DescribeNamespace, + DescribeNamespaceRequest(namespace="test_namespace"), + timeout=30 + ) + + def test_delete_namespace(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.delete_namespace("test_namespace") + self.index.runner.run.assert_called_once_with( + self.index.stub.DeleteNamespace, + DeleteNamespaceRequest(namespace="test_namespace"), + timeout=None + ) + + def test_delete_namespace_with_timeout(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.delete_namespace("test_namespace", timeout=30) + self.index.runner.run.assert_called_once_with( + self.index.stub.DeleteNamespace, + DeleteNamespaceRequest(namespace="test_namespace"), + timeout=30 + ) + + def test_list_namespaces_paginated(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.list_namespaces_paginated(limit=10, pagination_token="token123") + self.index.runner.run.assert_called_once_with( + self.index.stub.ListNamespaces, + ListNamespacesRequest(limit=10, pagination_token="token123"), + timeout=None + ) + + def test_list_namespaces_paginated_with_timeout(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.list_namespaces_paginated(limit=10, timeout=30) + self.index.runner.run.assert_called_once_with( + self.index.stub.ListNamespaces, + ListNamespacesRequest(limit=10), + timeout=30 + ) + + def test_list_namespaces_paginated_no_args(self, mocker): + mocker.patch.object(self.index.runner, "run", autospec=True) + self.index.list_namespaces_paginated() + self.index.runner.run.assert_called_once_with( + self.index.stub.ListNamespaces, + ListNamespacesRequest(), + timeout=None + ) \ No newline at end of file diff --git a/tests/unit_grpc/test_grpc_index_query.py b/tests/unit_grpc/test_grpc_index_query.py index 2dbc846b..d237aa98 100644 --- a/tests/unit_grpc/test_grpc_index_query.py +++ b/tests/unit_grpc/test_grpc_index_query.py @@ -2,7 +2,7 @@ from pinecone import Config from pinecone.grpc import GRPCIndex -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import QueryRequest +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import QueryRequest from pinecone.grpc.utils import dict_to_proto_struct diff --git a/tests/unit_grpc/test_grpc_index_update.py b/tests/unit_grpc/test_grpc_index_update.py index f9abd958..207cd09e 100644 --- a/tests/unit_grpc/test_grpc_index_update.py +++ b/tests/unit_grpc/test_grpc_index_update.py @@ -1,6 +1,6 @@ from pinecone import Config from pinecone.grpc import GRPCIndex -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import UpdateRequest +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import UpdateRequest from pinecone.grpc.utils import dict_to_proto_struct diff --git a/tests/unit_grpc/test_grpc_index_upsert.py b/tests/unit_grpc/test_grpc_index_upsert.py index 987053f7..1a65da1c 100644 --- a/tests/unit_grpc/test_grpc_index_upsert.py +++ b/tests/unit_grpc/test_grpc_index_upsert.py @@ -6,7 +6,7 @@ from pinecone import Config from pinecone.grpc import GRPCIndex -from pinecone.core.grpc.protos.db_data_2025_01_pb2 import ( +from pinecone.core.grpc.protos.db_data_2025_04_pb2 import ( Vector, UpsertRequest, UpsertResponse, From 443344d1ac2dc68c64a77cf749b6f87306dde7a1 Mon Sep 17 00:00:00 2001 From: rohanshah18 Date: Thu, 26 Jun 2025 13:46:24 -0400 Subject: [PATCH 2/4] fix unit tests --- pinecone/grpc/index_grpc.py | 2 +- tests/unit_grpc/test_grpc_index_namespace.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/pinecone/grpc/index_grpc.py b/pinecone/grpc/index_grpc.py index 7548c7a6..cd18ceb7 100644 --- a/pinecone/grpc/index_grpc.py +++ b/pinecone/grpc/index_grpc.py @@ -772,8 +772,8 @@ def list_namespaces_paginated( ("pagination_token", pagination_token), ] ) - request = ListNamespacesRequest(**args_dict, **kwargs) timeout = kwargs.pop("timeout", None) + request = ListNamespacesRequest(**args_dict, **kwargs) response = self.runner.run(self.stub.ListNamespaces, request, timeout=timeout) return parse_list_namespaces_response(response) diff --git a/tests/unit_grpc/test_grpc_index_namespace.py b/tests/unit_grpc/test_grpc_index_namespace.py index 2ae25f14..d44dd4e8 100644 --- a/tests/unit_grpc/test_grpc_index_namespace.py +++ b/tests/unit_grpc/test_grpc_index_namespace.py @@ -16,7 +16,7 @@ def setup_method(self): def test_describe_namespace(self, mocker): mocker.patch.object(self.index.runner, "run", autospec=True) - self.index.describe_namespace("test_namespace") + self.index.describe_namespace(namespace="test_namespace") self.index.runner.run.assert_called_once_with( self.index.stub.DescribeNamespace, DescribeNamespaceRequest(namespace="test_namespace"), @@ -25,7 +25,7 @@ def test_describe_namespace(self, mocker): def test_describe_namespace_with_timeout(self, mocker): mocker.patch.object(self.index.runner, "run", autospec=True) - self.index.describe_namespace("test_namespace", timeout=30) + self.index.describe_namespace(namespace="test_namespace", timeout=30) self.index.runner.run.assert_called_once_with( self.index.stub.DescribeNamespace, DescribeNamespaceRequest(namespace="test_namespace"), @@ -34,7 +34,7 @@ def test_describe_namespace_with_timeout(self, mocker): def test_delete_namespace(self, mocker): mocker.patch.object(self.index.runner, "run", autospec=True) - self.index.delete_namespace("test_namespace") + self.index.delete_namespace(namespace="test_namespace") self.index.runner.run.assert_called_once_with( self.index.stub.DeleteNamespace, DeleteNamespaceRequest(namespace="test_namespace"), @@ -43,7 +43,7 @@ def test_delete_namespace(self, mocker): def test_delete_namespace_with_timeout(self, mocker): mocker.patch.object(self.index.runner, "run", autospec=True) - self.index.delete_namespace("test_namespace", timeout=30) + self.index.delete_namespace(namespace="test_namespace", timeout=30) self.index.runner.run.assert_called_once_with( self.index.stub.DeleteNamespace, DeleteNamespaceRequest(namespace="test_namespace"), From c8e0e748f427b158d9a4000f5f2a2c0f29b9b55a Mon Sep 17 00:00:00 2001 From: rohanshah18 Date: Thu, 26 Jun 2025 14:49:34 -0400 Subject: [PATCH 3/4] add SparseValues back --- pinecone/grpc/index_grpc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pinecone/grpc/index_grpc.py b/pinecone/grpc/index_grpc.py index cd18ceb7..83fce126 100644 --- a/pinecone/grpc/index_grpc.py +++ b/pinecone/grpc/index_grpc.py @@ -48,7 +48,7 @@ DeleteNamespaceRequest, ListNamespacesRequest, ) -from pinecone import Vector +from pinecone import Vector, SparseValues from pinecone.db_data.query_results_aggregator import QueryNamespacesResults, QueryResultsAggregator from pinecone.core.grpc.protos.db_data_2025_04_pb2_grpc import VectorServiceStub from .base import GRPCIndexBase @@ -366,7 +366,7 @@ def query( include_values: Optional[bool] = None, include_metadata: Optional[bool] = None, sparse_vector: Optional[ - Union[GRPCSparseValues, SparseVectorTypedDict] + Union[SparseValues, GRPCSparseValues, SparseVectorTypedDict] ] = None, async_req: Optional[bool] = False, **kwargs, From 57a87d5c24a96d13c5e3845822717de2aad3e41b Mon Sep 17 00:00:00 2001 From: rohanshah18 Date: Fri, 27 Jun 2025 12:03:43 -0400 Subject: [PATCH 4/4] remove old proto files --- .../core/grpc/protos/db_data_2025_01_pb2.py | 130 -------- .../core/grpc/protos/db_data_2025_01_pb2.pyi | 259 --------------- .../grpc/protos/db_data_2025_01_pb2_grpc.py | 309 ------------------ 3 files changed, 698 deletions(-) delete mode 100644 pinecone/core/grpc/protos/db_data_2025_01_pb2.py delete mode 100644 pinecone/core/grpc/protos/db_data_2025_01_pb2.pyi delete mode 100644 pinecone/core/grpc/protos/db_data_2025_01_pb2_grpc.py diff --git a/pinecone/core/grpc/protos/db_data_2025_01_pb2.py b/pinecone/core/grpc/protos/db_data_2025_01_pb2.py deleted file mode 100644 index ab0491f8..00000000 --- a/pinecone/core/grpc/protos/db_data_2025_01_pb2.py +++ /dev/null @@ -1,130 +0,0 @@ -# -*- coding: utf-8 -*- -# Generated by the protocol buffer compiler. DO NOT EDIT! -# NO CHECKED-IN PROTOBUF GENCODE -# source: db_data_2025-01.proto -# Protobuf Python Version: 5.29.0 -"""Generated protocol buffer code.""" -from google.protobuf import descriptor as _descriptor -from google.protobuf import descriptor_pool as _descriptor_pool -from google.protobuf import runtime_version as _runtime_version -from google.protobuf import symbol_database as _symbol_database -from google.protobuf.internal import builder as _builder -_runtime_version.ValidateProtobufRuntimeVersion( - _runtime_version.Domain.PUBLIC, - 5, - 29, - 0, - '', - 'db_data_2025-01.proto' -) -# @@protoc_insertion_point(imports) - -_sym_db = _symbol_database.Default() - - -from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2 -from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2 -from google.api import field_behavior_pb2 as google_dot_api_dot_field__behavior__pb2 - - -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x64\x62_data_2025-01.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/api/field_behavior.proto\"J\n\x0cSparseValues\x12\x1d\n\x07indices\x18\x01 \x03(\rB\x03\xe0\x41\x02R\x07indices\x12\x1b\n\x06values\x18\x02 \x03(\x02\x42\x03\xe0\x41\x02R\x06values\"\x9e\x01\n\x06Vector\x12\x13\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x02R\x02id\x12\x16\n\x06values\x18\x02 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x04 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12\x33\n\x08metadata\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\"\xba\x01\n\x0cScoredVector\x12\x13\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x02R\x02id\x12\x14\n\x05score\x18\x02 \x01(\x02R\x05score\x12\x16\n\x06values\x18\x03 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x05 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12\x33\n\x08metadata\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\"\xa1\x01\n\x0cRequestUnion\x12(\n\x06upsert\x18\x01 \x01(\x0b\x32\x0e.UpsertRequestH\x00R\x06upsert\x12(\n\x06\x64\x65lete\x18\x02 \x01(\x0b\x32\x0e.DeleteRequestH\x00R\x06\x64\x65lete\x12(\n\x06update\x18\x03 \x01(\x0b\x32\x0e.UpdateRequestH\x00R\x06updateB\x13\n\x11RequestUnionInner\"U\n\rUpsertRequest\x12&\n\x07vectors\x18\x01 \x03(\x0b\x32\x07.VectorB\x03\xe0\x41\x02R\x07vectors\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\"7\n\x0eUpsertResponse\x12%\n\x0eupserted_count\x18\x01 \x01(\rR\rupsertedCount\"\x8f\x01\n\rDeleteRequest\x12\x10\n\x03ids\x18\x01 \x03(\tR\x03ids\x12\x1d\n\ndelete_all\x18\x02 \x01(\x08R\tdeleteAll\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12/\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\"\x10\n\x0e\x44\x65leteResponse\"C\n\x0c\x46\x65tchRequest\x12\x15\n\x03ids\x18\x01 \x03(\tB\x03\xe0\x41\x02R\x03ids\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\"\xd6\x01\n\rFetchResponse\x12\x35\n\x07vectors\x18\x01 \x03(\x0b\x32\x1b.FetchResponse.VectorsEntryR\x07vectors\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\x12!\n\x05usage\x18\x03 \x01(\x0b\x32\x06.UsageH\x00R\x05usage\x88\x01\x01\x1a\x43\n\x0cVectorsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x1d\n\x05value\x18\x02 \x01(\x0b\x32\x07.VectorR\x05value:\x02\x38\x01\x42\x08\n\x06_usage\"\xbd\x01\n\x0bListRequest\x12\x1b\n\x06prefix\x18\x01 \x01(\tH\x00R\x06prefix\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\rH\x01R\x05limit\x88\x01\x01\x12.\n\x10pagination_token\x18\x03 \x01(\tH\x02R\x0fpaginationToken\x88\x01\x01\x12\x1c\n\tnamespace\x18\x04 \x01(\tR\tnamespaceB\t\n\x07_prefixB\x08\n\x06_limitB\x13\n\x11_pagination_token\" \n\nPagination\x12\x12\n\x04next\x18\x01 \x01(\tR\x04next\"\x1a\n\x08ListItem\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\"\xbf\x01\n\x0cListResponse\x12#\n\x07vectors\x18\x01 \x03(\x0b\x32\t.ListItemR\x07vectors\x12\x30\n\npagination\x18\x02 \x01(\x0b\x32\x0b.PaginationH\x00R\npagination\x88\x01\x01\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12!\n\x05usage\x18\x04 \x01(\x0b\x32\x06.UsageH\x01R\x05usage\x88\x01\x01\x42\r\n\x0b_paginationB\x08\n\x06_usage\"\xbd\x01\n\x0bQueryVector\x12\x16\n\x06values\x18\x01 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x05 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12\x13\n\x05top_k\x18\x02 \x01(\rR\x04topK\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12/\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\"\xd1\x02\n\x0cQueryRequest\x12\x1c\n\tnamespace\x18\x01 \x01(\tR\tnamespace\x12\x18\n\x05top_k\x18\x02 \x01(\rB\x03\xe0\x41\x02R\x04topK\x12/\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\x12%\n\x0einclude_values\x18\x04 \x01(\x08R\rincludeValues\x12)\n\x10include_metadata\x18\x05 \x01(\x08R\x0fincludeMetadata\x12*\n\x07queries\x18\x06 \x03(\x0b\x32\x0c.QueryVectorB\x02\x18\x01R\x07queries\x12\x16\n\x06vector\x18\x07 \x03(\x02R\x06vector\x12\x32\n\rsparse_vector\x18\t \x01(\x0b\x32\r.SparseValuesR\x0csparseVector\x12\x0e\n\x02id\x18\x08 \x01(\tR\x02id\"[\n\x12SingleQueryResults\x12\'\n\x07matches\x18\x01 \x03(\x0b\x32\r.ScoredVectorR\x07matches\x12\x1c\n\tnamespace\x18\x02 \x01(\tR\tnamespace\"\xb6\x01\n\rQueryResponse\x12\x31\n\x07results\x18\x01 \x03(\x0b\x32\x13.SingleQueryResultsB\x02\x18\x01R\x07results\x12\'\n\x07matches\x18\x02 \x03(\x0b\x32\r.ScoredVectorR\x07matches\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12!\n\x05usage\x18\x04 \x01(\x0b\x32\x06.UsageH\x00R\x05usage\x88\x01\x01\x42\x08\n\x06_usage\":\n\x05Usage\x12\"\n\nread_units\x18\x01 \x01(\rH\x00R\treadUnits\x88\x01\x01\x42\r\n\x0b_read_units\"\xca\x01\n\rUpdateRequest\x12\x13\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x02R\x02id\x12\x16\n\x06values\x18\x02 \x03(\x02R\x06values\x12\x32\n\rsparse_values\x18\x05 \x01(\x0b\x32\r.SparseValuesR\x0csparseValues\x12:\n\x0cset_metadata\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x0bsetMetadata\x12\x1c\n\tnamespace\x18\x04 \x01(\tR\tnamespace\"\x10\n\x0eUpdateResponse\"L\n\x19\x44\x65scribeIndexStatsRequest\x12/\n\x06\x66ilter\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x06\x66ilter\"5\n\x10NamespaceSummary\x12!\n\x0cvector_count\x18\x01 \x01(\rR\x0bvectorCount\"\x9f\x03\n\x1a\x44\x65scribeIndexStatsResponse\x12K\n\nnamespaces\x18\x01 \x03(\x0b\x32+.DescribeIndexStatsResponse.NamespacesEntryR\nnamespaces\x12!\n\tdimension\x18\x02 \x01(\rH\x00R\tdimension\x88\x01\x01\x12%\n\x0eindex_fullness\x18\x03 \x01(\x02R\rindexFullness\x12,\n\x12total_vector_count\x18\x04 \x01(\rR\x10totalVectorCount\x12\x1b\n\x06metric\x18\x05 \x01(\tH\x01R\x06metric\x88\x01\x01\x12$\n\x0bvector_type\x18\x06 \x01(\tH\x02R\nvectorType\x88\x01\x01\x1aP\n\x0fNamespacesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\'\n\x05value\x18\x02 \x01(\x0b\x32\x11.NamespaceSummaryR\x05value:\x02\x38\x01\x42\x0c\n\n_dimensionB\t\n\x07_metricB\x0e\n\x0c_vector_type2\xb9\x04\n\rVectorService\x12\x45\n\x06Upsert\x12\x0e.UpsertRequest\x1a\x0f.UpsertResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/vectors/upsert:\x01*\x12X\n\x06\x44\x65lete\x12\x0e.DeleteRequest\x1a\x0f.DeleteResponse\"-\x82\xd3\xe4\x93\x02\'\"\x0f/vectors/delete:\x01*Z\x11*\x0f/vectors/delete\x12>\n\x05\x46\x65tch\x12\r.FetchRequest\x1a\x0e.FetchResponse\"\x16\x82\xd3\xe4\x93\x02\x10\x12\x0e/vectors/fetch\x12:\n\x04List\x12\x0c.ListRequest\x1a\r.ListResponse\"\x15\x82\xd3\xe4\x93\x02\x0f\x12\r/vectors/list\x12\x39\n\x05Query\x12\r.QueryRequest\x1a\x0e.QueryResponse\"\x11\x82\xd3\xe4\x93\x02\x0b\"\x06/query:\x01*\x12\x45\n\x06Update\x12\x0e.UpdateRequest\x1a\x0f.UpdateResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/vectors/update:\x01*\x12\x88\x01\n\x12\x44\x65scribeIndexStats\x12\x1a.DescribeIndexStatsRequest\x1a\x1b.DescribeIndexStatsResponse\"9\x82\xd3\xe4\x93\x02\x33\"\x15/describe_index_stats:\x01*Z\x17\x12\x15/describe_index_statsBS\n\x11io.pinecone.protoP\x01Z None: ... - -class Vector(_message.Message): - __slots__ = ("id", "values", "sparse_values", "metadata") - ID_FIELD_NUMBER: _ClassVar[int] - VALUES_FIELD_NUMBER: _ClassVar[int] - SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] - METADATA_FIELD_NUMBER: _ClassVar[int] - id: str - values: _containers.RepeatedScalarFieldContainer[float] - sparse_values: SparseValues - metadata: _struct_pb2.Struct - def __init__(self, id: _Optional[str] = ..., values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... - -class ScoredVector(_message.Message): - __slots__ = ("id", "score", "values", "sparse_values", "metadata") - ID_FIELD_NUMBER: _ClassVar[int] - SCORE_FIELD_NUMBER: _ClassVar[int] - VALUES_FIELD_NUMBER: _ClassVar[int] - SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] - METADATA_FIELD_NUMBER: _ClassVar[int] - id: str - score: float - values: _containers.RepeatedScalarFieldContainer[float] - sparse_values: SparseValues - metadata: _struct_pb2.Struct - def __init__(self, id: _Optional[str] = ..., score: _Optional[float] = ..., values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... - -class RequestUnion(_message.Message): - __slots__ = ("upsert", "delete", "update") - UPSERT_FIELD_NUMBER: _ClassVar[int] - DELETE_FIELD_NUMBER: _ClassVar[int] - UPDATE_FIELD_NUMBER: _ClassVar[int] - upsert: UpsertRequest - delete: DeleteRequest - update: UpdateRequest - def __init__(self, upsert: _Optional[_Union[UpsertRequest, _Mapping]] = ..., delete: _Optional[_Union[DeleteRequest, _Mapping]] = ..., update: _Optional[_Union[UpdateRequest, _Mapping]] = ...) -> None: ... - -class UpsertRequest(_message.Message): - __slots__ = ("vectors", "namespace") - VECTORS_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - vectors: _containers.RepeatedCompositeFieldContainer[Vector] - namespace: str - def __init__(self, vectors: _Optional[_Iterable[_Union[Vector, _Mapping]]] = ..., namespace: _Optional[str] = ...) -> None: ... - -class UpsertResponse(_message.Message): - __slots__ = ("upserted_count",) - UPSERTED_COUNT_FIELD_NUMBER: _ClassVar[int] - upserted_count: int - def __init__(self, upserted_count: _Optional[int] = ...) -> None: ... - -class DeleteRequest(_message.Message): - __slots__ = ("ids", "delete_all", "namespace", "filter") - IDS_FIELD_NUMBER: _ClassVar[int] - DELETE_ALL_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - FILTER_FIELD_NUMBER: _ClassVar[int] - ids: _containers.RepeatedScalarFieldContainer[str] - delete_all: bool - namespace: str - filter: _struct_pb2.Struct - def __init__(self, ids: _Optional[_Iterable[str]] = ..., delete_all: bool = ..., namespace: _Optional[str] = ..., filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... - -class DeleteResponse(_message.Message): - __slots__ = () - def __init__(self) -> None: ... - -class FetchRequest(_message.Message): - __slots__ = ("ids", "namespace") - IDS_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - ids: _containers.RepeatedScalarFieldContainer[str] - namespace: str - def __init__(self, ids: _Optional[_Iterable[str]] = ..., namespace: _Optional[str] = ...) -> None: ... - -class FetchResponse(_message.Message): - __slots__ = ("vectors", "namespace", "usage") - class VectorsEntry(_message.Message): - __slots__ = ("key", "value") - KEY_FIELD_NUMBER: _ClassVar[int] - VALUE_FIELD_NUMBER: _ClassVar[int] - key: str - value: Vector - def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Vector, _Mapping]] = ...) -> None: ... - VECTORS_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - USAGE_FIELD_NUMBER: _ClassVar[int] - vectors: _containers.MessageMap[str, Vector] - namespace: str - usage: Usage - def __init__(self, vectors: _Optional[_Mapping[str, Vector]] = ..., namespace: _Optional[str] = ..., usage: _Optional[_Union[Usage, _Mapping]] = ...) -> None: ... - -class ListRequest(_message.Message): - __slots__ = ("prefix", "limit", "pagination_token", "namespace") - PREFIX_FIELD_NUMBER: _ClassVar[int] - LIMIT_FIELD_NUMBER: _ClassVar[int] - PAGINATION_TOKEN_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - prefix: str - limit: int - pagination_token: str - namespace: str - def __init__(self, prefix: _Optional[str] = ..., limit: _Optional[int] = ..., pagination_token: _Optional[str] = ..., namespace: _Optional[str] = ...) -> None: ... - -class Pagination(_message.Message): - __slots__ = ("next",) - NEXT_FIELD_NUMBER: _ClassVar[int] - next: str - def __init__(self, next: _Optional[str] = ...) -> None: ... - -class ListItem(_message.Message): - __slots__ = ("id",) - ID_FIELD_NUMBER: _ClassVar[int] - id: str - def __init__(self, id: _Optional[str] = ...) -> None: ... - -class ListResponse(_message.Message): - __slots__ = ("vectors", "pagination", "namespace", "usage") - VECTORS_FIELD_NUMBER: _ClassVar[int] - PAGINATION_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - USAGE_FIELD_NUMBER: _ClassVar[int] - vectors: _containers.RepeatedCompositeFieldContainer[ListItem] - pagination: Pagination - namespace: str - usage: Usage - def __init__(self, vectors: _Optional[_Iterable[_Union[ListItem, _Mapping]]] = ..., pagination: _Optional[_Union[Pagination, _Mapping]] = ..., namespace: _Optional[str] = ..., usage: _Optional[_Union[Usage, _Mapping]] = ...) -> None: ... - -class QueryVector(_message.Message): - __slots__ = ("values", "sparse_values", "top_k", "namespace", "filter") - VALUES_FIELD_NUMBER: _ClassVar[int] - SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] - TOP_K_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - FILTER_FIELD_NUMBER: _ClassVar[int] - values: _containers.RepeatedScalarFieldContainer[float] - sparse_values: SparseValues - top_k: int - namespace: str - filter: _struct_pb2.Struct - def __init__(self, values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., top_k: _Optional[int] = ..., namespace: _Optional[str] = ..., filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... - -class QueryRequest(_message.Message): - __slots__ = ("namespace", "top_k", "filter", "include_values", "include_metadata", "queries", "vector", "sparse_vector", "id") - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - TOP_K_FIELD_NUMBER: _ClassVar[int] - FILTER_FIELD_NUMBER: _ClassVar[int] - INCLUDE_VALUES_FIELD_NUMBER: _ClassVar[int] - INCLUDE_METADATA_FIELD_NUMBER: _ClassVar[int] - QUERIES_FIELD_NUMBER: _ClassVar[int] - VECTOR_FIELD_NUMBER: _ClassVar[int] - SPARSE_VECTOR_FIELD_NUMBER: _ClassVar[int] - ID_FIELD_NUMBER: _ClassVar[int] - namespace: str - top_k: int - filter: _struct_pb2.Struct - include_values: bool - include_metadata: bool - queries: _containers.RepeatedCompositeFieldContainer[QueryVector] - vector: _containers.RepeatedScalarFieldContainer[float] - sparse_vector: SparseValues - id: str - def __init__(self, namespace: _Optional[str] = ..., top_k: _Optional[int] = ..., filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., include_values: bool = ..., include_metadata: bool = ..., queries: _Optional[_Iterable[_Union[QueryVector, _Mapping]]] = ..., vector: _Optional[_Iterable[float]] = ..., sparse_vector: _Optional[_Union[SparseValues, _Mapping]] = ..., id: _Optional[str] = ...) -> None: ... - -class SingleQueryResults(_message.Message): - __slots__ = ("matches", "namespace") - MATCHES_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - matches: _containers.RepeatedCompositeFieldContainer[ScoredVector] - namespace: str - def __init__(self, matches: _Optional[_Iterable[_Union[ScoredVector, _Mapping]]] = ..., namespace: _Optional[str] = ...) -> None: ... - -class QueryResponse(_message.Message): - __slots__ = ("results", "matches", "namespace", "usage") - RESULTS_FIELD_NUMBER: _ClassVar[int] - MATCHES_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - USAGE_FIELD_NUMBER: _ClassVar[int] - results: _containers.RepeatedCompositeFieldContainer[SingleQueryResults] - matches: _containers.RepeatedCompositeFieldContainer[ScoredVector] - namespace: str - usage: Usage - def __init__(self, results: _Optional[_Iterable[_Union[SingleQueryResults, _Mapping]]] = ..., matches: _Optional[_Iterable[_Union[ScoredVector, _Mapping]]] = ..., namespace: _Optional[str] = ..., usage: _Optional[_Union[Usage, _Mapping]] = ...) -> None: ... - -class Usage(_message.Message): - __slots__ = ("read_units",) - READ_UNITS_FIELD_NUMBER: _ClassVar[int] - read_units: int - def __init__(self, read_units: _Optional[int] = ...) -> None: ... - -class UpdateRequest(_message.Message): - __slots__ = ("id", "values", "sparse_values", "set_metadata", "namespace") - ID_FIELD_NUMBER: _ClassVar[int] - VALUES_FIELD_NUMBER: _ClassVar[int] - SPARSE_VALUES_FIELD_NUMBER: _ClassVar[int] - SET_METADATA_FIELD_NUMBER: _ClassVar[int] - NAMESPACE_FIELD_NUMBER: _ClassVar[int] - id: str - values: _containers.RepeatedScalarFieldContainer[float] - sparse_values: SparseValues - set_metadata: _struct_pb2.Struct - namespace: str - def __init__(self, id: _Optional[str] = ..., values: _Optional[_Iterable[float]] = ..., sparse_values: _Optional[_Union[SparseValues, _Mapping]] = ..., set_metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., namespace: _Optional[str] = ...) -> None: ... - -class UpdateResponse(_message.Message): - __slots__ = () - def __init__(self) -> None: ... - -class DescribeIndexStatsRequest(_message.Message): - __slots__ = ("filter",) - FILTER_FIELD_NUMBER: _ClassVar[int] - filter: _struct_pb2.Struct - def __init__(self, filter: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ... - -class NamespaceSummary(_message.Message): - __slots__ = ("vector_count",) - VECTOR_COUNT_FIELD_NUMBER: _ClassVar[int] - vector_count: int - def __init__(self, vector_count: _Optional[int] = ...) -> None: ... - -class DescribeIndexStatsResponse(_message.Message): - __slots__ = ("namespaces", "dimension", "index_fullness", "total_vector_count", "metric", "vector_type") - class NamespacesEntry(_message.Message): - __slots__ = ("key", "value") - KEY_FIELD_NUMBER: _ClassVar[int] - VALUE_FIELD_NUMBER: _ClassVar[int] - key: str - value: NamespaceSummary - def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[NamespaceSummary, _Mapping]] = ...) -> None: ... - NAMESPACES_FIELD_NUMBER: _ClassVar[int] - DIMENSION_FIELD_NUMBER: _ClassVar[int] - INDEX_FULLNESS_FIELD_NUMBER: _ClassVar[int] - TOTAL_VECTOR_COUNT_FIELD_NUMBER: _ClassVar[int] - METRIC_FIELD_NUMBER: _ClassVar[int] - VECTOR_TYPE_FIELD_NUMBER: _ClassVar[int] - namespaces: _containers.MessageMap[str, NamespaceSummary] - dimension: int - index_fullness: float - total_vector_count: int - metric: str - vector_type: str - def __init__(self, namespaces: _Optional[_Mapping[str, NamespaceSummary]] = ..., dimension: _Optional[int] = ..., index_fullness: _Optional[float] = ..., total_vector_count: _Optional[int] = ..., metric: _Optional[str] = ..., vector_type: _Optional[str] = ...) -> None: ... diff --git a/pinecone/core/grpc/protos/db_data_2025_01_pb2_grpc.py b/pinecone/core/grpc/protos/db_data_2025_01_pb2_grpc.py deleted file mode 100644 index 4c80e120..00000000 --- a/pinecone/core/grpc/protos/db_data_2025_01_pb2_grpc.py +++ /dev/null @@ -1,309 +0,0 @@ -# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! -"""Client and server classes corresponding to protobuf-defined services.""" -import grpc - -import pinecone.core.grpc.protos.db_data_2025_01_pb2 as db__data__2025__01__pb2 - - -class VectorServiceStub(object): - """The `VectorService` interface is exposed by Pinecone's vector index services. - This service could also be called a `gRPC` service or a `REST`-like api. - """ - - def __init__(self, channel) -> None: - """Constructor. - - Args: - channel: A grpc.Channel. - """ - self.Upsert = channel.unary_unary( - '/VectorService/Upsert', - request_serializer=db__data__2025__01__pb2.UpsertRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.UpsertResponse.FromString, - ) - self.Delete = channel.unary_unary( - '/VectorService/Delete', - request_serializer=db__data__2025__01__pb2.DeleteRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.DeleteResponse.FromString, - ) - self.Fetch = channel.unary_unary( - '/VectorService/Fetch', - request_serializer=db__data__2025__01__pb2.FetchRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.FetchResponse.FromString, - ) - self.List = channel.unary_unary( - '/VectorService/List', - request_serializer=db__data__2025__01__pb2.ListRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.ListResponse.FromString, - ) - self.Query = channel.unary_unary( - '/VectorService/Query', - request_serializer=db__data__2025__01__pb2.QueryRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.QueryResponse.FromString, - ) - self.Update = channel.unary_unary( - '/VectorService/Update', - request_serializer=db__data__2025__01__pb2.UpdateRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.UpdateResponse.FromString, - ) - self.DescribeIndexStats = channel.unary_unary( - '/VectorService/DescribeIndexStats', - request_serializer=db__data__2025__01__pb2.DescribeIndexStatsRequest.SerializeToString, - response_deserializer=db__data__2025__01__pb2.DescribeIndexStatsResponse.FromString, - ) - - -class VectorServiceServicer(object): - """The `VectorService` interface is exposed by Pinecone's vector index services. - This service could also be called a `gRPC` service or a `REST`-like api. - """ - - def Upsert(self, request, context): - """Upsert vectors - - Writes vectors into a namespace. If a new value is upserted for an existing vector ID, it will overwrite the previous value. - - For guidance and examples, see [Upsert data](https://docs.pinecone.io/guides/data/upsert-data). - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def Delete(self, request, context): - """Delete vectors - - Delete vectors by id from a single namespace. - - For guidance and examples, see [Delete data](https://docs.pinecone.io/guides/data/delete-data). - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def Fetch(self, request, context): - """Fetch vectors - - Look up and returns vectors by ID from a single namespace. The returned vectors include the vector data and/or metadata. - - For guidance and examples, see [Fetch data](https://docs.pinecone.io/guides/data/fetch-data). - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def List(self, request, context): - """List vector IDs - - List the IDs of vectors in a single namespace of a serverless index. An optional prefix can be passed to limit the results to IDs with a common prefix. - - This returns up to 100 IDs at a time by default in sorted order (bitwise/"C" collation). If the `limit` parameter is set, `list` returns up to that number of IDs instead. Whenever there are additional IDs to return, the response also includes a `pagination_token` that you can use to get the next batch of IDs. When the response does not include a `pagination_token`, there are no more IDs to return. - - For guidance and examples, see [List record IDs](https://docs.pinecone.io/guides/data/list-record-ids). - - **Note:** This is supported only for serverless indexes. - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def Query(self, request, context): - """Query vectors - - Searches a namespace, using a query vector. It retrieves the ids of the most similar items in a namespace, along with their similarity scores. - - For guidance and examples, see [Query data](https://docs.pinecone.io/guides/data/query-data). - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def Update(self, request, context): - """Update a vector - - Update a vector in a namespace. If a value is included, it will overwrite the previous value. If a `set_metadata` is included, the values of the fields specified in it will be added or overwrite the previous value. - - For guidance and examples, see [Update data](https://docs.pinecone.io/guides/data/update-data). - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def DescribeIndexStats(self, request, context): - """Get index stats - - Return statistics about the contents of an index, including the vector count per namespace, the number of dimensions, and the index fullness. - - Serverless indexes scale automatically as needed, so index fullness is relevant only for pod-based indexes. - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - -def add_VectorServiceServicer_to_server(servicer, server): - rpc_method_handlers = { - 'Upsert': grpc.unary_unary_rpc_method_handler( - servicer.Upsert, - request_deserializer=db__data__2025__01__pb2.UpsertRequest.FromString, - response_serializer=db__data__2025__01__pb2.UpsertResponse.SerializeToString, - ), - 'Delete': grpc.unary_unary_rpc_method_handler( - servicer.Delete, - request_deserializer=db__data__2025__01__pb2.DeleteRequest.FromString, - response_serializer=db__data__2025__01__pb2.DeleteResponse.SerializeToString, - ), - 'Fetch': grpc.unary_unary_rpc_method_handler( - servicer.Fetch, - request_deserializer=db__data__2025__01__pb2.FetchRequest.FromString, - response_serializer=db__data__2025__01__pb2.FetchResponse.SerializeToString, - ), - 'List': grpc.unary_unary_rpc_method_handler( - servicer.List, - request_deserializer=db__data__2025__01__pb2.ListRequest.FromString, - response_serializer=db__data__2025__01__pb2.ListResponse.SerializeToString, - ), - 'Query': grpc.unary_unary_rpc_method_handler( - servicer.Query, - request_deserializer=db__data__2025__01__pb2.QueryRequest.FromString, - response_serializer=db__data__2025__01__pb2.QueryResponse.SerializeToString, - ), - 'Update': grpc.unary_unary_rpc_method_handler( - servicer.Update, - request_deserializer=db__data__2025__01__pb2.UpdateRequest.FromString, - response_serializer=db__data__2025__01__pb2.UpdateResponse.SerializeToString, - ), - 'DescribeIndexStats': grpc.unary_unary_rpc_method_handler( - servicer.DescribeIndexStats, - request_deserializer=db__data__2025__01__pb2.DescribeIndexStatsRequest.FromString, - response_serializer=db__data__2025__01__pb2.DescribeIndexStatsResponse.SerializeToString, - ), - } - generic_handler = grpc.method_handlers_generic_handler( - 'VectorService', rpc_method_handlers) - server.add_generic_rpc_handlers((generic_handler,)) - - - # This class is part of an EXPERIMENTAL API. -class VectorService(object): - """The `VectorService` interface is exposed by Pinecone's vector index services. - This service could also be called a `gRPC` service or a `REST`-like api. - """ - - @staticmethod - def Upsert(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/Upsert', - db__data__2025__01__pb2.UpsertRequest.SerializeToString, - db__data__2025__01__pb2.UpsertResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) - - @staticmethod - def Delete(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/Delete', - db__data__2025__01__pb2.DeleteRequest.SerializeToString, - db__data__2025__01__pb2.DeleteResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) - - @staticmethod - def Fetch(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/Fetch', - db__data__2025__01__pb2.FetchRequest.SerializeToString, - db__data__2025__01__pb2.FetchResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) - - @staticmethod - def List(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/List', - db__data__2025__01__pb2.ListRequest.SerializeToString, - db__data__2025__01__pb2.ListResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) - - @staticmethod - def Query(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/Query', - db__data__2025__01__pb2.QueryRequest.SerializeToString, - db__data__2025__01__pb2.QueryResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) - - @staticmethod - def Update(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/Update', - db__data__2025__01__pb2.UpdateRequest.SerializeToString, - db__data__2025__01__pb2.UpdateResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) - - @staticmethod - def DescribeIndexStats(request, - target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/VectorService/DescribeIndexStats', - db__data__2025__01__pb2.DescribeIndexStatsRequest.SerializeToString, - db__data__2025__01__pb2.DescribeIndexStatsResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata)