yym68686 commited on
Commit
cd90ffd
·
1 Parent(s): ec11ef5

🐛 Bug: Fix the bug where GitHub models cannot retrieve the model when the model field is not set.

Browse files
Files changed (1) hide show
  1. utils.py +12 -1
utils.py CHANGED
@@ -205,9 +205,10 @@ def update_initial_model(api_url, api):
205
  endpoint_models_url = endpoint.v1_models
206
  if isinstance(api, list):
207
  api = api[0]
 
208
  response = httpx.get(
209
  endpoint_models_url,
210
- headers={"Authorization": f"Bearer {api}"},
211
  )
212
  models = response.json()
213
  if models.get("error"):
@@ -260,6 +261,16 @@ def update_config(config_data, use_config_url=False):
260
  safe_get(provider, "preferences", "api_key_rate_limit", default={"default": "999999/min"})
261
  )
262
 
 
 
 
 
 
 
 
 
 
 
263
  if not provider.get("model"):
264
  model_list = update_initial_model(provider['base_url'], provider['api'])
265
  if model_list:
 
205
  endpoint_models_url = endpoint.v1_models
206
  if isinstance(api, list):
207
  api = api[0]
208
+ headers = {"Authorization": f"Bearer {api}"}
209
  response = httpx.get(
210
  endpoint_models_url,
211
+ headers=headers,
212
  )
213
  models = response.json()
214
  if models.get("error"):
 
261
  safe_get(provider, "preferences", "api_key_rate_limit", default={"default": "999999/min"})
262
  )
263
 
264
+ if "models.inference.ai.azure.com" in provider['base_url'] and not provider.get("model"):
265
+ provider['model'] = [
266
+ "gpt-4o",
267
+ "gpt-4o-mini",
268
+ "o1-mini",
269
+ "o1-preview",
270
+ "text-embedding-3-small",
271
+ "text-embedding-3-large",
272
+ ]
273
+
274
  if not provider.get("model"):
275
  model_list = update_initial_model(provider['base_url'], provider['api'])
276
  if model_list: