qiw+d dZddlZddlZddlmZmZmZddlZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZee Zej,deDcic]}||c}eZee Zej,d eDcic]}||c}eZed Zej9d gd ddddej:ddfdeeeej<d fdeeeej<d fdeeeej<d fde de dededdfdZej9ddg dUdeeej@d fdeddfd!Z!ej9d"gd# dVd$eeej@d% fd&eeeej<d' fd(eeeej<d) fd*ee"ej<d+ fd,ee"ej<d- fdeddfd.Z#ej9d/gd0 dWdeeej@d fd$eeeej<d1 fd(eeeej<d2 fd3eee$ej<d4 fd*eee"ej<d5 fd6eeeej<d7 fdeddfd8Z%ej9d9d:d;g dXdeeej@d fdZ&ej9d?gd@ dYdeeej@d fdAeeej@dB fdCeeej@dD fdEeeeej<dF fd,ee"ej<dG fdeddfdHZ'ej9dIdJdKg dZdeeej@d fdLeeej@dM fdEeeeej<dN fd3eee$ej<dO fdeddf dPZ(ej9dQ dXdeeej@d fdLeeej@dR fd?create)z!hf collections create "My Models"z^hf collections create "My Models" --description "A collection of my favorite models" --privatez9hf collections create "Org Collection" --namespace my-orgtitlezThe title of the collection. namespacezMThe namespace (username or organization). Defaults to the authenticated user. descriptionz!A description for the collection.privatezCreate a private collection. exists_okz7Do not raise an error if the collection already exists.ct|}|j|||||}td|jtt j t |dy)z#Create a new collection on the Hub.r")r8r9r:r;r<zCollection created: r.r/N)rcreate_collectionr2urlr3r4r)r8r9r:r;r<rr%r's r)collections_creater@sb: 5 !C&& 'J  0 12 $**' 3A >?r6update)z@hf collections update username/my-collection --title "New Title"zPhf collections update username/my-collection --description "Updated description"zDhf collections update username/my-collection --private --theme greenz!The new title for the collection.z'The new description for the collection.positionz7The new position of the collection in the owner's list.z)Whether the collection should be private.themez;The theme color for the collection (e.g., 'green', 'blue').ct|}|j||||||}td|jtt j t |dy)z*Update a collection's metadata on the Hub.r")r,r8r:rBr;rCzCollection updated: r.r/N)rupdate_collection_metadatar2r?r3r4r) r,r8r:rBr;rCrr%r's r)collections_updaterFsfB 5 !C//' 0J  0 12 $**' 3A >?r6deletez,hf collections delete username/my-collectionz9hf collections delete username/my-collection --missing-ok missing_okz6Do not raise an error if the collection doesn't exist.c^t|}|j||td|y)z!Delete a collection from the Hub.r")rHzCollection deleted: N)rdelete_collectionr2)r,rHrr%s r)collections_deleterKs2 5 !C/jA  1 23r6zadd-item)zGhf collections add-item username/my-collection moonshotai/kimi-k2 modelz`hf collections add-item username/my-collection Qwen/DeepPlanning dataset --note "Useful dataset"zGhf collections add-item username/my-collection Tongyi-MAI/Z-Image spaceitem_idzCThe ID of the item to add (repo_id for repos, paper ID for papers). item_typez?The type of item (model, dataset, space, paper, or collection).notez2A note to attach to the item (max 500 characters).z?Do not raise an error if the item is already in the collection.ct|}|j|||j||}td|tt j t |dy)zAdd an item to a collection.r")r,rLrMrNr<zItem added to collection: r.r/N)radd_collection_itemr#r2r3r4r)r,rLrMrNr<rr%r's r)collections_add_itemrQsb8 5 !C(('//  )J &&7 89 $**' 3A >?r6z update-itemzVhf collections update-item username/my-collection ITEM_OBJECT_ID --note "Updated note"zMhf collections update-item username/my-collection ITEM_OBJECT_ID --position 0item_object_idzTThe ID of the item in the collection (from 'item_object_id' field, not the repo_id).z-A new note for the item (max 500 characters).z/The new position of the item in the collection.cbt|}|j||||td|y)zUpdate an item in a collection.r")r,rRrNrBzItem updated in collection: N)rupdate_collection_itemr2)r,rRrNrBrr%s r)collections_update_itemrUs@0 5 !C'%    ((9 :;r6z delete-itemznThe ID of the item in the collection (retrieved from `item_object_id` field returned by 'hf collections info'.z0Do not raise an error if the item doesn't exist.c`t|}|j|||td|y)z!Delete an item from a collection.r")r,rRrHzItem deleted from collection: N)rdelete_collection_itemr2)r,rRrHrr%s r)collections_delete_itemrX5s= 5 !C'%  *?*; <=r6)N)NNFFN)NNNNNN)FN)NFN)NNN)*__doc__enumr3typingrrrtyperhuggingface_hub.hf_apirr _cli_utilsr r r r r rrrr_COLLECTION_ITEM_TYPESEnumstrr_COLLECTION_SORT_OPTIONSrcollections_clicommandtableOptionr*Argumentr5boolr@intrFrKrQrUrX)tss00r)rls, 00 I   ""67TYY3DZ5[qad5[bef#$45+ @ @  89 ;@  ST V@$ %@& '@@B      -#@sNENN8q$rrs#@   => @ #@   CD F #@  ST V#@ EF H#@$   WX Z %#@, -#@. /#@#@L 6C  4sNENN8q$rrs 4  RS U 4  4  4 4(  #@sNENN8q$rrs@  ^U^^!f gg@ ]^ ` @   NO Q @  [\ ^@" #@$ %@@@`W  <sNENN8q$rrs< rs u<   IJ L  <  KL N< < <<6' >sNENN8q$rrs> B  >  LM O> > >(>w6\.Vs  R( < R-