chunkr-ai 0.1.0a7__py3-none-any.whl → 0.1.0a9__py3-none-any.whl

chunkr_ai/resources/tasks/extract.py

@@ -6,7 +6,7 @@ from typing import Optional
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -50,16 +50,16 @@ class ExtractResource(SyncAPIResource):
         *,
         file: str,
         schema: object,
-        expires_in: Optional[int] | NotGiven = NOT_GIVEN,
-        file_name: Optional[str] | NotGiven = NOT_GIVEN,
-        parse_configuration: Optional[ParseConfigurationParam] | NotGiven = NOT_GIVEN,
-        system_prompt: Optional[str] | NotGiven = NOT_GIVEN,
+        expires_in: Optional[int] | Omit = omit,
+        file_name: Optional[str] | Omit = omit,
+        parse_configuration: Optional[ParseConfigurationParam] | Omit = omit,
+        system_prompt: Optional[str] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> ExtractCreateResponse:
         """
@@ -129,25 +129,20 @@ class ExtractResource(SyncAPIResource):
         self,
         task_id: Optional[str],
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        wait_for_completion: bool | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        include_chunks: bool | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> ExtractGetResponse:
         """
-        Retrieves the current state of an extract task and, when requested, can wait for
-        completion.
+        Retrieves the current state of an extract task.
 
         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.
+        available), file metadata, and timestamps.
 
         Typical uses:
 
@@ -161,8 +156,6 @@ class ExtractResource(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
@@ -184,7 +177,6 @@ class ExtractResource(SyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
-                        "wait_for_completion": wait_for_completion,
                     },
                     extract_get_params.ExtractGetParams,
                 ),
@@ -218,16 +210,16 @@ class AsyncExtractResource(AsyncAPIResource):
         *,
         file: str,
         schema: object,
-        expires_in: Optional[int] | NotGiven = NOT_GIVEN,
-        file_name: Optional[str] | NotGiven = NOT_GIVEN,
-        parse_configuration: Optional[ParseConfigurationParam] | NotGiven = NOT_GIVEN,
-        system_prompt: Optional[str] | NotGiven = NOT_GIVEN,
+        expires_in: Optional[int] | Omit = omit,
+        file_name: Optional[str] | Omit = omit,
+        parse_configuration: Optional[ParseConfigurationParam] | Omit = omit,
+        system_prompt: Optional[str] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> ExtractCreateResponse:
         """
@@ -297,25 +289,20 @@ class AsyncExtractResource(AsyncAPIResource):
         self,
         task_id: Optional[str],
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        wait_for_completion: bool | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        include_chunks: bool | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> ExtractGetResponse:
         """
-        Retrieves the current state of an extract task and, when requested, can wait for
-        completion.
+        Retrieves the current state of an extract task.
 
         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.
+        available), file metadata, and timestamps.
 
         Typical uses:
 
@@ -329,8 +316,6 @@ class AsyncExtractResource(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
@@ -352,7 +337,6 @@ class AsyncExtractResource(AsyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
-                        "wait_for_completion": wait_for_completion,
                     },
                     extract_get_params.ExtractGetParams,
                 ),

chunkr_ai/resources/tasks/parse.py

@@ -7,7 +7,7 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -19,7 +19,6 @@ from ..._response import (
 )
 from ...types.tasks import parse_get_params, parse_create_params
 from ..._base_client import make_request_options
-from ...types.llm_processing_param import LlmProcessingParam
 from ...types.chunk_processing_param import ChunkProcessingParam
 from ...types.segment_processing_param import SegmentProcessingParam
 from ...types.tasks.parse_get_response import ParseGetResponse
@@ -52,21 +51,20 @@ class ParseResource(SyncAPIResource):
         self,
         *,
         file: str,
