99from zhipuai .types .audio import AudioSpeechParams
1010from ...types .audio import audio_customization_param
1111
12- from zhipuai .core import BaseAPI , maybe_transform , StreamResponse
12+ from zhipuai .core import BaseAPI , maybe_transform
1313from zhipuai .core import NOT_GIVEN , Body , Headers , NotGiven , FileTypes
1414from zhipuai .core import _legacy_response
1515
2020 make_request_options ,
2121)
2222from zhipuai .core import deepcopy_minimal
23- from ...types .audio .audio_speech_chunk import AudioSpeechChunk
2423
2524if TYPE_CHECKING :
2625 from zhipuai ._client import ZhipuAI
@@ -47,17 +46,15 @@ def speech(
4746 sensitive_word_check : Optional [SensitiveWordCheckRequest ] | NotGiven = NOT_GIVEN ,
4847 request_id : str = None ,
4948 user_id : str = None ,
50- stream : bool = False ,
5149 extra_headers : Headers | None = None ,
5250 extra_body : Body | None = None ,
5351 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
54- ) -> _legacy_response .HttpxBinaryResponseContent | StreamResponse [ AudioSpeechChunk ] :
52+ ) -> _legacy_response .HttpxBinaryResponseContent :
5553 body = deepcopy_minimal (
5654 {
5755 "model" : model ,
5856 "input" : input ,
5957 "voice" : voice ,
60- "stream" : stream ,
6158 "response_format" : response_format ,
6259 "sensitive_word_check" : sensitive_word_check ,
6360 "request_id" : request_id ,
@@ -66,13 +63,11 @@ def speech(
6663 )
6764 return self ._post (
6865 "/audio/speech" ,
69- body = body ,
66+ body = maybe_transform ( body , AudioSpeechParams ) ,
7067 options = make_request_options (
7168 extra_headers = extra_headers , extra_body = extra_body , timeout = timeout
7269 ),
73- cast_type = _legacy_response .HttpxBinaryResponseContent ,
74- stream = stream or False ,
75- stream_cls = StreamResponse [AudioSpeechChunk ]
70+ cast_type = _legacy_response .HttpxBinaryResponseContent
7671 )
7772
7873 def customization (
0 commit comments