diff --git a/src/openai/lib/_chat_completion_resources.py b/src/openai/lib/_chat_completion_resources.py index 038448c25f..8ceaa42a11 100644 --- a/src/openai/lib/_chat_completion_resources.py +++ b/src/openai/lib/_chat_completion_resources.py @@ -140,7 +140,7 @@ def parser(raw_completion: ChatCompletion) -> ParsedChatCompletion[ResponseForma "verbosity": verbosity, "web_search_options": web_search_options, }, - completion_create_params.CompletionCreateParams, + completion_create_params.CompletionCreateParamsNonStreaming, ), options=make_request_options( extra_headers=extra_headers, @@ -259,7 +259,7 @@ def parser(raw_completion: ChatCompletion) -> ParsedChatCompletion[ResponseForma "verbosity": verbosity, "web_search_options": web_search_options, }, - completion_create_params.CompletionCreateParams, + completion_create_params.CompletionCreateParamsNonStreaming, ), options=make_request_options( extra_headers=extra_headers, diff --git a/src/openai/lib/_response_resources.py b/src/openai/lib/_response_resources.py index 76dff89ae6..f4ca471efa 100644 --- a/src/openai/lib/_response_resources.py +++ b/src/openai/lib/_response_resources.py @@ -136,7 +136,9 @@ def parser(raw_response: Response) -> ParsedResponse[TextFormatT]: "user": user, "verbosity": verbosity, }, - response_create_params.ResponseCreateParams, + response_create_params.ResponseCreateParamsStreaming + if stream + else response_create_params.ResponseCreateParamsNonStreaming, ), options=make_request_options( extra_headers=extra_headers, @@ -252,7 +254,9 @@ def parser(raw_response: Response) -> ParsedResponse[TextFormatT]: "user": user, "verbosity": verbosity, }, - response_create_params.ResponseCreateParams, + response_create_params.ResponseCreateParamsStreaming + if stream + else response_create_params.ResponseCreateParamsNonStreaming, ), options=make_request_options( extra_headers=extra_headers,