o d۷i @sddlZddlZddlmZddlmZmZddlZddlm Z m Z m Z ddl m Z ddlmZmZddlmZddlmZmZd d lmZmZeeZGd d d eZddedefddZdefddZ dde dedBdefddZ!ej" ddededBdedfddZ#dS)N) Generator)AnyLiteral) HTTPErrorHTTPStatusError ReadTimeout)Segment) RichToolkitRichToolkitTheme)Progress) MinimalStyle TaggedStyle)Identitydelete_auth_configc sjeZdZddeffdd Z   ddeeefded ed e d dBd e e e eff fd d Z ZS) FastAPIStyle tag_widthcstj|ddS)Nr)super__init__)selfr __class__Q/home/ubuntu/vllm_env/lib/python3.10/site-packages/fastapi_cloud_cli/utils/cli.pyrszFastAPIStyle.__init__FNmetadata is_animateddoneanimation_status)startedstoppederrorreturncst|s tj||||dSgd}||jt|}|r#|d|d}|dkr)d}|jd}td|}t|g|fS) N)r)u🥚u🐣u🐤u🐥u🐓u🐔 done_emojir"u🟡rr)r_get_tag_segmentsanimation_counterlengetrmaxr)rrrrremojistag left_paddingrrrr&s   zFastAPIStyle._get_tag_segments)r)FFN)__name__ __module__ __qualname__intrdictstrrboolrtuplelistrr& __classcell__rrrrrs   rFminimalr#c Cs>|rtntdd}t|ddddddd d d d d }t|dS)Nrrzwhite on #009485zwhite on #007166grey62whitez#007166grey85z on #007166redzindian_red italic) z tag.titler, placeholdertextselectedresultprogressr" cancelled)styletheme)rD)r rr r )r8rCrDrrrget_rich_toolkit9s rEcCs2d}t}|jdkrt|d7}|S|d7}|S)Nz"The specified token is not valid. userz,Use `fastapi login` to generate a new token.zMake sure to use a valid token.)r auth_moder)messageidentityrrrhandle_unauthorizedNs rJr"default_messagecCsbd}t|tr&|jj}|dkrt|jn|dkr t}n|dkr&d}|s/|p.d|}|S)Niiiz,You don't have permissions for this resourcezZSomething went wrong while contacting the FastAPI Cloud server. Please try again later. ) isinstancerresponse status_codeloggerdebugjsonrJ)r"rKrHrNrrrhandle_http_error]s rRrA)NNNc cszdVWdSty#}zt||dtddd}~wtyC}zt|t||}||tddd}~ww)NzJThe request to the FastAPI Cloud server timed out. Please try again later.r)rrOrP set_errortyperExitrrR)rArKerHrrrhandle_http_errorsvs"       rW)F)N)$ contextlibloggingcollections.abcrtypingrrrThttpxrrr rich.segmentr rich_toolkitr r rich_toolkit.progressr rich_toolkit.stylesr r authrr getLoggerr.rOrr4rEr3rJrRcontextmanagerrWrrrrs2    '