o ~ri@sRdZddlZddlZddlZddlmZmZmZddlZddl m Z m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZee Zejdd d eDed Zd eed eefddZeddZejdgddddddddej ddf deeeej!dedfdeeeej!ddfdeeeej!ddfdeeeej!ddfd eeeej!d!dfd"ed#ed$ed%ed dfd&d'Z"ejd(gd)dd*ej ddfd+eeej#d,dfd"ed#ed$ed%ed df d-d.Z$ejd/d0gd d9d1eeej#d2dfd%ed dfd3d4Z%ejd5d6gd d9d1eeej#d2dfd%ed dfd7d8Z&dS):aContains commands to interact with papers on the Hugging Face Hub. Usage: # list daily papers (most recently submitted) hf papers ls # list trending papers hf papers ls --sort=trending # list papers from a specific date, ordered by upvotes hf papers ls --date=2025-01-23 # list today's papers, ordered by upvotes hf papers ls --date=today # list papers from a specific week hf papers ls --week=2025-W09 # list papers by a specific submitter hf papers ls --submitter=someuser # search papers hf papers search "vision language" # get info about a paper hf papers info 2502.08025 # read a paper as markdown hf papers read 2502.08025 N) AnnotatedOptionalget_args)CLIErrorHfHubHTTPError)DailyPapersSort_T) FormatOptLimitOpt OutputFormatQuietOptTokenOpt _format_cellapi_object_to_dict get_hf_apiprint_list_output typer_factory PaperSortEnumcCsi|]}||qSr).0srrh/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/huggingface_hub/cli/papers.py Fsr)typevaluereturncCs*|durdS|dkrtjS|S)z6Parse date option, converting 'today' to current date.Ntoday)lowerdatetimedater isoformat)rrrr _parse_dateIs  r!z Interact with papers on the Hub.)helpz list | ls)z hf papers lszhf papers ls --sort trendingzhf papers ls --date 2025-01-23zhf papers ls --week 2025-W09z hf papers ls --submitter akhaliqzhf papers ls --format json)examples2Frz+Date in ISO format (YYYY-MM-DD) or 'today'.)r"callbackweekz'ISO week to filter by, e.g. '2025-W09'.monthz;Month to filter by in ISO format (YYYY-MM), e.g. '2025-02'. submitterz$Filter by username of the submitter.sortz Sort results.limitformatquiettokenc  Csvt|d} |r |jnd} dd| j||||| |dD} gd} dtdttfd d } t| ||d | | d d d ddS)zList daily papers on the Hub.r-NcSg|]}t|qSrrr paper_inforrr szpapers_ls..)rr&r'r(r)r*)idtitleupvotescomments published_at submitted_byitemrc Ssr|dpi}|dp|dpd}|ddt|ddddt|d dt|d dt|d d|gS) Nr9fullnameusernamer4r5<max_lenr6r7r8getrstr)r:r9submitter_namerrr _paper_rows zpapers_ls.._paper_rowr4right)r6r7r+r,id_keyheadersrow_fn alignments)rrlist_daily_papersdictlistrCr)rr&r'r(r)r*r+r,r-apisort_keyresults_HEADERSrErrr papers_lsUs. )   rSsearch)z"hf papers search "vision language"z1hf papers search "attention mechanism" --limit 10z*hf papers search "diffusion" --format jsonqueryzSearch query string.c Cs^t|d}dd|j||dD}gd}dtdttfdd }t|||d ||d d id dS)zSearch papers on the Hub.r.cSr/rr0r1rrrr3sz!papers_search..)rVr*)r4r5r6r8r:rcSs<|ddt|ddddt|ddt|ddgS)Nr4r=r5Fr?r6r8rA)r:rrrrEs z!papers_search.._paper_rowr4r6rFrGN)r list_papersrMrNrCr) rVr*r+r,r-rOrQrRrErrr papers_searchs  rYinfozhf papers info 2601.15621paper_idz'The arXiv paper ID (e.g. '2502.08025').c Csnt|d}z|j|d}Wnty)}z|jjdkr$td|d|d}~wwttjt |dddS) zs  0       D