parallel-web 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

parallel/resources/beta/task_group.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 is_given, maybe_transform, strip_not_given, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -59,13 +59,13 @@ class TaskGroupResource(SyncAPIResource):
     def create(
         self,
         *,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, 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,
     ) -> TaskGroup:
         """
         Initiates a TaskGroup to group and track multiple runs.
@@ -99,7 +99,7 @@ class TaskGroupResource(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,
     ) -> TaskGroup:
         """
         Retrieves aggregated status across runs in a TaskGroup.
@@ -128,14 +128,14 @@ class TaskGroupResource(SyncAPIResource):
         task_group_id: str,
         *,
         inputs: Iterable[BetaRunInputParam],
-        default_task_spec: Optional[TaskSpecParam] | NotGiven = NOT_GIVEN,
-        betas: List[ParallelBetaParam] | NotGiven = NOT_GIVEN,
+        default_task_spec: Optional[TaskSpecParam] | Omit = omit,
+        betas: List[ParallelBetaParam] | 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,
     ) -> TaskGroupRunResponse:
         """
         Initiates multiple task runs within a TaskGroup.
@@ -163,7 +163,7 @@ class TaskGroupResource(SyncAPIResource):
         if not task_group_id:
             raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}")
         extra_headers = {
-            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else NOT_GIVEN}),
+            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}),
             **(extra_headers or {}),
         }
         return self._post(
@@ -185,20 +185,20 @@ class TaskGroupResource(SyncAPIResource):
         self,
         task_group_id: str,
         *,
-        last_event_id: Optional[str] | NotGiven = NOT_GIVEN,
-        api_timeout: Optional[float] | NotGiven = NOT_GIVEN,
+        last_event_id: Optional[str] | Omit = omit,
+        api_timeout: Optional[float] | 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,
     ) -> Stream[TaskGroupEventsResponse]:
         """
         Streams events from a TaskGroup: status updates and run completions.
 
-        The connection will remain open for up to 10 minutes as long as at least one run
-        in the TaskGroup is active.
+        The connection will remain open for up to an hour as long as at least one run in
+        the group is still active.
 
         Args:
           extra_headers: Send extra headers
@@ -238,23 +238,32 @@ class TaskGroupResource(SyncAPIResource):
         self,
         task_group_id: str,
         *,
-        include_input: bool | NotGiven = NOT_GIVEN,
-        include_output: bool | NotGiven = NOT_GIVEN,
-        last_event_id: Optional[str] | NotGiven = NOT_GIVEN,
+        include_input: bool | Omit = omit,
+        include_output: bool | Omit = omit,
+        last_event_id: Optional[str] | Omit = omit,
         status: Optional[
             Literal["queued", "action_required", "running", "completed", "failed", "cancelling", "cancelled"]
         ]
-        | NotGiven = NOT_GIVEN,
+        | 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,
     ) -> Stream[TaskGroupGetRunsResponse]:
         """
         Retrieves task runs in a TaskGroup and optionally their inputs and outputs.
 
+        All runs within a TaskGroup are returned as a stream. To get the inputs and/or
+        outputs back in the stream, set the corresponding `include_input` and
+        `include_output` parameters to `true`.
+
+        The stream is resumable using the `event_id` as the cursor. To resume a stream,
+        specify the `last_event_id` parameter with the `event_id` of the last event in
+        the stream. The stream will resume from the next event after the
+        `last_event_id`.
+
         Args:
           extra_headers: Send extra headers
 
@@ -315,13 +324,13 @@ class AsyncTaskGroupResource(AsyncAPIResource):
     async def create(
         self,
         *,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, 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,
     ) -> TaskGroup:
         """
         Initiates a TaskGroup to group and track multiple runs.
@@ -355,7 +364,7 @@ class AsyncTaskGroupResource(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,
     ) -> TaskGroup:
         """
         Retrieves aggregated status across runs in a TaskGroup.
@@ -384,14 +393,14 @@ class AsyncTaskGroupResource(AsyncAPIResource):
         task_group_id: str,
         *,
         inputs: Iterable[BetaRunInputParam],
-        default_task_spec: Optional[TaskSpecParam] | NotGiven = NOT_GIVEN,
-        betas: List[ParallelBetaParam] | NotGiven = NOT_GIVEN,
+        default_task_spec: Optional[TaskSpecParam] | Omit = omit,
+        betas: List[ParallelBetaParam] | 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,
     ) -> TaskGroupRunResponse:
         """
         Initiates multiple task runs within a TaskGroup.
