superdup95 commited on
Commit
02ffcdc
·
verified ·
1 Parent(s): 4590eea

Update api_usage.py

Browse files
Files changed (1) hide show
  1. api_usage.py +5 -1
api_usage.py CHANGED
@@ -200,7 +200,11 @@ def get_models(session, key, org: str = None):
200
  headers = get_headers(key)
201
  rq = session.get(f"{BASE_URL}/models", headers=headers, timeout=10)
202
  avai_models = rq.json()
203
- return [model["id"] for model in avai_models["data"]] #[model["id"] for model in avai_models["data"] if model["id"] in GPT_TYPES]
 
 
 
 
204
 
205
  def check_key_availability(session, key):
206
  try:
 
200
  headers = get_headers(key)
201
  rq = session.get(f"{BASE_URL}/models", headers=headers, timeout=10)
202
  avai_models = rq.json()
203
+ try:
204
+ list_models = [model["id"] for model in avai_models["data"]] #[model["id"] for model in avai_models["data"] if model["id"] in GPT_TYPES]
205
+ except:
206
+ list_models = []
207
+ return list_models
208
 
209
  def check_key_availability(session, key):
210
  try: