
    -:jN                        d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ  G d d	ee      Z G d
 dee      Zej                   ej"                  ej$                  ej&                  f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e G d d             Ze G d  d!             Zd"d"d"d#d$d%ed&e e   d'e!eef   d"z  d(e!eef   d"z  d)eez  d"z  d*e"e#z  ez  d"z  d+e!eef   d"z  d,e e
   d"z  d-e e"   d"z  d.e$d/e!eef   fd0Z%y")1    )	dataclass)datetime)Enum)Any)	constants)Volume)parse_datetimec                   x    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZy)JobHardwarea  
    Enumeration of hardware flavors available to run Jobs on the Hub.

    Value can be compared to a string:
    ```py
    assert JobHardware.CPU_BASIC == "cpu-basic"
    ```

    Both enums are kept in sync with the Hub API by `utils/check_hardware_flavors.py`.
    z	cpu-basiczcpu-upgradezcpu-performancezcpu-xlzt4-smallz	t4-mediuml4x1l4x4l40sx1l40sx4l40sx8z
a10g-smallz
a10g-largeza10g-largex2za10g-largex4z
a100-largea100x4a100x8h200h200x2h200x4h200x8zrtx-pro-6000zrtx-pro-6000x2zrtx-pro-6000x4zrtx-pro-6000x8N)__name__
__module____qualname____doc__	CPU_BASICCPU_UPGRADECPU_PERFORMANCECPU_XLT4_SMALL	T4_MEDIUML4X1L4X4L40SX1L40SX4L40SX8
A10G_SMALL
A10G_LARGEA10G_LARGEX2A10G_LARGEX4
A100_LARGEA100X4A100X8H200H200X2H200X4H200X8RTX_PRO_6000RTX_PRO_6000X2RTX_PRO_6000X4RTX_PRO_6000X8     [/var/www/html/tokenscope/api/venv/lib/python3.12/site-packages/huggingface_hub/_jobs_api.pyr   r      s    	 IK'OF HIDDFFFJJ!L!LJFFDFFF!L%N%N%Nr6   r   c                   (    e Zd ZdZdZdZdZdZdZdZ	y)	JobStageav  
    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`, `SCHEDULING`, `RUNNING`.
    Taken from https://github.com/huggingface/moon-landing/blob/main/server/job_types/JobInfo.ts#L61 (private url).
    	COMPLETEDCANCELEDERRORDELETED
SCHEDULINGRUNNINGN)
r   r   r   r   r:   r;   r<   r=   r>   r?   r5   r6   r7   r9   r9   C   s'    	 IHEGJGr6   r9   c                   N    e Zd ZU eed<   edz  ed<   ee   dz  ed<   edz  ed<   y)	JobStatusstageNmessageexpose_urlsssh_url)r   r   r   r9   __annotations__strlistr5   r6   r7   rA   rA   \   s*    O4ZcT!!4Zr6   rA   c                   ,    e Zd ZU eed<   eed<   eed<   y)JobOwneridnametypeN)r   r   r   rG   rF   r5   r6   r7   rJ   rJ   d   s    G
I
Ir6   rJ   c                   J    e Zd ZU dZedz  ed<   edz  ed<   edz  ed<   ddZy)JobDurationsau  
    Timing breakdown for a Job, computed server-side.

    Args:
        scheduling_secs (`int` or `None`):
            Seconds the job spent in the scheduling stage before starting to run.
            `None` if the job never reached the running stage.
        running_secs (`int` or `None`):
            Seconds the job has been or was running. Recomputed on each request
            while the job is in progress. `None` if the job never started running.
        total_secs (`int` or `None`):
            Total seconds elapsed since the job was created. Recomputed on each
            request while the job is in progress.
    Nscheduling_secsrunning_secs
total_secsc                     |j                  d|j                  d            | _        |j                  d|j                  d            | _        |j                  d|j                  d            | _        y )NschedulingSecsrP   runningSecsrQ   	totalSecsrR   )getrP   rQ   rR   selfkwargss     r7   __init__zJobDurations.__init__   sW    %zz*:FJJGX<YZ"JJ}fjj6PQ **[&**\2JKr6   returnN)r   r   r   r   intrF   r[   r5   r6   r7   rO   rO   k   s-     4Z*d
Lr6   rO   c                   :    e Zd ZU dZeed<   eed<   dZedz  ed<   y)JobInitiatoraJ  
    Contains information about what triggered a Job.

    Args:
        type (`str`): Initiator kind, for example `"user"`, `"org"`, `"scheduled-job"`, or `"duplicated-job"`.
        id (`str`): Identifier of the initiator.
        name (`str` or `None`): Human-readable name when available, usually for user/org initiators.
    rM   rK   NrL   )r   r   r   r   rG   rF   rL   r5   r6   r7   r`   r`      s!     IGD#*r6   r`   c                   \   e Zd ZU dZeed<   edz  ed<   edz  ed<   edz  ed<   edz  ed<   edz  ed<   ee   dz  ed	<   ee   dz  ed
<   eee	f   dz  ed<   eee	f   dz  ed<   e
dz  ed<   eeef   dz  ed<   ee   dz  ed<   eed<   edz  ed<   eed<   edz  ed<   eed<   eed<   ddZy)JobInfoa  
    Contains information about a Job.

    Args:
        id (`str`):
            Job ID.
        created_at (`datetime` or `None`):
            When the Job was created.
        started_at (`datetime` or `None`):
            When the Job started running. None while the Job is still scheduling.
        finished_at (`datetime` or `None`):
            When the Job finished. None while the Job is still scheduling or running.
        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. See [`JobHardware`] for possible values.
            E.g. `"cpu-basic"`.
        labels (`dict[str, str]` or `None`):
            Labels to attach to the job (key-value pairs).
        volumes (`list[Volume]` or `None`):
            Volumes mounted in the job container (buckets, models, datasets, spaces).
        status: (`JobStatus` or `None`):
            Status of the Job, e.g. `JobStatus(stage="RUNNING", message=None)`
            See [`JobStage`] for possible stage values.
        durations (`JobDurations` or `None`):
            Timing breakdown of the Job. Present for all job states including SCHEDULING.
        owner: (`JobOwner` or `None`):
            Owner of the Job, e.g. `JobOwner(id="5e9ecfc04957053f60648a3e", name="lhoestq", type="user")`
        initiator (`JobInitiator` or `None`):
            What triggered the Job, e.g. `JobInitiator(type="scheduled-job", id="...")` for a cron-triggered run.
        expose_urls (`list[str]` or `None`):
            Public URLs through which the Job's exposed ports are reachable (one per port exposed via `expose=`),
            e.g. `["https://687fb701029421ae5549d998--8000.hf.jobs"]`. `None` when no port is exposed.
            Accessing a URL requires an HF token with read access to the Job's namespace.
        ssh_url (`str` or `None`):
            SSH endpoint of the Job, e.g. `"ssh://687fb701029421ae5549d998@ssh.hf.jobs"`. Only present when the Job
            was started with `ssh=True`. Connecting requires write access to the Job's namespace and an SSH public
            key registered on the Hub (https://huggingface.co/settings/keys).

    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), started_at=datetime.datetime(2025, 7, 22, 16, 6, 31, 79000, tzinfo=datetime.timezone.utc), finished_at=None, 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), durations=JobDurations(scheduling_secs=6, running_secs=2, total_secs=8), owner=JobOwner(id='5e9ecfc04957053f60648a3e', name='lhoestq', type='user'), initiator=JobInitiator(type='user', id='5e9ecfc04957053f60648a3e', name='lhoestq'), 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'
    ```
    rK   N
created_at
started_atfinished_atdocker_imagespace_idcommand	argumentsenvironmentsecretsflavorlabelsvolumesstatus	durationsowner	initiatorendpointurlc                    |d   | _         |j                  d      xs |j                  d      }|rt        |      nd | _        |j                  d      xs |j                  d      }|rt        |      nd | _        |j                  d      xs |j                  d      }|rt        |      nd | _        |j                  d      xs |j                  d	      | _        |j                  d
      xs |j                  d      | _        |j                  di       }t        |d   |d   |d         | _	        |j                  d      | _
        |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d      }|r|D cg c]  }t!        d#i | c}nd | _        |j                  di       }t%        |d   |j                  d      |j                  d      |j                  d            | _        |j                  d      }	|	rt)        d#i |	nd | _        |j                  d      }
|
r#t-        |
d   |
d   |
j                  d            nd | _        |j                  d t0        j2                        | _        | j4                   d!| j                  j6                   d"| j                    | _        y c c}w )$NrK   	createdAtrc   	startedAtrd   
finishedAtre   dockerImagerf   spaceIdrg   rq   rL   rM   rK   rL   rM   rh   ri   rj   rk   rl   rm   rn   ro   rB   rC   
exposeUrlssshUrl)rB   rC   rD   rE   rp   rr   )rM   rK   rL   rs   z/jobs//r5   )rK   rW   r	   rc   rd   re   rf   rg   rJ   rq   rh   ri   rj   rk   rl   rm   r   rn   rA   ro   rO   rp   r`   rr   r   ENDPOINTrs   rL   rt   )rY   rZ   rc   rd   re   rq   rn   vro   rp   rr   s              r7   r[   zJobInfo.__init__   sq   ,ZZ,H

<0H
8B.4ZZ,H

<0H
8B.4jj.K&**]2K:E>+64"JJ}5SN9S

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

