From e80b0932ee0cc355ba6b378c5805afbd73014f3c Mon Sep 17 00:00:00 2001 From: Ben Chambers <35960+bjchambers@users.noreply.github.com> Date: Tue, 23 Jul 2024 07:28:06 -0700 Subject: [PATCH] community[patch]: small fixes to link extractors (#24528) - **Description:** small fixes to imports / types in the link extraction work --------- Co-authored-by: Eugene Yurtsev --- .../graph_vectorstores/extractors/__init__.py | 15 +++++++-------- .../extractors/link_extractor_transformer.py | 4 ++-- 2 files changed, 9 insertions(+), 10 deletions(-) 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