Poetry fixes (#1382)

embedchain/loaders/excel_file.py

@@ -2,10 +2,11 @@ import hashlib
 import importlib.util
 
 try:
+    import unstructured  # noqa: F401
     from langchain_community.document_loaders import UnstructuredExcelLoader
 except ImportError:
     raise ImportError(
-        'Excel file requires extra dependencies. Install with `pip install --upgrade "embedchain[dataloaders]"`'
+        'Excel file requires extra dependencies. Install with `pip install "unstructured[local-inference, all-docs]"`'
     ) from None
 
 if importlib.util.find_spec("openpyxl") is None and importlib.util.find_spec("xlrd") is None:

embedchain/loaders/google_drive.py

@@ -9,7 +9,14 @@ except ImportError:
     ) from None
 
 from langchain_community.document_loaders import GoogleDriveLoader as Loader
-from langchain_community.document_loaders import UnstructuredFileIOLoader
+
+try:
+    import unstructured  # noqa: F401
+    from langchain_community.document_loaders import UnstructuredFileIOLoader
+except ImportError:
+    raise ImportError(
+        'Unstructured file requires extra dependencies. Install with `pip install "unstructured[local-inference, all-docs]"`'  # noqa: E501
+    ) from None
 
 from embedchain.helpers.json_serializable import register_deserializable
 from embedchain.loaders.base_loader import BaseLoader

embedchain/loaders/unstructured_file.py

@@ -10,11 +10,11 @@ class UnstructuredLoader(BaseLoader):
     def load_data(self, url):
         """Load data from an Unstructured file."""
         try:
-            from langchain_community.document_loaders import \
-                UnstructuredFileLoader
+            import unstructured  # noqa: F401
+            from langchain_community.document_loaders import UnstructuredFileLoader
         except ImportError:
             raise ImportError(
-                'Unstructured file requires extra dependencies. Install with `pip install --upgrade "embedchain[dataloaders]"`'  # noqa: E501
+                'Unstructured file requires extra dependencies. Install with `pip install "unstructured[local-inference, all-docs]"`'  # noqa: E501
             ) from None
 
         loader = UnstructuredFileLoader(url)

embedchain/loaders/xml.py

@@ -1,10 +1,11 @@
 import hashlib
 
 try:
+    import unstructured  # noqa: F401
     from langchain_community.document_loaders import UnstructuredXMLLoader
 except ImportError:
     raise ImportError(
-        'XML file requires extra dependencies. Install with `pip install --upgrade "embedchain[dataloaders]"`'
+        'XML file requires extra dependencies. Install with `pip install "unstructured[local-inference, all-docs]"`'
     ) from None
 from embedchain.helpers.json_serializable import register_deserializable
 from embedchain.loaders.base_loader import BaseLoader