o
    Û·iÆ  ã                   @   s&   d dl Z d dlmZ G dd„ dƒZdS )é    N)ÚOrchestratorAggregatorc                   @   s*   e Zd ZdZddedejdB fdd„ZdS )ÚClientRequestStatez>Tracks the state of an individual request in the orchestrator.NÚ
request_idÚqueuec                 C   s,   || _ d | _|d ur|nt ¡ | _d | _d S ©N)r   Ústage_idÚasyncioÚQueuer   Úmetrics)Úselfr   r   © r   ú`/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/entrypoints/client_request_state.pyÚ__init__	   s   
zClientRequestState.__init__r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ústrr   r	   r   r   r   r   r   r      s    r   )r   Úvllm_omni.metricsr   r   r   r   r   r   Ú<module>   s    