
    .`i                        d dl mZ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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)Response)ProfilerConfig)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/profile/api_router.pyr   r      s    ;**    z/start_profileraw_requestc                    K   t                               d           t          |                                            d {V  t                               d           t	          d          S )NzStarting profiler...zProfiler started.   status_code)loggerinfor   start_profiler   r   s    r   r   r      si      
KK&'''

$
$
2
2
4
44444444
KK#$$$$$$$r   z/stop_profilec                    K   t                               d           t          |                                            d {V  t                               d           t	          d          S )NzStopping profiler...zProfiler stopped.r   r   )r   r   r   stop_profiler   r   s    r   r   r      si      
KK&'''

$
$
1
1
3
33333333
KK#$$$$$$$r   r   c                     t          | j        j        dd           }|t          |t                    sJ |C|j        >t                              d|j                   |                     t                     d S d S d S )Nprofiler_configzeProfiler with mode '%s' is enabled in the API server. This should ONLY be used for local development!)
getattrr   args
isinstancer   profilerr   warning_onceinclude_routerrouter)r   r   s     r   attach_routerr&   %   s    cin.?FFO"j.&Q&Q""Q"'?'KJ$	
 	
 	

 	6""""" #"'K'Kr   N)fastapir   r   r   fastapi.responsesr   vllm.configr   vllm.engine.protocolr   vllm.loggerr	   __name__r   r%   r   postr   r   r&    r   r   <module>r/      s7  
 0 / / / / / / / / / & & & & & & & & & & & & - - - - - - # # # # # #	X			+7 +| + + + + %W % % % % _%G % % % %	#w 	# 	# 	# 	# 	# 	#r   