nucliadb 6.3.1.post3577__py3-none-any.whl → 6.3.1.post3584__py3-none-any.whl

nucliadb/tasks/retries.py

@@ -20,7 +20,7 @@
 import functools
 import logging
 from enum import Enum
-from typing import Optional
+from typing import Callable, Optional
 
 from pydantic import BaseModel
 
@@ -73,7 +73,8 @@ class TaskRetryHandler:
         self.kbid = kbid
         self.task_type = task_type
         self.task_id = task_id
-        self.max_retries = max_retries
+        # Limit max retries to 50
+        self.max_retries = min(max_retries, 50)
         self.context = context
 
     @property
@@ -94,7 +95,7 @@ class TaskRetryHandler:
             await txn.set(self.metadata_key, metadata.model_dump_json().encode())
             await txn.commit()
 
-    def wrap(self, func):
+    def wrap(self, func: Callable) -> Callable:
         @functools.wraps(func)
         async def wrapper(*args, **kwargs):
             func_result = None
@@ -110,7 +111,7 @@ class TaskRetryHandler:
 
             if metadata.status in (TaskMetadata.Status.COMPLETED, TaskMetadata.Status.FAILED):
                 logger.info(
-                    f"{self.type} task is {metadata.status.value}. Skipping",
+                    f"{self.task_type} task is {metadata.status.value}. Skipping",
                     extra={"kbid": self.kbid, "task_type": self.task_type, "task_id": self.task_id},
                 )
                 return
@@ -130,7 +131,7 @@ class TaskRetryHandler:
             except Exception as ex:
                 metadata.retries += 1
                 metadata.error_messages.append(str(ex))
-                logger.info(
+                logger.exception(
                     f"Task failed. Will be retried",
                     extra={"kbid": self.kbid, "task_type": self.task_type, "task_id": self.task_id},
                 )

{nucliadb-6.3.1.post3577.dist-info → nucliadb-6.3.1.post3584.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nucliadb
-Version: 6.3.1.post3577
+Version: 6.3.1.post3584
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -20,11 +20,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: nucliadb-telemetry[all]>=6.3.1.post3577
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.3.1.post3577
-Requires-Dist: nucliadb-protos>=6.3.1.post3577
-Requires-Dist: nucliadb-models>=6.3.1.post3577
-Requires-Dist: nidx-protos>=6.3.1.post3577
+Requires-Dist: nucliadb-telemetry[all]>=6.3.1.post3584
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.3.1.post3584
+Requires-Dist: nucliadb-protos>=6.3.1.post3584
+Requires-Dist: nucliadb-models>=6.3.1.post3584
+Requires-Dist: nidx-protos>=6.3.1.post3584
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn

{nucliadb-6.3.1.post3577.dist-info → nucliadb-6.3.1.post3584.dist-info}/RECORD

@@ -272,7 +272,7 @@ nucliadb/tasks/consumer.py,sha256=xc0Ql3N1Iq52dJ3t4YYGJFj1NCQAly0J5W_brfLa_F8,68
 nucliadb/tasks/logger.py,sha256=C7keOEO_mjLVp5VbqAZ2QXfqVB2Hot7NgBlUP_SDSMw,924
 nucliadb/tasks/models.py,sha256=qrZKi5DNDQ07waMsp5L4_Fi7WRs57YiO-kmXlrBzEAA,1168
 nucliadb/tasks/producer.py,sha256=UnpJAzhj_GElsCoO5G6T4m6MshsgOaqR2tVzJmEta64,2625
-nucliadb/tasks/retries.py,sha256=Zv-3Hys-SKayG9VQ7_7EIflkegE5j-xPGrf-nwaxsfY,5075
+nucliadb/tasks/retries.py,sha256=tLNtwAutGEfFV3Adr4a9ew-Wg4vuxBd72GGDE_Mma8s,5160
 nucliadb/tasks/utils.py,sha256=tV1AbWdFc3qfIULX44Veqj41FCD1B6XYjG6brULBeiw,1459
 nucliadb/tests/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb/tests/config.py,sha256=JN_Jhgj-fwM9_8IeO9pwxr6C1PiwRDrXxm67Y38rU30,2080
@@ -347,8 +347,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
 nucliadb/writer/tus/s3.py,sha256=vF0NkFTXiXhXq3bCVXXVV-ED38ECVoUeeYViP8uMqcU,8357
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.3.1.post3577.dist-info/METADATA,sha256=7cMll6LH15F3_kAg1yrlOSDK8XFk77amc4TEa0kApug,4291
-nucliadb-6.3.1.post3577.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-nucliadb-6.3.1.post3577.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.3.1.post3577.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.3.1.post3577.dist-info/RECORD,,
+nucliadb-6.3.1.post3584.dist-info/METADATA,sha256=1LFsE32EtmIRlKK9EAN1-kK6P5pfLq3pIRewpO0k2Ug,4291
+nucliadb-6.3.1.post3584.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+nucliadb-6.3.1.post3584.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.3.1.post3584.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.3.1.post3584.dist-info/RECORD,,