\j6dZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZdd lmZmZGd d eej4ZGd d eej4ZddhZeeeddfZdedzdedzdedzfdZ edZ!e!jEdgdejFejHddejJdfde deeeddd fd!eeed"d#d$fd%e d&e d'ed(eddfd)Z&e!jEd*d+d,gejJdfde d-ed'ed(eddf d.Z'e!jEd/gd0ddd1ejJdfde d2eeed3d4fdeedzed5d6fdeedzed7d8fd9ee(ed:d;d<fd'ed(eddfd=Z)e!jEd>d?d@gddejJdfde d-edeedzed5dAfdeedzed7dBfd'ed(eddfdCZ*e!jEdDdEdFgddejJdfde d-edGeeedHfdeedzed5dIfdeedzed7dJfd'ed(eddfdKZ+e!jEdLdMdNgdd1ejJdfde d-ed>eedzedOdPfdQee(edRdSdTfd'ed(eddfdUZ,e!jEdVdWdXgdd1ejJdfde d-ed>eedzedOdYfdQee(edRdSdTfd'ed(eddfdZZ-e!jEd[d\gejJdfde d-ed]eeed^fd'ed(eddf d_Z.e!jEd`dadbgdd1ejJdfde d-ed>eedzedOdcfdQee(edRdSdTfd'ed(eddfddZ/e!jEdedfgejJdfde d-ed'ed(eddf dgZ0y)hzYContains commands to interact with discussions and pull requests on the Hugging Face Hub.N)Path) Annotated) constants) AuthorOptLimitOpt RepoIdArgRepoType RepoTypeOptTokenOpt get_hf_api typer_factory)ArgumentOption)_dataclass_to_dictoutc eZdZdZdZdZdZdZy)DiscussionStatusopenclosedmergeddraftallN)__name__ __module__ __qualname__rrrrrR/root/.hermes/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!/s 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)click BadParameterstrsysstdinread read_text)r&r's r _read_bodyr3Es[ 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_typetokenct|}|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| gddycc} w) z-List discussions and pull requests on a repo.r<rrN)r6r9discussion_typediscussion_statusr;)numtitleis_pull_requestr7r9 created_atrA)headersid_key)r rrrr!rvalueget_repo_discussions_CLIENT_SIDE_STATUSESr7appendlenrrtable) r6r7r8r9r:r;r<api api_statusapi_discussion_type discussionsditemss rdiscussion_listrSSs B 5 !C!&&& #** *   ~!!!""jjK  % %+$// &  <<0 0QXX5M 1 { u $  -8 8q  " 8E 8II U 9sDinfoz'hf discussions info username/my-model 5z5hf discussions info username/my-model 5 --format jsonrAct|}|j|||j}tj|y)z,Get info about a discussion or pull request.r>r6discussion_numr;N)r get_discussion_detailsrGrdictr6rAr;r<rMdetailss rdiscussion_infor\s> 5 !C((//)G HHWrcreate)z)r6rB descriptionr;r#z pull requestr"zrefs/pr/NzCreated z #z on )rAurlref)r3r create_discussionrGrArresultr`) r6rBr&r'r#r;r<r_rMr"r8ras rdiscussion_createrdsXT9-K 5 !C&&//! 'J*>|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.N)Either --body or --body-file is required.r>r6rWrer;zCommented on # in rArepo)r3r,r-r comment_discussionrGrrc)r6rAr&r'r;r<rerMs rdiscussion_commentrmss8y)G  !LMM 5 !C//  JJuD 2'JreditzHhf discussions edit username/my-model 5 abc123 --body "Updated comment."zChf discussions edit username/my-model 5 abc123 --body-file fixed.md comment_idzLThe ID of the comment to edit (see 'hf discussions info ... --format json').z)The new comment text (supports Markdown).z4Read the new comment from a file. Use '-' for stdin.ct||}|tjdt|}|j |||||j t jd|d|d||||y) z9Edit an existing comment on a discussion or pull request.Nrgr>)r6rWro new_contentr;zEdited comment z on #ri)rArkro)r3r,r-r edit_discussion_commentrGrrc) r6rAror&r'r;r<rqrMs rdiscussion_editrssDT9-K  !LMM 5 !C//  JJ E#d7)D#T[hrsrclosez(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 ''?rur>rr6rW new_statusrer;zClosed #rirjNrconfirmr change_discussion_statusrGrcr6rArerur;r<rMs rdiscussion_closerDsp: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 #rwrxryr>rrzz Reopened #rirjNr|rs rdiscussion_reopenrmsp: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>)r6rWrr;z Renamed #ri)rArkrBN)r rename_discussionrGrrc)r6rArr;r<rMs rdiscussion_renamersR& 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 #rwrxryr>rhzMerged #rirjN)rr}r merge_pull_requestrGrcrs 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>rVzNo diff available.N)r rXrGrrtextrZs rdiscussion_diffrsV 5 !C((//)G ||  %&r)1__doc__enumr/pathlibrtypingrr,huggingface_hubr _cli_utilsrrr r r r r r _frameworkrr_outputrrr.Enumrr!rIintDiscussionNumArgr3discussions_clicommandrrmodelrSr\boolrdrmrsrrrrrrrrrsF`  %   ),sDII"S$))""7+ 5 S4ZD4KC$J %WX$  %^^+; ;   G   ;   B   ;$ %;& ';();* +;, -;;| 1?&^^             0   %^^A.r .r   ?   .r  d  7   .r  t  G  !.r.   A  /.r>?.r@ A.rB C.r.rb SC  %^^%K K K  d  8   K t  C  K"#K$ %K& 'KKD RM,  %^^1&t &t &t _  &t  d  <   &t  t  G  !&t./&t0 1&t2 3&t&tR 2S  %^^'E E E d  <  E    ,   E$%E& 'E( )EED 3d  %^^'G G G d  >  G    ,   G$%G& 'G( )GGD C&^^W W W !  WW W W  W. 2O  %^^'E E E d  <  E    ,   E$%E& 'E( )EEB 1&^^ ' ' ''  '  '  'r