aaaaasss/0406
0
1a
2 7*d� � @ s@ d dl mZmZmZmZ d dlZd dlZG dd� d�Ze� ZdS )� )�COMPLETION_URL�BALANCE_API_URL�
USAGE_API_URL�API_HOSTNc @ sX e Zd ZdZdZeZeZe Z3dd� Z dd� Z dd� Z
dd � Zd4d � Zd d
� Zdd� ZdS )�StateFc C s5 d| _ d S �NT�� interrupted��self� r �G/Users/treediagram/Documents/Developer/ChuanhuChatGPT/modules/shared.py� interrupt s zState.interruptc C s6 d| _ d S �NFr r7 r r r
�recover s z
State.recoverc C s@ d|� d�| _ d|� d�| _d|� d�| _d|� d�tjd< d S )N�https://z/v1/chat/completionsz /dashboard/billing/credit_grantsz/dashboard/billing/usage�/v1�OPENAI_API_BASE)�completion_url�balance_api_url�
usage_api_url�os�environ)r �api_hostr r r
� set_api_host s zState.set_api_hostc C s( t | _t| _t| _dt� d�tjd< tS )Nr r r ) r r r r r r r r r r8 r r r
�reset_api_host s9 zState.reset_api_hostc C s d| _ t| _d S r )r r r r10 r r r
� reset_all s zState.reset_allc C s* d| _ t�� | _|D ]}| j�|� qd S r )�
multi_api_key�queue�Queue�
api_key_queue�put)r � api_key_list�api_keyr r r
�set_api_key_queue# s 11zState.set_api_key_queuec s t �d�s� S � �fdd�}|S )Nr c s@ �j �� }t| �dd � } � |g| �R i |��}�j �|� |S )N� )r �get�listr! )�args�kwargsr# �ret��funcr r r
�wrapped- s12 13 z(State.switching_api_key.<locals>.wrapped)�hasattr)r r, r- r r+ r
�switching_api_key) s 14zState.switching_api_keyN)�__name__�15__module__� __qualname__r r r r r r r r r r r r r r$ r/ r r r r
r s r ) �modules.presetsr r r r r r r �stater r r r
�<module> s 2