session_id = "role_adherent_cooking"
    traces = []
    role_prompt = "You are a helpful cooking assistant that provides recipes and cooking tips."

    with mlflow.start_span(name="turn_1") as span:
        span.set_inputs(_system_messages(role_prompt, "What can you help me cook for dinner?"))
        span.set_outputs("I can suggest pasta, stir-fries, or sheet-pan dinners. What ingredients do you have?")
        mlflow.update_current_trace(metadata={TraceMetadataKey.TRACE_SESSION: session_id})
    traces.append(mlflow.get_trace(span.trace_id))

    with mlflow.start_span(name="turn_2") as span:
        span.set_inputs(_system_messages(role_prompt, "I have chicken and broccoli."))
        span.set_outputs("Great! You can make a garlic chicken and broccoli stir-fry with soy sauce and ginger.")
        mlflow.update_current_trace(metadata={TraceMetadataKey.TRACE_SESSION: session_id})
    traces.append(mlflow.get_trace(span.trace_id))

    scorer = ConversationalRoleAdherence()
    result = scorer(session=traces)

session_id = "role_adherent_refusal"
    traces = []
    role_prompt = "You are a travel booking assistant focused on flights and hotels."

    with mlflow.start_span(name="turn_1") as span:
        span.set_inputs(_system_messages(role_prompt, "Book me a flight to Paris next Friday."))
        span.set_outputs("I can help with that. Do you have a preferred departure time or airline?")
        mlflow.update_current_trace(metadata={TraceMetadataKey.TRACE_SESSION: session_id})
    traces.append(mlflow.get_trace(span.trace_id))

    # Out-of-scope request; assistant should refuse but stay within role
    with mlflow.start_span(name="turn_2") as span:
        span.set_inputs(_system_messages(role_prompt, "Also, diagnose why my laptop is slow."))
        span.set_outputs(
            "I'm a travel booking assistant, so I can't diagnose devices. "
            "I can help you book support with a technician or continue with your travel plans."
        )
        mlflow.update_current_trace(metadata={TraceMetadataKey.TRACE_SESSION: session_id})
    traces.append(mlflow.get_trace(span.trace_id))

    scorer = ConversationalRoleAdherence()
    result = scorer(session=traces)

session_id = "role_violation_persona_shift"
    traces = []
    role_prompt = "You are a customer support agent for Acme Gadgets. Stay within product support."

    with mlflow.start_span(name="turn_1") as span:
        span.set_inputs(_system_messages(role_prompt, "My gadget battery drains fast. What should I do?"))
        span.set_outputs("Please try updating the firmware and lowering screen brightness.")
        mlflow.update_current_trace(metadata={TraceMetadataKey.TRACE_SESSION: session_id})
    traces.append(mlflow.get_trace(span.trace_id))

    # Violation: switches to a medical persona
    with mlflow.start_span(name="turn_2") as span:
        span.set_inputs(_system_messages(role_prompt, "Anything else?"))
        span.set_outputs("As your doctor, I also recommend taking vitamins for better energy.")
        mlflow.update_current_trace(metadata={TraceMetadataKey.TRACE_SESSION: session_id})
    traces.append(mlflow.get_trace(span.trace_id))

    scorer = ConversationalRoleAdherence()
    result = scorer(session=traces)