Pi >ddlmZddlmZmZGddeZdS))List)MessagePromptTemplateInterfacecHeZdZdZdddddZdeedeefdZdS) 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 @g}|D]}|jdkrtdd|j|jddg|jzd|j|jddgz}|t |j||j|j|j|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 r z7System prompts are not supported in MistralChatTemplatetextr)typecontent)rolermaskedr eot) r ValueErrortemplaterappendrrr r)selfr formatted_dialoguemessagers }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/models/mistral/_prompt_template.py__call__zMistralChatTemplate.__call__&s"   G|x'' M %w|1LQ1OPPQo& &4=3Nq3QRRST  % % #">#O     "!)__name__ __module__ __qualname____doc__rrrrrrrr sf(( H&"w-&" g&"&"&"&"&"&"rrN)typingrtorchtune.datarrrr$rrr'sp ;;;;;;;;A"A"A"A"A"1A"A"A"A"A"r