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

update chat

Browse Source
This commit is contained in:
ACBBZ
2024-05-23 02:52:32 +00:00
committed by superobk
parent e2367a7aa8
commit b42b13d0e6
2 changed files with 26 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/blackbox/blackbox_factory.py
View File

@ -1,10 +1,8 @@
from . import melotts
from .audio_chat import AudioChat
from .sentiment import Sentiment
from .tts import TTS
from .asr import ASR
from .audio_to_text import AudioToText
#from .emotion import Emotion
from .blackbox import Blackbox
# from .text_to_audio import TextToAudio
# from .tesou import Tesou
@ -25,12 +23,10 @@ class BlackboxFactory:
@inject
def __init__(self,
audio_to_text: AudioToText,
text_to_audio: TextToAudio,
asr: ASR,
tts: TTS,
sentiment_engine: Sentiment,
#emotion: Emotion,
tesou: Tesou,
fastchat: Fastchat,
audio_chat: AudioChat,
g2e: G2E,
@ -39,23 +35,22 @@ class BlackboxFactory:
#chroma_upsert: ChromaUpsert,
#chroma_chat: ChromaChat,
melotts: MeloTTS,
vlms: VLMS) -> None:
vlms: VLMS,
chroma_query: ChromaQuery,
chroma_upsert: ChromaUpsert,
chroma_chat: ChromaChat) -> None:
self.models["audio_to_text"] = audio_to_text
self.models["text_to_audio"] = text_to_audio
self.models["asr"] = asr
self.models["tts"] = tts
self.models["sentiment_engine"] = sentiment_engine
self.models["tesou"] = tesou
#self.models["emotion"] = emotion
self.models["fastchat"] = fastchat
self.models["audio_chat"] = audio_chat
self.models["g2e"] = g2e
self.models["text_and_image"] = text_and_image
#self.models["chroma_query"] = chroma_query
#self.models["chroma_upsert"] = chroma_upsert
#self.models["chroma_chat"] = chroma_chat
self.models["melotts"] = melotts
self.models["vlms"] = vlms
self.models["chroma_query"] = chroma_query
self.models["chroma_upsert"] = chroma_upsert
self.models["chroma_chat"] = chroma_chat
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)