Files
dify-plugin-sdks/python/examples/code_based_workflow/tools/tts.py
T
-LAN- b3e0612e53 chore: Apply format
Signed-off-by: -LAN- <laipz8200@outlook.com>
2024-12-02 21:18:36 +08:00

22 lines
687 B
Python

import binascii
from collections.abc import Generator
from dify_plugin.entities.model.tts import TTSModelConfig
from dify_plugin.entities.tool import ToolInvokeMessage
from dify_plugin.interfaces.tool import Tool
class TTS(Tool):
def _invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
response = self.session.model.tts.invoke(
model_config=TTSModelConfig(
provider="openai",
model="tts-1",
voice="alloy",
),
content_text="Hello, world!",
)
for chunk in response:
yield self.create_text_message(binascii.hexlify(chunk).decode())