added await to tests, and async functions to audio

Author: winzamark123

jigsawstack/audio.py

@@ -79,15 +79,9 @@ def __init__(
     @overload
     def speech_to_text(self, params: SpeechToTextParams) -> SpeechToTextResponse: ...
     @overload
-    def speech_to_text(
-        self, file: bytes, options: Optional[SpeechToTextParams] = None
-    ) -> SpeechToTextResponse: ...
+    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:
+    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
@@ -104,14 +98,7 @@ def speech_to_text(
         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:
@@ -126,43 +113,23 @@ def text_to_speech(self, params: TextToSpeechParams) -> TextToSpeechResponse:
 
     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()
+        resp = Request(config=self.config, path=path, params=cast(Dict[Any, Any], params), verb="post").perform_with_content()
 
         return resp
 
     def get_clones(self, params: GetTTSVoiceClonesParams) -> TextToSpeechResponse:
         path = "/ai/tts/clone"
-        resp = Request(
-            config=self.config,
-            path=path,
-            params=cast(Dict[Any, Any], params),
-            verb="get",
-        ).perform_with_content()
+        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()
+        resp = Request(config=self.config, path=path, params={}, verb="delete").perform_with_content()
         return resp
 
 
@@ -183,9 +150,7 @@ def __init__(
         )
 
     @overload
-    async def speech_to_text(
-        self, params: SpeechToTextParams
-    ) -> SpeechToTextResponse: ...
+    async def speech_to_text(self, params: SpeechToTextParams) -> SpeechToTextResponse: ...
     @overload
     async def speech_to_text(
         self, file: bytes, options: Optional[SpeechToTextParams] = None
@@ -239,3 +204,33 @@ async def speaker_voice_accents(self) -> TextToSpeechResponse:
             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 get_clones(self, params: GetTTSVoiceClonesParams) -> 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

setup.py

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

tests/test_audio.py

@@ -18,7 +18,7 @@ async def _test():
 
         """Test converting text to speech"""
         try:
-            response = client.audio.text_to_speech(
+            response = await client.audio.text_to_speech(
                 {
                     "text": "Hello world, this is a test of the JigsawStack text to speech API."
                 }
@@ -33,16 +33,17 @@ async def _test():
 
 
 def test_speaker_voice_accents():
-    client = AsyncJigsawStack()
+    async def _test():
+        client = AsyncJigsawStack()
 
-    """Test getting available voice accents"""
-    try:
-        response = client.audio.speaker_voice_accents()
-        print("Speaker voice accents response:", response)
-        assert response["success"] == True
+        """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}")
+        except Exception as e:
+            print(f"Error in speaker voice accents test: {e}")
 
 
 def test_create_clone():
@@ -54,7 +55,7 @@ async def _test():
             audio_url = (
                 "https://jigsawstack.com/audio/test.mp3"  # Replace with an actual URL
             )
-            clone_response_url = client.audio.create_clone(
+            clone_response_url = await client.audio.create_clone(
                 {"url": audio_url, "name": "Test Voice Clone URL"}
             )
 
@@ -81,7 +82,7 @@ async def _test():
         """Test getting voice clones"""
         try:
             # List available voice clones
-            clones_response = client.audio.get_clones({"limit": 10, "page": 1})
+            clones_response = await client.audio.get_clones({"limit": 10, "page": 1})
 
             assert clones_response["success"] == True
 
@@ -96,13 +97,13 @@ async def _test():
         client = AsyncJigsawStack()
         """Test getting a voice clone"""
         try:
-            create_clone_response = client.audio.create_clone(
+            create_clone_response = await client.audio.create_clone(
                 {"name": "Test Voice Clone URL", "file_store_key": "hello_audio"}
             )
-            clones = client.audio.get_clones()
+            clones = await client.audio.get_clones({"limit": 10, "page": 1})
             print("Clones:", clones)
-            clone_id = clones[0]["id"]
-            delete_clone_response = client.audio.delete_clone(clone_id)
+            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