File: //usr/local/lib/python3.10/dist-packages/openai/cli/_api/__pycache__/models.cpython-310.pyc
o
���g � @ s| d dl mZ d dlmZ d dlmZ ddlmZmZ ddl m
Z
er(d dlm Z dd d
�Z G dd� de
�Z
G dd� d�ZdS )� )� annotations)�
TYPE_CHECKING)�ArgumentParser� )�
get_client� print_model)� BaseModel)�_SubParsersAction� subparser�!_SubParsersAction[ArgumentParser]�return�Nonec C st | � d�}|jtjd� | � d�}|jddddd� |jtjtd � | � d
�}|jddddd� |jtjtd � d S ) Nz models.list)�funczmodels.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 �A/usr/local/lib/python3.10/dist-packages/openai/cli/_api/models.py�register
s
r c @ s e Zd ZU ded< dS )r �str�idN)�__name__�
__module__� __qualname__�__annotations__r r r r r s
r c @ s6 e Zd Zed dd��Zed dd��Zed
d d
��Zd S )r �argsr r r
c C � t � jj| jd�}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+ zCLIModels.deletec C s" t � j�� } | D ]}t|� qd S )N)r r( r r )r( r' r r r r ) s
�zCLIModels.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