from modelscope.utils.constant import Tasks
from .pipeline_tool import ModelscopePipelineTool


class TranslationEn2ZhTool(ModelscopePipelineTool):
    default_model = 'damo/nlp_csanmt_translation_en2zh'
    description = '根据输入指令，将相应的英文文本翻译成中文回复'
    name = 'modelscope_text-translation-en2zh'
    task = Tasks.translation
    parameters: list = [{
        'name': 'input',
        'description': '用户输入的英文文本',
        'required': True
    }]

    def _parse_output(self, origin_result, *args, **kwargs):
        return {'result': origin_result['translation']}