Skip to content

contextvars_utils

with_contextvars(wrapped)

Source code in griptape/utils/contextvars_utils.py
def with_contextvars(wrapped: Callable) -> Callable:
    ctx = contextvars.copy_context()

    def wrapper(*args, **kwargs) -> Any:
        return ctx.run(wrapped, *args, **kwargs)

    return wrapper