Fix deadlock issues in training

2025-11-05 18:47:20 +01:00
parent fd0a96e254
commit 74215d3e85
3 changed files with 620 additions and 131 deletions
--- a/services/training/app/services/training_service.py
+++ b/services/training/app/services/training_service.py
@@ -244,7 +244,8 @@ class EnhancedTrainingService:
                training_results = await self.trainer.train_tenant_models(
                    tenant_id=tenant_id,
                    training_dataset=training_dataset,
-                    job_id=job_id
+                    job_id=job_id,
+                    session=session  # Pass the main session to avoid nested sessions
                )

                await self.training_log_repo.update_log_progress(