o
    }o™iÒ  ã                   @   s(   d Z ddlmZmZ G dd„ deƒZdS )z[
Implemented following the guide at https://www.promptingguide.ai/models/phi-2#phi-2-usage
é    )ÚModalityÚPromptFormatterc                   @   s<   e Zd ZdZdZdZdddejidœeddejidœiZd	S )
ÚExamplePromptFormatterzç
    The simplest possible prompt formatter implementation.

    It defines a dialog of the form:

        User: Hi.
        Assistant: Hi, how can I help you?
        User: What's the time?
        Assistant: It's 9 o'clock.

    Úexample_prompt_formatÚ	assistantÚuserzUser: |message|
Úmessage)ÚtemplateÚslotszAssistant: |message|
N)	Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚNAMEÚOUTPUT_ROLEr   ÚTextÚTEMPLATE© r   r   ú[/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/common/prompts/example.pyr      s    ÿþÿþùr   N)r   Ú)nemo.collections.common.prompts.formatterr   r   r   r   r   r   r   Ú<module>   s   