Change HF embedding library (#1440)

embedchain/embedder/huggingface.py

@@ -2,7 +2,14 @@ import os
 from typing import Optional
 
 from langchain_community.embeddings import HuggingFaceEmbeddings
-from langchain_community.embeddings.huggingface import HuggingFaceInferenceAPIEmbeddings
+
+try:
+    from langchain_huggingface import HuggingFaceEndpointEmbeddings
+except ModuleNotFoundError:
+    raise ModuleNotFoundError(
+        "The required dependencies for HuggingFaceHub are not installed."
+        "Please install with `pip install langchain_huggingface`"
+    ) from None
 
 from embedchain.config import BaseEmbedderConfig
 from embedchain.embedder.base import BaseEmbedder
@@ -19,10 +26,9 @@ class HuggingFaceEmbedder(BaseEmbedder):
                     "Please set the HUGGINGFACE_ACCESS_TOKEN environment variable or pass API Key in the config."
                 )
 
-            embeddings = HuggingFaceInferenceAPIEmbeddings(
-                model_name=self.config.model,
-                api_url=self.config.endpoint,
-                api_key=self.config.api_key or os.getenv("HUGGINGFACE_ACCESS_TOKEN"),
+            embeddings = HuggingFaceEndpointEmbeddings(
+                model=self.config.endpoint,
+                huggingfacehub_api_token=self.config.api_key or os.getenv("HUGGINGFACE_ACCESS_TOKEN"),
             )
         else:
             embeddings = HuggingFaceEmbeddings(model_name=self.config.model)