add stream reponse to audio.speech

Author: yuhongxiao

src/zai/api_resource/audio/audio.py

@@ -3,6 +3,7 @@
 from typing import TYPE_CHECKING, Mapping, Optional, cast
 
 import httpx
+from httpx import stream
 
 from zai.core import (
 	NOT_GIVEN,
@@ -58,6 +59,7 @@ def speech(
 		encode_format: str = None,
 		speed: float | None = 1.0,
 		volume: float | None = 1.0,
+		stream: bool | None = False
 	) -> HttpxBinaryResponseContent:
 		"""
 		Generate speech audio from text input
@@ -86,6 +88,7 @@ def speech(
 				'user_id': user_id,
 				'speed': speed,
 				'volume': volume,
+				'stream': stream
 			}
 		)
 		return self._post(

tests/integration_tests/test_audio.py

@@ -17,6 +17,7 @@ def test_audio_speech(logging_conf):
 			voice='female',
 			response_format='pcm',
 			encode_format='base64',
+			stream=False,
 			speed=1.0,
 			volume=1.0,
 		)