mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
util: verify_access_key
Verify that the user provided a valid openai token by creating a request to the openAI endpoint.
This commit is contained in:
parent
c3ed7e2b29
commit
261db15bcf
3 changed files with 23 additions and 0 deletions
|
@ -2554,3 +2554,18 @@ def trim_messages(
|
|||
except: # [NON-Blocking, if error occurs just return final_messages
|
||||
return messages
|
||||
|
||||
|
||||
# Verify that the user has passed in a valid and active api key
|
||||
def verify_access_key(access_key:str):
|
||||
openai.api_key = access_key
|
||||
try:
|
||||
test_response = openai.ChatCompletion.create(
|
||||
model="gpt-3.5-turbo",
|
||||
messages=[
|
||||
{"role": "user", "content": "test"},
|
||||
],
|
||||
max_tokens = 10
|
||||
)
|
||||
return False
|
||||
except:
|
||||
return True
|
Loading…
Add table
Add a link
Reference in a new issue