nucliadb-utils 5.0.0.post941__py3-none-any.whl → 5.0.0.post944__py3-none-any.whl

nucliadb_utils/aiopynecone/client.py

@@ -523,3 +523,31 @@ async def async_batchify(async_iterable: AsyncIterable, batch_size: int):
             batch = []
     if batch:
         yield batch
+
+
+class FilterOperator:
+    """
+    Filter operators for metadata queries.
+    https://docs.pinecone.io/guides/data/filter-with-metadata#metadata-query-language
+    """
+
+    EQUALS = "$eq"
+    NOT_EQUALS = "$ne"
+    GREATER_THAN = "$gt"
+    GREATER_THAN_OR_EQUAL = "$gte"
+    LESS_THAN = "$lt"
+    LESS_THAN_OR_EQUAL = "$lte"
+    IN = "$in"
+    NOT_IN = "$nin"
+    EXISTS = "$exists"
+
+
+class LogicalOperator:
+    """
+    Logical operators for metadata queries.
+    https://docs.pinecone.io/guides/data/filter-with-metadata#metadata-query-language
+    """
+
+    AND = "$and"
+    OR = "$or"
+    NOT = "$not"

nucliadb_utils/aiopynecone/exceptions.py

@@ -60,6 +60,14 @@ class PineconeRateLimitError(PineconeAPIError):
     pass
 
 
+class MetadataTooLargeError(ValueError):
+    """
+    Raised when the metadata of a vector to be upserted is too large.
+    """
+
+    pass
+
+
 def raise_for_status(operation: str, response: httpx.Response):
     try:
         response.raise_for_status()

nucliadb_utils/aiopynecone/models.py

@@ -24,6 +24,8 @@ import pydantic
 from pydantic import BaseModel, Field, field_validator
 from typing_extensions import Annotated
 
+from nucliadb_utils.aiopynecone.exceptions import MetadataTooLargeError
+
 KILO_BYTE = 1024
 MAX_METADATA_SIZE = 40 * KILO_BYTE
 MAX_INDEX_NAME_LENGTH = 45
@@ -71,7 +73,7 @@ class Vector(BaseModel):
     def validate_metadata_size(cls, value):
         json_value = json.dumps(value)
         if len(json_value) > MAX_METADATA_SIZE:
-            raise ValueError("metadata size is too large")
+            raise MetadataTooLargeError(f"metadata size is too large: {len(json_value)} bytes")
         return value
 
 

{nucliadb_utils-5.0.0.post941.dist-info → nucliadb_utils-5.0.0.post944.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 5.0.0.post941
+Version: 5.0.0.post944
 Home-page: https://nuclia.com
 License: BSD
 Classifier: Development Status :: 4 - Beta
@@ -23,8 +23,8 @@ Requires-Dist: PyNaCl
 Requires-Dist: pyjwt >=2.4.0
 Requires-Dist: memorylru >=1.1.2
 Requires-Dist: mrflagly
-Requires-Dist: nucliadb-protos >=5.0.0.post941
-Requires-Dist: nucliadb-telemetry >=5.0.0.post941
+Requires-Dist: nucliadb-protos >=5.0.0.post944
+Requires-Dist: nucliadb-telemetry >=5.0.0.post944
 Provides-Extra: cache
 Requires-Dist: redis >=4.3.4 ; extra == 'cache'
 Requires-Dist: orjson >=3.6.7 ; extra == 'cache'

{nucliadb_utils-5.0.0.post941.dist-info → nucliadb_utils-5.0.0.post944.dist-info}/RECORD

@@ -18,9 +18,9 @@ nucliadb_utils/store.py,sha256=kQ35HemE0v4_Qg6xVqNIJi8vSFAYQtwI3rDtMsNy62Y,890
 nucliadb_utils/transaction.py,sha256=mwcI3aIHAvU5KOGqd_Uz_d1XQzXhk_-NWY8NqU1lfb0,7307
 nucliadb_utils/utilities.py,sha256=trAez4j2TYYT5n1qpimyHOEAnsQqDqD2qCFgVSDi184,16747
 nucliadb_utils/aiopynecone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/aiopynecone/client.py,sha256=JsFmHmPjs73JU3ZQY3LmrpOma4oBAhuXxIHgQ987Yp8,19464
-nucliadb_utils/aiopynecone/exceptions.py,sha256=hFhq-UEY4slqNWjObXr_LPnRf_AQ1vpcG4SF2XRFd1E,2873
-nucliadb_utils/aiopynecone/models.py,sha256=B_ihJhHZGp3ivQVUxhV49uoUnHe1PLDKxTgHNbHgSS0,2937
+nucliadb_utils/aiopynecone/client.py,sha256=pl3WWGpjwoc-qzgDChuIE2LU7Mopu1aKL9iPHkr8QD0,20069
+nucliadb_utils/aiopynecone/exceptions.py,sha256=EEE0XoGs1zIB5yOJ_fy6yoG4uIb4cWIawYdJeNe4eDo,3012
+nucliadb_utils/aiopynecone/models.py,sha256=tUugQ-ACQAyT-lhsfLwKAOgb6ilLchLwZfnRV7xwFck,3047
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/audit/audit.py,sha256=kpjiTLgWtLXA5DNgJcmeIaodel11Mz0Y-JfQt-yfhis,2641
 nucliadb_utils/audit/basic.py,sha256=LwCCvUAwy8LNzYdrw3NebxqNdUumCEyRQ244dJ93JfE,3190
@@ -64,8 +64,8 @@ nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQ
 nucliadb_utils/tests/local.py,sha256=7nuP8EFUAiA8ZH50R1iPV9EUXBySQxOanVm3Zht_e0g,1835
 nucliadb_utils/tests/nats.py,sha256=xqpww4jZjTKY9oPGlJdDJG67L3FIBQsa9qDHxILR8r8,7687
 nucliadb_utils/tests/s3.py,sha256=IdMxK_cNdSHLvO1u8BwsKFzD87Hk1MVPDZ57zx6h-rA,3656
-nucliadb_utils-5.0.0.post941.dist-info/METADATA,sha256=KvXDjN4ZbzwDj-QrqChx6mHzZqLvsNUzP4gNoUwQGyU,2073
-nucliadb_utils-5.0.0.post941.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-nucliadb_utils-5.0.0.post941.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.0.0.post941.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.0.0.post941.dist-info/RECORD,,
+nucliadb_utils-5.0.0.post944.dist-info/METADATA,sha256=5LUVJjFXK-iP4wwMuJ28VQ8_PAhFaDxeI0mb8ZM0c3s,2073
+nucliadb_utils-5.0.0.post944.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
+nucliadb_utils-5.0.0.post944.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.0.0.post944.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.0.0.post944.dist-info/RECORD,,