ai_helper.put_labels(): add optional keys= parameter