src/infra/context.py

from contextvars import ContextVar

session_context: ContextVar[str] = ContextVar("session_context", default=None)

def set_session_id(session_id: str):
    return session_context.set(session_id)

def get_session_id() -> str:
    return session_context.get()