From 3461ef4b1457580363002b6e1b3def134c0d256e Mon Sep 17 00:00:00 2001
From: Mike Makuch <1mikemakuch@gmail.com>
Date: Tue, 4 Jul 2023 15:00:30 -0500
Subject: [PATCH] add: Allow setting OpenAI organization along with api_key
 (#84)

---
 embedchain/vectordb/chroma_db.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/embedchain/vectordb/chroma_db.py b/embedchain/vectordb/chroma_db.py
index e4981394..30de7bac 100644
--- 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"
 )
 
@@ -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,
-        )
\ No newline at end of file
+        )