o "i@sRddlZddlmZddlmZddlmZddlmZdede de fd d Z dS) N)Session)SFTTask)DPOTask)TrainingManagerdbtask_id task_typec st|} z<|dkr|ttj|k}n |ttj|k}|||IdH|r6|jdvr9WdSt dIdHWn$t yf}zt d|d|t dIdHWYd}~nd}~wwq) z'Background task to sync training statusTsftN) completedfailed cancelledzError syncing task z: ) rqueryrfilteridfirstrupdate_task_statusstatusasynciosleep Exceptionprint)rrrmanagertaskerL/lsinfo/ai/hellotax_ai/training_center/backend/app/services/training_sync.pysync_training_statuss"r) rsqlalchemy.ormrapp.models.sft_taskrapp.models.dpo_taskrapp.services.training_managerrintstrrrrrrs