Преглед изворни кода

fix: Dont initialize chroma and embedding function in init config. (#289)

Taranjeet Singh пре 2 година
родитељ
комит
2889799f10
1 измењених фајлова са 1 додато и 8 уклоњено
  1. 1 8
      embedchain/config/InitConfig.py

+ 1 - 8
embedchain/config/InitConfig.py

@@ -22,15 +22,8 @@ class InitConfig(BaseConfig):
         :param port: Optional. Port for the database server.
         """
         self._setup_logging(log_level)
-
-        if db is None:
-            from embedchain.vectordb.chroma_db import ChromaDB
-
-            self.db = ChromaDB(ef=ef)
-        else:
-            self.db = db
-
         self.ef = ef
+        self.db = db
         self.host = host
         self.port = port
         self.id = id