New enterprise feature

services/training/app/repositories/artifact_repository.py

@@ -260,44 +260,44 @@ class ArtifactRepository(TrainingBaseRepository):
             base_filters = {}
             if tenant_id:
                 base_filters["tenant_id"] = tenant_id
-            
+
             # Get basic counts
             total_artifacts = await self.count(filters=base_filters)
-            
+
             # Get artifacts by type
             type_query_params = {}
             type_query_filter = ""
             if tenant_id:
                 type_query_filter = "WHERE tenant_id = :tenant_id"
                 type_query_params["tenant_id"] = tenant_id
-            
+
             type_query = text(f"""
-                SELECT artifact_type, COUNT(*) as count 
-                FROM model_artifacts 
+                SELECT artifact_type, COUNT(*) as count
+                FROM model_artifacts
                 {type_query_filter}
                 GROUP BY artifact_type
                 ORDER BY count DESC
             """)
-            
+
             result = await self.session.execute(type_query, type_query_params)
             artifacts_by_type = {row.artifact_type: row.count for row in result.fetchall()}
-            
+
             # Get storage location stats
             location_query = text(f"""
-                SELECT 
-                    storage_location, 
+                SELECT
+                    storage_location,
                     COUNT(*) as count,
                     SUM(COALESCE(file_size_bytes, 0)) as total_size_bytes
-                FROM model_artifacts 
+                FROM model_artifacts
                 {type_query_filter}
                 GROUP BY storage_location
                 ORDER BY count DESC
             """)
-            
+
             location_result = await self.session.execute(location_query, type_query_params)
             storage_stats = {}
             total_size_bytes = 0
-            
+
             for row in location_result.fetchall():
                 storage_stats[row.storage_location] = {
                     "artifact_count": row.count,
@@ -305,10 +305,10 @@ class ArtifactRepository(TrainingBaseRepository):
                     "total_size_mb": round((row.total_size_bytes or 0) / (1024 * 1024), 2)
                 }
                 total_size_bytes += row.total_size_bytes or 0
-            
+
             # Get expired artifacts count
             expired_artifacts = len(await self.get_expired_artifacts())
-            
+
             return {
                 "total_artifacts": total_artifacts,
                 "expired_artifacts": expired_artifacts,
@@ -321,7 +321,7 @@ class ArtifactRepository(TrainingBaseRepository):
                     "total_size_gb": round(total_size_bytes / (1024 * 1024 * 1024), 2)
                 }
             }
-            
+
         except Exception as e:
             logger.error("Failed to get artifact statistics",
                         tenant_id=tenant_id,