From 5c5f4426494ad24e168e1c93f601a78feff37b43 Mon Sep 17 00:00:00 2001 From: jiangfy-ihep <108909320+jiangfy-ihep@users.noreply.github.com> Date: Thu, 18 Apr 2024 16:24:29 +0800 Subject: [PATCH] Fix: openai project API key pattern (#1721) Co-authored-by: Fayu Jiang --- shared_utils/key_pattern_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/shared_utils/key_pattern_manager.py b/shared_utils/key_pattern_manager.py index 44ad949..8feb064 100644 --- a/shared_utils/key_pattern_manager.py +++ b/shared_utils/key_pattern_manager.py @@ -14,7 +14,7 @@ def is_openai_api_key(key): if len(CUSTOM_API_KEY_PATTERN) != 0: API_MATCH_ORIGINAL = re.match(CUSTOM_API_KEY_PATTERN, key) else: - API_MATCH_ORIGINAL = re.match(r"sk-[a-zA-Z0-9]{48}$|sess-[a-zA-Z0-9]{40}$", key) + API_MATCH_ORIGINAL = re.match(r"sk-[a-zA-Z0-9]{48}$|sk-proj-[a-zA-Z0-9]{48}$|sess-[a-zA-Z0-9]{40}$", key) return bool(API_MATCH_ORIGINAL)