community: Use default load() implementation in doc loaders (#18385)

Following https://github.com/langchain-ai/langchain/pull/18289
This commit is contained in:
Christophe Bornet
2024-03-01 20:46:52 +01:00
committed by GitHub
parent 42341bc787
commit 177f51c7bd
49 changed files with 22 additions and 253 deletions

View File

@@ -1,4 +1,4 @@
from typing import Any, Callable, Iterator, List, Mapping, Optional
from typing import Any, Callable, Iterator, Mapping, Optional
from langchain_core.documents import Document
from langchain_core.utils.utils import guard_import
@@ -53,9 +53,6 @@ class AirbyteCDKLoader(BaseLoader):
self._stream_name = stream_name
self._state = state
def load(self) -> List[Document]:
return list(self.lazy_load())
def lazy_load(self) -> Iterator[Document]:
return self._integration._load_data(
stream_name=self._stream_name, state=self._state