bakery-admin/bakery-ia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve the event messaging for training service

Browse Source
This commit is contained in:
Urtzi Alfaro
2025-07-30 21:21:02 +02:00
parent 5e3fbc5493
commit 923b2d48d2
3 changed files with 428 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
services/training/app/services/training_service.py
View File

@@ -61,7 +61,16 @@ class TrainingService:
logger.info(f"Starting training job {job_id} for tenant {tenant_id}")
from app.services.messaging import TrainingStatusPublisher
status_publisher = TrainingStatusPublisher(job_id, tenant_id)
try:
await status_publisher.job_started({
"bakery_location": bakery_location,
"has_custom_dates": bool(requested_start or requested_end)
}, 0) # Will be updated when we know product count
# Step 1: Prepare training dataset with date alignment and orchestration
logger.info("Step 1: Preparing and aligning training data")
training_dataset = await self.orchestrator.prepare_training_data(
@@ -101,10 +110,12 @@ class TrainingService:
}
logger.info(f"Training job {job_id} completed successfully")
await status_publisher.job_completed(final_result)
return TrainingService.create_detailed_training_response(final_result)
except Exception as e:
logger.error(f"Training job {job_id} failed: {str(e)}")
await status_publisher.job_failed(str(e))
# Return error response in same detailed format
final_result = {
"job_id": job_id,