add: Allow setting OpenAI organization along with api_key (#84)

This commit is contained in:
Mike Makuch
2023-07-04 15:00:30 -05:00
committed by GitHub
parent 5fa2f5c5a2
commit 3461ef4b14

View File

@@ -7,6 +7,7 @@ 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"
)
@@ -27,4 +28,4 @@ class ChromaDB(BaseVectorDB):
def _get_or_create_collection(self):
return self.client.get_or_create_collection(
'embedchain_store', embedding_function=openai_ef,
)
)