o
}oi @ s< d Z ddlmZmZ dZdZdZdZdZG dd d eZ d
S ) zz
Implemented following the guide at https://www.promptingguide.ai/models/mistral-7b#chat-template-for-mistral-7b-instruct
)ModalityPromptFormatterzz[INST]z[/INST]zz
c @ sR e Zd ZdZdZddeide de ddej id ed
e
dej id iZ d S ) MistralPromptFormattermistral assistantpreambletemplateuserz |message| message)r slotsz |message|N) __name__
__module__ __qualname__NAME OUTPUT_ROLE MISTRAL_BOSMISTRAL_PROMPT_BEGINMISTRAL_PROMPT_ENDr TextMISTRAL_END_OF_TURNTEMPLATE r r [/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/common/prompts/mistral.pyr s" r N)
__doc__)nemo.collections.common.prompts.formatterr r r r r r
MISTRAL_NLr r r r r s