fix agent panic in RunStep

Signed-off-by: Patrick Zhao <[email protected]>
koderover · Jan 21, 2025 · b088d67 · b088d67
1 parent feabc99
commit b088d67
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/cli/zadig-agent/internal/agent/step/step.go b/pkg/cli/zadig-agent/internal/agent/step/step.go
@@ -18,6 +18,7 @@ package step
 
 import (
 	"context"
+	"errors"
 	"fmt"
 
 	"github.com/koderover/zadig/v2/pkg/cli/zadig-agent/helper/log"
@@ -48,6 +49,11 @@ func RunStep(ctx context.Context, jobCtx *jobctl.JobContext, step *commonmodels.
 	var stepInstance Step
 	var err error
 
+	if step == nil {
+		log.Errorf("step is nil")
+		return errors.New("step is nil")
+	}
+
 	switch step.StepType {
 	case "batch_file":
 		stepInstance, err = script.NewBatchFileStep(jobCtx.Outputs, step.Spec, dirs, envs, secretEnvs, logger)