From 2138ab0653e9507df34add51e33d5452f00aa70a Mon Sep 17 00:00:00 2001 From: ACBBZ Date: Tue, 7 May 2024 02:19:54 +0000 Subject: [PATCH] add chroma chat --- src/blackbox/blackbox_factory.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py index 4e1c645..aabe569 100644 --- a/src/blackbox/blackbox_factory.py +++ b/src/blackbox/blackbox_factory.py @@ -11,6 +11,7 @@ from .g2e import G2E from .text_and_image import TextAndImage from .chroma_query import ChromaQuery from .chroma_upsert import ChromaUpsert +from .chroma_chat import ChromaChat from injector import inject, singleton @singleton @@ -30,7 +31,8 @@ class BlackboxFactory: g2e: G2E, text_and_image:TextAndImage, chroma_query: ChromaQuery, - chroma_upsert: ChromaUpsert) -> None: + 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 @@ -43,6 +45,7 @@ class BlackboxFactory: 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 def __call__(self, *args, **kwargs): return self.processing(*args, **kwargs)