✨ feat: 添加代理配置选项以支持国内访问 fish-audio

Author: Cvandia

README.md

@@ -22,12 +22,16 @@ _⭐文本生成语音`tts`插件⭐_
 
 > 或者使用官方在线api -> [fish-audio](https://fish.audio/zh-CN/)即可享受快速云端的语音生成。
 
+> [!WARNING]
+> 由于目前国内无法正常访问fish-audio,需要配置[代理](#️-配置)或者使用离线fish-speech生成语音。
+
 ## 📜 免责声明
 
-> [!note]
+> [!CAUTION]
 > 本插件仅供**学习**和**研究**使用，使用者需自行承担使用插件的风险。作者不对插件的使用造成的任何损失或问题负责。请合理使用插件，**遵守相关法律法规。**
 使用**本插件即表示您已阅读并同意遵守以上免责声明**。如果您不同意或无法遵守以上声明，请不要使用本插件。
 
+---
 
 ## 💿 安装
 
@@ -78,7 +82,7 @@ git clone https://github.com/Cvandia/nonebot-plugin-fishspeech-tts
 |  tts_is_online   |  bool   |   是   |       True        |                       是否使用云端api                       |
 | tts_chunk_length | literal |   否   |     "normal"      | 请求时音频分片长度，默认为normal，可选：short, normal, long |
 |  tts_audio_path  |   str   |   否   | "./data/参考音频" |            语音素材路径，默认为"./data/参考音频"            |
-|    tts_prefix    |   str   |   否   |       None        |                  触发前缀，默认为None                   |
+|    tts_prefix    |   str   |   否   |       None        |                    触发前缀，默认为None                     |
 
 **注：参考音频的文件名格式为：［角色名］音频对应的文字标签.[音频后缀名]**
 
@@ -95,6 +99,7 @@ ___
 | :------------------: | :---: | :----: | :-----: | :--------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | online_authorization |  str  |   是   | "xxxxx" |                                                    官网api鉴权秘钥，详见[链接](https://fish.audio/zh-CN/go-api/api-keys/)                                                    |
 |  online_model_first  | bool  |   否   |  True   | 如果你想调用官方模型，通过自己的参考音频，定制角色音色，将此项设为`False`。当然，如果你没有准备参考音频，也会调用官网已经有的音色，具体详见[链接](https://fish.audio/zh-CN/) |
+|   online_api_proxy   |  str  |   否   |  None   |                                                                     代理地址，如：http://127.0.0.1:7890                                                                      |
 
 ---
 

nonebot_plugin_fishspeech_tts/config.py

@@ -17,6 +17,8 @@ class Config(BaseModel):
     # 区分配置
     online_authorization: Optional[str] = "xxxxx"
     online_model_first: bool = True
+    # 设置代理地址
+    online_api_proxy: Optional[str] = None
 
     offline_api_url: str = "http://127.0.0.1:8080"
 

nonebot_plugin_fishspeech_tts/fish_audio_api.py

@@ -29,6 +29,7 @@
 
 
 is_reference_id_first = config.online_model_first
+online_api_proxy = config.online_api_proxy
 
 
 class FishAudioAPI:
@@ -37,8 +38,9 @@ class FishAudioAPI:
     """
 
     def __init__(self):
-        self.url = "https://api.fish.audio/v1/tts"
+        self.url: str = "https://api.fish.audio/v1/tts"
         self.path_audio: Path = Path(config.tts_audio_path)
+        self.proxy = online_api_proxy
 
         # 如果在线授权码为空, 且使用在线api, 则抛出异常
         if not config.online_authorization and config.tts_is_online:
@@ -70,7 +72,7 @@ async def _get_reference_id_by_speaker(self, speaker: str) -> str:
         """
         request_api = "https://api.fish.audio/model"
         sort_options = ["score", "task_count", "created_at"]
-        async with AsyncClient() as client:
+        async with AsyncClient(proxies=self.proxy) as client:
             for sort_by in sort_options:
                 params = {"title": speaker, "sort_by": sort_by}
                 response = await client.get(
@@ -147,7 +149,7 @@ async def generate_tts(self, request: ServeTTSRequest) -> bytes:
         if request.references:
             self.headers["content-type"] = "application/msgpack"
             try:
-                async with AsyncClient() as client:
+                async with AsyncClient(proxies=self.proxy) as client:
                     async with client.stream(
                         "POST",
                         self.url,
@@ -166,11 +168,13 @@ async def generate_tts(self, request: ServeTTSRequest) -> bytes:
                 HTTPStatusError,
             ) as e:
                 logger.error(f"获取TTS音频失败: {e}")
+                if self.proxy:
+                    raise HTTPException("代理地址错误, 请检查代理地址是否正确")
                 raise HTTPException("网络错误, 请检查网络连接")
         else:
             self.headers["content-type"] = "application/json"
             try:
-                async with AsyncClient() as client:
+                async with AsyncClient(proxies=self.proxy) as client:
                     response = await client.post(
                         self.url,
                         headers=self.headers,
@@ -186,14 +190,16 @@ async def generate_tts(self, request: ServeTTSRequest) -> bytes:
                 HTTPStatusError,
             ) as e:
                 logger.error(f"获取TTS音频失败: {e}")
+                if self.proxy:
+                    raise HTTPException("代理地址错误, 请检查代理地址是否正确")
                 raise HTTPException("网络错误, 请检查网络连接")
 
     async def get_balance(self) -> float:
         """
         获取账户余额
         """
         balance_url = "https://api.fish.audio/wallet/self/api-credit"
-        async with AsyncClient() as client:
+        async with AsyncClient(proxies=self.proxy) as client:
             response = await client.get(balance_url, headers=self.headers)
             try:
                 return response.json()["credit"]

pyproject.toml

@@ -10,7 +10,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "nonebot-plugin-fishspeech-tts"
-version = "0.3.3"
+version = "0.3.4"
 description = "小样本TTS,通过调用在线或本地api发送TTS语音"
 authors = [
     {name = "Cvandia",email = "106718176+Cvandia@users.noreply.github.com"},