o ̳i @s0ddlmZddlmZmZGdddeZdS))List)MessagePromptTemplateInterfacec@s8eZdZdZdddddZdeedeefdd ZdS) MistralChatTemplatea Formats according to Mistral's `instruct model `_. It is identical to :class:`~torchtune.data.Llama2ChatTemplate`, except it does not support system prompts. Note: This template is only recommended for Mistral's Instruct-v0.1 and Instruct-v0.2 models. Instruct-v0.3 adds additional tags for tool calls, which is not yet supported by this template. .. code-block:: text "[INST] I am going to Paris, what should I see? [/INST] Paris, the capital of France, is known for its stunning architecture..." N)z[INST] z [/INST] )r)systemuser assistantipythonmessagesreturnc Cszg}|D]6}|jdkrtdd|j|jddg|jd|j|jddg}|t|j||j|j|jdq|S)aS Format user and system messages with appropriate tags. Args: messages (List[Message]): a single conversation, structured as a list of `Message` objects Returns: The formatted list of messages Raises: ValueError: If system prompts are provided rz7System prompts are not supported in MistralChatTemplatetextr)typecontent)rolermaskedr eot) r ValueErrortemplaterappendrrr r)selfr formatted_dialoguemessagerr]/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/models/mistral/_prompt_template.py__call__&s,  zMistralChatTemplate.__call__)__name__ __module__ __qualname____doc__rrrrrrrrr srN)typingrtorchtune.datarrrrrrrs