Add role-based filtering and imporve code

services/training/app/models/__init__.py

@@ -4,6 +4,13 @@ Training Service Models Package
 Import all models to ensure they are registered with SQLAlchemy Base.
 """
 
+# Import AuditLog model for this service
+from shared.security import create_audit_log_model
+from shared.database.base import Base
+
+# Create audit log model for this service
+AuditLog = create_audit_log_model(Base)
+
 # Import all models to register them with the Base metadata
 from .training import (
     TrainedModel,
@@ -20,4 +27,5 @@ __all__ = [
     "ModelPerformanceMetric",
     "TrainingJobQueue",
     "ModelArtifact",
+    "AuditLog",
 ]