mirror of
https://github.com/langgenius/dify.git
synced 2025-07-06 00:31:23 +00:00
26 lines
847 B
Python
26 lines
847 B
Python
|
import openai
|
||
|
|
from models.provider import ProviderName
|
||
|
|
from core.llm.error_handle_wraps import handle_llm_exceptions
|
||
|
|
from core.llm.provider.base import BaseProvider
|
||
|
|
|
||
|
|
|
||
|
|
class Whisper:
|
||
|
|
|
||
|
|
def __init__(self, provider: BaseProvider):
|
||
|
|
self.provider = provider
|
||
|
|
|
||
|
|
if self.provider.get_provider_name() == ProviderName.OPENAI:
|
||
|
|
self.client = openai.Audio
|
||
|
|
self.credentials = provider.get_credentials()
|
||
|
|
|
||
|
|
@handle_llm_exceptions
|
||
|
|
def transcribe(self, file):
|
||
|
|
return self.client.transcribe(
|
||
|
|
model='whisper-1',
|
||
|
|
file=file,
|
||
|
|
api_key=self.credentials.get('openai_api_key'),
|
||
|
|
api_base=self.credentials.get('openai_api_base'),
|
||
|
|
api_type=self.credentials.get('openai_api_type'),
|
||
|
|
api_version=self.credentials.get('openai_api_version'),
|
||
|
|
)
|