feat(playht): add Play3.0-mini engine support (livekit#1140)

Ganjiali · Nov 27, 2024 · 7b601b6 · 7b601b6
1 parent 415b07a
commit 7b601b6
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/.changeset/warm-suns-fly.md b/.changeset/warm-suns-fly.md
@@ -0,0 +1,5 @@
+---
+"livekit-plugins-playht": patch
+---
+
+feat(playht): add Play3.0-mini engine support
diff --git a/livekit-plugins/livekit-plugins-playht/livekit/plugins/playht/models.py b/livekit-plugins/livekit-plugins-playht/livekit/plugins/playht/models.py
@@ -1,6 +1,11 @@
 from typing import Literal
 
-TTSEngines = Literal["PlayHT2.0", "PlayHT1.0", "PlayHT2.0-turbo"]
+TTSEngines = Literal[
+    "PlayHT2.0",
+    "PlayHT1.0",
+    "PlayHT2.0-turbo",
+    "Play3.0-mini",
+]
 
 TTSEncoding = Literal[
     "mp3_22050_32",

diff --git a/livekit-plugins/livekit-plugins-playht/livekit/plugins/playht/tts.py b/livekit-plugins/livekit-plugins-playht/livekit/plugins/playht/tts.py
@@ -32,6 +32,8 @@ def _encoding_from_format(output_format: TTSEncoding) -> _Encoding:
         return "mp3"
     elif output_format.startswith("pcm"):
         return "pcm"
+    elif output_format.startswith("wav"):
+        return "pcm"
 
     raise ValueError(f"Unknown format: {output_format}")
 
@@ -46,7 +48,7 @@ class Voice:
 DEFAULT_VOICE = Voice(
     id="s3://peregrine-voices/mel22/manifest.json",
     name="Will",
-    voice_engine="PlayHT2.0",
+    voice_engine="Play3.0-mini",
 )
 
 ACCEPT_HEADER = {