(ti32 dZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZGd d eej2ZGd d eej2Zd dhZeeej<ddfZdedzdedzdedzfdZ edZ!e!jEdgdejFejHddejJejLdfdedeeejNdddfd eeejNd!d"d#fd$e d%e d&ed'e d(eddfd)Z(e!jEd*d+d,gejJejLdfded-ed&ed'e d(eddf d.Z)e!jEd/gd0ddd1ejJejLdfded2eeejNd3d4fdeedzejNd5d6fdeedzejNd7d8fd9ee*ejNd:d;d<fd&ed'e d(eddfd=Z+e!jEd>d?d@gddejJejLdfded-edeedzejNd5dAfdeedzejNd7dBfd&ed'e d(eddfdCZ,e!jEdDdEdFgdd1ejJejLdfded-ed>eedzejNdGdHfdIee*ejNdJdKdLfd&ed'e d(eddfdMZ-e!jEdNdOdPgdd1ejJejLdfded-ed>eedzejNdGdQfdIee*ejNdJdKdLfd&ed'e d(eddfdRZ.e!jEdSdTgejJejLdfded-edUeeej<dVfd&ed'e d(eddfdWZ/e!jEdXdYdZgdd1ejJejLdfded-ed>eedzejNdGd[fdIee*ejNdJdKdLfd&ed'e d(eddfd\Z0e!jEd]d^gejJejLdfded-ed&ed'e d(eddf d_Z1y)`zYContains commands to interact with discussions and pull requests on the Hugging Face Hub.N)Path) Annotated) constants) AuthorOptFormatWithAutoOptLimitOpt RepoIdArgRepoType RepoTypeOptTokenOptapi_object_to_dict get_hf_api typer_factory)OutputFormatWithAutooutc eZdZdZdZdZdZdZy)DiscussionStatusopenclosedmergeddraftallN)__name__ __module__ __qualname__rrrrrn/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/huggingface_hub/cli/discussions.pyrr(s D F F E CrrceZdZdZdZdZy)DiscussionKindr discussion pull_requestN)rrrrr"r#rrrr!r!0s CJ!Lrr!rrz&The discussion or pull request number.)helpminbody body_filereturnc||tjd|>t|dk(rtjj S|j dS|S)zFResolve body text from --body or --body-file (supports '-' for stdin).z'Cannot use both --body and --body-file.-zutf-8)encoding)typer BadParameterstrsysstdinread read_text)r&r's r _read_bodyr3Fs[ I1  !JKK y>S 99>># #""G"44 Krz0Manage discussions and pull requests on the Hub.)r$z list | ls)z%hf discussions list username/my-modelzIhf discussions list username/my-model --kind pull_request --status mergedzFhf discussions list username/my-dataset --type dataset --status closedzBhf discussions list username/my-model --author alice --format json)examplesrepo_idstatusz-sz--statusz4Filter by status (open, closed, merged, draft, all).kindz-kz--kindz/Filter by kind (discussion, pull_request, all).authorlimit repo_typeformattokenct|}|tjk(rd} n|tjk(rd} nd} |tj k(rd} n |j } g} |j||| | |j D]O} |j tvr| j|j k7r/| j| t| |k\sOn| D cgc] } t| } } tj| gddddi ycc} w) z-List discussions and pull requests on a repo.r=rrN)r6r9discussion_typediscussion_statusr;)numtitleis_pull_requestr7r9 created_atrBright)headersid_key alignments)rrrrr!rvalueget_repo_discussions_CLIENT_SIDE_STATUSESr7appendlenrrtable)r6r7r8r9r:r;r<r=api api_statusapi_discussion_type discussionsditemss rdiscussion_listrVTs D 5 !C!&&& #** *   ~!!!""jjK  % %+$// & <<0 0QXX5M 1 { u $ -8 8Kq  "KE 8II U7#  9sDinfoz'hf discussions info username/my-model 5z5hf discussions info username/my-model 5 --format jsonrBct|}|j|||j}tj|y)z,Get info about a discussion or pull request.r?r6discussion_numr;N)rget_discussion_detailsrJrdictr6rBr;r<r=rPdetailss rdiscussion_infor_s> 5 !C((//)G HHWrcreate)z|D)5HZ^^$ %4CJJ$r*..!1gY?Z^^YcYgYgmpqrcommentzIhf discussions comment username/my-model 5 --body "Thanks for reporting!"z9hf discussions comment username/my-model 5 --body "LGTM!"z%The comment text (supports Markdown).z0Read the comment from a file. Use '-' for stdin.ct||}|tjdt|}|j ||||j t jd|d|||y)z(Comment on a discussion or pull request.Nz)Either --body or --body-file is required.r?r6rZrhr;zCommented on # in rBrepo)r3r,r-rcomment_discussionrJrrf) r6rBr&r'r;r<r=rhrPs rdiscussion_commentross:y)G  !LMM 5 !C//  JJuD 2'Jrclosez(hf discussions close username/my-model 5zIhf discussions close username/my-model 5 --comment "Closing as resolved."z --commentz)An optional comment to post when closing.yesz--yesz-yzSkip confirmation prompt.ctjd|d|d|t|}|j||d||jtj d|d ||| y ) z#Close a discussion or pull request.zClose # on ''?rqr?rr6rZ new_statusrhr;zClosed #rkrlNrconfirmrchange_discussion_statusrJrfr6rBrhrqr;r<r=rPs rdiscussion_closer|sp<KK'#eG9B/S9 5 !C  // !JJ#d7),#GDrreopenz)hf discussions reopen username/my-model 5zZhf discussions reopen username/my-model 5 --comment "Reopening for further investigation."z+An optional comment to post when reopening.ctjd|d|d|t|}|j||d||jtj d|d ||| y ) z+Reopen a closed discussion or pull request.zReopen #rsrtrur?rrvz Reopened #rkrlNrxr{s rdiscussion_reopenrDsp<KK(3%uWIR0c: 5 !C  // !JJC5WI.CgFrrenamez9hf discussions rename username/my-model 5 "Updated title" new_titlezThe new title.ct|}|j||||jtjd|d||||y)z$Rename a discussion or pull request.r?)r6rZrr;z Renamed #rk)rBrmrCN)rrename_discussionrJrrf)r6rBrr;r<r=rPs rdiscussion_renamernsR( 5 !C//  JJ3%tG9-3WIVrmergez(hf discussions merge username/my-model 5zEhf discussions merge username/my-model 5 --comment "Merging, thanks!"z)An optional comment to post when merging.ctjd|d|d|t|}|j||||jtj d|d||| y ) zMerge a pull request.zMerge #rsrtrur?rjzMerged #rkrlN)rryrmerge_pull_requestrJrfr{s rdiscussion_mergersm<KK'#eG9B/S9 5 !C//  JJ#d7),#GDrdiffz'hf discussions diff username/my-model 5ct|}|j|||j}|jr t j |jyt j dy)z Show the diff of a pull request.r?rYzNo diff available.N)rr[rJrrtextr]s rdiscussion_diffrsV 5 !C((//)G ||  %&r)2__doc__enumr/pathlibrtypingrr,huggingface_hubr _cli_utilsrrr r r r r rrr_outputrrr.Enumrr!rLintArgumentDiscussionNumArgr3discussions_clicommandrrmodelautoOptionrVr_boolrgror|rrrrrrrrsc`  %   /sDII"S$))""7+ENN 5 S4ZD4KC$J %WX$  %^^ 4 9 9-= =   G   =   B   =$ %=& '=()=* +=, -=. /==@ 1?&^^ 4 9 9         " 0   %^^ 4 9 9C/r /r   ?   /r  d  7   /r  t  G  !/r.   A  //r>?/r@ A/rB C/rD E/r/rd SC  %^^ 4 9 9' K  K  K  d  8    K t  C   K"# K$ % K& ' K( ) K KF 2S  %^^ 4 9 9) E  E  E d  <   E    ,    E$% E& ' E( ) E* + E EF 3d  %^^ 4 9 9) G  G  G d  >   G    ,    G$% G& ' G( ) G* + G GF C&^^ 4 9 9W W W !  WW W W W  W0 2O  %^^ 4 9 9)E E E d  <  E    ,   E$%E& 'E( )E* +EED 1&^^ 4 9 9 ' ' ''  '  '  '  'r