i, dZddlZddlmZmZddlZddlmZmZddl m Z m Z m Z m Z mZmZddlmZmZeeZej(deDcic]}||c}e ZeeZej(d eDcic]}||c}e Zed Zej5d gdddddej6dfdeedzej8d fdeedzej8d fdeedzej8d fde de de ddfdZej5ddgej6dfdeeej<d fde de ddfd Zej5d!gd"ddd#d#ej6dfd$eeej<d% fd&eedzej8d' fd(eedzej8d) fd*ee ej8d+ fd,ee ej8d- fde de ddfd.Z!ej5d/gd0dddddej6dfdeeej<d fd$eedzej8d1 fd(eedzej8d2 fd3ee"dzej8d4 fd*ee dzej8d5 fd6eedzej8d7 fde de ddfd8Z#ej5d9d:d;gd#ej6dfdeeej<d fdZ$ej5d?gd@dd#ej6dfdeeej<d fdAeeej<dB fdCeeej<dD fdEeedzej8dF fd,ee ej8dG fde de ddfdHZ%ej5dIdJdKgddej6dfdeeej<d fdLeeej<dM fdEeedzej8dN fd3ee"dzej8dO fde de ddfdPZ&ej5dQd#ej6dfdeeej<d fdLeeej<dR fdzCollection updatedr6N)r update_collection_metadatarr:r7r8) r)r0r2r=r3r>rrr"r$s r&collections_updaterAsQD 5 !C//' 0JJJ#*//z~~Nr.deletez,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.cpt|}|j||tjd|y)z!Delete a collection from the Hub.r)rCzCollection deletedr7N)r delete_collectionrr:)r)rCrrr"s r&collections_deleterGs0" 5 !C/jAJJ#/:r.zadd-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_typezGThe type of item (model, dataset, space, paper, collection, or bucket).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||}tjd||j y)zAdd an item to a collection.r)r)rHrIrJr4zItem added to collectionr6N)r add_collection_itemrrr:r8) r)rHrIrJr4rrr"r$s r&collections_add_itemrMsM: 5 !C(('//  )JJJ)Z^^Tr.z 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.ctt|}|j||||tjd|y)zUpdate an item in a collection.r)r)rNrJr=zItem updated in collectionrEN)r update_collection_itemrr:)r)rNrJr=rrr"s r&collections_update_itemrQ s>2 5 !C'%   JJ+/Br.z 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.crt|}|j|||tjd|y)z!Delete an item from a collection.r)r)rNrCzItem deleted from collectionrEN)r delete_collection_itemrr:)r)rNrCrrr"s r&collections_delete_itemrT-s;" 5 !C'% JJ-ODr.)(__doc__enumtypingrrtyperhuggingface_hub.hf_apirr _cli_utilsrr r r r r _outputrr_COLLECTION_ITEM_TYPESEnumstrr_COLLECTION_SORT_OPTIONSrcollections_clicommandautoOptionr'Argumentr-boolr;intrArGrMrQrT)tss00r&risB, & Ill.""67TYY3DZ5[qad5[bef#$45+ @ O  89 ;O  ST VO$ %O& 'O( )OOB      4 9 9/#OsNENN8q$rrs#O  d  => @ #O  d  CD F #O d  ST V#O t  EF H#O$  d  WX Z %#O, -#O. /#O0 1#O#OL 6C  4 9 9 ;sNENN8q$rrs ;  RS U ;  ;  ;  ; ;(   4 9 9%UsNENN8q$rrsU  ^U^^!f ggU ef h U  d  NO Q U  [\ ^U" #U$ %U& 'UU@`W  4 9 9CsNENN8q$rrsC rs uC  d  IJ L  C d  KL NC C C  !CC8'  4 9 9EsNENN8q$rrsE B  E  LM OE E E E(E{6\.Vs S< 8 T