import importlib import time import os from functools import lru_cache from colorful import print亮红, print亮绿, print亮蓝 pj = os.path.join default_user_name = 'default_user' def read_env_variable(arg, default_value): """ 环境变量可以是 `GPT_ACADEMIC_CONFIG`(优先),也可以直接是`CONFIG` 例如在windows cmd中,既可以写: set USE_PROXY=True set API_KEY=sk-j7caBpkRoxxxxxxxxxxxxxxxxxxxxxxxxxxxx set proxies={"http":"http://127.0.0.1:10085", "https":"http://127.0.0.1:10085",} set AVAIL_LLM_MODELS=["gpt-3.5-turbo", "chatglm"] set AUTHENTICATION=[("username", "password"), ("username2", "password2")] 也可以写: set GPT_ACADEMIC_USE_PROXY=True set GPT_ACADEMIC_API_KEY=sk-j7caBpkRoxxxxxxxxxxxxxxxxxxxxxxxxxxxx set GPT_ACADEMIC_proxies={"http":"http://127.0.0.1:10085", "https":"http://127.0.0.1:10085",} set GPT_ACADEMIC_AVAIL_LLM_MODELS=["gpt-3.5-turbo", "chatglm"] set GPT_ACADEMIC_AUTHENTICATION=[("username", "password"), ("username2", "password2")] """ arg_with_prefix = "GPT_ACADEMIC_" + arg if arg_with_prefix in os.environ: env_arg = os.environ[arg_with_prefix] elif arg in os.environ: env_arg = os.environ[arg] else: raise KeyError print(f"[ENV_VAR] 尝试加载{arg},默认值:{default_value} --> 修正值:{env_arg}") try: if isinstance(default_value, bool): env_arg = env_arg.strip() if env_arg == 'True': r = True elif env_arg == 'False': r = False else: print('Enter True or False, but have:', env_arg); r = default_value elif isinstance(default_value, int): r = int(env_arg) elif isinstance(default_value, float): r = float(env_arg) elif isinstance(default_value, str): r = env_arg.strip() elif isinstance(default_value, dict): r = eval(env_arg) elif isinstance(default_value, list): r = eval(env_arg) elif default_value is None: assert arg == "proxies" r = eval(env_arg) else: print亮红(f"[ENV_VAR] 环境变量{arg}不支持通过环境变量设置! ") raise KeyError except: print亮红(f"[ENV_VAR] 环境变量{arg}加载失败! ") raise KeyError(f"[ENV_VAR] 环境变量{arg}加载失败! ") print亮绿(f"[ENV_VAR] 成功读取环境变量{arg}") return r @lru_cache(maxsize=128) def read_single_conf_with_lru_cache(arg): from shared_utils.key_pattern_manager import is_any_api_key try: # 优先级1. 获取环境变量作为配置 default_ref = getattr(importlib.import_module('config'), arg) # 读取默认值作为数据类型转换的参考 r = read_env_variable(arg, default_ref) except: # try: # # 优先级2. 获取config_private中的配置 # r = getattr(importlib.import_module('config_private'), arg) # except: # # 优先级3. 获取config中的配置 r = getattr(importlib.import_module('config'), arg) # 在读取API_KEY时,检查一下是不是忘了改config if arg == 'API_URL_REDIRECT': oai_rd = r.get("https://api.openai.com/v1/chat/completions", None) # API_URL_REDIRECT填写格式是错误的,请阅读`https://github.com/binary-husky/gpt_academic/wiki/项目配置说明` if oai_rd and not oai_rd.endswith('/completions'): print亮红("\n\n[API_URL_REDIRECT] API_URL_REDIRECT填错了。请阅读`https://github.com/binary-husky/gpt_academic/wiki/项目配置说明`。如果您确信自己没填错,无视此消息即可。") time.sleep(5) if arg == 'API_KEY': print亮蓝(f"[API_KEY] 本项目现已支持OpenAI和Azure的api-key。也支持同时填写多个api-key,如API_KEY=\"openai-key1,openai-key2,azure-key3\"") print亮蓝(f"[API_KEY] 您既可以在config.py中修改api-key(s),也可以在问题输入区输入临时的api-key(s),然后回车键提交后即可生效。") if is_any_api_key(r): print亮绿(f"[API_KEY] 您的 API_KEY 是: {r[:15]}*** API_KEY 导入成功") else: print亮红("[API_KEY] 您的 API_KEY 不满足任何一种已知的密钥格式,请在config文件中修改API密钥之后再运行。") if arg == 'proxies': if not read_single_conf_with_lru_cache('USE_PROXY'): r = None # 检查USE_PROXY,防止proxies单独起作用 if r is None: print亮红('[PROXY] 网络代理状态:未配置。无代理状态下很可能无法访问OpenAI家族的模型。建议:检查USE_PROXY选项是否修改。') else: print亮绿('[PROXY] 网络代理状态:已配置。配置信息如下:', r) assert isinstance(r, dict), 'proxies格式错误,请注意proxies选项的格式,不要遗漏括号。' return r @lru_cache(maxsize=128) def get_conf(*args): """ 本项目的所有配置都集中在config.py中。 修改配置有三种方法,您只需要选择其中一种即可: - 直接修改config.py - 创建并修改config_private.py - 修改环境变量(修改docker-compose.yml等价于修改容器内部的环境变量) 注意:如果您使用docker-compose部署,请修改docker-compose(等价于修改容器内部的环境变量) """ res = [] for arg in args: r = read_single_conf_with_lru_cache(arg) res.append(r) if len(res) == 1: return res[0] return res def set_conf(key, value): from toolbox import read_single_conf_with_lru_cache read_single_conf_with_lru_cache.cache_clear() get_conf.cache_clear() os.environ[key] = str(value) altered = get_conf(key) return altered def set_multi_conf(dic): for k, v in dic.items(): set_conf(k, v) return