7B't5=uV}U
zz),K0!::m4zz),jj*jj***Y'9@W55dHb)/JJy)

<0JJx(	
 JJ{+	6?2	2TJJ{+	dmLi/IdO)--X^J_`sw 	
 

:y/A/ABmm_F4::??*;1TWWIF# 6s   K?r\   )r   r   r   r   rG   rF   r   rH   dictr   r   r   rA   rO   rJ   r`   r[   r5   r6   r7   rb   rb      s    DL 	G44D *Dj#YCy4c3h$&&#s(^d""$cNT!!&\D  d""Od"" M	H$Gr6   rb   c                      e Zd ZU edz  ed<   edz  ed<   ee   dz  ed<   ee   dz  ed<   eeef   dz  ed<   eeef   dz  ed<   edz  ed<   e	dz  ed	<   ee   dz  ed
<   edz  ed<   eeef   dz  ed<   ee
   dz  ed<   ddZy)JobSpecNrf   rg   rh   ri   rj   rk   rl   timeouttagsarchrm   rn   c                    |j                  d      xs |j                  d      | _        |j                  d      xs |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d	      | _        |j                  d
      | _        |j                  d      | _	        |j                  d      | _
        |j                  d      | _        |j                  d      }|r|D cg c]  }t        di | c}| _        y d | _        y c c}w )Nry   rf   rz   rg   rh   ri   rj   rk   rl   r   r   r   rm   rn   r5   )rW   rf   rg   rh   ri   rj   rk   rl   r   r   r   rm   r   rn   )rY   rZ   rn   r   s       r7   r[   zJobSpec.__init__*  s   "JJ}5SN9S

9-GJ1Gzz),K0!::m4zz),jj*zz),JJv&	JJv&	jj***Y'9@W55d5s   0Er\   )r   r   r   rG   rF   rH   r   r   r   r^   r   r[   r5   r6   r7   r   r     s    *Dj#YCy4c3h$&&#s(^d""$4Z
s)d

