mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -04:00
Yeuoly
25 lines
798 B
Python
25 lines
798 B
Python
from binascii import hexlify
|
|
from typing import IO
|
|
|
|
from dify_plugin.core.entities.invocation import InvokeType
|
|
from dify_plugin.core.runtime import BackwardsInvocation
|
|
from dify_plugin.entities.model.speech2text import Speech2TextModelConfig, Speech2TextResult
|
|
|
|
|
|
class Speech2TextInvocation(BackwardsInvocation[Speech2TextResult]):
|
|
def invoke(self, model_config: Speech2TextModelConfig, file: IO[bytes]) -> str:
|
|
"""
|
|
Invoke speech2text
|
|
"""
|
|
for data in self._backwards_invoke(
|
|
InvokeType.Speech2Text,
|
|
Speech2TextResult,
|
|
{
|
|
**model_config.model_dump(),
|
|
"file": hexlify(file.read()),
|
|
},
|
|
):
|
|
return data.result
|
|
|
|
raise Exception("No response from speech2text")
|