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