alexfremont commited on
Commit
c2cd706
·
1 Parent(s): ec6e669

Fix model logging to use display_name instead of name field

Browse files
Files changed (2) hide show
  1. db/models.py +1 -1
  2. models/loader.py +1 -1
db/models.py CHANGED
@@ -78,7 +78,7 @@ async def fetch_model_by_id(model_id: str) -> Union[Dict[str, Any], None]:
78
  row = await conn.fetchrow(query, model_id)
79
 
80
  if row:
81
- logger.info(f"Found model with ID '{model_id}': {row['name']}")
82
  return dict(row)
83
  else:
84
  logger.warning(f"No model found with ID '{model_id}'")
 
78
  row = await conn.fetchrow(query, model_id)
79
 
80
  if row:
81
+ logger.info(f"Found model with ID '{model_id}': {row['display_name']}")
82
  return dict(row)
83
  else:
84
  logger.warning(f"No model found with ID '{model_id}'")
models/loader.py CHANGED
@@ -106,7 +106,7 @@ async def load_models(models_data: List[Dict[str, Any]]) -> None:
106
  except Exception as e:
107
  # Log l'échec mais continue avec les autres modèles
108
  logger.error(f"Failed to load model ID {model_id}: {e}")
109
- failed_models.append(model_data.get('name', f'ID {model_id}'))
110
 
111
  logger.info(f"Finished loading models. Successfully loaded: {loaded_count}/{len(models_data)}")
112
  if failed_models:
 
106
  except Exception as e:
107
  # Log l'échec mais continue avec les autres modèles
108
  logger.error(f"Failed to load model ID {model_id}: {e}")
109
+ failed_models.append(model_data.get('display_name', f'ID {model_id}'))
110
 
111
  logger.info(f"Finished loading models. Successfully loaded: {loaded_count}/{len(models_data)}")
112
  if failed_models: