github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-16 23:13:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Add async methods to AstraDBCache (#17415)

Browse Source 
Adds async methods to AstraDBCache
This commit is contained in:
Christophe Bornet
2024-02-15 05:10:08 +01:00
committed by GitHub
parent e438fe6be9
commit ca2d4078f3
6 changed files with 465 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -19,7 +19,7 @@ from langchain_core.documents import Document
from langchain_core.runnables import run_in_executor
from langchain_community.document_loaders.base import BaseLoader
from langchain_community.utilities.astradb import AstraDBEnvironment
from langchain_community.utilities.astradb import _AstraDBEnvironment
if TYPE_CHECKING:
from astrapy.db import AstraDB, AsyncAstraDB
@@ -44,7 +44,7 @@ class AstraDBLoader(BaseLoader):
nb_prefetched: int = 1000,
extraction_function: Callable[[Dict], str] = json.dumps,
) -> None:
astra_env = AstraDBEnvironment(
astra_env = _AstraDBEnvironment(
token=token,
api_endpoint=api_endpoint,
astra_db_client=astra_db_client,