Add load_errors to health endpoint for debugging

ml_service/app/api/routes.py

@@ -70,14 +70,15 @@ async def predict(request: PredictRequest):
 async def health_check() -> Dict[str, Any]:
     return {
         "status": "healthy",
-        "backend": False, # Gateway sets this to true later
+        "backend": False,
         "ml_service": True,
-        "database": False, # Gateway sets this to true later
+        "database": False,
         "models_loaded": {
             "logistic_regression": model_manager.models_loaded.get("logistic_regression", False),
             "lstm": model_manager.models_loaded.get("lstm", False),
             "bert": model_manager.models_loaded.get("bert", False)
         },
+        "load_errors": model_manager.load_errors,
         "version": "1.0.0"
     }
 

ml_service/app/services/model_loader.py

@@ -34,6 +34,7 @@ class ModelManager:
             "lstm": False,
             "bert": False
         }
+        self.load_errors = {}
         
         # LR specific
         self.lr_model = None
@@ -89,6 +90,7 @@ class ModelManager:
                 self.lr_model = joblib.load(model_path)
                 self.models_loaded["logistic_regression"] = True
         except Exception as e:
+            self.load_errors["lr"] = str(e)
             print(f"Failed to load Logistic Regression: {e}")
 
     def load_lstm(self):
@@ -126,6 +128,7 @@ class ModelManager:
                     self.lstm_config = pickle.load(f)
                 self.models_loaded["lstm"] = True
         except Exception as e:
+            self.load_errors["lstm"] = str(e)
             print(f"Failed to load LSTM: {e}")
 
     def load_bert(self):
@@ -143,6 +146,7 @@ class ModelManager:
                         
                 self.models_loaded["bert"] = True
         except Exception as e:
+            self.load_errors["bert"] = str(e)
             print(f"Failed to load BERT: {e}")
 
 model_manager = ModelManager()