o
    ?Hh                     @  s|   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
 er(d dlmZ dddZG dd de
ZG dd dZdS )    )annotations)TYPE_CHECKING)ArgumentParser   )
get_clientprint_model)	BaseModel)_SubParsersAction	subparser!_SubParsersAction[ArgumentParser]returnNonec                 C  st   |  d}|jtjd |  d}|jddddd |jtjtd	 |  d
}|jddddd |jtjtd	 d S )Nzmodels.list)funczmodels.retrievez-iz--idTzThe model ID)requiredhelp)r   
args_modelzmodels.delete)
add_parserset_defaults	CLIModelslistadd_argumentgetCLIModelIDArgsdelete)r
   sub r   V/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/openai/cli/_api/models.pyregister   s   


r   c                   @  s   e Zd ZU ded< dS )r   stridN)__name__
__module____qualname____annotations__r   r   r   r   r      s   
 r   c                   @  s6   e Zd ZedddZedddZedd	d
ZdS )r   argsr   r   r   c                 C     t  jj| jd}t| d S N)model)r   modelsretriever   r   r$   r'   r   r   r   r         zCLIModels.getc                 C  r%   r&   )r   r(   r   r   r   r*   r   r   r   r   $   r+   zCLIModels.deletec                  C  s"   t  j } | D ]}t| qd S )N)r   r(   r   r   )r(   r'   r   r   r   r   )   s   
zCLIModels.listN)r$   r   r   r   )r   r   )r    r!   r"   staticmethodr   r   r   r   r   r   r   r      s    r   N)r
   r   r   r   )
__future__r   typingr   argparser   _utilsr   r   _modelsr   r	   r   r   r   r   r   r   r   <module>   s    
