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

add chroma upsert

Browse Source
This commit is contained in:
ACBBZ
2024-05-07 01:58:06 +00:00
parent 62b37e7f20
commit 62c3c6852b
2 changed files with 144 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/blackbox/blackbox_factory.py
View File

@ -9,6 +9,7 @@ from .tesou import Tesou
from .fastchat import Fastchat
from .g2e import G2E
from .text_and_image import TextAndImage
from .chroma_query import ChromaQuery
from injector import inject, singleton
@singleton
@ -26,7 +27,8 @@ class BlackboxFactory:
fastchat: Fastchat,
audio_chat: AudioChat,
g2e: G2E,
text_and_image:TextAndImage ) -> None:
text_and_image:TextAndImage,
chroma_query: ChromaQuery) -> None:
self.models["audio_to_text"] = audio_to_text
self.models["text_to_audio"] = text_to_audio
self.models["asr"] = asr
@ -37,6 +39,7 @@ class BlackboxFactory:
self.models["audio_chat"] = audio_chat
self.models["g2e"] = g2e
self.models["text_and_image"] = text_and_image
self.models["chroma_query"] = chroma_query
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)