fix: make bs4 an optional dependency

Sinaptik-AI · Oct 10, 2023 · 89b6a1a · 89b6a1a
1 parent 720811f
commit 89b6a1a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/pandasai/helpers/from_google_sheets.py b/pandasai/helpers/from_google_sheets.py
@@ -1,5 +1,4 @@
 import requests
-from bs4 import BeautifulSoup
 import pandas as pd
 import re
 
@@ -15,6 +14,14 @@ def get_google_sheet(src) -> list:
         list: A 2D array representing content of the Google Sheet.
     """
 
+    try:
+        from bs4 import BeautifulSoup
+    except ImportError:
+        raise ImportError(
+            "Could not import beautifulsoup4 python package. "
+            "Please install it with `pip install beautifulsoup4`."
+        )
+
     # The size of the Google sheet that can be read is limited
     raw_html = requests.get(src).text
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,7 @@ matplotlib = "^3.7.1"
 pydantic = "^1"
 sqlalchemy = "^1.4.49"
 duckdb = "^0.8.1"
-beautifulsoup4 = "^4.12.2"
+beautifulsoup4 = {version="^4.12.2", optional = true}
 google-generativeai = {version = "^0.1.0rc2", optional = true}
 google-cloud-aiplatform = {version = "^1.26.1", optional = true}
 langchain = {version = "^0.0.199", optional = true}