فهرست منبع

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

Mike Makuch 2 سال پیش
والد
کامیت
3461ef4b14
1فایلهای تغییر یافته به همراه2 افزوده شده و 1 حذف شده
  1. 2 1
      embedchain/vectordb/chroma_db.py

+ 2 - 1
embedchain/vectordb/chroma_db.py

@@ -7,6 +7,7 @@ from embedchain.vectordb.base_vector_db import BaseVectorDB
 
 
 openai_ef = embedding_functions.OpenAIEmbeddingFunction(
 openai_ef = embedding_functions.OpenAIEmbeddingFunction(
     api_key=os.getenv("OPENAI_API_KEY"),
     api_key=os.getenv("OPENAI_API_KEY"),
+    organization_id=os.getenv("OPENAI_ORGANIZATION"),
     model_name="text-embedding-ada-002"
     model_name="text-embedding-ada-002"
 )
 )
 
 
@@ -27,4 +28,4 @@ class ChromaDB(BaseVectorDB):
     def _get_or_create_collection(self):
     def _get_or_create_collection(self):
         return self.client.get_or_create_collection(
         return self.client.get_or_create_collection(
             'embedchain_store', embedding_function=openai_ef,
             'embedchain_store', embedding_function=openai_ef,
-        )
+        )