chunkr-ai 0.1.0a4__py3-none-any.whl → 0.1.0a6__py3-none-any.whl

chunkr_ai/__init__.py

@@ -72,9 +72,6 @@ __all__ = [
 ]
 
 if not _t.TYPE_CHECKING:
-    # Load custom helpers that monkey-patch generated types.
-    # This keeps custom code separate from generated files, per Stainless guidance.
-    from .lib import tasks_poll as _tasks_poll  # noqa: F401
     from ._utils._resources_proxy import resources as resources
 
 _setup_logging()

chunkr_ai/_client.py

@@ -95,6 +95,8 @@ class Chunkr(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._idempotency_header = "Idempotency-Key"
+
         self.tasks = tasks.TasksResource(self)
         self.files = files.FilesResource(self)
         self.health = health.HealthResource(self)
@@ -267,6 +269,8 @@ class AsyncChunkr(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._idempotency_header = "Idempotency-Key"
+
         self.tasks = tasks.AsyncTasksResource(self)
         self.files = files.AsyncFilesResource(self)
         self.health = health.AsyncHealthResource(self)

chunkr_ai/_constants.py

@@ -5,10 +5,10 @@ import httpx
 RAW_RESPONSE_HEADER = "X-Stainless-Raw-Response"
 OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
 
-# default timeout is 1 minute
-DEFAULT_TIMEOUT = httpx.Timeout(timeout=60, connect=5.0)
-DEFAULT_MAX_RETRIES = 2
+# default timeout is 30 seconds
+DEFAULT_TIMEOUT = httpx.Timeout(timeout=30, connect=5.0)
+DEFAULT_MAX_RETRIES = 50
 DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
 
-INITIAL_RETRY_DELAY = 0.5
-MAX_RETRY_DELAY = 8.0
+INITIAL_RETRY_DELAY = 1.0
+MAX_RETRY_DELAY = 10.0

chunkr_ai/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "chunkr_ai"
-__version__ = "0.1.0-alpha.4"  # x-release-please-version
+__version__ = "0.1.0-alpha.6"  # x-release-please-version

chunkr_ai/resources/files.py

@@ -59,6 +59,7 @@ class FilesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> File:
         """
         Accepts multipart/form-data with fields:
@@ -78,6 +79,8 @@ class FilesResource(SyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         body = deepcopy_minimal(
             {
@@ -95,7 +98,11 @@ class FilesResource(SyncAPIResource):
             body=maybe_transform(body, file_create_params.FileCreateParams),
             files=files,
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=File,
         )
@@ -170,6 +177,7 @@ class FilesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> Delete:
         """Delete file contents and scrub sensitive metadata.
 
@@ -184,13 +192,19 @@ class FilesResource(SyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         if not file_id:
             raise ValueError(f"Expected a non-empty value for `file_id` but received {file_id!r}")
         return self._delete(
             f"/files/{file_id}",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=Delete,
         )
@@ -353,6 +367,7 @@ class AsyncFilesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> File:
         """
         Accepts multipart/form-data with fields:
@@ -372,6 +387,8 @@ class AsyncFilesResource(AsyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         body = deepcopy_minimal(
             {
@@ -389,7 +406,11 @@ class AsyncFilesResource(AsyncAPIResource):
             body=await async_maybe_transform(body, file_create_params.FileCreateParams),
             files=files,
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=File,
         )
@@ -464,6 +485,7 @@ class AsyncFilesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> Delete:
         """Delete file contents and scrub sensitive metadata.
 
@@ -478,13 +500,19 @@ class AsyncFilesResource(AsyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         if not file_id:
             raise ValueError(f"Expected a non-empty value for `file_id` but received {file_id!r}")
         return await self._delete(
             f"/files/{file_id}",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=Delete,
         )

chunkr_ai/resources/tasks/parse.py

@@ -63,26 +63,21 @@ class ParseResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> Task:
         """
-        Queues a document for processing and returns a TaskResponse containing:
+        Queues a document for processing and returns a `TaskResponse` with the assigned
+        `task_id`, initial configuration, file metadata, and timestamps. The initial
+        status is `Starting`.
 
-        - Task ID for status polling
-        - Initial configuration
-        - File metadata
-        - Processing status
-        - Creation timestamp
-        - Presigned URLs for file access
-
-        The returned task will typically be in a `Starting` or `Processing` state. Use
-        the `GET /tasks/{task_id}` endpoint to poll for completion.
+        Creates a task and returns its metadata immediately.
 
         Args:
           file:
               The file to be uploaded. Supported inputs:
 
-              - `ch://files/{file_id}`: References a previously uploaded file you own
-                (authorization enforced)
+              - `ch://files/{file_id}`: Reference to an existing file. Upload via the Files
+                API
               - `http(s)://...`: Remote URL to fetch
               - `data:*;base64,...` or raw base64 string
 
@@ -114,22 +109,29 @@ class ParseResource(SyncAPIResource):
 
           segment_processing: Defines how each segment type is handled when generating the final output.
 
-              Each segment uses one of three strategies. The chosen strategy controls: •
-              Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-              content is produced (rule-based vs. LLM). • The output format (`Html` or
-              `Markdown`).
+              Each segment uses one of three strategies. The chosen strategy controls:
+
+              - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+              - How the content is produced (rule-based vs. LLM).
+              - The output format (`Html` or `Markdown`).
 
               Optional flags such as image **cropping**, **extended context**, and
               **descriptions** further refine behaviour.
 
-              **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-              `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-              (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-              cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-              `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+              **Default strategy per segment**
+
+              - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+                (Markdown, description off)
+              - `Table` → **LLM** (HTML, description on)
+              - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+              - `Formula`, `Page` → **LLM** (Markdown, description off)
+              - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
 
-              **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-              generate content with an LLM. • **Ignore** – exclude the segment entirely.
+              **Strategy reference**
+
+              - **Auto** – rule-based content generation.
+              - **LLM** – generate content with an LLM.
+              - **Ignore** – exclude the segment entirely.
 
           segmentation_strategy:
               Controls the segmentation strategy:
@@ -147,6 +149,8 @@ class ParseResource(SyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         return self._post(
             "/tasks/parse",
@@ -166,7 +170,11 @@ class ParseResource(SyncAPIResource):
                 parse_create_params.ParseCreateParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=Task,
         )
@@ -190,20 +198,19 @@ class ParseResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> Task:
         """Updates an existing task's configuration and reprocesses the document.
 
         The
-        original configuration will be used for all values that are not provided in the
-        update.
+        current configuration is used as the base; only provided fields are changed.
 
         Requirements:
 
-        - Task must have status `Succeeded` or `Failed`
-        - New configuration must be different from the current one
+        - Task must be in a terminal state (`Succeeded` or `Failed`).
+        - The new configuration must differ from the current configuration.
 
-        The returned task will typically be in a `Starting` or `Processing` state. Use
-        the `GET /tasks/{task_id}` endpoint to poll for completion.
+        Updates a task and returns its new metadata immediately.
 
         Args:
           chunk_processing: Controls the setting for the chunking and post-processing of each chunk.
@@ -235,22 +242,29 @@ class ParseResource(SyncAPIResource):
 
           segment_processing: Defines how each segment type is handled when generating the final output.
 
-              Each segment uses one of three strategies. The chosen strategy controls: •
-              Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-              content is produced (rule-based vs. LLM). • The output format (`Html` or
-              `Markdown`).
+              Each segment uses one of three strategies. The chosen strategy controls:
+
+              - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+              - How the content is produced (rule-based vs. LLM).
+              - The output format (`Html` or `Markdown`).
 
               Optional flags such as image **cropping**, **extended context**, and
               **descriptions** further refine behaviour.
 
-              **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-              `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-              (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-              cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-              `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+              **Default strategy per segment**
+
+              - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+                (Markdown, description off)
+              - `Table` → **LLM** (HTML, description on)
+              - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+              - `Formula`, `Page` → **LLM** (Markdown, description off)
+              - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
 
-              **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-              generate content with an LLM. • **Ignore** – exclude the segment entirely.
+              **Strategy reference**
+
+              - **Auto** – rule-based content generation.
+              - **LLM** – generate content with an LLM.
+              - **Ignore** – exclude the segment entirely.
 
           segmentation_strategy:
               Controls the segmentation strategy:
@@ -268,6 +282,8 @@ class ParseResource(SyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         if not task_id:
             raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
@@ -288,7 +304,11 @@ class ParseResource(SyncAPIResource):
                 parse_update_params.ParseUpdateParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=Task,
         )
@@ -333,26 +353,21 @@ class AsyncParseResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> Task:
         """
-        Queues a document for processing and returns a TaskResponse containing:
+        Queues a document for processing and returns a `TaskResponse` with the assigned
+        `task_id`, initial configuration, file metadata, and timestamps. The initial
+        status is `Starting`.
 
-        - Task ID for status polling
-        - Initial configuration
-        - File metadata
-        - Processing status
-        - Creation timestamp
-        - Presigned URLs for file access
-
-        The returned task will typically be in a `Starting` or `Processing` state. Use
-        the `GET /tasks/{task_id}` endpoint to poll for completion.
+        Creates a task and returns its metadata immediately.
 
         Args:
           file:
               The file to be uploaded. Supported inputs:
 
-              - `ch://files/{file_id}`: References a previously uploaded file you own
-                (authorization enforced)
+              - `ch://files/{file_id}`: Reference to an existing file. Upload via the Files
+                API
               - `http(s)://...`: Remote URL to fetch
               - `data:*;base64,...` or raw base64 string
 
@@ -384,22 +399,29 @@ class AsyncParseResource(AsyncAPIResource):
 
           segment_processing: Defines how each segment type is handled when generating the final output.
 
-              Each segment uses one of three strategies. The chosen strategy controls: •
-              Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-              content is produced (rule-based vs. LLM). • The output format (`Html` or
-              `Markdown`).
+              Each segment uses one of three strategies. The chosen strategy controls:
+
+              - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+              - How the content is produced (rule-based vs. LLM).
+              - The output format (`Html` or `Markdown`).
 
               Optional flags such as image **cropping**, **extended context**, and
               **descriptions** further refine behaviour.
 
-              **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-              `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-              (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-              cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-              `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+              **Default strategy per segment**
+
+              - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+                (Markdown, description off)
+              - `Table` → **LLM** (HTML, description on)
+              - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+              - `Formula`, `Page` → **LLM** (Markdown, description off)
+              - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
 
-              **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-              generate content with an LLM. • **Ignore** – exclude the segment entirely.
+              **Strategy reference**
+
+              - **Auto** – rule-based content generation.
+              - **LLM** – generate content with an LLM.
+              - **Ignore** – exclude the segment entirely.
 
           segmentation_strategy:
               Controls the segmentation strategy:
@@ -417,6 +439,8 @@ class AsyncParseResource(AsyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         return await self._post(
             "/tasks/parse",
@@ -436,7 +460,11 @@ class AsyncParseResource(AsyncAPIResource):
                 parse_create_params.ParseCreateParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=Task,
         )
@@ -460,20 +488,19 @@ class AsyncParseResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> Task:
         """Updates an existing task's configuration and reprocesses the document.
 
         The
-        original configuration will be used for all values that are not provided in the
-        update.
+        current configuration is used as the base; only provided fields are changed.
 
         Requirements:
 
-        - Task must have status `Succeeded` or `Failed`
-        - New configuration must be different from the current one
+        - Task must be in a terminal state (`Succeeded` or `Failed`).
+        - The new configuration must differ from the current configuration.
 
-        The returned task will typically be in a `Starting` or `Processing` state. Use
-        the `GET /tasks/{task_id}` endpoint to poll for completion.
+        Updates a task and returns its new metadata immediately.
 
         Args:
           chunk_processing: Controls the setting for the chunking and post-processing of each chunk.
@@ -505,22 +532,29 @@ class AsyncParseResource(AsyncAPIResource):
 
           segment_processing: Defines how each segment type is handled when generating the final output.
 
-              Each segment uses one of three strategies. The chosen strategy controls: •
-              Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-              content is produced (rule-based vs. LLM). • The output format (`Html` or
-              `Markdown`).
+              Each segment uses one of three strategies. The chosen strategy controls:
+
+              - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+              - How the content is produced (rule-based vs. LLM).
+              - The output format (`Html` or `Markdown`).
 
               Optional flags such as image **cropping**, **extended context**, and
               **descriptions** further refine behaviour.
 
-              **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-              `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-              (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-              cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-              `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+              **Default strategy per segment**
+
+              - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+                (Markdown, description off)
+              - `Table` → **LLM** (HTML, description on)
+              - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+              - `Formula`, `Page` → **LLM** (Markdown, description off)
+              - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
 
-              **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-              generate content with an LLM. • **Ignore** – exclude the segment entirely.
+              **Strategy reference**
+
+              - **Auto** – rule-based content generation.
+              - **LLM** – generate content with an LLM.
+              - **Ignore** – exclude the segment entirely.
 
           segmentation_strategy:
               Controls the segmentation strategy:
@@ -538,6 +572,8 @@ class AsyncParseResource(AsyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         if not task_id:
             raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
@@ -558,7 +594,11 @@ class AsyncParseResource(AsyncAPIResource):
                 parse_update_params.ParseUpdateParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=Task,
         )

chunkr_ai/resources/tasks/tasks.py

@@ -138,6 +138,7 @@ class TasksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> None:
         """
         Delete a task by its ID.
@@ -154,6 +155,8 @@ class TasksResource(SyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         if not task_id:
             raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
@@ -161,7 +164,11 @@ class TasksResource(SyncAPIResource):
         return self._delete(
             f"/tasks/{task_id}",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=NoneType,
         )
@@ -213,6 +220,7 @@ class TasksResource(SyncAPIResource):
         *,
         base64_urls: bool | NotGiven = NOT_GIVEN,
         include_chunks: bool | NotGiven = NOT_GIVEN,
+        wait_for_completion: bool | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -221,20 +229,20 @@ class TasksResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Task:
         """
-        Retrieves detailed information about a task by its ID, including:
+        Retrieves the current state of a task and, when requested, waits briefly for
+        completion.
 
-        - Processing status
-        - Task configuration
-        - Output data (if processing is complete)
-        - File metadata (name, page count)
-        - Timestamps (created, started, finished)
-        - Presigned URLs for accessing files
+        Returns task details such as processing status, configuration, output (when
+        available), file metadata, and timestamps. If `wait_for_completion=true` is
+        provided, the server will hold the request briefly. If the task does not reach a
+        terminal state during that window, the response will indicate a retry with
+        appropriate headers.
 
-        This endpoint can be used to:
+        Typical uses:
 
-        1. Poll the task status during processing
-        2. Retrieve the final output once processing is complete
-        3. Access task metadata and configuration
+        - Poll a task during processing
+        - Retrieve the final output once processing is complete
+        - Access task metadata and configuration
 
         Args:
           base64_urls: Whether to return base64 encoded URLs. If false, the URLs will be returned as
@@ -242,6 +250,8 @@ class TasksResource(SyncAPIResource):
 
           include_chunks: Whether to include chunks in the output response
 
+          wait_for_completion: Whether to wait for the task to complete
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -263,6 +273,7 @@ class TasksResource(SyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
+                        "wait_for_completion": wait_for_completion,
                     },
                     task_get_params.TaskGetParams,
                 ),
@@ -375,6 +386,7 @@ class AsyncTasksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        idempotency_key: str | None = None,
     ) -> None:
         """
         Delete a task by its ID.
@@ -391,6 +403,8 @@ class AsyncTasksResource(AsyncAPIResource):
           extra_body: Add additional JSON properties to the request
 
           timeout: Override the client-level default timeout for this request, in seconds
+
+          idempotency_key: Specify a custom idempotency key for this request
         """
         if not task_id:
             raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
@@ -398,7 +412,11 @@ class AsyncTasksResource(AsyncAPIResource):
         return await self._delete(
             f"/tasks/{task_id}",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                idempotency_key=idempotency_key,
             ),
             cast_to=NoneType,
         )
@@ -450,6 +468,7 @@ class AsyncTasksResource(AsyncAPIResource):
         *,
         base64_urls: bool | NotGiven = NOT_GIVEN,
         include_chunks: bool | NotGiven = NOT_GIVEN,
+        wait_for_completion: bool | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -458,20 +477,20 @@ class AsyncTasksResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Task:
         """
-        Retrieves detailed information about a task by its ID, including:
+        Retrieves the current state of a task and, when requested, waits briefly for
+        completion.
 
-        - Processing status
-        - Task configuration
-        - Output data (if processing is complete)
-        - File metadata (name, page count)
-        - Timestamps (created, started, finished)
-        - Presigned URLs for accessing files
+        Returns task details such as processing status, configuration, output (when
+        available), file metadata, and timestamps. If `wait_for_completion=true` is
+        provided, the server will hold the request briefly. If the task does not reach a
+        terminal state during that window, the response will indicate a retry with
+        appropriate headers.
 
-        This endpoint can be used to:
+        Typical uses:
 
-        1. Poll the task status during processing
-        2. Retrieve the final output once processing is complete
-        3. Access task metadata and configuration
+        - Poll a task during processing
+        - Retrieve the final output once processing is complete
+        - Access task metadata and configuration
 
         Args:
           base64_urls: Whether to return base64 encoded URLs. If false, the URLs will be returned as
@@ -479,6 +498,8 @@ class AsyncTasksResource(AsyncAPIResource):
 
           include_chunks: Whether to include chunks in the output response
 
+          wait_for_completion: Whether to wait for the task to complete
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -500,6 +521,7 @@ class AsyncTasksResource(AsyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
+                        "wait_for_completion": wait_for_completion,
                     },
                     task_get_params.TaskGetParams,
                 ),

chunkr_ai/types/task.py

@@ -827,22 +827,29 @@ class Configuration(BaseModel):
     segment_processing: ConfigurationSegmentProcessing
     """Defines how each segment type is handled when generating the final output.
 
-    Each segment uses one of three strategies. The chosen strategy controls: •
-    Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-    content is produced (rule-based vs. LLM). • The output format (`Html` or
-    `Markdown`).
+    Each segment uses one of three strategies. The chosen strategy controls:
+
+    - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+    - How the content is produced (rule-based vs. LLM).
+    - The output format (`Html` or `Markdown`).
 
     Optional flags such as image **cropping**, **extended context**, and
     **descriptions** further refine behaviour.
 
-    **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-    `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-    (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-    cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-    `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+    **Default strategy per segment**
+
+    - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+      (Markdown, description off)
+    - `Table` → **LLM** (HTML, description on)
+    - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+    - `Formula`, `Page` → **LLM** (Markdown, description off)
+    - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+
+    **Strategy reference**
 
-    **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-    generate content with an LLM. • **Ignore** – exclude the segment entirely.
+    - **Auto** – rule-based content generation.
+    - **LLM** – generate content with an LLM.
+    - **Ignore** – exclude the segment entirely.
     """
 
     segmentation_strategy: Literal["LayoutAnalysis", "Page"]

chunkr_ai/types/task_get_params.py

@@ -16,3 +16,6 @@ class TaskGetParams(TypedDict, total=False):
 
     include_chunks: bool
     """Whether to include chunks in the output response"""
+
+    wait_for_completion: bool
+    """Whether to wait for the task to complete"""

chunkr_ai/types/tasks/parse_create_params.py

@@ -36,8 +36,8 @@ class ParseCreateParams(TypedDict, total=False):
     file: Required[str]
     """The file to be uploaded. Supported inputs:
 
-    - `ch://files/{file_id}`: References a previously uploaded file you own
-      (authorization enforced)
+    - `ch://files/{file_id}`: Reference to an existing file. Upload via the Files
+      API
     - `http(s)://...`: Remote URL to fetch
     - `data:*;base64,...` or raw base64 string
     """
@@ -83,22 +83,29 @@ class ParseCreateParams(TypedDict, total=False):
     segment_processing: Optional[SegmentProcessing]
     """Defines how each segment type is handled when generating the final output.
 
-    Each segment uses one of three strategies. The chosen strategy controls: •
-    Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-    content is produced (rule-based vs. LLM). • The output format (`Html` or
-    `Markdown`).
+    Each segment uses one of three strategies. The chosen strategy controls:
+
+    - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+    - How the content is produced (rule-based vs. LLM).
+    - The output format (`Html` or `Markdown`).
 
     Optional flags such as image **cropping**, **extended context**, and
     **descriptions** further refine behaviour.
 
-    **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-    `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-    (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-    cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-    `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+    **Default strategy per segment**
+
+    - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+      (Markdown, description off)
+    - `Table` → **LLM** (HTML, description on)
+    - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+    - `Formula`, `Page` → **LLM** (Markdown, description off)
+    - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+
+    **Strategy reference**
 
-    **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-    generate content with an LLM. • **Ignore** – exclude the segment entirely.
+    - **Auto** – rule-based content generation.
+    - **LLM** – generate content with an LLM.
+    - **Ignore** – exclude the segment entirely.
     """
 
     segmentation_strategy: Optional[Literal["LayoutAnalysis", "Page"]]

chunkr_ai/types/tasks/parse_update_params.py

@@ -77,22 +77,29 @@ class ParseUpdateParams(TypedDict, total=False):
     segment_processing: Optional[SegmentProcessing]
     """Defines how each segment type is handled when generating the final output.
 
-    Each segment uses one of three strategies. The chosen strategy controls: •
-    Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`). • How the
-    content is produced (rule-based vs. LLM). • The output format (`Html` or
-    `Markdown`).
+    Each segment uses one of three strategies. The chosen strategy controls:
+
+    - Whether the segment is kept (`Auto`, `LLM`) or skipped (`Ignore`).
+    - How the content is produced (rule-based vs. LLM).
+    - The output format (`Html` or `Markdown`).
 
     Optional flags such as image **cropping**, **extended context**, and
     **descriptions** further refine behaviour.
 
-    **Default strategy per segment** • `Title`, `SectionHeader`, `Text`, `ListItem`,
-    `Caption`, `Footnote` → **Auto** (Markdown, description off) • `Table` → **LLM**
-    (HTML, description on) • `Picture` → **LLM** (Markdown, description off,
-    cropping _All_) • `Formula`, `Page` → **LLM** (Markdown, description off) •
-    `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+    **Default strategy per segment**
+
+    - `Title`, `SectionHeader`, `Text`, `ListItem`, `Caption`, `Footnote` → **Auto**
+      (Markdown, description off)
+    - `Table` → **LLM** (HTML, description on)
+    - `Picture` → **LLM** (Markdown, description off, cropping _All_)
+    - `Formula`, `Page` → **LLM** (Markdown, description off)
+    - `PageHeader`, `PageFooter` → **Ignore** (removed from output)
+
+    **Strategy reference**
 
-    **Strategy reference** • **Auto** – rule-based content generation. • **LLM** –
-    generate content with an LLM. • **Ignore** – exclude the segment entirely.
+    - **Auto** – rule-based content generation.
+    - **LLM** – generate content with an LLM.
+    - **Ignore** – exclude the segment entirely.
     """
 
     segmentation_strategy: Optional[Literal["LayoutAnalysis", "Page"]]

{chunkr_ai-0.1.0a4.dist-info → chunkr_ai-0.1.0a6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: chunkr-ai
-Version: 0.1.0a4
+Version: 0.1.0a6
 Summary: The official Python library for the chunkr API
 Project-URL: Homepage, https://github.com/lumina-ai-inc/chunkr-python
 Project-URL: Repository, https://github.com/lumina-ai-inc/chunkr-python
@@ -299,7 +299,7 @@ Error codes are as follows:
 
 ### Retries
 
-Certain errors are automatically retried 2 times by default, with a short exponential backoff.
+Certain errors are automatically retried 50 times by default, with a short exponential backoff.
 Connection errors (for example, due to a network connectivity problem), 408 Request Timeout, 409 Conflict,
 429 Rate Limit, and >=500 Internal errors are all retried by default.
 
@@ -322,7 +322,7 @@ client.with_options(max_retries=5).tasks.parse.create(
 
 ### Timeouts
 
-By default requests time out after 1 minute. You can configure this with a `timeout` option,
+By default requests time out after 30 seconds. You can configure this with a `timeout` option,
 which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/timeouts/#fine-tuning-the-configuration) object:
 
 ```python
@@ -330,7 +330,7 @@ from chunkr_ai import Chunkr
 
 # Configure the default for all requests:
 client = Chunkr(
-    # 20 seconds (default is 1 minute)
+    # 20 seconds (default is 30 seconds)
     timeout=20.0,
 )
 

{chunkr_ai-0.1.0a4.dist-info → chunkr_ai-0.1.0a6.dist-info}/RECORD

@@ -1,8 +1,8 @@
-chunkr_ai/__init__.py,sha256=RqteAJ-1Ma7DnoNz_AJJIjmynGeoGJ3F4ZKrSnjp9zs,2793
+chunkr_ai/__init__.py,sha256=scS30uHiCpLbaalKTAJSCFSTqnu_b9R5JCkTu2hmbzU,2587
 chunkr_ai/_base_client.py,sha256=Nv5b_rmVdmmPbF42mlOfymbSC6lxcYsrsvBhKSBDXWQ,67038
-chunkr_ai/_client.py,sha256=fseZHGtnXGw3uSa1Le8SxH2oSBeHczn6mOsLeLGj4rY,15867
+chunkr_ai/_client.py,sha256=yn0QdzDkm0M6Ft2-ItmfJpUxQnJVoWa29tSv_2g3KDQ,15975
 chunkr_ai/_compat.py,sha256=VWemUKbj6DDkQ-O4baSpHVLJafotzeXmCQGJugfVTIw,6580
-chunkr_ai/_constants.py,sha256=S14PFzyN9-I31wiV7SmIlL5Ga0MLHxdvegInGdXH7tM,462
+chunkr_ai/_constants.py,sha256=SZppb_i55UWs0n0_MRbw7s0Hy_TOGQu9q7FVd-fCwgM,466
 chunkr_ai/_exceptions.py,sha256=ClgXUcwf4qhBTXnK4LzUPQCFdFldRxAlcYdOFFgpTxA,3220
 chunkr_ai/_files.py,sha256=SUFtic_gwSzbvhLtMdQ7TBem8szrqZE2nZFFMRa0KTw,3619
 chunkr_ai/_models.py,sha256=KvjsMfb88XZlFUKVoOxr8OyDj47MhoH2OKqWNEbBhk4,30010
@@ -11,7 +11,7 @@ chunkr_ai/_resource.py,sha256=f5tiwjxcKdbeMor8idoHtMFTUhqD9yc2xXtq5rqeLLk,1100
 chunkr_ai/_response.py,sha256=xXNpF53hiYARmAW7npKuxQ5UHAEjgAzm7ME_L3eIstY,28800
 chunkr_ai/_streaming.py,sha256=ZmyrVWk7-AWkLAATR55WgNxnyFzYmaqJt2LthA_PTqQ,10100
 chunkr_ai/_types.py,sha256=dnzU2Q2tLcuk29QFEcnPC1wp0-4XB4Cpef_3AnRhV5Y,6200
-chunkr_ai/_version.py,sha256=hJYiv4ePWLGN-Ur1VkK5zJERczdAZjPDNh7APrmHgBE,169
+chunkr_ai/_version.py,sha256=Wa4VwBGgQILEshUij6ZGjQhFefcxwDJV1YTUEWRgG6c,169
 chunkr_ai/pagination.py,sha256=bT-ErcJ80YlKBV6tWq2s9uqg-wv7o66SKe_AgUAGrKc,3533
 chunkr_ai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chunkr_ai/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
@@ -25,13 +25,12 @@ chunkr_ai/_utils/_transform.py,sha256=n7kskEWz6o__aoNvhFoGVyDoalNe6mJwp-g7BWkdj8
 chunkr_ai/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,4602
 chunkr_ai/_utils/_utils.py,sha256=ts4CiiuNpFiGB6YMdkQRh2SZvYvsl7mAF-JWHCcLDf4,12312
 chunkr_ai/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
-chunkr_ai/lib/tasks_poll.py,sha256=3yosl_hH5j6NVNH9mANqneAW0FJSbIV9dMoTcF-OdJU,3341
 chunkr_ai/resources/__init__.py,sha256=K-axuAEg2pJQl45N5ao1tm8AnRwpQVVNp_b6qSMgB6A,1426
-chunkr_ai/resources/files.py,sha256=Dez080pD_xUr1jOW3y6QSg92sSSZhEYObPze2RWktoY,26304
+chunkr_ai/resources/files.py,sha256=iX6LbX2PqM6kFKNoLxS_R9OGaVSnnZJ8U0dCUxNBGIM,27184
 chunkr_ai/resources/health.py,sha256=XTvUtRs5hEK-uccb_40mcIex85eEUo1a171nQUjpSOs,4965
 chunkr_ai/resources/tasks/__init__.py,sha256=W-sclAx_Kfm7OBGlSs694QzNCMkewtz9LU9KRcb8Ud0,976
-chunkr_ai/resources/tasks/parse.py,sha256=um0sw2ZU7bY6AK7LKAS0GxAHUyuSzav_NlhxXPjNjxY,28491
-chunkr_ai/resources/tasks/tasks.py,sha256=UC15zZNpY7u85X_JJudDuNrnpaeULiISaBde4BRHGSw,21653
+chunkr_ai/resources/tasks/parse.py,sha256=cMKxGPFqg8h5HutUx0l7J_namzsYzayWKoQVnlnJ6qA,28999
+chunkr_ai/resources/tasks/tasks.py,sha256=XkNulmXZz4N6UXaG-6EdS-WAyncTgwMY3BYbJBqeEGw,22745
 chunkr_ai/types/__init__.py,sha256=DSRAMgXVRTZM2t8s2yrFU-FHt3FTs_wpZfVILH1zjJ0,728
 chunkr_ai/types/delete.py,sha256=EU78fjXpc8-fqvgcFTuJ0ejs5u_UjbhOz5frkeUHvxY,225
 chunkr_ai/types/file.py,sha256=kOxR0g-3A-qOxz2cjuTcq0wFMqPoph9uQuLYQ56zb-c,718
@@ -41,13 +40,13 @@ chunkr_ai/types/file_url.py,sha256=L434WnOXkNmt59dJiaAgT1_3pN3BIsxm2q14zHQK6xY,3
 chunkr_ai/types/file_url_params.py,sha256=ZHfKiy_6B25StdDemulavGcsPggNNMKLWf6KN7xfPTY,413
 chunkr_ai/types/files_list_response.py,sha256=ggSRWhTzZWjcDXxStyCzrYICXXB5TqnL2j-SN9mHH_g,506
 chunkr_ai/types/health_check_response.py,sha256=6Zn5YYHCQf2RgMjDlf39mtiTPqfaBfC9Vv599U_rKCI,200
-chunkr_ai/types/task.py,sha256=aew6aT0ngKtwgfUCSCvMTJOBQL1Xp0F0otB_wxumIGQ,46703
-chunkr_ai/types/task_get_params.py,sha256=Nx2luhebcoaiuRln4KP4FarWvBPd1OYi__efi56zHPM,460
+chunkr_ai/types/task.py,sha256=L8vE_q0Hej_YuJM_rd_bZOg8kHbithsFx6fOQYpH0cY,46702
+chunkr_ai/types/task_get_params.py,sha256=yGMHRfkbLzQpRLdF_Dj-8TqcioEhDNWyVbEt50xDAP0,542
 chunkr_ai/types/task_list_params.py,sha256=fCku42QW6QUsLmZgKJBaxisGvUcmcQ5fa6LgHHRIwiQ,1043
 chunkr_ai/types/tasks/__init__.py,sha256=VdLEmQvgPoiykSEYaRhkMYVaIueGDkR4P_MjCq9SbQY,267
-chunkr_ai/types/tasks/parse_create_params.py,sha256=PBg2VR_OnBdB8K4NihuefGJXgUXBn7v5317LZG7PDks,34340
-chunkr_ai/types/tasks/parse_update_params.py,sha256=B1cKfdX_cNDh0m2zDoH0FiZP_Qc-a5GFy-5iXHDHuy8,34113
-chunkr_ai-0.1.0a4.dist-info/METADATA,sha256=241RRJb1pTZBg9IG3oGx3_WASrJDN8a2myyJhQ9TUNE,16441
-chunkr_ai-0.1.0a4.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-chunkr_ai-0.1.0a4.dist-info/licenses/LICENSE,sha256=3FDRL-L-DFkrFy8yJpb1Nxhuztm0PB2kawcCgK5utFg,11336
-chunkr_ai-0.1.0a4.dist-info/RECORD,,
+chunkr_ai/types/tasks/parse_create_params.py,sha256=lUSnRHq_nT4nW9boB80HvpVcnjHwIpfQoxBKugFRC7Y,34324
+chunkr_ai/types/tasks/parse_update_params.py,sha256=VrzK-Y4S02fficwIr4EatD73_4i40lOAxMEYDN3KZEQ,34112
+chunkr_ai-0.1.0a6.dist-info/METADATA,sha256=Kbt8umfSyxxm5O7Xol6ypvmc7yprG33YFE3L-TqGx6Y,16446
+chunkr_ai-0.1.0a6.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+chunkr_ai-0.1.0a6.dist-info/licenses/LICENSE,sha256=3FDRL-L-DFkrFy8yJpb1Nxhuztm0PB2kawcCgK5utFg,11336
+chunkr_ai-0.1.0a6.dist-info/RECORD,,

chunkr_ai/lib/tasks_poll.py

@@ -1,122 +0,0 @@
-from __future__ import annotations
-
-"""
-Custom helpers for task polling.
-
-This module adds `Task.poll()` and `Task.apoll()` methods at runtime to the
-generated `Task` model, without modifying generated code directly.
-
-Usage:
-    task = client.tasks.get(task_id)
-    task = task.poll(client)  # blocks until terminal state
-
-    # async
-    task = await async_client.tasks.get(task_id)
-    task = await task.apoll(async_client)
-"""
-
-import time
-import asyncio
-from typing import Protocol, cast
-
-from .._types import NOT_GIVEN, NotGiven
-from .._client import Chunkr, AsyncChunkr
-from ..types.task import Task as _Task
-from .._exceptions import ChunkrError
-
-TERMINAL_STATUSES = {"Succeeded", "Failed", "Cancelled"}
-
-
-def _task_poll(
-    self: _Task,
-    client: Chunkr,
-    *,
-    interval: float = 0.5,
-    timeout: float = 600.0,
-    include_chunks: bool | NotGiven = NOT_GIVEN,
-    base64_urls: bool | NotGiven = NOT_GIVEN,
-) -> _Task:
-    """Poll the task until it reaches a terminal status.
-
-    Args:
-        client: Synchronous Chunkr client instance.
-        interval: Seconds to sleep between polls.
-        timeout: Maximum total seconds to wait before raising an error.
-        include_chunks: Whether to include chunks in the output response for each poll.
-        base64_urls: Whether to return base64 encoded URLs.
-    """
-    start_time = time.monotonic()
-    current: _Task = self
-
-    class _TasksGetProtocol(Protocol):
-        def get(
-            self,
-            task_id: str,
-            *,
-            base64_urls: bool | NotGiven = NOT_GIVEN,
-            include_chunks: bool | NotGiven = NOT_GIVEN,
-        ) -> _Task: ...
-
-    resource = cast(_TasksGetProtocol, client.tasks)
-
-    while current.status not in TERMINAL_STATUSES:
-        if time.monotonic() - start_time > timeout:
-            raise ChunkrError("Task polling timed out.")
-
-        if interval > 0:
-            time.sleep(interval)
-
-        current = resource.get(
-            current.task_id,
-            include_chunks=include_chunks,
-            base64_urls=base64_urls,
-        )
-
-    return current
-
-
-async def _task_apoll(
-    self: _Task,
-    client: AsyncChunkr,
-    *,
-    interval: float = 0.5,
-    timeout: float = 600.0,
-    include_chunks: bool | NotGiven = NOT_GIVEN,
-    base64_urls: bool | NotGiven = NOT_GIVEN,
-) -> _Task:
-    """Async poll the task until it reaches a terminal status."""
-    start_time = time.monotonic()
-    current: _Task = self
-
-    class _AsyncTasksGetProtocol(Protocol):
-        async def get(
-            self,
-            task_id: str,
-            *,
-            base64_urls: bool | NotGiven = NOT_GIVEN,
-            include_chunks: bool | NotGiven = NOT_GIVEN,
-        ) -> _Task: ...
-
-    aresource = cast(_AsyncTasksGetProtocol, client.tasks)
-
-    while current.status not in TERMINAL_STATUSES:
-        if time.monotonic() - start_time > timeout:
-            raise ChunkrError("Task polling timed out.")
-
-        if interval > 0:
-            await asyncio.sleep(interval)
-
-        current = await aresource.get(
-            current.task_id,
-            include_chunks=include_chunks,
-            base64_urls=base64_urls,
-        )
-
-    return current
-
-
-# Attach methods to the generated Task model
-_Task.poll = _task_poll  # type: ignore[attr-defined]
-_Task.apoll = _task_apoll  # type: ignore[attr-defined]
-
-