jigsawstack 0.2.2__tar.gz → 0.2.4__tar.gz

{jigsawstack-0.2.2 → jigsawstack-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jigsawstack
-Version: 0.2.2
+Version: 0.2.4
 Summary: JigsawStack - The AI SDK for Python
 Home-page: https://github.com/jigsawstack/jigsawstack-python
 Author: JigsawStack

{jigsawstack-0.2.2 → jigsawstack-0.2.4}/jigsawstack/audio.py

@@ -16,6 +16,17 @@ class TextToSpeechParams(TypedDict):
     speaker_clone_file_store_key: NotRequired[str]
 
 
+class TTSCloneParams(TypedDict):
+    url: NotRequired[str]
+    file_store_key: NotRequired[str]
+    name: str
+
+
+class ListTTSVoiceClonesParams(TypedDict):
+    limit: NotRequired[int]
+    page: NotRequired[int]
+
+
 class TextToSpeechResponse(TypedDict):
     success: bool
     text: str
@@ -70,12 +81,10 @@ class Audio(ClientConfig):
     @overload
     def speech_to_text(self, file: bytes, options: Optional[SpeechToTextParams] = None) -> SpeechToTextResponse: ...
 
-    def speech_to_text(
-        self,
-        blob: Union[SpeechToTextParams, bytes],
-        options: Optional[SpeechToTextParams] = None,
-    ) -> SpeechToTextResponse:
-        if isinstance(blob, dict): # If params is provided as a dict, we assume it's the first argument
+    def speech_to_text(self, blob: Union[SpeechToTextParams, bytes], options: Optional[SpeechToTextParams] = None) -> SpeechToTextResponse:
+        if isinstance(
+            blob, dict
+        ):  # If params is provided as a dict, we assume it's the first argument
             resp = Request(
                 config=self.config,
                 path="/ai/transcribe",
@@ -89,17 +98,9 @@ class Audio(ClientConfig):
         content_type = options.get("content_type", "application/octet-stream")
         headers = {"Content-Type": content_type}
 
-        resp = Request(
-            config=self.config,
-            path=path,
-            params=options,
-            data=blob,
-            headers=headers,
-            verb="post",
-        ).perform_with_content()
+        resp = Request(config=self.config, path=path, params=options, data=blob, headers=headers, verb="post").perform_with_content()
         return resp
 
-
     def text_to_speech(self, params: TextToSpeechParams) -> TextToSpeechResponse:
         path = "/ai/tts"
         resp = Request(
@@ -112,12 +113,23 @@ class Audio(ClientConfig):
 
     def speaker_voice_accents(self) -> TextToSpeechResponse:
         path = "/ai/tts"
-        resp = Request(
-            config=self.config,
-            path=path,
-            params={},
-            verb="get",
-        ).perform_with_content()
+        resp = Request(config=self.config, path=path, params={}, verb="get").perform_with_content()
+        return resp
+
+    def create_clone(self, params: TTSCloneParams) -> TextToSpeechResponse:
+        path = "/ai/tts/clone"
+        resp = Request(config=self.config, path=path, params=cast(Dict[Any, Any], params), verb="post").perform_with_content()
+
+        return resp
+
+    def list_clones(self, params: ListTTSVoiceClonesParams) -> TextToSpeechResponse:
+        path = "/ai/tts/clone"
+        resp = Request(config=self.config, path=path, params=cast(Dict[Any, Any], params), verb="get").perform_with_content()
+        return resp
+
+    def delete_clone(self, voice_id: str) -> TextToSpeechResponse:
+        path = f"/ai/tts/clone/{voice_id}"
+        resp = Request(config=self.config, path=path, params={}, verb="delete").perform_with_content()
         return resp
 
 
@@ -140,7 +152,9 @@ class AsyncAudio(ClientConfig):
     @overload
     async def speech_to_text(self, params: SpeechToTextParams) -> SpeechToTextResponse: ...
     @overload
-    async def speech_to_text(self, file: bytes, options: Optional[SpeechToTextParams] = None) -> SpeechToTextResponse: ...
+    async def speech_to_text(
+        self, file: bytes, options: Optional[SpeechToTextParams] = None
+    ) -> SpeechToTextResponse: ...
 
     async def speech_to_text(
         self,
@@ -155,7 +169,7 @@ class AsyncAudio(ClientConfig):
                 verb="post",
             ).perform_with_content()
             return resp
-        
+
         options = options or {}
         path = build_path(base_path="/ai/transcribe", params=options)
         content_type = options.get("content_type", "application/octet-stream")
@@ -190,3 +204,33 @@ class AsyncAudio(ClientConfig):
             verb="get",
         ).perform_with_content()
         return resp
+
+    async def create_clone(self, params: TTSCloneParams) -> TextToSpeechResponse:
+        path = "/ai/tts/clone"
+        resp = await AsyncRequest(
+            config=self.config, 
+            path=path, 
+            params=cast(Dict[Any, Any], params), 
+            verb="post"
+        ).perform_with_content()
+        return resp
+
+    async def list_clones(self, params: ListTTSVoiceClonesParams) -> TextToSpeechResponse:
+        path = "/ai/tts/clone"
+        resp = await AsyncRequest(
+            config=self.config, 
+            path=path, 
+            params=cast(Dict[Any, Any], params), 
+            verb="get"
+        ).perform_with_content()
+        return resp
+
+    async def delete_clone(self, voice_id: str) -> TextToSpeechResponse:
+        path = f"/ai/tts/clone/{voice_id}"
+        resp = await AsyncRequest(
+            config=self.config, 
+            path=path, 
+            params={}, 
+            verb="delete"
+        ).perform_with_content()
+        return resp

{jigsawstack-0.2.2 → jigsawstack-0.2.4}/jigsawstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jigsawstack
-Version: 0.2.2
+Version: 0.2.4
 Summary: JigsawStack - The AI SDK for Python
 Home-page: https://github.com/jigsawstack/jigsawstack-python
 Author: JigsawStack

{jigsawstack-0.2.2 → jigsawstack-0.2.4}/setup.py

@@ -6,7 +6,7 @@ install_requires = open("requirements.txt").readlines()
 
 setup(
     name="jigsawstack",
-    version="0.2.2",
+    version="0.2.4",
     description="JigsawStack - The AI SDK for Python",
     long_description=open("README.md", encoding="utf8").read(),
     long_description_content_type="text/markdown",

jigsawstack-0.2.4/tests/test_audio.py

@@ -0,0 +1,112 @@
+from unittest.mock import MagicMock
+import unittest
+from jigsawstack.exceptions import JigsawStackError
+from jigsawstack import AsyncJigsawStack
+import pytest
+import asyncio
+import logging
+from jigsawstack import AsyncJigsawStack
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+def test_text_to_speech():
+    async def _test():
+        client = AsyncJigsawStack()
+
+        """Test converting text to speech"""
+        try:
+            response = await client.audio.text_to_speech(
+                {
+                    "text": "Hello world, this is a test of the JigsawStack text to speech API."
+                }
+            )
+            print("Text to speech response:", response)
+            assert response["success"] == True
+
+        except Exception as e:
+            print(f"Error in text_to_speech test: {e}")
+
+    asyncio.run(_test())
+
+
+def test_speaker_voice_accents():
+    async def _test():
+        client = AsyncJigsawStack()
+
+        """Test getting available voice accents"""
+        try:
+            response = await client.audio.speaker_voice_accents()
+            print("Speaker voice accents response:", response)
+            assert response["success"] == True
+
+        except Exception as e:
+            print(f"Error in speaker voice accents test: {e}")
+
+
+def test_create_clone():
+    async def _test():
+        client = AsyncJigsawStack()
+
+        """Test creating a voice clone with URL"""
+        try:
+            audio_url = (
+                "https://jigsawstack.com/audio/test.mp3"  # Replace with an actual URL
+            )
+            clone_response_url = await client.audio.create_clone(
+                {"url": audio_url, "name": "Test Voice Clone URL"}
+            )
+
+            assert clone_response_url["success"] == True
+
+            clone_response_file_store_key = client.audio.create_clone(
+                {
+                    "file_store_key": "hello_audio",
+                    "name": "Test Voice Clone File Store Key",
+                }
+            )
+
+            assert clone_response_file_store_key["success"] == True
+
+        except Exception as e:
+            print(f"Error in voice_cloning test: {e}")
+
+    asyncio.run(_test())
+
+
+def test_list_clones():
+    async def _test():
+        client = AsyncJigsawStack()
+        """Test listing voice clones"""
+        try:
+            # List available voice clones
+            clones_response = await client.audio.list_clones({"limit": 10, "page": 1})
+
+            assert clones_response["success"] == True
+
+        except Exception as e:
+            print(f"Error in voice_cloning test: {e}")
+
+    asyncio.run(_test())
+
+
+def test_delete_clone():
+    async def _test():
+        client = AsyncJigsawStack()
+        """Test getting a voice clone"""
+        try:
+            create_clone_response = await client.audio.create_clone(
+                {"name": "Test Voice Clone URL", "file_store_key": "hello_audio"}
+            )
+            clones = await client.audio.list_clones({"limit": 10, "page": 1})
+            print("Clones:", clones)
+            clone_id = clones["data"][0]["id"]
+            delete_clone_response = await client.audio.delete_clone(clone_id)
+            print("Delete clone response:", delete_clone_response)
+            assert delete_clone_response["success"] == True
+
+        except Exception as e:
+            print(f"Error in list_clones test: {e}")
+
+    asyncio.run(_test())

jigsawstack-0.2.2/tests/test_audio.py

@@ -1,22 +0,0 @@
-from unittest.mock import MagicMock
-import unittest
-from jigsawstack.exceptions import JigsawStackError
-from jigsawstack import AsyncJigsawStack
-import pytest
-import asyncio
-import logging
-
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-
-def test_async_speaker_voice_accents_response():
-    async def _test():
-        client = AsyncJigsawStack()
-        try:
-            result = await client.audio.speaker_voice_accents()
-            assert result["success"] == True
-        except JigsawStackError as e:
-            pytest.fail(f"Unexpected JigsawStackError: {e}")
-
-    asyncio.run(_test())