mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
4cb5f4c353
This PR should make it easier for linters to do type checking and for IDEs to jump to definition of code. See #20050 as a template for this PR. - As a byproduct: Added 3 missed `test_imports`. - Added missed `SolarChat` in to __init___.py Added it into test_import ut. - Added `# type: ignore` to fix linting. It is not clear, why linting errors appear after ^ changes. --------- Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
26 lines
553 B
Python
26 lines
553 B
Python
from langchain_community.graphs import __all__, _module_lookup
|
|
|
|
EXPECTED_ALL = [
|
|
"MemgraphGraph",
|
|
"NetworkxEntityGraph",
|
|
"Neo4jGraph",
|
|
"NebulaGraph",
|
|
"BaseNeptuneGraph",
|
|
"NeptuneAnalyticsGraph",
|
|
"NeptuneGraph",
|
|
"NeptuneRdfGraph",
|
|
"KuzuGraph",
|
|
"HugeGraph",
|
|
"RdfGraph",
|
|
"ArangoGraph",
|
|
"FalkorDBGraph",
|
|
"TigerGraph",
|
|
"OntotextGraphDBGraph",
|
|
"GremlinGraph",
|
|
]
|
|
|
|
|
|
def test_all_imports() -> None:
|
|
assert set(__all__) == set(EXPECTED_ALL)
|
|
assert set(__all__) == set(_module_lookup.keys())
|