o
    -wi8                     @   s0   d dl mZ d dlmZ G dd dZdgZdS )    )Any)Apic                   @   s  e Zd ZdZdededdfddZdd	 Zd
d Zede	fddZ
edd Zedd Zedd Zedd Zedd Zedd ZdefddZdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zdd*d+Zd,d- Zdd.d/Zd0d1 Zd2d3 Z	4dd5ed6ed7eddfd8d9Z 	4dd5ed:ed6ed7eddf
d;d<Z!d=d> Z"d?d@ Z#dAdB Z$defdCdDZ%dEdF Z&dGdH Z'dIdJ Z(dKdL Z)dMdN Z*dOdP Z+dQdR Z,dSdT Z-dUdV Z.dWdX Z/dYdZ Z0d[d\ Z1d]d^ Z2d_d` Z3dadb Z4dcdd Z5dedf Z6dgdh Z7didj Z8dkdl Z9dmdn Z:dodp Z;dqdr Z<dsdt Z=dudv Z>dwdx Z?dydz Z@d{d| ZAd}d~ ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdS )r   z,Internal proxy to the official internal API.argskwargsreturnNc                 O   s   || _ || _d | _d S N)	_api_args_api_kwargs_apiselfr   r    r   P/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/apis/internal.py__init__   s   
zApi.__init__c                 C   s   | j  }|d= |S )zUUse for serializing.

        self._api is not serializable, so it's dropped
        r
   )__dict__copyr   stater   r   r   __getstate__   s   
zApi.__getstate__c                 C   s   | j | d| _dS )zkUsed for deserializing.

        Don't need to set self._api because it's constructed when needed.
        N)r   updater
   r   r   r   r   __setstate__   s   
zApi.__setstate__c                 C   s$   | j d u rt| ji | j| _ | j S r   )r
   InternalApir   r	   r   r   r   r   api!   s   
zApi.apic                 C      | j jS r   )r   api_keyr   r   r   r   r   *      zApi.api_keyc                 C   s   | j jd up| j jd uS r   )r   access_tokenr   r   r   r   r   is_authenticated.   s   zApi.is_authenticatedc                 C   r   r   )r   api_urlr   r   r   r   r   2   r   zApi.api_urlc                 C   r   r   )r   app_urlr   r   r   r   r    6   r   zApi.app_urlc                 C   r   r   )r   default_entityr   r   r   r   r!   :   r   zApi.default_entityc                 C   r   r   )r   gitr   r   r   r   r"   >   r   zApi.gitc                 C   
   | j  S )z>Returns whether the API key stored on initialization is valid.)r   validate_api_keyr   r   r   r   r$   B   s   
