cartesia 0.1.1__py2.py3-none-any.whl → 1.0.1__py2.py3-none-any.whl

cartesia/__init__.py

@@ -1,3 +1,3 @@
-from cartesia.tts import AsyncCartesiaTTS, CartesiaTTS
+from cartesia.client import Cartesia, AsyncCartesia
 
-__all__ = ["CartesiaTTS", "AsyncCartesiaTTS"]
+__all__ = ["Cartesia", "AsyncCartesia"]

cartesia/_types.py

@@ -1,43 +1,75 @@
-from enum import Enum
-from typing import List, Optional, TypedDict, Union
+from typing import List, TypedDict
+from cartesia.utils.deprecated import deprecated
 
-try:
-    import numpy as np
 
-    _NUMPY_AVAILABLE = True
-except ImportError:
-    _NUMPY_AVAILABLE = False
+class OutputFormatMapping:
+    _format_mapping = {
+        "raw_pcm_f32le_44100": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 44100},
+        "raw_pcm_s16le_44100": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 44100},
+        "raw_pcm_f32le_24000": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 24000},
+        "raw_pcm_s16le_24000": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 24000},
+        "raw_pcm_f32le_22050": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 22050},
+        "raw_pcm_s16le_22050": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 22050},
+        "raw_pcm_f32le_16000": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 16000},
+        "raw_pcm_s16le_16000": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 16000},
+        "raw_pcm_f32le_8000": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 8000},
+        "raw_pcm_s16le_8000": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 8000},
+        "raw_pcm_mulaw_8000": {"container": "raw", "encoding": "pcm_mulaw", "sample_rate": 8000},
+        "raw_pcm_alaw_8000": {"container": "raw", "encoding": "pcm_alaw", "sample_rate": 8000},
+    }
 
+    @classmethod
+    def get_format(cls, format_name):
+        if format_name in cls._format_mapping:
+            return cls._format_mapping[format_name]
+        else:
+            raise ValueError(f"Unsupported format: {format_name}")
 
-class AudioDataReturnType(Enum):
-    BYTES = "bytes"
-    ARRAY = "array"
 
+class DeprecatedOutputFormatMapping:
+    """Deprecated formats as of v1.0.1. These will be removed in v1.2.0. Use :class:`OutputFormatMapping` instead."""
 
-class AudioOutputFormat(Enum):
-    """Supported output formats for the audio."""
+    _format_mapping = {
+        "fp32": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 44100},
+        "pcm": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 44100},
+        "fp32_8000": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 8000},
+        "fp32_16000": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 16000},
+        "fp32_22050": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 22050},
+        "fp32_24000": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 24000},
+        "fp32_44100": {"container": "raw", "encoding": "pcm_f32le", "sample_rate": 44100},
+        "pcm_8000": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 8000},
+        "pcm_16000": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 16000},
+        "pcm_22050": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 22050},
+        "pcm_24000": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 24000},
+        "pcm_44100": {"container": "raw", "encoding": "pcm_s16le", "sample_rate": 44100},
+        "mulaw_8000": {"container": "raw", "encoding": "pcm_mulaw", "sample_rate": 8000},
+        "alaw_8000": {"container": "raw", "encoding": "pcm_alaw", "sample_rate": 8000},
+    }
 
-    FP32 = "fp32"  # float32
-    PCM = "pcm"  # 16-bit signed integer PCM
-    FP32_16000 = "fp32_16000"  # float32, 16 kHz
-    FP32_22050 = "fp32_22050"  # float32, 22.05 kHz
-    FP32_44100 = "fp32_44100"  # float32, 44.1 kHz
-    PCM_16000 = "pcm_16000"  # 16-bit signed integer PCM, 16 kHz
-    PCM_22050 = "pcm_22050"  # 16-bit signed integer PCM, 22.05 kHz
-    PCM_44100 = "pcm_44100"  # 16-bit signed integer PCM, 44.1 kHz
-    MULAW_8000 = "mulaw_8000"  # 8-bit mu-law, 8 kHz
-
-
-class AudioOutput(TypedDict):
-    audio: Union[bytes, "np.ndarray"]
-    sampling_rate: int
-
-
-Embedding = List[float]
+    @deprecated(
+        vdeprecated="1.0.1",
+        vremove="1.2.0",
+        reason="Old output format names are being deprecated in favor of names aligned with the Cartesia API. Use names from `OutputFormatMapping` instead.",
+    )
+    def get_format_deprecated(self, format_name):
+        if format_name in self._format_mapping:
+            return self._format_mapping[format_name]
+        else:
+            raise ValueError(f"Unsupported format: {format_name}")
 
 
 class VoiceMetadata(TypedDict):
     id: str
     name: str
     description: str
-    embedding: Optional[Embedding]
+    embedding: List[float]
+    is_public: bool
+    user_id: str
+    created_at: str
+    language: str
+
+
+class OutputFormat(TypedDict):
+    container: str
+    encoding: str
+    sample_rate: int