smallestai 3.0.2__py3-none-any.whl → 3.1.0__py3-none-any.whl

smallestai/waves/async_waves_client.py

@@ -89,9 +89,9 @@ class AsyncWavesClient:
         return False
     
 
-    def get_languages(self) -> List[str]:
+    def get_languages(self, model="lightning") -> List[str]:
         """Returns a list of available languages."""
-        return get_smallest_languages()
+        return get_smallest_languages(model)
 
     def get_cloned_voices(self) -> str:
         """Returns a list of your cloned voices."""
@@ -176,7 +176,7 @@ class AsyncWavesClient:
             validate_input(text, opts.model, opts.sample_rate, opts.speed, opts.consistency, opts.similarity, opts.enhancement)
 
             self.chunk_size = 250
-            if opts.model == 'lightning-large':
+            if opts.model == 'lightning-large' or opts.model == "lightning-v2":
                 self.chunk_size = 140
 
             chunks = chunk_text(text, self.chunk_size)
@@ -192,7 +192,7 @@ class AsyncWavesClient:
                         "model": opts.model
                     }
                     
-                    if opts.model == "lightning-large":
+                    if opts.model == "lightning-large" or opts.model == "lightning-v2":
                         if opts.consistency is not None:
                             payload["consistency"] = opts.consistency
                         if opts.similarity is not None:

smallestai/waves/models.py

@@ -1,5 +1,8 @@
-TTSLanguages = ["en", "hi"]
+TTSLanguages_lightning = ["en", "hi"]
+TTSLanguages_lightning_large = ["en", "hi"]
+TTSLanguages_lightning_v2 = ["en", "hi", "mr", "kn", "ta", "bn", "gu", "de", "fr", "es", "it", "pl", "nl", "ru", "ar", "he"]
 TTSModels = [
     "lightning", 
-    "lightning-large"
-]
+    "lightning-large",
+    "lightning-v2"
+]

smallestai/waves/utils.py

@@ -6,7 +6,7 @@ from pydub import AudioSegment
 from dataclasses import dataclass
 
 from smallestai.waves.exceptions import ValidationError
-from smallestai.waves.models import TTSModels, TTSLanguages
+from smallestai.waves.models import TTSModels, TTSLanguages_lightning, TTSLanguages_lightning_large, TTSLanguages_lightning_v2
 
 
 API_BASE_URL = "https://waves-api.smallest.ai/api/v1"
@@ -90,8 +90,15 @@ def chunk_text(text: str, chunk_size: int = 250) -> List[str]:
     return chunks
 
 
-def get_smallest_languages() -> List[str]:
-    return TTSLanguages
+def get_smallest_languages(model: str = 'lightning') -> List[str]:
+    if model == 'lightning':
+        return TTSLanguages_lightning
+    elif model == 'lightning-large':
+        return TTSLanguages_lightning_large
+    elif model == 'lightning-v2':
+        return TTSLanguages_lightning_v2
+    else:
+        raise ValidationError(f"Invalid model: {model}. Must be one of {TTSModels}")
 
 def get_smallest_models() -> List[str]:
     return TTSModels

smallestai/waves/waves_client.py

@@ -66,9 +66,9 @@ class WavesClient:
         )
     
         
-    def get_languages(self) -> List[str]:
+    def get_languages(self, model:str="lightning") -> List[str]:
         """Returns a list of available languages."""
-        return get_smallest_languages()
+        return get_smallest_languages(model)
     
     def get_cloned_voices(self) -> str:
         """Returns a list of your cloned voices."""
@@ -144,7 +144,7 @@ class WavesClient:
         validate_input(text, opts.model, opts.sample_rate, opts.speed, opts.consistency, opts.similarity, opts.enhancement)
 
         self.chunk_size = 250
-        if opts.model == "lightning-large":
+        if opts.model == "lightning-large" or opts.model == "lightning-v2":
             self.chunk_size = 140
 
         chunks = chunk_text(text, self.chunk_size)
@@ -159,7 +159,7 @@ class WavesClient:
                     "speed": opts.speed,
                 }
 
