U di!@sddlmZmZddlmZdddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddМZGdd҄deZdS))AutoModelForSeq2SeqLM AutoTokenizer) PipelineToolZace_ArabZace_LatnZacm_ArabZacq_ArabZaeb_ArabZafr_LatnZajp_ArabZaka_LatnZamh_EthiZapc_ArabZarb_ArabZarb_LatnZars_ArabZary_ArabZarz_ArabZasm_BengZast_LatnZawa_DevaZayr_LatnZazb_ArabZazj_LatnZbak_CyrlZbam_LatnZban_LatnZbel_CyrlZbem_LatnZben_BengZbho_DevaZbjn_ArabZbjn_LatnZbod_TibtZbos_LatnZbug_LatnZbul_CyrlZcat_LatnZceb_LatnZces_LatnZcjk_LatnZckb_ArabZcrh_LatnZcym_LatnZdan_LatnZdeu_LatnZdik_LatnZdyu_LatnZdzo_TibtZell_GrekZeng_LatnZepo_LatnZest_LatnZeus_LatnZewe_LatnZfao_LatnZfij_LatnZfin_LatnZfon_LatnZfra_LatnZfur_LatnZfuv_LatnZgla_LatnZgle_LatnZglg_LatnZgrn_LatnZguj_GujrZhat_LatnZhau_LatnZheb_HebrZhin_DevaZhne_DevaZhrv_LatnZhun_LatnZhye_ArmnZibo_LatnZilo_LatnZind_LatnZisl_LatnZita_LatnZjav_LatnZjpn_JpanZkab_LatnZkac_LatnZkam_LatnZkan_KndaZkas_ArabZkas_DevaZkat_GeorZknc_ArabZknc_LatnZkaz_CyrlZkbp_LatnZkea_LatnZkhm_KhmrZkik_LatnZkin_LatnZkir_CyrlZkmb_LatnZkmr_LatnZkon_LatnZkor_HangZlao_LaooZlij_LatnZlim_LatnZlin_LatnZlit_LatnZlmo_LatnZltg_LatnZltz_LatnZlua_LatnZlug_LatnZluo_LatnZlus_LatnZlvs_LatnZmag_DevaZmai_DevaZmal_MlymZmar_DevaZmin_ArabZmin_LatnZmkd_CyrlZplt_LatnZmlt_LatnZmni_BengZkhk_CyrlZmos_LatnZmri_LatnZmya_MymrZnld_LatnZnno_LatnZnob_LatnZnpi_DevaZnso_LatnZnus_LatnZnya_LatnZoci_LatnZgaz_LatnZory_OryaZpag_LatnZpan_GuruZpap_LatnZpes_ArabZpol_LatnZpor_LatnZprs_ArabZpbt_ArabZquy_LatnZron_LatnZrun_LatnZrus_CyrlZsag_LatnZsan_DevaZsat_OlckZscn_LatnZshn_MymrZsin_SinhZslk_LatnZslv_LatnZsmo_LatnZsna_LatnZsnd_ArabZsom_LatnZsot_LatnZspa_LatnZals_LatnZsrd_LatnZsrp_CyrlZssw_LatnZsun_LatnZswe_LatnZswh_LatnZszl_LatnZtam_TamlZtat_CyrlZtel_TeluZtgk_CyrlZtgl_LatnZtha_ThaiZtir_EthiZtaq_LatnZtaq_TfngZtpi_LatnZtsn_LatnZtso_LatnZtuk_LatnZtum_LatnZtur_LatnZtwi_LatnZtzm_TfngZuig_ArabZukr_CyrlZumb_LatnZurd_ArabZuzn_LatnZvec_LatnZvie_LatnZwar_LatnZwol_LatnZxho_LatnZydd_HebrZyor_LatnZyue_HantZzho_HansZzho_HantZzsm_LatnZzul_Latn)zAcehnese ArabiczAcehnese LatinzMesopotamian ArabiczTa'izzi-Adeni ArabiczTunisian ArabicZ AfrikaanszSouth Levantine ArabicZAkanZAmhariczNorth Levantine ArabiczModern Standard Arabicz Modern Standard Arabic Romanizedz Najdi ArabiczMoroccan ArabiczEgyptian ArabicZAssameseZAsturianZAwadhizCentral AymarazSouth AzerbaijanizNorth AzerbaijaniZBashkirZBambaraZBalineseZ BelarusianZBembaZBengaliZBhojpuriz Banjar Arabicz Banjar LatinzStandard TibetanZBosnianZBuginese BulgarianZCatalanZCebuanoZCzechZChokwezCentral Kurdishz Crimean TatarZWelshZDanishZGermanzSouthwestern DinkaZDyulaZDzongkhaGreekZEnglishZ EsperantoZEstonianZBasqueZEweZFaroeseZFijianZFinnishZFonZFrenchZFriulianzNigerian FulfuldezScottish GaelicZIrishZGalicianZGuaraniZGujaratizHaitian CreoleZHausaHebrewZHindiZ ChhattisgarhiZCroatianZ HungarianZArmenianZIgboZIlocanoZ IndonesianZ IcelandicZItalianZJavaneseJapaneseZKabyleZJingphoZKambaZKannadazKashmiri ArabiczKashmiri DevanagariZGeorgianzCentral Kanuri ArabiczCentral Kanuri LatinZKazakhuKabiyèZ KabuverdianuZKhmerZKikuyuZ KinyarwandaZKyrgyzZKimbunduzNorthern KurdishZKikongoKoreanZLaoZLigurianZ LimburgishZLingalaZ LithuanianZLombardZ LatgalianZ Luxembourgishz Luba-KasaiZGandaZLuoZMizozStandard LatvianZMagahiZMaithiliZ MalayalamZMarathizMinangkabau Arabic zMinangkabau LatinZ MacedonianzPlateau MalagasyZMaltesezMeitei BengalizHalh MongolianZMossiZMaoriZBurmeseZDutchzNorwegian NynorskuNorwegian BokmålZNepalizNorthern SothoZNuerZNyanjaZOccitanzWest Central OromoZOdiaZ PangasinanzEastern PanjabiZ PapiamentozWestern PersianZPolishZ PortugueseZDarizSouthern PashtozAyacucho QuechuaZRomanianZRundiRussianZSangoZSanskritZSantaliZSicilianZShanZSinhalaZSlovakZ SlovenianZSamoanZShonaZSindhiZSomalizSouthern SothoZSpanishz Tosk AlbanianZ SardinianZSerbianZSwatiZ SundaneseZSwedishZSwahiliZSilesianZTamilZTatarZTeluguZTajikZTagalogThaiZTigrinyazTamasheq LatinzTamasheq Tifinaghz Tok PisinZTswanaZTsongaZTurkmenZTumbukaTurkishZTwizCentral Atlas TamazightZUyghurZ UkrainianZUmbunduZUrduzNorthern UzbekZVenetianZ VietnameseZWarayZWolofZXhosazEastern YiddishZYorubaz Yue ChinesezChinese SimplifiedzChinese TraditionalzStandard MalayZZuluc@sreZdZdZeZdZdeedZ dZ e Z e Zddddd ddd dd ZdZd d ZddZddZdS)TranslationToolz Example: ```py from transformers.agents import TranslationTool translator = TranslationTool() translator("This is a super nice API!", src_lang="English", tgt_lang="French") ``` z facebook/nllb-200-distilled-600MzThis is a tool that translates text from a language to another.Both `src_lang`and `tgt_lang` should belong to this list of languages: .Z translatorstringzThe text to translate)type descriptionzbThe language of the text to translate. Written in plain English, such as 'Romanian', or 'Albanian'zhThe language for the desired ouput language. Written in plain English, such as 'Romanian', or 'Albanian')textsrc_langtgt_langcCsX||jkrt|d||jkr0t|d|j|}|j|}|jj|d||dS)Nz is not a supported language.pt)Zreturn_tensorsrr) lang_to_code ValueErrorZ pre_processorZ_build_translation_inputs)selfrrrrC/tmp/pip-unpacked-wheel-bm_b0l5e/transformers/agents/translation.pyencodes    zTranslationTool.encodecCs|jjf|S)N)modelgenerate)rinputsrrrforwardszTranslationTool.forwardcCs|jj|dddS)NT)Zskip_special_tokens)Zpost_processordecodetolist)routputsrrrr"szTranslationTool.decodeN)__name__ __module__ __qualname____doc__LANGUAGE_CODESrZdefault_checkpointlistkeysrnamerZpre_processor_classrZ model_classr output_typerr r"rrrrrs(   rN)Z models.autorrZtoolsrr)rrrrrs Q