o …wÖi)ã@s\ddlmZmZddlmZddlmZmZGdd„deƒZeeeƒdedefdd „ƒZ d S) é)ÚCutÚMixedCut)Úregistered_prompt_format_fn)ÚModalityÚPromptFormatterc@s<eZdZdZdZdZdddejidœeddejidœiZdS) ÚPlainPromptFormatterz` Plain prompt formatter: there is nothing being added to the user and assistants turns. ÚplainÚ assistantÚuserz |message|Úmessage)ÚtemplateÚslotsN) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚNAMEÚ OUTPUT_ROLErÚTextÚTEMPLATE©rrúb/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/common/prompts/plain.pyrsÿþÿþùrÚcutÚpromptcCsht|tƒr|j}| d¡r|j}nd}dd|idœg}|jdj}dur/| dd|idœ¡| |¡S)NÚcontextÚr r )Úroler rr ) Ú isinstancerÚfirst_non_padding_cutÚ has_customrÚ supervisionsÚtextÚappendÚ encode_dialog)rrÚctxÚturnsÚanswerrrrr+s   rN) Ú lhotse.cutrrÚ&nemo.collections.common.data.prompt_fnrÚ)nemo.collections.common.prompts.formatterrrrrrrrrÚs