o
    Zh                     @   sT   d dl mZ d dlmZmZmZmZmZ dedefddZ	dee defdd	Z
d
S )    )List)	AIMessageBaseMessageChatMessageHumanMessageSystemMessagemessagereturnc                 C   s   t | trd| j  d| j }|S t | tr!d| j d}|S t | tr,| j }|S t | tr:d| j d}|S td|  )Nz

z: z[INST] z [/INST]z<<SYS>> z	 <</SYS>>zGot unknown type )	
isinstancer   Zrole
capitalizecontentr   r   r   
ValueError)r   message_text r   [/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/chat_models/meta.py"_convert_one_message_to_text_llama   s   
	


r   messagesc                 C   s   d dd | D S )z1Convert a list of messages to a prompt for llama.
c                 S   s   g | ]}t |qS r   )r   ).0r   r   r   r   
<listcomp>   s    z4convert_messages_to_prompt_llama.<locals>.<listcomp>)join)r   r   r   r    convert_messages_to_prompt_llama   s   r   N)typingr   Zlangchain_core.messagesr   r   r   r   r   strr   r   r   r   r   r   <module>   s    	