Skip to content

Commit 903bbbd

Browse files
use union
1 parent 847d4b2 commit 903bbbd

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

sentry_sdk/integrations/openai.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ def _calculate_token_usage(
195195

196196
def _get_input_messages(
197197
kwargs: "dict[str, Any]",
198-
) -> "Optional[Iterable[Any] | list[str]]":
198+
) -> "Optional[Union[Iterable[Any], list[str]]]":
199199
# Input messages (the prompt or data sent to the model)
200200
messages = kwargs.get("messages")
201201
if messages is None:
@@ -243,7 +243,9 @@ def _set_responses_api_input_data(
243243
kwargs: "dict[str, Any]",
244244
integration: "OpenAIIntegration",
245245
):
246-
messages: "Optional[ResponseInputParam | list[str]]" = _get_input_messages(kwargs) # type: ignore
246+
messages: "Optional[Union[ResponseInputParam, list[str]]]" = _get_input_messages(
247+
kwargs
248+
) # type: ignore
247249

248250
if (
249251
messages is not None
@@ -268,7 +270,7 @@ def _set_completions_api_input_data(
268270
kwargs: "dict[str, Any]",
269271
integration: "OpenAIIntegration",
270272
):
271-
messages: "Optional[Iterable[ChatCompletionMessageParam] | list[str]]" = (
273+
messages: "Optional[Union[Iterable[ChatCompletionMessageParam], list[str]]]" = (
272274
_get_input_messages(kwargs)
273275
)
274276

0 commit comments

Comments
 (0)