o
    -im                     @   sJ   d dl mZ d dlmZ d dlmZ d dlmZ eeG dd dZdS )    )Literal)Field)	dataclass)configc                   @   s   e Zd ZU dZdZeed< 	 eddZe	d ed< 	 dZ
eed	< 	 dZedB ed
< 	 dZeed< 	 dZeed< 	 dZeed< 	 dZeed< 	 dd ZdS )KVEventsConfigz&Configuration for KV event publishing.Fenable_kv_cache_eventsN)default)nullzmq	publisherztcp://*:5557endpointreplay_endpointi'  buffer_stepsi hwmmax_queue_size topicc                 C   s"   | j d u r| jr
dnd| _ d S d S )Nr
   r	   )r   r   )self r   R/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/config/kv_events.py__post_init__6   s   
zKVEventsConfig.__post_init__)__name__
__module____qualname____doc__r   bool__annotations__r   r   r   r   strr   r   intr   r   r   r   r   r   r   r   r      s&   
 r   N)	typingr   pydanticr   pydantic.dataclassesr   vllm.config.utilsr   r   r   r   r   r   <module>   s   