From a6e6e9bb86d3667803dfe693522a3a4c991f6c7d Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Wed, 9 Aug 2023 16:13:06 -0400 Subject: [PATCH] Fix airbyte loader (#8998) Fix airbyte loader https://github.com/langchain-ai/langchain/issues/8996 --- libs/langchain/langchain/document_loaders/airbyte.py | 3 +-- .../tests/unit_tests/document_loaders/test_airbyte.py | 9 +++++++++ 2 files changed, 10 insertions(+), 2 deletions(-) create mode 100644 libs/langchain/tests/unit_tests/document_loaders/test_airbyte.py diff --git a/libs/langchain/langchain/document_loaders/airbyte.py b/libs/langchain/langchain/document_loaders/airbyte.py index 1a6aca6c0c..aa67070465 100644 --- a/libs/langchain/langchain/document_loaders/airbyte.py +++ b/libs/langchain/langchain/document_loaders/airbyte.py @@ -1,10 +1,9 @@ """Loads local airbyte json files.""" from typing import Any, Callable, Iterator, List, Mapping, Optional -from libs.langchain.langchain.utils.utils import guard_import - from langchain.docstore.document import Document from langchain.document_loaders.base import BaseLoader +from langchain.utils.utils import guard_import RecordHandler = Callable[[Any, Optional[str]], Document] diff --git a/libs/langchain/tests/unit_tests/document_loaders/test_airbyte.py b/libs/langchain/tests/unit_tests/document_loaders/test_airbyte.py new file mode 100644 index 0000000000..8e3b93037f --- /dev/null +++ b/libs/langchain/tests/unit_tests/document_loaders/test_airbyte.py @@ -0,0 +1,9 @@ +"""Test the airbyte document loader. + +Light test to ensure that the airbyte document loader can be imported. +""" + + +def test_airbyte_import() -> None: + """Test that the airbyte document loader can be imported.""" + from langchain.document_loaders import airbyte # noqa