o iO"@sddlZddlZddlmZddlmZmZmZmZddl Z ddl m Z dZ GdddZ Gd d d ZGd d d ZGd ddZGdddZdS)N) expanduser)DictListOptionalTuple) REPO_TYPESzhttps://huggingface.coc@seZdZdZddZdS)RepoObjzl HuggingFace git-based system, data structure that represents a file belonging to the current user. cKs"|D] \}}t|||qdSN)itemssetattr)selfkwargskvrQ/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/huggingface_hub/hf_api.py__init__#szRepoObj.__init__N)__name__ __module__ __qualname____doc__rrrrrr s r c@seZdZdZdefddZdS) ModelFileze Data structure that represents a public file inside a model, accessible from huggingface.co rfilenamecKs(||_|D] \}}t|||qdSr )rr r )r rrrrrrrr-szModelFile.__init__N)rrrrstrrrrrrr(src @sReZdZdZddgddfdeedeedeedeedeeef dd ZdS) ModelInfozB Info about a public model accessible from huggingface.co NmodelIdshatags pipeline_tagsiblingsc KsV||_||_||_||_|durdd|Dnd|_|D] \}}t|||qdS)NcSg|] }tdi|qSr)r.0xrrr Hz&ModelInfo.__init__..)rrrrr r r ) r rrrrr rrrrrrr8s zModelInfo.__init__) rrrrrrrrrrrrrr3s(rc@s<eZdZd#ddZdededefddZd edeeeeffd d Zd eddfd d Z d#de edee fddZ dee fddZ d$dede ed e ede fddZ d#d ede edeefddZ     d%d edede ede ede ede edefdd Z  d$d edede ede efd!d"ZdS)&HfApiNcCs|dur ||_dSt|_dSr )ENDPOINTendpoint)r r*rrrrOszHfApi.__init__usernamepasswordreturncCs8d|j}tj|||dd}||}|dS)z Call HF API to sign in a user and get a token if credentials are valid. Outputs: token if credentials are valid Throws: requests.exceptions.HTTPError if credentials are invalid z {}/api/login)r+r,)jsontoken)formatr*requestspostraise_for_statusr.)r r+r,pathrdrrrloginRs z HfApi.loginr/cCsDd|j}tj|dd|id}||}|d|dfS)z. Call HF API to know "whoami" z {}/api/whoami authorization Bearer {}headersuserorgsr0r*r1getr3r.)r r/r4r5r6rrrwhoami`s z HfApi.whoamicCs0d|j}tj|dd|id}|dS)z) Call HF API to log out. z {}/api/logoutr8r9r:N)r0r*r1r2r3)r r/r4r5rrrlogoutjs  z HfApi.logoutfiltercCsNd|j}|dur|ddnd}tj||d}||}dd|DS)zI Get the public list of all the models on huggingface.co z {}/api/modelsNT)rBfull)paramscSr!r")rr#rrrr&{r'z%HfApi.list_models..r>)r rBr4rDr5r6rrr list_modelsrs zHfApi.list_modelscCstd|S)z Deprecated method name, renamed to `list_models`. Get the public list of all the models on huggingface.co zfThis method has been renamed to `list_models` for consistency and will be removed in a future version.)warningswarnrE)r rrr model_list}szHfApi.model_listrepo_idrevisioncCsp|dur dj|j|dndj|j||d}|dur dd|ind}tj||d}||}td i|S) z Get info on one specific model on huggingface.co Model can be private if you pass an acceptable token. Nz{}/api/models/{repo_id})rIz+{}/api/models/{repo_id}/revision/{revision})rIrJr8r9r:r)r0r*r1r?r3r.r)r rIrJr/r4r;r5r6rrr model_infos zHfApi.model_info organizationcCsXd|j}|durd|ind}tj||dd|id}||}dd|DS) z HuggingFace git-based system, used for models. Call HF API to list all stored files for user (or one of their organizations). z{}/api/repos/lsNrLr8r9)rDr;cSr!r")r r#rrrr&r'z)HfApi.list_repos_objs..r>)r r/rLr4rDr5r6rrrlist_repos_objss zHfApi.list_repos_objsFnameprivate repo_typelfsmultipartthreshc Csd|j}|tvrtd|||d} |dur|| d<|dur$|| d<tj|dd|i| d } |r@| jd kr@| } | d S| | } | d S) a HuggingFace git-based system, used for models. Call HF API to create a whole repo. Params: private: Whether the model repo should be private (requires a paid huggingface.co account) repo_type: Set to "dataset" if creating a dataset, default is model exist_ok: Do not raise an error if repo already exists lfsmultipartthresh: Optional: internal param for testing purposes. Returns: URL to the newly created repo. z{}/api/repos/createInvalid repo type)rNrLrONtyperQr8r9r;r.iurl) r0r*r ValueErrorr1r2 status_coder.r3) r r/rNrLrOrPexist_okrQr4r.r5r6rrr create_repos&   zHfApi.create_repocCs\d|j}|tvrtd||d}|dur||d<tj|dd|i|d}|dS) z HuggingFace git-based system, used for models. Call HF API to delete a whole repo. CAUTION(this is irreversible). z{}/api/repos/deleterR)rNrLNrSr8r9rT)r0r*rrVr1deleter3)r r/rNrLrPr4r.r5rrr delete_repos    zHfApi.delete_repor )NN)NNNFN)rrrrrr7rrr@rArrrErHrKr rMboolintrYr[rrrrr(Nsp       5r(c@s8eZdZedZeddZeddZeddZdS) HfFolderz~/.huggingface/tokencCsTtjtj|jddt|jd}||WddS1s#wYdS)z8 Save token, creating folder as needed. T)rXzw+N)osmakedirsr4dirname path_tokenopenwrite)clsr/frrr save_tokens "zHfFolder.save_tokencCsRzt|jd }|WdWS1swYWdSty(YdSw)z4 Get token or None if not existent. r5N)rcrbreadFileNotFoundError)rerfrrr get_token s( zHfFolder.get_tokencCs(z t|jWdStyYdSw)zD Delete token. Do not fail if token does not exist. N)r_removerbri)rerrr delete_tokens  zHfFolder.delete_tokenN) rrrrrb classmethodrgrjrlrrrrr^s   r^)r_rFos.pathrtypingrrrrr1 constantsrr)r rrr(r^rrrrs    3