diff --git a/frontend/src/api/services/trainingService.ts b/frontend/src/api/services/trainingService.ts index ad6ae4e0..592d3444 100644 --- a/frontend/src/api/services/trainingService.ts +++ b/frontend/src/api/services/trainingService.ts @@ -5,7 +5,7 @@ import { } from '../types/api'; export interface TrainingJobStatus { - id: string; + job_id: string; tenant_id: string; status: 'queued' | 'running' | 'completed' | 'failed' | 'cancelled'; progress: number; @@ -76,7 +76,7 @@ export class TrainingService { * Start new training job */ async startTraining(config: TrainingConfiguration): Promise { - const response = await apiClient.post>( + const response = await apiClient.post( '/api/v1/training/jobs', config ); diff --git a/frontend/src/pages/onboarding.tsx b/frontend/src/pages/onboarding.tsx index d6db140a..8f8b50ed 100644 --- a/frontend/src/pages/onboarding.tsx +++ b/frontend/src/pages/onboarding.tsx @@ -141,7 +141,7 @@ const OnboardingPage = () => { // Update form data with training job ID setFormData(prev => ({ ...prev, - trainingTaskId: trainingJob.id, + trainingTaskId: trainingJob.job_id, trainingStatus: 'running' })); diff --git a/services/training/app/services/training_service.py b/services/training/app/services/training_service.py index 35ef834c..8a863182 100644 --- a/services/training/app/services/training_service.py +++ b/services/training/app/services/training_service.py @@ -411,8 +411,7 @@ class TrainingService: # Call data service to get sales data async with httpx.AsyncClient() as client: params = { - "tenant_id": tenant_id, - "include_all": True + "tenant_id": tenant_id } if hasattr(request, 'start_date') and request.start_date: @@ -425,7 +424,7 @@ class TrainingService: params["limit"] = limit response = await client.get( - f"{settings.DATA_SERVICE_URL}/api/sales", + f"{settings.DATA_SERVICE_URL}/api/v1/sales/", params=params, timeout=30.0 )