
    i8                        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
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dedee   d eeef   dz  d!eeef   dz  d"e
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ef   fd'Zy)(    )	dataclass)datetime)Enum)Any)	constants)SpaceHardwareVolume)parse_datetimec                   $    e Zd ZdZdZdZdZdZdZy)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        j/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/huggingface_hub/_jobs_api.pyr   r      s"    	 IHEGGr   r   c                   (    e Zd ZU eed<   edz  ed<   y)	JobStatusstageNmessage)r   r   r   r   __annotations__strr   r   r   r   r   ,   s    O4Zr   r   c                   ,    e Zd ZU eed<   eed<   eed<   y)JobOwneridnametypeN)r   r   r   r   r   r   r   r   r    r    2   s    G
I
Ir   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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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.
        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).
        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.
        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!   N
created_atdocker_imagespace_idcommand	argumentsenvironmentsecretsflavorlabelsvolumesstatusownerendpointurlc                 $   |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t$        j&                        | _        | j(                   d| j                  j*                   d| j                    | _        y c c}w )Nr!   	createdAtr&   dockerImager'   spaceIdr(   r1   r"   r#   r!   r"   r#   r)   r*   r+   r,   r-   r.   r/   r0   r   r   )r   r   r2   z/jobs//r   )r!   getr
   r&   r'   r(   r    r1   r)   r*   r+   r,   r-   r.   r	   r/   r   r0   r   ENDPOINTr2   r"   r3   )selfkwargsr&   r1   r/   vr0   s          r   __init__zJobInfo.__init__   s   ,ZZ,H

<0H
8B.4"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)fWovzz)?TU 

:y/A/ABmm_F4::??*;1TWWIF 6s   HreturnN)r   r   r   r   r   r   r   listdictr   r   r	   r   r    r?   r   r   r   r%   r%   9   s    4l 	G4*Dj#YCy4c3h$&&#s(^d""D  cNT!!&\D  O M	HGr   r%   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)JobSpecNr'   r(   r)   r*   r+   r,   r-   timeouttagsarchr.   r/   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 )Nr6   r'   r7   r(   r)   r*   r+   r,   r-   rF   rG   rH   r.   r/   r   )r:   r'   r(   r)   r*   r+   r,   r-   rF   rG   rH   r.   r	   r/   )r<   r=   r/   r>   s       r   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   r   r   rB   rC   r   r   intr	   r?   r   r   r   rE   rE      s    *Dj#YCy4c3h$&&#s(^d""D  4Z
s)d

*cNT!!&\D  Kr   rE   c                   *    e Zd ZU eed<   eed<   ddZy)LastJobInfor!   atNc                 >    |d   | _         t        |d         | _        y )Nr!   rM   )r!   r
   rM   r<   r=   s     r   r?   zLastJobInfo.__init__   s    , .r   r@   )r   r   r   r   r   r   r?   r   r   r   rL   rL      s    GL/r   rL   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lastJobrR   nextJobRunAtrS   r   )r:   rL   rR   r
   r   rS   )r<   r=   rR   rS   s       r   r?   zScheduledJobStatus.__init__   sj    ::i(BFJJz,B3;/h/ **^4U

CT8UGV~c/.BC\`r   r@   )r   r   r   rL   r   r   r?   r   r   r   rQ   rQ      s    D  _$ar   rQ   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)
    ```
    r!   Nr&   job_specschedulesuspendconcurrencyr0   r1   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 )Nr!   r5   r&   rY   jobSpecrZ   r[   r\   r0   rR   rU   rS   rV   )rR   rS   r1   r"   r#   r8   r   )r!   r:   r
   r&   rE   rY   rZ   r[   r\   rQ   r0   r    r1   )r<   r=   r&   r0   r1   s        r   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
r   r@   )r   r   r   r   r   r   r   rE   boolrQ   r    r?   r   r   r   rX   rX      sN    &P 	G4DjD[OVr   rX   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"`.
    r#   modelquantityvrammanufacturerNc                 h    |d   | _         |d   | _        |d   | _        |d   | _        |d   | _        y )Nr#   rb   rc   rd   re   )r#   rb   rc   rd   re   rO   s     r   r?   zJobAccelerator.__init__,  s<    6N	G_
z*6N	">2r   r@   )r   r   r   r   r   r   r?   r   r   r   ra   ra     s(      IJM
I3r   ra   c                   p    e Zd ZU dZ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)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ramNacceleratorunit_cost_micro_usdunit_cost_usd
unit_labelc                     |d   | _         |d   | _        |d   | _        |d   | _        |j	                  d      }|rt        d	i |nd | _        |d   | _        |d   | _        |d   | _	        y )
Nr"   
prettyNamerj   rk   rl   unitCostMicroUSDunitCostUSD	unitLabelr   )
r"   ri   rj   rk   r:   ra   rl   rm   rn   ro   )r<   r=   rl   s      r   r?   zJobHardware.__init__`  sy    6N	!,/%=%=jj/<G>8K8T#)*<#= #M2 -r   r@   )
r   r   r   r   r   r   ra   rJ   floatr?   r   r   r   rh   rh   4  s@    B I	H	H$&&O	.r   rh   N)r.   r/   imager)   envr,   r-   rF   r.   r/   rA   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<   dD ]'  }| j                  |      s| t        |      d  |d<    |S  | |d<   |S c c}
w )N)r)   r*   r+   r-   r,      <   i  iQ )smhdtimeoutSecondsr.   r/   )zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/zhf.co/spaces/r7   r6   )	r   	CPU_BASIC
isinstancer   rJ   ru   to_dict
startswithlen)rv   r)   rw   r,   r-   rF   r.   r/   rY   time_units_factorsvolprefixs               r   _create_job_specr   l  s    yb3M33	 H %#$2DyIgs#7I(I),U73B<-@CUV]^`VaCb-b)cH%&),WH%&#8?@s{{}@ 
( F#"'F"6HY O
( #(O As   C)dataclassesr   r   enumr   typingr   huggingface_hubr   huggingface_hub._space_apir   r	   huggingface_hub.utils._datetimer
   r   r   r   r    r%   rE   rL   rQ   rX   ra   rh   rB   rC   rJ   ru   r   r   r   r   <module>r      s   "    % < :sD (   
    ^G ^G ^GB K K K< / / / a a a @V @V @VF 3 3 3> 4. 4. 4.~ %)#'.. #Y. 
c3h$		.
 #s(^d". D . 5[3%. cNT!. &\D . 
#s(^.r   