chore(types): change optional parameter type from NotGiven to Omit

Author: stainless-app[bot]

src/replicate/_client.py

@@ -29,7 +29,6 @@
 from ._qs import Querystring
 from .types import client_search_params
 from ._types import (
-    NOT_GIVEN,
     Body,
     Omit,
     Query,
@@ -39,6 +38,8 @@
     Transport,
     ProxiesTypes,
     RequestOptions,
+    omit,
+    not_given,
 )
 from ._utils import (
     is_given,
@@ -103,7 +104,7 @@ def __init__(
         *,
         bearer_token: str | None = None,
         base_url: str | httpx.URL | None = None,
-        timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
+        timeout: float | Timeout | None | NotGiven = not_given,
         max_retries: int = DEFAULT_MAX_RETRIES,
         default_headers: Mapping[str, str] | None = None,
         default_query: Mapping[str, object] | None = None,
@@ -241,7 +242,7 @@ def run(
         *,
         file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
         use_file_output: bool = True,
-        wait: Union[int, bool, NotGiven] = NOT_GIVEN,
+        wait: Union[int, bool, NotGiven] = not_given,
         **params: Unpack[PredictionCreateParamsWithoutVersion],
     ) -> Any:
         """
@@ -325,9 +326,9 @@ def copy(
         *,
         bearer_token: str | None = None,
         base_url: str | httpx.URL | None = None,
-        timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | Timeout | None | NotGiven = not_given,
         http_client: httpx.Client | None = None,
-        max_retries: int | NotGiven = NOT_GIVEN,
+        max_retries: int | NotGiven = not_given,
         default_headers: Mapping[str, str] | None = None,
         set_default_headers: Mapping[str, str] | None = None,
         default_query: Mapping[str, object] | None = None,
@@ -375,13 +376,13 @@ def search(
         self,
         *,
         query: str,
-        limit: int | NotGiven = NOT_GIVEN,
+        limit: 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,
     ) -> SearchResponse:
         """
         Search for public models, collections, and docs using a text query.
@@ -478,7 +479,7 @@ def __init__(
         *,
         bearer_token: str | None = None,
         base_url: str | httpx.URL | None = None,
-        timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
+        timeout: float | Timeout | None | NotGiven = not_given,
         max_retries: int = DEFAULT_MAX_RETRIES,
         default_headers: Mapping[str, str] | None = None,
         default_query: Mapping[str, object] | None = None,
@@ -616,7 +617,7 @@ async def run(
         *,
         use_file_output: bool = True,
         file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
-        wait: Union[int, bool, NotGiven] = NOT_GIVEN,
+        wait: Union[int, bool, NotGiven] = not_given,
         **params: Unpack[PredictionCreateParamsWithoutVersion],
     ) -> Any:
         """
@@ -700,9 +701,9 @@ def copy(
         *,
         bearer_token: str | None = None,
         base_url: str | httpx.URL | None = None,
-        timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | Timeout | None | NotGiven = not_given,
         http_client: httpx.AsyncClient | None = None,
-        max_retries: int | NotGiven = NOT_GIVEN,
+        max_retries: int | NotGiven = not_given,
         default_headers: Mapping[str, str] | None = None,
         set_default_headers: Mapping[str, str] | None = None,
         default_query: Mapping[str, object] | None = None,
@@ -750,13 +751,13 @@ async def search(
         self,
         *,
         query: str,
-        limit: int | NotGiven = NOT_GIVEN,
+        limit: 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,
     ) -> SearchResponse:
         """
         Search for public models, collections, and docs using a text query.

src/replicate/resources/predictions.py

@@ -11,7 +11,7 @@
 from replicate.lib._files import FileEncodingStrategy, encode_json, async_encode_json
 
 from ..types import prediction_list_params, prediction_create_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, strip_not_given, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -63,17 +63,17 @@ def create(
         *,
         input: object,
         version: str,
-        stream: bool | NotGiven = NOT_GIVEN,
-        webhook: str | NotGiven = NOT_GIVEN,
-        webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | NotGiven = NOT_GIVEN,
-        prefer: str | NotGiven = NOT_GIVEN,
+        stream: bool | Omit = omit,
+        webhook: str | Omit = omit,
+        webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
+        prefer: str | Omit = omit,
         file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
         # 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,
     ) -> Prediction:
         """
         Create a prediction for the model version and inputs you provide.
@@ -208,14 +208,14 @@ def create(
     def list(
         self,
         *,
-        created_after: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        created_before: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        created_after: Union[str, datetime] | Omit = omit,
+        created_before: Union[str, datetime] | 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,
     ) -> SyncCursorURLPageWithCreatedFilters[Prediction]:
         """
         Get a paginated list of all predictions created by the user or organization
@@ -330,7 +330,7 @@ def cancel(
         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,
     ) -> Prediction:
         """
         Cancel a prediction that is currently running.
@@ -386,7 +386,7 @@ def get(
         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,
     ) -> Prediction:
         """
         Get the current state of a prediction.
@@ -514,17 +514,17 @@ async def create(
         *,
         input: object,
         version: str,
-        stream: bool | NotGiven = NOT_GIVEN,
-        webhook: str | NotGiven = NOT_GIVEN,
-        webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | NotGiven = NOT_GIVEN,
-        prefer: str | NotGiven = NOT_GIVEN,
+        stream: bool | Omit = omit,
+        webhook: str | Omit = omit,
+        webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
+        prefer: str | Omit = omit,
         file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
         # 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,
     ) -> Prediction:
         """
         Create a prediction for the model version and inputs you provide.
@@ -661,14 +661,14 @@ async def create(
     def list(
         self,
         *,
-        created_after: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        created_before: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        created_after: Union[str, datetime] | Omit = omit,
+        created_before: Union[str, datetime] | 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[Prediction, AsyncCursorURLPageWithCreatedFilters[Prediction]]:
         """
         Get a paginated list of all predictions created by the user or organization
@@ -783,7 +783,7 @@ async def cancel(
         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,
     ) -> Prediction:
         """
         Cancel a prediction that is currently running.
@@ -839,7 +839,7 @@ async def get(
         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,
     ) -> Prediction:
         """
         Get the current state of a prediction.