From 808e23c98af2e88a0259ae84001ed31a3e5fb990 Mon Sep 17 00:00:00 2001
From: Jia Xinglong <jiaxinglong1996@outlook.com>
Date: Fri, 31 Mar 2023 17:38:39 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8=20re=20=E6=A8=A1=E5=9D=97?=
 =?UTF-8?q?=E7=9A=84=20match=20=E5=87=BD=E6=95=B0=E5=8F=AF=E4=BB=A5?=
 =?UTF-8?q?=E6=9B=B4=E7=B2=BE=E5=87=86=E7=9A=84=E5=8C=B9=E9=85=8D=E5=92=8C?=
 =?UTF-8?q?=E7=A1=AE=E8=AE=A4=20API=5FKEY=20=E6=98=AF=E5=90=A6=E6=AD=A3?=
 =?UTF-8?q?=E7=A1=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 toolbox.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/toolbox.py b/toolbox.py
index d96b3f6..75dd8bc 100644
--- a/toolbox.py
+++ b/toolbox.py
@@ -1,6 +1,7 @@
 import markdown, mdtex2html, threading, importlib, traceback
 from show_math import convert as convert_math
 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=''):
     """
@@ -226,9 +227,14 @@ def get_conf(*args):
         except: r = getattr(importlib.import_module('config'), arg)
         res.append(r)
         # 在读取API_KEY时，检查一下是不是忘了改config
-        if arg=='API_KEY' and len(r) != 51:
-            assert False, "正确的API_KEY密钥是51位，请在config文件中修改API密钥, 添加海外代理之后再运行。" + \
-                        "（如果您刚更新过代码，请确保旧版config_private文件中没有遗留任何新增键值）"
+        if arg=='API_KEY':
+            # 正确的 API_KEY 是 "sk-" + 48 位大小写字母数字的组合
+            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
 
 def clear_line_break(txt):