github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-03 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use Postponed Evaluation of Annotations in Astra and Cassandra doc loaders (#16694)

Browse Source 
Minor/cosmetic change
This commit is contained in:
Christophe Bornet
2024-01-29 01:39:27 +01:00
committed by GitHub
parent bc7607a4e9
commit 2e3af04080
3 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,3 +1,5 @@
from __future__ import annotations
import json
import logging
import threading
@@ -31,8 +33,8 @@ class AstraDBLoader(BaseLoader):
collection_name: str,
token: Optional[str] = None,
api_endpoint: Optional[str] = None,
astra_db_client: Optional["AstraDB"] = None,
async_astra_db_client: Optional["AsyncAstraDB"] = None,
astra_db_client: Optional[AstraDB] = None,
async_astra_db_client: Optional[AsyncAstraDB] = None,
namespace: Optional[str] = None,
filter_criteria: Optional[Dict[str, Any]] = None,
projection: Optional[Dict[str, Any]] = None,

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

@@ -1,3 +1,5 @@
from __future__ import annotations
from typing import (
TYPE_CHECKING,
Any,
@@ -25,9 +27,9 @@ class CassandraLoader(BaseLoader):
def __init__(
self,
table: Optional[str] = None,
session: Optional["Session"] = None,
session: Optional[Session] = None,
keyspace: Optional[str] = None,
query: Optional[Union[str, "Statement"]] = None,
query: Optional[Union[str, Statement]] = None,
page_content_mapper: Callable[[Any], str] = str,
metadata_mapper: Callable[[Any], dict] = lambda _: {},
*,
@@ -37,7 +39,7 @@ class CassandraLoader(BaseLoader):
query_custom_payload: dict = None,
query_execution_profile: Any = _NOT_SET,
query_paging_state: Any = None,
query_host: "Host" = None,
query_host: Host = None,
query_execute_as: str = None,
) -> None:
"""