ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-14 00:53:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

doc: update swagger

Browse Source
This commit is contained in:
superobk
2024-04-25 15:07:18 +08:00
parent 201d0f374c
commit c062abbe19
4 changed files with 19 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/blackbox/blackbox_factory.py
View File

@ -1,11 +1,9 @@
from .audio_chat import AudioChat
from .sum import SUM
from .sentiment import Sentiment
from .tts import TTS
from .asr import ASR
from .audio_to_text import AudioToText
from .blackbox import Blackbox
from .calculator import Calculator
from .text_to_audio import TextToAudio
from .tesou import Tesou
from .fastchat import Fastchat
@ -18,8 +16,6 @@ class BlackboxFactory:
self.tts = TTS()
self.asr = ASR(".env.yaml")
self.sentiment = Sentiment()
self.sum = SUM()
self.calculator = Calculator()
self.audio_to_text = AudioToText()
self.text_to_audio = TextToAudio()
self.tesou = Tesou()
@ -36,16 +32,12 @@ class BlackboxFactory:
return self.audio_to_text
if blackbox_name == "text_to_audio":
return self.text_to_audio
if blackbox_name == "calculator":
return self.calculator
if blackbox_name == "asr":
return self.asr
if blackbox_name == "tts":
return self.tts
if blackbox_name == "sentiment_engine":
return self.sentiment
if blackbox_name == "sum":
return self.sum
if blackbox_name == "tesou":
return self.tesou
if blackbox_name == "fastchat":