-        chunk_processing: ChunkProcessingParam | NotGiven = NOT_GIVEN,
-        error_handling: Literal["Fail", "Continue"] | NotGiven = NOT_GIVEN,
-        expires_in: Optional[int] | NotGiven = NOT_GIVEN,
-        file_name: Optional[str] | NotGiven = NOT_GIVEN,
-        llm_processing: LlmProcessingParam | NotGiven = NOT_GIVEN,
-        ocr_strategy: Literal["All", "Auto"] | NotGiven = NOT_GIVEN,
-        pipeline: Literal["Azure", "Chunkr"] | NotGiven = NOT_GIVEN,
-        segment_processing: Optional[SegmentProcessingParam] | NotGiven = NOT_GIVEN,
-        segmentation_strategy: Literal["LayoutAnalysis", "Page"] | NotGiven = NOT_GIVEN,
+        chunk_processing: ChunkProcessingParam | Omit = omit,
+        error_handling: Literal["Fail", "Continue"] | Omit = omit,
+        expires_in: Optional[int] | Omit = omit,
+        file_name: Optional[str] | Omit = omit,
+        ocr_strategy: Literal["All", "Auto"] | Omit = omit,
+        pipeline: Literal["Azure", "Chunkr"] | Omit = omit,
+        segment_processing: Optional[SegmentProcessingParam] | Omit = omit,
+        segmentation_strategy: Literal["LayoutAnalysis", "Page"] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> ParseCreateResponse:
         """
@@ -99,8 +97,6 @@ class ParseResource(SyncAPIResource):
 
           file_name: The name of the file to be parsed. If not set a name will be generated.
 
-          llm_processing: Controls the LLM used for the task.
-
           ocr_strategy: Controls the Optical Character Recognition (OCR) strategy.
 
               - `All`: Processes all pages with OCR. (Latency penalty: ~0.5 seconds per page)
@@ -150,7 +146,6 @@ class ParseResource(SyncAPIResource):
                     "error_handling": error_handling,
                     "expires_in": expires_in,
                     "file_name": file_name,
-                    "llm_processing": llm_processing,
                     "ocr_strategy": ocr_strategy,
                     "pipeline": pipeline,
                     "segment_processing": segment_processing,
@@ -172,25 +167,20 @@ class ParseResource(SyncAPIResource):
         self,
         task_id: Optional[str],
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        wait_for_completion: bool | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        include_chunks: bool | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> ParseGetResponse:
         """
-        Retrieves the current state of a parse task and, when requested, can wait for
-        completion.
+        Retrieves the current state of a parse task.
 
         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.
+        available), file metadata, and timestamps.
 
         Typical uses:
 
@@ -204,8 +194,6 @@ class ParseResource(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
@@ -227,7 +215,6 @@ class ParseResource(SyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
-                        "wait_for_completion": wait_for_completion,
                     },
                     parse_get_params.ParseGetParams,
                 ),
@@ -260,21 +247,20 @@ class AsyncParseResource(AsyncAPIResource):
         self,
         *,
         file: str,
-        chunk_processing: ChunkProcessingParam | NotGiven = NOT_GIVEN,
-        error_handling: Literal["Fail", "Continue"] | NotGiven = NOT_GIVEN,
-        expires_in: Optional[int] | NotGiven = NOT_GIVEN,
-        file_name: Optional[str] | NotGiven = NOT_GIVEN,
-        llm_processing: LlmProcessingParam | NotGiven = NOT_GIVEN,
-        ocr_strategy: Literal["All", "Auto"] | NotGiven = NOT_GIVEN,
-        pipeline: Literal["Azure", "Chunkr"] | NotGiven = NOT_GIVEN,
-        segment_processing: Optional[SegmentProcessingParam] | NotGiven = NOT_GIVEN,
-        segmentation_strategy: Literal["LayoutAnalysis", "Page"] | NotGiven = NOT_GIVEN,
+        chunk_processing: ChunkProcessingParam | Omit = omit,
+        error_handling: Literal["Fail", "Continue"] | Omit = omit,
+        expires_in: Optional[int] | Omit = omit,
+        file_name: Optional[str] | Omit = omit,
+        ocr_strategy: Literal["All", "Auto"] | Omit = omit,
+        pipeline: Literal["Azure", "Chunkr"] | Omit = omit,
+        segment_processing: Optional[SegmentProcessingParam] | Omit = omit,
+        segmentation_strategy: Literal["LayoutAnalysis", "Page"] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> ParseCreateResponse:
         """
@@ -307,8 +293,6 @@ class AsyncParseResource(AsyncAPIResource):
 
           file_name: The name of the file to be parsed. If not set a name will be generated.
 
-          llm_processing: Controls the LLM used for the task.
-
           ocr_strategy: Controls the Optical Character Recognition (OCR) strategy.
 
               - `All`: Processes all pages with OCR. (Latency penalty: ~0.5 seconds per page)
@@ -358,7 +342,6 @@ class AsyncParseResource(AsyncAPIResource):
                     "error_handling": error_handling,
                     "expires_in": expires_in,
                     "file_name": file_name,
-                    "llm_processing": llm_processing,
                     "ocr_strategy": ocr_strategy,
                     "pipeline": pipeline,
                     "segment_processing": segment_processing,
@@ -380,25 +363,20 @@ class AsyncParseResource(AsyncAPIResource):
         self,
         task_id: Optional[str],
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        wait_for_completion: bool | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        include_chunks: bool | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> ParseGetResponse:
         """