*cNT!!&\D  Kr6   r   c                   *    e Zd ZU eed<   eed<   ddZy)LastJobInforK   atNc                 >    |d   | _         t        |d         | _        y )NrK   r   )rK   r	   r   rX   s     r7   r[   zLastJobInfo.__init__?  s    , .r6   r\   )r   r   r   rG   rF   r   r[   r5   r6   r7   r   r   :  s    GL/r6   r   c                   6    e Zd ZU edz  ed<   edz  ed<   ddZy)ScheduledJobStatusNlast_jobnext_job_run_atc                    |j                  d      xs |j                  d      }|rt        di |nd | _        |j                  d      xs |j                  d      }|rt        t	        |            | _        y d | _        y )NlastJobr   nextJobRunAtr   r5   )rW   r   r   r	   rG   r   )rY   rZ   r   r   s       r7   r[   zScheduledJobStatus.__init__I  sj    ::i(BFJJz,B3;/h/ **^4U

CT8UGV~c/.BC\`r6   r\   )r   r   r   r   rF   r   r[   r5   r6   r7   r   r   D  s    D  _$ar6   r   c                       e Zd ZU dZeed<   edz  ed<   eed<   edz  ed<   edz  ed<   edz  ed<   e	ed	<   e
ed
<   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)
    ```
    rK   Nrc   job_specschedulesuspendconcurrencyro   rq   c                    |d   | _         |j                  d      xs |j                  d      }|rt        |      nd | _        t	        di |j                  d      xs |j                  di       | _        |j                  d      | _        |j                  d      | _        |j                  d      | _        |j                  d	i       }t        |j                  d
      xs |j                  d      |j                  d      xs |j                  d            | _
        |j                  di       }t        |d   |d   |d         | _        y )NrK   rv   rc   r   jobSpecr   r   r   ro   r   r   r   r   )r   r   rq   rL   rM   r{   r5   )rK   rW   r	   rc   r   r   r   r   r   r   ro   rJ   rq   )rY   rZ   rc   ro   rq   s        r7   r[   zScheduledJobInfo.__init__  s   ,ZZ,H

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

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

7B't5=uV}U
r6   r\   )r   r   r   r   rG   rF   r   r   boolr   rJ   r[   r5   r6   r7   r   r   P  sN    &P 	G4DjD[OVr6   r   c                   L    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y)
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"`.
    rM   modelquantityvrammanufacturerNc                 h    |d   | _         |d   | _        |d   | _        |d   | _        |d   | _        y )NrM   r   r   r   r   )rM   r   r   r   r   rX   s     r7   r[   zJobAccelerator.__init__  s<    6N	G_
z*6N	">2r6   r\   )r   r   r   r   rG   rF   r[   r5   r6   r7   r   r     s(      IJM
I3r6   r   c                   z    e Zd ZU dZeed<   eed<   eed<   eed<   eed<   edz  ed<   eed	<   eed
<   eed<   ddZ	y)JobHardwareInfoa$  
    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"`.
        ephemeral_storage (`str`):
            Ephemeral storage specification, e.g. `"20 GB"`, `"100 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]
    JobHardwareInfo(name='cpu-basic', pretty_name='CPU Basic', cpu='2 vCPU', ram='16 GB', ephemeral_storage='20 GB', accelerator=None, unit_cost_micro_usd=167, unit_cost_usd=0.000167, unit_label='minute')
    >>> hardware_list[0].name
    'cpu-basic'
    ```
    rL   pretty_namecpuramephemeral_storageNacceleratorunit_cost_micro_usdunit_cost_usd
