
    Zi6                        d dl mZ d dlmZ d dlmZ d dlmZmZmZ d dl	m
Z
 d dlmZ d dlmZ  G d d	ee          Ze G d
 d                      Ze G d d                      Ze G d d                      Ze G d d                      Ze G d d                      Ze G d d                      Ze G d d                      Ze G d d                      Ze G d d                      Zdddedee         d eeeef                  d!eeeef                  d"ee         d#eeeeef                  d$eeeef                  d%eeef         fd&ZdS )'    )	dataclass)datetime)Enum)AnyOptionalUnion)	constants)SpaceHardware)parse_datetimec                   &    e Zd ZdZdZdZdZdZdZdS )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).
    	COMPLETEDCANCELEDERRORDELETEDRUNNINGN)	__name__
__module____qualname____doc__r   r   r   r   r        T/root/projects/butler/venv/lib/python3.11/site-packages/huggingface_hub/_jobs_api.pyr   r      s3        	 	 IHEGGGGr   r   c                   0    e Zd ZU eed<   ee         ed<   dS )	JobStatusstagemessageN)r   r   r   r   __annotations__r   strr   r   r   r   r   -   s+         OOOc]r   r   c                   .    e Zd ZU eed<   eed<   eed<   dS )JobOwneridnametypeN)r   r   r   r   r   r   r   r   r!   r!   3   s+         GGG
III
IIIIIr   r!   c                   \   e Zd ZU dZeed<   ee         ed<   ee         ed<   ee         ed<   eee                  ed<   eee                  ed<   ee	ee
f                  ed<   ee	ee
f                  ed	<   ee         ed
<   ee	eef                  ed<   eed<   eed<   eed<   eed<   ddZdS )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_imagespace_idcommand	argumentsenvironmentsecretsflavorlabelsstatusownerendpointurlreturnNc                 0   |d         | _         |                    d          p|                    d          }|rt          |          nd | _        |                    d          p|                    d          | _        |                    d          p|                    d          | _        |                    di           }t          |d         |d	         |d
                   | _        |                    d          | _        |                    d          | _	        |                    d          | _
        |                    d          | _        |                    d          | _        |                    d          | _        |                    di           }t          |d         |                    d                    | _        |                    dt           j                  | _        | j         d| j        j         d| j          | _        d S )Nr"   	createdAtr'   dockerImager(   spaceIdr)   r1   r#   r$   r"   r#   r$   r*   r+   r,   r-   r.   r/   r0   r   r   )r   r   r2   z/jobs//)r"   getr   r'   r(   r)   r!   r1   r*   r+   r,   r-   r.   r/   r   r0   r	   ENDPOINTr2   r#   r3   )selfkwargsr'   r1   r0   s        r   __init__zJobInfo.__init__   s   ,ZZ,,H

<0H0H
8BL.444"JJ}55SN9S9S

9--GJ1G1G

7B''t5=uV}UUU
zz),,K00!::m44zz),,jj**jj**Hb))fWovzz)?T?TUUU 

:y/ABBmFF4:?FFTWFFr   r4   N)r   r   r   r   r   r   r   r   listdictr   r
   r   r!   r?   r   r   r   r&   r&   :   s&        2 2h 	GGG""""3-smd3i    S	""""$sCx.))))d38n%%%%]####T#s(^$$$$OOO MMM	HHHG G G G G Gr   r&   c                   ^   e Zd ZU ee         ed<   ee         ed<   eee                  ed<   eee                  ed<   eeeef                  ed<   eeeef                  ed<   ee	         ed<   ee
         ed<   eee                  ed	<   ee         ed
<   eeeef                  ed<   ddZdS )JobSpecr(   r)   r*   r+   r,   r-   r.   timeouttagsarchr/   r4   Nc                    |                     d          p|                     d          | _        |                     d          p|                     d          | _        |                     d          | _        |                     d          | _        |                     d          | _        |                     d          | _        |                     d	          | _        |                     d
          | _        |                     d          | _	        |                     d          | _
        |                     d          | _        d S )Nr7   r(   r8   r)   r*   r+   r,   r-   r.   rE   rF   rG   r/   )r;   r(   r)   r*   r+   r,   r-   r.   rE   rF   rG   r/   r=   r>   s     r   r?   zJobSpec.__init__   s    "JJ}55SN9S9S

