yym68686 commited on
Commit
141a8b8
·
1 Parent(s): 501b1f1

Default to all models if model field is not set

Browse files

---

For more details, open the [Copilot Workspace session](https://copilot-workspace.githubnext.com/yym68686/uni-api?shareId=XXXX-XXXX-XXXX-XXXX).

Files changed (1) hide show
  1. utils.py +5 -1
utils.py CHANGED
@@ -119,6 +119,10 @@ def update_config(config_data, use_config_url=False):
119
  config_data['api_keys'][index]['weights'] = weights_dict
120
  config_data['api_keys'][index]['model'] = models
121
  api_keys_db[index]['model'] = models
 
 
 
 
122
 
123
  api_list = [item["api"] for item in api_keys_db]
124
  # logger.info(json.dumps(config_data, indent=4, ensure_ascii=False))
@@ -385,4 +389,4 @@ def safe_get(data, *keys, default=None):
385
  data = data[key] if isinstance(data, (dict, list)) else data.get(key)
386
  except (KeyError, IndexError, AttributeError, TypeError):
387
  return default
388
- return data
 
119
  config_data['api_keys'][index]['weights'] = weights_dict
120
  config_data['api_keys'][index]['model'] = models
121
  api_keys_db[index]['model'] = models
122
+ else:
123
+ # Default to all models if 'model' field is not set
124
+ config_data['api_keys'][index]['model'] = ["all"]
125
+ api_keys_db[index]['model'] = ["all"]
126
 
127
  api_list = [item["api"] for item in api_keys_db]
128
  # logger.info(json.dumps(config_data, indent=4, ensure_ascii=False))
 
389
  data = data[key] if isinstance(data, (dict, list)) else data.get(key)
390
  except (KeyError, IndexError, AttributeError, TypeError):
391
  return default
392
+ return data