}tjcdZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddl mZddlmZddlmZmZddlZddlmZmZmZdd lmZmZd d lmZd d lmZm Z d d l!m"Z"dZ#edZ$dZ%dZ&ee&Z'dZ(dZ)dZ*ej+e,Z-Gdde.eZ/e GddZ0e'1dgd dkdej2dee.ed fd!ee3e d"d#fd$dfd%Z4e'1d&gd' dld(ee.dzed)fd$dfd*Z5e'1d+d,dd,d-d.d/g0dej2d(ee.ed1fd$dfd2Z6e'1d3d4gdmd5Z7e'1d6d7gdmd8Z8e'1d9d:gd(ee.ed;fd$dfd<Z9d$e:e0fd=Z;d$e:eZ=d?e:e.d@e>e.d$e?dzfdAZ@dBe.d$edzfdCZAd(e.d$e0fdDZBdEe0d$e/fdFZCdddGdHe.dIe.dBe.dJe.dKe.dzdLe.dzd$e0fdMZDd,dNdHe.dIe.dJe.dOe3d$e.dzf dPZEdHe.dIe.dJe.dBe.d$eFf dQZGdRedBe.dSeFd$efdTZHdRedHe.dIe.dBe.dJe.d$ef dUZIdHe.dIe.dJe.dVe.dzd$e.dzf dWZJdBe.d$efdXZKdddYdZe.d[eLdzd\eLdzfd]ZMdHe.dIe.d$ez*ExtensionManifest.load..Ps,]]]zsESCD\=\=\U=\=\=\r$r0r#) MANIFEST_FILENAMEis_filer jsonloads read_textitemsr fromisoformatr')r;r3 manifest_pathdatas` r%loadzExtensionManifest.loadIs00 $$&& ljjjjkk kz-113344]]]]TZZ\\]]]'5d>6JKK^ ((4(((r$c|tz }|jddt|}|j|d<|tj|dddS)NTparentsexist_okr0)indent sort_keys) r=parentmkdirrr0 isoformat write_textr?dumps)selfr3rDrEs r%savezExtensionManifest.saveTsx00 ""4$"???d||#0::<<^  Dd!K!K!KLLLLLr$) rrrstr__annotations__r rr1r2 classmethodrrFrTr#r$r%r'r'=s JJJ III LLLOOO $ %%%%"Kt"""!Jd !!!))!4)))[)MM$MMMMMMr$r'install)zhf extensions install hf-claudez-hf extensions install hanouticelina/hf-claudez(hf extensions install alvarobartt/hf-mem)examplesFctxr*z:GitHub extension repository in `[OWNER/]hf-` format.forcez--forcezOverwrite if already installed.r4ct|\}}}|}tt|jdi}||vrt d|d|dt|}|r|st d|dt||\} } |rtj |t|||| | } | j } tj| d |d |d | tjd |dS)zInstall an extension from a public GitHub repository. Security warning: this installs a third-party executable or Python package. Install only from sources you trust. commandszCannot install extension 'z,' because it conflicts with an existing `hf z ` command. Extension 'z1' is already installed. Use --force to overwrite.r( repo_namer(r`r+branchr1z extension installed/hf )sourcecommandzRun it with: hf N)_normalize_repo_id find_rootsetgetattrrfkeysr _get_extension_direxists_fetch_github_repo_infoshutilrmtree_install_extensionr/ capitalizerresulthint) rZr*r[r(r`r+root_ctxreserved_commands extension_dirrbr1manifestext_types r%extension_installrz\s*$6g#>#> E9j}}HGH$4j"EEJJLLMM&&& w w wak w w w   'z22MdedbZbbbccc1SSSFK% m$$$!yZ\gH}''))HJ )))%%)%%"j"" H , , ,-----r$update)zhf extensions updatezhf extensions update hf-claudez'hf extensions update alvarobartt/hf-memnamezExtension to update (with or without `hf-` prefix, optionally as `OWNER/hf-`). If omitted, all installed extensions are checked and the outdated ones are updated.c  |t|}t|}|xtjkr#t jd|j|jnFxtjkr&t jd|jd|jntj kr dSt}|s*t j dt j ddSg}g}|D]}t j d |jd |jd  t|}n?#t$r2}t j d |jd |jd |dYd}~od}~wwxYw|xtjkr||jxtjkr||jtj krt jd|rd|nd|rd|nddS)z6Update installed extension(s) to their latest version.NzExtension updated)r|rer^z' is already up to dater|zNo extensions installed.z1Install one with: hf extensions install z Checking '' (z)...zCould not update '): . Skipping.zExtensions update completez, )rr)$_load_installed_extension_for_update_update_installed_extensionrr rrsr+r*r!r"_list_installed_extensionswarningrtlog Exceptionappendjoin)r|rx update_status manifestsrrerrors r%extension_updatersa( 7==3H== /'//// .X5HQYQabbbbb2'2222 U)<UUU\d\oppppp'///*,,I  ./// DEEEGJ KX0KKX5EKKKLLL 7AAMM    KlX-@llXEUllZ_lll m m m HHHH /'////x233332'2222!!("56666'///0J$&-7 '"""4,6@499Z(((DsD E(E  EexecT)allow_extra_argsallow_interspersed_argsignore_unknown_optionsz#hf extensions exec claude -- --helpz/hf extensions exec claude --model zai-org/GLM-5)context_settingsrYz.Extension name (with or without `hf-` prefix).ct|}t|}|std|dt |t |j}tj |)zExecute an installed extension.r^' is not installed.r,args)code) _normalize_extension_name"_resolve_installed_executable_pathr>r _execute_extension_binarylistrclick exceptionsExit)rZr|r+r, exit_codes r%extension_execrs +400J8DDO  " " $ $FDZDDDEEE)/PTUXU]P^P^___I    Y  / //r$z list | lszhf extensions listcbdtD}tj|ddS)z"List installed extension commands.cg|]S}d|jt|jt|j|jd|jdTS)rdz%Y-%m-%d)rfrer/ installedr1)r+rUr*r/r0strftimer1)r8rxs r% z"extension_list..st     3X022(*++ &&!.77 CC#/      r$rfid_keyN)rrtable)rowss r%extension_listrsE  344   DId9%%%%%%r$searchzhf extensions searchc tddtdddd}|}dtD}g}|d gD]m}|d d }|||d |d d|dpd||vrdnddntj|ddS)zISearch extensions available on GitHub (tagged with 'hf-extension' topic).z*https://api.github.com/search/repositoriesztopic:starsdescd)qsortorderper_page)paramsch|] }|j Sr#r+)r8ms r% z#extension_search..sDDD!DDDr$rBr|hf- full_namestargazers_countrr1yes)r|r)rr1rr)rN) _github_get_EXTENSIONS_GITHUB_TOPICr?rget removeprefixrrr)responserErrr)r+s r%extension_searchrs48688'TZhkllH ==??DDD'A'C'CDDDI D"%%   &\..u55  "[)"4a88#xx 66<"&0I&=&=UU2       Id6""""""r$z remove | rmzhf extensions remove claudez8Extension name to remove (with or without `hf-` prefix).ct|}t|}|std|dt j|t jd|dS)zRemove an installed extension.r^rzExtension removedr~N)rrlis_dirr rorprrs)r|r+rws r%extension_removersy+400J&z22M    ! !FDZDDDEEE M-   J"444444r$ct}|sgSg}t|D]}|r|jds1 |t |`#t$r/}t d|jd|Yd}~d}~wwxYw|S)zPReturn manifests for all validly-installed extensions, sorted by directory name.rz'Failed to load manifest for extension '': N) EXTENSIONS_ROOT expanduserrsortediterdirr| startswithrr'rFrloggerdebug)root_dirrrwes r%rr%s))++H ??   I 0 0 2 233 ##%% ]-?-J-J5-Q-Q      .33MBB C C C C    LL]=CU]]Z[]] ^ ^ ^ HHHH  s-B33 C,=%C''C,cg}tD]>}d|jd}|j |jd|n|}||j|f?|S)Nz [extension ] )rr*r1rr+)entriesrxtag help_texts r%"list_installed_extensions_for_helpr6suG.0099/H,///7?7K7Wx+33c333]` +Y78888 Nr$rknown_commandsc|sdS|d}|drdSd|D}||vrdS t|d|}n#t$rYdSwxYw t |}n#t $rt |}YnwxYw||sdSt|t|ddS)Nr-cfh|].}|dD]}|/S)|)splitstrip)r8cmdas r%rz7dispatch_unknown_top_level_extension..Fs7MMMsciinnMMMMMMr$roriginal_inputrr) r_validate_extension_short_namerr rr _auto_install_official_extensionr>rr)rr command_name all_knownr+r,s r%$dispatch_unknown_top_level_extensionr?s) t7Ls##tMMnMMMIy  t3L4M4Me4T4Teqrrr ttG> Z Z ZZs#$A A*)A*.A>>BBr+c td|}}t|rdS t||\}}n#t$rYdSwxYw t jd|d|d|dn#t$rYdSwxYw t|||||}t|j  S#t$rYdSwxYw) zKTry to auto-install huggingface/hf-. Returns executable path or None.rNr_'z)' is an official Hugging Face extension (rcz). Install it?ra) DEFAULT_EXTENSION_OWNERrlrmrnrrconfirmr rqrr,r)r+r(r`rbr1rxs r%rr[s2.0Bj0B0B9E*%%,,..t5EYWWW tt n nnUnnU^nnnoooo tt%9F`k   H,--88::: tts5A AAA77 BB 9C CCc6t|}t|}|sN|d\}}}|r|d|nd|}t d|d|t |S)Nrcz/hf-rr^zA' is not installed. Install it first with: hf extensions install )rrlrr rpartitionr r'rF)r|r+rwr(_install_targets r%rrss*400J&z22M    ! ! jjll--c22 q!7<TE33z333BT BTBT w* w wgu w w     ! !- 0 00r$rxc |j|j|j}}} t||\}}nD#t$r7}t jd|d|d|d|d tjcYd}~Sd}~wwxYwt|||d}|*t jd|d|d|d tjS||j kr tj St|||||| tj S) Nr_zCould not check updates for 'rrcrrF)r(r`rbwarnz#): GitHub is unreachable. Skipping.)r(r`r+rbr1r2)r(r)r+rnrrrrr"_fetch_latest_commit_shar2r!rqr )rxr(r`r+rbr1r latest_shas r%rrsH#+>8=(BUj9E.5EYWWW ... kJkk5kk9kkY^kkklll%-------.*SY`efffJ qJ q q5 q q9 q q q   &--X(((%00  " ))s, A-,A("A-(A-)r1r2r(r`rbr1r2ct|}| }d} t||||} n#t$rd} YnwxYw| t ||| } nt |||||} t |||d||t| | dndtj tj t|||| |pt|||  } | |d }| |s|rtj|d SSS#t"$rt$j$r$} t)dt*d|d|d| d} ~ wt$j$r$} t)d|d|d| jd| d} ~ wt$r} t)d|d|d| | d} ~ wwxYw#|s|rtj|d wwwxYw)aFetch and install an extension (binary or Python package), then persist its manifest. Installs in place: an existing install is overwritten without being removed first, so a failed update keeps the previous version working. A failed fresh install is cleaned up entirely. F)r(r`rbr+N)rwr+r-)rwr(r`r+rbrcr-r.)r(r`rbcandidate_description)r(r`rb) r(r)r*r+r,r/r0r1r2T) ignore_errorszPip install timed out after zs for 'z$'. See pip output above for details.z$Failed to install pip package from 'z ' (exit code z$). See pip output above for details.z"Failed to install extension from 'r)rlrm_fetch_remote_binaryr_install_binary_extension_install_python_extensionr'rUrnowrutc_try_fetch_remote_descriptionrrTrorpr subprocessTimeoutExpiredr _EXTENSIONS_PIP_INSTALL_TIMEOUTCalledProcessError returncode) r(r`r+rbr1r2rw fresh_installrr-r,rxrs r%rqrqs'z22M%,,...MI/= )SYfpqqqFF   FFF   7+ 6OO8+5IZdmsO%**y**!00#/X!hl335yWb"n%=EU^gm%n%n%n      m$$$   =] = M-t < < < < < = =   $& 0+J 0 0SX 0 0[d 0 0 0     (& 05 0 09 0 0STS_ 0 0 0    nnn&'eE'e'eT]'e'ebc'e'efflmmn =] = M-t < < < < < = =sW=D A  D A  B1DF.3EF.$F F.F))F..F11G)rrc  td|d|d|ddi}|jpdS#t$r-}|r t jd|d |d|d |Yd}~dSd}~wwxYw) zZBest-effort fetch of the latest commit SHA for a branch, used to detect available updates.https://api.github.com/repos/rcz /commits/Acceptzapplication/vnd.github.sha)headersNz'Could not fetch latest commit SHA for 'rr)rtextrrrr)r(r`rbrrrs r%rrs  PE P PI P P P P;<   }""$$,,   o Km)mmPUmmXammfkmm n n nttttts69 A0"A++A0cht|}d|d|d|d|}t|}|jS)N"https://raw.githubusercontent.com/rc /refs/heads/)_get_executable_namercontent)r(r`rbr+executable_nameraw_urlrs r%rrsL*:66Ol5ll9llRXll[jllG7##H  r$rwr-c|dd|t|z }||tjdkrtj|d|S)NTrHnti)rOr  write_bytesosr|chmod)rwr+r-r,s r%rrset444#&::&F&FFO''' w$ %((( r$c d|d|d|d}|dz }t|tjdkrdnd}tjd }t j} |s| d || d d |r'tj |dt|gd n5tjd t|| d|| d||r|ddddt||g} nt|dddddd|g} tj | d t"| d|t%|} t|| } | st'd|d|d| d| S)Nzhttps://github.com/rcz/archive/refs/heads/z.zipvenvrz python.exer.uvz Creating virtual environment in TrH)check)with_pipzVirtual environment created in zInstalling package from piprXz --reinstallz--pythonz-mz--disable-pip-version-checkz --no-inputz--force-reinstall)rtimeoutzPackage installed from zInstalled package from 'z/' does not expose the required console script ''.)_get_venv_bin_pathrr|rowhichrstatusr>r{rOrrunrUr EnvBuildercreatedonerr r resolve) rwr(r`r+rb source_urlvenv_dir venv_pythonuv_pathr install_cmdrvenv_executables r%rrsF[uZZyZZfZZZJv%H$Xrw$||T\]]Kl4  G Z\\F    B CCCDDDD4888  A NGVS]];4 H H H H H OT * * * 1 1#h-- @ @ @ @h@@AAA MM9Z99:::  y-SQ\M]M]_ij        )     N;d4STTTT KK6*66777*:66O(?CCO  " " $ $  $u $ $y $ $ $ $ $     " " $ $$r$rcDd|d|d|} t|dt}|d}t |t r|Sn#t $rYnwxYw t|d}|jD]m}| }| drB| d\}}} | dcSnn#t $rYnwxYw|S)zTry to fetch project description either from: - manifest.json - pyproject.toml Only best effort, no error handling. r rcr r1z/pyproject.toml=z"') rr=r?r isinstancerUrr splitlinesrr partition) r(r`rbrbaserr1linerr:s r%rr.sc X W W W Wv W WD $<<):<<==mmoo))-88 k3 ' '           $77788M,,.. 2 2D::<|stdtfddDrtd|ddvsdvrtd|dtstd|dS) NzExtension name cannot be empty.c3 K|]}|vV dSr=r#)r8sepr|s r% z1_validate_extension_short_name..s' . .33$; . . . . . .r$)rc\zInvalid extension name 'rz..:z9'. Allowed characters: letters, digits, '.', '_' and '-'.)rr any_ALLOWED_EXTENSION_NAME fullmatch)r+rr|s @r%rrs     D :8999 . . . .+ . . ...FD.DDDEEE t||sd{{D.DDDEEE " , ,T 2 2  p~ p p p    Kr$cd|vrtd|d}t|dkrt}|d}n;t|dkrt |r|\}}ntd|d|d std |d|d }|std t|| |||fS) Nz://zEOnly GitHub repositories in `[OWNER/]hf-` format are supported.rcrrrKz%Expected `[OWNER/]REPO` format, got 'rrz6Extension repository name must start with 'hf-', got 'z(Invalid extension repository name 'hf-'.r)r rlenrallrrr)r*partsr(r`r+s r%rgrgs ^___ MM#  E 5zzQ'!H UqSZZ yyJwJJJKKK    & &_]PY]]]^^^''..J CABBB":gFFFF )Z ''r$c|ddd}t|d|S)Nrcrrr)rrsplitrr)r|r`s r%rrsF ##C++B/I ))*@*@*G*GX\ ] ] ]]r$r,c tjt|g|zS#t$rV}tjdks|jt jkrtjdt|g|zcYd}~Sd}~wwxYw)Nrsh)rcallrUOSErrorrr|errnoENOEXEC)r,rrs r%rrsDO 4 45<=== DDD 7d??ag66 c/&:&:;dBCCCCCCCCDs$' BA B<BB)Fr=)r4N)X__doc__rUr?rrerorr dataclassesrrrrenumrpathlibrtypingr r rhuggingface_hub.errorsr r r huggingface_hub.utilsrr _cli_utilsr _frameworkrr_outputrrrr=EXTENSIONS_HELPextensions_clirr6r get_loggerrrrUrr'rfContextboolrzrrrrrrrtuplerriintrrrrrqrbytesrrrrrldictrrnr rrcompilerHrrgrrr#r$r%rlsE 76  ))))))))''''''''%%%%%%%% XXXXXXXXXX66666666%%%%%%((((((((($566#+ O444)!"%  H % %S$  MMMMMMM M< Y^ &.&. &.  RSSS U&. T66)2STTTT U &. &.&.&.&.R " 33  d f     3 3333l *.5lpqq-90 0  FGGG I 0  0000$ /C.DEE & & &FE &,B+CDD###ED#4 1N0OPP5  PQQQ S 5  555QP5(D):$;"DsCx,A[tCy[#c([WZ]aWa[[[[80 1s 17H 1 1 1 1**;*@V****H#!@=@=@= @=@= @=  @= t @=d @=@=@=@=@=FW[   s s C t _bei_i    33QTY^#uY]+%+%#&+%36+%DG+%QT+% +%+%+%+%\"! "!"!(+"!DG$J"!4Z"!"!"!"!J=3=4====48PT   S TD[ $+     ;c;c;eCtO>T;;;;SS73747777 ....... %"*%DEE s s s    ((c3m(<((((0^C^C^^^^ DtD49DDDDDDDr$