9--GJ1G1Gzz),,K00!::m44zz),,jj**zz),,JJv&&	JJv&&	jj**r   r@   )r   r   r   r   r   r   rA   rB   r   r
   intr?   r   r   r   rD   rD      s        3-smd3i    S	""""$sCx.))))d38n%%%%]####c]
49

3-T#s(^$$$$+ + + + + +r   rD   c                   ,    e Zd ZU eed<   eed<   ddZdS )LastJobInfor"   atr4   Nc                 T    |d         | _         t          |d                   | _        d S )Nr"   rM   )r"   r   rM   rI   s     r   r?   zLastJobInfo.__init__   s$    , ..r   r@   )r   r   r   r   r   r   r?   r   r   r   rL   rL      s<         GGGLLL/ / / / / /r   rL   c                   D    e Zd ZU ee         ed<   ee         ed<   ddZdS )ScheduledJobStatuslast_jobnext_job_run_atr4   Nc                 "   |                     d          p|                     d          }|rt          di |nd | _        |                     d          p|                     d          }|rt          t	          |                    nd | _        d S )NlastJobrQ   nextJobRunAtrR   r   )r;   rL   rQ   r   r   rR   )r=   r>   rQ   rR   s       r   r?   zScheduledJobStatus.__init__   s    ::i((BFJJz,B,B3;E//h/// **^44U

CT8U8UGV`~c/.B.BCCC\`r   r@   )r   r   r   r   rL   r   r   r?   r   r   r   rP   rP      sR         {####h''''a a a a a ar   rP   c                       e Zd ZU dZeed<   ee         ed<   eed<   ee         ed<   ee	         ed<   ee	         ed<   e
ed<   eed	<   ddZdS )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_specschedulesuspendconcurrencyr0   r1   r4   Nc                 *   |d         | _         |                    d          p|                    d          }|rt          |          nd | _        t	          di |                    d          p|                    di           | _        |                    d          | _        |                    d          | _        |                    d          | _        |                    d	i           }t          |                    d
          p|                    d          |                    d          p|                    d                    | _
        |                    di           }t          |d         |d         |d                   | _        d S )Nr"   r6   r'   rX   jobSpecrY   rZ   r[   r0   rQ   rT   rR   rU   )rQ   rR   r1   r#   r$   r9   r   )r"   r;   r   r'   rD   rX   rY   rZ   r[   rP   r0   r!   r1   )r=   r>   r'   r0   r1   s        r   r?   zScheduledJobInfo.__init__   s\   ,ZZ,,H

<0H0H
8BL.444XX6::j#9#9#VVZZ	SU=V=VXX

:..zz),,!::m44Hb))(ZZ
++Dvzz)/D/D"JJ'899WVZZ=W=W
 
 
 

7B''t5=uV}UUU


r   r@   )r   r   r   r   r   r   r   r   rD   boolrP   r!   r?   r   r   r   rW   rW      s         & &P 	GGG""""smd^$OOOV V V V V Vr   rW   c                   N    e Zd ZU dZeed<   eed<   eed<   eed<   eed<   d
d	ZdS )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$   modelquantityvrammanufacturerr4   Nc                     |d         | _         |d         | _        |d         | _        |d         | _        |d         | _        d S )Nr$   ra   rb   rc   rd   )r$   ra   rb   rc   rd   rI   s     r   r?   zJobAccelerator.__init__%  sB    6N	G_
z*6N	">2r   r@   )r   r   r   r   r   r   r?   r   r   r   r`   r`     sf            IIIJJJMMM
III3 3 3 3 3 3r   r`   c                   x    e Zd ZU dZeed<   eed<   eed<   eed<   ee         ed<   eed<   e	ed<   eed	<   ddZ
dS )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_namecpuramacceleratorunit_cost_micro_usdunit_cost_usd
unit_labelr4   Nc                    |d         | _         |d         | _        |d         | _        |d         | _        |                    d          }|rt          d	i |nd | _        |d         | _        |d         | _        |d         | _	        d S )
Nr#   
prettyNameri   rj   rk   unitCostMicroUSDunitCostUSD	unitLabelr   )
r#   rh   ri   rj   r;   r`   rk   rl   rm   rn   )r=   r>   rk   s      r   r?   zJobHardware.__init__Y  s    6N	!,/%=%=jj//<GQ>88K888T#)*<#= #M2 -r   r@   )r   r   r   r   r   r   r   r`   rJ   floatr?   r   r   r   rg   rg   -  s          B III	HHH	HHH.))))OOO	. 	. 	. 	. 	. 	.r   rg   N)r/   imager*   envr-   r.   rE   r/   r4   c                    |g |pi |pt           j        d}|r||d<   |roddddd}t          |t                    rA|d         |v r7t	          t          |d d                   ||d                  z            |d	<   nt	          |          |d	<   |r||d
<   dD ]3}	|                     |	          r| t          |	          d          |d<    n4| |d<   |S )N)r*   r+   r,   r.   r-      <   i  iQ )smhdtimeoutSecondsr/   )zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/zhf.co/spaces/r8   r7   )r
   	CPU_BASIC
isinstancer   rJ   rt   
startswithlen)
ru   r*   rv   r-   r.   rE   r/   rX   time_units_factorsprefixs
             r   _create_job_specr   e  s/    yb3M3	   H  &% 6#$2DyIIgs## 	67I(I(I),U73B3<-@-@CUV]^`VaCb-b)c)cH%&&),WH%& $# 
( 
( F## 	"'F"6HYE	 #(Or   )dataclassesr   r   enumr   typingr   r   r   huggingface_hubr	   huggingface_hub._space_apir
   huggingface_hub.utils._datetimer   r   r   r   r!   r&   rD   rL   rP   rW   r`   rg   rA   rB   rJ   rt   r   r   r   r   <module>r      s   " ! ! ! ! !             ' ' ' ' ' ' ' ' ' ' % % % % % % 4 4 4 4 4 4 : : : : : :    sD   (        
         YG YG YG YG YG YG YG YGx + + + + + + + +6 / / / / / / / / a a a a a a a a @V @V @V @V @V @V @V @VF 3 3 3 3 3 3 3 3> 4. 4. 4. 4. 4. 4. 4. 4.~ (,* * ** #Y* 
$sCx.	!	*
 d38n%* ]#* eCO,-* T#s(^$* 
#s(^* * * * * *r   