diff --git a/libs/community/langchain_community/graph_vectorstores/extractors/__init__.py b/libs/community/langchain_community/graph_vectorstores/extractors/__init__.py index 5acd800003..8d6a829ef6 100644 --- a/libs/community/langchain_community/graph_vectorstores/extractors/__init__.py +++ b/libs/community/langchain_community/graph_vectorstores/extractors/__init__.py @@ -6,22 +6,21 @@ from langchain_community.graph_vectorstores.extractors.hierarchy_link_extractor HierarchyInput, HierarchyLinkExtractor, ) +from langchain_community.graph_vectorstores.extractors.html_link_extractor import ( + HtmlInput, + HtmlLinkExtractor, +) from langchain_community.graph_vectorstores.extractors.keybert_link_extractor import ( KeybertInput, KeybertLinkExtractor, ) - -from .html_link_extractor import ( - HtmlInput, - HtmlLinkExtractor, -) -from .link_extractor import ( +from langchain_community.graph_vectorstores.extractors.link_extractor import ( LinkExtractor, ) -from .link_extractor_adapter import ( +from langchain_community.graph_vectorstores.extractors.link_extractor_adapter import ( LinkExtractorAdapter, ) -from .link_extractor_transformer import ( +from langchain_community.graph_vectorstores.extractors.link_extractor_transformer import ( # noqa: E501 LinkExtractorTransformer, ) diff --git a/libs/community/langchain_community/graph_vectorstores/extractors/link_extractor_transformer.py b/libs/community/langchain_community/graph_vectorstores/extractors/link_extractor_transformer.py index 52e4347d45..3a5b4e83a5 100644 --- a/libs/community/langchain_community/graph_vectorstores/extractors/link_extractor_transformer.py +++ b/libs/community/langchain_community/graph_vectorstores/extractors/link_extractor_transformer.py @@ -1,4 +1,4 @@ -from typing import Any, Iterable, Sequence +from typing import Any, Sequence from langchain_core.documents import Document from langchain_core.documents.transformers import BaseDocumentTransformer @@ -21,7 +21,7 @@ class LinkExtractorTransformer(BaseDocumentTransformer): extract_links.transform_documents(docs) """ - def __init__(self, link_extractors: Iterable[LinkExtractor[Document]]): + def __init__(self, link_extractors: Sequence[LinkExtractor[Document]]): """Create a DocumentTransformer which adds extracted links to each document.""" self.link_extractors = link_extractors