fix: dependencies (#416)

embedchain/apps/CustomApp.py

@@ -68,7 +68,7 @@ class CustomApp(EmbedChain):
                 return CustomApp._get_azure_openai_answer(prompt, config)
 
         except ImportError as e:
-            raise ImportError(e.msg) from None
+            raise ModuleNotFoundError(e.msg) from None
 
     @staticmethod
     def _get_openai_answer(prompt: str, config: ChatConfig) -> str:

embedchain/apps/OpenSourceApp.py

@@ -43,8 +43,8 @@ class OpenSourceApp(EmbedChain):
         try:
             from gpt4all import GPT4All
         except ModuleNotFoundError:
-            raise ValueError(
-                "The GPT4All python package is not installed. Please install it with `pip install GPT4All`"
+            raise ModuleNotFoundError(
+                "The GPT4All python package is not installed. Please install it with `pip install embedchain[opensource]`"  # noqa E501
             ) from None
 
         return GPT4All(model)

embedchain/config/apps/OpenSourceAppConfig.py

@@ -50,4 +50,10 @@ class OpenSourceAppConfig(BaseAppConfig):
 
         :returns: The default embedding function
         """
-        return embedding_functions.SentenceTransformerEmbeddingFunction(model_name="all-MiniLM-L6-v2")
+        try:
+            return embedding_functions.SentenceTransformerEmbeddingFunction(model_name="all-MiniLM-L6-v2")
+        except ValueError as e:
+            print(e)
+            raise ModuleNotFoundError(
+                "The open source app requires extra dependencies. Install with `pip install embedchain[opensource]`"
+            ) from None