Track users and sessions

from openlayer.lib import set_user_session_context, clear_user_session_context

# In your middleware or request handler
def handle_request(request): # Extract user and session from your authentication system
    user_id = get_user_id_from_request(request)
    session_id = get_session_id_from_request(request)

    # Set default context for all traces in this request
    set_user_session_context(user_id=user_id, session_id=session_id)

    try:
        # Your application logic with traced functions
        result = process_user_request(request.data)
        return result
    finally:
        # Clean up context when request is complete
        clear_user_session_context()

from flask import Flask, request
from openlayer.lib import set_user_session_context, clear_user_session_context, trace

app = Flask(__name__)

@app.before_request
def before_request():
    # Extract from headers, cookies, or JWT tokens
    user_id = request.headers.get('X-User-ID')
    session_id = request.headers.get('X-Session-ID')

    if user_id or session_id:
        set_user_session_context(user_id=user_id, session_id=session_id)

@app.after_request
def after_request(response):
    clear_user_session_context()
    return response

@app.route('/chat')
@trace()
def chat_endpoint():
    # This trace will automatically include user_id and session_id
    user_message = request.json.get('message')
    response = generate_chat_response(user_message)
    return {'response': response}

from openlayer.lib import update_trace_user_session

@trace()
def process_request():
    update_trace_user_session(
        user_id="different_user_123",
        session_id="different_session_123"
    )
    return "result"