🚚 修改css文件命名，支持配置超时时间

Author: FrostN0v0

nonebot_plugin_deepseek/__init__.py

@@ -280,7 +280,7 @@ def parse_model_dict(model_dict: dict[str, dict[str, list[str]]], start_index: i
         model_lines = "".join(f"<div>{line}</div>" for line in model_list_msg.split("\n") if line)
         model_html = f"<h2 class='header-title'>{header_msg}</h2><div class='models-container'>{model_lines}</div>"
         final_html = header_html + model_html
-        css_path = str(Path(__file__).parent / "resources/text.css")
+        css_path = str(Path(__file__).parent / "resources/tts_models.css")
         await deepseek.finish(UniMessage.image(raw=await text_to_pic(text=final_html, css_path=css_path, width=1440)))
     await deepseek.finish(message)
 

nonebot_plugin_deepseek/apis/request.py

@@ -70,7 +70,7 @@ async def get_tts_models(cls) -> list[TTSModelInfo]:
                     f"{tts_config.base_url}/models",
                     headers={**cls._headers},
                     json={"version": tts_config.tts_version},
-                    timeout=30,
+                    timeout=tts_config.timeout,
                 )
             if response.status_code != 200:
                 raise RequestException(f"获取 TTS 模型列表失败，状态码: {response.status_code}")
@@ -92,6 +92,7 @@ async def text_to_speach(cls, text: str, model: str) -> bytes:
             "app_key": tts_config.access_token,
             "access_token": tts_config.access_token,
             "version": tts_config.tts_version,
+            "dl_url": tts_config.dl_url,
             **model_config.to_dict(),
         }
 
@@ -102,12 +103,12 @@ async def text_to_speach(cls, text: str, model: str) -> bytes:
                     f"{tts_config.base_url}/infer_single",
                     headers={**cls._headers, "Authorization": f"Bearer {tts_config.access_token}"},
                     json=json,
-                    timeout=50,
+                    timeout=tts_config.timeout,
                 )
             tts_logger("DEBUG", f"Response: {response.status_code} {response.text}")
             if audio_url := response.json().get("audio_url"):
                 async with httpx.AsyncClient() as client:
-                    response = await client.get(audio_url, timeout=50)
+                    response = await client.get(audio_url, timeout=tts_config.timeout)
                     return response.content
             else:
                 raise RequestException("语音合成失败")

nonebot_plugin_deepseek/config.py

@@ -310,6 +310,16 @@ class ScopedTTSConfig(BaseModel):
     """Your GPT-Sovits API Access Token"""
     tts_version: str = "v4"
     """Your GPT-Sovits API Version"""
+    dl_url: str = ""
+    """audio download url"""
+    timeout: int = Field(default=60)
+
+    @model_validator(mode="before")
+    @classmethod
+    def check_audio_dl_url(cls, data: dict) -> dict:
+        if not data.get("audio_dl_url") and data.get("base_url"):
+            data["audio_dl_url"] = data["base_url"]
+        return data
 
     def get_enable_tts(self) -> list[str]:
         if isinstance(self.enable_models, bool):