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

add chroma upsert

Browse Source
This commit is contained in:
ACBBZ
2024-05-07 01:59:41 +00:00
parent 62c3c6852b
commit d0b597fba0
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/blackbox/blackbox_factory.py
View File

@ -10,6 +10,7 @@ from .fastchat import Fastchat
from .g2e import G2E from .g2e import G2E
from .text_and_image import TextAndImage from .text_and_image import TextAndImage
from .chroma_query import ChromaQuery from .chroma_query import ChromaQuery
from .chroma_upsert import ChromaUpsert
from injector import inject, singleton from injector import inject, singleton
@singleton @singleton
@ -28,7 +29,8 @@ class BlackboxFactory:
audio_chat: AudioChat, audio_chat: AudioChat,
g2e: G2E, g2e: G2E,
text_and_image:TextAndImage, text_and_image:TextAndImage,
chroma_query: ChromaQuery) -> None: chroma_query: ChromaQuery,
chroma_upsert: ChromaUpsert) -> None:
self.models["audio_to_text"] = audio_to_text self.models["audio_to_text"] = audio_to_text
self.models["text_to_audio"] = text_to_audio self.models["text_to_audio"] = text_to_audio
self.models["asr"] = asr self.models["asr"] = asr
@ -40,6 +42,7 @@ class BlackboxFactory:
self.models["g2e"] = g2e self.models["g2e"] = g2e
self.models["text_and_image"] = text_and_image self.models["text_and_image"] = text_and_image
self.models["chroma_query"] = chroma_query self.models["chroma_query"] = chroma_query
self.models["chroma_upsert"] = chroma_upsert
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs) return self.processing(*args, **kwargs)