Benedict King commited on
Commit
00ec216
·
1 Parent(s): 19a2bc1

if api key is invalid, raise except

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
utils.py CHANGED
@@ -46,6 +46,8 @@ def update_initial_model(api_url, api):
46
  headers={"Authorization": f"Bearer {api}"},
47
  )
48
  models = response.json()
 
 
49
  # print(models)
50
  models_list = models["data"]
51
  models_id = [model["id"] for model in models_list]
 
46
  headers={"Authorization": f"Bearer {api}"},
47
  )
48
  models = response.json()
49
+ if models.get("error"):
50
+ raise Exception({"error": models.get("error"), "endpoint": endpoint_models_url, "api": api})
51
  # print(models)
52
  models_list = models["data"]
53
  models_id = [model["id"] for model in models_list]