
    .`i
                     `   U d dl mZ d dlmZ d dlmZ d dlmZ  ee          Z	 ee          j
        Zeeegedz  f         z  Zeed<   dededz  fd	Zdededz  fd
Zedz  edz  edz  edz  edz  edz  eedz  eedz  edz  dZeeef         ed<   dededdfdZdedededz  fdZdS )    )Callable)Path)	TypeAlias)init_loggerNChatTemplatePathtokenizer_name_or_pathreturnc                 T    |                      d          r
t          dz  S t          dz  S )Nz-Chattemplate_chatml.jinjatemplate_basic.jinja)endswithCHAT_TEMPLATES_DIRr   s    /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/transformers_utils/chat_templates/registry.py _get_qwen_chat_template_fallbackr      s0    &&w// <!$;;; 666    c                 :    d| v sd| v r
t           dz  S t           dz  S )Nz4.54_5ztemplate_minicpmv45.jinjar   )r   r   s    r   $_get_minicpmv_chat_template_fallbackr      s3    &&&%3I*I*I!$???  777r   ztemplate_blip2.jinjar   ztemplate_deepseek_ocr.jinjaztemplate_deepseek_vl2.jinjaztemplate_fuyu.jinja)zblip-2	chameleonclipdeepseek_ocrdeepseek_vl_v2fuyuminicpmv	paligemmaqwensiglipsiglip2%_MODEL_TYPE_TO_CHAT_TEMPLATE_FALLBACK
model_typechat_templatec                 d    | t           v rt                              d| |           |t           | <   d S )NziModel type %s already has a chat template registered. It will be overwritten by the new chat template %s.)r    loggerwarning)r!   r"   s     r   $register_chat_template_fallback_pathr&   /   sF     :::B		
 	
 	
 9F)*555r   c                 v    t                               |           }t          |          r ||          }|d S |S )N)r    getcallable)r!   r   r"   s      r   get_chat_template_fallback_pathr*   >   sH     :==jIIM >%&<==tr   )collections.abcr   pathlibr   typingr   vllm.loggerr   __file__r$   parentr   strr   __annotations__r   r   r    dictr&   r*    r   r   <module>r5      s   % $ $ $ $ $ $             # # # # # #	X		T(^^* "XseTD[.@%AA ) A A A7S 7TD[ 7 7 7 78 8PT 8 8 8 8 !#99#&<<!77&)FF(+HH!664#&<<, #99!$::F F %tC1A,A'B   FF#F 
F F F F 
D[     r   