Files
dify-plugin-sdks/python/examples/code_based_workflow/tools/tts.py
T
2024-12-02 21:10:36 +08:00

24 lines
701 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())