refactor: rename _get_headers to get_headers and update usage across the codebase

Author: twangodev

src/fishaudio/core/client_wrapper.py

@@ -60,10 +60,10 @@ def __init__(
             )
         self.base_url = base_url
 
-    def _get_headers(
+    def get_headers(
         self, additional_headers: Optional[Dict[str, str]] = None
     ) -> Dict[str, str]:
-        """Build headers including authentication."""
+        """Build headers including authentication and user agent."""
         headers = {
             "Authorization": f"Bearer {self.api_key}",
             "User-Agent": f"fish-audio/python/{__version__}",
@@ -77,7 +77,7 @@ def _prepare_request_kwargs(
     ) -> None:
         """Prepare request kwargs by merging headers, timeout, and query params."""
         # Merge headers
-        headers = self._get_headers()
+        headers = self.get_headers()
         if request_options and request_options.additional_headers:
             headers.update(request_options.additional_headers)
         kwargs["headers"] = {**headers, **kwargs.get("headers", {})}
@@ -113,7 +113,7 @@ def __init__(
             self._client = httpx.Client(
                 base_url=base_url,
                 timeout=httpx.Timeout(timeout),
-                headers=self._get_headers(),
+                headers=self.get_headers(),
             )
 
     def request(
@@ -185,7 +185,7 @@ def __init__(
             self._client = httpx.AsyncClient(
                 base_url=base_url,
                 timeout=httpx.Timeout(timeout),
-                headers=self._get_headers(),
+                headers=self.get_headers(),
             )
 
     async def request(

src/fishaudio/resources/tts.py

@@ -329,10 +329,7 @@ def text_generator():
             with connect_ws(
                 "/v1/tts/live",
                 client=self._client.client,
-                headers={
-                    "model": model,
-                    "Authorization": f"Bearer {self._client.api_key}",
-                },
+                headers=self._client.get_headers({"model": model}),
                 **ws_kwargs,
             ) as ws:
 
@@ -630,7 +627,7 @@ async def text_generator():
         async with aconnect_ws(
             "/v1/tts/live",
             client=self._client.client,
-            headers={"model": model, "Authorization": f"Bearer {self._client.api_key}"},
+            headers=self._client.get_headers({"model": model}),
             **ws_kwargs,
         ) as ws:
 

tests/unit/test_core.py

@@ -111,13 +111,13 @@ def test_init_with_env_var(self, mock_api_key):
 
     def test_get_headers(self, mock_api_key):
         wrapper = ClientWrapper(api_key=mock_api_key)
-        headers = wrapper._get_headers()
+        headers = wrapper.get_headers()
         assert headers["Authorization"] == f"Bearer {mock_api_key}"
         assert "User-Agent" in headers
 
     def test_get_headers_with_additional(self, mock_api_key):
         wrapper = ClientWrapper(api_key=mock_api_key)
-        headers = wrapper._get_headers({"X-Custom": "value"})
+        headers = wrapper.get_headers({"X-Custom": "value"})
         assert headers["X-Custom"] == "value"
         assert headers["Authorization"] == f"Bearer {mock_api_key}"
 
@@ -139,6 +139,6 @@ def test_init_without_api_key_raises(self):
 
     def test_get_headers(self, mock_api_key):
         wrapper = AsyncClientWrapper(api_key=mock_api_key)
-        headers = wrapper._get_headers()
+        headers = wrapper.get_headers()
         assert headers["Authorization"] == f"Bearer {mock_api_key}"
         assert "User-Agent" in headers

tests/unit/test_tts_realtime.py

@@ -16,6 +16,12 @@ def mock_client_wrapper(mock_api_key):
     wrapper.api_key = mock_api_key
     # Mock the underlying httpx.Client
     wrapper._client = Mock()
+    # Mock get_headers to return a dict with the additional headers merged
+    wrapper.get_headers = lambda additional=None: {
+        "Authorization": f"Bearer {mock_api_key}",
+        "User-Agent": "fish-audio/python/test",
+        **(additional or {}),
+    }
     return wrapper
 
 
@@ -26,6 +32,12 @@ def async_mock_client_wrapper(mock_api_key):
     wrapper.api_key = mock_api_key
     # Mock the underlying httpx.AsyncClient
     wrapper._client = Mock()
+    # Mock get_headers to return a dict with the additional headers merged
+    wrapper.get_headers = lambda additional=None: {
+        "Authorization": f"Bearer {mock_api_key}",
+        "User-Agent": "fish-audio/python/test",
+        **(additional or {}),
+    }
     return wrapper