diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py
index 0945cfa..4e1c645 100644
--- a/src/blackbox/blackbox_factory.py
+++ b/src/blackbox/blackbox_factory.py
@@ -10,6 +10,7 @@ from .fastchat import Fastchat
 from .g2e import G2E
 from .text_and_image import TextAndImage
 from .chroma_query import ChromaQuery
+from .chroma_upsert import ChromaUpsert
 from injector import inject, singleton
 
 @singleton
@@ -28,7 +29,8 @@ class BlackboxFactory:
                  audio_chat: AudioChat,
                  g2e: G2E,
                  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["text_to_audio"] = text_to_audio
         self.models["asr"] = asr
@@ -40,6 +42,7 @@ class BlackboxFactory:
         self.models["g2e"] = g2e
         self.models["text_and_image"] = text_and_image
         self.models["chroma_query"] = chroma_query
+        self.models["chroma_upsert"] = chroma_upsert
 
     def __call__(self, *args, **kwargs):
         return self.processing(*args, **kwargs)