feat:豆神一期，添加音色，查询音色列表

Author: 赵嘉琦

tests/integration_tests/test_tts.py

@@ -0,0 +1,33 @@
+import os
+import pytest
+from zhipuai.api_resource.tts import TTSApi
+
+# 请根据实际情况填写
+BASE_URL_LIST = "http://localhost:9203"
+BASE_URL_ADD = "http://localhost:9203"
+API_TOKEN = "eyJhbGciOiJIUzUxMiJ9.eyJ1c2VyX3R5cGUiOiJTRVJWSUNFIiwidXNlcl9pZCI6MywiYXBpX2tleSI6IjM0ZDhiM2ExNDQ2OTQ0MmY5MTJkNzg5MTJjZjZlMjU1IiwidXNlcl9rZXkiOiIyNzQyNjM0My1iYjI0LTQ4ZDktODVjNy01ODFmMjUwNzBiMzMiLCJjdXN0b21lcl9pZCI6IjEwMDAwMyIsInVzZXJuYW1lIjoiY3BjMTk4NiJ9.3MsaFjRAyArDp2WZsVbvKGPrVjkKvGB5EfBafprcFogbDeZv4s9VCQt4wUaR4FVon1tiL_pnqtMaI6qGGXs0Qg"
+AUDIO_FILE_PATH = r"C:\Users\zjq18\Downloads\北京今天的天气.wav"
+
+@pytest.mark.integration
+def test_list_voices():
+    tts_api = TTSApi(base_url=BASE_URL_LIST, api_key=API_TOKEN)
+    resp = tts_api.list_voices()
+    assert resp.code == 200
+    assert resp.success
+    assert isinstance(resp.data, list)
+    if resp.data:
+        assert hasattr(resp.data[0], "voiceId")
+
+@pytest.mark.integration
+def test_add_voice():
+    if not os.path.exists(AUDIO_FILE_PATH):
+        pytest.skip("音频文件不存在，跳过上传测试")
+    tts_api = TTSApi(base_url=BASE_URL_ADD, api_key=API_TOKEN)
+    resp = tts_api.add_voice(
+        voice_name="磁性嗓音123",
+        voice_text="今天北京的天气怎么样？",
+        file_path=AUDIO_FILE_PATH
+    )
+    assert resp.code == 200
+    assert resp.success
+    assert resp.data.voiceName == "磁性嗓音123" 

zhipuai/api_resource/tts/__init__.py

@@ -0,0 +1 @@
+from .tts import TTSApi 

zhipuai/api_resource/tts/tts.py

@@ -0,0 +1,32 @@
+import requests
+from typing import Optional
+from zhipuai.types.tts.voice import VoiceListResponse, VoiceAddResponse
+
+class TTSApi:
+    def __init__(self, base_url: str, api_key: Optional[str] = None):
+        self.base_url = base_url.rstrip('/')
+        self.api_key = api_key
+
+    def list_voices(self) -> VoiceListResponse:
+        url = f"{self.base_url}/v4/voice/list"
+        headers = {}
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+
+        resp = requests.post(url,headers=headers)
+        resp.raise_for_status()
+        return VoiceListResponse.parse_obj(resp.json())
+
+    def add_voice(self, voice_name: str, voice_text: str, file_path: str) -> VoiceAddResponse:
+        url = f"{self.base_url}/v4/voice/add"
+        headers = {}
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+        files = {
+            "voiceName": (None, voice_name),
+            "voiceText": (None, voice_text),
+            "file": open(file_path, "rb"),
+        }
+        resp = requests.post(url, headers=headers, files=files)
+        resp.raise_for_status()
+        return VoiceAddResponse.parse_obj(resp.json()) 

zhipuai/types/tts/__init__.py

@@ -0,0 +1 @@
+from .voice import Voice, VoiceListResponse, VoiceAddResponse 

zhipuai/types/tts/voice.py

@@ -0,0 +1,24 @@
+from typing import List
+from pydantic import BaseModel
+
+class Voice(BaseModel):
+    id: int
+    voiceId: str
+    voiceName: str
+    voiceType: str
+    voiceStatus: str
+    voiceText: str
+    createTime: str
+    updateTime: str
+
+class VoiceListResponse(BaseModel):
+    code: int
+    msg: str
+    data: List[Voice]
+    success: bool
+
+class VoiceAddResponse(BaseModel):
+    code: int
+    msg: str
+    data: Voice
+    success: bool