chunkr-ai 0.0.31__tar.gz → 0.0.33__tar.gz

{chunkr_ai-0.0.31/src/chunkr_ai.egg-info → chunkr_ai-0.0.33}/PKG-INFO

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

{chunkr_ai-0.0.31 → chunkr_ai-0.0.33}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chunkr-ai"
-version = "0.0.31"
+version = "0.0.33"
 authors = [{"name" = "Ishaan Kapoor", "email" = "ishaan@lumina.sh"}]
 description = "Python client for Chunkr: open source document intelligence"
 readme = "README.md"

{chunkr_ai-0.0.31 → chunkr_ai-0.0.33}/src/chunkr_ai/api/decorators.py

@@ -1,6 +1,7 @@
-import functools
 import asyncio
+import functools
 import httpx
+import nest_asyncio
 from typing import Callable, Any, TypeVar, Awaitable, Union, overload
 try:
     from typing import ParamSpec
@@ -21,6 +22,12 @@ def anywhere():
         @functools.wraps(async_func)
         def wrapper(*args: P.args, **kwargs: P.kwargs) -> Union[Awaitable[T], T]:
             global _sync_loop
+
+            try:
+                nest_asyncio.apply()
+            except ImportError:
+                pass
+
             try:
                 asyncio.get_running_loop()
                 return async_func(*args, **kwargs) 
@@ -61,11 +68,11 @@ def require_task() -> Callable[[Callable[P, Awaitable[T]]], Callable[P, Awaitabl
         return wrapper
     return decorator
 
-def retry_on_429(max_retries: int = 25, initial_delay: float = 0.5) -> Callable[[Callable[P, Awaitable[T]]], Callable[P, Awaitable[T]]]:
+def retry_on_429(max_retries: int = 3, 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: 25)
+        max_retries: Maximum number of retry attempts (default: 3)
         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]]:
@@ -78,10 +85,10 @@ def retry_on_429(max_retries: int = 25, initial_delay: float = 0.5) -> Callable[
                     return await async_func(*args, **kwargs)
                 except httpx.HTTPStatusError as e:
                     if e.response.status_code != 429:
-                        raise
+                        raise e
                     if retries >= max_retries:
                         print("Max retries reached")
-                        raise
+                        raise e
                     retries += 1
                     delay = initial_delay * (2 ** retries)
                     # Use Retry-After header if available

{chunkr_ai-0.0.31 → chunkr_ai-0.0.33/src/chunkr_ai.egg-info}/PKG-INFO

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

{chunkr_ai-0.0.31 → chunkr_ai-0.0.33}/tests/test_chunkr.py

@@ -24,6 +24,10 @@ def sample_path():
 def sample_image():
     return Image.open("tests/files/test.jpg")
 
+@pytest.fixture
+def sample_url():
+    return "https://chunkr-web.s3.us-east-1.amazonaws.com/landing_page/input/science.pdf"
+
 @pytest.fixture
 def client():
     client = Chunkr()
@@ -36,6 +40,13 @@ async def test_send_file_path(client, sample_path):
     assert response.status == "Succeeded"
     assert response.output is not None
 
+@pytest.mark.asyncio
+async def test_send_file_url(client, sample_url):
+    response = await client.upload(sample_url)
+    assert response.task_id is not None
+    assert response.status == "Succeeded"
+    assert response.output is not None
+
 @pytest.mark.asyncio
 async def test_send_file_path_str(client, sample_path):
     response = await client.upload(str(sample_path))