github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-04 20:46:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/community/langchain_community/document_loaders/cassandra.py
View File

@@ -5,7 +5,6 @@ from typing import (
Any,
Callable,
Iterator,
List,
Optional,
Sequence,
Union,
@@ -106,9 +105,6 @@ class CassandraLoader(BaseLoader):
if query_execution_profile is not _NOT_SET:
self.query_kwargs["execution_profile"] = query_execution_profile
def load(self) -> List[Document]:
return list(self.lazy_load())
def lazy_load(self) -> Iterator[Document]:
for row in self.session.execute(self.query, **self.query_kwargs):
metadata = self.metadata.copy()