o
    
Û¾i{  ã                   @   s8   d dl Z d dlmZmZ d dlmZ G dd„ deƒZdS )é    N)Úadd_cli_argsÚmain)ÚBenchmarkSubcommandBasec                   @   sH   e Zd ZdZdZdZedejddfdd„ƒZ	e
d	ejddfd
d„ƒZdS )ÚBenchmarkServingSubcommandz(The `serve` subcommand for `vllm bench`.Úservez(Benchmark the online serving throughput.ÚparserÚreturnNc                 C   s   t |ƒ d S ©N)r   )Úclsr   © r   úX/home/ubuntu/.local/lib/python3.10/site-packages/vllm/entrypoints/cli/benchmark/serve.pyr      ó   z'BenchmarkServingSubcommand.add_cli_argsÚargsc                 C   s   t | ƒ d S r	   )r   )r   r   r   r   Úcmd   r   zBenchmarkServingSubcommand.cmd)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚnameÚhelpÚclassmethodÚargparseÚArgumentParserr   ÚstaticmethodÚ	Namespacer   r   r   r   r   r   	   s    r   )r   Úvllm.benchmarks.server   r   Ú#vllm.entrypoints.cli.benchmark.baser   r   r   r   r   r   Ú<module>   s   