community[patch]: import flattening fix (#20110)

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>
This commit is contained in:
Leonid Ganeline
2024-04-10 10:01:19 -07:00
committed by GitHub
parent 12190ad728
commit 4cb5f4c353
60 changed files with 2973 additions and 163 deletions

View File

@@ -1,7 +1,68 @@
"""**Graphs** provide a natural language interface to graph databases."""
import importlib
from typing import Any
from typing import TYPE_CHECKING, Any
if TYPE_CHECKING:
from langchain_community.graphs.arangodb_graph import (
ArangoGraph, # noqa: F401
)
from langchain_community.graphs.falkordb_graph import (
FalkorDBGraph, # noqa: F401
)
from langchain_community.graphs.gremlin_graph import (
GremlinGraph, # noqa: F401
)
from langchain_community.graphs.hugegraph import (
HugeGraph, # noqa: F401
)
from langchain_community.graphs.kuzu_graph import (
KuzuGraph, # noqa: F401
)
from langchain_community.graphs.memgraph_graph import (
MemgraphGraph, # noqa: F401
)
from langchain_community.graphs.nebula_graph import (
NebulaGraph, # noqa: F401
)
from langchain_community.graphs.neo4j_graph import (
Neo4jGraph, # noqa: F401
)
from langchain_community.graphs.neptune_graph import (
NeptuneGraph, # noqa: F401
)
from langchain_community.graphs.neptune_rdf_graph import (
NeptuneRdfGraph, # noqa: F401
)
from langchain_community.graphs.networkx_graph import (
NetworkxEntityGraph, # noqa: F401
)
from langchain_community.graphs.ontotext_graphdb_graph import (
OntotextGraphDBGraph, # noqa: F401
)
from langchain_community.graphs.rdf_graph import (
RdfGraph, # noqa: F401
)
from langchain_community.graphs.tigergraph_graph import (
TigerGraph, # noqa: F401
)
__all__ = [
"ArangoGraph",
"FalkorDBGraph",
"GremlinGraph",
"HugeGraph",
"KuzuGraph",
"MemgraphGraph",
"NebulaGraph",
"Neo4jGraph",
"NeptuneGraph",
"NeptuneRdfGraph",
"NetworkxEntityGraph",
"OntotextGraphDBGraph",
"RdfGraph",
"TigerGraph",
]
_module_lookup = {
"ArangoGraph": "langchain_community.graphs.arangodb_graph",