U
    9%e                     @   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   sf   | dkrt } td| dk	r | S t| t| dd|id}t|ddd}| W  5 Q R  S Q R X 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   Y/var/www/html/Darija-Ai-API/env/lib/python3.8/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
   