@@ -419,7 +428,7 @@ class AsyncTaskGroupResource(AsyncAPIResource):
         if not task_group_id:
             raise ValueError(f"Expected a non-empty value for `task_group_id` but received {task_group_id!r}")
         extra_headers = {
-            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else NOT_GIVEN}),
+            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}),
             **(extra_headers or {}),
         }
         return await self._post(
@@ -441,20 +450,20 @@ class AsyncTaskGroupResource(AsyncAPIResource):
         self,
         task_group_id: str,
         *,
-        last_event_id: Optional[str] | NotGiven = NOT_GIVEN,
-        api_timeout: Optional[float] | NotGiven = NOT_GIVEN,
+        last_event_id: Optional[str] | Omit = omit,
+        api_timeout: Optional[float] | 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,
     ) -> AsyncStream[TaskGroupEventsResponse]:
         """
         Streams events from a TaskGroup: status updates and run completions.
 
-        The connection will remain open for up to 10 minutes as long as at least one run
-        in the TaskGroup is active.
+        The connection will remain open for up to an hour as long as at least one run in
+        the group is still active.
 
         Args:
           extra_headers: Send extra headers
@@ -494,23 +503,32 @@ class AsyncTaskGroupResource(AsyncAPIResource):
         self,
         task_group_id: str,
         *,
-        include_input: bool | NotGiven = NOT_GIVEN,
-        include_output: bool | NotGiven = NOT_GIVEN,
-        last_event_id: Optional[str] | NotGiven = NOT_GIVEN,
+        include_input: bool | Omit = omit,
+        include_output: bool | Omit = omit,
+        last_event_id: Optional[str] | Omit = omit,
         status: Optional[
             Literal["queued", "action_required", "running", "completed", "failed", "cancelling", "cancelled"]
         ]
-        | NotGiven = NOT_GIVEN,
+        | 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,
     ) -> AsyncStream[TaskGroupGetRunsResponse]:
         """
         Retrieves task runs in a TaskGroup and optionally their inputs and outputs.
 
+        All runs within a TaskGroup are returned as a stream. To get the inputs and/or
+        outputs back in the stream, set the corresponding `include_input` and
+        `include_output` parameters to `true`.
+
+        The stream is resumable using the `event_id` as the cursor. To resume a stream,
+        specify the `last_event_id` parameter with the `event_id` of the last event in
+        the stream. The stream will resume from the next event after the
+        `last_event_id`.
+
         Args:
           extra_headers: Send extra headers
 

parallel/resources/beta/task_run.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Union, Iterable, Optional, cast
 
 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 is_given, maybe_transform, strip_not_given, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -56,19 +56,19 @@ class TaskRunResource(SyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        enable_events: Optional[bool] | NotGiven = NOT_GIVEN,
-        mcp_servers: Optional[Iterable[McpServerParam]] | NotGiven = NOT_GIVEN,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        source_policy: Optional[SourcePolicy] | NotGiven = NOT_GIVEN,
-        task_spec: Optional[TaskSpecParam] | NotGiven = NOT_GIVEN,
-        webhook: Optional[WebhookParam] | NotGiven = NOT_GIVEN,
-        betas: List[ParallelBetaParam] | NotGiven = NOT_GIVEN,
+        enable_events: Optional[bool] | Omit = omit,
+        mcp_servers: Optional[Iterable[McpServerParam]] | Omit = omit,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        source_policy: Optional[SourcePolicy] | Omit = omit,
+        task_spec: Optional[TaskSpecParam] | Omit = omit,
+        webhook: Optional[WebhookParam] | Omit = omit,
+        betas: List[ParallelBetaParam] | 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,
     ) -> TaskRun:
         """
         Initiates a task run.
@@ -87,13 +87,13 @@ class TaskRunResource(SyncAPIResource):
               [Task Run events](https://platform.parallel.ai/api-reference) endpoint. When
               false, no progress events are tracked. Note that progress tracking cannot be
               enabled after a run has been created. The flag is set to true by default for
-              premium processors (pro and above). This feature is not available via the Python
-              SDK. To enable this feature in your API requests, specify the `parallel-beta`
-              header with `events-sse-2025-07-24` value.
+              premium processors (pro and above). To enable this feature in your requests,
+              specify `events-sse-2025-07-24` as one of the values in `parallel-beta` header
+              (for API calls) or `betas` param (for the SDKs).
 
-          mcp_servers: Optional list of MCP servers to use for the run. This feature is not available
-              via the Python SDK. To enable this feature in your API requests, specify the
-              `parallel-beta` header with `mcp-server-2025-07-17` value.
+          mcp_servers: Optional list of MCP servers to use for the run. To enable this feature in your
+              requests, specify `mcp-server-2025-07-17` as one of the values in
+              `parallel-beta` header (for API calls) or `betas` param (for the SDKs).
 
           metadata: User-provided metadata stored with the run. Keys and values must be strings with
               a maximum length of 16 and 512 characters respectively.
@@ -122,7 +122,7 @@ class TaskRunResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         extra_headers = {
-            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else NOT_GIVEN}),
+            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}),
             **(extra_headers or {}),
         }
         return self._post(
@@ -155,7 +155,7 @@ class TaskRunResource(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,
     ) -> Stream[TaskRunEventsResponse]:
         """
         Streams events for a task run.
