update: import bs4 lazily.

This commit lazily loads the bs4 import. This is needed so that not everyone has to install the bs4 package
mem0ai · Nov 17, 2023 · 5724c5d · 5724c5d
1 parent 023a614
commit 5724c5d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/embedchain/utils.py b/embedchain/utils.py
@@ -5,7 +5,6 @@
 import string
 from typing import Any
 
-from bs4 import BeautifulSoup
 from schema import Optional, Or, Schema
 
 from embedchain.models.data_type import DataType
@@ -16,6 +15,8 @@ def parse_content(content, type):
     if type not in implemented:
         raise ValueError(f"Parser type {type} not implemented. Please choose one of {implemented}")
 
+    from bs4 import BeautifulSoup
+
     soup = BeautifulSoup(content, type)
     original_size = len(str(soup.get_text()))