chunkr-ai 0.0.28__py3-none-any.whl → 0.0.30__py3-none-any.whl

chunkr_ai/api/decorators.py

@@ -61,25 +61,29 @@ def require_task() -> Callable[[Callable[P, Awaitable[T]]], Callable[P, Awaitabl
         return wrapper
     return decorator
 
-def retry_on_429(max_retries: int = 10, initial_delay: float = 0.5) -> Callable[[Callable[P, Awaitable[T]]], Callable[P, Awaitable[T]]]:
+def retry_on_429(max_retries: int = 25, initial_delay: float = 0.5) -> Callable[[Callable[P, Awaitable[T]]], Callable[P, Awaitable[T]]]:
     """Decorator that retries the request when encountering 429 Too Many Requests errors.
     
     Args:
-        max_retries: Maximum number of retry attempts (default: 3)
-        initial_delay: Initial delay in seconds, will be exponentially increased (default: 1.0)
+        max_retries: Maximum number of retry attempts (default: 25)
+        initial_delay: Initial delay in seconds, will be exponentially increased with jitter (default: 0.5)
     """
     def decorator(async_func: Callable[P, Awaitable[T]]) -> Callable[P, Awaitable[T]]:
         @functools.wraps(async_func)
         async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+            import random 
             retries = 0
             while True:
                 try:
                     return await async_func(*args, **kwargs)
                 except httpx.HTTPStatusError as e:
-                    if e.response.status_code != 429 or retries >= max_retries:
+                    if e.response.status_code != 429:
+                        raise
+                    if retries >= max_retries:
+                        print("Max retries reached")
                         raise
                     retries += 1
-                    delay = initial_delay
+                    delay = initial_delay * (2 ** retries)
                     # Use Retry-After header if available
                     retry_after = e.response.headers.get('Retry-After')
                     if retry_after:
@@ -87,6 +91,7 @@ def retry_on_429(max_retries: int = 10, initial_delay: float = 0.5) -> Callable[
                             delay = float(retry_after)
                         except (ValueError, TypeError):
                             pass
-                    await asyncio.sleep(delay)
+                    jitter = random.uniform(0, 0.25) * delay
+                    await asyncio.sleep(delay + jitter)
         return wrapper
     return decorator

{chunkr_ai-0.0.28.dist-info → chunkr_ai-0.0.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.28
+Version: 0.0.30
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 License: MIT License

{chunkr_ai-0.0.28.dist-info → chunkr_ai-0.0.30.dist-info}/RECORD

@@ -5,12 +5,12 @@ chunkr_ai/api/auth.py,sha256=hlv0GiUmlsbFO1wLL9sslqOnsBSoBqkL_6Mk2SDvxgE,413
 chunkr_ai/api/chunkr.py,sha256=VnbuAPlWLqyf8xCCU_kpdybgjVPTwZLarDQoD3uozY0,3065
 chunkr_ai/api/chunkr_base.py,sha256=giW56fL7xxJphdOTpIH52dXxpNt7OdP8pNiPSqbNjGM,5835
 chunkr_ai/api/configuration.py,sha256=m8QL6s2j1_xSo9bERzJ3UG5j8cE6o-cZIiQGxsSf7Ow,3805
-chunkr_ai/api/decorators.py,sha256=HSq3vcxOeUJkaWaf7HOvCyg9dWkVo8cG5BrU-jhbhmc,4053
+chunkr_ai/api/decorators.py,sha256=hu3wdd0s7P6askR3Td8g6OM-RoUCggDGv06HQUsMI6E,4284
 chunkr_ai/api/misc.py,sha256=5PBI6pvOXr0x-3WieSKLrC8MA0iGPa-IG-5FEZ3vnr0,5724
 chunkr_ai/api/protocol.py,sha256=LjPrYSq52m1afIlAo0yVGXlGZxPRh8J6g7S4PAit3Zo,388
 chunkr_ai/api/task_response.py,sha256=uvM5uNIImDnivsgujjOh2kxRYuUoqKDhLE1nOH2dJjo,6091
-chunkr_ai-0.0.28.dist-info/LICENSE,sha256=w3R12yNDyZpMiy2lxy_hvNbsldC75ww79sF0u11rkho,1069
-chunkr_ai-0.0.28.dist-info/METADATA,sha256=Cgl0mTQaF6Np72pVGUPS96ZJBH9hmJuAfv1GieKln5k,6996
-chunkr_ai-0.0.28.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-chunkr_ai-0.0.28.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
-chunkr_ai-0.0.28.dist-info/RECORD,,
+chunkr_ai-0.0.30.dist-info/LICENSE,sha256=w3R12yNDyZpMiy2lxy_hvNbsldC75ww79sF0u11rkho,1069
+chunkr_ai-0.0.30.dist-info/METADATA,sha256=7prC0mZFxek20uv901qM7s4rFmteCYNnK8BI3ZelXXo,6996
+chunkr_ai-0.0.30.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+chunkr_ai-0.0.30.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
+chunkr_ai-0.0.30.dist-info/RECORD,,