mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 02:15:22 -04:00
Yeuoly
23 lines
724 B
Python
23 lines
724 B
Python
from binascii import unhexlify
|
|
from collections.abc import Generator
|
|
|
|
from dify_plugin.core.entities.invocation import InvokeType
|
|
from dify_plugin.core.runtime import BackwardsInvocation
|
|
from dify_plugin.entities.model.tts import TTSModelConfig, TTSResult
|
|
|
|
|
|
class TTSInvocation(BackwardsInvocation[TTSResult]):
|
|
def invoke(self, model_config: TTSModelConfig, content_text: str) -> Generator[bytes, None, None]:
|
|
"""
|
|
Invoke tts
|
|
"""
|
|
for data in self._backwards_invoke(
|
|
InvokeType.TTS,
|
|
TTSResult,
|
|
{
|
|
**model_config.model_dump(),
|
|
"content_text": content_text,
|
|
},
|
|
):
|
|
yield unhexlify(data.result)
|