Skip to content

Commit

Permalink
Harrison/unstructured validation (#2111)
Browse files Browse the repository at this point in the history
Co-authored-by: kravetsmic <79907559+kravetsmic@users.noreply.github.com>
  • Loading branch information
hwchase17 and kravetsmic committed Mar 28, 2023
1 parent b25dbcb commit 6e85cbc
Showing 1 changed file with 24 additions and 2 deletions.
26 changes: 24 additions & 2 deletions langchain/document_loaders/url.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,17 +11,36 @@
class UnstructuredURLLoader(BaseLoader):
"""Loader that uses unstructured to load HTML files."""

def __init__(self, urls: List[str], continue_on_failure: bool = True):
def __init__(
self, urls: List[str], continue_on_failure: bool = True, headers: dict = {}
):
"""Initialize with file path."""
try:
import unstructured # noqa:F401
from unstructured.__version__ import __version__ as __unstructured_version__

self.__version = __unstructured_version__
except ImportError:
raise ValueError(
"unstructured package not found, please install it with "
"`pip install unstructured`"
)

if not self.__is_headers_available() and len(headers.keys()) != 0:
logger.warning(
"You are using old version of unstructured. "
"The headers parameter is ignored"
)

self.urls = urls
self.continue_on_failure = continue_on_failure
self.headers = headers

def __is_headers_available(self) -> bool:
_unstructured_version = self.__version.split("-")[0]
unstructured_version = tuple([int(x) for x in _unstructured_version.split(".")])

return unstructured_version >= (0, 5, 7)

def load(self) -> List[Document]:
"""Load file."""
Expand All @@ -30,7 +49,10 @@ def load(self) -> List[Document]:
docs: List[Document] = list()
for url in self.urls:
try:
elements = partition_html(url=url)
if self.__is_headers_available():
elements = partition_html(url=url, headers=self.headers)
else:
elements = partition_html(url=url)
except Exception as e:
if self.continue_on_failure:
logger.error(f"Error fetching or processing {url}, exeption: {e}")
Expand Down

0 comments on commit 6e85cbc

Please sign in to comment.