From da7d0cdbbe723a7a536098b79ea9c41e0ad01ad9 Mon Sep 17 00:00:00 2001 From: Kilerd Chan Date: Wed, 2 Apr 2025 15:38:02 +0800 Subject: [PATCH 1/2] feat: add model 1.6 in supported backend --- src/fish_audio_sdk/schemas.py | 2 +- tests/test_apis.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/src/fish_audio_sdk/schemas.py b/src/fish_audio_sdk/schemas.py index 0bdbb75..9dae6fe 100644 --- a/src/fish_audio_sdk/schemas.py +++ b/src/fish_audio_sdk/schemas.py @@ -5,7 +5,7 @@ from pydantic import BaseModel, Field -Backends = Literal["speech-1.5", "agent-x0"] +Backends = Literal["speech-1.5", "speech-1.6", "agent-x0"] Item = TypeVar("Item") diff --git a/tests/test_apis.py b/tests/test_apis.py index fc6be26..760d7c8 100644 --- a/tests/test_apis.py +++ b/tests/test_apis.py @@ -10,6 +10,12 @@ def test_tts(session: Session): buffer.extend(chunk) assert len(buffer) > 0 +def test_tts_model_1_6(session: Session): + buffer = bytearray() + for chunk in session.tts(TTSRequest(text="Hello, world!"), backend="speech-1.6"): + buffer.extend(chunk) + assert len(buffer) > 0 + async def test_tts_async(session: Session): buffer = bytearray() From 475ef327ccf7b91d9eadc042fea638c304e07767 Mon Sep 17 00:00:00 2001 From: Kilerd Chan Date: Wed, 2 Apr 2025 15:40:50 +0800 Subject: [PATCH 2/2] feat: add top_p and temperature parameters to TTSRequest schema --- src/fish_audio_sdk/schemas.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/fish_audio_sdk/schemas.py b/src/fish_audio_sdk/schemas.py index 9dae6fe..b7e0c7b 100644 --- a/src/fish_audio_sdk/schemas.py +++ b/src/fish_audio_sdk/schemas.py @@ -37,6 +37,8 @@ class TTSRequest(BaseModel): normalize: bool = True latency: Literal["normal", "balanced"] = "balanced" prosody: Prosody | None = None + top_p: float = 0.7 + temperature: float = 0.7 class ASRRequest(BaseModel):