def get_assume_role_input(role_arn, duration):
    """Create input for assume_role."""
    return {
        'RoleArn': role_arn,
        'RoleSessionName': 'terrasnow_enterprise',
        'DurationSeconds': duration
    }