o
    §iw*  ã                   @   sL  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	„ d	eeƒZeG d
d„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZ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f fdd „Zd!S )"é    )Ú	dataclass)Údatetime)ÚEnum)ÚAnyÚOptionalÚUnion)Ú	constants)ÚSpaceHardware)Úparse_datetimec                   @   s$   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   © r   r   ú_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/huggingface_hub/_jobs_api.pyr      s    r   c                   @   s"   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   
 r   c                   @   s&   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   
 r   c                   @   s¼   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d< eed< eed< eed< 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"`.
        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', 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ÚstatusÚownerÚendpointÚurlÚreturnNc                 K   s  |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| _| dtj¡| _| j› d| jj› d| j › | _d S )Nr   Ú	createdAtr!   ÚdockerImager"   ÚspaceIdr#   r*   r   r   ©r   r   r   r$   r%   r&   r'   r(   r)   r   r   )r   r   r+   z/jobs/ú/)r   Úgetr
   r!   r"   r#   r   r*   r$   r%   r&   r'   r(   r   r)   r   ÚENDPOINTr+   r   r,   )ÚselfÚkwargsr!   r*   r)   r   r   r   Ú__init__~   s    
"zJobInfo.__init__©r-   N)r   r   r   r   r   r   r   r   ÚlistÚdictr   r	   r   r   r7   r   r   r   r   r    :   s    
 2r    c                   @   s¬   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
< ddd„ZdS )ÚJobSpecr"   r#   r$   r%   r&   r'   r(   ÚtimeoutÚtagsÚarchr-   Nc                 K   s   |  d¡p	|  d¡| _|  d¡p|  d¡| _|  d¡| _|  d¡| _|  d¡| _|  d¡| _|  d	¡| _|  d
¡| _|  d¡| _	|  d¡| _
d S )Nr/   r"   r0   r#   r$   r%   r&   r'   r(   r<   r=   r>   )r3   r"   r#   r$   r%   r&   r'   r(   r<   r=   r>   ©r5   r6   r   r   r   r7       s   zJobSpec.__init__r8   )r   r   r   r   r   r   r9   r:   r   r	   Úintr7   r   r   r   r   r;   “   s   
 r;   c                   @   s(   e Zd ZU eed< eed< ddd„ZdS )ÚLastJobInfor   Úatr-   Nc                 K   s   |d | _ t|d ƒ| _d S )Nr   rB   )r   r
   rB   r?   r   r   r   r7   ²   s   
zLastJobInfo.__init__r8   )r   r   r   r   r   r   r7   r   r   r   r   rA   ­   s   
 rA   c                   @   s0   e Zd ZU ee ed< ee ed< ddd„ZdS )ÚScheduledJobStatusÚlast_jobÚnext_job_run_atr-   Nc                 K   s`   |  d¡p	|  d¡}|rtdi |¤Žnd | _|  d¡p|  d¡}|r+tt|ƒƒ| _d S d | _d S )NÚlastJobrD   ÚnextJobRunAtrE   r   )r3   rA   rD   r
   r   rE   )r5   r6   rD   rE   r   r   r   r7   ¼   s    zScheduledJobStatus.__init__r8   )r   r   r   r   rA   r   r   r7   r   r   r   r   rC   ·   s   
 rC   c                   @   sl   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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Úconcurrencyr)   r*   r-   Nc                 K   sÜ   |d | _ | d¡p| d¡}|rt|ƒnd | _tdi | d¡p%| di ¡¤Ž| _| d¡| _| d¡| _| d¡| _| d	i ¡}t	| d
¡pK| d¡| d¡pT| d¡d| _
| di ¡}t|d |d |d d| _d S )Nr   r.   r!   rI   ÚjobSpecrJ   rK   rL   r)   rD   rF   rE   rG   )rD   rE   r*   r   r   r1   r   )r   r3   r
   r!   r;   rI   rJ   rK   rL   rC   r)   r   r*   )r5   r6   r!   r)   r*   r   r   r   r7   ö   s   
"þ zScheduledJobInfo.__init__r8   )r   r   r   r   r   r   r   r   r;   ÚboolrC   r   r7   r   r   r   r   rH   Ã   s   
 (rH   Úimager$   Úenvr'   r(   r<   r-   c           	      C   s¸   |g |pi |p	t jdœ}|r||d< |r?dddddœ}t|tƒr9|d |v r9tt|d d… ƒ||d   ƒ|d	< nt|ƒ|d	< d
D ]}|  |¡rU| t|ƒd … |d<  |S qA| |d< |S )N)r$   r%   r&   r(   r'   é   é<   i  i€Q )ÚsÚmÚhÚdéÿÿÿÿÚtimeoutSeconds)zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/zhf.co/spaces/r0   r/   )r	   Ú	CPU_BASICÚ
isinstancer   r@   ÚfloatÚ
startswithÚlen)	rO   r$   rP   r'   r(   r<   rI   Útime_units_factorsÚprefixr   r   r   Ú_create_job_spec  s(   ü&
ûr`   N)Údataclassesr   r   Úenumr   Útypingr   r   r   Úhuggingface_hubr   Úhuggingface_hub._space_apir	   Úhuggingface_hub.utils._datetimer
   r   r   r   r   r    r;   rA   rC   rH   r9   r:   r@   r[   r`   r   r   r   r   Ú<module>   sJ   X	Cþýüûúù
ø