Ziw+6 dZddlZddlZddlmZmZmZddlZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZee ZejddeDe Zee Zejd d eDe Zed Zedgdddddejddfdeeeejd fdeeeejd fdeeeejd fde de dededdfdZedd g dWd!eeej d" fdeddfd#Z!ed$gd% dXd&eeej d' fd(eeeejd) fd*eeeejd+ fd,ee"ejd- fd.ee"ejd/ fdeddfd0Z#ed1gd2 dYd!eeej d" fd&eeeejd3 fd*eeeejd4 fd5eee$ejd6 fd,eee"ejd7 fd8eeeejd9 fdeddfd:Z%ed;dee"ejd? fdeddfd@Z&edAgdB d[d!eeej d" fdCeeej dD fdEeeej dF fdGeeeejdH fd.ee"ejdI fdeddfdJZ'edKdLdMg d\d!eeej d" fdNeeej dO fdGeeeejdP fd5eee$ejdQ fdeddf dRZ(edS dZd!eeej d" fdNeeej dT fd>ee"ejdU fdeddf dVZ)dS)]a_Contains commands to interact with collections on the Hugging Face Hub. Usage: # list collections on the Hub hf collections ls # list collections for a specific user hf collections ls --owner username # get info about a collection hf collections info username/collection-slug # create a new collection hf collections create "My Collection" --description "A collection of models" # add an item to a collection hf collections add-item username/collection-slug username/model-name model # delete a collection hf collections delete username/collection-slug N) AnnotatedOptionalget_args)CollectionItemType_TCollectionSort_T) FormatOptLimitOpt OutputFormatQuietOptTokenOptapi_object_to_dict get_hf_apiprint_list_output typer_factoryCollectionItemTypeci|]}||Sr).0ts Z/root/projects/butler/venv/lib/python3.11/site-packages/huggingface_hub/cli/collections.py r;s5[5[5[qa5[5[5[)typeCollectionSortci|]}||Srr)rss rrr>s-U-U-Uqa-U-U-Urz%Interact with collections on the Hub.)helpls)zhf collections lsz hf collections ls --owner nvidiazLhf collections ls --item models/teknium/OpenHermes-2.5-Mistral-7B --limit 10)examples Fownerz)Filter by owner username or organization.itemzkFilter collections containing a specific item (e.g., "models/gpt2", "datasets/squad", "papers/2311.12983").sortz4Sort results by last modified, trending, or upvotes.limitformatquiettokenreturnct|}|r|jnd}d|||||D} t| ||dS)zList collections on the Hub.r(Nc,g|]}t|Sr)r)r collections r z"collections_ls..cs0  :&&r)r"r#r$r%)r&r')rvaluelist_collectionsr) r"r#r$r%r&r'r(apisort_keyresultss rcollections_lsr4Ds: 5 ! ! !C!+tzztH.. /  GgfE::::::rinfoz/hf collections info username/my-collection-slugcollection_slugz7The collection slug (e.g., 'username/collection-slug').ct|}||}ttjt |ddS)z'Get info about a collection on the Hub.r+indentN)rget_collectionprintjsondumpsr)r6r(r1r-s rcollections_infor?osS 5 ! ! !C##O44J $*' 33A > > >?????rcreate)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|}||||||}td|jtt jt |ddS)z#Create a new collection on the Hub.r+)rArBrCrDrEzCollection created: r8r9N)rcreate_collectionr<urlr=r>r)rArBrCrDrEr(r1r-s rcollections_createrIs: 5 ! ! !C&& 'J 1 1 1222 $*' 33A > > >?????rupdate)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|}|||||||}td|jtt jt |ddS)z*Update a collection's metadata on the Hub.r+)r6rArCrKrDrLzCollection updated: r8r9N)rupdate_collection_metadatar<rHr=r>r) r6rArCrKrDrLr(r1r-s rcollections_updaterOsB 5 ! ! !C//' 0J 1 1 1222 $*' 33A > > >?????rdeletez,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.cxt|}|||td|dS)z!Delete a collection from the Hub.r+)rQzCollection deleted: N)rdelete_collectionr<)r6rQr(r1s rcollections_deleterTsJ 5 ! ! !C/jAAA 2 2 233333rzadd-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||}td|tt jt |ddS)zAdd an item to a collection.r+)r6rUrVrWrEzItem added to collection: r8r9N)radd_collection_itemr/r<r=r>r)r6rUrVrWrEr(r1r-s rcollections_add_itemrZs8 5 ! ! !C(('/  )J 8 8 8999 $*' 33A > > >?????rz 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.c|t|}|||||td|dS)zUpdate an item in a collection.r+)r6r[rWrKzItem updated in collection: N)rupdate_collection_itemr<)r6r[rWrKr(r1s rcollections_update_itemr^s\0 5 ! ! !C'%    : : :;;;;;rz 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.czt|}||||td|dS)z!Delete an item from a collection.r+)r6r[rQzItem deleted from collection: N)rdelete_collection_itemr<)r6r[rQr(r1s rcollections_delete_itemra5sY 5 ! ! !C'%  rss , 0000000000 IIIIIIII                      ""677TY35[5[DZ5[5[5[befff#8$455+-U-U>> @ @  8999 ;@  STTT V@$ %@& '@@@@B      -#@#@sNEN8q$r$r$rrs#@   =>>> @ #@   CDDD F #@  STTT V#@ EFFF H#@$   WXXX Z %#@, -#@. /#@#@#@#@L 6C  4 4sNEN8q$r$r$rrs 4  RSSS U 4  4  4 4 4 4(  #@@sNEN8q$r$r$rrs@  ^U^!f g g gg@ ]^^^ ` @   NOOO Q @  [\\\ ^@" #@$ %@@@@@`W  <<sNEN8q$r$r$rrs< rsss u<   IJJJ L  <  KLLL N< < <<<<6'' >>sNEN8q$r$r$rrs> B    >  LMMM O> > >>>('>>>r