-        Retrieves the current state of a parse task and, when requested, can wait for
-        completion.
+        Retrieves the current state of a parse task.
 
         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.
+        available), file metadata, and timestamps.
 
         Typical uses:
 
@@ -412,8 +390,6 @@ class AsyncParseResource(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
@@ -435,7 +411,6 @@ class AsyncParseResource(AsyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
-                        "wait_for_completion": wait_for_completion,
                     },
                     parse_get_params.ParseGetParams,
                 ),

chunkr_ai/resources/tasks/tasks.py

@@ -25,7 +25,7 @@ from .extract import (
     ExtractResourceWithStreamingResponse,
     AsyncExtractResourceWithStreamingResponse,
 )
-from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
+from ..._types import Body, Omit, Query, Headers, NoneType, NotGiven, omit, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -73,21 +73,21 @@ class TasksResource(SyncAPIResource):
     def list(
         self,
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        cursor: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        end: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
-        sort: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
-        start: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        statuses: List[Literal["Starting", "Processing", "Succeeded", "Failed", "Cancelled"]] | NotGiven = NOT_GIVEN,
-        task_types: List[Literal["Parse", "Extract"]] | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        cursor: Union[str, datetime] | Omit = omit,
+        end: Union[str, datetime] | Omit = omit,
+        include_chunks: bool | Omit = omit,
+        limit: int | Omit = omit,
+        sort: Literal["asc", "desc"] | Omit = omit,
+        start: Union[str, datetime] | Omit = omit,
+        statuses: List[Literal["Starting", "Processing", "Succeeded", "Failed", "Cancelled"]] | Omit = omit,
+        task_types: List[Literal["Parse", "Extract"]] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SyncTasksPage[TaskResponse]:
         """
         Lists tasks for the authenticated user with cursor-based pagination and optional
@@ -157,7 +157,7 @@ class TasksResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> None:
         """
@@ -202,7 +202,7 @@ class TasksResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> None:
         """
         Cancel a task that hasn't started processing yet:
@@ -238,25 +238,20 @@ class TasksResource(SyncAPIResource):
         self,
         task_id: Optional[str],
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        wait_for_completion: bool | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        include_chunks: bool | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskResponse:
         """
-        Retrieves the current state of a task and, when requested, can wait for
-        completion.
+        Retrieves the current state of a task.
 
         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.
+        available), file metadata, and timestamps.
 
         Typical uses:
 
@@ -270,8 +265,6 @@ 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
@@ -293,7 +286,6 @@ class TasksResource(SyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
-                        "wait_for_completion": wait_for_completion,
                     },
                     task_get_params.TaskGetParams,
                 ),
@@ -333,21 +325,21 @@ class AsyncTasksResource(AsyncAPIResource):
     def list(
         self,
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        cursor: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        end: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
-        sort: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
-        start: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        statuses: List[Literal["Starting", "Processing", "Succeeded", "Failed", "Cancelled"]] | NotGiven = NOT_GIVEN,
-        task_types: List[Literal["Parse", "Extract"]] | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        cursor: Union[str, datetime] | Omit = omit,
+        end: Union[str, datetime] | Omit = omit,
+        include_chunks: bool | Omit = omit,
+        limit: int | Omit = omit,
+        sort: Literal["asc", "desc"] | Omit = omit,
+        start: Union[str, datetime] | Omit = omit,
+        statuses: List[Literal["Starting", "Processing", "Succeeded", "Failed", "Cancelled"]] | Omit = omit,
+        task_types: List[Literal["Parse", "Extract"]] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AsyncPaginator[TaskResponse, AsyncTasksPage[TaskResponse]]:
         """
         Lists tasks for the authenticated user with cursor-based pagination and optional
@@ -417,7 +409,7 @@ class AsyncTasksResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> None:
         """
@@ -462,7 +454,7 @@ class AsyncTasksResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> None:
         """
         Cancel a task that hasn't started processing yet:
@@ -498,25 +490,20 @@ class AsyncTasksResource(AsyncAPIResource):
         self,
         task_id: Optional[str],
         *,
-        base64_urls: bool | NotGiven = NOT_GIVEN,
-        include_chunks: bool | NotGiven = NOT_GIVEN,
-        wait_for_completion: bool | NotGiven = NOT_GIVEN,
+        base64_urls: bool | Omit = omit,
+        include_chunks: bool | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskResponse:
         """
-        Retrieves the current state of a task and, when requested, can wait for
-        completion.
+        Retrieves the current state of a task.
 
         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.
+        available), file metadata, and timestamps.
 
         Typical uses:
 
