o
    ۷i                     @   s2   d Z ddlZddlZdd Zedkre  dS dS )z>
CLI entry point for vLLM-Omni that intercepts vLLM commands.
    Nc                  C   s(  dt jvrddlm}  |   dS ddlm}m} ddlm} ddl	}ddl
}|jjj|jjjjg}|  |d|jdd	d
}|jdddtjdd |jddd}i }|D ]}	|	 }
|
D ]}||j|jd |||j< q^qV| }|j|v r||j | t|dr|| dS |  dS )z3Main CLI entry point that intercepts vLLM commands.z--omnir   )mainN)VLLM_SUBCMD_PARSER_EPILOGcli_env_setup)FlexibleArgumentParserzvLLM OMNI CLIz[subcommand])subcmd)descriptionepilogz-vz	--versionversion	vllm_omni)actionr	   F	subparser)requireddest)dispatch_functionr   ) sysargvvllm.entrypoints.cli.mainr   vllm.entrypoints.utilsr   r   vllm.utils.argparse_utilsr   (vllm_omni.entrypoints.cli.benchmark.mainvllm_omni.entrypoints.cli.serveentrypointscliserve	benchmarkformatadd_argument	importlibmetadatar	   add_subparserscmd_initsubparser_initset_defaultscmdname
parse_argsr   validatehasattrr   
print_help)	vllm_mainr   r   r   r
   CMD_MODULESparser
subparserscmds
cmd_modulenew_cmdsr#   args r1   T/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/entrypoints/cli/main.pyr   	   sH   





r   __main__)__doc__importlib.metadatar   r   r   __name__r1   r1   r1   r2   <module>   s    1
