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

community: Fix some mypy types in cassandra doc loader (#17570)

Browse Source 
Thank you!
This commit is contained in:
Christophe Bornet
2024-02-15 15:45:22 +01:00
committed by GitHub
parent f3e4a0e27f
commit ff1f985a2a
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -29,18 +29,18 @@ class CassandraLoader(BaseLoader):
table: Optional[str] = None,
session: Optional[Session] = None,
keyspace: Optional[str] = None,
query: Optional[Union[str, Statement]] = None,
query: Union[str, Statement, None] = None,
page_content_mapper: Callable[[Any], str] = str,
metadata_mapper: Callable[[Any], dict] = lambda _: {},
*,
query_parameters: Union[dict, Sequence] = None, # type: ignore[assignment]
query_parameters: Union[dict, Sequence, None] = None,
query_timeout: Optional[float] = _NOT_SET, # type: ignore[assignment]
query_trace: bool = False,
query_custom_payload: dict = None, # type: ignore[assignment]
query_custom_payload: Optional[dict] = None,
query_execution_profile: Any = _NOT_SET,
query_paging_state: Any = None,
query_host: Host = None,
query_execute_as: str = None, # type: ignore[assignment]
query_host: Optional[Host] = None,
query_execute_as: Optional[str] = None,
) -> None:
"""
Document Loader for Apache Cassandra.