-                if opts.model == "lightning-large":
+                if opts.model == "lightning-large" or opts.model == "lightning-v2":
                     if opts.consistency is not None:
                         payload["consistency"] = opts.consistency
                     if opts.similarity is not None:

{smallestai-3.0.2.dist-info → smallestai-3.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smallestai
-Version: 3.0.2
+Version: 3.1.0
 Summary: Official Python client for the Smallest AI API
 Author-email: Smallest <support@smallest.ai>
 License: MIT

{smallestai-3.0.2.dist-info → smallestai-3.1.0.dist-info}/RECORD

@@ -74,14 +74,14 @@ smallestai/atoms/models/update_agent_request_synthesizer_voice_config_one_of.py,
 smallestai/atoms/models/update_agent_request_synthesizer_voice_config_one_of1.py,sha256=9AJxgngoNSMvDbceajIqnG23PY4rw84coTh7yUTNS3c,3487
 smallestai/atoms/models/upload_text_to_knowledge_base_request.py,sha256=Sxg0vRv_naT15odE8fBUeyjwLpEYOmQwGcJuzRRr90A,2587
 smallestai/waves/__init__.py,sha256=Hkq7N2nuz_wS7pC6QeUnIU1MzQnX_nrhfXGpjGSvFhQ,244
-smallestai/waves/async_waves_client.py,sha256=hv9rQ8-ykWuHoAcmZPhwtX_-AAQT4H4G3H8c4BhO5-0,12658
+smallestai/waves/async_waves_client.py,sha256=sZrBnH_oi3wumL-w_2ccn1MZkLz7eYOyUFZEwjC-2vY,12746
 smallestai/waves/exceptions.py,sha256=nY6I8fCXe2By54CytQ0-i3hFiYtt8TYAKj0g6OYsCjc,585
-smallestai/waves/models.py,sha256=g2e_4nU5P48vyXZandKLWqZC1TkoEGeLvYKqJIqurSI,83
+smallestai/waves/models.py,sha256=FaMVkOFyNCVpWvyMCmqkv3t1wmnfCs1HIULxLr1L8XE,283
 smallestai/waves/stream_tts.py,sha256=Ppjwp1jXpUSpyNkwCnesMYQbAdyzKLMj_1o1iTb3jaA,10958
-smallestai/waves/utils.py,sha256=0VqMA4apJ-9U7abOznVXqUYEEAxQ2JkpLGyFhcJ_Kbw,3307
-smallestai/waves/waves_client.py,sha256=XKdPVWs-HZDzlxzF1x3cMdJQ_q71ZFS1P5oltzj2KO4,10740
-smallestai-3.0.2.dist-info/licenses/LICENSE,sha256=kK3HNKhN7luQhkjkNWIvy9_gizbEDUM4mSv_HWq9uuM,1068
-smallestai-3.0.2.dist-info/METADATA,sha256=w5DR17sxy9oTxYum6ktWRfo3hktGb6Dx2kP1w_k7bFE,20392
-smallestai-3.0.2.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-smallestai-3.0.2.dist-info/top_level.txt,sha256=pdJzm1VC2J6RxoobATz45L9U3cki4AFLigsfvETz7Io,11
-smallestai-3.0.2.dist-info/RECORD,,
+smallestai/waves/utils.py,sha256=UPThOnh1O5_6nZ9ZBJoY--iDOnFKBP9Nq3qGHhuDPbc,3692
+smallestai/waves/waves_client.py,sha256=KMUpWolrKqEP24DTRkNzHWpYuWxO3buVpX1cHn90gGY,10832
+smallestai-3.1.0.dist-info/licenses/LICENSE,sha256=kK3HNKhN7luQhkjkNWIvy9_gizbEDUM4mSv_HWq9uuM,1068
+smallestai-3.1.0.dist-info/METADATA,sha256=Gs-xjdLWK7ihF3JH98PpZd4afswx-6Qj7BT3IxUyHoA,20392
+smallestai-3.1.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+smallestai-3.1.0.dist-info/top_level.txt,sha256=pdJzm1VC2J6RxoobATz45L9U3cki4AFLigsfvETz7Io,11
+smallestai-3.1.0.dist-info/RECORD,,

{smallestai-3.0.2.dist-info → smallestai-3.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any