
    -`iS                         d dl Z d dlZd dlmZ d dlmZ d dlmZ ej        rd dl	m
Z
 ne j        Z
 G d de          Zdee         fd	ZdS )
    N)BenchmarkSubcommandBase)CLISubcommand)VLLM_SUBCMD_PARSER_EPILOG)FlexibleArgumentParserc                   x    e Zd ZdZdZdZedej        ddfd            Z	dej        ddfdZ
d	ej        defd
ZdS )BenchmarkSubcommandz(The `bench` subcommand for the vLLM CLI.benchzvLLM bench subcommand.argsreturnNc                 0    |                      |            d S Ndispatch_function)r
   s    w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/entrypoints/cli/benchmark/main.pycmdzBenchmarkSubcommand.cmd   s    t$$$$$    c                     d S r    )selfr
   s     r   validatezBenchmarkSubcommand.validate   s    r   
subparsersc                    |                     | j        | j        | j        d| j         d          }|                    dd          }t	          j                    D ]}|                     |j        |j        |j        d| j         d|j         d          }|                    |j        	           |                    |           t          j
        | j         d|j         
          |_        |S )Nzvllm z <bench_type> [options])helpdescriptionusageT
bench_type)requireddest z
 [options]r   )subcmd)
add_parsernamer   add_subparsersr   __subclasses__set_defaultsr   add_cli_argsr   formatepilog)r   r   bench_parserbench_subparserscmd_clscmd_subparsers         r   subparser_initz"BenchmarkSubcommand.subparser_init   s    ",,I	<$)<<<	 - 
 
 (66<6XX.=?? 	 	G,77\#LBdiBB',BBB	 8  M &&&EEE  ///#<#C)44gl44$ $ $M   r   )__name__
__module____qualname____doc__r"   r   staticmethodargparse	Namespacer   r   _SubParsersActionr   r-   r   r   r   r   r      s        22D#D%($ % % % % \%X/ D    "4	     r   r   r   c                       t                      gS r   )r   r   r   r   cmd_initr7   8   s    !!""r   )r3   typing#vllm.entrypoints.cli.benchmark.baser   vllm.entrypoints.cli.typesr   vllm.entrypoints.utilsr   TYPE_CHECKINGvllm.utils.argparse_utilsr   ArgumentParserr   listr7   r   r   r   <module>r@      s      G G G G G G 4 4 4 4 4 4 < < < < < <	 5@@@@@@@%4$ $ $ $ $- $ $ $N#$}% # # # # # #r   