Skip to content

Commit edcb1d0

Browse files
committed
feat: update websocket integration tests to check for deprecated models
1 parent f745f70 commit edcb1d0

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/integration/test_tts_websocket_integration.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from fishaudio import WebSocketOptions
88
from fishaudio.types import FlushEvent, Prosody, TextEvent, TTSConfig
9-
from fishaudio.types.shared import Model
9+
from fishaudio.types.shared import DEPRECATED_MODELS, Model
1010

1111
from .conftest import TEST_REFERENCE_ID
1212

@@ -44,7 +44,7 @@ def text_stream():
4444
reason="WebSocket unreliable for legacy models"
4545
),
4646
)
47-
if not m.startswith("s1")
47+
if m in DEPRECATED_MODELS
4848
else m
4949
for m in get_args(Model)
5050
],
@@ -233,7 +233,7 @@ async def text_stream():
233233
reason="WebSocket unreliable for legacy models"
234234
),
235235
)
236-
if not m.startswith("s1")
236+
if m in DEPRECATED_MODELS
237237
else m
238238
for m in get_args(Model)
239239
],

0 commit comments

Comments
 (0)