zApi.validate_api_keyc                 G   s   | j j| S r   )r   file_current)r   r   r   r   r   r%   F      zApi.file_currentc                 O      | j j|i |S r   )r   download_filer   r   r   r   r(   I      zApi.download_filec                 O   r'   r   )r   download_write_filer   r   r   r   r*   L   r)   zApi.download_write_filec                 C      | j |S r   )r   set_current_run_id)r   run_idr   r   r   r,   O   r&   zApi.set_current_run_idc                 C   r#   r   )r   viewerr   r   r   r   r.   R      
z
Api.viewerc                 C   r#   r   )r   max_cli_versionr   r   r   r   r0   U   r/   zApi.max_cli_versionc                 C   r#   r   )r   viewer_server_infor   r   r   r   r1   X   r/   zApi.viewer_server_infoc                 C   s   | j j|dS )N)entity)r   list_projects)r   r2   r   r   r   r3   [   s   zApi.list_projectsc                 C   r+   r   )r   format_project)r   projectr   r   r   r4   ^   r&   zApi.format_projectc                 C   s   | j j||||dS )N)iddescriptionr2   )r   upsert_project)r   r5   r6   r7   r2   r   r   r   r8   a   s   zApi.upsert_projectc                 O   r'   r   )r   
upsert_runr   r   r   r   r9   f   r)   zApi.upsert_runc                 O   r'   r   )r   settingsr   r   r   r   r:   i   r)   zApi.settingsFkeygloballypersistc                 C   s   | j |||S r   )r   clear_setting)r   r;   r<   r=   r   r   r   r>   l   s   zApi.clear_settingvaluec                 C   s   | j ||||S r   )r   set_setting)r   r;   r?   r<   r=   r   r   r   r@   q   s   zApi.set_settingc                 O   r'   r   )r   
parse_slugr   r   r   r   rA   v   r)   zApi.parse_slugc                 O   r'   r   )r   download_urlr   r   r   r   rB   y   r)   zApi.download_urlc                 O   r'   r   )r   download_urlsr   r   r   r   rC   |   r)   zApi.download_urlsc                 C   r#   r   )r   create_anonymous_api_keyr   r   r   r   rD      r/   zApi.create_anonymous_api_keyc                 O   r'   r   )r   pushr   r   r   r   rE      r)   zApi.pushc                 O   r'   r   )r   sweepr   r   r   r   rF      r)   z	Api.sweepc                 O   r'   r   )r   upsert_sweepr   r   r   r   rG      r)   zApi.upsert_sweepc                 O   r'   r   )r   set_sweep_stater   r   r   r   rH      r)   zApi.set_sweep_statec                 O   r'   r   )r   get_sweep_stater   r   r   r   rI      r)   zApi.get_sweep_statec                 O   r'   r   )r   
stop_sweepr   r   r   r   rJ      r)   zApi.stop_sweepc                 O   r'   r   )r   cancel_sweepr   r   r   r   rK      r)   zApi.cancel_sweepc                 O   r'   r   )r   pause_sweepr   r   r   r   rL      r)   zApi.pause_sweepc                 O   r'   r   )r   resume_sweepr   r   r   r   rM      r)   zApi.resume_sweepc                 O   r'   r   )r   register_agentr   r   r   r   rN      r)   zApi.register_agentc                 O   r'   r   )r   agent_heartbeatr   r   r   r   rO      r)   zApi.agent_heartbeatc                 O   r'   r   )r   use_artifactr   r   r   r   rP      r)   zApi.use_artifactc                 O   r'   r   )r   create_artifactr   r   r   r   rQ      r)   zApi.create_artifactc                 O   r'   r   )r   "complete_multipart_upload_artifactr   r   r   r   rR      r)   z&Api.complete_multipart_upload_artifactc                 O   r'   r   )r   
run_configr   r   r   r   rS      r)   zApi.run_configc                 O   r'   r   )r   upload_file_retryr   r   r   r   rT      r)   zApi.upload_file_retryc                 O   r'   r   )r   !upload_multipart_file_chunk_retryr   r   r   r   rU      r)   z%Api.upload_multipart_file_chunk_retryc                 O   r'   r   )r   get_run_infor   r   r   r   rV      r)   zApi.get_run_infoc                 O   r'   r   )r   get_run_stater   r   r   r   rW      r)   zApi.get_run_statec                 O   r'   r   )r   entity_is_teamr   r   r   r   rX      r)   zApi.entity_is_teamc                 O   r'   r   )r   get_project_run_queuesr   r   r   r   rY      r)   zApi.get_project_run_queuesc                 O   r'   r   )r   push_to_run_queuer   r   r   r   rZ      r)   zApi.push_to_run_queuec                 O   r'   r   )r   pop_from_run_queuer   r   r   r   r[      r)   zApi.pop_from_run_queuec                 O   r'   r   )r   ack_run_queue_itemr   r   r   r   r\      r)   zApi.ack_run_queue_itemc                 O   r'   r   )r   create_launch_agentr   r   r   r   r]      r)   zApi.create_launch_agentc                 O   r'   r   )r   create_default_resource_configr   r   r   r   r^      r)   z"Api.create_default_resource_configc                 O   r'   r   )r   create_run_queuer   r   r   r   r_      r)   zApi.create_run_queuec                 O   r'   r   )r   upsert_run_queuer   r   r   r   r`      r)   zApi.upsert_run_queuec                 O   r'   r   )r   create_custom_chartr   r   r   r   ra      r)   zApi.create_custom_chartc                 O   r'   r   )r   update_launch_agent_statusr   r   r   r   rb      r)   zApi.update_launch_agent_statusc                 O   r'   r   )r   launch_agent_introspectionr   r   r   r   rc      r)   zApi.launch_agent_introspectionc                 O   r'   r   )r   !fail_run_queue_item_introspectionr   r   r   r   rd      r)   z%Api.fail_run_queue_item_introspectionc                 O   r'   r   )r   fail_run_queue_itemr   r   r   r   re      r)   zApi.fail_run_queue_itemc                 O   r'   r   )r   update_run_queue_item_warningr   r   r   r   rf      r)   z!Api.update_run_queue_item_warningc                 O   r'   r   )r   get_launch_agentr   r   r   r   rg      r)   zApi.get_launch_agentc                 O   r'   r   )r   stop_runr   r   r   r   rh      r)   zApi.stop_runr   )NNN)FF)J__name__
__module____qualname____doc__r   r   r   r   propertyr   r   r   r   r   r    r!   r"   boolr$   r%   r(   r*   r,   r.   r0   r1   r3   r4   r8   r9   r:   strr>   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   r   r   r   r   r      s    	









r   N)typingr   wandb.sdk.internal.internal_apir   r   __all__r   r   r   r   <module>   s
     
j