✨ feat(config): 添加 tts_prefix 配置项

- 在 Config 类中添加 tts_prefix 配置项
- 优化 matcher.py 中的命令前缀处理逻辑
- 更新 README.md 中的配置项说明
- 将版本号从 0.3.1 升级到 0.3.2

Author: Cvandia

README.md

@@ -73,11 +73,12 @@ 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_audio_path|str|否|"./data/参考音频"|语音素材路径，默认为"./data/参考音频"|
+|     基础配置     |  类型   | 必填项 |      默认值       |                            说明                             |
+| :--------------: | :-----: | :----: | :---------------: | :---------------------------------------------------------: |
+|  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                   |
 
 **注：参考音频的文件名格式为：［角色名］音频对应的文字标签.[音频后缀名]**
 
@@ -90,18 +91,18 @@ ___
 
 如果你想使用官方的api，请将配置项`tts_is_online`设置为`True`并配置以下
 
-| 配置项 | 类型 | 必填项 | 默认值 | 说明 |
-|:-----:|:----:|:----:|:---:|:----:|
-|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_authorization |  str  |   是   | "xxxxx" |                                                    官网api鉴权秘钥，详见[链接](https://fish.audio/zh-CN/go-api/api-keys/)                                                    |
+|  online_model_first  | bool  |   否   |  True   | 如果你想调用官方模型，通过自己的参考音频，定制角色音色，将此项设为`False`。当然，如果你没有准备参考音频，也会调用官网已经有的音色，具体详见[链接](https://fish.audio/zh-CN/) |
 
 ---
 
 如果你想使用[自搭](#离线搭建fish-speech)或者其他的[fish-speech](https://github.com/fishaudio/fish-speech)项目的api,请将配置项`tts_is_online`设置为`Fasle`并配置以下
 
-| 配置项 | 类型 | 必填项 | 默认值 | 说明 |
-|:----:|:----:|:----:|:---:|:----:|
-|offline_api_url|str|是|"http://127.0.0.1:8080"|你的`fish-speech`api地址|
+|     配置项      | 类型  | 必填项 |         默认值          |           说明           |
+| :-------------: | :---: | :----: | :---------------------: | :----------------------: |
+| offline_api_url |  str  |   是   | "http://127.0.0.1:8080" | 你的`fish-speech`api地址 |
 
 ## ⭐ 使用
 
@@ -110,11 +111,11 @@ ___
 
 ### 指令：
 
-| 指令 | 需要@ | 范围 | 说明 |权限|
-|:---:|:---:|:---:|:---:|:---:|
-|xxx说xxx|根据配置|all|tts语音生成|all|
-|语音列表|是|all|获取所有角色列表|all|
-|语音余额|是|all|查询api余额|all|
+|   指令   |  需要@   | 范围  |       说明       | 权限  |
+| :------: | :------: | :---: | :--------------: | :---: |
+| xxx说xxx | 根据配置 |  all  |   tts语音生成    |  all  |
+| 语音列表 |    是    |  all  | 获取所有角色列表 |  all  |
+| 语音余额 |    是    |  all  |   查询api余额    |  all  |
 
 ## 🌙 Todo
  - [x] 添加更多配置项

nonebot_plugin_fishspeech_tts/config.py

@@ -13,6 +13,7 @@ class Config(BaseModel):
     tts_is_online: bool = True
     tts_chunk_length: Literal["normal", "short", "long"] = "normal"
     tts_audio_path: str = "./data/参考音频"
+    tts_prefix: Optional[str] = None
 
     # 区分配置
     online_authorization: Optional[str] = "xxxxx"

nonebot_plugin_fishspeech_tts/matcher.py

@@ -14,22 +14,25 @@
 """
 
 
-IS_ONLINE = config.tts_is_online  # True 为在线模式，False 为本地模式
-PREFIX = list(get_driver().config.command_start)[0]  # 命令前缀
-
-
-chunk_length_map = {
+is_online = config.tts_is_online  # True 为在线模式，False 为本地模式
+prefix = (
+    config.tts_prefix
+    if config.tts_prefix
+    else list(get_driver().config.command_start)[0]
+)  # 命令前缀
+
+CHUNKLENGTH_MAP = {
     "normal": ChunkLength.NORMAL,
     "short": ChunkLength.SHORT,
     "long": ChunkLength.LONG,
 }
 
-CHUNKLENGTH = chunk_length_map.get(
+chunk_length = CHUNKLENGTH_MAP.get(
     config.tts_chunk_length, ChunkLength.NORMAL
 )  # 请求语音长度
 
 
-tts_handler = on_regex(rf"^{PREFIX}(.+?)说([\s\S]*)", block=False)
+tts_handler = on_regex(rf"^{prefix}(.+?)说([\s\S]*)", block=False)
 speaker_list = on_command(
     "语音列表", aliases={"语音角色列表"}, block=True, rule=to_me()
 )
@@ -54,16 +57,16 @@ async def tts_handle(message: UniMsg):
     try:
         fish_audio_api = FishAudioAPI()
         fish_speech_api = FishSpeechAPI()
-        if IS_ONLINE:
+        if is_online:
             await tts_handler.send("正在通过在线api合成语音, 请稍等")
             request = await fish_audio_api.generate_servettsrequest(
-                text, speaker, CHUNKLENGTH
+                text, speaker, chunk_length
             )
             audio = await fish_audio_api.generate_tts(request)
         else:
             await tts_handler.send("正在通过本地api合成语音, 请稍等")
             request = await fish_speech_api.generate_servettsrequest(
-                text, speaker, CHUNKLENGTH
+                text, speaker, chunk_length
             )
             audio = await fish_speech_api.generate_tts(request)
         await UniMessage.voice(raw=audio).finish()
@@ -77,7 +80,7 @@ async def speaker_list_handle(event: Event):
     try:
         fish_audio_api = FishAudioAPI()
         fish_speech_api = FishSpeechAPI()
-        if IS_ONLINE:
+        if is_online:
             _list = fish_audio_api.get_speaker_list()
             await speaker_list.finish("语音角色列表: " + ", ".join(_list))
         else:
@@ -91,7 +94,7 @@ async def speaker_list_handle(event: Event):
 async def balance_handle(event: Event):
     try:
         fish_audio_api = FishAudioAPI()
-        if IS_ONLINE:
+        if is_online:
             await balance.send("正在查询在线语音余额, 请稍等")
             balance_float = await fish_audio_api.get_balance()
             await balance.finish(f"语音余额为: {balance_float}")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot-plugin-fishspeech-tts"
-version = "0.3.1"
+version = "0.3.2"
 description = "小样本TTS,通过调用在线或本地api发送TTS语音"
 authors = ["Cvandia <106718176+Cvandia@users.noreply.github.com>"]
 license = "MIT"