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

@@ -76,9 +76,6 @@ class AstraDBLoader(BaseLoader):
self.nb_prefetched = nb_prefetched
self.extraction_function = extraction_function
def load(self) -> List[Document]:
return list(self.lazy_load())
def lazy_load(self) -> Iterator[Document]:
for doc in self.collection.paginated_find(
filter=self.filter,