polign 0.1.0__py3-none-any.whl

polign/__init__.py

@@ -0,0 +1,66 @@
+"""polign — Python client for polign_db.
+
+HTTP client (zero dependencies):
+
+    from polign import Client
+    c = Client("http://localhost:23000")
+
+gRPC client (pip install polign[grpc]):
+
+    from polign import GrpcClient
+    c = GrpcClient("localhost:23001")
+
+Both expose the same operations: put, put_many, get, list, delete, search.
+"""
+
+from .client import Client
+from .errors import (
+    AuthenticationError,
+    ConflictError,
+    ConnectionError,
+    InvalidArgumentError,
+    NotEnabledError,
+    NotFoundError,
+    NotOwnerError,
+    PermissionDeniedError,
+    PolignError,
+    RateLimitError,
+    ServerError,
+    UnavailableError,
+)
+from .types import CollectionBackend, CollectionInfo, Fusion, Hit, Vector, VectorPage
+
+__version__ = "0.1.0rc8"
+
+__all__ = [
+    "Client",
+    "GrpcClient",
+    "Vector",
+    "Hit",
+    "Fusion",
+    "VectorPage",
+    "PolignError",
+    "ConnectionError",
+    "InvalidArgumentError",
+    "NotFoundError",
+    "AuthenticationError",
+    "PermissionDeniedError",
+    "RateLimitError",
+    "NotOwnerError",
+    "ServerError",
+    "ConflictError",
+    "NotEnabledError",
+    "UnavailableError",
+    "CollectionBackend",
+    "CollectionInfo",
+    "__version__",
+]
+
+
+def __getattr__(name):
+    # Lazy import so the base package works without grpcio installed.
+    if name == "GrpcClient":
+        from .grpc_client import GrpcClient
+
+        return GrpcClient
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

polign/_filter.py

