fix: rename

This commit is contained in:
Dan Chen
2024-03-18 17:58:23 +08:00
parent f6dd6159a8
commit 2612d5d370
4 changed files with 59 additions and 18 deletions

View File

@ -0,0 +1,17 @@
from blackbox.audio_to_text import AudioToText
from blackbox.blackbox import Blackbox
from blackbox.calculator import Calculator
from blackbox.text_to_audio import TextToAudio
class BlackboxFactory:
def create_blackbox(self, blackbox_type: str, blackbox_config: dict) -> Blackbox:
if blackbox_type == "audio_to_text":
return AudioToText(blackbox_config)
if blackbox_type == "text_to_audio":
return TextToAudio(blackbox_config)
if blackbox_type == "calculator":
return Calculator(blackbox_config)
raise ValueError("Invalid blockbox type")