phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update check_valid_key

Browse source
This commit is contained in:
ishaan-jaff 2023-09-12 11:48:49 -07:00
parent 05f2f3dde7
commit 8b60440b4b
1 changed files with 4 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

25
litellm/utils.py
View file

@ -1499,16 +1499,16 @@ def valid_model(model):
# check valid api key # check valid api key
def check_valid_key(model: str, api_key: str): def check_valid_key(model: str, api_key: str):
# returns True if key is valid for the model
# returns False if key is invalid for the model
messages = [{"role": "user", "content": "Hey, how's it going?"}] messages = [{"role": "user", "content": "Hey, how's it going?"}]
try: try:
litellm.completion(model=model, messages=messages, api_key=api_key) litellm.completion(model=model, messages=messages, api_key=api_key, max_tokens=10)
return True return True
except AuthenticationError as e: except AuthenticationError as e:
return False return False
except Exception as e: except Exception as e:
raise ValueError(str(e)) return False
# integration helper function # integration helper function
def modify_integration(integration_name, integration_params): def modify_integration(integration_name, integration_params):
@ -2606,23 +2606,6 @@ def trim_messages(
except: # [NON-Blocking, if error occurs just return final_messages except: # [NON-Blocking, if error occurs just return final_messages
return 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 True
except:
return False
# this helper reads the .env and returns a list of supported llms for user # this helper reads the .env and returns a list of supported llms for user
def get_valid_models(): def get_valid_models():
try: try: