Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
e241d94
1
Parent(s):
078f3c3
Remove system status endpoint and monitoring functionality
Browse files- api/management.py +0 -23
api/management.py
CHANGED
@@ -5,7 +5,6 @@ from fastapi import APIRouter, HTTPException, Depends
|
|
5 |
from api.dependencies import verify_management_api_key
|
6 |
from db.models import fetch_model_by_id
|
7 |
from models.loader import model_pipelines, _load_single_model_pipeline
|
8 |
-
from utils.system_monitor import get_system_status
|
9 |
|
10 |
logger = logging.getLogger(__name__)
|
11 |
|
@@ -117,25 +116,3 @@ async def delete_single_model(model_db_id: Any):
|
|
117 |
if model_db_id in model_pipelines:
|
118 |
del model_pipelines[model_db_id]
|
119 |
raise HTTPException(status_code=500, detail=f"Internal server error while unloading model {model_db_id}")
|
120 |
-
|
121 |
-
@router.get("/system_status", summary="Get system resource usage information")
|
122 |
-
async def get_resource_usage():
|
123 |
-
"""Récupère des informations sur l'utilisation des ressources système.
|
124 |
-
|
125 |
-
Fournit des détails sur la mémoire disponible, l'utilisation CPU,
|
126 |
-
les GPUs (si disponibles) et l'utilisation mémoire des modèles chargés.
|
127 |
-
|
128 |
-
Returns:
|
129 |
-
Un dictionnaire contenant les informations de ressources système
|
130 |
-
"""
|
131 |
-
try:
|
132 |
-
# Obtenir toutes les informations système, y compris les modèles chargés
|
133 |
-
status = get_system_status(model_pipelines)
|
134 |
-
|
135 |
-
# Ajouter des statistiques sur les modèles
|
136 |
-
status["loaded_models_count"] = len(model_pipelines)
|
137 |
-
|
138 |
-
return status
|
139 |
-
except Exception as e:
|
140 |
-
logger.exception(f"Error getting system status: {e}")
|
141 |
-
raise HTTPException(status_code=500, detail=f"Failed to retrieve system status: {str(e)}")
|
|
|
5 |
from api.dependencies import verify_management_api_key
|
6 |
from db.models import fetch_model_by_id
|
7 |
from models.loader import model_pipelines, _load_single_model_pipeline
|
|
|
8 |
|
9 |
logger = logging.getLogger(__name__)
|
10 |
|
|
|
116 |
if model_db_id in model_pipelines:
|
117 |
del model_pipelines[model_db_id]
|
118 |
raise HTTPException(status_code=500, detail=f"Internal server error while unloading model {model_db_id}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|