a
    f                     @  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rPd dlmZ dd	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)_SubParsersActionz!_SubParsersAction[ArgumentParser]None)	subparserreturnc                 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   Z
args_modelzmodels.delete)
add_parserset_defaults	CLIModelslistadd_argumentgetCLIModelIDArgsdelete)r   sub r   O/var/www/ai-form-bot/venv/lib/python3.9/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                   @  sF   e Zd ZedddddZedddddZeddd	d
ZdS )r   r   r
   )argsr   c                 C  s   t  jj| jd}t| d S N)model)r   modelsretriever   r   r"   r$   r   r   r   r      s    zCLIModels.getc                 C  s   t  jj| jd}t| d S r#   )r   r%   r   r   r   r'   r   r   r   r   $   s    zCLIModels.delete)r   c                  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    staticmethodr   r   r   r   r   r   r   r      s   r   N)
__future__r   typingr   argparser   Z_utilsr   r   Z_modelsr   r	   r   r   r   r   r   r   r   <module>   s   