@@ -0,0 +1,161 @@
+"""Converts the dict filter language into the wire FilterExpr tree.
+
+The gRPC client accepts the same metadata-filter dicts as the HTTP client
+(the syntax of docs/FILTERING.md) and converts them client-side into the
+``polign.v1.FilterExpr`` proto. The semantics mirror the server's JSON parser
+(internal/filter/json.go): bare values are equality, per-key operator objects
+($eq, $ne, $in, $gt, $gte, $lt, $lte, $exists) AND together with the four
+range operators merged into one range, and $and/$or/$not compose sub-filters.
+
+Only imported by the gRPC transport — needs the ``grpc`` extra for protobuf.
+"""
+
+from __future__ import annotations
+
+from typing import Any, Dict, List, Optional
+
+from . import errors
+from ._pb import vectordb_pb2 as pb
+
+_RANGE_OPS = {"$gt": "gt", "$gte": "gte", "$lt": "lt", "$lte": "lte"}
+
+
+def filter_expr_from_dict(obj: Optional[Dict[str, Any]]) -> Optional["pb.FilterExpr"]:
+    """Convert a filter dict to a FilterExpr, or None for no filter.
+
+    Raises :class:`polign.InvalidArgumentError` on a malformed filter, with
+    the same messages the server would produce.
+    """
+    if not obj:
+        return None
+    if not isinstance(obj, dict):
+        raise errors.InvalidArgumentError("filter: not a JSON object")
+    return _parse_object(obj)
+
+
+def _parse_object(obj: Dict[str, Any]) -> Optional["pb.FilterExpr"]:
+    exprs: List[pb.FilterExpr] = []
+    for key in sorted(obj):
+        val = obj[key]
+        if key in ("$and", "$or"):
+            children = _parse_object_list(key, val)
+            junction = pb.FilterJunction(exprs=children)
+            if key == "$and":
+                exprs.append(pb.FilterExpr(**{"and": junction}))
+            else:
+                exprs.append(pb.FilterExpr(**{"or": junction}))
+        elif key == "$not":
+            if not isinstance(val, dict):
+                raise errors.InvalidArgumentError("filter: $not takes a filter object")
+            child = _parse_object(val)
+            if child is None:
+                raise errors.InvalidArgumentError(
+                    "filter: $not takes a non-empty filter object"
+                )
+            exprs.append(pb.FilterExpr(**{"not": child}))
+        elif key.startswith("$"):
+            raise errors.InvalidArgumentError(f"filter: unknown operator {key!r}")
+        else:
+            exprs.append(_parse_field(key, val))
+    return _and_collapse(exprs)
+
+
+def _parse_object_list(op: str, val: Any) -> List["pb.FilterExpr"]:
+    if not isinstance(val, list) or not val:
+        raise errors.InvalidArgumentError(
+            f"filter: {op} takes a non-empty array of filter objects"
+        )
+    children = []
+    for item in val:
+        if not isinstance(item, dict):
+            raise errors.InvalidArgumentError(
+                f"filter: {op} takes a non-empty array of filter objects"
+            )
+        child = _parse_object(item)
+        if child is None:
+            raise errors.InvalidArgumentError(f"filter: {op}: empty filter object")
+        children.append(child)
+    return children
+
+
+def _parse_field(key: str, val: Any) -> "pb.FilterExpr":
+    if isinstance(val, dict):
+        return _parse_field_ops(key, val)
+    return _cond(pb.FilterCond(key=key, eq=_scalar(key, val)))
+
+
+def _parse_field_ops(key: str, ops: Dict[str, Any]) -> "pb.FilterExpr":
+    if not ops:
+        raise errors.InvalidArgumentError(f"filter: key {key!r}: empty operator object")
+    exprs: List[pb.FilterExpr] = []
+    bounds: Dict[str, str] = {}
+    numeric_set = lex_set = False
+    for op in sorted(ops):
+        val = ops[op]
+        if op == "$eq":
+            exprs.append(_cond(pb.FilterCond(key=key, eq=_scalar(key, val))))
+        elif op == "$ne":
+            eq = _cond(pb.FilterCond(key=key, eq=_scalar(key, val)))
+            exprs.append(pb.FilterExpr(**{"not": eq}))
+        elif op == "$in":
+            if not isinstance(val, list):
+                raise errors.InvalidArgumentError(f"filter: key {key!r}: $in takes an array")
+            values = [_scalar(key, item) for item in val]
+            exprs.append(_cond(pb.FilterCond(key=key, **{"in": pb.ValueList(values=values)})))
+        elif op in _RANGE_OPS:
+            if isinstance(val, str):
+                lex_set = True
+            elif isinstance(val, (int, float)) and not isinstance(val, bool):
+                numeric_set = True
+            else:
+                raise errors.InvalidArgumentError(
+                    f"filter: key {key!r}: {op} takes a string or number"
+                )
+            bounds[_RANGE_OPS[op]] = val if isinstance(val, str) else str(val)
+        elif op == "$exists":
+            if not isinstance(val, bool):
+                raise errors.InvalidArgumentError(
+                    f"filter: key {key!r}: $exists takes a boolean"
+                )
+            exprs.append(_cond(pb.FilterCond(key=key, exists=val)))
+        else:
+            raise errors.InvalidArgumentError(
+                f"filter: key {key!r}: unknown operator {op!r}"
+            )
+    if bounds:
+        if numeric_set and lex_set:
+            raise errors.InvalidArgumentError(
+                f"filter: key {key!r}: range bounds mix numbers and strings"
+            )
+        rng = pb.FilterRange(numeric=numeric_set, **bounds)
+        exprs.append(_cond(pb.FilterCond(key=key, range=rng)))
+    collapsed = _and_collapse(exprs)
+    assert collapsed is not None  # ops was non-empty
+    return collapsed
+
+
+def _and_collapse(exprs: List["pb.FilterExpr"]) -> Optional["pb.FilterExpr"]:
+    if not exprs:
+        return None
+    if len(exprs) == 1:
+        return exprs[0]
+    return pb.FilterExpr(**{"and": pb.FilterJunction(exprs=exprs)})
+
+
+def _cond(c: "pb.FilterCond") -> "pb.FilterExpr":
+    return pb.FilterExpr(cond=c)
+
+
+def _scalar(key: str, val: Any) -> str:
+    """A scalar's metadata string form: strings as-is, numbers by their
+    literal ("0.5"), booleans as "true"/"false" — matching what the HTTP
+    client's JSON encoding would send."""
+    if isinstance(val, bool):  # before int: bool subclasses int
+        return "true" if val else "false"
+    if isinstance(val, str):
+        return val
+    if isinstance(val, (int, float)):
+        return str(val)
+    raise errors.InvalidArgumentError(
+        f"filter: key {key!r}: expected a string, number or boolean"
+    )

polign/_pb/vectordb_pb2.py

