"""Model training wrappers for trajectory_rag."""