fix: Enhance API key validation in is_any_api_key function (#2113)

This commit is contained in:
Steven Moder 2025-01-29 21:40:30 +08:00 committed by GitHub
parent 5ddd657ebc
commit 44fe78fff5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions

View File

@ -45,6 +45,13 @@ def is_cohere_api_key(key):
def is_any_api_key(key): def is_any_api_key(key):
# key 一般只包含字母、数字、下划线、逗号、中划线
if not re.match(r"^[a-zA-Z0-9_\-,]+$", key):
# 如果配置了 CUSTOM_API_KEY_PATTERN再检查以下以免误杀
if CUSTOM_API_KEY_PATTERN := get_conf('CUSTOM_API_KEY_PATTERN'):
return bool(re.match(CUSTOM_API_KEY_PATTERN, key))
return False
if ',' in key: if ',' in key:
keys = key.split(',') keys = key.split(',')
for k in keys: for k in keys: