postgres-ai
diff --git a/‎internal/retrieval/retrieval.go
Copy file name to clipboardExpand all lines: internal/retrieval/retrieval.go
+16-10Lines changed: 16 additions & 10 deletions b/‎internal/retrieval/retrieval.go
Copy file name to clipboardExpand all lines: internal/retrieval/retrieval.go
+16-10Lines changed: 16 additions & 10 deletions
@@ -80,16 +80,7 @@ func New(cfg *dblabCfg.Config, engineProps global.EngineProps, docker *client.Cl
 		},
 	}
 
-	for _, jobName := range r.cfg.Jobs {
-		jobSpec, ok := r.cfg.JobsSpec[jobName]
-		if !ok {
-			continue
-		}
-
-		jobSpec.Name = jobName
-		r.jobSpecs[jobName] = jobSpec
-	}
-
+	r.formatJobsSpec()
 	r.defineRetrievalMode()
 
 	return r
@@ -99,6 +90,8 @@ func New(cfg *dblabCfg.Config, engineProps global.EngineProps, docker *client.Cl
 func (r *Retrieval) Reload(ctx context.Context, cfg *dblabCfg.Config) {
 	*r.cfg = cfg.Retrieval
 
+	r.formatJobsSpec()
+
 	for _, job := range r.jobs {
 		cfg, ok := r.cfg.JobsSpec[job.Name()]
 		if !ok {
@@ -111,9 +104,22 @@ func (r *Retrieval) Reload(ctx context.Context, cfg *dblabCfg.Config) {
 		}
 	}
 
+	r.stopScheduler()
 	r.setupScheduler(ctx)
 }
 
+func (r *Retrieval) formatJobsSpec() {
+	for _, jobName := range r.cfg.Jobs {
+		jobSpec, ok := r.cfg.JobsSpec[jobName]
+		if !ok {
+			continue
+		}
+
+		jobSpec.Name = jobName
+		r.jobSpecs[jobName] = jobSpec
+	}
+}
+
 // Run start retrieving process.
 func (r *Retrieval) Run(ctx context.Context) error {
 	runCtx, cancel := context.WithCancel(ctx)