Merge pull request #66 from voicevox-client/tuna2134-patch-1

tuna2134 · web-flow · commit a832a68f009a · 2024-04-30T19:06:40.000+09:00
Add initialize function
diff --git a/vvclient/client.py b/vvclient/client.py
@@ -71,3 +71,25 @@ async def fetch_core_versions(self) -> List[str]:
             VOICEVOX Core versions
         """
         return await self.http.core_versions()
+
+    async def init_speaker(self, speaker: int, *, skip_reinit: bool = False, core_version: Optional[str] = None) -> None:
+        """
+        Initialize speaker
+
+        Parameters
+        ----------
+        speaker: intn
+            Speaker ID
+        skip_reinit: bool
+            Skip initialized speaker
+        core_version: Optional[str]
+            VOICEVOX Core version
+        """
+        params = {
+            "speaker": speaker
+        }
+        if skip_reinit:
+            params["skip_reinit"] = "true"
+        if core_version:
+            params["core_version"] = core_version
+        await self.http.initialize_speaker(params)
diff --git a/vvclient/http.py b/vvclient/http.py
@@ -31,6 +31,8 @@ async def request(self, route: Route, **kwargs) -> Any:
                 return await response.json()
             else:
                 return await response.read()
+        elif response.status == 204:
+            return await response.read()
         elif response.status == 404:
             raise NotFoundError("Not found")
         else:
@@ -56,3 +58,6 @@ async def engine_version(self) -> str:
 
     async def core_versions(self) -> List[str]:
         return await self.request(Route("GET", "/core_versions"))
+
+    async def initialize_speaker(self, params: Dict[str, Union[str, int]]) -> None:
+        return await self.request(Route("POST", "/initialize_speaker"), params=params)
