fix: 简化get_audio方法中的提示文本生成逻辑,清除冗余判断逻辑

This commit is contained in:
Raven95676
2025-06-25 10:55:51 +08:00
parent c0810a674f
commit dc279dde4a
@@ -44,10 +44,7 @@ class ProviderGeminiTTSAPI(TTSProvider):
async def get_audio(self, text: str) -> str:
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
path = os.path.join(temp_dir, f"gemini_tts_{uuid.uuid4()}.wav")
if self.prefix:
prompt = f"{self.prefix}: {text}"
else:
prompt = text
prompt = f"{self.prefix}: {text}" if self.prefix else text
response = await self.client.models.generate_content(
model=self.model,
contents=prompt,
@@ -79,7 +76,4 @@ class ProviderGeminiTTSAPI(TTSProvider):
wf.setframerate(24000)
wf.writeframes(response.candidates[0].content.parts[0].inline_data.data)
if not os.path.exists(path):
raise Exception(f"Failed to save audio to {path}.")
return path