
    .`i                     Z   d dl mZ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  ee          Zdede
fdZ e            Ze                    d	          d
efd            Ze                    d          d
efd            Ze                    d          d
efd            ZdefdZdS )    )	APIRouterFastAPIRequest)JSONResponseResponseN)EngineClient)init_loggerrequestreturnc                 $    | j         j        j        S N)appstateengine_client)r
   s    {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/entrypoints/serve/sleep/api_router.pyr   r      s    ;**    z/sleepraw_requestc                    K   | j                             dd          }t          |                               t	          |                     d {V  t          d          S )Nlevel1   status_code)query_paramsgetr   sleepintr   )r   r   s     r   r   r      sg       $((#66E

$
$
*
*3u::
6
66666666 $$$$r   z/wake_upc                    K   | j                             d          }|g k    rd }t                              d|           t	          |                               |           d {V  t          d          S )Ntagsz wake up the engine with tags: %sr   r   )r   getlistloggerinfor   wake_upr   )r   r   s     r   r#   r#       s      #++F33Drzz
KK2D999

$
$
,
,T
2
22222222 $$$$r   z/is_sleepingc                    K   t                               d           t          |                                            d {V }t	          d|i          S )Nz$check whether the engine is sleepingis_sleeping)content)r!   r"   r   r%   r   )r   r%   s     r   r%   r%   -   sZ      
KK6777%k22>>@@@@@@@@K <====r   r   c                 V    t           j        sd S |                     t                     d S r   )envsVLLM_SERVER_DEV_MODEinclude_routerrouter)r   s    r   attach_routerr,   4   s,    $ vr   )fastapir   r   r   fastapi.responsesr   r   	vllm.envsr(   vllm.engine.protocolr   vllm.loggerr	   __name__r!   r   r+   postr   r#   r   r%   r,    r   r   <module>r5      sr  
 0 / / / / / / / / / 4 4 4 4 4 4 4 4       - - - - - - # # # # # #	X		+7 +| + + + + 
 X%W % % % % Z	%w 	% 	% 	% 	% N>7 > > > >w      r   