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