unit_labelc                    |d   | _         |d   | _        |d   | _        |d   | _        |j	                  dd      | _        |j	                  d      }|rt        di |nd | _        |d   | _        |d	   | _	        |d
   | _
        y )NrL   
prettyNamer   r   ephemeralStoragezN/Ar   unitCostMicroUSDunitCostUSD	unitLabelr5   )rL   r   r   r   rW   r   r   r   r   r   r   )rY   rZ   r   s      r7   r[   zJobHardwareInfo.__init__  s    6N	!,/%=%=!',>!Fjj/<G>8K8T#)*<#= #M2 -r6   r\   )
r   r   r   r   rG   rF   r   r^   floatr[   r5   r6   r7   r   r     sF    !F I	H	H$&&O
.r6   r   NF)rm   rn   exposesshimagerh   envrk   rl   r   rm   rn   r   r   r]   c        
            |g |xs i |xs t         j                  d}
|r||
d<   |rPddddd}t        |t              r+|d   |v r$t	        t        |d d       ||d      z        |
d	<   nt	        |      |
d	<   |r||
d
<   |r!|D cg c]  }|j                          c}|
d<   |rd|i|
d<   |	rddi|
d<   dD ]'  }| j                  |      s| t        |      d  |
d<    |
S  | |
d<   |
S c c}w )N)rh   ri   rj   rl   rk      <   i  iQ )smhdtimeoutSecondsrm   rn   portsr   enabledTr   )zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/zhf.co/spaces/rz   ry   )	r   r   
isinstancerG   r^   r   to_dict
startswithlen)r   rh   r   rk   rl   r   rm   rn   r   r   r   time_units_factorsvolprefixs                 r7   _create_job_specr     s>    yb1K11	 H %#$2DyIgs#7I(I),U73B<-@CUV]^`VaCb-b)cH%&),WH%&#8?@s{{}@%v.
$d+ 
( F#"'F"6HY O
( #(O' As   C&)&dataclassesr   r   enumr   typingr   huggingface_hubr   huggingface_hub._space_apir   huggingface_hub.utils._datetimer	   rG   r   r9   r:   r;   r<   r=   TERMINAL_JOB_STAGESrA   rJ   rO   r`   rb   r   r   r   r   r   r   rH   r   r^   r   r   r   r5   r6   r7   <module>r      sg   "    % - :(&#t (&VsD ,  ))8+<+<hnnhN^N^_        L L L4    AG AG AGH K K K< / / / a a a @V @V @VF 3 3 3> 8. 8. 8.F %)#'#66 #Y6 
c3h$		6
 #s(^d"6 #$6 5[3%6 cNT!6 &\D 6 I6 
6 
#s(^6r6   