fix: update import statements and formatting across multiple files

Author: twangodev

examples/getting-started/01_simple_tts.py

@@ -19,6 +19,7 @@
 """
 
 import os
+
 from fishaudio import FishAudio
 from fishaudio.utils import save
 

examples/getting_started.ipynb

@@ -39,7 +39,18 @@
     }
    },
    "outputs": [],
-   "source": "from dotenv import load_dotenv\nfrom fishaudio import FishAudio\nfrom fishaudio.utils import play\n# from fishaudio.utils import save  # Uncomment if saving audio to file\n\nload_dotenv()\n\nclient = FishAudio()"
+   "source": [
+    "from dotenv import load_dotenv\n",
+    "\n",
+    "from fishaudio import FishAudio\n",
+    "from fishaudio.utils import play\n",
+    "\n",
+    "# from fishaudio.utils import save  # Uncomment if saving audio to file\n",
+    "\n",
+    "load_dotenv()\n",
+    "\n",
+    "client = FishAudio()"
+   ]
   },
   {
    "cell_type": "markdown",

scripts/copy_docs.py

@@ -10,6 +10,8 @@
     python scripts/copy_docs.py sdk docs  # In CI context
 """
 
+from __future__ import annotations
+
 import argparse
 import shutil
 from pathlib import Path

src/fish_audio_sdk/__init__.py

@@ -1,18 +1,18 @@
 from .apis import Session
 from .exceptions import HttpCodeErr, WebSocketErr
 from .schemas import (
+    APICreditEntity,
     ASRRequest,
-    TTSRequest,
-    ReferenceAudio,
-    Prosody,
-    PaginatedResponse,
+    CloseEvent,
     ModelEntity,
-    APICreditEntity,
+    PaginatedResponse,
+    Prosody,
+    ReferenceAudio,
     StartEvent,
     TextEvent,
-    CloseEvent,
+    TTSRequest,
 )
-from .websocket import WebSocketSession, AsyncWebSocketSession
+from .websocket import AsyncWebSocketSession, WebSocketSession
 
 __all__ = [
     "Session",

src/fish_audio_sdk/apis.py

@@ -1,4 +1,6 @@
-from typing import Generator, Literal
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Literal
 
 import ormsgpack
 
@@ -7,13 +9,16 @@
     APICreditEntity,
     ASRRequest,
     ASRResponse,
-    ModelEntity,
     Backends,
+    ModelEntity,
     PackageEntity,
     PaginatedResponse,
     TTSRequest,
 )
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 
 class Session(RemoteCall):
     @convert_stream

src/fish_audio_sdk/io.py

@@ -1,21 +1,20 @@
+from __future__ import annotations
+
 import dataclasses
 import typing
+from collections.abc import AsyncGenerator, Awaitable, Generator
 from http.client import responses as http_responses
 from typing import (
     Any,
-    AsyncGenerator,
-    Awaitable,
     Callable,
-    Generator,
     Generic,
     TypeVar,
 )
 
-from typing_extensions import Concatenate, ParamSpec
-
 import httpx
 import httpx._client
 import httpx._types
+from typing_extensions import Concatenate, ParamSpec
 
 from .exceptions import HttpCodeErr
 
@@ -194,8 +193,7 @@ def sync_wrapper(self: RemoteCall, *args: P.args, **kwargs: P.kwargs) -> R:
             return exc.value
         raise RuntimeError("Generator did not stop")
 
-    call = IOCallDescriptor(async_wrapper, sync_wrapper)
-    return call
+    return IOCallDescriptor(async_wrapper, sync_wrapper)
 
 
 GStream = G[Generator[bytes, bytes, None]]
@@ -257,5 +255,4 @@ def sync_wrapper(
 
         raise RuntimeError("Generator did not stop")
 
-    call = StreamIOCallDescriptor(async_wrapper, sync_wrapper)
-    return call
+    return StreamIOCallDescriptor(async_wrapper, sync_wrapper)

src/fish_audio_sdk/schemas.py

@@ -1,9 +1,12 @@
-import datetime
-import decimal
-from typing import Annotated, Generic, Literal, TypeVar
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Annotated, Generic, Literal, TypeVar
 
 from pydantic import BaseModel, Field
 
+if TYPE_CHECKING:
+    import datetime
+    import decimal
 
 Backends = Literal["speech-1.5", "speech-1.6", "agent-x0", "s1", "s1-mini"]
 

src/fish_audio_sdk/websocket.py

@@ -1,14 +1,13 @@
 import asyncio
+from collections.abc import AsyncGenerator, AsyncIterable, Generator, Iterable
 from concurrent.futures import ThreadPoolExecutor
-from typing import AsyncGenerator, AsyncIterable, Generator, Iterable
 
 import httpx
 import ormsgpack
-from httpx_ws import WebSocketDisconnect, connect_ws, aconnect_ws
+from httpx_ws import WebSocketDisconnect, aconnect_ws, connect_ws
 
 from .exceptions import WebSocketErr
-
-from .schemas import Backends, CloseEvent, StartEvent, TTSRequest, TextEvent
+from .schemas import Backends, CloseEvent, StartEvent, TextEvent, TTSRequest
 
 
 class WebSocketSession:

src/fishaudio/client.py

@@ -6,8 +6,8 @@
 
 from .core import AsyncClientWrapper, ClientWrapper
 from .resources import (
-    ASRClient,
     AccountClient,
+    ASRClient,
     AsyncAccountClient,
     AsyncASRClient,
     AsyncTTSClient,

src/fishaudio/core/client_wrapper.py

@@ -2,19 +2,20 @@
 
 import os
 from json import JSONDecodeError
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 import httpx
 
-from .._version import __version__
-from ..exceptions import (
+from fishaudio._version import __version__
+from fishaudio.exceptions import (
     APIError,
     AuthenticationError,
     NotFoundError,
     PermissionError,
     RateLimitError,
     ServerError,
 )
+
 from .request_options import RequestOptions
 
 
@@ -32,16 +33,15 @@ def _raise_for_status(response: httpx.Response) -> None:
     # Raise specific exception based on status code
     if status == 401:
         raise AuthenticationError(status, message, response.text)
-    elif status == 403:
+    if status == 403:
         raise PermissionError(status, message, response.text)
-    elif status == 404:
+    if status == 404:
         raise NotFoundError(status, message, response.text)
-    elif status == 429:
+    if status == 429:
         raise RateLimitError(status, message, response.text)
-    elif status >= 500:
+    if status >= 500:
         raise ServerError(status, message, response.text)
-    else:
-        raise APIError(status, message, response.text)
+    raise APIError(status, message, response.text)
 
 
 class BaseClientWrapper:
@@ -61,8 +61,8 @@ def __init__(
         self.base_url = base_url
 
     def get_headers(
-        self, additional_headers: Optional[Dict[str, str]] = None
-    ) -> Dict[str, str]:
+        self, additional_headers: Optional[dict[str, str]] = None
+    ) -> dict[str, str]:
         """Build headers including authentication and user agent."""
         headers = {
             "Authorization": f"Bearer {self.api_key}",
@@ -73,7 +73,7 @@ def get_headers(
         return headers
 
     def _prepare_request_kwargs(
-        self, request_options: Optional[RequestOptions], kwargs: Dict[str, Any]
+        self, request_options: Optional[RequestOptions], kwargs: dict[str, Any]
     ) -> None:
         """Prepare request kwargs by merging headers, timeout, and query params."""
         # Merge headers