extend the KnowledgeGraph such it can get a Graph reference instead of creating a connection.
db = FalkorDB(host='localhost', port=6379)
db.select_graph("kg_name")
kg = KnowledgeGraph(
model_config=KnowledgeGraphModelConfig.with_model(model),
graph=graph
)