@@ -0,0 +1,117 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: vectordb.proto
+# Protobuf Python Version: 6.31.1
+"""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,
+    6,
+    31,
+    1,
+    '',
+    'vectordb.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0evectordb.proto\x12\tpolign.v1\"t\n\x11\x43ollectionBackend\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x10\n\x08role_arn\x18\x02 \x01(\t\x12\x13\n\x0b\x65xternal_id\x18\x03 \x01(\t\x12\x0e\n\x06region\x18\x04 \x01(\t\x12\x1b\n\x13gcs_service_account\x18\x05 \x01(\t\"\xfe\x01\n\x0e\x43ollectionInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06status\x18\x02 \x01(\t\x12-\n\x07\x62\x61\x63kend\x18\x03 \x01(\x0b\x32\x1c.polign.v1.CollectionBackend\x12\x12\n\nbackend_id\x18\x04 \x01(\t\x12\x1d\n\x15verified_capabilities\x18\x05 \x03(\t\x12\x17\n\x0f\x63reated_at_unix\x18\x06 \x01(\x03\x12\x18\n\x10verified_at_unix\x18\x07 \x01(\x03\x12\x13\n\x0b\x63laim_token\x18\x08 \x01(\t\x12\x12\n\nclaim_path\x18\t \x01(\t\x12\x10\n\x08warnings\x18\n \x03(\t\"\\\n\x17\x43reateCollectionRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12-\n\x07\x62\x61\x63kend\x18\x02 \x01(\x0b\x32\x1c.polign.v1.CollectionBackend\"C\n\x18\x43reateCollectionResponse\x12\'\n\x04info\x18\x01 \x01(\x0b\x32\x19.polign.v1.CollectionInfo\"*\n\x14GetCollectionRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\"@\n\x15GetCollectionResponse\x12\'\n\x04info\x18\x01 \x01(\x0b\x32\x19.polign.v1.CollectionInfo\"\x18\n\x16ListCollectionsRequest\"I\n\x17ListCollectionsResponse\x12.\n\x0b\x63ollections\x18\x01 \x03(\x0b\x32\x19.polign.v1.CollectionInfo\"-\n\x17\x44\x65leteCollectionRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\"+\n\x18\x44\x65leteCollectionResponse\x12\x0f\n\x07\x64\x65leted\x18\x01 \x01(\x08\"-\n\x17VerifyCollectionRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\"C\n\x18VerifyCollectionResponse\x12\'\n\x04info\x18\x01 \x01(\x0b\x32\x19.polign.v1.CollectionInfo\"\x88\x01\n\x06Vector\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0e\n\x06values\x18\x02 \x03(\x02\x12\x31\n\x08metadata\x18\x03 \x03(\x0b\x32\x1f.polign.v1.Vector.MetadataEntry\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"I\n\x10PutVectorRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12!\n\x06vector\x18\x02 \x01(\x0b\x32\x11.polign.v1.Vector\"\x1f\n\x11PutVectorResponse\x12\n\n\x02id\x18\x01 \x01(\t\"K\n\x11PutVectorsRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\"\n\x07vectors\x18\x02 \x03(\x0b\x32\x11.polign.v1.Vector\"!\n\x12PutVectorsResponse\x12\x0b\n\x03ids\x18\x01 \x03(\t\"2\n\x10GetVectorRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\t\"6\n\x11GetVectorResponse\x12!\n\x06vector\x18\x01 \x01(\x0b\x32\x11.polign.v1.Vector\"4\n\x11GetVectorsRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0b\n\x03ids\x18\x02 \x03(\t\"8\n\x12GetVectorsResponse\x12\"\n\x07vectors\x18\x01 \x03(\x0b\x32\x11.polign.v1.Vector\"G\n\x12ListVectorsRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\r\n\x05limit\x18\x02 \x01(\x05\x12\x0e\n\x06offset\x18\x03 \x01(\x05\"H\n\x13ListVectorsResponse\x12\"\n\x07vectors\x18\x01 \x03(\x0b\x32\x11.polign.v1.Vector\x12\r\n\x05total\x18\x02 \x01(\x05\"5\n\x13\x44\x65leteVectorRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\t\"\'\n\x14\x44\x65leteVectorResponse\x12\x0f\n\x07\x64\x65leted\x18\x01 \x01(\x08\"\xc9\x02\n\x14SearchVectorsRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06values\x18\x02 \x03(\x02\x12\t\n\x01k\x18\x03 \x01(\x05\x12\n\n\x02\x65\x66\x18\x04 \x01(\x05\x12\x0c\n\x04\x63old\x18\x05 \x01(\x08\x12\x0e\n\x06nprobe\x18\x06 \x01(\x05\x12;\n\x06\x66ilter\x18\x07 \x03(\x0b\x32+.polign.v1.SearchVectorsRequest.FilterEntry\x12\x0c\n\x04text\x18\x08 \x01(\t\x12!\n\x06\x66usion\x18\t \x01(\x0b\x32\x11.polign.v1.Fusion\x12*\n\x0b\x66ilter_expr\x18\n \x01(\x0b\x32\x15.polign.v1.FilterExpr\x12\x0f\n\x07rescore\x18\x0b \x01(\x05\x1a-\n\x0b\x46ilterEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xb4\x01\n\nFilterExpr\x12(\n\x03\x61nd\x18\x01 \x01(\x0b\x32\x19.polign.v1.FilterJunctionH\x00\x12\'\n\x02or\x18\x02 \x01(\x0b\x32\x19.polign.v1.FilterJunctionH\x00\x12$\n\x03not\x18\x03 \x01(\x0b\x32\x15.polign.v1.FilterExprH\x00\x12%\n\x04\x63ond\x18\x04 \x01(\x0b\x32\x15.polign.v1.FilterCondH\x00\x42\x06\n\x04\x65xpr\"6\n\x0e\x46ilterJunction\x12$\n\x05\x65xprs\x18\x01 \x03(\x0b\x32\x15.polign.v1.FilterExpr\"\x8c\x01\n\nFilterCond\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x0c\n\x02\x65q\x18\x02 \x01(\tH\x00\x12\"\n\x02in\x18\x03 \x01(\x0b\x32\x14.polign.v1.ValueListH\x00\x12\'\n\x05range\x18\x04 \x01(\x0b\x32\x16.polign.v1.FilterRangeH\x00\x12\x10\n\x06\x65xists\x18\x05 \x01(\x08H\x00\x42\x04\n\x02op\"\x1b\n\tValueList\x12\x0e\n\x06values\x18\x01 \x03(\t\"\x82\x01\n\x0b\x46ilterRange\x12\x0f\n\x02gt\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x10\n\x03gte\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x0f\n\x02lt\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x10\n\x03lte\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x0f\n\x07numeric\x18\x05 \x01(\x08\x42\x05\n\x03_gtB\x06\n\x04_gteB\x05\n\x03_ltB\x06\n\x04_lte\"6\n\x06\x46usion\x12\x0e\n\x06method\x18\x01 \x01(\t\x12\r\n\x05\x61lpha\x18\x02 \x01(\x01\x12\r\n\x05rrf_k\x18\x03 \x01(\x05\"\x9f\x01\n\tSearchHit\x12\n\n\x02id\x18\x01 \x01(\t\x12\x10\n\x08\x64istance\x18\x02 \x01(\x02\x12\x34\n\x08metadata\x18\x03 \x03(\x0b\x32\".polign.v1.SearchHit.MetadataEntry\x12\r\n\x05score\x18\x04 \x01(\x02\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\";\n\x15SearchVectorsResponse\x12\"\n\x04hits\x18\x01 \x03(\x0b\x32\x14.polign.v1.SearchHit2\xe8\x07\n\x08VectorDB\x12\x46\n\tPutVector\x12\x1b.polign.v1.PutVectorRequest\x1a\x1c.polign.v1.PutVectorResponse\x12I\n\nPutVectors\x12\x1c.polign.v1.PutVectorsRequest\x1a\x1d.polign.v1.PutVectorsResponse\x12\x46\n\tGetVector\x12\x1b.polign.v1.GetVectorRequest\x1a\x1c.polign.v1.GetVectorResponse\x12I\n\nGetVectors\x12\x1c.polign.v1.GetVectorsRequest\x1a\x1d.polign.v1.GetVectorsResponse\x12L\n\x0bListVectors\x12\x1d.polign.v1.ListVectorsRequest\x1a\x1e.polign.v1.ListVectorsResponse\x12O\n\x0c\x44\x65leteVector\x12\x1e.polign.v1.DeleteVectorRequest\x1a\x1f.polign.v1.DeleteVectorResponse\x12R\n\rSearchVectors\x12\x1f.polign.v1.SearchVectorsRequest\x1a .polign.v1.SearchVectorsResponse\x12[\n\x10\x43reateCollection\x12\".polign.v1.CreateCollectionRequest\x1a#.polign.v1.CreateCollectionResponse\x12R\n\rGetCollection\x12\x1f.polign.v1.GetCollectionRequest\x1a .polign.v1.GetCollectionResponse\x12X\n\x0fListCollections\x12!.polign.v1.ListCollectionsRequest\x1a\".polign.v1.ListCollectionsResponse\x12[\n\x10\x44\x65leteCollection\x12\".polign.v1.DeleteCollectionRequest\x1a#.polign.v1.DeleteCollectionResponse\x12[\n\x10VerifyCollection\x12\".polign.v1.VerifyCollectionRequest\x1a#.polign.v1.VerifyCollectionResponseB,Z*github.com/Polign/polign_db/internal/pb;pbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'vectordb_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z*github.com/Polign/polign_db/internal/pb;pb'
+  _globals['_VECTOR_METADATAENTRY']._loaded_options = None
+  _globals['_VECTOR_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_SEARCHVECTORSREQUEST_FILTERENTRY']._loaded_options = None
+  _globals['_SEARCHVECTORSREQUEST_FILTERENTRY']._serialized_options = b'8\001'
+  _globals['_SEARCHHIT_METADATAENTRY']._loaded_options = None
+  _globals['_SEARCHHIT_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_COLLECTIONBACKEND']._serialized_start=29
+  _globals['_COLLECTIONBACKEND']._serialized_end=145
+  _globals['_COLLECTIONINFO']._serialized_start=148
+  _globals['_COLLECTIONINFO']._serialized_end=402
+  _globals['_CREATECOLLECTIONREQUEST']._serialized_start=404
+  _globals['_CREATECOLLECTIONREQUEST']._serialized_end=496
+  _globals['_CREATECOLLECTIONRESPONSE']._serialized_start=498
+  _globals['_CREATECOLLECTIONRESPONSE']._serialized_end=565
+  _globals['_GETCOLLECTIONREQUEST']._serialized_start=567
+  _globals['_GETCOLLECTIONREQUEST']._serialized_end=609
+  _globals['_GETCOLLECTIONRESPONSE']._serialized_start=611
+  _globals['_GETCOLLECTIONRESPONSE']._serialized_end=675
+  _globals['_LISTCOLLECTIONSREQUEST']._serialized_start=677
+  _globals['_LISTCOLLECTIONSREQUEST']._serialized_end=701
+  _globals['_LISTCOLLECTIONSRESPONSE']._serialized_start=703
+  _globals['_LISTCOLLECTIONSRESPONSE']._serialized_end=776
+  _globals['_DELETECOLLECTIONREQUEST']._serialized_start=778
+  _globals['_DELETECOLLECTIONREQUEST']._serialized_end=823
+  _globals['_DELETECOLLECTIONRESPONSE']._serialized_start=825
+  _globals['_DELETECOLLECTIONRESPONSE']._serialized_end=868
+  _globals['_VERIFYCOLLECTIONREQUEST']._serialized_start=870
+  _globals['_VERIFYCOLLECTIONREQUEST']._serialized_end=915
+  _globals['_VERIFYCOLLECTIONRESPONSE']._serialized_start=917
+  _globals['_VERIFYCOLLECTIONRESPONSE']._serialized_end=984
+  _globals['_VECTOR']._serialized_start=987
+  _globals['_VECTOR']._serialized_end=1123
+  _globals['_VECTOR_METADATAENTRY']._serialized_start=1076
+  _globals['_VECTOR_METADATAENTRY']._serialized_end=1123
+  _globals['_PUTVECTORREQUEST']._serialized_start=1125
+  _globals['_PUTVECTORREQUEST']._serialized_end=1198
+  _globals['_PUTVECTORRESPONSE']._serialized_start=1200
+  _globals['_PUTVECTORRESPONSE']._serialized_end=1231
+  _globals['_PUTVECTORSREQUEST']._serialized_start=1233
+  _globals['_PUTVECTORSREQUEST']._serialized_end=1308
+  _globals['_PUTVECTORSRESPONSE']._serialized_start=1310
+  _globals['_PUTVECTORSRESPONSE']._serialized_end=1343
+  _globals['_GETVECTORREQUEST']._serialized_start=1345
+  _globals['_GETVECTORREQUEST']._serialized_end=1395
+  _globals['_GETVECTORRESPONSE']._serialized_start=1397
+  _globals['_GETVECTORRESPONSE']._serialized_end=1451
+  _globals['_GETVECTORSREQUEST']._serialized_start=1453
+  _globals['_GETVECTORSREQUEST']._serialized_end=1505
+  _globals['_GETVECTORSRESPONSE']._serialized_start=1507
+  _globals['_GETVECTORSRESPONSE']._serialized_end=1563
+  _globals['_LISTVECTORSREQUEST']._serialized_start=1565
+  _globals['_LISTVECTORSREQUEST']._serialized_end=1636
+  _globals['_LISTVECTORSRESPONSE']._serialized_start=1638
+  _globals['_LISTVECTORSRESPONSE']._serialized_end=1710
+  _globals['_DELETEVECTORREQUEST']._serialized_start=1712
+  _globals['_DELETEVECTORREQUEST']._serialized_end=1765
+  _globals['_DELETEVECTORRESPONSE']._serialized_start=1767
+  _globals['_DELETEVECTORRESPONSE']._serialized_end=1806
+  _globals['_SEARCHVECTORSREQUEST']._serialized_start=1809
+  _globals['_SEARCHVECTORSREQUEST']._serialized_end=2138
+  _globals['_SEARCHVECTORSREQUEST_FILTERENTRY']._serialized_start=2093
+  _globals['_SEARCHVECTORSREQUEST_FILTERENTRY']._serialized_end=2138
+  _globals['_FILTEREXPR']._serialized_start=2141
+  _globals['_FILTEREXPR']._serialized_end=2321
+  _globals['_FILTERJUNCTION']._serialized_start=2323
+  _globals['_FILTERJUNCTION']._serialized_end=2377
+  _globals['_FILTERCOND']._serialized_start=2380
+  _globals['_FILTERCOND']._serialized_end=2520
+  _globals['_VALUELIST']._serialized_start=2522
+  _globals['_VALUELIST']._serialized_end=2549
+  _globals['_FILTERRANGE']._serialized_start=2552
+  _globals['_FILTERRANGE']._serialized_end=2682
+  _globals['_FUSION']._serialized_start=2684
+  _globals['_FUSION']._serialized_end=2738
+  _globals['_SEARCHHIT']._serialized_start=2741
+  _globals['_SEARCHHIT']._serialized_end=2900
+  _globals['_SEARCHHIT_METADATAENTRY']._serialized_start=1076
+  _globals['_SEARCHHIT_METADATAENTRY']._serialized_end=1123
+  _globals['_SEARCHVECTORSRESPONSE']._serialized_start=2902
+  _globals['_SEARCHVECTORSRESPONSE']._serialized_end=2961
+  _globals['_VECTORDB']._serialized_start=2964
+  _globals['_VECTORDB']._serialized_end=3964
+# @@protoc_insertion_point(module_scope)

polign/_pb/vectordb_pb2.pyi

@@ -0,0 +1,326 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class CollectionBackend(_message.Message):
+    __slots__ = ("uri", "role_arn", "external_id", "region", "gcs_service_account")
+    URI_FIELD_NUMBER: _ClassVar[int]
+    ROLE_ARN_FIELD_NUMBER: _ClassVar[int]
+    EXTERNAL_ID_FIELD_NUMBER: _ClassVar[int]
+    REGION_FIELD_NUMBER: _ClassVar[int]
+    GCS_SERVICE_ACCOUNT_FIELD_NUMBER: _ClassVar[int]
+    uri: str
+    role_arn: str
+    external_id: str
+    region: str
+    gcs_service_account: str
+    def __init__(self, uri: _Optional[str] = ..., role_arn: _Optional[str] = ..., external_id: _Optional[str] = ..., region: _Optional[str] = ..., gcs_service_account: _Optional[str] = ...) -> None: ...
+
+class CollectionInfo(_message.Message):
+    __slots__ = ("name", "status", "backend", "backend_id", "verified_capabilities", "created_at_unix", "verified_at_unix", "claim_token", "claim_path", "warnings")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    BACKEND_FIELD_NUMBER: _ClassVar[int]
+    BACKEND_ID_FIELD_NUMBER: _ClassVar[int]
+    VERIFIED_CAPABILITIES_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_UNIX_FIELD_NUMBER: _ClassVar[int]
+    VERIFIED_AT_UNIX_FIELD_NUMBER: _ClassVar[int]
+    CLAIM_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    CLAIM_PATH_FIELD_NUMBER: _ClassVar[int]
+    WARNINGS_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    status: str
+    backend: CollectionBackend
+    backend_id: str
+    verified_capabilities: _containers.RepeatedScalarFieldContainer[str]
+    created_at_unix: int
+    verified_at_unix: int
+    claim_token: str
+    claim_path: str
+    warnings: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, name: _Optional[str] = ..., status: _Optional[str] = ..., backend: _Optional[_Union[CollectionBackend, _Mapping]] = ..., backend_id: _Optional[str] = ..., verified_capabilities: _Optional[_Iterable[str]] = ..., created_at_unix: _Optional[int] = ..., verified_at_unix: _Optional[int] = ..., claim_token: _Optional[str] = ..., claim_path: _Optional[str] = ..., warnings: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class CreateCollectionRequest(_message.Message):
+    __slots__ = ("collection", "backend")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    BACKEND_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    backend: CollectionBackend
+    def __init__(self, collection: _Optional[str] = ..., backend: _Optional[_Union[CollectionBackend, _Mapping]] = ...) -> None: ...
+
+class CreateCollectionResponse(_message.Message):
+    __slots__ = ("info",)
+    INFO_FIELD_NUMBER: _ClassVar[int]
+    info: CollectionInfo
+    def __init__(self, info: _Optional[_Union[CollectionInfo, _Mapping]] = ...) -> None: ...
+
+class GetCollectionRequest(_message.Message):
+    __slots__ = ("collection",)
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    def __init__(self, collection: _Optional[str] = ...) -> None: ...
+
+class GetCollectionResponse(_message.Message):
+    __slots__ = ("info",)
+    INFO_FIELD_NUMBER: _ClassVar[int]
+    info: CollectionInfo
+    def __init__(self, info: _Optional[_Union[CollectionInfo, _Mapping]] = ...) -> None: ...
+
+class ListCollectionsRequest(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class ListCollectionsResponse(_message.Message):
+    __slots__ = ("collections",)
+    COLLECTIONS_FIELD_NUMBER: _ClassVar[int]
+    collections: _containers.RepeatedCompositeFieldContainer[CollectionInfo]
+    def __init__(self, collections: _Optional[_Iterable[_Union[CollectionInfo, _Mapping]]] = ...) -> None: ...
+
+class DeleteCollectionRequest(_message.Message):
+    __slots__ = ("collection",)
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    def __init__(self, collection: _Optional[str] = ...) -> None: ...
+
+class DeleteCollectionResponse(_message.Message):
+    __slots__ = ("deleted",)
+    DELETED_FIELD_NUMBER: _ClassVar[int]
+    deleted: bool
+    def __init__(self, deleted: bool = ...) -> None: ...
+
+class VerifyCollectionRequest(_message.Message):
+    __slots__ = ("collection",)
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    def __init__(self, collection: _Optional[str] = ...) -> None: ...
+
+class VerifyCollectionResponse(_message.Message):
+    __slots__ = ("info",)
+    INFO_FIELD_NUMBER: _ClassVar[int]
+    info: CollectionInfo
+    def __init__(self, info: _Optional[_Union[CollectionInfo, _Mapping]] = ...) -> None: ...
+
+class Vector(_message.Message):
+    __slots__ = ("id", "values", "metadata")
+    class MetadataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ID_FIELD_NUMBER: _ClassVar[int]
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    values: _containers.RepeatedScalarFieldContainer[float]
+    metadata: _containers.ScalarMap[str, str]
+    def __init__(self, id: _Optional[str] = ..., values: _Optional[_Iterable[float]] = ..., metadata: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class PutVectorRequest(_message.Message):
+    __slots__ = ("collection", "vector")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    VECTOR_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    vector: Vector
+    def __init__(self, collection: _Optional[str] = ..., vector: _Optional[_Union[Vector, _Mapping]] = ...) -> None: ...
+
+class PutVectorResponse(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    def __init__(self, id: _Optional[str] = ...) -> None: ...
+
+class PutVectorsRequest(_message.Message):
+    __slots__ = ("collection", "vectors")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    VECTORS_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    vectors: _containers.RepeatedCompositeFieldContainer[Vector]
+    def __init__(self, collection: _Optional[str] = ..., vectors: _Optional[_Iterable[_Union[Vector, _Mapping]]] = ...) -> None: ...
+
+class PutVectorsResponse(_message.Message):
+    __slots__ = ("ids",)
+    IDS_FIELD_NUMBER: _ClassVar[int]
+    ids: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, ids: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class GetVectorRequest(_message.Message):
+    __slots__ = ("collection", "id")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    ID_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    id: str
+    def __init__(self, collection: _Optional[str] = ..., id: _Optional[str] = ...) -> None: ...
+
+class GetVectorResponse(_message.Message):
+    __slots__ = ("vector",)
+    VECTOR_FIELD_NUMBER: _ClassVar[int]
+    vector: Vector
+    def __init__(self, vector: _Optional[_Union[Vector, _Mapping]] = ...) -> None: ...
+
+class GetVectorsRequest(_message.Message):
+    __slots__ = ("collection", "ids")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    IDS_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    ids: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, collection: _Optional[str] = ..., ids: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class GetVectorsResponse(_message.Message):
+    __slots__ = ("vectors",)
+    VECTORS_FIELD_NUMBER: _ClassVar[int]
+    vectors: _containers.RepeatedCompositeFieldContainer[Vector]
+    def __init__(self, vectors: _Optional[_Iterable[_Union[Vector, _Mapping]]] = ...) -> None: ...
+
+class ListVectorsRequest(_message.Message):
+    __slots__ = ("collection", "limit", "offset")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    limit: int
+    offset: int
+    def __init__(self, collection: _Optional[str] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ...) -> None: ...
+
+class ListVectorsResponse(_message.Message):
+    __slots__ = ("vectors", "total")
+    VECTORS_FIELD_NUMBER: _ClassVar[int]
+    TOTAL_FIELD_NUMBER: _ClassVar[int]
+    vectors: _containers.RepeatedCompositeFieldContainer[Vector]
+    total: int
+    def __init__(self, vectors: _Optional[_Iterable[_Union[Vector, _Mapping]]] = ..., total: _Optional[int] = ...) -> None: ...
+
+class DeleteVectorRequest(_message.Message):
+    __slots__ = ("collection", "id")
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    ID_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    id: str
+    def __init__(self, collection: _Optional[str] = ..., id: _Optional[str] = ...) -> None: ...
+
+class DeleteVectorResponse(_message.Message):
+    __slots__ = ("deleted",)
+    DELETED_FIELD_NUMBER: _ClassVar[int]
+    deleted: bool
+    def __init__(self, deleted: bool = ...) -> None: ...
+
+class SearchVectorsRequest(_message.Message):
+    __slots__ = ("collection", "values", "k", "ef", "cold", "nprobe", "filter", "text", "fusion", "filter_expr", "rescore")
+    class FilterEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    COLLECTION_FIELD_NUMBER: _ClassVar[int]
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    K_FIELD_NUMBER: _ClassVar[int]
+    EF_FIELD_NUMBER: _ClassVar[int]
+    COLD_FIELD_NUMBER: _ClassVar[int]
+    NPROBE_FIELD_NUMBER: _ClassVar[int]
+    FILTER_FIELD_NUMBER: _ClassVar[int]
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    FUSION_FIELD_NUMBER: _ClassVar[int]
+    FILTER_EXPR_FIELD_NUMBER: _ClassVar[int]
+    RESCORE_FIELD_NUMBER: _ClassVar[int]
+    collection: str
+    values: _containers.RepeatedScalarFieldContainer[float]
+    k: int
+    ef: int
+    cold: bool
+    nprobe: int
+    filter: _containers.ScalarMap[str, str]
+    text: str
+    fusion: Fusion
+    filter_expr: FilterExpr
+    rescore: int
+    def __init__(self, collection: _Optional[str] = ..., values: _Optional[_Iterable[float]] = ..., k: _Optional[int] = ..., ef: _Optional[int] = ..., cold: bool = ..., nprobe: _Optional[int] = ..., filter: _Optional[_Mapping[str, str]] = ..., text: _Optional[str] = ..., fusion: _Optional[_Union[Fusion, _Mapping]] = ..., filter_expr: _Optional[_Union[FilterExpr, _Mapping]] = ..., rescore: _Optional[int] = ...) -> None: ...
+
+class FilterExpr(_message.Message):
+    __slots__ = ("cond",)
+    AND_FIELD_NUMBER: _ClassVar[int]
+    OR_FIELD_NUMBER: _ClassVar[int]
+    NOT_FIELD_NUMBER: _ClassVar[int]
+    COND_FIELD_NUMBER: _ClassVar[int]
+    cond: FilterCond
+    def __init__(self, cond: _Optional[_Union[FilterCond, _Mapping]] = ..., **kwargs) -> None: ...
+
+class FilterJunction(_message.Message):
+    __slots__ = ("exprs",)
+    EXPRS_FIELD_NUMBER: _ClassVar[int]
+    exprs: _containers.RepeatedCompositeFieldContainer[FilterExpr]
+    def __init__(self, exprs: _Optional[_Iterable[_Union[FilterExpr, _Mapping]]] = ...) -> None: ...
+
+class FilterCond(_message.Message):
+    __slots__ = ("key", "eq", "range", "exists")
+    KEY_FIELD_NUMBER: _ClassVar[int]
+    EQ_FIELD_NUMBER: _ClassVar[int]
+    IN_FIELD_NUMBER: _ClassVar[int]
+    RANGE_FIELD_NUMBER: _ClassVar[int]
+    EXISTS_FIELD_NUMBER: _ClassVar[int]
+    key: str
+    eq: str
+    range: FilterRange
+    exists: bool
+    def __init__(self, key: _Optional[str] = ..., eq: _Optional[str] = ..., range: _Optional[_Union[FilterRange, _Mapping]] = ..., exists: bool = ..., **kwargs) -> None: ...
+
+class ValueList(_message.Message):
+    __slots__ = ("values",)
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    values: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, values: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class FilterRange(_message.Message):
+    __slots__ = ("gt", "gte", "lt", "lte", "numeric")
+    GT_FIELD_NUMBER: _ClassVar[int]
+    GTE_FIELD_NUMBER: _ClassVar[int]
+    LT_FIELD_NUMBER: _ClassVar[int]
+    LTE_FIELD_NUMBER: _ClassVar[int]
+    NUMERIC_FIELD_NUMBER: _ClassVar[int]
+    gt: str
+    gte: str
+    lt: str
+    lte: str
+    numeric: bool
+    def __init__(self, gt: _Optional[str] = ..., gte: _Optional[str] = ..., lt: _Optional[str] = ..., lte: _Optional[str] = ..., numeric: bool = ...) -> None: ...
+
+class Fusion(_message.Message):
+    __slots__ = ("method", "alpha", "rrf_k")
+    METHOD_FIELD_NUMBER: _ClassVar[int]
+    ALPHA_FIELD_NUMBER: _ClassVar[int]
+    RRF_K_FIELD_NUMBER: _ClassVar[int]
+    method: str
+    alpha: float
+    rrf_k: int
+    def __init__(self, method: _Optional[str] = ..., alpha: _Optional[float] = ..., rrf_k: _Optional[int] = ...) -> None: ...
+
+class SearchHit(_message.Message):
+    __slots__ = ("id", "distance", "metadata", "score")
+    class MetadataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ID_FIELD_NUMBER: _ClassVar[int]
+    DISTANCE_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    SCORE_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    distance: float
+    metadata: _containers.ScalarMap[str, str]
+    score: float
+    def __init__(self, id: _Optional[str] = ..., distance: _Optional[float] = ..., metadata: _Optional[_Mapping[str, str]] = ..., score: _Optional[float] = ...) -> None: ...
+
+class SearchVectorsResponse(_message.Message):
+    __slots__ = ("hits",)
+    HITS_FIELD_NUMBER: _ClassVar[int]
+    hits: _containers.RepeatedCompositeFieldContainer[SearchHit]
+    def __init__(self, hits: _Optional[_Iterable[_Union[SearchHit, _Mapping]]] = ...) -> None: ...