superdup95 commited on
Commit
e6ab908
1 Parent(s): 91a1375

Update api_usage.py

Browse files
Files changed (1) hide show
  1. api_usage.py +1 -1
api_usage.py CHANGED
@@ -84,7 +84,7 @@ def check_gpt4_32k_availability(available_models):
84
  def check_key_availability():
85
  try:
86
  avai_models = openai.Model.list()
87
- return [model["root"] for model in avai_models["data"] if model["root"] in GPT_TYPES]
88
  except:
89
  return False
90
 
 
84
  def check_key_availability():
85
  try:
86
  avai_models = openai.Model.list()
87
+ return [model["id"] for model in avai_models["data"] if model["id"] in GPT_TYPES]
88
  except:
89
  return False
90