mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
58360a1e53
Fix some __init__ files and add a unit test
96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
"""**Graphs** provide a natural language interface to graph databases."""
|
|
|
|
import importlib
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
if TYPE_CHECKING:
|
|
from langchain_community.graphs.arangodb_graph import (
|
|
ArangoGraph,
|
|
)
|
|
from langchain_community.graphs.falkordb_graph import (
|
|
FalkorDBGraph,
|
|
)
|
|
from langchain_community.graphs.gremlin_graph import (
|
|
GremlinGraph,
|
|
)
|
|
from langchain_community.graphs.hugegraph import (
|
|
HugeGraph,
|
|
)
|
|
from langchain_community.graphs.kuzu_graph import (
|
|
KuzuGraph,
|
|
)
|
|
from langchain_community.graphs.memgraph_graph import (
|
|
MemgraphGraph,
|
|
)
|
|
from langchain_community.graphs.nebula_graph import (
|
|
NebulaGraph,
|
|
)
|
|
from langchain_community.graphs.neo4j_graph import (
|
|
Neo4jGraph,
|
|
)
|
|
from langchain_community.graphs.neptune_graph import (
|
|
BaseNeptuneGraph,
|
|
NeptuneAnalyticsGraph,
|
|
NeptuneGraph,
|
|
)
|
|
from langchain_community.graphs.neptune_rdf_graph import (
|
|
NeptuneRdfGraph,
|
|
)
|
|
from langchain_community.graphs.networkx_graph import (
|
|
NetworkxEntityGraph,
|
|
)
|
|
from langchain_community.graphs.ontotext_graphdb_graph import (
|
|
OntotextGraphDBGraph,
|
|
)
|
|
from langchain_community.graphs.rdf_graph import (
|
|
RdfGraph,
|
|
)
|
|
from langchain_community.graphs.tigergraph_graph import (
|
|
TigerGraph,
|
|
)
|
|
|
|
__all__ = [
|
|
"ArangoGraph",
|
|
"FalkorDBGraph",
|
|
"GremlinGraph",
|
|
"HugeGraph",
|
|
"KuzuGraph",
|
|
"BaseNeptuneGraph",
|
|
"MemgraphGraph",
|
|
"NebulaGraph",
|
|
"Neo4jGraph",
|
|
"NeptuneGraph",
|
|
"NeptuneRdfGraph",
|
|
"NeptuneAnalyticsGraph",
|
|
"NetworkxEntityGraph",
|
|
"OntotextGraphDBGraph",
|
|
"RdfGraph",
|
|
"TigerGraph",
|
|
]
|
|
|
|
_module_lookup = {
|
|
"ArangoGraph": "langchain_community.graphs.arangodb_graph",
|
|
"FalkorDBGraph": "langchain_community.graphs.falkordb_graph",
|
|
"GremlinGraph": "langchain_community.graphs.gremlin_graph",
|
|
"HugeGraph": "langchain_community.graphs.hugegraph",
|
|
"KuzuGraph": "langchain_community.graphs.kuzu_graph",
|
|
"MemgraphGraph": "langchain_community.graphs.memgraph_graph",
|
|
"NebulaGraph": "langchain_community.graphs.nebula_graph",
|
|
"Neo4jGraph": "langchain_community.graphs.neo4j_graph",
|
|
"BaseNeptuneGraph": "langchain_community.graphs.neptune_graph",
|
|
"NeptuneAnalyticsGraph": "langchain_community.graphs.neptune_graph",
|
|
"NeptuneGraph": "langchain_community.graphs.neptune_graph",
|
|
"NeptuneRdfGraph": "langchain_community.graphs.neptune_rdf_graph",
|
|
"NetworkxEntityGraph": "langchain_community.graphs.networkx_graph",
|
|
"OntotextGraphDBGraph": "langchain_community.graphs.ontotext_graphdb_graph",
|
|
"RdfGraph": "langchain_community.graphs.rdf_graph",
|
|
"TigerGraph": "langchain_community.graphs.tigergraph_graph",
|
|
}
|
|
|
|
|
|
def __getattr__(name: str) -> Any:
|
|
if name in _module_lookup:
|
|
module = importlib.import_module(_module_lookup[name])
|
|
return getattr(module, name)
|
|
raise AttributeError(f"module {__name__} has no attribute {name}")
|