Browse Source

refactor: no default embedding function (#238)

cachho 2 years ago
parent
commit
e283315efd
1 changed files with 1 additions and 8 deletions
  1. 1 8
      embedchain/vectordb/chroma_db.py

+ 1 - 8
embedchain/vectordb/chroma_db.py

@@ -11,14 +11,7 @@ class ChromaDB(BaseVectorDB):
     """Vector database using ChromaDB."""
 
     def __init__(self, db_dir=None, ef=None, host=None, port=None):
-        if ef:
-            self.ef = ef
-        else:
-            self.ef = embedding_functions.OpenAIEmbeddingFunction(
-                api_key=os.getenv("OPENAI_API_KEY"),
-                organization_id=os.getenv("OPENAI_ORGANIZATION"),
-                model_name="text-embedding-ada-002",
-            )
+        self.ef = ef
 
         if host and port:
             logging.info(f"Connecting to ChromaDB server: {host}:{port}")