o
    )i                     @  st   d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm	Z	m
Z
 ejr+d dlmZ G dd deZdddZdS )    )annotationsN)BenchmarkSubcommandBase)CLISubcommand)VLLM_SUBCMD_PARSER_EPILOG)show_filtered_argument_or_group_from_help)FlexibleArgumentParserc                   @  s:   e Zd ZdZdZdZeddd	Zdd
dZdddZ	dS )BenchmarkSubcommandz* The `bench` subcommand for the vLLM CLI. benchzvLLM bench subcommand.argsargparse.NamespacereturnNonec                 C  s   |  |  d S NZdispatch_function)r
    r   o/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/vllm/entrypoints/cli/benchmark/main.pycmd   s   zBenchmarkSubcommand.cmdc                 C  s   d S r   r   )selfr
   r   r   r   validate   s   zBenchmarkSubcommand.validate
subparsersargparse._SubParsersActionr   c                 C  s   |j | j| j| jdd}|jddd}t D ]*}|j |j|j|jd|j dd}|j|jd || t	|d	|jg t
|_q|S )
Nz!vllm bench <bench_type> [options])helpdescriptionusageTZ
bench_type)requireddestzvllm bench z
 [options]r   r	   )
add_parsernamer   add_subparsersr   __subclasses__set_defaultsr   Zadd_cli_argsr   r   epilog)r   r   Zbench_parserZbench_subparsersZcmd_clsZcmd_subparserr   r   r   subparser_init   s.   
z"BenchmarkSubcommand.subparser_initN)r
   r   r   r   )r   r   r   r   )
__name__
__module____qualname____doc__r   r   staticmethodr   r   r"   r   r   r   r   r      s    
r   r   list[CLISubcommand]c                   C  s   t  gS r   )r   r   r   r   r   cmd_init9   s   r)   )r   r(   )
__future__r   argparsetypingZ#vllm.entrypoints.cli.benchmark.baser   Zvllm.entrypoints.cli.typesr   Zvllm.entrypoints.utilsr   r   TYPE_CHECKINGZ
vllm.utilsr   r   r)   r   r   r   r   <module>   s   '