Checking onboardin flow - fix 2

services/training/app/schemas/training.py

@@ -8,6 +8,7 @@ from pydantic import BaseModel, Field, validator
 from typing import List, Optional, Dict, Any, Union
 from datetime import datetime
 from enum import Enum
+from uuid import UUID
 
 
 class TrainingStatus(str, Enum):
@@ -58,7 +59,6 @@ class SingleProductTrainingRequest(BaseModel):
     weekly_seasonality: bool = Field(True, description="Enable weekly seasonality")
     yearly_seasonality: bool = Field(True, description="Enable yearly seasonality")
 
-
 class TrainingJobResponse(BaseModel):
     """Response schema for training job creation"""
     job_id: str = Field(..., description="Unique training job identifier")
@@ -67,7 +67,17 @@ class TrainingJobResponse(BaseModel):
     tenant_id: str = Field(..., description="Tenant identifier")
     created_at: datetime = Field(..., description="Job creation timestamp")
     estimated_duration_minutes: int = Field(..., description="Estimated completion time in minutes")
-
+    
+    # ✅ FIX: Add custom validator to convert UUID to string
+    @validator('tenant_id', 'job_id', pre=True)
+    def convert_uuid_to_string(cls, v):
+        """Convert UUID objects to strings for JSON serialization"""
+        if isinstance(v, UUID):
+            return str(v)
+        return v
+    
+    class Config:
+        from_attributes = True
 
 class TrainingJobStatus(BaseModel):
     """Response schema for training job status checks"""
@@ -81,6 +91,16 @@ class TrainingJobStatus(BaseModel):
     products_completed: int = Field(0, description="Number of products completed")
     products_failed: int = Field(0, description="Number of products that failed")
     error_message: Optional[str] = Field(None, description="Error message if failed")
+    
+    @validator('job_id', pre=True)
+    def convert_uuid_to_string(cls, v):
+        """Convert UUID objects to strings for JSON serialization"""
+        if isinstance(v, UUID):
+            return str(v)
+        return v
+    
+    class Config:
+        from_attributes = True
 
 
 class TrainingJobProgress(BaseModel):
@@ -94,6 +114,16 @@ class TrainingJobProgress(BaseModel):
     products_total: int = Field(0, description="Total number of products")
     estimated_time_remaining_minutes: Optional[int] = Field(None, description="Estimated time remaining")
     timestamp: datetime = Field(default_factory=datetime.now, description="Progress update timestamp")
+    
+    @validator('job_id', pre=True)
+    def convert_uuid_to_string(cls, v):
+        """Convert UUID objects to strings for JSON serialization"""
+        if isinstance(v, UUID):
+            return str(v)
+        return v
+    
+    class Config:
+        from_attributes = True
 
 
 class DataValidationRequest(BaseModel):
@@ -157,6 +187,16 @@ class TrainingResultsResponse(BaseModel):
     training_results: Dict[str, ProductTrainingResult] = Field(..., description="Per-product results")
     summary: Dict[str, Any] = Field(..., description="Training summary statistics")
     completed_at: datetime = Field(..., description="Job completion timestamp")
+    
+    @validator('tenant_id', 'job_id', pre=True)
+    def convert_uuid_to_string(cls, v):
+        """Convert UUID objects to strings for JSON serialization"""
+        if isinstance(v, UUID):
+            return str(v)
+        return v
+    
+    class Config:
+        from_attributes = True
 
 
 class TrainingValidationResult(BaseModel):
@@ -232,6 +272,16 @@ class TrainedModelResponse(BaseModel):
     created_at: datetime = Field(..., description="Model creation timestamp")
     data_period_start: Optional[datetime] = Field(None, description="Training data start date")
     data_period_end: Optional[datetime] = Field(None, description="Training data end date")
+    
+    @validator('tenant_id', 'model_id', pre=True)
+    def convert_uuid_to_string(cls, v):
+        """Convert UUID objects to strings for JSON serialization"""
+        if isinstance(v, UUID):
+            return str(v)
+        return v
+    
+    class Config:
+        from_attributes = True
 
 
 class ModelTrainingStats(BaseModel):