Dwild/ChuanhuChatGPT
0
1a
2 �K,d� � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlZddlm Z g d�Z3ej � d�r�e
ddd d4��Ze�e�ZW d � q�1 s�0 Y ni Ze�d d �Zej�d
�dkr�dZe�dd�Zej�de�Ze�dd �Ze�r e�dg �Zee�d k�re�d� e�d� e j�e� e�dg �Z ee �d kZ ej�de�dd��Z e �r\e j� e � e�r�edk�r�e�d� e�d� ej�d�Z ej�d�Z!e"e e#d���s�e"e!e#d���s�e �$ej�d�ej�d�f� dZ �ne�s8ej � d ��r8ej �%d ��r8e
d d�� Ze�&� �'� ZW d � n1 �s.0 Y ej � d ��r�dZ e
d dd d5��vZe�e�Z(e(D ]VZ)e(e) d �r�e(e) d �r�e �$e(e) d e(e) d f� ne�d � e�d� �qhW d � n1 �s�0 Y ed2d!d"��Z*e�d#d$�Z+ej,e+d%d&� e�d'd�a-e�d(d�a.ej�d)t-�a-ej�d*t.�a.dejd)< dejd*< ed3d+d,��Z/ed-d.� �a0t0�1e�d/i �� d0d1� Z2dS )4� )� defaultdict)�contextmanagerN� )�shared) �6my_api_key�authflag� auth_list�7dockerflag�retrieve_proxy� log_level� advance_docs�update_doc_config�
multi_api_keyz config.json�rzutf-8)�encodingr FZ dockerrun�yesTZopenai_api_key� r r � api_key_listuG 多账号模式已开启，但api_key_list为空，请检查config.json�users�api_host�emptyzPlease give a api key!�USERNAMEZPASSWORDz api_key.txtz auth.json�username�passwordu3 请检查auth.json文件中的用户名和密码！c c sF t j�dd�}| d u r(tt jd< tV n| t jd< | V |t jd< d S )NZOPENAI_API_KEYr )�os�environ�getr )�api_keyZ old_api_key� r �D/Users/treediagram/Documents/Developer/huggingface/modules/config.py�retrieve_openai_apiV s 89r r �INFOzA%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s)�level�format�10http_proxy� https_proxy�11HTTP_PROXY� HTTPS_PROXYc c sf | dur | a | at tfV nFtjd tjd f}t tjd< ttjd< t tfV |\tjd< tjd< dS )u� 12 1, 如果proxy = NONE，设置环境变量，并返回最新设置的代理13 2，如果proxy ！= NONE，更新当前的代理配置，但是不更新环境变量14 Nr& r' )r$ r% r r )�proxyZold_varr r r r15 r s 161718r19 c C s t t�S )N)r �dictr r r r �<lambda>� � r* r c C s2 | rdt d d< n dt d d< t�dt � �� d S )NT�pdf�20two_columnFu 更新后的文件参数为：)r �logging�info)Ztwo_column_pdfr r r r
� s r
)N)N)3� collectionsr �21contextlibr r r. �sys�jsonr r �__all__�path�exists�open�f�load�configr r r r r r �len�error�exit�stateZset_api_key_queuer r r Z set_api_hostr r �22isinstance�type�append�getsize�read�strip�auth�_r r � basicConfigr$ r% r23 r �updater
r r r r �<module> s� * 2425 262728 �29�30� ,31 32. � 3334