refactor: no default embedding function (#238)

This commit is contained in:
cachho
2023-07-14 08:12:17 +02:00
committed by GitHub
parent c393c214dc
commit e283315efd

View File

@@ -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}")