File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -195,7 +195,7 @@ def _calculate_token_usage(
195195
196196def _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
You can’t perform that action at this time.
0 commit comments