nucliadb-utils 5.0.0.post833__py3-none-any.whl → 5.0.0.post838__py3-none-any.whl

nucliadb_utils/aiopynecone/client.py

@@ -64,6 +64,7 @@ BASE_API_HEADERS = {
 MEGA_BYTE = 1024 * 1024
 MAX_UPSERT_PAYLOAD_SIZE = 2 * MEGA_BYTE
 MAX_DELETE_BATCH_SIZE = 1000
+MAX_LIST_PAGE_SIZE = 100
 
 
 RETRIABLE_EXCEPTIONS = (
@@ -162,6 +163,9 @@ class DataPlane:
         - `vectors`: The vectors to upsert.
         - `timeout`: to control the request timeout. If not set, the default timeout is used.
         """
+        if len(vectors) == 0:
+            # Nothing to upsert.
+            return
         headers = {"Api-Key": self.api_key}
         payload = UpsertRequest(vectors=vectors)
         post_kwargs: dict[str, Any] = {
@@ -257,7 +261,7 @@ class DataPlane:
     async def list_page(
         self,
         id_prefix: Optional[str] = None,
-        limit: int = 100,
+        limit: int = MAX_LIST_PAGE_SIZE,
         pagination_token: Optional[str] = None,
         timeout: Optional[float] = None,
     ) -> ListResponse:
@@ -270,6 +274,8 @@ class DataPlane:
            if there are more pages to fetch.
         - `timeout`: to control the request timeout. If not set, the default timeout is used.
         """
+        if limit > MAX_LIST_PAGE_SIZE:  # pragma: no cover
+            raise ValueError(f"Maximum limit is {MAX_LIST_PAGE_SIZE}.")
         headers = {"Api-Key": self.api_key}
         params = {"limit": str(limit)}
         if id_prefix is not None:
@@ -292,7 +298,10 @@ class DataPlane:
         return ListResponse.model_validate(response.json())
 
     async def list_all(
-        self, id_prefix: Optional[str] = None, page_size: int = 100, page_timeout: Optional[float] = None
+        self,
+        id_prefix: Optional[str] = None,
+        page_size: int = MAX_LIST_PAGE_SIZE,
+        page_timeout: Optional[float] = None,
     ) -> AsyncGenerator[str, None]:
         """
         Iterate over all vector ids from the index in a paginated manner.
@@ -368,9 +377,7 @@ class DataPlane:
                 await self.delete(ids=batch, timeout=batch_timeout)
 
         tasks = []
-        async_iterable = self.list_all(
-            id_prefix=id_prefix, page_size=batch_size, page_timeout=batch_timeout
-        )
+        async_iterable = self.list_all(id_prefix=id_prefix, page_timeout=batch_timeout)
         async for batch in async_batchify(async_iterable, batch_size):
             tasks.append(asyncio.create_task(_delete_batch(batch)))
 

nucliadb_utils/aiopynecone/models.py

@@ -49,8 +49,7 @@ def validate_index_name(value, handler, info):
 
 IndexNameStr = Annotated[
     str,
-    pydantic.StringConstraints(pattern=IndexNamePattern),
-    pydantic.StringConstraints(min_length=1, max_length=MAX_INDEX_NAME_LENGTH),
+    pydantic.StringConstraints(pattern=IndexNamePattern, min_length=1, max_length=MAX_INDEX_NAME_LENGTH),
     pydantic.WrapValidator(validate_index_name),
 ]
 
@@ -63,7 +62,7 @@ class CreateIndexRequest(BaseModel):
 
 
 class Vector(BaseModel):
-    id: str = Field(max_length=512)
+    id: str = Field(min_length=1, max_length=512)
     values: list[float]
     metadata: dict[str, Any] = {}
 
@@ -73,6 +72,7 @@ class Vector(BaseModel):
         json_value = json.dumps(value)
         if len(json_value) > MAX_METADATA_SIZE:
             raise ValueError("metadata size is too large")
+        return value
 
 
 class UpsertRequest(BaseModel):

{nucliadb_utils-5.0.0.post833.dist-info → nucliadb_utils-5.0.0.post838.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 5.0.0.post833
+Version: 5.0.0.post838
 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.post833
-Requires-Dist: nucliadb-telemetry >=5.0.0.post833
+Requires-Dist: nucliadb-protos >=5.0.0.post838
+Requires-Dist: nucliadb-telemetry >=5.0.0.post838
 Provides-Extra: cache
 Requires-Dist: redis >=4.3.4 ; extra == 'cache'
 Requires-Dist: orjson >=3.6.7 ; extra == 'cache'

{nucliadb_utils-5.0.0.post833.dist-info → nucliadb_utils-5.0.0.post838.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=oz3tEODG2g3todnyvA-nW1Ou6xXDveL_tMKTDGdWXM4,15287
 nucliadb_utils/aiopynecone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/aiopynecone/client.py,sha256=wUQIUZHKvhMhmLVfwrOF_nMBcf9l-4mXCvlSI0l0H24,18472
+nucliadb_utils/aiopynecone/client.py,sha256=jlTLOdphrLZElKW9ajVzDcebN7CSAbjNyX6tnD37eqY,18721
 nucliadb_utils/aiopynecone/exceptions.py,sha256=hFhq-UEY4slqNWjObXr_LPnRf_AQ1vpcG4SF2XRFd1E,2873
-nucliadb_utils/aiopynecone/models.py,sha256=sEmifzQ6rvqIB8nbkJbh8-hrCW4j8J9lJ_xLRaTAqro,2934
+nucliadb_utils/aiopynecone/models.py,sha256=B_ihJhHZGp3ivQVUxhV49uoUnHe1PLDKxTgHNbHgSS0,2937
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/audit/audit.py,sha256=dn5ZnCVQUlCcvdjzaORghbrjk9QgVGrtkfIftq30Bp8,2819
 nucliadb_utils/audit/basic.py,sha256=NViey6mKbCXqRTLDBX2xNTcCg9I-2e4oB2xkekuhDvM,3392
@@ -64,8 +64,8 @@ nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQ
 nucliadb_utils/tests/local.py,sha256=c3gZJJWmvOftruJkIQIwB3q_hh3uxEhqGIAVWim1Bbk,1343
 nucliadb_utils/tests/nats.py,sha256=Tosonm9A9cusImyji80G4pgdXEHNVPaCLT5TbFK_ra0,7543
 nucliadb_utils/tests/s3.py,sha256=YB8QqDaBXxyhHonEHmeBbRRDmvB7sTOaKBSi8KBGokg,2330
-nucliadb_utils-5.0.0.post833.dist-info/METADATA,sha256=MXk_bhsqpp74b_A9eLOWcCQSEguuztiGcf0llEGsVFA,2073
-nucliadb_utils-5.0.0.post833.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-nucliadb_utils-5.0.0.post833.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.0.0.post833.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.0.0.post833.dist-info/RECORD,,
+nucliadb_utils-5.0.0.post838.dist-info/METADATA,sha256=gmSTcyZplCaAvM5mu6UvxeWhR8SlgbAV_S-YdGykS2U,2073
+nucliadb_utils-5.0.0.post838.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+nucliadb_utils-5.0.0.post838.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.0.0.post838.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.0.0.post838.dist-info/RECORD,,