o
    iC                     @   s   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eZe Zdedefdd	Zejd
eddedefddZdd ZdS )    )	APIRouterRequest)Response)EngineClient)init_logger)EngineDeadErrorrequestreturnc                 C   s
   | j jjS N)appstateengine_client)r    r   b/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/entrypoints/serve/instrumentator/health.pyr      s   
r   z/health)response_classraw_requestc                    s>   zt |  I dH  tddW S  ty   tdd Y S w )zHealth check.N   )status_codei  )r   check_healthr   r   )r   r   r   r   health   s   r   c                 C   s   |  t d S r
   )include_routerrouter)r   r   r   r   attach_router    s   r   N)fastapir   r   fastapi.responsesr   vllm.engine.protocolr   vllm.loggerr   vllm.v1.engine.exceptionsr   __name__loggerr   r   getr   r   r   r   r   r   <module>   s   	