Commit 8086d617
Changed files (8)
src
openai
_utils
resources
audio
beta
threads
chat
completions
responses
src/openai/_utils/_transform.py
@@ -5,7 +5,7 @@ import base64
import pathlib
from typing import Any, Mapping, TypeVar, cast
from datetime import date, datetime
-from typing_extensions import Literal, get_args, override, get_type_hints
+from typing_extensions import Literal, get_args, override, get_type_hints as _get_type_hints
import anyio
import pydantic
@@ -13,6 +13,7 @@ import pydantic
from ._utils import (
is_list,
is_given,
+ lru_cache,
is_mapping,
is_iterable,
)
@@ -109,6 +110,7 @@ def transform(
return cast(_T, transformed)
+@lru_cache(maxsize=8096)
def _get_annotated_type(type_: type) -> type | None:
"""If the given type is an `Annotated` type then it is returned, if not `None` is returned.
@@ -433,3 +435,13 @@ async def _async_transform_typeddict(
else:
result[_maybe_transform_key(key, type_)] = await _async_transform_recursive(value, annotation=type_)
return result
+
+
+@lru_cache(maxsize=8096)
+def get_type_hints(
+ obj: Any,
+ globalns: dict[str, Any] | None = None,
+ localns: Mapping[str, Any] | None = None,
+ include_extras: bool = False,
+) -> dict[str, Any]:
+ return _get_type_hints(obj, globalns=globalns, localns=localns, include_extras=include_extras)
src/openai/_utils/_typing.py
@@ -13,6 +13,7 @@ from typing_extensions import (
get_origin,
)
+from ._utils import lru_cache
from .._types import InheritsGeneric
from .._compat import is_union as _is_union
@@ -66,6 +67,7 @@ def is_type_alias_type(tp: Any, /) -> TypeIs[typing_extensions.TypeAliasType]:
# Extracts T from Annotated[T, ...] or from Required[Annotated[T, ...]]
+@lru_cache(maxsize=8096)
def strip_annotated_type(typ: type) -> type:
if is_required_type(typ) or is_annotated_type(typ):
return strip_annotated_type(cast(type, get_args(typ)[0]))
src/openai/resources/audio/transcriptions.py
@@ -321,7 +321,12 @@ class Transcriptions(SyncAPIResource):
extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
return self._post( # type: ignore[return-value]
"/audio/transcriptions",
- body=maybe_transform(body, transcription_create_params.TranscriptionCreateParams),
+ body=maybe_transform(
+ body,
+ transcription_create_params.TranscriptionCreateParamsStreaming
+ if stream
+ else transcription_create_params.TranscriptionCreateParamsNonStreaming,
+ ),
files=files,
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -616,7 +621,12 @@ class AsyncTranscriptions(AsyncAPIResource):
extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
return await self._post(
"/audio/transcriptions",
- body=await async_maybe_transform(body, transcription_create_params.TranscriptionCreateParams),
+ body=await async_maybe_transform(
+ body,
+ transcription_create_params.TranscriptionCreateParamsStreaming
+ if stream
+ else transcription_create_params.TranscriptionCreateParamsNonStreaming,
+ ),
files=files,
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
src/openai/resources/beta/threads/runs/runs.py
@@ -587,7 +587,7 @@ class Runs(SyncAPIResource):
"top_p": top_p,
"truncation_strategy": truncation_strategy,
},
- run_create_params.RunCreateParams,
+ run_create_params.RunCreateParamsStreaming if stream else run_create_params.RunCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers,
@@ -1324,7 +1324,9 @@ class Runs(SyncAPIResource):
"tool_outputs": tool_outputs,
"stream": stream,
},
- run_submit_tool_outputs_params.RunSubmitToolOutputsParams,
+ run_submit_tool_outputs_params.RunSubmitToolOutputsParamsStreaming
+ if stream
+ else run_submit_tool_outputs_params.RunSubmitToolOutputsParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -1996,7 +1998,7 @@ class AsyncRuns(AsyncAPIResource):
"top_p": top_p,
"truncation_strategy": truncation_strategy,
},
- run_create_params.RunCreateParams,
+ run_create_params.RunCreateParamsStreaming if stream else run_create_params.RunCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers,
@@ -2732,7 +2734,9 @@ class AsyncRuns(AsyncAPIResource):
"tool_outputs": tool_outputs,
"stream": stream,
},
- run_submit_tool_outputs_params.RunSubmitToolOutputsParams,
+ run_submit_tool_outputs_params.RunSubmitToolOutputsParamsStreaming
+ if stream
+ else run_submit_tool_outputs_params.RunSubmitToolOutputsParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
src/openai/resources/beta/threads/threads.py
@@ -717,7 +717,9 @@ class Threads(SyncAPIResource):
"top_p": top_p,
"truncation_strategy": truncation_strategy,
},
- thread_create_and_run_params.ThreadCreateAndRunParams,
+ thread_create_and_run_params.ThreadCreateAndRunParamsStreaming
+ if stream
+ else thread_create_and_run_params.ThreadCreateAndRunParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -1564,7 +1566,9 @@ class AsyncThreads(AsyncAPIResource):
"top_p": top_p,
"truncation_strategy": truncation_strategy,
},
- thread_create_and_run_params.ThreadCreateAndRunParams,
+ thread_create_and_run_params.ThreadCreateAndRunParamsStreaming
+ if stream
+ else thread_create_and_run_params.ThreadCreateAndRunParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
src/openai/resources/chat/completions/completions.py
@@ -947,7 +947,9 @@ class Completions(SyncAPIResource):
"user": user,
"web_search_options": web_search_options,
},
- completion_create_params.CompletionCreateParams,
+ completion_create_params.CompletionCreateParamsStreaming
+ if stream
+ else completion_create_params.CompletionCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -2033,7 +2035,9 @@ class AsyncCompletions(AsyncAPIResource):
"user": user,
"web_search_options": web_search_options,
},
- completion_create_params.CompletionCreateParams,
+ completion_create_params.CompletionCreateParamsStreaming
+ if stream
+ else completion_create_params.CompletionCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
src/openai/resources/responses/responses.py
@@ -623,7 +623,9 @@ class Responses(SyncAPIResource):
"truncation": truncation,
"user": user,
},
- response_create_params.ResponseCreateParams,
+ response_create_params.ResponseCreateParamsStreaming
+ if stream
+ else response_create_params.ResponseCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -1435,7 +1437,9 @@ class AsyncResponses(AsyncAPIResource):
"truncation": truncation,
"user": user,
},
- response_create_params.ResponseCreateParams,
+ response_create_params.ResponseCreateParamsStreaming
+ if stream
+ else response_create_params.ResponseCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
src/openai/resources/completions.py
@@ -559,7 +559,9 @@ class Completions(SyncAPIResource):
"top_p": top_p,
"user": user,
},
- completion_create_params.CompletionCreateParams,
+ completion_create_params.CompletionCreateParamsStreaming
+ if stream
+ else completion_create_params.CompletionCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -1101,7 +1103,9 @@ class AsyncCompletions(AsyncAPIResource):
"top_p": top_p,
"user": user,
},
- completion_create_params.CompletionCreateParams,
+ completion_create_params.CompletionCreateParamsStreaming
+ if stream
+ else completion_create_params.CompletionCreateParamsNonStreaming,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout