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 chat

Browse Source
This commit is contained in:
ACBBZ
2024-05-07 02:19:54 +00:00
parent 95d6c3785c
commit 2138ab0653
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

@ -11,6 +11,7 @@ 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 .chroma_upsert import ChromaUpsert
from .chroma_chat import ChromaChat
from injector import inject, singleton from injector import inject, singleton
@singleton @singleton
@ -30,7 +31,8 @@ class BlackboxFactory:
g2e: G2E, g2e: G2E,
text_and_image:TextAndImage, text_and_image:TextAndImage,
chroma_query: ChromaQuery, chroma_query: ChromaQuery,
chroma_upsert: ChromaUpsert) -> None: chroma_upsert: ChromaUpsert,
chroma_chat: ChromaChat) -> 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
@ -43,6 +45,7 @@ class BlackboxFactory:
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 self.models["chroma_upsert"] = chroma_upsert
self.models["chroma_chat"] = chroma_chat
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs) return self.processing(*args, **kwargs)