fix: Fix dependency of openai env variables for OpenSourceApp (#144)

This commit fixes dependency of initializing openai env variables
for OpenSourceApp.
This commit is contained in:
Taranjeet Singh
2023-07-05 23:03:15 +05:30
committed by GitHub
parent 659e26e3d7
commit 200f11a0e0
2 changed files with 16 additions and 14 deletions

View File

@@ -5,15 +5,17 @@ from chromadb.utils import embedding_functions
from embedchain.vectordb.base_vector_db import BaseVectorDB
openai_ef = embedding_functions.OpenAIEmbeddingFunction(
api_key=os.getenv("OPENAI_API_KEY"),
organization_id=os.getenv("OPENAI_ORGANIZATION"),
model_name="text-embedding-ada-002"
)
class ChromaDB(BaseVectorDB):
def __init__(self, db_dir=None, ef=None):
self.ef = ef if ef is not None else openai_ef
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"
)
if db_dir is None:
db_dir = "db"
self.client_settings = chromadb.config.Settings(