14555a25d0
* feat: simple graph pipeline * feat: implement incremental graph generation * fix: various bug fixes * fix: upgrade weaviate-client --------- Co-authored-by: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
"""Factory function to get the appropriate graph client based on the graph type."""
|
|
|
|
from .config import get_graph_config
|
|
from .graph_db_interface import GraphDBInterface
|
|
from .networkx.adapter import NetworkXAdapter
|
|
|
|
|
|
async def get_graph_engine() -> GraphDBInterface :
|
|
"""Factory function to get the appropriate graph client based on the graph type."""
|
|
config = get_graph_config()
|
|
|
|
if config.graph_database_provider == "neo4j":
|
|
try:
|
|
from .neo4j_driver.adapter import Neo4jAdapter
|
|
|
|
return Neo4jAdapter(
|
|
graph_database_url = config.graph_database_url,
|
|
graph_database_username = config.graph_database_username,
|
|
graph_database_password = config.graph_database_password
|
|
)
|
|
except:
|
|
pass
|
|
|
|
elif config.graph_database_provider == "falkorb":
|
|
try:
|
|
from .falkordb.adapter import FalcorDBAdapter
|
|
|
|
return FalcorDBAdapter(
|
|
graph_database_url = config.graph_database_url,
|
|
graph_database_username = config.graph_database_username,
|
|
graph_database_password = config.graph_database_password,
|
|
graph_database_port = config.graph_database_port
|
|
)
|
|
except:
|
|
pass
|
|
|
|
graph_client = NetworkXAdapter(filename = config.graph_file_path)
|
|
|
|
if graph_client.graph is None:
|
|
await graph_client.load_graph_from_file()
|
|
|
|
return graph_client
|