Skip to content

Commit 442c10b

Browse files
saberustersonichi
andauthored
support openai 'sk-proj-' apikey format (#2426)
Co-authored-by: Chi Wang <[email protected]>
1 parent 2df0f39 commit 442c10b

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

autogen/oai/openai_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def is_valid_api_key(api_key: str) -> bool:
9090
Returns:
9191
bool: A boolean that indicates if input is valid OpenAI API key.
9292
"""
93-
api_key_re = re.compile(r"^sk-[A-Za-z0-9]{32,}$")
93+
api_key_re = re.compile(r"^sk-(proj-)?[A-Za-z0-9]{32,}$")
9494
return bool(re.fullmatch(api_key_re, api_key))
9595

9696

test/oai/test_utils.py

+1
Original file line numberDiff line numberDiff line change
@@ -382,6 +382,7 @@ def test_is_valid_api_key():
382382
assert not is_valid_api_key("sk-asajsdjsd22372%23kjdfdfdf2329ffUUDSDS")
383383
assert is_valid_api_key("sk-asajsdjsd22372X23kjdfdfdf2329ffUUDSDS")
384384
assert is_valid_api_key("sk-asajsdjsd22372X23kjdfdfdf2329ffUUDSDS1212121221212sssXX")
385+
assert is_valid_api_key("sk-proj-asajsdjsd22372X23kjdfdfdf2329ffUUDSDS12121212212")
385386
assert is_valid_api_key(MOCK_OPEN_AI_API_KEY)
386387

387388

0 commit comments

Comments
 (0)