From 45f2ef124a00c5c66b72702f21f695ea8015469c Mon Sep 17 00:00:00 2001 From: based Date: Fri, 1 Mar 2024 12:23:55 +1000 Subject: [PATCH] update main.py --- main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9971486..6bc2fb6 100644 --- a/main.py +++ b/main.py @@ -147,6 +147,7 @@ def validate_vertexai(key: APIKey): oai_regex = re.compile('(sk-[A-Za-z0-9]{20}T3BlbkFJ[A-Za-z0-9]{20})') anthropic_regex = re.compile(r'sk-ant-api03-[A-Za-z0-9\-_]{93}AA') +anthropic_secondary_regex = re.compile(r'sk-ant-[A-Za-z0-9\-_]{86}') ai21_and_mistral_regex = re.compile('[A-Za-z0-9]{32}') makersuite_regex = re.compile(r'AIzaSy[A-Za-z0-9\-_]{33}') aws_regex = re.compile(r'^(AKIA[0-9A-Z]{16}):([A-Za-z0-9+/]{40})$') @@ -167,8 +168,8 @@ async def validate_keys(): continue key_obj = APIKey(Provider.VERTEXAI, key) futures.append(executor.submit(validate_vertexai, key_obj)) - elif "ant-api03" in key: - match = anthropic_regex.match(key) + elif "sk-ant-" in key[:7]: + match = anthropic_regex.match(key) if "ant-api03" in key else anthropic_secondary_regex.match(key) if not match: continue key_obj = APIKey(Provider.ANTHROPIC, key)