from swift.llm import TemplateType

if __name__ == '__main__':
    template_name_list = TemplateType.get_template_name_list()
    tn_gen = ', '.join([tn for tn in template_name_list if 'generation' in tn])
    tn_chat = ', '.join([tn for tn in template_name_list if 'generation' not in tn])
    print(f'Text Generation: {tn_gen}')
    print(f'Chat: {tn_chat}')