o
    ॵi3                     @   sv   d dl Z de jfddZdd Zdd Zed	kr9d dlZe d
Zee e Z	ee	Z
eje
e	je	jd dS dS )    Nparserc                 C   sl   | j ddtdd | j ddtdd | j ddd	d
 | j dtddd | j dddd
 | j dtddd d S )Nz
--model_idTzThe target model id)requiredtypehelpz
--revisionzModel revisionz--hostz0.0.0.0zHost to listen)defaultr   z--porti@  zServer port)r   r   r   z--debugdebugzSet debug level.z--llm_firstz%Use LLMPipeline first for llm models.)add_argumentstrintbool)r    r   P/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/server/api_server.pyadd_server_args   s   
r   c              
   C   sb   zdd l }t| }|j|| j| jd W d S  ty0 } zt| td W Y d }~d S d }~ww )Nr   hostporta0  To execute the server command, first install the domain dependencies with: pip install modelscope[DOMAIN] -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html the "DOMAIN" include [cv|nlp|audio|multi-modal|science] and then install server dependencies with: pip install modelscope[server])uvicornget_apprunr   r   ModuleNotFoundErrorprint)argsr   apper   r   r   
run_server   s   r   c                 C   sr   ddl m} ddlm} ddlm}m} |dddddid	}| |j_|	| |
d
|| |
d|| |S )Nr   )FastAPI)
api_router)start_app_handlerstop_app_handlermodelscope_serverz0.1TtryItOutEnabled)titleversionr   swagger_ui_parametersstartupshutdown)fastapir   $modelscope.server.api.routers.routerr   %modelscope.server.core.event_handlersr   r   stater   include_routeradd_event_handler)r   r   r   r   r   r   r   r   r   r   #   s   
r   __main__r   r   )argparseArgumentParserr   r   r   __name__r   r   
parse_argsr   r   r   r   r   r   r   r   r   <module>   s    
