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 + )