community[patch]: Add driver config param for neo4j graph (#20772)

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Tomaz Bratanic
2024-04-24 23:14:41 +02:00
committed by GitHub
parent 13751c3297
commit 9efab3ed66
2 changed files with 24 additions and 1 deletions

View File

@@ -273,3 +273,21 @@ def test_neo4j_filtering_labels() -> None:
# Assert both are empty
assert graph.structured_schema["node_props"] == {}
assert graph.structured_schema["relationships"] == []
def test_driver_config() -> None:
"""Test that neo4j works with driver config."""
url = os.environ.get("NEO4J_URI")
username = os.environ.get("NEO4J_USERNAME")
password = os.environ.get("NEO4J_PASSWORD")
assert url is not None
assert username is not None
assert password is not None
graph = Neo4jGraph(
url=url,
username=username,
password=password,
driver_config={"max_connection_pool_size": 1},
)
graph.query("RETURN 'foo'")