improve code

Author: yuhongxiao

src/zai/api_resource/audio/audio.py

@@ -3,7 +3,6 @@
 from typing import TYPE_CHECKING, Mapping, Optional, cast
 
 import httpx
-from httpx import stream
 
 from zai.core import (
 	NOT_GIVEN,
@@ -23,8 +22,8 @@
 from zai.types.sensitive_word_check import SensitiveWordCheckRequest
 
 from .transcriptions import Transcriptions
-from ...core import StreamResponse
-from ...types.audio import AudioSpeechChunk
+from zai.core._streaming import StreamResponse
+from zai.types.audio import AudioSpeechChunk
 
 if TYPE_CHECKING:
 	from zai._client import ZaiClient
@@ -85,7 +84,6 @@ def speech(
 				'voice': voice,
 				'response_format': response_format,
 				'encode_format': encode_format,
-				'sensitive_word_check': sensitive_word_check,
 				'request_id': request_id,
 				'user_id': user_id,
 				'speed': speed,
@@ -98,6 +96,8 @@ def speech(
 			body=maybe_transform(body, AudioSpeechParams),
 			options=make_request_options(extra_headers=extra_headers, extra_body=extra_body, timeout=timeout),
 			cast_type=HttpxBinaryResponseContent,
+			stream=stream or False,
+			stream_cls=StreamResponse[AudioSpeechChunk]
 		)
 
 	def customization(

src/zai/types/audio/audio_speech_params.py

@@ -29,3 +29,7 @@ class AudioSpeechParams(TypedDict, total=False):
 	sensitive_word_check: Optional[SensitiveWordCheckRequest]
 	request_id: str
 	user_id: str
+	encode_format: str
+	speed: float
+	volume: float
+	stream: bool

tests/integration_tests/test_audio.py

@@ -6,12 +6,12 @@
 import zai
 from zai import ZaiClient
 
-from src.zai import ZhipuAiClient
+from zai import ZhipuAiClient
 
 
 def test_audio_speech(logging_conf):
 	logging.config.dictConfig(logging_conf)  # type: ignore
-	client = ZhipuAiClient(base_url='https://open.bigmodel.cn/api/paas/v4', api_key='adf953faf621426da79103110eb41473.3FCVJZcTaq0Q7i3W')  # Fill in your own API Key
+	client = ZhipuAiClient()  # Fill in your own API Key
 	try:
 		speech_file_path = Path(__file__).parent / 'asr1.pcm'
 		response = client.audio.speech(
@@ -20,7 +20,7 @@ def test_audio_speech(logging_conf):
 			voice='female',
 			response_format='pcm',
 			encode_format='base64',
-			stream=False,
+			stream=True,
 			speed=1.0,
 			volume=1.0,
 		)
@@ -29,9 +29,9 @@ def test_audio_speech(logging_conf):
 				choice = item.choices[0]
 				index = choice.index
 				finish_reason = choice.finish_reason
-				audio_delta = choice.delta.content
-				if finish_reason is not None:
+				if choice.delta is None:
 					break
+				audio_delta = choice.delta.content
 				f.write(base64.b64decode(audio_delta))
 
 	except zai.core._errors.APIRequestFailedError as err: