[Feature] Add support for OpenAI assistants and support openai version >=1.0.0 (#921)

embedchain/embedder/openai.py

@@ -7,13 +7,7 @@ from embedchain.config import BaseEmbedderConfig
 from embedchain.embedder.base import BaseEmbedder
 from embedchain.models import VectorDimensions
 
-try:
-    from chromadb.utils import embedding_functions
-except RuntimeError:
-    from embedchain.utils import use_pysqlite3
-
-    use_pysqlite3()
-    from chromadb.utils import embedding_functions
+from .chroma_embeddings import OpenAIEmbeddingFunction
 
 
 class OpenAIEmbedder(BaseEmbedder):
@@ -30,11 +24,10 @@ class OpenAIEmbedder(BaseEmbedder):
                 raise ValueError(
                     "OPENAI_API_KEY or OPENAI_ORGANIZATION environment variables not provided"
                 )  # noqa:E501
-            embedding_fn = embedding_functions.OpenAIEmbeddingFunction(
+            embedding_fn = OpenAIEmbeddingFunction(
                 api_key=os.getenv("OPENAI_API_KEY"),
                 organization_id=os.getenv("OPENAI_ORGANIZATION"),
                 model_name=self.config.model,
             )
-
         self.set_embedding_fn(embedding_fn=embedding_fn)
         self.set_vector_dimension(vector_dimension=VectorDimensions.OPENAI.value)