"""Neo4j connector helpers."""

from contextlib import contextmanager
from typing import Iterator

from neo4j import Driver, Session


@contextmanager
def neo4j_session(driver: Driver, access_mode: str = "READ") -> Iterator[Session]:
    """Provide a Neo4j session for a single unit of work."""
    with driver.session(
        default_access_mode=access_mode,
    ) as session:
        yield session