|
29 | 29 | from ._qs import Querystring |
30 | 30 | from .types import client_search_params |
31 | 31 | from ._types import ( |
32 | | - NOT_GIVEN, |
33 | 32 | Body, |
34 | 33 | Omit, |
35 | 34 | Query, |
|
39 | 38 | Transport, |
40 | 39 | ProxiesTypes, |
41 | 40 | RequestOptions, |
| 41 | + omit, |
| 42 | + not_given, |
42 | 43 | ) |
43 | 44 | from ._utils import ( |
44 | 45 | is_given, |
@@ -103,7 +104,7 @@ def __init__( |
103 | 104 | *, |
104 | 105 | bearer_token: str | None = None, |
105 | 106 | base_url: str | httpx.URL | None = None, |
106 | | - timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN, |
| 107 | + timeout: float | Timeout | None | NotGiven = not_given, |
107 | 108 | max_retries: int = DEFAULT_MAX_RETRIES, |
108 | 109 | default_headers: Mapping[str, str] | None = None, |
109 | 110 | default_query: Mapping[str, object] | None = None, |
@@ -241,7 +242,7 @@ def run( |
241 | 242 | *, |
242 | 243 | file_encoding_strategy: Optional["FileEncodingStrategy"] = None, |
243 | 244 | use_file_output: bool = True, |
244 | | - wait: Union[int, bool, NotGiven] = NOT_GIVEN, |
| 245 | + wait: Union[int, bool, NotGiven] = not_given, |
245 | 246 | **params: Unpack[PredictionCreateParamsWithoutVersion], |
246 | 247 | ) -> Any: |
247 | 248 | """ |
@@ -325,9 +326,9 @@ def copy( |
325 | 326 | *, |
326 | 327 | bearer_token: str | None = None, |
327 | 328 | base_url: str | httpx.URL | None = None, |
328 | | - timeout: float | Timeout | None | NotGiven = NOT_GIVEN, |
| 329 | + timeout: float | Timeout | None | NotGiven = not_given, |
329 | 330 | http_client: httpx.Client | None = None, |
330 | | - max_retries: int | NotGiven = NOT_GIVEN, |
| 331 | + max_retries: int | NotGiven = not_given, |
331 | 332 | default_headers: Mapping[str, str] | None = None, |
332 | 333 | set_default_headers: Mapping[str, str] | None = None, |
333 | 334 | default_query: Mapping[str, object] | None = None, |
@@ -375,13 +376,13 @@ def search( |
375 | 376 | self, |
376 | 377 | *, |
377 | 378 | query: str, |
378 | | - limit: int | NotGiven = NOT_GIVEN, |
| 379 | + limit: int | Omit = omit, |
379 | 380 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
380 | 381 | # The extra values given here take precedence over values defined on the client or passed to this method. |
381 | 382 | extra_headers: Headers | None = None, |
382 | 383 | extra_query: Query | None = None, |
383 | 384 | extra_body: Body | None = None, |
384 | | - timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN, |
| 385 | + timeout: float | httpx.Timeout | None | NotGiven = not_given, |
385 | 386 | ) -> SearchResponse: |
386 | 387 | """ |
387 | 388 | Search for public models, collections, and docs using a text query. |
@@ -478,7 +479,7 @@ def __init__( |
478 | 479 | *, |
479 | 480 | bearer_token: str | None = None, |
480 | 481 | base_url: str | httpx.URL | None = None, |
481 | | - timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN, |
| 482 | + timeout: float | Timeout | None | NotGiven = not_given, |
482 | 483 | max_retries: int = DEFAULT_MAX_RETRIES, |
483 | 484 | default_headers: Mapping[str, str] | None = None, |
484 | 485 | default_query: Mapping[str, object] | None = None, |
@@ -616,7 +617,7 @@ async def run( |
616 | 617 | *, |
617 | 618 | use_file_output: bool = True, |
618 | 619 | file_encoding_strategy: Optional["FileEncodingStrategy"] = None, |
619 | | - wait: Union[int, bool, NotGiven] = NOT_GIVEN, |
| 620 | + wait: Union[int, bool, NotGiven] = not_given, |
620 | 621 | **params: Unpack[PredictionCreateParamsWithoutVersion], |
621 | 622 | ) -> Any: |
622 | 623 | """ |
@@ -700,9 +701,9 @@ def copy( |
700 | 701 | *, |
701 | 702 | bearer_token: str | None = None, |
702 | 703 | base_url: str | httpx.URL | None = None, |
703 | | - timeout: float | Timeout | None | NotGiven = NOT_GIVEN, |
| 704 | + timeout: float | Timeout | None | NotGiven = not_given, |
704 | 705 | http_client: httpx.AsyncClient | None = None, |
705 | | - max_retries: int | NotGiven = NOT_GIVEN, |
| 706 | + max_retries: int | NotGiven = not_given, |
706 | 707 | default_headers: Mapping[str, str] | None = None, |
707 | 708 | set_default_headers: Mapping[str, str] | None = None, |
708 | 709 | default_query: Mapping[str, object] | None = None, |
@@ -750,13 +751,13 @@ async def search( |
750 | 751 | self, |
751 | 752 | *, |
752 | 753 | query: str, |
753 | | - limit: int | NotGiven = NOT_GIVEN, |
| 754 | + limit: int | Omit = omit, |
754 | 755 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
755 | 756 | # The extra values given here take precedence over values defined on the client or passed to this method. |
756 | 757 | extra_headers: Headers | None = None, |
757 | 758 | extra_query: Query | None = None, |
758 | 759 | extra_body: Body | None = None, |
759 | | - timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN, |
| 760 | + timeout: float | httpx.Timeout | None | NotGiven = not_given, |
760 | 761 | ) -> SearchResponse: |
761 | 762 | """ |
762 | 763 | Search for public models, collections, and docs using a text query. |
|
0 commit comments