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

Checking onboardin flow - fix 4

Browse Source
This commit is contained in:
Urtzi Alfaro
2025-07-27 16:29:53 +02:00
parent 0b14cf9eb2
commit e63a99b818
8 changed files with 497 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -19,6 +19,7 @@ from app.schemas.training import TrainingJobRequest, SingleProductTrainingReques
from app.services.messaging import publish_job_completed, publish_job_failed
from app.core.config import settings
from shared.monitoring.metrics import MetricsCollector
from app.services.data_client import DataServiceClient
logger = logging.getLogger(__name__)
metrics = MetricsCollector("training-service")
@@ -31,6 +32,7 @@ class TrainingService:
def __init__(self):
self.ml_trainer = BakeryMLTrainer()
self.data_client = DataServiceClient()
async def execute_training_job_simple(self, job_id: str, tenant_id_str: str, request: TrainingJobRequest):
"""Simple wrapper that creates its own database session"""
@@ -136,7 +138,7 @@ class TrainingService:
await self._update_job_status(db, job_id, "running", 5, "Fetching training data")
# Fetch sales data from data service
sales_data = await self._fetch_sales_data(tenant_id, request)
sales_data = await self.data_client.fetch_sales_data(tenant_id)
# Fetch external data if requested
weather_data = []
@@ -144,11 +146,11 @@ class TrainingService:
if request.include_weather:
await self._update_job_status(db, job_id, "running", 15, "Fetching weather data")
weather_data = await self._fetch_weather_data(tenant_id, request)
weather_data = await self.data_client.fetch_weather_data(tenant_id)
if request.include_traffic:
await self._update_job_status(db, job_id, "running", 25, "Fetching traffic data")
traffic_data = await self._fetch_traffic_data(tenant_id, request)
traffic_data = await self.data_client.fetch_traffic_data(tenant_id)
# Execute ML training
await self._update_job_status(db, job_id, "running", 35, "Processing training data")