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

community[minor]: Add Cassandra ByteStore (#22064)

Browse Source
This commit is contained in:
Christophe Bornet
2024-05-23 16:46:23 +02:00
committed by GitHub
parent fea6b99b16
commit 74947ec894
4 changed files with 350 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -14,7 +14,7 @@ from typing import (
from langchain_core.documents import Document
from langchain_community.document_loaders.base import BaseLoader
from langchain_community.utilities.cassandra import wrapped_response_future
from langchain_community.utilities.cassandra import aexecute_cql
_NOT_SET = object()
@@ -118,11 +118,7 @@ class CassandraLoader(BaseLoader):
)
async def alazy_load(self) -> AsyncIterator[Document]:
for row in await wrapped_response_future(
self.session.execute_async,
self.query,
**self.query_kwargs,
):
for row in await aexecute_cql(self.session, self.query, **self.query_kwargs):
metadata = self.metadata.copy()
metadata.update(self.metadata_mapper(row))
yield Document(