@@ -192,14 +192,14 @@ class TaskRunResource(SyncAPIResource):
         self,
         run_id: str,
         *,
-        api_timeout: int | NotGiven = NOT_GIVEN,
-        betas: List[ParallelBetaParam] | NotGiven = NOT_GIVEN,
+        api_timeout: int | Omit = omit,
+        betas: List[ParallelBetaParam] | 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,
     ) -> BetaTaskRunResult:
         """
         Retrieves a run result by run_id, blocking until the run is completed.
@@ -218,7 +218,7 @@ class TaskRunResource(SyncAPIResource):
         if not run_id:
             raise ValueError(f"Expected a non-empty value for `run_id` but received {run_id!r}")
         extra_headers = {
-            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else NOT_GIVEN}),
+            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}),
             **(extra_headers or {}),
         }
         return self._get(
@@ -259,19 +259,19 @@ class AsyncTaskRunResource(AsyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        enable_events: Optional[bool] | NotGiven = NOT_GIVEN,
-        mcp_servers: Optional[Iterable[McpServerParam]] | NotGiven = NOT_GIVEN,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        source_policy: Optional[SourcePolicy] | NotGiven = NOT_GIVEN,
-        task_spec: Optional[TaskSpecParam] | NotGiven = NOT_GIVEN,
-        webhook: Optional[WebhookParam] | NotGiven = NOT_GIVEN,
-        betas: List[ParallelBetaParam] | NotGiven = NOT_GIVEN,
+        enable_events: Optional[bool] | Omit = omit,
+        mcp_servers: Optional[Iterable[McpServerParam]] | Omit = omit,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        source_policy: Optional[SourcePolicy] | Omit = omit,
+        task_spec: Optional[TaskSpecParam] | Omit = omit,
+        webhook: Optional[WebhookParam] | Omit = omit,
+        betas: List[ParallelBetaParam] | 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,
     ) -> TaskRun:
         """
         Initiates a task run.
