File: //proc/self/root/usr/local/lib/python3.10/dist-packages/openai/cli/__pycache__/_cli.cpython-310.pyc
o
���g{ � @ sV d dl mZ d dlZd dlZd dlZd dlmZmZmZm Z d dl
m Z d dl Z d dl
Z
d dlZddlmZ ddlmZmZ dd lmZ dd
lmZ dd lmZmZ dd lmZm Z m Z dd
l m Z ddl m!Z! e�"� Z#e�$d�Z%e�&ej'�Z(e(�)e%� e#�*e(� G dd� de �Z+d!dd�Z,d"dd�Z-d#dd �Z.d$d d �Z/e0d kr�e�1e-� � dS dS )%� )� annotationsN)�Any�List�Type�Optional)�ClassVar� )�_tools� )�_ApiType� __version__)�register_commands)�
can_use_http2)�CLIError�
display_error)� PYDANTIC_V2�
ConfigDict� model_parse)� BaseModel)�APIErrorz[%(asctime)s] %(message)sc @ s� e Zd ZU eredd�Zded< n G dd� dej�Z ded< d Z
d
ed < d
ed < d
ed
< d
ed< ded< d Z ded< d Z d
ed< d Z
d
ed< d Zd
ed< d Zded< g Zded< dZded < d S ) � Arguments�ignore)�extrazClassVar[ConfigDict]� model_configc @ s e Zd ZU ejjZded< dS )zArguments.Configr r N)�__name__�
__module__� __qualname__�pydantic�Extrar r �__annotations__� r r �:/usr/local/lib/python3.10/dist-packages/openai/cli/_cli.py�Config% s
r" �int� verbosityNz
Optional[str]�version�api_key�api_base� organizationzOptional[List[str]]�proxyzOptional[_ApiType]�api_type� api_version�azure_endpoint�azure_ad_tokenzOptional[Type[BaseModel]]�
args_modelz List[str]� unknown_argsF�bool�allow_unknown_args)r r r r r r r r �
BaseConfigr" r% r* r+ r, r- r. r/ r1 r r r r! r s&
� r �return�argparse.ArgumentParserc s t jd dd�� � jddddddd � � jd
d d d
� � jdddd
� � jddddd� � jdddd
� � jddtdd d � � jd d d
� � jd d!d
� � jd"d#d
� � jd$d%d&d't d(� d2� fd+d,� } � j| d-� � �� }|jd.d/d
�}t|� |jd0d1d
�}t �||� � S )3N�openai)� description�progz-vz --verbose�countr$ r zSet verbosity.)�action�dest�default�helpz-bz
--api-basezWhat API base url to use.)r< z-kz --api-keyzWhat API key to use.z-pz--proxy�+zWhat proxy to use.)�nargsr< z-oz--organizationzRWhich organization to run as (will use your default organization if not specified)z-tz
--api-type)r5 �azurez4The backend API to call, must be `openai` or `azure`)�type�choicesr< z
--api-versionzvThe Azure API version, e.g. 'https://learn.microsoft.com/en-us/azure/ai-services/openai/reference#rest-api-versioning'z--azure-endpointz<The Azure endpoint, e.g. 'https://endpoint.openai.azure.com'z--azure-ad-tokenzyA token from Azure Active Directory, https://www.microsoft.com/en-us/security/business/identity-access/microsoft-entra-idz-Vz --versionr% z %(prog)s )r9 r% r3 �Nonec s � � � d S �N)�
print_helpr ��parserr r! r< n s z_build_parser.<locals>.help)�func�apizDirect API calls�toolsz!Client side tools for convenience�r3 rB )
�argparse�ArgumentParser� add_argument�strr � set_defaults�add_subparsers�
add_parserr
r )r< �
subparsers�sub_api� sub_toolsr rE r! �
_build_parser> sd ������� rU r# c
C sb zt � W dS tttjfy! } z t| � W Y d } ~ dS d } ~ w ty0 tj� d� Y dS w )Nr �
r )
�_mainr r r �ValidationErrorr �KeyboardInterrupt�sys�stderr�write)�errr r r! �main~ s �� �r^ rF �/tuple[argparse.Namespace, Arguments, list[str]]c C s� dt jv rt j�d�}t jd|� }t j|d � }n t jdd � }g }| �|�\}}|�|� ttt|��}|js=| � � |||fS )Nz--r )
rZ �argv�index�parse_known_args�extendr r �varsr1 �
parse_args)rF �idx�
known_argsr/ �parsed�remaining_unknown�argsr r r! � _parse_args� s
rk rB c C s� t � } t| �\}}}|jdkrtj�d� i }|jd urE|jD ]%}|�d�r(dnd}||v r6td|� d���t j
t � t � |��d�||< q t j
|pJd t� d �}|t_|jrY|jt_|jr`|jt_|jrg|jt_|jd urp|jt_|jd ury|jt_|jd ur�|jt_|jd ur�|jt_z3|jr�|�t|ji d
d � t |�� � D ��d |i��� n|�� W z|� � W d S t y� Y d S w z|� � W w t y� Y w w )
Nr z)Warning: --verbosity isn't supported yet
�httpszhttps://zhttp://z Multiple z0 proxies given - only the last one would be used)r) )�mounts�http2c S s i | ] \}}|d ur||�qS rC r )�.0�key�valuer r r! �
<dictcomp>� s
�z_main.<locals>.<dictcomp>r/ ) rU rk r$ rZ r[ r\ r) �
startswithr �httpx�
HTTPTransport�Proxy�URL�Clientr r5 � http_clientr( r& r' �base_urlr* r, r+ r- r. rG r rd �items�close� Exception)rF rh rj �unknown�proxiesr) rp ry r r r! rW � sp
�
������ �� �rW �__main__)r3 r4 )r3 r# )rF r4 r3 r_ rJ )2�
__future__r rZ �loggingrK �typingr r r r �typing_extensionsr rt r r5 � r r r �_apir
�_utilsr �_errorsr r �_compatr r r �_modelsr � _exceptionsr � getLogger�logger� Formatter� formatter�
StreamHandlerr[ �handler� setFormatter�
addHandlerr rU r^ rk rW r �exitr r r r! �<module> s<
@
F