File: //home/arjun/.local/lib/python3.10/site-packages/openai/cli/_api/__pycache__/files.cpython-310.pyc
o
���g) � @ s� d dl mZ d dlmZmZmZ d dlmZ ddlm Z m
Z
ddl m Z ddl
mZ er2d dlmZ dd
d�ZG dd� de �ZG dd� de �ZG dd� d�ZdS )� )� annotations)�
TYPE_CHECKING�Any�cast)�ArgumentParser� )�
get_client� print_model)� BaseModel)� BufferReader)�_SubParsersAction� subparser�!_SubParsersAction[ArgumentParser]�return�Nonec C s� | � d�}|jddddd� |jddd dd
� |jtjtd � | � d �}|jd
dddd� |jtjtd � | � d�}|jd
dddd� |jtjtd � | � d�}|jtjd� d S )Nz files.createz-fz--fileTzFile to upload)�required�helpz-pz --purposezcWhy are you uploading this file? (see https://platform.openai.com/docs/api-reference/ for purposes))r r )�func�
args_modelzfiles.retrievez-iz--idz The files IDz files.deletez
files.list)r ) �
add_parser� add_argument� set_defaults�CLIFile�create�CLIFileCreateArgs�get�delete�list)r
�sub� r �H/home/arjun/.local/lib/python3.10/site-packages/openai/cli/_api/files.py�register s,
��
r! c @ s e Zd ZU ded< dS )�
CLIFileIDArgs�str�idN��__name__�
__module__� __qualname__�__annotations__r r r r r" + s
r" c @ s e Zd ZU ded< ded< dS )r r# �file�purposeNr% r r r r r / s
r c @ sD e Zd Zeddd��Zeddd ��Zedd
d ��Zedd d
��ZdS )r �argsr r r c C sh t | jd��}t|�� dd�}W d � n1 sw Y t� jj| j|ftt| j �d�}t
|� d S )N�rbzUpload progress)�desc)r* r+ ) �openr* r �readr �filesr r r r+ r )r, � file_reader�
buffer_readerr* r r r r 5 s �
� zCLIFile.creater" c C � t � jj| jd�}t|� d S �N)�file_id)r r1 �retriever$ r �r, r* r r r r B � z CLIFile.getc C r4 r5 )r r1 r r$ r r8 r r r r G r9 zCLIFile.deletec C s" t � j�� } | D ]}t|� qd S )N)r r1 r r )r1 r* r r r r L s
�z CLIFile.listN)r, r r r )r, r" r r )r r )r& r' r( � staticmethodr r r r r r r r r 4 s r N)r
r r r )�
__future__r �typingr r r �argparser �_utilsr r �_modelsr
� _progressr r r! r" r r r r r r �<module> s