@@ -530,8 +517,6 @@ 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
@@ -553,7 +538,6 @@ class AsyncTasksResource(AsyncAPIResource):
                     {
                         "base64_urls": base64_urls,
                         "include_chunks": include_chunks,
-                        "wait_for_completion": wait_for_completion,
                     },
                     task_get_params.TaskGetParams,
                 ),

chunkr_ai/resources/webhooks.py

@@ -7,7 +7,7 @@ from typing import Mapping, cast
 
 import httpx
 
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import Body, Query, Headers, NotGiven, not_given
 from .._compat import cached_property
 from .._models import construct_type
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -79,7 +79,7 @@ class WebhooksResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WebhookURLResponse:
         """Get or create webhook for user and return dashboard URL"""
         return self._get(
@@ -145,7 +145,7 @@ class AsyncWebhooksResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WebhookURLResponse:
         """Get or create webhook for user and return dashboard URL"""
         return await self._get(

chunkr_ai/types/__init__.py

@@ -15,7 +15,6 @@ from .ocr_result import OcrResult as OcrResult
 from .bounding_box import BoundingBox as BoundingBox
 from .version_info import VersionInfo as VersionInfo
 from .task_response import TaskResponse as TaskResponse
-from .llm_processing import LlmProcessing as LlmProcessing
 from .file_url_params import FileURLParams as FileURLParams
 from .task_get_params import TaskGetParams as TaskGetParams
 from .chunk_processing import ChunkProcessing as ChunkProcessing
@@ -26,7 +25,6 @@ from .file_create_params import FileCreateParams as FileCreateParams
 from .segment_processing import SegmentProcessing as SegmentProcessing
 from .files_list_response import FilesListResponse as FilesListResponse
 from .parse_configuration import ParseConfiguration as ParseConfiguration
-from .llm_processing_param import LlmProcessingParam as LlmProcessingParam
 from .unwrap_webhook_event import UnwrapWebhookEvent as UnwrapWebhookEvent
 from .webhook_url_response import WebhookURLResponse as WebhookURLResponse
 from .extract_configuration import ExtractConfiguration as ExtractConfiguration

chunkr_ai/types/extract_output_response.py

@@ -7,13 +7,56 @@ __all__ = ["ExtractOutputResponse"]
 
 class ExtractOutputResponse(BaseModel):
     citations: object
-    """Mirror of `results`; leaves are `Vec<Citation>` for the corresponding field"""
+    """Mirror of `results`; leaves are `Vec<Citation>` for the corresponding field
+
+    Example:
+
+    ```json
+    {
+      "field_name": [
+        {
+          "citation_id": "abc1234",
+          "citation_type": "Segment",
+          "bboxes": [
+            {
+              "left": 10,
+              "top": 20,
+              "width": 100,
+              "height": 18
+            }
+          ],
+          "content": "Example content",
+          "segment_id": "seg_001",
+          "segment_type": "Text",
+          "page_number": 1,
+          "page_height": 297,
+          "page_width": 210,
+          "ss_ranges": ["A1:C10"],
+          "ss_sheet_name": "Sheet1"
+        }
+      ]
+    }
+    ```
+    """
 
     metrics: object
     """
     Mirror of `results`; leaves contain a `Metrics` object for the corresponding
     field
+
+    Example:
+
+    ```json
+    { "field_name": { "confidence": "High" } }
+    ```
     """
 
     results: object
-    """JSON data that matches the provided schema"""
+    """JSON data that matches the provided schema
+
+    Example:
+
+    ```json
+    { "field_name": "value" }
+    ```
+    """