@@ -290,13 +290,13 @@ class AsyncTaskRunResource(AsyncAPIResource):
               [Task Run events](https://platform.parallel.ai/api-reference) endpoint. When
               false, no progress events are tracked. Note that progress tracking cannot be
               enabled after a run has been created. The flag is set to true by default for
-              premium processors (pro and above). This feature is not available via the Python
-              SDK. To enable this feature in your API requests, specify the `parallel-beta`
-              header with `events-sse-2025-07-24` value.
+              premium processors (pro and above). To enable this feature in your requests,
+              specify `events-sse-2025-07-24` as one of the values in `parallel-beta` header
+              (for API calls) or `betas` param (for the SDKs).
 
-          mcp_servers: Optional list of MCP servers to use for the run. This feature is not available
-              via the Python SDK. To enable this feature in your API requests, specify the
-              `parallel-beta` header with `mcp-server-2025-07-17` value.
+          mcp_servers: Optional list of MCP servers to use for the run. To enable this feature in your
+              requests, specify `mcp-server-2025-07-17` as one of the values in
+              `parallel-beta` header (for API calls) or `betas` param (for the SDKs).
 
           metadata: User-provided metadata stored with the run. Keys and values must be strings with
               a maximum length of 16 and 512 characters respectively.
@@ -325,7 +325,7 @@ class AsyncTaskRunResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         extra_headers = {
-            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else NOT_GIVEN}),
+            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}),
             **(extra_headers or {}),
         }
         return await self._post(
@@ -358,7 +358,7 @@ class AsyncTaskRunResource(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,
     ) -> AsyncStream[TaskRunEventsResponse]:
         """
         Streams events for a task run.
@@ -395,14 +395,14 @@ class AsyncTaskRunResource(AsyncAPIResource):
         self,
         run_id: str,
         *,
-        api_timeout: int | NotGiven = NOT_GIVEN,
-        betas: List[ParallelBetaParam] | NotGiven = NOT_GIVEN,
+        api_timeout: int | Omit = omit,
+        betas: List[ParallelBetaParam] | 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,
     ) -> BetaTaskRunResult:
         """
         Retrieves a run result by run_id, blocking until the run is completed.
@@ -421,7 +421,7 @@ class AsyncTaskRunResource(AsyncAPIResource):
         if not run_id:
             raise ValueError(f"Expected a non-empty value for `run_id` but received {run_id!r}")
         extra_headers = {
-            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else NOT_GIVEN}),
+            **strip_not_given({"parallel-beta": ",".join(str(e) for e in betas) if is_given(betas) else not_given}),
             **(extra_headers or {}),
         }
         return await self._get(

parallel/resources/task_run.py

@@ -10,7 +10,7 @@ import httpx
 from parallel.lib._time import prepare_timeout_float
 
 from ..types import task_run_create_params, task_run_result_params
-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
@@ -61,15 +61,15 @@ class TaskRunResource(SyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        source_policy: Optional[SourcePolicy] | NotGiven = NOT_GIVEN,
-        task_spec: Optional[TaskSpecParam] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        source_policy: Optional[SourcePolicy] | Omit = omit,
+        task_spec: Optional[TaskSpecParam] | 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,
     ) -> TaskRun:
         """
         Initiates a task run.
@@ -132,7 +132,7 @@ class TaskRunResource(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,
     ) -> TaskRun:
         """
         Retrieves run status by run_id.
@@ -162,13 +162,13 @@ class TaskRunResource(SyncAPIResource):
         self,
         run_id: str,
         *,
-        api_timeout: int | NotGiven = NOT_GIVEN,
+        api_timeout: int | 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,
     ) -> TaskRunResult:
         """
         Retrieves a run result by run_id, blocking until the run is completed.
@@ -201,7 +201,7 @@ class TaskRunResource(SyncAPIResource):
         *,
         run_id: str,
         deadline: float,
-        output: Optional[OutputSchema] | Type[OutputT] | NotGiven = NOT_GIVEN,
+        output: Optional[OutputSchema] | Type[OutputT] | 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,
@@ -229,14 +229,14 @@ class TaskRunResource(SyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        output: Optional[OutputSchema] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        output: Optional[OutputSchema] | 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,
     ) -> TaskRunResult: ...
     @overload
     def execute(
@@ -244,28 +244,28 @@ class TaskRunResource(SyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
         output: Type[OutputT],
         # 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,
     ) -> ParsedTaskRunResult[OutputT]: ...
     def execute(
         self,
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        output: Optional[OutputSchema] | Type[OutputT] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        output: Optional[OutputSchema] | Type[OutputT] | 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,
     ) -> TaskRunResult | ParsedTaskRunResult[OutputT]:
         """
         Convenience method to create and execute a task run in a single call.
@@ -350,15 +350,15 @@ class AsyncTaskRunResource(AsyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        source_policy: Optional[SourcePolicy] | NotGiven = NOT_GIVEN,
-        task_spec: Optional[TaskSpecParam] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        source_policy: Optional[SourcePolicy] | Omit = omit,
+        task_spec: Optional[TaskSpecParam] | 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,
     ) -> TaskRun:
         """
         Initiates a task run.
@@ -421,7 +421,7 @@ class AsyncTaskRunResource(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,
     ) -> TaskRun:
         """
         Retrieves run status by run_id.
@@ -451,13 +451,13 @@ class AsyncTaskRunResource(AsyncAPIResource):
         self,
         run_id: str,
         *,
-        api_timeout: int | NotGiven = NOT_GIVEN,
+        api_timeout: int | 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,
     ) -> TaskRunResult:
         """
         Retrieves a run result by run_id, blocking until the run is completed.
@@ -492,7 +492,7 @@ class AsyncTaskRunResource(AsyncAPIResource):
         *,
         run_id: str,
         deadline: float,
-        output: Optional[OutputSchema] | Type[OutputT] | NotGiven = NOT_GIVEN,
+        output: Optional[OutputSchema] | Type[OutputT] | 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,
@@ -520,12 +520,12 @@ class AsyncTaskRunResource(AsyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        output: Optional[OutputSchema] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        output: Optional[OutputSchema] | Omit = omit,
         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,
     ) -> TaskRunResult: ...
     @overload
     async def execute(
@@ -533,26 +533,26 @@ class AsyncTaskRunResource(AsyncAPIResource):
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
         output: Type[OutputT],
         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,
     ) -> ParsedTaskRunResult[OutputT]: ...
     async def execute(
         self,
         *,
         input: Union[str, Dict[str, object]],
         processor: str,
-        metadata: Optional[Dict[str, Union[str, float, bool]]] | NotGiven = NOT_GIVEN,
-        output: Optional[OutputSchema] | Type[OutputT] | NotGiven = NOT_GIVEN,
+        metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
+        output: Optional[OutputSchema] | Type[OutputT] | 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,
     ) -> TaskRunResult | ParsedTaskRunResult[OutputT]:
         """
         Convenience method to create and execute a task run in a single call.