✨ feature: 预添加语速参数支持，更新相关API和正则表达式

Cvandia · Cvandia · commit b3e68c7515fe · 2024-12-10T09:17:06.000+08:00
diff --git a/README.md b/README.md
@@ -84,13 +84,13 @@ git clone https://github.com/Cvandia/nonebot-plugin-fishspeech-tts
 
 **在.env中添加以下配置**
 
-|     基础配置     |  类型   | 必填项 |      默认值       |                            说明                             |
-| :--------------: | :-----: | :----: | :---------------: | :---------------------------------------------------------: |
-|  tts_is_online   |  bool   |   是   |       True        |                       是否使用云端api                       |
-| tts_chunk_length | literal |   否   |     "normal"      | 请求时音频分片长度，默认为normal，可选：short, normal, long |
-| tts_max_new_tokens   |   int   |   否   |       800       |                   最大音频长度，默认为800,设置为0则为不限制                   |
-|  tts_audio_path  |   str   |   否   | "./data/参考音频" |            语音素材路径，默认为"./data/参考音频"            |
-|    tts_prefix    |   str   |   否   |       None        |                    触发前缀，默认为None                     |
+|      基础配置      |  类型   | 必填项 |      默认值       |                            说明                             |
+| :----------------: | :-----: | :----: | :---------------: | :---------------------------------------------------------: |
+|   tts_is_online    |  bool   |   是   |       True        |                       是否使用云端api                       |
+|  tts_chunk_length  | literal |   否   |     "normal"      | 请求时音频分片长度，默认为normal，可选：short, normal, long |
+| tts_max_new_tokens |   int   |   否   |        800        |          最大音频长度，默认为800,设置为0则为不限制          |
+|   tts_audio_path   |   str   |   否   | "./data/参考音频" |            语音素材路径，默认为"./data/参考音频"            |
+|     tts_prefix     |   str   |   否   |       None        |                    触发前缀，默认为None                     |
 
 **注：参考音频的文件名格式为：［角色名］音频对应的文字标签.[音频后缀名]**
 
@@ -103,12 +103,12 @@ ___
 
 如果你想使用官方的api，请将配置项`tts_is_online`设置为`True`并配置以下
 
-|        配置项        | 类型  | 必填项 | 默认值  |                                                                                     说明                                                                                     |
-| :------------------: | :---: | :----: | :-----: | :--------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
-| online_api_url      |  str  |   否   | "https://api.fish-audio.cn"| 官网api地址  |
-| 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                                                                      |
+|        配置项        | 类型  | 必填项 |           默认值            |                                                                                     说明                                                                                     |
+| :------------------: | :---: | :----: | :-------------------------: | :--------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
+|    online_api_url    |  str  |   否   | "https://api.fish-audio.cn" |                                                                                 官网api地址，可选https://api.fish.audio(被墙)或者默认值                                                                                  |
+| 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                                                                      |
 
 ---
 
diff --git a/nonebot_plugin_fishspeech_tts/fish_audio_api.py b/nonebot_plugin_fishspeech_tts/fish_audio_api.py
@@ -89,6 +89,7 @@ async def generate_servettsrequest(
         text: str,
         speaker_name: str,
         chunk_length: ChunkLength = ChunkLength.NORMAL,
+        # TODO: speed: int = 0,
     ) -> ServeTTSRequest:
         """
         生成TTS请求
@@ -97,6 +98,7 @@ async def generate_servettsrequest(
             text: 待合成文本
             speaker_name: 说话人姓名
             chunk_length: 分片长度
+            TODO:speed: 语速
 
         Returns:
             ServeTTSRequest: TTS请求
diff --git a/nonebot_plugin_fishspeech_tts/fish_speech_api.py b/nonebot_plugin_fishspeech_tts/fish_speech_api.py
@@ -43,13 +43,16 @@ async def generate_servettsrequest(
         text: str,
         speaker_name: str,
         chunk_length: ChunkLength = ChunkLength.NORMAL,
+        # TODO: speed: int = 0,
     ) -> ServeTTSRequest:
         """
         生成TTS请求
 
         Args:
             text: 文本
             speaker_name: 说话人姓名
+            chunk_length: 请求语音的切片长度
+            TODO:speed: 语速
         Returns:
             ServeTTSRequest: TTS请求
         """
diff --git a/nonebot_plugin_fishspeech_tts/matcher.py b/nonebot_plugin_fishspeech_tts/matcher.py
@@ -1,7 +1,8 @@
 from nonebot import get_driver, on_command, on_regex
 from nonebot.adapters import Message
+from nonebot.params import RegexDict
 from nonebot.rule import to_me
-from nonebot_plugin_alconna import Reply, Text, UniMessage, UniMsg
+from nonebot_plugin_alconna import Reply, UniMessage, UniMsg
 
 from .config import config
 from .exception import APIException
@@ -26,16 +27,19 @@
     config.tts_chunk_length, ChunkLength.NORMAL
 )  # 请求语音长度
 
-
-tts_handler = on_regex(rf"^{prefix}(.+?)说([\s\S]*)", block=False)
+# "xxx说xxx -s int"
+tts_handler = on_regex(
+    rf"^{prefix}(?P<speaker>.+?)说(?P<text>.+)?(?:\s+-s\s+(?P<speed>.+))?$",
+    priority=15,
+)
 speaker_list = on_command(
     "语音列表", aliases={"语音角色列表"}, block=True, rule=to_me()
 )
 balance = on_command("语音余额", block=True, rule=to_me())
 
 
 @tts_handler.handle()
-async def tts_handle(message: UniMsg):
+async def tts_handle(message: UniMsg, regex_group: dict = RegexDict()):  # noqa: B008
     if message.has(Reply):
         front_reply = message[Reply, 0].msg
         if isinstance(front_reply, Message):
@@ -44,11 +48,12 @@ async def tts_handle(message: UniMsg):
             text = front_reply
         else:
             text = str(front_reply)
-        reply_msg = message[Text, 0].text
-        speaker = reply_msg.split("说", 1)[0]
+        speaker = regex_group["speaker"]
+        # TODO: speed = regex_group["speed"]
     else:
-        speaker, text = (message[Text, 0].text).split("说", 1)
-        speaker = speaker.replace(prefix, "").strip()  # 去除前缀
+        text = regex_group["text"]
+        speaker = regex_group["speaker"]
+        # TODO: speed = regex_group["speed"]
 
     try:
         fish_audio_api = FishAudioAPI()
@@ -58,12 +63,14 @@ async def tts_handle(message: UniMsg):
             request = await fish_audio_api.generate_servettsrequest(
                 text, speaker, chunk_length
             )
+            # TODO: request = await fish_audio_api.generate_ttsrequest(text, speaker, speed)
             audio = await fish_audio_api.generate_tts(request)
         else:
             await tts_handler.send("正在通过本地api合成语音, 请稍等")
             request = await fish_speech_api.generate_servettsrequest(
                 text, speaker, chunk_length
             )
+            # TODO: request = await fish_speech_api.generate_ttsrequest(text, speaker, speed)
             audio = await fish_speech_api.generate_tts(request)
         await UniMessage.voice(raw=audio).finish()
 
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
