relaxai 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

relaxai/_utils/_transform.py

@@ -16,18 +16,20 @@ from ._utils import (
     lru_cache,
     is_mapping,
     is_iterable,
+    is_sequence,
 )
 from .._files import is_base64_file_input
+from ._compat import get_origin, is_typeddict
 from ._typing import (
     is_list_type,
     is_union_type,
     extract_type_arg,
     is_iterable_type,
     is_required_type,
+    is_sequence_type,
     is_annotated_type,
     strip_annotated_type,
 )
-from .._compat import get_origin, model_dump, is_typeddict
 
 _T = TypeVar("_T")
 
@@ -167,6 +169,8 @@ def _transform_recursive(
 
             Defaults to the same value as the `annotation` argument.
     """
+    from .._compat import model_dump
+
     if inner_type is None:
         inner_type = annotation
 
@@ -184,6 +188,8 @@ def _transform_recursive(
         (is_list_type(stripped_type) and is_list(data))
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
+        # Sequence[T]
+        or (is_sequence_type(stripped_type) and is_sequence(data) and not isinstance(data, str))
     ):
         # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
         # intended as an iterable, so we don't transform it.
@@ -262,7 +268,7 @@ def _transform_typeddict(
     annotations = get_type_hints(expected_type, include_extras=True)
     for key, value in data.items():
         if not is_given(value):
-            # we don't need to include `NotGiven` values here as they'll
+            # we don't need to include omitted values here as they'll
             # be stripped out before the request is sent anyway
             continue
 
@@ -329,6 +335,8 @@ async def _async_transform_recursive(
 
             Defaults to the same value as the `annotation` argument.
     """
+    from .._compat import model_dump
+
     if inner_type is None:
         inner_type = annotation
 
@@ -346,6 +354,8 @@ async def _async_transform_recursive(
         (is_list_type(stripped_type) and is_list(data))
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
+        # Sequence[T]
+        or (is_sequence_type(stripped_type) and is_sequence(data) and not isinstance(data, str))
     ):
         # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
         # intended as an iterable, so we don't transform it.
@@ -424,7 +434,7 @@ async def _async_transform_typeddict(
     annotations = get_type_hints(expected_type, include_extras=True)
     for key, value in data.items():
         if not is_given(value):
-            # we don't need to include `NotGiven` values here as they'll
+            # we don't need to include omitted values here as they'll
             # be stripped out before the request is sent anyway
             continue
 

relaxai/_utils/_typing.py

@@ -15,7 +15,7 @@ from typing_extensions import (
 
 from ._utils import lru_cache
 from .._types import InheritsGeneric
-from .._compat import is_union as _is_union
+from ._compat import is_union as _is_union
 
 
 def is_annotated_type(typ: type) -> bool:
@@ -26,6 +26,11 @@ def is_list_type(typ: type) -> bool:
     return (get_origin(typ) or typ) == list
 
 
+def is_sequence_type(typ: type) -> bool:
+    origin = get_origin(typ) or typ
+    return origin == typing_extensions.Sequence or origin == typing.Sequence or origin == _c_abc.Sequence
+
+
 def is_iterable_type(typ: type) -> bool:
     """If the given type is `typing.Iterable[T]`"""
     origin = get_origin(typ) or typ

relaxai/_utils/_utils.py

@@ -21,8 +21,7 @@ from typing_extensions import TypeGuard
 
 import sniffio
 
-from .._types import NotGiven, FileTypes, NotGivenOr, HeadersLike
-from .._compat import parse_date as parse_date, parse_datetime as parse_datetime
+from .._types import Omit, NotGiven, FileTypes, HeadersLike
 
 _T = TypeVar("_T")
 _TupleT = TypeVar("_TupleT", bound=Tuple[object, ...])
@@ -64,7 +63,7 @@ def _extract_items(
     try:
         key = path[index]
     except IndexError:
-        if isinstance(obj, NotGiven):
+        if not is_given(obj):
             # no value was provided - we can safely ignore
             return []
 
@@ -127,8 +126,8 @@ def _extract_items(
     return []
 
 
-def is_given(obj: NotGivenOr[_T]) -> TypeGuard[_T]:
-    return not isinstance(obj, NotGiven)
+def is_given(obj: _T | NotGiven | Omit) -> TypeGuard[_T]:
+    return not isinstance(obj, NotGiven) and not isinstance(obj, Omit)
 
 
 # Type safe methods for narrowing types with TypeVars.

relaxai/_version.py

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

relaxai/resources/__init__.py

@@ -8,6 +8,14 @@ from .chat import (
     ChatResourceWithStreamingResponse,
     AsyncChatResourceWithStreamingResponse,
 )
+from .tools import (
+    ToolsResource,
+    AsyncToolsResource,
+    ToolsResourceWithRawResponse,
+    AsyncToolsResourceWithRawResponse,
+    ToolsResourceWithStreamingResponse,
+    AsyncToolsResourceWithStreamingResponse,
+)
 from .models import (
     ModelsResource,
     AsyncModelsResource,
@@ -24,6 +32,14 @@ from .embeddings import (
     EmbeddingsResourceWithStreamingResponse,
     AsyncEmbeddingsResourceWithStreamingResponse,
 )
+from .deep_research import (
+    DeepResearchResource,
+    AsyncDeepResearchResource,
+    DeepResearchResourceWithRawResponse,
+    AsyncDeepResearchResourceWithRawResponse,
+    DeepResearchResourceWithStreamingResponse,
+    AsyncDeepResearchResourceWithStreamingResponse,
+)
 
 __all__ = [
     "ChatResource",
@@ -44,4 +60,16 @@ __all__ = [
     "AsyncModelsResourceWithRawResponse",
     "ModelsResourceWithStreamingResponse",
     "AsyncModelsResourceWithStreamingResponse",
+    "ToolsResource",
+    "AsyncToolsResource",
+    "ToolsResourceWithRawResponse",
+    "AsyncToolsResourceWithRawResponse",
+    "ToolsResourceWithStreamingResponse",
+    "AsyncToolsResourceWithStreamingResponse",
+    "DeepResearchResource",
+    "AsyncDeepResearchResource",
+    "DeepResearchResourceWithRawResponse",
+    "AsyncDeepResearchResourceWithRawResponse",
+    "DeepResearchResourceWithStreamingResponse",
+    "AsyncDeepResearchResourceWithStreamingResponse",
 ]

relaxai/resources/chat.py

@@ -2,12 +2,12 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Iterable
+from typing import Dict, Iterable
 
 import httpx
 
 from ..types import chat_create_completion_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -51,40 +51,40 @@ class ChatResource(SyncAPIResource):
         *,
         messages: Iterable[ChatCompletionMessageParam],
         model: str,
-        query_stream: bool | NotGiven = NOT_GIVEN,
-        chat_template_kwargs: chat_create_completion_params.ChatTemplateKwargs | NotGiven = NOT_GIVEN,
-        frequency_penalty: float | NotGiven = NOT_GIVEN,
-        function_call: chat_create_completion_params.FunctionCall | NotGiven = NOT_GIVEN,
-        functions: Iterable[FunctionDefinitionParam] | NotGiven = NOT_GIVEN,
-        logit_bias: Dict[str, int] | NotGiven = NOT_GIVEN,
-        logprobs: bool | NotGiven = NOT_GIVEN,
-        max_completion_tokens: int | NotGiven = NOT_GIVEN,
-        max_tokens: int | NotGiven = NOT_GIVEN,
-        metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        n: int | NotGiven = NOT_GIVEN,
-        parallel_tool_calls: chat_create_completion_params.ParallelToolCalls | NotGiven = NOT_GIVEN,
-        prediction: chat_create_completion_params.Prediction | NotGiven = NOT_GIVEN,
-        presence_penalty: float | NotGiven = NOT_GIVEN,
-        reasoning_effort: str | NotGiven = NOT_GIVEN,
-        response_format: chat_create_completion_params.ResponseFormat | NotGiven = NOT_GIVEN,
-        seed: int | NotGiven = NOT_GIVEN,
-        stop: List[str] | NotGiven = NOT_GIVEN,
-        store: bool | NotGiven = NOT_GIVEN,
-        body_stream: bool | NotGiven = NOT_GIVEN,
-        stream_options: StreamOptionsParam | NotGiven = NOT_GIVEN,
-        temperature: float | NotGiven = NOT_GIVEN,
-        tool_choice: chat_create_completion_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[chat_create_completion_params.Tool] | NotGiven = NOT_GIVEN,
-        top_logprobs: int | NotGiven = NOT_GIVEN,
-        top_p: float | NotGiven = NOT_GIVEN,
-        user: str | NotGiven = NOT_GIVEN,
-        web_search_options: chat_create_completion_params.WebSearchOptions | NotGiven = NOT_GIVEN,
+        query_stream: bool | Omit = omit,
+        chat_template_kwargs: chat_create_completion_params.ChatTemplateKwargs | Omit = omit,
+        frequency_penalty: float | Omit = omit,
+        function_call: chat_create_completion_params.FunctionCall | Omit = omit,
+        functions: Iterable[FunctionDefinitionParam] | Omit = omit,
+        logit_bias: Dict[str, int] | Omit = omit,
+        logprobs: bool | Omit = omit,
+        max_completion_tokens: int | Omit = omit,
+        max_tokens: int | Omit = omit,
+        metadata: Dict[str, str] | Omit = omit,
+        n: int | Omit = omit,
+        parallel_tool_calls: chat_create_completion_params.ParallelToolCalls | Omit = omit,
+        prediction: chat_create_completion_params.Prediction | Omit = omit,
+        presence_penalty: float | Omit = omit,
+        reasoning_effort: str | Omit = omit,
+        response_format: chat_create_completion_params.ResponseFormat | Omit = omit,
+        seed: int | Omit = omit,
+        stop: SequenceNotStr[str] | Omit = omit,
+        store: bool | Omit = omit,
+        body_stream: bool | Omit = omit,
+        stream_options: StreamOptionsParam | Omit = omit,
+        temperature: float | Omit = omit,
+        tool_choice: chat_create_completion_params.ToolChoice | Omit = omit,
+        tools: Iterable[chat_create_completion_params.Tool] | Omit = omit,
+        top_logprobs: int | Omit = omit,
+        top_p: float | Omit = omit,
+        user: str | Omit = omit,
+        web_search_options: chat_create_completion_params.WebSearchOptions | 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,
     ) -> ChatCompletionResponse:
         """
         Creates a chat completion for the given model
@@ -175,40 +175,40 @@ class AsyncChatResource(AsyncAPIResource):
         *,
         messages: Iterable[ChatCompletionMessageParam],
         model: str,
-        query_stream: bool | NotGiven = NOT_GIVEN,
-        chat_template_kwargs: chat_create_completion_params.ChatTemplateKwargs | NotGiven = NOT_GIVEN,
-        frequency_penalty: float | NotGiven = NOT_GIVEN,
-        function_call: chat_create_completion_params.FunctionCall | NotGiven = NOT_GIVEN,
-        functions: Iterable[FunctionDefinitionParam] | NotGiven = NOT_GIVEN,
-        logit_bias: Dict[str, int] | NotGiven = NOT_GIVEN,
-        logprobs: bool | NotGiven = NOT_GIVEN,
-        max_completion_tokens: int | NotGiven = NOT_GIVEN,
-        max_tokens: int | NotGiven = NOT_GIVEN,
-        metadata: Dict[str, str] | NotGiven = NOT_GIVEN,
-        n: int | NotGiven = NOT_GIVEN,
-        parallel_tool_calls: chat_create_completion_params.ParallelToolCalls | NotGiven = NOT_GIVEN,
-        prediction: chat_create_completion_params.Prediction | NotGiven = NOT_GIVEN,
-        presence_penalty: float | NotGiven = NOT_GIVEN,
-        reasoning_effort: str | NotGiven = NOT_GIVEN,
-        response_format: chat_create_completion_params.ResponseFormat | NotGiven = NOT_GIVEN,
-        seed: int | NotGiven = NOT_GIVEN,
-        stop: List[str] | NotGiven = NOT_GIVEN,
-        store: bool | NotGiven = NOT_GIVEN,
-        body_stream: bool | NotGiven = NOT_GIVEN,
-        stream_options: StreamOptionsParam | NotGiven = NOT_GIVEN,
-        temperature: float | NotGiven = NOT_GIVEN,
-        tool_choice: chat_create_completion_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[chat_create_completion_params.Tool] | NotGiven = NOT_GIVEN,
-        top_logprobs: int | NotGiven = NOT_GIVEN,
-        top_p: float | NotGiven = NOT_GIVEN,
-        user: str | NotGiven = NOT_GIVEN,
-        web_search_options: chat_create_completion_params.WebSearchOptions | NotGiven = NOT_GIVEN,
+        query_stream: bool | Omit = omit,
+        chat_template_kwargs: chat_create_completion_params.ChatTemplateKwargs | Omit = omit,
+        frequency_penalty: float | Omit = omit,
+        function_call: chat_create_completion_params.FunctionCall | Omit = omit,
+        functions: Iterable[FunctionDefinitionParam] | Omit = omit,
+        logit_bias: Dict[str, int] | Omit = omit,
+        logprobs: bool | Omit = omit,
+        max_completion_tokens: int | Omit = omit,
+        max_tokens: int | Omit = omit,
+        metadata: Dict[str, str] | Omit = omit,
+        n: int | Omit = omit,
+        parallel_tool_calls: chat_create_completion_params.ParallelToolCalls | Omit = omit,
+        prediction: chat_create_completion_params.Prediction | Omit = omit,
+        presence_penalty: float | Omit = omit,
+        reasoning_effort: str | Omit = omit,
+        response_format: chat_create_completion_params.ResponseFormat | Omit = omit,
+        seed: int | Omit = omit,
+        stop: SequenceNotStr[str] | Omit = omit,
+        store: bool | Omit = omit,
+        body_stream: bool | Omit = omit,
+        stream_options: StreamOptionsParam | Omit = omit,
+        temperature: float | Omit = omit,
+        tool_choice: chat_create_completion_params.ToolChoice | Omit = omit,
+        tools: Iterable[chat_create_completion_params.Tool] | Omit = omit,
+        top_logprobs: int | Omit = omit,
+        top_p: float | Omit = omit,
+        user: str | Omit = omit,
+        web_search_options: chat_create_completion_params.WebSearchOptions | 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,
     ) -> ChatCompletionResponse:
         """
         Creates a chat completion for the given model

relaxai/resources/deep_research.py

@@ -0,0 +1,165 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..types import deep_research_create_params
+from .._types import Body, Query, Headers, NoneType, NotGiven, not_given
+from .._utils import maybe_transform, async_maybe_transform
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from .._base_client import make_request_options
+from ..types.deepresearch_request_param import DeepresearchRequestParam
+
+__all__ = ["DeepResearchResource", "AsyncDeepResearchResource"]
+
+
+class DeepResearchResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> DeepResearchResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/relax-ai/python-sdk#accessing-raw-response-data-eg-headers
+        """
+        return DeepResearchResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> DeepResearchResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/relax-ai/python-sdk#with_streaming_response
+        """
+        return DeepResearchResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        body: DeepresearchRequestParam,
+        # 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,
+    ) -> None:
+        """
+        Performs deep research on a given topic and returns a detailed report.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return self._post(
+            "/v1/deep-research",
+            body=maybe_transform({"body": body}, deep_research_create_params.DeepResearchCreateParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class AsyncDeepResearchResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncDeepResearchResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/relax-ai/python-sdk#accessing-raw-response-data-eg-headers
+        """
+        return AsyncDeepResearchResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncDeepResearchResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/relax-ai/python-sdk#with_streaming_response
+        """
+        return AsyncDeepResearchResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        body: DeepresearchRequestParam,
+        # 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,
+    ) -> None:
+        """
+        Performs deep research on a given topic and returns a detailed report.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return await self._post(
+            "/v1/deep-research",
+            body=await async_maybe_transform({"body": body}, deep_research_create_params.DeepResearchCreateParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class DeepResearchResourceWithRawResponse:
+    def __init__(self, deep_research: DeepResearchResource) -> None:
+        self._deep_research = deep_research
+
+        self.create = to_raw_response_wrapper(
+            deep_research.create,
+        )
+
+
+class AsyncDeepResearchResourceWithRawResponse:
+    def __init__(self, deep_research: AsyncDeepResearchResource) -> None:
+        self._deep_research = deep_research
+
+        self.create = async_to_raw_response_wrapper(
+            deep_research.create,
+        )
+
+
+class DeepResearchResourceWithStreamingResponse:
+    def __init__(self, deep_research: DeepResearchResource) -> None:
+        self._deep_research = deep_research
+
+        self.create = to_streamed_response_wrapper(
+            deep_research.create,
+        )
+
+
+class AsyncDeepResearchResourceWithStreamingResponse:
+    def __init__(self, deep_research: AsyncDeepResearchResource) -> None:
+        self._deep_research = deep_research
+
+        self.create = async_to_streamed_response_wrapper(
+            deep_research.create,
+        )

relaxai/resources/embeddings.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import httpx
 
 from ..types import embedding_create_embedding_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
@@ -46,15 +46,15 @@ class EmbeddingsResource(SyncAPIResource):
         *,
         input: object,
         model: str,
-        dimensions: int | NotGiven = NOT_GIVEN,
-        encoding_format: str | NotGiven = NOT_GIVEN,
-        user: str | NotGiven = NOT_GIVEN,
+        dimensions: int | Omit = omit,
+        encoding_format: str | Omit = omit,
+        user: 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,
     ) -> EmbeddingResponse:
         """
         Creates an embedding vector representing the input text.
@@ -112,15 +112,15 @@ class AsyncEmbeddingsResource(AsyncAPIResource):
         *,
         input: object,
         model: str,
-        dimensions: int | NotGiven = NOT_GIVEN,
-        encoding_format: str | NotGiven = NOT_GIVEN,
-        user: str | NotGiven = NOT_GIVEN,
+        dimensions: int | Omit = omit,
+        encoding_format: str | Omit = omit,
+        user: 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,
     ) -> EmbeddingResponse:
         """
         Creates an embedding vector representing the input text.

relaxai/resources/models.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 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 .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -48,7 +48,7 @@ class ModelsResource(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,
     ) -> ModelList:
         """List all the available models"""
         return self._get(
@@ -68,7 +68,7 @@ class ModelsResource(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,
     ) -> Model:
         """
         Get the details of the given model
@@ -121,7 +121,7 @@ class AsyncModelsResource(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,
     ) -> ModelList:
         """List all the available models"""
         return await self._get(
@@ -141,7 +141,7 @@ class AsyncModelsResource(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,
     ) -> Model:
         """
         Get the details of the given model