Add CustomVoiceResponse TypedDict for create_custom_voice

- Added `CustomVoiceResponse` TypedDict for `create_custom_voice` return type - Updated `create_custom_voice` to return `CustomVoiceResponse`
cr2007 · Jul 10, 2024 · 2d96696 · 2d96696
1 parent a642341
commit 2d96696
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cambai/__init__.py b/cambai/__init__.py
@@ -127,6 +127,11 @@ class TaskStatus(TypedDict):
  run_id: Optional[int]
 
 
+# ---------- Custom Voice Information ---------- #
+
+class CustomVoiceResponse(TypedDict):
+ voice_id: int
+
 # ---------- Transcription Result ---------- #
 
 
@@ -367,7 +372,7 @@ def get_languages(
 
  def create_custom_voice(
  self, /, voice_name: str, file: str, gender: Gender, *, age: Optional[int] = 30
- ) -> dict[str, str]:
+ ) -> CustomVoiceResponse:
  """
  Creates a custom voice profile using the provided voice sample.