auth/jwt_utils.py

from flask_jwt_extended import create_access_token

def generate_jwt(user_id):
    return create_access_token(identity=user_id)