nucliadb-utils 5.0.1.post1107__py3-none-any.whl → 5.0.1.post1116__py3-none-any.whl

nucliadb_utils/aiopynecone/exceptions.py

@@ -60,6 +60,14 @@ class PineconeRateLimitError(PineconeAPIError):
     pass
 
 
+class PineconeNeedsPlanUpgradeError(PineconeAPIError):
+    """
+    Raised when the client needs to upgrade the plan to continue using the service.
+    """
+
+    pass
+
+
 class MetadataTooLargeError(ValueError):
     """
     Raised when the metadata of a vector to be upserted is too large.
@@ -79,12 +87,19 @@ def raise_for_status(operation: str, response: httpx.Response):
         try:
             resp_json = response.json()
             error = resp_json.get("error") or {}
-            code = error.get("code")
-            message = error.get("message")
-            details = error.get("details")
+            code = resp_json.get("code") or error.get("code")
+            message = resp_json.get("message") or error.get("message") or ""
+            details = resp_json.get("details") or error.get("details")
         except Exception:  # pragma: no cover
             message = response.text
         if response.status_code == 429:
+            if "month" in message:
+                raise PineconeNeedsPlanUpgradeError(
+                    http_status_code=response.status_code,
+                    code=code,
+                    message=message,
+                    details=details,
+                )
             raise PineconeRateLimitError(
                 http_status_code=response.status_code,
                 code=code,

nucliadb_utils/const.py

@@ -83,3 +83,4 @@ class Features:
     PG_CATALOG_WRITE = "nucliadb_pg_catalog_write"
     VECTORSETS_V0 = "vectorsets_v0_new_kbs_with_multiple_vectorsets"
     SKIP_EXTERNAL_INDEX = "nucliadb_skip_external_index"
+    NATS_ASYNC_ACK = "nucliadb_nats_async_ack"

nucliadb_utils/featureflagging.py

@@ -81,6 +81,10 @@ DEFAULT_FLAG_DATA: dict[str, Any] = {
         "rollout": 0,
         "variants": {"environment": ["none"]},
     },
+    const.Features.NATS_ASYNC_ACK: {
+        "rollout": 0,
+        "variants": {"environment": ["local"]},
+    },
 }
 
 

nucliadb_utils/nats.py

@@ -74,6 +74,7 @@ class MessageProgressUpdater:
         try:
             await self._task
         except asyncio.CancelledError:  # pragma: no cover
+            logger.info("MessageProgressUpdater cancelled")
             pass
         except Exception as exc:  # pragma: no cover
             capture_exception(exc)

{nucliadb_utils-5.0.1.post1107.dist-info → nucliadb_utils-5.0.1.post1116.dist-info}/METADATA

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

{nucliadb_utils-5.0.1.post1107.dist-info → nucliadb_utils-5.0.1.post1116.dist-info}/RECORD

@@ -1,14 +1,14 @@
 nucliadb_utils/__init__.py,sha256=EvBCH1iTODe-AgXm48aj4kVUt_Std3PeL8QnwimR5wI,895
 nucliadb_utils/asyncio_utils.py,sha256=h8Y-xpcFFRgNzaiIW0eidz7griAQa7ggbNk34-tAt2c,2888
 nucliadb_utils/authentication.py,sha256=N__d2Ez3JHJv5asYK5TgUcIkKqcAC8ZTLlnfLhfSneM,5837
-nucliadb_utils/const.py,sha256=ERzAa78DCPnE7ln9ChjP7A3ILdA_qGd9HStosBZkqwQ,2676
+nucliadb_utils/const.py,sha256=AYtxOqOoyDXYzsca0UTXbt6ACZsgoyfItorybDGKHdc,2723
 nucliadb_utils/debug.py,sha256=Q56Nx9Dp7V2ae3CU2H0ztaZcHTJXdlflPLKLeOPZ170,2436
 nucliadb_utils/exceptions.py,sha256=y_3wk77WLVUtdo-5FtbBsdSkCtK_DsJkdWb5BoPn3qo,1094
-nucliadb_utils/featureflagging.py,sha256=0nN3vmtaEs6rB2uFQn0kI88It9568gcZ0GWWeWI1AOk,3308
+nucliadb_utils/featureflagging.py,sha256=H_qW7nE2k5wS7mv1m5SXayuvDDhTHYPwqTfW5-eVzEI,3422
 nucliadb_utils/grpc.py,sha256=apu0uePnkGHCAT7GRQ9YZfRYyFj26kJ440i8jitbM3U,3314
 nucliadb_utils/helpers.py,sha256=nPw8yod3hP-pxq80VF8QC36s7ygSg0dBUdfI-LatvCs,1600
 nucliadb_utils/indexing.py,sha256=Luaqcar3CySpdYOFp6Q9Fyr8ZYwhYhaKRHQ_VGL78f8,2318
-nucliadb_utils/nats.py,sha256=b7_ZqBTtacnfTeErfeozXTLXbTL27JwqF1-SUEe9VmM,8370
+nucliadb_utils/nats.py,sha256=prs98dIrGh2-WchQcRBloc5wYhpjEnB0VtCERPnoReg,8430
 nucliadb_utils/partition.py,sha256=jBgy4Hu5Iwn4gjbPPcthSykwf-qNx-GcLAIwbzPd1d0,1157
 nucliadb_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb_utils/run.py,sha256=HpAIM8xbR7UpVC2_7xOjB4fYbUVykyPP6yHrv2RD3DI,1707
@@ -19,7 +19,7 @@ nucliadb_utils/transaction.py,sha256=mwcI3aIHAvU5KOGqd_Uz_d1XQzXhk_-NWY8NqU1lfb0
 nucliadb_utils/utilities.py,sha256=idajCm_4Sojh7b3HTkP0fTfG2Mb6PIB9xtMmcfB7Nl0,15758
 nucliadb_utils/aiopynecone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/aiopynecone/client.py,sha256=3DGJuHXO9Vm83atMLD4fd2FTz8SAr8RvPCGU_bXH9Ho,22333
-nucliadb_utils/aiopynecone/exceptions.py,sha256=EEE0XoGs1zIB5yOJ_fy6yoG4uIb4cWIawYdJeNe4eDo,3012
+nucliadb_utils/aiopynecone/exceptions.py,sha256=yfUXIADRuhPb2rNynoFRlpQ6QINm1EkmvT_BbZ7dv8I,3536
 nucliadb_utils/aiopynecone/models.py,sha256=ketK2IYLWiwFZ76rnJmwfcuopFJrCAtCUszdTSurm_Q,3236
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/audit/audit.py,sha256=Tbi8poUP6BKZy4ZsaFaeTaQ1K5ys8GXym5ps389-XHE,2966
@@ -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.1.post1107.dist-info/METADATA,sha256=tiIC4Np-x49gLu5SMMyBIwNmtiq_5L0HNuMIGX3UgyA,2071
-nucliadb_utils-5.0.1.post1107.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-nucliadb_utils-5.0.1.post1107.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.0.1.post1107.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.0.1.post1107.dist-info/RECORD,,
+nucliadb_utils-5.0.1.post1116.dist-info/METADATA,sha256=KpdqhX5ztjM8y5_ScL4v-NM9eAhMDLxbtJDEzeoXv6s,2071
+nucliadb_utils-5.0.1.post1116.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+nucliadb_utils-5.0.1.post1116.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.0.1.post1116.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.0.1.post1116.dist-info/RECORD,,