o
    پi                     @   st   d dl mZ d dlmZ d dlmZ d dlmZ dee fddZ	dee fdd	Z
dddZedkr8e  d
S d
S )    )CLISubcommand)GenerateSubcommand)ServeSubcommand)FlexibleArgumentParserreturnc                   C   s   t  t gS )N)r   r    r   r   f/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/runtime/entrypoints/cli/main.pygenerate_cmd_init   s   r	   c                  C   s   g } |  t  | S )z-Initialize all commands from separate modules)extendr	   )commandsr   r   r   cmd_init   s   r   Nc                  C   s   t dd} | jddddd | jdd	d
}i }t D ]}||j|jd |||j< q|  }|j	|v r=||j	 
| t|drI|| d S |   d S )Nzsglang-diffusion CLI)descriptionz-vz	--versionversionz0.1.0)actionr   F	subparser)requireddest)dispatch_functionr   )r   add_argumentadd_subparsersr   subparser_initset_defaultscmdname
parse_argsr   validatehasattrr   
print_help)parser
subparserscmdsr   argsr   r   r   main   s   



r"   __main__)r   N)7sglang.multimodal_gen.runtime.entrypoints.cli.cli_typesr   6sglang.multimodal_gen.runtime.entrypoints.cli.generater   3sglang.multimodal_gen.runtime.entrypoints.cli.server   sglang.multimodal_gen.utilsr   listr	   r   r"   __name__r   r   r   r   <module>   s   

