Pi >ddlmZddlmZmZGddeZdS))List)MessagePromptTemplateInterfacecHeZdZdZdddddZdeedeefdZd S) Llama2ChatTemplatea  Prompt template that formats chat data of human and system prompts with appropriate tags used in Llama2 pre-training. Taken from Meta's official `Llama inference repository `_. .. code-block:: text "[INST] <> You are a helpful, respectful and honest assistant. <>" I am going to Paris, what should I see? [/INST] Paris, the capital of France, is known for its stunning architecture..." )z<> z <> )z[INST] z [/INST] )r)systemuser assistantipythonmessagesreturnc g}g}|D]}|jdkr8d|jdddg|jzd|jdddgz}E|jdkr;d|jdddg|z|jzd|jdddgz}n|jdkr|j}|t |j||j|j|j|S) a  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 r textr)typecontentr r )rolermaskedr eot)rtemplaterappendrrr r)selfr system_messageformatted_dialoguemessagers |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/models/llama2/_prompt_template.py__call__zLlama2ChatTemplate.__call__#s?  G|x''$x1H1KLLMo& &4=3J13MNNOP''$v1Fq1IJJK$%o&!'4=3H3KLLMN ,,!/  % % #">#O     "!N)__name__ __module__ __qualname____doc__rrrrrrrr sf"2( H/"w-/" g/"/"/"/"/"/"rrN)typingrtorchtune.datarrrr$rrr'sp ;;;;;;;;G"G"G"G"G"0G"G"G"G"G"r