chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

使用 re 模块的 match 函数可以更精准的匹配和确认 API_KEY 是否正确

Browse Source
This commit is contained in:
Jia Xinglong 2023-03-31 17:38:39 +08:00
parent 3cc0635628
commit 808e23c98a
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
toolbox.py
View File

@ -1,6 +1,7 @@
import markdown, mdtex2html, threading, importlib, traceback import markdown, mdtex2html, threading, importlib, traceback
from show_math import convert as convert_math from show_math import convert as convert_math
from functools import wraps from functools import wraps
import re
def predict_no_ui_but_counting_down(i_say, i_say_show_user, chatbot, top_p, temperature, history=[], sys_prompt=''): def predict_no_ui_but_counting_down(i_say, i_say_show_user, chatbot, top_p, temperature, history=[], sys_prompt=''):
""" """
@ -226,9 +227,14 @@ def get_conf(*args):
except: r = getattr(importlib.import_module('config'), arg) except: r = getattr(importlib.import_module('config'), arg)
res.append(r) res.append(r)
# 在读取API_KEY时，检查一下是不是忘了改config # 在读取API_KEY时，检查一下是不是忘了改config
if arg=='API_KEY' and len(r) != 51: if arg=='API_KEY':
assert False, "正确的API_KEY密钥是51位，请在config文件中修改API密钥, 添加海外代理之后再运行。" + \ # 正确的 API_KEY 是 "sk-" + 48 位大小写字母数字的组合
"（如果您刚更新过代码，请确保旧版config_private文件中没有遗留任何新增键值）" API_MATCH = re.match(r"sk-[a-zA-Z0-9]{48}$", r)
if API_MATCH:
print("您的 API_KEY 是: ", r, "\nAPI_KEY 导入成功")
else:
assert False, "正确的 API_KEY 是 'sk-' + '48 位大小写字母数字' 的组合，请在config文件中修改API密钥, 添加海外代理之后再运行。" + \
"（如果您刚更新过代码，请确保旧版config_private文件中没有遗留任何新增键值）"
return res return res
def clear_line_break(txt): def clear_line_break(txt):