rh"TdZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZmZddlZddlmZmZmZddlmZdd lmZd d lmZej6eZeDcgc]}|j<d k7s|j< c}ZGd deZ GddeZ!GddeZ"de e ee#e$fde e#de#fdZ%GddeZ&GddeZ'GddeZ(GddeZ)de e#e#ffdZ*ycc}w) aContains commands to interact with jobs on the Hugging Face Hub. Usage: # run a job hf jobs run # List running or completed jobs hf jobs ps [-a] [-f key=value] [--format TEMPLATE] # Stream logs from a job hf jobs logs # Inspect detailed information about a job hf jobs inspect # Cancel a running job hf jobs cancel N) Namespace_SubParsersAction)asdict)Path)DictListOptionalUnion)HfApi SpaceHardware get_token)logging) load_dotenv)BaseHuggingfaceCLICommandz zero-a10gc"eZdZedefdZy) JobsCommandsparsercv|jddjd}jfdtj |t j |t j |tj |tj |tj |y)NjobszRun and manage Jobs on the Hub.helpz$huggingface.co jobs related commandsc$jSN) print_help)args jobs_parsers k/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/huggingface_hub/cli/jobs.pyz2JobsCommands.register_subcommand..>s;3I3I3Kfunc) add_parseradd_subparsers set_defaultsInspectCommandregister_subcommand LogsCommand PsCommand RunCommand CancelCommand UvCommand)rjobs_subparsersrs @rr'z JobsCommands.register_subcommand8s''5V'W %44:`4a   &K L **?;''8%%o6&&7))/:%%o6r N)__name__ __module__ __qualname__ staticmethodrr'r rrr7s 7$5 7 7r rc>eZdZededdfdZdeddfdZddZy) r*rreturnNcz|jdd}|jdtd|jddd d |jd d d d |jdtd|jdtd|jdtddjtd|jdtd|jdddd |jdtd|jdtd |jd!d"d#$|j t %y)&Nrunz Run a JobrimagezThe Docker image to use.typer-e--envappendz/Set environment variables. E.g. --env ENV=valueactionr-s --secretsvSet secret environment variables. E.g. --secrets SECRET=value or `--secrets HF_TOKEN` to pass your Hugging Face token. --env-file(Read in a file of environment variables.--secrets-file/Read in a file of secret environment variables.--flavorTFlavor for the hardware, as in HF Spaces. Defaults to `cpu-basic`. Possible values: , . --timeoutzVMax duration: int/float with s (seconds, default), m (minutes), h (hours) or d (days).-d--detach store_truez3Run the Job in the background and print the Job ID. --namespaceVThe namespace where the Job will be created. Defaults to the current user's namespace.--tokenIA User Access Token generated from https://huggingface.co/settings/tokenscommand...zThe command to run.nargsrr!)r# add_argumentstrjoinSUGGESTED_FLAVORSr%r*r run_parsers rr'zRunCommand.register_subcommandJs&&u;&? c8RSghEvw  K    3=gh 0sArs ghlhqhqsDiEhFFGH   i    F   i   \   =RSZ0r rc|j|_|j|_i|_|jrc|jj t t |jjtjj|jxsgD]D}|jj t |tjjFi|_ t}|jrG|jj t t |jj||jxsgD](}|jj t ||*|j|_|j|_|j |_|j"|_|j$|_y)Nenviron)r7rRenvenv_fileupdaterr read_textosr^copysecrets_get_extended_environ secrets_fileflavortimeoutdetach namespacetokenselfr env_valueextended_environsecrets r__init__zRunCommand.__init__wsL** "&,, -/ == HHOOKT]](;(E(E(GQSQ[Q[Q`Q`Qbc dR OI HHOOK 2::??;LM N O13 02    LL   D1B1B,C,M,M,OYi j kll(b OF LL   FeZdZededdfdZdeddfdZddZy) r(rr4Nc|jdd}|jdtd|jdtd|jd td |jt y) NlogszFetch the logs of a JobrrxJob IDr8rNQThe namespace where the job is running. Defaults to the current user's namespace.rPrQr!)r#rVrWr%r(rZs rr'zLogsCommand.register_subcommandsy&&v4M&N sB d   C&q   [1r rcj|j|_|j|_|j|_yrrxrkrlrnrs rrrzLogsCommand.__init__s#;; (,$(JJ r ct|j}|j|j|jD] }t |yNrtrxrk)r rlr}rxrkrz)rnrrs rr6zLogsCommand.runs>$**%%%T[[DNN%S C #J r rrr2r rr(r(s; 2$5 2$ 2 2/Y/4/ r r(rowsheadersr4c tg||Dcgc]}td|D}}ttjjt |dz}t |t |z|kDr|jt|}||xxdzcc<t |t |z|kr#|t |z t |z ||z||<t |t |z|kDrdt |zj|}g}|j|j||j|j|Dcgc]}d|z c}|D]r} t| |D cgc];\} } t t| | kDrt| d| dz dzn t| =} } } |j|j| td j|Scc}wcc}wcc} } w) z Inspired by: - stackoverflow.com/a/8356620/593036 - stackoverflow.com/questions/9535954/printing-lists-as-tabular-data c3DK|]}tt|ywr)lenrW).0xs r z_tabulate..s/ac#a&k/s  z{{:{}}} -NrS ) zipmaxrcget_terminal_sizecolumnsrsumindexformatr<rWrX) rrcol col_widthsterminal_widthcol_to_minimize row_formatlineswrowr col_widthrow_format_argss r _tabulaters 9$**3z?;?#)# w<#j/ )^ ;*83z?*JSQX\*Y\fgv\w*wJ ' g,Z (> 9 4*s7|+33Z@J E LL"""G,- LL"""j$AS1W$ABC:!$C 4 9033q6{Y/FCF?Y] #e +CPQF R    &Z&&89 : 99U #P%B sG* G/ 6AG4c^eZdZededdfdZdeddfdZd dZde e e fde fd Z d Z y) r)rr4Nc<|jdd}|jdddd|jd td |jd td |jdddgd|jdtd |jty)Npsz List Jobsrz-az--allrMz*Show all Jobs (default shows just running)r=rNzUThe namespace from where it lists the jobs. Defaults to the current user's namespace.r8rPrQz-fz--filterr<z>Filter output based on conditions provided (format: key=value))r>defaultrz--formatz%Format output using a custom templater!)r#rVrWr%r)rZs rr'zPsCommand.register_subcommands&&t+&>   =   h   \    Q    8  Y/r rcH|j|_|j|_|j|_|j|_i|_|j D]H}d|vr3|j dd\}}||j|j<:td|dJy)N=rz)Warning: Ignoring invalid filter format 'z'. Use key=value format.) allrkrlrfiltersfiltersplitlowerrz)rnrfkeyvalues rrrzPsCommand.__init__s(,$(JJ %)[[ ')  _AaxWWS!_ U,1 SYY[)A!D\]^  _r c  t|j}|j|j}gd}g}|D]}|jr|jj nd}|j s|dvr8|j}|jxsd}|jxsg} | rdj| nd} |jxsd} |||j| d} |j| s|j||| | |g|sad } |jrDd d j|jj!Dcgc] \}}|d |c}}} t#d | y|j%||ycc}}w#t&j($r}t#d|Yd}~yd}~wt*t,t.f$r}t#d|Yd}~yd}~wt0$r.}t#dt3|j4d|Yd}~yd}~wwxYw)z Fetch and display job information for the current user. Uses Docker-style filtering with -f/--filter flag and key=value pairs. rt)rk)zJOB IDz IMAGE/SPACECOMMANDCREATEDSTATUSUNKNOWN)RUNNINGUPDATINGzN/A )r{r7statusrRz matching filters: rHrz No jobs foundNzError fetching jobs data: zError processing jobs data: zUnexpected error - z: )r rl list_jobsrkrstagerr{ docker_imagerRrX created_atr_matches_filtersr<ritemsrz _print_outputrequestsRequestExceptionKeyError ValueError TypeError Exceptionr9r.)rnrr table_headersrrrrximage_or_spacerR command_strrjob_properties filters_msgkves rr6z PsCommand.run s  A Adjj)C==4>>=:DVMD" W-0ZZ))YxxF2I$I"%!1!1!:U+++3:chhw/ !^^4u !+$lln* ",,^< V^[*fUVE" WJ <<$7 Z^ZfZfZlZlZnBoRVRSUVaS!:Bo8p7q"rK k]34   t] 3 Cp(( 4 .qc2 3 3*i0 6 04 5 5 A 'Q(8(8'9A3? @ @ AsHD;F=E: F'F:FH F&&H =G H $HH rcP|jjD]\}}||vryd|vsd|vrQ|jddjdd}tjd|d||tj rcy|j ||j vsyy) z+Check if job matches all specified filters.F*?z.*rI^$T)rrreplaceresearch IGNORECASEr)rnrrpattern regex_patterns rrzPsCommand._matches_filtersRs LL..0 LC.(g~ 'T : B B3 L yy1]O1!5~c7JBMMZ s(;(A(A(CC r c |jr_|j}|D]M}|}tgdD].\}}d|d}||vs|j|t||}0t |Oyt t ||y)z,Print output according to the chosen format.)r{r7rRcreatedrz{{.z}})rN)r enumeraterrWrzr) rnrrtemplaterlineifield placeholders rrzPsCommand._print_outputes ;;{{H  )*Y ZFHAu$)%"5K"d*#||KSVEFd   # r r)r.r/r0r1rr'rrrr6rrWboolrrr2r rr)r)sb 0$5 0$ 0 0D _Y _4 _FAPtCH~$&r r)c>eZdZededdfdZdeddfdZddZy) r&rr4Nc|jdd}|jdtd|jdtd|jd d d |jt y)Ninspectz0Display detailed information on one or more JobsrrNrr8rPrQjob_idsrSzThe jobs to inspectrTr!)r#rVrWr%r&rZs rr'z"InspectCommand.register_subcommand|sz&&y7i&j  d   C&q    =RS^4r rcj|j|_|j|_|j|_yr)rkrlrrs rrrzInspectCommand.__init__s#(,$(JJ "&,, r c $t|j}|jDcgc]}|j||j!}}t t j|Dcgc] }t|c}dtycc}wcc}w)Nrtr)indentr) r rlr inspect_jobrkrzjsondumpsrrW)rnrrxrrs rr6zInspectCommand.runsg$**%W[WcWcdVvHdd djj6#&+6q#NOe6s $B#B rrr2r rr&r&{s< 5$5 5$ 5 5/Y/4/ Pr r&c>eZdZededdfdZdeddfdZddZy) r+rr4Nc|jdd}|jdtd|jdtd|jd td |jt y) Ncancelz Cancel a Jobrrxrr8rNrrPrQr!)r#rVrWr%r+rZs rr'z!CancelCommand.register_subcommandsx&&xn&E sB d   C&q   ]3r rcj|j|_|j|_|j|_yrrrs rrrzCancelCommand.__init__s#;; $(JJ r c~t|j}|j|j|jyr)r rl cancel_jobrxrk)rnrs rr6zCancelCommand.runs($**% dkkT^^Dr rrr2r rr+r+s< 4$5 4$ 4 4/Y/4/ Er r+c8eZdZdZedZdeddfdZddZy) r,z.Run UV scripts on Hugging Face infrastructure.cr|jdd}|jddd}|jdd }|jd d |jd d dg|jdtd|jdd|jdtddj t d|jdddd|jdddd |jd!td"|jd#td$|jd%td&|jd'd(d)d*|jd+td,|jd-td.|jd/dd0d12|jd3d4td5|j t6y7)8zRegister UV run subcommand.uvzERun UV scripts (Python with inline dependencies) on HF infrastructurer uv_commandz UV commandsT)destrrequiredr6z8Run a UV script (local file or URL) on HF infrastructurescriptz$UV script to run (local file or URL) script_argsrSzArguments for the script)rUrrz--imagez.Use a custom Docker image with `uv` installed.r8z--repozCRepository name for the script (creates ephemeral if not specified)rFrGrHrIr:r;r<zEnvironment variablesr=r?r@rArBrCrDrErJz Max duration (e.g., 30s, 5m, 1h)rKrLrMzRun in backgroundrNrOrPzHF tokenz--withz%Run with the given packages installedwith_)r>rrz-pz--pythonz5The Python interpreter to use for the run environmentr!N)r#r$rVrWrXrYr%r,)r uv_parser subparsersr[s rr'zUvCommand.register_subcommands%% X& --r"s& 1..;;)5'   H %,9WDTZZ;=VTZZW7,7$V*Vr+4DeCHo./$s)6a)aHP.P4E-E2q)qhtCH~]Xs %C<:C<