qi6 ddlmZddlmZddlmZddlmZmZmZddl m Z ddl m Z ddl mZGdd eeZeGd d ZeGd d ZeGddZeGddZeGddZeGddZeGddZeGddZeGddZdddedeed eeeefd!eeeefd"ee d#eeeeefd$eeeefd%eeeffd&Zy)') dataclass)datetime)Enum)AnyOptionalUnion) constants) SpaceHardware)parse_datetimec$eZdZdZdZdZdZdZdZy)JobStageah Enumeration of possible stage of a Job on the Hub. Value can be compared to a string: ```py assert JobStage.COMPLETED == "COMPLETED" ``` Possible values are: `COMPLETED`, `CANCELED`, `ERROR`, `DELETED`, `RUNNING`. Taken from https://github.com/huggingface/moon-landing/blob/main/server/job_types/JobInfo.ts#L61 (private url). COMPLETEDCANCELEDERRORDELETEDRUNNINGN) __name__ __module__ __qualname____doc__rrrrrK/opt/pipecat/venv/lib/python3.12/site-packages/huggingface_hub/_jobs_api.pyr r s" IH EGGrr c(eZdZUeed<eeed<y) JobStatusstagemessageN)rrrr __annotations__rstrrrrrr-s O c]rrc,eZdZUeed<eed<eed<y)JobOwneridnametypeN)rrrrrrrrr!r!3s G I Irr!ceZdZUdZeed<eeed<eeed<eeed<eeeed<eeeed<ee ee fed<ee ee fed <ee ed <ee eefed <e ed <e ed <eed<eed<ddZy)JobInfoa Contains information about a Job. Args: id (`str`): Job ID. created_at (`datetime` or `None`): When the Job was created. docker_image (`str` or `None`): The Docker image from Docker Hub used for the Job. Can be None if space_id is present instead. space_id (`str` or `None`): The Docker image from Hugging Face Spaces used for the Job. Can be None if docker_image is present instead. command (`list[str]` or `None`): Command of the Job, e.g. `["python", "-c", "print('hello world')"]` arguments (`list[str]` or `None`): Arguments passed to the command environment (`dict[str]` or `None`): Environment variables of the Job as a dictionary. secrets (`dict[str]` or `None`): Secret environment variables of the Job (encrypted). flavor (`str` or `None`): Flavor for the hardware, as in Hugging Face Spaces. See [`SpaceHardware`] for possible values. E.g. `"cpu-basic"`. labels (`dict[str, str]` or `None`): Labels to attach to the job (key-value pairs). status: (`JobStatus` or `None`): Status of the Job, e.g. `JobStatus(stage="RUNNING", message=None)` See [`JobStage`] for possible stage values. owner: (`JobOwner` or `None`): Owner of the Job, e.g. `JobOwner(id="5e9ecfc04957053f60648a3e", name="lhoestq", type="user")` Example: ```python >>> from huggingface_hub import run_job >>> job = run_job( ... image="python:3.12", ... command=["python", "-c", "print('Hello from the cloud!')"] ... ) >>> job JobInfo(id='687fb701029421ae5549d998', created_at=datetime.datetime(2025, 7, 22, 16, 6, 25, 79000, tzinfo=datetime.timezone.utc), docker_image='python:3.12', space_id=None, command=['python', '-c', "print('Hello from the cloud!')"], arguments=[], environment={}, secrets={}, flavor='cpu-basic', labels=None, status=JobStatus(stage='RUNNING', message=None), owner=JobOwner(id='5e9ecfc04957053f60648a3e', name='lhoestq', type='user'), endpoint='https://huggingface.co', url='https://huggingface.co/jobs/lhoestq/687fb701029421ae5549d998') >>> job.id '687fb701029421ae5549d998' >>> job.url 'https://huggingface.co/jobs/lhoestq/687fb701029421ae5549d998' >>> job.status.stage 'RUNNING' ``` r" created_at docker_imagespace_idcommand arguments environmentsecretsflavorlabelsstatusownerendpointurlNc |d|_|jdxs|jd}|r t|nd|_|jdxs|jd|_|jdxs|jd|_|jdi}t |d|d |d  |_|jd |_|jd |_ |jd|_ |jd|_ |jd|_ |jd|_ |jdi}t|d|jd|_|jdt j"|_|j$d|jj&d|j|_y)Nr" createdAtr' dockerImager(spaceIdr)r1r#r$r"r#r$r*r+r,r-r.r/r0rr)rrr2z/jobs//)r"getr r'r(r)r!r1r*r+r,r-r.r/rr0r ENDPOINTr2r#r3)selfkwargsr'r1r0s r__init__zJobInfo.__init__sq,ZZ ,H <0H 8B.4"JJ}5SN9S 9-GJ1G  7B't5=uV}U zz), K0!::m4zz), jj* jj* Hb)fWovzz)?TU  :y/A/AB mm_F4::??*;1TWWIFrreturnN)rrrrrrrrlistdictrr rr!r>rrrr&r&:s2h G""3-sm d3i  S ""$sCx.)) d38n %% ] ## T#s(^ $$  OM HGrr&ceZdZUeeed<eeed<eeeed<eeeed<eeeefed<eeeefed<ee ed<ee ed<eeeed <eeed <eeeefed <dd Z y )JobSpecr(r)r*r+r,r-r.timeouttagsarchr/Nc 4|jdxs|jd|_|jdxs|jd|_|jd|_|jd|_|jd|_|jd|_|jd |_|jd |_|jd |_ |jd |_ |jd |_ y)Nr6r(r7r)r*r+r,r-r.rErFrGr/) r:r(r)r*r+r,r-r.rErFrGr/r<r=s rr>zJobSpec.__init__s"JJ}5SN9S 9-GJ1G zz), K0!::m4zz), jj* zz), JJv& JJv& jj* rr?) rrrrrrrArBrr intr>rrrrDrDs3-sm d3i  S ""$sCx.)) d38n %% ] ## c] 49  3- T#s(^ $$ +rrDc*eZdZUeed<eed<ddZy) LastJobInfor"atNc >|d|_t|d|_y)Nr"rM)r"r rMrIs rr>zLastJobInfo.__init__s, .rr?)rrrrrrr>rrrrLrLs GL/rrLc6eZdZUeeed<eeed<ddZy)ScheduledJobStatuslast_jobnext_job_run_atNc |jdxs|jd}|r tdi|nd|_|jdxs|jd}|rtt ||_yd|_y)NlastJobrQ nextJobRunAtrRr)r:rLrQr rrR)r<r=rQrRs rr>zScheduledJobStatus.__init__sj::i(BFJJz,B3; /h/  **^4U CT8UGV~c/.BC\`rr?)rrrrrLrrr>rrrrPrPs{##h''arrPceZdZUdZeed<eeed<eed<eeed<ee ed<ee ed<e ed<e ed <d d Z y ) ScheduledJobInfoa Contains information about a Job. Args: id (`str`): Scheduled Job ID. created_at (`datetime` or `None`): When the scheduled Job was created. tags (`list[str]` or `None`): The tags of the scheduled Job. schedule (`str` or `None`): One of "@annually", "@yearly", "@monthly", "@weekly", "@daily", "@hourly", or a CRON schedule expression (e.g., '0 9 * * 1' for 9 AM every Monday). suspend (`bool` or `None`): Whether the scheduled job is suspended (paused). concurrency (`bool` or `None`): Whether multiple instances of this Job can run concurrently. status (`ScheduledJobStatus` or `None`): Status of the scheduled Job. owner: (`JobOwner` or `None`): Owner of the scheduled Job, e.g. `JobOwner(id="5e9ecfc04957053f60648a3e", name="lhoestq", type="user")` job_spec: (`JobSpec` or `None`): Specifications of the Job. Example: ```python >>> from huggingface_hub import run_job >>> scheduled_job = create_scheduled_job( ... image="python:3.12", ... command=["python", "-c", "print('Hello from the cloud!')"], ... schedule="@hourly", ... ) >>> scheduled_job.id '687fb701029421ae5549d999' >>> scheduled_job.status.next_job_run_at datetime.datetime(2025, 7, 22, 17, 6, 25, 79000, tzinfo=datetime.timezone.utc) ``` r"r'job_specschedulesuspend concurrencyr0r1Nc |d|_|jdxs|jd}|r t|nd|_t di|jdxs|jdi|_|jd|_|jd|_|jd|_|jd i}t|jd xs|jd |jd xs|jd |_ |jdi}t|d|d|d|_ y)Nr"r5r'rXjobSpecrYrZr[r0rQrTrRrU)rQrRr1r#r$r8r) r"r:r r'rDrXrYrZr[rPr0r!r1)r<r=r'r0r1s rr>zScheduledJobInfo.__init__s,ZZ ,H <0H 8B.4X6::j#9#VVZZ SU=VX  :. zz), !::m4Hb)(ZZ +Dvzz)/D"JJ'89WVZZ=W   7B't5=uV}U rr?) rrrrrrrrrDboolrPr!r>rrrrWrWsO&P G""sm d^$  OVrrWcLeZdZUdZeed<eed<eed<eed<eed<d dZy) JobAcceleratora Contains information about a Job accelerator (GPU). Args: type (`str`): Type of accelerator, e.g. `"gpu"`. model (`str`): Model of accelerator, e.g. `"T4"`, `"A10G"`, `"A100"`, `"L4"`, `"L40S"`. quantity (`str`): Number of accelerators, e.g. `"1"`, `"2"`, `"4"`, `"8"`. vram (`str`): Total VRAM, e.g. `"16 GB"`, `"24 GB"`. manufacturer (`str`): Manufacturer of the accelerator, e.g. `"Nvidia"`. r$modelquantityvram manufacturerNc h|d|_|d|_|d|_|d|_|d|_y)Nr$rarbrcrd)r$rarbrcrdrIs rr>zJobAccelerator.__init__%s<6N G_ z* 6N ">2rr?)rrrrrrr>rrrr`r` s( I JM I3rr`cpeZdZUdZeed<eed<eed<eed<eeed<eed<e ed<eed <d d Z y ) JobHardwarea Contains information about available Job hardware. Args: name (`str`): Machine identifier, e.g. `"cpu-basic"`, `"a10g-large"`. pretty_name (`str`): Human-readable name, e.g. `"CPU Basic"`, `"Nvidia A10G - large"`. cpu (`str`): CPU specification, e.g. `"2 vCPU"`, `"12 vCPU"`. ram (`str`): RAM specification, e.g. `"16 GB"`, `"46 GB"`. accelerator (`JobAccelerator` or `None`): GPU/accelerator details if available. unit_cost_micro_usd (`int`): Cost in micro-dollars per unit, e.g. `167` (= $0.000167). unit_cost_usd (`float`): Cost in USD per unit, e.g. `0.000167`. unit_label (`str`): Cost unit period, e.g. `"minute"`. Example: ```python >>> from huggingface_hub import list_jobs_hardware >>> hardware_list = list_jobs_hardware() >>> hardware_list[0] JobHardware(name='cpu-basic', pretty_name='CPU Basic', cpu='2 vCPU', ram='16 GB', accelerator=None, unit_cost_micro_usd=167, unit_cost_usd=0.000167, unit_label='minute') >>> hardware_list[0].name 'cpu-basic' ``` r# pretty_namecpuram acceleratorunit_cost_micro_usd unit_cost_usd unit_labelNc |d|_|d|_|d|_|d|_|j d}|r t d i|nd|_|d|_|d|_|d|_ y) Nr# prettyNamerirjrkunitCostMicroUSD unitCostUSD unitLabelr) r#rhrirjr:r`rkrlrmrn)r<r=rks rr>zJobHardware.__init__Ysy6N !,/%=%=jj/ 8K8T#)*<#= #M2 -rr?) rrrrrrrr`rJfloatr>rrrrgrg-s@B I H H.))O .rrgN)r/imager*envr-r.rEr/r@cb|g|xsi|xstjd}|r||d<|rPddddd}t|tr+|d|vr$t t |dd||dz|d <nt ||d <|r||d <d D]'} |j | s|t| d|d <|S||d <|S)N)r*r+r,r.r-<iiQ)smhdtimeoutSecondsr/)zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/z hf.co/spaces/r7r6)r CPU_BASIC isinstancerrJrt startswithlen) rur*rvr-r.rEr/rXtime_units_factorsprefixs r_create_job_specresyb3M33  H%#$2DyI gs # 7I(I),U73B<-@CUV]^`VaCb-b)cH% &),WH% & # (   F #"'F "6HY   O (#( Or) dataclassesrrenumrtypingrrrhuggingface_hubr huggingface_hub._space_apir huggingface_hub.utils._datetimer rr rr!r&rDrLrPrWr`rgrArBrJrtrrrrrs"''%4:sD(       YGYG YGx ++ +6 // / aa a @V@V @VF 33 3> 4.4. 4.~(,* *#Y* $sCx. ! * d38n % * ] # *eCO, -* T#s(^ $* #s(^*r