Add user role
This commit is contained in:
@@ -442,6 +442,24 @@ async def publish_data_validation_completed(
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
async def publish_models_deleted_event(tenant_id: str, deletion_stats: Dict[str, Any]):
|
||||
"""Publish models deletion event to message queue"""
|
||||
try:
|
||||
await training_publisher.publish_event(
|
||||
exchange="training_events",
|
||||
routing_key="training.tenant.models.deleted",
|
||||
message={
|
||||
"event_type": "tenant_models_deleted",
|
||||
"tenant_id": tenant_id,
|
||||
"timestamp": datetime.utcnow().isoformat(),
|
||||
"deletion_stats": deletion_stats
|
||||
}
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error("Failed to publish models deletion event", error=str(e))
|
||||
|
||||
|
||||
# =========================================
|
||||
# UTILITY FUNCTIONS FOR BATCH PUBLISHING
|
||||
# =========================================
|
||||
@@ -549,4 +567,5 @@ class TrainingStatusPublisher:
|
||||
async def job_failed(self, error: str, error_details: Optional[Dict] = None):
|
||||
"""Publish job failure with clean error details"""
|
||||
clean_error_details = safe_json_serialize(error_details) if error_details else None
|
||||
await publish_job_failed(self.job_id, self.tenant_id, error, clean_error_details)
|
||||
await publish_job_failed(self.job_id, self.tenant_id, error, clean_error_details)
|
||||
|
||||
Reference in New Issue
Block a user