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

2023-07-04 15:00:30 -05:00
parent 5fa2f5c5a2
commit 3461ef4b14
1 changed files with 2 additions and 1 deletions
--- a/embedchain/vectordb/chroma_db.py
+++ b/embedchain/vectordb/chroma_db.py
@@ -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"
 )