
    .`i
                        d dl mZmZ d dlmZmZ d dlmZ d dlm	Z	 d dl
mZ ddlmZ erd dlmZ  ee          Zd	d
ddddZe G d d                      Z ed e                                D                       Z	 ddZdS )    )	dataclassfield)TYPE_CHECKINGAny)init_logger)tokenizer_args_from_config)resolve_obj_by_qualname   )RendererLike)ModelConfig)deepseek_v32DeepseekV32Renderer)hf
HfRenderer)grok2Grok2Renderer)mistralMistralRenderer)
terratorchTerratorchRenderer)r   r   r   r   r   c                       e Zd ZU  ee          Zeeeeef         f         ed<   dedededdfdZ	dede
e         fd	Zded
ddeeef         defdZdS )RendererRegistry)default_factory	renderersrenderer_modemodule
class_namereturnNc                 j    || j         v rt                              d|||           ||f| j         |<   d S )NzS%s.%s is already registered for renderer_mode=%r. It is overwritten by the new one.)r   loggerwarningselfr   r   r   s       k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/renderers/registry.pyregisterzRendererRegistry.register    sK    DN**NN4   *0(<}%t    c                     || j         vrt          d|d          | j         |         \  }}t                              d| d|           t	          | d|           S )Nz)No renderer registered for renderer_mode=.zLoading z for renderer_mode=)r   
ValueErrorr    
debug_oncer	   r"   s       r$   load_renderer_clsz"RendererRegistry.load_renderer_cls.   s{    ..N=NNNOOO!^M:
HZHHmHHIII&&'?'?:'?'?@@@r&   configr   tokenizer_kwargsc                 X    |                      |          }|                    ||          S )N)r+   from_config)r#   r   r,   r-   renderer_clss        r$   load_rendererzRendererRegistry.load_renderer7   s.     --m<<''0@AAAr&   )__name__
__module____qualname__r   dictr   strtuple__annotations__r%   typer   r+   r   r1    r&   r$   r   r      s          -2E$,G,G,GItCsCx()GGGc 3 C D    As AtL7I A A A ABB B sCx.	B
 
B B B B B Br&   r   c                 *    i | ]\  }\  }}|d | |fS )zvllm.renderers.r:   ).0modemod_relnamecls_names       r$   
<dictcomp>r@   B   sA       )D); 	...9  r&   r,   r   c                     t          | fi |\  }}}}| j        dk    r| j        dk    rd}n|}t                              || i |d|i          S )Nautor   tokenizer_name)r-   )r   tokenizer_mode
model_implRENDERER_REGISTRYr1   )r,   kwargsrD   rC   argsr   s         r$   renderer_from_configrI   J   s    3M4 44 40NND& &&6+<+L+L$&**EFE$4nEE +   r&   N)r,   r   )dataclassesr   r   typingr   r   vllm.loggerr   vllm.tokenizers.registryr   vllm.utils.import_utilsr	   protocolr   vllm.configr   r2   r    _VLLM_RENDERERSr   itemsrF   rI   r:   r&   r$   <module>rS      sb   ) ( ( ( ( ( ( ( % % % % % % % % # # # # # # ? ? ? ? ? ? ; ; ; ; ; ; " " " " " " (''''''	X		 <
'-6  "B "B "B "B "B "B "B "BJ %$ -<-B-B-D-D     .     r&   