klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: no default embedding function (#238)

Browse Source
This commit is contained in:
cachho
2023-07-14 08:12:17 +02:00
committed by GitHub
parent c393c214dc
commit e283315efd
1 changed files with 1 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
embedchain/vectordb/chroma_db.py
View File

@@ -11,14 +11,7 @@ class ChromaDB(BaseVectorDB):
"""Vector database using ChromaDB.""" """Vector database using ChromaDB."""
def __init__(self, db_dir=None, ef=None, host=None, port=None): def __init__(self, db_dir=None, ef=None, host=None, port=None):
if ef: self.ef = 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",
)
if host and port: if host and port:
logging.info(f"Connecting to ChromaDB server: {host}:{port}") logging.info(f"Connecting to ChromaDB server: {host}:{port}")