a
    d                     @   s4   d dl Z ddlmZ dZdZdddZdd
dZdS )    N   )cached_filez
Human: <<task>>

Assistant: z!huggingface-tools/default-promptszchat_prompt_template.txtzrun_prompt_template.txt)Zchatrunr   c                 C   sr   | du rt } td| dur | S t| t| dd|id}t|ddd}| W  d   S 1 sd0    Y  dS )	z\
    Downloads and caches the prompt from a repo and returns it contents (if necessary)
    Nz\sZdatasetZagent)Z	repo_type
user_agentrzutf-8)encoding)DEFAULT_PROMPTS_REPOresearchr   PROMPT_FILESopenread)Zprompt_or_repo_idZ
agent_namemodeZprompt_filef r   c/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/tools/prompts.pydownload_prompt!   s    r   )r   )r	   utilsr   ZCHAT_MESSAGE_PROMPTr   r   r   r   r   r   r   <module>   s
   
