maximedenes commited on
Commit
87189e1
·
1 Parent(s): 8e3c82c

Anonymize models in finished queue

Browse files
Files changed (1) hide show
  1. backend/app/api/endpoints/models.py +10 -0
backend/app/api/endpoints/models.py CHANGED
@@ -5,6 +5,7 @@ from app.services.models import ModelService
5
  from app.api.dependencies import get_model_service
6
  from app.core.fastapi_cache import cached
7
  from app.core.formatting import LogFormatter
 
8
 
9
  logger = logging.getLogger(__name__)
10
  router = APIRouter(tags=["models"])
@@ -18,6 +19,15 @@ async def get_models_status(
18
  try:
19
  logger.info(LogFormatter.info("Fetching status for all models"))
20
  result = await model_service.get_models()
 
 
 
 
 
 
 
 
 
21
  stats = {
22
  status: len(models) for status, models in result.items()
23
  }
 
5
  from app.api.dependencies import get_model_service
6
  from app.core.fastapi_cache import cached
7
  from app.core.formatting import LogFormatter
8
+ import json
9
 
10
  logger = logging.getLogger(__name__)
11
  router = APIRouter(tags=["models"])
 
19
  try:
20
  logger.info(LogFormatter.info("Fetching status for all models"))
21
  result = await model_service.get_models()
22
+
23
+ # TODO remove is model anonymisation is no longer wanted
24
+ result = json.loads(json.dumps(result)) # Thread-safe deep copy
25
+ for status, models in result.items():
26
+ if status == "finished":
27
+ for i, model in enumerate(models):
28
+ model["name"] = "(modèle non révélé)"
29
+ model["revision"] = "(...)"
30
+
31
  stats = {
32
  status: len(models) for status, models in result.items()
33
  }