datalab-python-sdk 0.1.6__tar.gz → 0.1.7__tar.gz

{datalab_python_sdk-0.1.6 → datalab_python_sdk-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datalab-python-sdk
-Version: 0.1.6
+Version: 0.1.7
 Summary: SDK for the Datalab document intelligence API
 Author-email: Datalab Team <hi@datalab.to>
 License-Expression: MIT

{datalab_python_sdk-0.1.6 → datalab_python_sdk-0.1.7}/datalab_sdk/cli.py

@@ -42,6 +42,12 @@ def common_options(func):
     func = click.option("--skip_cache", help="Skip the cache when running inference")(
         func
     )
+    func = click.option(
+        "--max_polls", default=300, type=int, help="Maximum number of polling attempts"
+    )(func)
+    func = click.option(
+        "--poll_interval", default=1, type=int, help="Polling interval in seconds"
+    )(func)
     return func
 
 
@@ -106,6 +112,8 @@ async def process_files_async(
     max_concurrent: int = 5,
     api_key: str | None = None,
     base_url: str | None = None,
+    max_polls: int = 300,
+    poll_interval: int = 1,
 ) -> List[dict]:
     """Process files asynchronously"""
     semaphore = asyncio.Semaphore(max_concurrent)
@@ -124,11 +132,13 @@ async def process_files_async(
                 ) as client:
                     if method == "convert":
                         result = await client.convert(
-                            file_path, options=options, save_output=output_path
+                            file_path, options=options, save_output=output_path,
+                            max_polls=max_polls, poll_interval=poll_interval
                         )
                     else:  # method == 'ocr'
                         result = await client.ocr(
-                            file_path, options=options, save_output=output_path
+                            file_path, options=options, save_output=output_path,
+                            max_polls=max_polls, poll_interval=poll_interval
                         )
 
                 return {
@@ -218,6 +228,8 @@ def process_documents(
     base_url: str,
     page_range: Optional[str],
     skip_cache: bool,
+    max_polls: int,
+    poll_interval: int,
     # Convert-specific options
     output_format: Optional[str] = None,
     force_ocr: bool = False,
@@ -288,6 +300,8 @@ def process_documents(
                 max_concurrent=max_concurrent,
                 api_key=api_key,
                 base_url=base_url,
+                max_polls=max_polls,
+                poll_interval=poll_interval,
             )
         )
 
@@ -320,6 +334,8 @@ def convert(
     base_url: str,
     page_range: Optional[str],
     skip_cache: bool,
+    max_polls: int,
+    poll_interval: int,
     output_format: str,
     force_ocr: bool,
     format_lines: bool,
@@ -342,6 +358,8 @@ def convert(
         base_url=base_url,
         page_range=page_range,
         skip_cache=skip_cache,
+        max_polls=max_polls,
+        poll_interval=poll_interval,
         output_format=output_format,
         force_ocr=force_ocr,
         format_lines=format_lines,
@@ -367,6 +385,8 @@ def ocr(
     base_url: str,
     page_range: Optional[str],
     skip_cache: bool,
+    max_polls: int,
+    poll_interval: int,
 ):
     """Perform OCR on documents"""
     process_documents(
@@ -380,6 +400,8 @@ def ocr(
         base_url=base_url,
         page_range=page_range,
         skip_cache=skip_cache,
+        max_polls=max_polls,
+        poll_interval=poll_interval,
     )
 
 

{datalab_python_sdk-0.1.6 → datalab_python_sdk-0.1.7}/datalab_sdk/models.py

@@ -13,7 +13,7 @@ import base64
 class ProcessingOptions:
     # Common options
     max_pages: Optional[int] = None
-    skip_cache: bool = True
+    skip_cache: bool = False
     page_range: Optional[str] = None
 
     def to_form_data(self) -> Dict[str, Any]:

{datalab_python_sdk-0.1.6 → datalab_python_sdk-0.1.7}/pyproject.toml

@@ -7,7 +7,7 @@ readme = "README.md"
 license = "MIT"
 repository = "https://github.com/datalab-to/sdk"
 keywords = ["datalab", "sdk", "document-intelligence", "api"]
-version = "0.1.6"
+version = "0.1.7"
 description = "SDK for the Datalab document intelligence API"
 requires-python = ">=3.10"
 dependencies = [

{datalab_python_sdk-0.1.6 → datalab_python_sdk-0.1.7}/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 2
+revision = 3
 requires-python = ">=3.10"
 
 [[package]]
@@ -169,7 +169,7 @@ wheels = [
 
 [[package]]
 name = "datalab-python-sdk"
-version = "0.1.6"
+version = "0.1.7"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },