fix: resolve compilation errors in training.go - fix xsync.Map API usage and ModelConfig methods

Author: localai-bot

core/services/training.go

@@ -121,7 +121,10 @@ func (s *TrainingJobService) CreateJob(req schema.TrainingJobRequest) (schema.Tr
 
 // GetJob returns a training job by ID
 func (s *TrainingJobService) GetJob(id string) (schema.TrainingJob, bool) {
-	return s.jobs.Get(id)
+	if !s.jobs.Exists(id) {
+		return schema.TrainingJob{}, false
+	}
+	return s.jobs.Get(id), true
 }
 
 // ListJobs returns all training jobs, sorted by creation time (newest first)
@@ -139,17 +142,17 @@ func (s *TrainingJobService) ListJobs() []schema.TrainingJob {
 
 // CancelJob cancels a running training job
 func (s *TrainingJobService) CancelJob(id string) error {
-	job, ok := s.jobs.Get(id)
-	if !ok {
+	if !s.jobs.Exists(id) {
 		return fmt.Errorf("job not found: %s", id)
 	}
+	job := s.jobs.Get(id)
 
 	if job.Status != schema.TrainingJobStatusPending && job.Status != schema.TrainingJobStatusRunning {
 		return fmt.Errorf("job %s is not cancellable (status: %s)", id, job.Status)
 	}
 
-	cancelFn, ok := s.cancellations.Get(id)
-	if ok {
+	if s.cancellations.Exists(id) {
+		cancelFn := s.cancellations.Get(id)
 		cancelFn()
 	}
 
@@ -165,15 +168,15 @@ func (s *TrainingJobService) CancelJob(id string) error {
 
 // DeleteJob removes a job from the store
 func (s *TrainingJobService) DeleteJob(id string) error {
-	job, ok := s.jobs.Get(id)
-	if !ok {
+	if !s.jobs.Exists(id) {
 		return fmt.Errorf("job not found: %s", id)
 	}
+	job := s.jobs.Get(id)
 
 	// Cancel if still running
 	if job.Status == schema.TrainingJobStatusRunning || job.Status == schema.TrainingJobStatusPending {
-		if cancelFn, ok := s.cancellations.Get(id); ok {
-			cancelFn()
+		if s.cancellations.Exists(id) {
+			s.cancellations.Get(id)()
 		}
 	}
 
@@ -185,10 +188,10 @@ func (s *TrainingJobService) DeleteJob(id string) error {
 
 // executeJob runs the training job via the backend gRPC call
 func (s *TrainingJobService) executeJob(id string) {
-	job, ok := s.jobs.Get(id)
-	if !ok {
+	if !s.jobs.Exists(id) {
 		return
 	}
+	job := s.jobs.Get(id)
 
 	ctx, cancel := context.WithCancel(s.ctx)
 	s.cancellations.Set(id, cancel)
@@ -229,29 +232,21 @@ func (s *TrainingJobService) executeJob(id string) {
 
 	modelConfig := config.ModelConfig{
 		Name:    job.Model,
-		Model:   job.Model,
 		Backend: backendName,
 	}
 
 	// Attempt to load from existing model configs
-	if cfgs := s.configLoader.ListConfigs(); len(cfgs) > 0 {
-		for _, name := range cfgs {
-			if name == job.Model {
-				if cfg, exists := s.configLoader.GetConfig(name); exists {
-					modelConfig = cfg
-					if backendName != "" {
-						modelConfig.Backend = backendName
-					}
-				}
-				break
-			}
+	if cfg, exists := s.configLoader.GetModelConfig(job.Model); exists {
+		modelConfig = cfg
+		if backendName != "" {
+			modelConfig.Backend = backendName
 		}
 	}
 
 	// Load backend and call TrainStream directly (avoids import cycle with core/backend)
 	opts := []model.Option{
 		model.WithBackendString(modelConfig.Backend),
-		model.WithModel(modelConfig.Model),
+		model.WithModel(modelConfig.Name),
 		model.WithContext(s.appConfig.Context),
 		model.WithModelID(modelConfig.Name),
 	}
@@ -267,7 +262,7 @@ func (s *TrainingJobService) executeJob(id string) {
 	}
 
 	// Final status update
-	job, _ = s.jobs.Get(id)
+	job = s.jobs.Get(id)
 	completedAt := time.Now()
 
 	if err != nil {
@@ -296,10 +291,10 @@ func (s *TrainingJobService) executeJob(id string) {
 
 // updateJobFromResponse updates job state from a streaming TrainResponse
 func (s *TrainingJobService) updateJobFromResponse(id string, resp *proto.TrainResponse) {
-	job, ok := s.jobs.Get(id)
-	if !ok {
+	if !s.jobs.Exists(id) {
 		return
 	}
+	job := s.jobs.Get(id)
 
 	job.Progress = resp.Progress
 	job.CurrentEpoch = resp.CurrentEpoch