g= U d Z ddlZddlZddlZddlmZ ddlmZmZm Z m
Z
m Z ddl m
Z
mZmZ er
ddlZddlmZ e rddlmZmZ dd lmZ ndxZxZZdae d
ed < ej d du Ze
G d
d
Z e
dde e de fd Z de fdZ de e ddfdZ de de de fdZ!de de de fdZ"dS )zSContains `WebhooksServer` and `webhook_endpoint` to create a webhook server easily. N)wraps)
TYPE_CHECKINGAnyCallableDictOptional ) experimentalis_fastapi_availableis_gradio_available)Request)FastAPIr
) JSONResponseWebhooksServer _global_appSPACE_IDc e Zd ZdZd fd Z dded dee ddfdZdd ee defd
Z dd e
d
e ddfdZ ddZ
xZS )r aB
The [`WebhooksServer`] class lets you create an instance of a Gradio app that can receive Huggingface webhooks.
These webhooks can be registered using the [`~WebhooksServer.add_webhook`] decorator. Webhook endpoints are added to
the app as a POST endpoint to the FastAPI router. Once all the webhooks are registered, the `launch` method has to be
called to start the app.
It is recommended to accept [`WebhookPayload`] as the first argument of the webhook function. It is a Pydantic
model that contains all the information about the webhook event. The data will be parsed automatically for you.
Check out the [webhooks guide](../guides/webhooks_server) for a step-by-step tutorial on how to setup your
WebhooksServer and deploy it on a Space.
`WebhooksServer` is experimental. Its API is subject to change in the future.
You must have `gradio` installed to use `WebhooksServer` (`pip install --upgrade gradio`).
Args:
ui (`gradio.Blocks`, optional):
A Gradio UI instance to be used as the Space landing page. If `None`, a UI displaying instructions
about the configured webhooks is created.
webhook_secret (`str`, optional):
A secret key to verify incoming webhook requests. You can set this value to any secret you want as long as
you also configure it in your [webhooks settings panel](https://huggingface.co/settings/webhooks). You
can also set this value as the `WEBHOOK_SECRET` environment variable. If no secret is provided, the
webhook endpoints are opened without any security.
Example:
```python
import gradio as gr
from huggingface_hub import WebhooksServer, WebhookPayload
with gr.Blocks() as ui:
...
app = WebhooksServer(ui=ui, webhook_secret="my_secret_key")
@app.add_webhook("/say_hello")
async def hello(payload: WebhookPayload):
return {"message": "hello"}
app.launch()
```
returnc t st d t st d t | S )NzjYou must have `gradio` installed to use `WebhooksServer`. Please run `pip install --upgrade gradio` first.zlYou must have `fastapi` installed to use `WebhooksServer`. Please run `pip install --upgrade fastapi` first.)r ImportErrorr super__new__)clsargskwargs __class__s \/var/www/html/ai-engine/env/lib/python3.11/site-packages/huggingface_hub/_webhooks_server.pyr zWebhooksServer.__new__a sg "$$
$%%
wws### Nui gr.Blockswebhook_secretc || _ |pt j d | _ i | _ t | j d S )NWEBHOOK_SECRET)_uiosgetenvr! registered_webhooks_warn_on_empty_secret)selfr r! s r __init__zWebhooksServer.__init__n sB
, K :J0K0K8: d 122222r pathc t r S t t j fd }|S )ax
Decorator to add a webhook to the [`WebhooksServer`] server.
Args:
path (`str`, optional):
The URL path to register the webhook function. If not provided, the function name will be used as the
path. In any case, all webhooks are registered under `/webhooks`.
Raises:
ValueError: If the provided path is already registered as a webhook.
Example:
```python
from huggingface_hub import WebhooksServer, WebhookPayload
app = WebhooksServer()
@app.add_webhook
async def trigger_training(payload: WebhookPayload):
if payload.repo.type == "dataset" and payload.event.action == "update":
# Trigger a training job if a dataset is updated
...
app.launch()
```
c | d }dp|j d }|j v rt d| d |j |< d S )Nr z
/webhooks//zWebhook z already exists.)__name__stripr'
ValueError)r r funcabs_pathr+ r) s r _inner_postz/WebhooksServer.add_webhook.._inner_post si 7DHT%:T]$A$A#$F$FHHH4333 !FH!F!F!FGGG15D $X . . .r )callable add_webhookr r post)r) r+ r4 s `` r r6 zWebhooksServer.add_webhooky sm 8 D>> ,%4##%%d++ +
w| 6 6 6 6 6
6 r Fprevent_thread_lock
launch_kwargsc | j p| }| dt |j d
ddi|\ | _ }}| j D ]E\ }}| j t || j } | j
| | Ft j
d }|d|z n
|j p|j d d }|d
d
fd | j D z z
}|d z
}t% | |s| dS dS )zLaunch the Gradio app and register webhooks to the underlying FastAPI server.
Input parameters are forwarded to Gradio when launching the app.
sharer8 TNr!
SPACE_HOSTzhttps://r. z/
Webhooks are correctly setup and ready to use:
c 3 & K | ] }d | V dS )z - POST N ).0webhookurls r z(WebhooksServer.launch.. s1 #g#g7$>$>W$>$>#g#g#g#g#g#gr zG
Go to https://huggingface.co/settings/webhooks to setup your webhooks.r@ )r$ _get_default_ui
setdefault _is_locallaunch fastapi_appr' itemsr! _wrap_webhook_to_check_secretr7 r% environget share_url local_urlr0 joinprint block_thread)
r) r8 r9 r _r+ r2
space_hostmessagerC s
@r rH zWebhooksServer.launch s|
X
/--//
)444!*!U!Ut!U}!U!U !Q 288:: . .JD$".4T$J]^^^
(D ! !$ ' ' - - - - Z^^L11
)3)?j:%%blFbVXVbiinnD4$))#g#g#g#gdNf#g#g#ggggg]]
g" OO r c ddl } |j 5 } |j d |j d |j t | j ddz d d | j D z |j t rd nd
ddd n # 1 swxY w Y |S ) zLDefault UI if not provided (lists webhooks and provides basic instructions).r Nu) # This is an app to process 🤗 WebhooksaT Webhooks are a foundation for MLOps-related features. They allow you to listen for new changes on specific repos or to all repos belonging to particular set of users/organizations (not just your repos, but any repo). Check out this [guide](https://huggingface.co/docs/hub/webhooks) to get to know more about webhooks on the Huggingface Hub.z webhook(s) are registered:z
z
c 3 T K | ]#\ }}d | dt |j | dV $dS )z- [z]()N)_get_webhook_doc_urlr/ )rA webhook_pathrB s r rD z1WebhooksServer._get_default_ui.. s] - g b,aa*>w?OQ]*^*^aaa r zGo to https://huggingface.co/settings/webhooks to setup your webhooks.
You app is running locally. Please look at the logs to check the full URL you need to set.z
This app is running on a Space. You can find the corresponding URL in the options menu (top-right) > 'Embed the Space'. The URL looks like 'https://{username}-{repo_name}.hf.space'.)gradioBlocksMarkdownlenr' rP rJ rG )r) grr s r rE z WebhooksServer._get_default_ui sP
RY[[ B BKC D D D BKD
BKt/00MMM** 151I1O1O1Q1Q
BK q qv
! 2 s BB99B= B=)r r )NNN)F)r r )r/
__module__ __qualname____doc__r r strr* r r6 boolr rH rE
__classcell__)r s @r r r * s 3 3j $ $ $ $ $ $ %)(, 3 3 [ ! 3 !
3
3 3 3 3) )
) ) ) ) )V $ QU D r r+ r c t r t S t t j dt
dt
f fd }|S )a Decorator to start a [`WebhooksServer`] and register the decorated function as a webhook endpoint.
This is a helper to get started quickly. If you need more flexibility (custom landing page or webhook secret),
you can use [`WebhooksServer`] directly. You can register multiple webhook endpoints (to the same server) by using
this decorator multiple times.
Check out the [webhooks guide](../guides/webhooks_server) for a step-by-step tutorial on how to setup your
server and deploy it on a Space.
`webhook_endpoint` is experimental. Its API is subject to change in the future.
You must have `gradio` installed to use `webhook_endpoint` (`pip install --upgrade gradio`).
Args:
path (`str`, optional):
The URL path to register the webhook function. If not provided, the function name will be used as the path.
In any case, all webhooks are registered under `/webhooks`.
Examples:
The default usage is to register a function as a webhook endpoint. The function name will be used as the path.
The server will be started automatically at exit (i.e. at the end of the script).
```python
from huggingface_hub import webhook_endpoint, WebhookPayload
@webhook_endpoint
async def trigger_training(payload: WebhookPayload):
if payload.repo.type == "dataset" and payload.event.action == "update":
# Trigger a training job if a dataset is updated
...
# Server is automatically started at the end of the script.
```
Advanced usage: register a function as a webhook endpoint and start the server manually. This is useful if you
are running it in a notebook.
```python
from huggingface_hub import webhook_endpoint, WebhookPayload
@webhook_endpoint
async def trigger_training(payload: WebhookPayload):
if payload.repo.type == "dataset" and payload.event.action == "update":
# Trigger a training job if a dataset is updated
...
# Start the server manually
trigger_training.launch()
```
r2 r c t | t j dk rt j j t j fd }|| _ | S )Nr c b t j j d S r` )atexit
unregisterrH )apps r _launch_nowz5webhook_endpoint.._inner.._launch_now. s)
cj ) ) ) JJLLLLLr )_get_global_appr6 r^ r' rj registerrH r )r2 rm rl r+ s @r _innerz webhook_endpoint.._inner& s d### s& ' '1 , , OCJ ' ' ' sz
" r )r5 webhook_endpointr r r6 r )r+ rp s ` r rq rq st x ~~ (! !!$'''
> %&&X ( '& Mr c : t t a t S r` )r r r@ r r rn rn : s $&& r r! c | /t d t d t d d S t d d S )NzZWebhook secret is not defined. This means your webhook endpoints will be open to everyone.zTo add a secret, set `WEBHOOK_SECRET` as environment variable or pass it at initialization:
`app = WebhooksServer(webhook_secret='my_secret', ...)`zpFor more details about webhook secrets, please refer to https://huggingface.co/docs/hub/webhooks#webhook-secret.z$Webhook secret is correctly defined.)rQ r< s r r( r( A sj
jkkk
J
H
455555r webhook_namerZ c @ d| z | dd z dz S )z@Returns the anchor to a given webhook in the docs (experimental)z/docs#/default/r. rS _post)replace)rt rZ s r rY rY P s' | +l.B.B3.L.L Lw VVr r2 c f t j t dt f fd }dj vrl t j dt j j t ft j z |_
|S )a Wraps a webhook function to check the webhook secret before calling the function.
This is a hacky way to add the `request` parameter to the function signature. Since FastAPI based itself on route
parameters to inject the values to the function, we need to hack the function signature to retrieve the `Request`
object (and hence the headers). A far cleaner solution would be to use a middleware. However, since
`fastapi==0.90.1`, a middleware cannot be added once the app has started. And since the FastAPI app is started by
Gradio internals (and not by us), we cannot add a middleware.
This method is called only when a secret has been defined by the user. If a request is sent without the
"x-webhook-secret", the function will return a 401 error (unauthorized). If the header is sent but is incorrect,
the function will return a 403 error (forbidden).
Inspired by https://stackoverflow.com/a/33112180.
requestc K | j d }|t ddid S |k rt ddid S dj v r| |d< t j r d i | d {V S d i |S )
Nzx-webhook-secreterrorz x-webhook-secret header not set.i ) status_codezInvalid webhook secret.i ry r@ )headersrM r
parametersinspectiscoroutinefunction)ry r request_secretr2 initial_sigr! s r _protected_funcz6_wrap_webhook_to_check_secret.._protected_funcf s ,,-?@@ ! *L M[^___ _ ^ + + *C DRUVVV V . . . 'F9 &t , , " '''''' '4>>&>> !r )namekind
annotation)r~ ) r signaturer r
r~ rw ParameterPOSITIONAL_OR_KEYWORDtuplevalues
__signature__)r2 r! r r s `` @r rK rK U s #D))K
4[["w " " " " " " " ["$ ...(3(;(;!yw7H7^krsss K*1133445 )< )
)
% r r` )#rc rj r r% functoolsr typingr r r r r utilsr
r r r[ r_ fastapir
r fastapi.responsesr r __annotations__rL rM rG r rd rq rn r( rY rK r@ r r r sM Z Y Y
? ? ? ? ? ? ? ? ? ? ? ? ? ? J J J J J J J J J J ,((((((((....... (,+G+g +/ X&
' . . . JNN: & &$ . x x x x x x x xv P P8C= PH P P P Pf 6(3- 6D 6 6 6 6 Ws W# W# W W W W
- -# -( - - - - - -r