github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-19 13:23:35 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 7 deletions
Show Stats 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, table: Optional[str] = None,
session: Optional[Session] = None, session: Optional[Session] = None,
keyspace: Optional[str] = None, keyspace: Optional[str] = None,
query: Optional[Union[str, Statement]] = None, query: Union[str, Statement, None] = None,
page_content_mapper: Callable[[Any], str] = str, page_content_mapper: Callable[[Any], str] = str,
metadata_mapper: Callable[[Any], dict] = lambda _: {}, 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_timeout: Optional[float] = _NOT_SET, # type: ignore[assignment]
query_trace: bool = False, 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_execution_profile: Any = _NOT_SET,
query_paging_state: Any = None, query_paging_state: Any = None,
query_host: Host = None, query_host: Optional[Host] = None,
query_execute_as: str = None, # type: ignore[assignment] query_execute_as: Optional[str] = None,
) -> None: ) -> None:
""" """
Document Loader for Apache Cassandra. Document Loader for Apache Cassandra.

4
libs/community/tests/integration_tests/document_loaders/test_cassandra.py
View File

@ -2,7 +2,7 @@
Test of Cassandra document loader class `CassandraLoader` Test of Cassandra document loader class `CassandraLoader`
""" """
import os import os
from typing import Any from typing import Any, Iterator
import pytest import pytest
from langchain_core.documents import Document from langchain_core.documents import Document
@ -14,7 +14,7 @@ CASSANDRA_TABLE = "docloader_test_table"
@pytest.fixture(autouse=True, scope="session") @pytest.fixture(autouse=True, scope="session")
def keyspace() -> str: # type: ignore[misc] def keyspace() -> Iterator[str]:
import cassio import cassio
from cassandra.cluster import Cluster from cassandra.cluster import Cluster
from cassio.config import check_resolve_session, resolve_keyspace from cassio.config import check_resolve_session, resolve_keyspace