o ~ri&@sdZddlZddlZddlmZmZmZmZddlZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZeee dZejd d d eDed Zee Zejd dd eDed Zdeedee fddZeddZejdgddej ddfdedededdfddZ!ejddgd dRd eeej"d!dfdeddfd"d#Z#ejd$gd%d     dSd&eeeej$d'd(dfd)eeeej$d*dfd+eeeej$d,d-dfd.eeeeej$d/d0dfd1eeeej$d2dfdeddfd3d4Z%ejd5gd6d     dSd eeej"d7dfd)eeeej$d8dfd&eeeeej$d'd9dfd.eeeeej$d/d:dfd1eeeej$d;dfdeddfdd?gd dRd eeej"d@dfdeddfdAdBZ'ejdCdDgd dRd eeej"dEdfdeddfdFdGZ(ejdHdIdJgd  dTd eeej"dKdfdLee)ej$dMdNdOdfdeddfdPdQZ*dS)UaContains commands to manage webhooks on the Hugging Face Hub. Usage: # list all webhooks hf webhooks ls # show details of a single webhook hf webhooks info # create a new webhook hf webhooks create --url https://example.com/hook --watch model:bert-base-uncased # create a webhook watching multiple items and domains hf webhooks create --url https://example.com/hook --watch org:HuggingFace --watch model:gpt2 --domain repo # update a webhook hf webhooks update --url https://new-url.com/hook # enable / disable a webhook hf webhooks enable hf webhooks disable # delete a webhook hf webhooks delete N) AnnotatedOptionalget_argsget_type_hints)WEBHOOK_DOMAIN_T)WebhookWatchedItem) FormatOpt OutputFormatQuietOptTokenOptapi_object_to_dict get_hf_apiprint_list_output typer_factorytypeWatchedItemTypecCi|]}||qSr).0trrj/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/huggingface_hub/cli/webhooks.py ?r)r WebhookDomaincCrrrrdrrrrBrvaluesreturnc Csg}tt}|D]9}d|vrtd|dd|d|dd\}}||vr8td|dd|d|t||dq|S) aYParse 'type:name' strings into WebhookWatchedItem objects. Args: values: List of strings in the format 'type:name' (e.g., 'model:bert-base-uncased', 'org:HuggingFace'). Returns: List of WebhookWatchedItem objects. Raises: typer.BadParameter: If any value doesn't match the expected format. :zCExpected format 'type:name' (e.g. 'model:bert-base-uncased'), got 'z'. Valid types: , .rzInvalid type ')rname)tuple_WATCHED_TYPEStyper BadParameterjoinsplitappendr)ritems valid_typesvkindr"rrr _parse_watchEs r.zManage webhooks on the Hub.)helpz list | ls)zhf webhooks lszhf webhooks ls --format jsonzhf webhooks ls -q)examplesFformatquiettokencCs:t|d}dd|D}t|||gdddddS) z'List all webhooks for the current user.r3cSsg|]}t|qSr)r rwrrr sszwebhooks_ls..)idurldisableddomainswatchedc SsV|dd|dp dt|ddd|dpgdd d |d p&gDgS) Nr8r9z(job)r:Fr r;css8|]}t|tr|dd|dnt|VqdS)rrr"N) isinstancedictstrr5rrr ~s( z0webhooks_ls....r<)getr@r')itemrrrys    zwebhooks_ls..)r1r2headersrow_fnN)r list_webhooksr)r1r2r3apiresultsrrr webhooks_lsds  rJinfozhf webhooks info abc123 webhook_idzThe ID of the webhook.cCs.t|d}||}ttjt|dddS)z/Show full details for a single webhook as JSON.r4indentN)r get_webhookprintjsondumpsr rLr3rHwebhookrrr webhooks_infos rVcreate)zQhf webhooks create --url https://example.com/hook --watch model:bert-base-uncasedzjhf webhooks create --url https://example.com/hook --watch org:HuggingFace --watch model:gpt2 --domain repozJhf webhooks create --job-id 687f911eaea852de79c4a50a --watch user:julien-cwatchz--watchzRItem to watch, in 'type:name' format (e.g. 'model:bert-base-uncased'). Repeatable.r9zBURL to send webhook payloads to. Mutually exclusive with --job-id.job_idz--job-idz]ID of a Job to trigger (from job.id) instead of pinging a URL. Mutually exclusive with --url.domainz--domainzNDomain to watch: 'repo' or 'discussions'. Repeatable. Defaults to all domains.secretz.Optional secret used to sign webhook payloads.c Cs|dur |dur td|dur|durtdt|d}t|}|r,dd|Dnd}|j|||||d} td| jttjt | d d dS) z|Create a new webhook. Provide either --url (to ping a remote server) or --job-id (to trigger a Job), but not both. Nz+Provide either --url or --job-id, not both.z!Provide either --url or --job-id.r4cSg|]}|jqSrvaluerrrrr7rz#webhooks_create..)r9rYr<r;r[zWebhook created: rMrN) r%r&rr.create_webhookrQr8rRrSr ) rXr9rYrZr[r3rH watched_itemsr;rUrrrwebhooks_creates,   raupdate)z8hf webhooks update abc123 --url https://new-url.com/hookz:hf webhooks update abc123 --watch model:gpt2 --domain repoz,hf webhooks update abc123 --secret newsecretz The ID of the webhook to update.z$New URL to send webhook payloads to.ziNew list of items to watch, in 'type:name' format. Repeatable. Replaces the entire existing watched list.zBNew list of domains to watch: 'repo' or 'discussions'. Repeatable.z)New secret used to sign webhook payloads.c Csnt|d}|r t|nd}|rdd|Dnd}|j|||||d} td| jttjt| dddS) z>Update an existing webhook. Only provided options are changed.r4NcSr\rr]rrrrr7rz#webhooks_update..)r9r<r;r[zWebhook updated: rMrN)rr.update_webhookrQr8rRrSr ) rLr9rXrZr[r3rHr`r;rUrrrwebhooks_updates &rdenablezhf webhooks enable abc123z The ID of the webhook to enable.cC(t|d}||}td|jdS)zEnable a disabled webhook.r4zWebhook enabled: N)renable_webhookrQr8rTrrrwebhooks_enable rhdisablezhf webhooks disable abc123z!The ID of the webhook to disable.cCrf)zDisable an active webhook.r4zWebhook disabled: N)rdisable_webhookrQr8rTrrrwebhooks_disable rirldeletezhf webhooks delete abc123zhf webhooks delete abc123 --yesz The ID of the webhook to delete.yesz--yesz-yzSkip confirmation prompt.cCsP|std|d}|stdtt|d}||td|dS)zDelete a webhook permanently.z)Are you sure you want to delete webhook 'z'?zAborted.r4zWebhook deleted: N)r%confirmrQAbortrdelete_webhook)rLrnr3rorHrrrwebhooks_deletes  rr)N)NNNNN)FN)+__doc__enumrRtypingrrrrr%huggingface_hub.constantsrhuggingface_hub.hf_apir _cli_utilsr r r r r rrrr$Enumr@r _DOMAIN_TYPESrlistr. webhooks_clicommandtablerJArgumentrVOptionrardrhrlboolrrrrrrs   (           0       &